From b75849f8af72ae16ef72ac0e30edda9bd8f822df Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Tue, 26 Feb 2019 18:07:59 +0400
Subject: [PATCH] Save extra ztplot with more than 4 unique DOMs

---
 scripts/ztplot.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/scripts/ztplot.py b/scripts/ztplot.py
index 6851ed6..e7ec5c5 100755
--- a/scripts/ztplot.py
+++ b/scripts/ztplot.py
@@ -120,6 +120,7 @@ class ZTPlot(Module):
     def create_plot(self, event_info, hits):
         print(self.__class__.__name__ + ": updating plot.")
         dus = set(hits.du)
+        doms = set(hits.dom_id)
         fontsize = 16
 
         time_offset = np.min(hits[hits.triggered == True].time)
@@ -186,6 +187,8 @@ class ZTPlot(Module):
         f = os.path.join(self.plots_path, filename + '.png')
         f_tmp = os.path.join(self.plots_path, filename + '_tmp.png')
         plt.savefig(f_tmp, dpi=120, bbox_inches="tight")
+        if len(doms) > 4:
+            plt.savefig(os.path.join(self.plots_path, filename + '_5doms.png'))
         plt.close('all')
         shutil.move(f_tmp, f)
 
-- 
GitLab