commit 6003b25df67a75ce5d09c70d10a31eacb348701f
author Javi Merino <javi.merino@arm.com> Tue Aug 12 15:32:17 2014 +0100
committer Javi Merino <javi.merino@arm.com> Wed Aug 12 14:52:41 2015 +0100
tree ba95cf8d2d1beaf192cd9b68ff2b092d6f11659e
parent 77a29853ceb7d5a1ebac5fd5b1b31a7a3e4381b4

thermal: make plot_output_power() accept an ax argument

cr2/thermal.py

diff --git a/cr2/thermal.py b/cr2/thermal.py
index c55145b..9182e84 100644
--- a/cr2/thermal.py
+++ b/cr2/thermal.py
@@ -268,7 +268,7 @@
         plot_dfr.plot(ax=ax)
         post_plot_setup(ax, title=title)
 
-    def plot_output_power(self, actor_order, title="", width=None, height=None):
+    def plot_output_power(self, actor_order, title="", width=None, height=None, ax=None):
         """Plot output power
 
         actor_order is an array with the order in which the actors were registered.
@@ -283,7 +283,9 @@
 
         title = normalize_title("Output Power", title)
 
-        ax = pre_plot_setup(width, height)
+        if not ax:
+            ax = pre_plot_setup(width, height)
+
         plot_dfr.plot(ax=ax)
         post_plot_setup(ax, title=title)