thermal: make plot_input_power accept an ax argument
diff --git a/cr2/thermal.py b/cr2/thermal.py
index 0f3fc1d..c55145b 100644
--- a/cr2/thermal.py
+++ b/cr2/thermal.py
@@ -244,7 +244,7 @@
with open("thermal.csv", "w") as fout:
fout.write(self.data_csv)
- def plot_input_power(self, actor_order, title="", width=None, height=None):
+ def plot_input_power(self, actor_order, title="", width=None, height=None, ax=None):
"""Plot input power
actor_order is an array with the order in which the actors were registered.
@@ -262,7 +262,9 @@
title = normalize_title("Input 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)