Javi Merino | 491cf73 | 2014-03-31 17:34:44 +0100 | [diff] [blame] | 1 | #!/usr/bin/python |
Javi Merino | 572049d | 2014-03-31 16:45:23 +0100 | [diff] [blame] | 2 | """Process the output of the power allocator trace in the current directory's trace.dat""" |
| 3 | |
| 4 | import os |
Javi Merino | ee56c36 | 2014-03-31 17:30:34 +0100 | [diff] [blame] | 5 | import re |
Javi Merino | 952815a | 2014-03-31 18:08:32 +0100 | [diff] [blame] | 6 | from StringIO import StringIO |
Javi Merino | f78ea5b | 2014-03-31 17:51:38 +0100 | [diff] [blame] | 7 | import pandas as pd |
Javi Merino | a6399fb | 2014-03-31 19:17:08 +0100 | [diff] [blame^] | 8 | from matplotlib import pyplot as plt |
Javi Merino | 572049d | 2014-03-31 16:45:23 +0100 | [diff] [blame] | 9 | |
| 10 | class Thermal(object): |
| 11 | def __init__(self): |
| 12 | if not os.path.isfile("trace.txt"): |
| 13 | self.__run_trace_cmd_report() |
Javi Merino | 952815a | 2014-03-31 18:08:32 +0100 | [diff] [blame] | 14 | self.data_csv = "" |
Javi Merino | f78ea5b | 2014-03-31 17:51:38 +0100 | [diff] [blame] | 15 | self.data_frame = False |
Javi Merino | 572049d | 2014-03-31 16:45:23 +0100 | [diff] [blame] | 16 | |
| 17 | |
| 18 | def __run_trace_cmd_report(self): |
| 19 | """Run "trace-cmd report > trace.txt". Overwrites the contents of trace.txt if it exists.""" |
Javi Merino | ee56c36 | 2014-03-31 17:30:34 +0100 | [diff] [blame] | 20 | from subprocess import check_output |
| 21 | |
Javi Merino | 1a3725a | 2014-03-31 18:35:15 +0100 | [diff] [blame] | 22 | if not os.path.isfile("trace.dat"): |
| 23 | raise IOError("No such file or directory: trace.dat") |
| 24 | |
Javi Merino | 572049d | 2014-03-31 16:45:23 +0100 | [diff] [blame] | 25 | with open(os.devnull) as devnull: |
Javi Merino | ee56c36 | 2014-03-31 17:30:34 +0100 | [diff] [blame] | 26 | out = check_output(["trace-cmd", "report"], stderr=devnull) |
Javi Merino | 572049d | 2014-03-31 16:45:23 +0100 | [diff] [blame] | 27 | |
| 28 | with open("trace.txt", "w") as f: |
| 29 | f.write(out) |
Javi Merino | c08ca68 | 2014-03-31 17:29:27 +0100 | [diff] [blame] | 30 | |
Javi Merino | 952815a | 2014-03-31 18:08:32 +0100 | [diff] [blame] | 31 | def __parse_into_csv(self): |
| 32 | """Create a csv representation of the thermal data and store it in self.data_csv""" |
Javi Merino | c08ca68 | 2014-03-31 17:29:27 +0100 | [diff] [blame] | 33 | pat_timestamp = re.compile(r"([0-9]+\.[0-9]+):") |
| 34 | pat_data = re.compile(r"[A-Za-z0-9_]+=([0-9]+) ") |
| 35 | header = "" |
| 36 | |
Javi Merino | 952815a | 2014-03-31 18:08:32 +0100 | [diff] [blame] | 37 | with open("trace.txt") as fin: |
| 38 | for line in fin: |
| 39 | if not re.search("Ptot_out", line): |
| 40 | continue |
| 41 | |
| 42 | line = line[:-1] |
| 43 | |
| 44 | m = re.search(pat_timestamp, line) |
| 45 | timestamp = m.group(1) |
| 46 | |
| 47 | semi_idx = line.index(" : ") |
| 48 | data_str = line[semi_idx + 3:] |
| 49 | |
| 50 | if not header: |
| 51 | header = re.sub(r"([A-Za-z0-9_]+)=[0-9]+ ", r"\1,", data_str) |
| 52 | header = header[:-1] |
| 53 | header = "time," + header + "\n" |
| 54 | self.data_csv = header |
| 55 | |
| 56 | parsed_data = re.sub(pat_data, r"\1,", data_str) |
| 57 | # Drop the last comma |
| 58 | parsed_data = parsed_data[:-1] |
| 59 | |
| 60 | parsed_data = timestamp + "," + parsed_data + "\n" |
| 61 | self.data_csv += parsed_data |
| 62 | |
| 63 | def write_thermal_csv(self): |
| 64 | """Write the csv info in thermal.csv""" |
| 65 | if not self.data_csv: |
| 66 | self.__parse_into_csv() |
| 67 | |
Javi Merino | c08ca68 | 2014-03-31 17:29:27 +0100 | [diff] [blame] | 68 | with open("thermal.csv", "w") as fout: |
Javi Merino | 952815a | 2014-03-31 18:08:32 +0100 | [diff] [blame] | 69 | fout.write(self.data_csv) |
Javi Merino | f78ea5b | 2014-03-31 17:51:38 +0100 | [diff] [blame] | 70 | |
| 71 | def get_data_frame(self): |
| 72 | """Return a pandas data frame for the run""" |
| 73 | if self.data_frame: |
| 74 | return self.data_frame |
| 75 | |
Javi Merino | 952815a | 2014-03-31 18:08:32 +0100 | [diff] [blame] | 76 | if not self.data_csv: |
| 77 | self.__parse_into_csv() |
Javi Merino | f78ea5b | 2014-03-31 17:51:38 +0100 | [diff] [blame] | 78 | |
Javi Merino | 952815a | 2014-03-31 18:08:32 +0100 | [diff] [blame] | 79 | self.data_frame = pd.read_csv(StringIO(self.data_csv)).set_index("time") |
Javi Merino | f78ea5b | 2014-03-31 17:51:38 +0100 | [diff] [blame] | 80 | return self.data_frame |
Javi Merino | df8316a | 2014-03-31 18:39:42 +0100 | [diff] [blame] | 81 | |
Javi Merino | a6399fb | 2014-03-31 19:17:08 +0100 | [diff] [blame^] | 82 | def __default_plot_settings(self, title=""): |
| 83 | plt.xlabel("Time") |
| 84 | if title: |
| 85 | plt.title(title) |
| 86 | |
Javi Merino | df8316a | 2014-03-31 18:39:42 +0100 | [diff] [blame] | 87 | def plot_temperature(self): |
| 88 | """Plot the temperature""" |
| 89 | df = self.get_data_frame() |
| 90 | (df["currT"] / 1000).plot() |
Javi Merino | a6399fb | 2014-03-31 19:17:08 +0100 | [diff] [blame^] | 91 | self.__default_plot_settings(title="Temperature") |