From 477a403134cc4626aec20038bcafaa6aeb78853a Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Sat, 23 Nov 2019 08:44:20 +0100
Subject: [PATCH] Close plots to cleanup memory
---
scripts/ztplot.py | 15 +++++++++------
1 file changed, 9 insertions(+), 6 deletions(-)
diff --git a/scripts/ztplot.py b/scripts/ztplot.py
index 01ce123..fdd98ca 100755
--- a/scripts/ztplot.py
+++ b/scripts/ztplot.py
@@ -179,12 +179,12 @@ class ZTPlot(kp.Module):
f = os.path.join(self.plots_path, filename + '.png')
f_tmp = os.path.join(self.plots_path, filename + '_tmp.png')
- ztplot(hits,
- filename=f_tmp,
- title=title,
- max_z=self.max_z,
- ytick_distance=self.ytick_distance,
- grid_lines=grid_lines)
+ fig = ztplot(hits,
+ filename=f_tmp,
+ title=title,
+ max_z=self.max_z,
+ ytick_distance=self.ytick_distance,
+ grid_lines=grid_lines)
shutil.move(f_tmp, f)
if overlays > self.records['overlays'] or n_hits > self.records[
@@ -215,6 +215,9 @@ class ZTPlot(kp.Module):
])
shutil.copy(f, plot_filename)
+ plt.close(fig)
+ plt.close('all')
+
def finish(self):
self.run = False
--
GitLab