blob: 5a2246c73694ecfa1978454e7020cc852f4570f3 [file] [log] [blame]
Javi Merino491cf732014-03-31 17:34:44 +01001#!/usr/bin/python
Javi Merino5bd3d442014-04-08 12:55:13 +01002"""Process the output of the power allocator trace in the current
3directory's trace.dat"""
Javi Merino572049d2014-03-31 16:45:23 +01004
5import os
Javi Merinoee56c362014-03-31 17:30:34 +01006import re
Javi Merino952815a2014-03-31 18:08:32 +01007from StringIO import StringIO
Javi Merinof78ea5b2014-03-31 17:51:38 +01008import pandas as pd
Javi Merinoa6399fb2014-03-31 19:17:08 +01009from matplotlib import pyplot as plt
Javi Merino572049d2014-03-31 16:45:23 +010010
Javi Merino7686f362014-04-02 11:26:40 +010011GOLDEN_RATIO = 1.618034
12
Javi Merinoc2ec5682014-04-01 15:16:06 +010013class BaseThermal(object):
Javi Merino5bd3d442014-04-08 12:55:13 +010014 """Base class to parse trace.dat dumps.
15
16 Don't use directly, create a subclass that defines the unique_word
17 you want to match in the output"""
Javi Merino68461552014-04-08 11:40:09 +010018 def __init__(self, basepath, unique_word):
19 if basepath is None:
20 basepath = "."
Javi Merinoc2ec5682014-04-01 15:16:06 +010021
Javi Merino68461552014-04-08 11:40:09 +010022 self.basepath = basepath
Javi Merino952815a2014-03-31 18:08:32 +010023 self.data_csv = ""
Javi Merinof78ea5b2014-03-31 17:51:38 +010024 self.data_frame = False
Javi Merino68461552014-04-08 11:40:09 +010025 self.unique_word = unique_word
26
27 if not os.path.isfile(os.path.join(basepath, "trace.txt")):
28 self.__run_trace_cmd_report()
Javi Merino572049d2014-03-31 16:45:23 +010029
Javi Merino572049d2014-03-31 16:45:23 +010030 def __run_trace_cmd_report(self):
Javi Merino5bd3d442014-04-08 12:55:13 +010031 """Run "trace-cmd report > trace.txt".
32
33 Overwrites the contents of trace.txt if it exists."""
Javi Merinoee56c362014-03-31 17:30:34 +010034 from subprocess import check_output
35
Javi Merino68461552014-04-08 11:40:09 +010036 if not os.path.isfile(os.path.join(self.basepath, "trace.dat")):
Javi Merino1a3725a2014-03-31 18:35:15 +010037 raise IOError("No such file or directory: trace.dat")
38
Javi Merino68461552014-04-08 11:40:09 +010039 previous_path = os.getcwd()
40 os.chdir(self.basepath)
Javi Merino572049d2014-03-31 16:45:23 +010041
Javi Merino68461552014-04-08 11:40:09 +010042 # This would better be done with a context manager (i.e.
43 # http://stackoverflow.com/a/13197763/970766)
44 try:
45 with open(os.devnull) as devnull:
46 out = check_output(["trace-cmd", "report"], stderr=devnull)
47
48 finally:
49 os.chdir(previous_path)
50
Javi Merino5bd3d442014-04-08 12:55:13 +010051 with open(os.path.join(self.basepath, "trace.txt"), "w") as fout:
52 fout.write(out)
Javi Merinoc08ca682014-03-31 17:29:27 +010053
Javi Merinoc2ec5682014-04-01 15:16:06 +010054 def parse_into_csv(self):
Javi Merino5bd3d442014-04-08 12:55:13 +010055 """Create a csv representation of the thermal data and store
56 it in self.data_csv"""
Javi Merinoc08ca682014-03-31 17:29:27 +010057 pat_timestamp = re.compile(r"([0-9]+\.[0-9]+):")
Javi Merinoe284d632014-04-02 18:57:06 +010058 pat_data = re.compile(r"[A-Za-z0-9_]+=([-a-f0-9]+)")
59 pat_header = re.compile(r"([A-Za-z0-9_]+)=[-a-f0-9]+")
Javi Merinoc08ca682014-03-31 17:29:27 +010060 header = ""
61
Javi Merino68461552014-04-08 11:40:09 +010062 with open(os.path.join(self.basepath, "trace.txt")) as fin:
Javi Merino952815a2014-03-31 18:08:32 +010063 for line in fin:
Javi Merinoc2ec5682014-04-01 15:16:06 +010064 if not re.search(self.unique_word, line):
Javi Merino952815a2014-03-31 18:08:32 +010065 continue
66
67 line = line[:-1]
68
Javi Merino5bd3d442014-04-08 12:55:13 +010069 timestamp_match = re.search(pat_timestamp, line)
70 timestamp = timestamp_match.group(1)
Javi Merino952815a2014-03-31 18:08:32 +010071
Javi Merinoc2ec5682014-04-01 15:16:06 +010072 data_start_idx = re.search(r"[A-Za-z0-9_]+=", line).start()
73 data_str = line[data_start_idx:]
Javi Merino952815a2014-03-31 18:08:32 +010074
75 if not header:
Javi Merino0af47212014-04-02 16:23:23 +010076 header = re.sub(pat_header, r"\1", data_str)
77 header = re.sub(r" ", r",", header)
Javi Merino952815a2014-03-31 18:08:32 +010078 header = "time," + header + "\n"
79 self.data_csv = header
80
Javi Merino0af47212014-04-02 16:23:23 +010081 parsed_data = re.sub(pat_data, r"\1", data_str)
82 parsed_data = re.sub(r" ", r",", parsed_data)
Javi Merino952815a2014-03-31 18:08:32 +010083
84 parsed_data = timestamp + "," + parsed_data + "\n"
85 self.data_csv += parsed_data
86
Javi Merinof78ea5b2014-03-31 17:51:38 +010087 def get_data_frame(self):
88 """Return a pandas data frame for the run"""
89 if self.data_frame:
90 return self.data_frame
91
Javi Merino952815a2014-03-31 18:08:32 +010092 if not self.data_csv:
Javi Merinoc2ec5682014-04-01 15:16:06 +010093 self.parse_into_csv()
Javi Merinof78ea5b2014-03-31 17:51:38 +010094
Javi Merino04f27492014-04-02 09:59:23 +010095 try:
Javi Merino5bd3d442014-04-08 12:55:13 +010096 unordered_df = pd.read_csv(StringIO(self.data_csv))
97 self.data_frame = unordered_df.set_index("time")
Javi Merino04f27492014-04-02 09:59:23 +010098 except StopIteration:
99 if not self.data_frame:
100 return pd.DataFrame()
101 raise
102
Javi Merinof78ea5b2014-03-31 17:51:38 +0100103 return self.data_frame
Javi Merinodf8316a2014-03-31 18:39:42 +0100104
Javi Merinofb2e8fd2014-04-08 12:27:38 +0100105def set_plot_size(width, height):
106 """Set the plot size.
107
108 This has to be called before calls to .plot()
109 """
110 if height is None:
111 if width is None:
112 height = 6
113 width = 10
114 else:
115 height = width / GOLDEN_RATIO
116 else:
117 if width is None:
118 width = height * GOLDEN_RATIO
119
120 plt.figure(figsize=(width, height))
121
Javi Merino05983ef2014-04-08 12:54:20 +0100122def default_plot_settings(title=""):
123 """Set xlabel and title of the plot
124
125 This has to be called after calls to .plot()
126 """
127
128 plt.xlabel("Time")
129 if title:
130 plt.title(title)
131
Javi Merinoc2ec5682014-04-01 15:16:06 +0100132class Thermal(BaseThermal):
Javi Merino5bd3d442014-04-08 12:55:13 +0100133 """Process the power allocator data in a ftrace dump"""
Javi Merino68461552014-04-08 11:40:09 +0100134 def __init__(self, path=None):
Javi Merinoc2ec5682014-04-01 15:16:06 +0100135 super(Thermal, self).__init__(
Javi Merino68461552014-04-08 11:40:09 +0100136 basepath=path,
137 unique_word="Ptot_out",
Javi Merinoc2ec5682014-04-01 15:16:06 +0100138 )
139
140 def write_thermal_csv(self):
141 """Write the csv info in thermal.csv"""
142 if not self.data_csv:
143 self.parse_into_csv()
144
145 with open("thermal.csv", "w") as fout:
146 fout.write(self.data_csv)
147
Javi Merino7686f362014-04-02 11:26:40 +0100148 def plot_temperature(self, width=None, height=None):
Javi Merinodf8316a2014-03-31 18:39:42 +0100149 """Plot the temperature"""
Javi Merino5bd3d442014-04-08 12:55:13 +0100150 dfr = self.get_data_frame()
Javi Merino9e186092014-04-08 15:41:41 +0100151 control_temp_series = (dfr["currT"] + dfr["deltaT"]) / 1000
152
153 set_plot_size(width, height)
154
Javi Merino5bd3d442014-04-08 12:55:13 +0100155 (dfr["currT"] / 1000).plot()
Javi Merino9e186092014-04-08 15:41:41 +0100156 control_temp_series.plot(color="y", linestyle="--",
157 label="control temperature")
158
Javi Merino05983ef2014-04-08 12:54:20 +0100159 default_plot_settings(title="Temperature")
Javi Merino9e186092014-04-08 15:41:41 +0100160 plt.legend()
Javi Merino749b26a2014-03-31 19:17:30 +0100161
Javi Merinof1dcb2d2014-04-02 16:52:11 +0100162 def plot_multivalue(self, values, title, width, height):
163 """Plot multiple values of the DataFrame
164
165 values is an array with the keys of the DataFrame to plot
166 """
Javi Merinofb2e8fd2014-04-08 12:27:38 +0100167 set_plot_size(width, height)
Javi Merino5bd3d442014-04-08 12:55:13 +0100168 dfr = self.get_data_frame()
169 dfr[values].plot()
Javi Merino05983ef2014-04-08 12:54:20 +0100170 default_plot_settings(title=title)
Javi Merinof1dcb2d2014-04-02 16:52:11 +0100171
172 def plot_input_power(self, width=None, height=None):
173 """Plot input power"""
Javi Merino5bd3d442014-04-08 12:55:13 +0100174 self.plot_multivalue(["Pa7_in", "Pa15_in", "Pgpu_in"], "Input Power",
175 width, height)
Javi Merino9c010772014-04-02 16:54:41 +0100176
177 def plot_output_power(self, width=None, height=None):
178 """Plot output power"""
Javi Merino5bd3d442014-04-08 12:55:13 +0100179 self.plot_multivalue(["Pa7_out", "Pa15_out", "Pgpu_out"],
180 "Output Power", width, height)