aboutsummaryrefslogtreecommitdiffstats
path: root/correlate_with_ref.py
diff options
context:
space:
mode:
authorMatthias P. Braendli <matthias.braendli@mpb.li>2016-07-25 20:22:55 +0200
committerMatthias P. Braendli <matthias.braendli@mpb.li>2016-07-25 20:22:55 +0200
commitf29396c9f814bdd7d35bdc8699530689235c7090 (patch)
treea1601d4a7e286d9a2a6fd3b97e929b96158f5479 /correlate_with_ref.py
parent6b13f6da09bef4ec8137c70c6c18d07b6a884bd5 (diff)
downloadodr-dab-cir-f29396c9f814bdd7d35bdc8699530689235c7090.tar.gz
odr-dab-cir-f29396c9f814bdd7d35bdc8699530689235c7090.tar.bz2
odr-dab-cir-f29396c9f814bdd7d35bdc8699530689235c7090.zip
Fix memory leak related to matplotlib
Diffstat (limited to 'correlate_with_ref.py')
-rwxr-xr-xcorrelate_with_ref.py14
1 files changed, 10 insertions, 4 deletions
diff --git a/correlate_with_ref.py b/correlate_with_ref.py
index ae783a6..997081b 100755
--- a/correlate_with_ref.py
+++ b/correlate_with_ref.py
@@ -17,10 +17,11 @@ import matplotlib
# matplotlib backend won't work. In case we are running as a module by cir_measure,
# switch to the Agg backend
# See http://matplotlib.org/faq/howto_faq.html#matplotlib-in-a-web-application-server
+# And http://matplotlib.org/examples/api/agg_oo.html#api-agg-oo
if __name__ != "__main__":
- matplotlib.use("Agg")
+ from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
-import matplotlib.pyplot as pp
+import matplotlib.figure
import sys
# T = 1/2048000 s
@@ -117,7 +118,12 @@ class CIR_Correlate:
self.calc_one_cir_(i * T_TF)
for i in range(num_correlations) ])
- fig = pp.figure()
+ if plot_file:
+ fig = matplotlib.figure.Figure()
+ canvas = FigureCanvas(fig)
+ else:
+ fig = matplotlib.figure()
+
fig.suptitle(title)
ax1 = fig.add_subplot(211)
ax1.plot(cirs.sum(axis=0))
@@ -126,7 +132,7 @@ class CIR_Correlate:
if plot_file:
print("Save to file {}".format(plot_file))
- fig.savefig(plot_file)
+ canvas.print_figure(plot_file)
else:
print("Plotting to screen")
fig.show()