analyze-arche: Add --plot-traces option

2 files changed, 24 insertions, 5 deletions

diff --git a/bin/analyze-archive.py b/bin/analyze-archive.py
index f7847c5..b254e76 100755
--- a/bin/analyze-archive.py
+++ b/bin/analyze-archive.py
@@ -21,6 +21,9 @@ Options:
     parameters. Also plots the corresponding measurements.
     If gplearn function is set, it is plotted using dashed lines.
 
+--plot-traces=<name>
+    Plot power trace for state or transition <name>.
+
 --export-traces=<directory>
     Export power traces of all states and transitions to <directory>.
     Creates a JSON file for each state and transition. Each JSON file
@@ -242,7 +245,7 @@ if __name__ == '__main__':
 
     try:
         optspec = (
-            'plot-unparam= plot-param= param-info show-models= show-quality= '
+            'plot-unparam= plot-param= plot-traces= param-info show-models= show-quality= '
             'ignored-trace-indexes= discard-outliers= function-override= '
             'export-traces= '
             'filter-param= '
@@ -293,7 +296,7 @@ if __name__ == '__main__':
         print(err)
         sys.exit(2)
 
-    raw_data = RawData(args, with_traces=('export-traces' in opts))
+    raw_data = RawData(args, with_traces=('export-traces' in opts or 'plot-traces' in opts))
 
     preprocessed_data = raw_data.get_preprocessed_data()
 
@@ -313,6 +316,14 @@ if __name__ == '__main__':
             with open(target, 'w') as f:
                 json.dump(data, f)
 
+    if 'plot-traces' in opts:
+        traces = list()
+        for trace in preprocessed_data:
+            for state_or_transition in trace['trace']:
+                if state_or_transition['name'] == opts['plot-traces']:
+                    traces.extend(map(lambda x: x['uW'], state_or_transition['offline']))
+        plotter.plot_y(traces, xlabel='t [1e-5 s]', ylabel='P [uW]', title=opts['plot-traces'], family=True)
+
     if raw_data.preprocessing_stats['num_valid'] == 0:
         print('No valid data available. Abort.')
         sys.exit(2)
diff --git a/lib/plotter.py b/lib/plotter.py
index b9d5c3e..deed93a 100755
--- a/lib/plotter.py
+++ b/lib/plotter.py
@@ -100,10 +100,13 @@ def plot_substate_thresholds_p(model, aggregate):
 
 
 def plot_y(Y, **kwargs):
-    plot_xy(np.arange(len(Y)), Y, **kwargs)
+    if 'family' in kwargs and kwargs['family']:
+        plot_xy(None, Y, **kwargs)
+    else:
+        plot_xy(np.arange(len(Y)), Y, **kwargs)
 
 
-def plot_xy(X, Y, xlabel=None, ylabel=None, title=None, output=None):
+def plot_xy(X, Y, xlabel=None, ylabel=None, title=None, output=None, family=False):
     fig, ax1 = plt.subplots(figsize=(10, 6))
     if title is not None:
         fig.canvas.set_window_title(title)
@@ -112,7 +115,12 @@ def plot_xy(X, Y, xlabel=None, ylabel=None, title=None, output=None):
     if ylabel is not None:
         ax1.set_ylabel(ylabel)
     plt.subplots_adjust(left=0.1, bottom=0.1, right=0.99, top=0.99)
-    plt.plot(X, Y, "bo", markersize=2)
+    if family:
+        cm = plt.get_cmap('brg', len(Y))
+        for i, YY in enumerate(Y):
+            plt.plot(np.arange(len(YY)), YY, "-", markersize=2, color=cm(i))
+    else:
+        plt.plot(X, Y, "bo", markersize=2)
     if output:
         plt.savefig(output)
         with open('{}.txt'.format(output), 'w') as f: