From 4bcda9015c674f0c77bab618f1f6f5a49af465a0 Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Tue, 3 Dec 2019 10:48:31 +0100
Subject: [PATCH] Write raw data for top10 selected hits
---
scripts/ztplot.py | 19 +++++++++++++------
1 file changed, 13 insertions(+), 6 deletions(-)
diff --git a/scripts/ztplot.py b/scripts/ztplot.py
index 2419042..d527818 100755
--- a/scripts/ztplot.py
+++ b/scripts/ztplot.py
@@ -123,7 +123,8 @@ class ZTPlot(kp.Module):
# print("Event queue size: {0}".format(self.queue.qsize()))
if self.queue.qsize() < self.max_queue:
- self.queue.put((event_info, hits))
+ raw_data = blob["CHData"]
+ self.queue.put((event_info, hits, raw_data))
else:
self.cprint("Skipping, queue is full...")
@@ -132,13 +133,13 @@ class ZTPlot(kp.Module):
def plot(self):
while self.run:
try:
- event_info, hits = self.queue.get(timeout=50)
+ event_info, hits, raw_data = self.queue.get(timeout=50)
except queue.Empty:
continue
with lock:
- self.create_plot(event_info, hits)
+ self.create_plot(event_info, hits, raw_data)
- def create_plot(self, event_info, hits):
+ def create_plot(self, event_info, hits, raw_data):
trigger_mask = event_info.trigger_mask[0]
det_id = event_info.det_id[0]
@@ -200,10 +201,12 @@ class ZTPlot(kp.Module):
"New record! Overlays: {}, hits: {}, triggered hits: {}".
format(overlays, n_hits, n_triggered_hits))
- plot_filename = os.path.join(
+ base_filename = os.path.join(
self.plots_path,
"event_selection/ztplot_{:08d}_{:08d}_FI{}_TC{}".format(
- det_id, run_id, frame_index, trigger_counter) + ".png")
+ det_id, run_id, frame_index, trigger_counter))
+ plot_filename = base_filename + ".png"
+ rawdata_filename = base_filename + ".dat"
self.services["insert_row"](self.event_selection_table, [
"overlays", "n_hits", "n_triggered_hits", "n_dus",
@@ -214,6 +217,10 @@ class ZTPlot(kp.Module):
run_id, det_id, frame_index, trigger_counter, utc_timestamp
])
shutil.copy(f, plot_filename)
+
+ with open(rawdata_filename, "wb") as fobj:
+ fobj.write(raw_data)
+
self._update_lower_limits()
self.services['post_elog'](
logbook=self.logbook,
--
GitLab