From 1bb5e3a8d89f12224134b106d1f3b2d196574f97 Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Sat, 23 Nov 2019 00:48:00 +0100
Subject: [PATCH] Table prototype
---
scripts/ztplot.py | 57 ++++++++++++++++++++++++++---------------------
1 file changed, 31 insertions(+), 26 deletions(-)
diff --git a/scripts/ztplot.py b/scripts/ztplot.py
index 51bd36d..ac2843f 100755
--- a/scripts/ztplot.py
+++ b/scripts/ztplot.py
@@ -33,6 +33,7 @@ import numpy as np
import km3pipe as kp
from km3pipe.io.daq import is_3dmuon, is_3dshower, is_mxshower
+from km3modules.common import LocalDBService
from km3modules.hits import count_multiplicities
from km3modules.plot import ztplot
import km3pipe.style
@@ -58,6 +59,17 @@ class ZTPlot(kp.Module):
self.index = 0
+ if not self.services["LocalDBService"].table_exists("nice_events"):
+ self.services["LocalDBService"].create_table(
+ "nice_events", [
+ "overlays", "n_hits", "n_triggered_hits", "n_dus",
+ "filename", "run_id", "det_id", "frame_index",
+ "trigger_counter", "utc_timestamp"
+ ], [
+ "INT", "INT", "INT", "INT", "TEXT", "INT", "INT", "INT",
+ "INT", "INT"
+ ])
+
self._update_calibration()
self.run = True
@@ -87,10 +99,8 @@ class ZTPlot(kp.Module):
n_triggered_dus = len(np.unique(hits[hits.triggered == True].du))
n_triggered_doms = len(np.unique(hits[hits.triggered == True].dom_id))
if n_triggered_dus < self.min_dus or n_triggered_doms < self.min_doms:
- print(
- f"Skipping event with {n_triggered_dus} DUs "
- f"and {n_triggered_doms} DOMs."
- )
+ print(f"Skipping event with {n_triggered_dus} DUs "
+ f"and {n_triggered_doms} DOMs.")
return blob
print("OK")
@@ -120,9 +130,9 @@ class ZTPlot(kp.Module):
& (self.calib.detector.pmts.channel_id == 0)]
trigger_params = ' '.join([
- trig for trig, trig_check in (("MX",
- is_mxshower), ("3DM", is_3dmuon),
- ("3DS", is_3dshower))
+ trig
+ for trig, trig_check in (("MX", is_mxshower), ("3DM", is_3dmuon),
+ ("3DS", is_3dshower))
if trig_check(int(event_info.trigger_mask[0]))
])
@@ -133,17 +143,13 @@ class ZTPlot(kp.Module):
event_info.det_id[0], self.t0set, event_info.run_id[0],
event_info.frame_index[0], event_info.trigger_counter[0],
event_info.overlays[0], trigger_params,
- datetime.utcfromtimestamp(event_info.utc_seconds)
- )
- )[0]
-
- fig = ztplot(
- hits,
- title,
- max_z=self.max_z,
- ytick_distance=self.ytick_distance,
- grid_lines=grid_lines
- )
+ datetime.utcfromtimestamp(event_info.utc_seconds)))[0]
+
+ fig = ztplot(hits,
+ title,
+ max_z=self.max_z,
+ ytick_distance=self.ytick_distance,
+ grid_lines=grid_lines)
filename = 'ztplot'
f = os.path.join(self.plots_path, filename + '.png')
@@ -168,14 +174,13 @@ def main():
ligier_port = int(args['-p'])
pipe = kp.Pipeline()
- pipe.attach(
- kp.io.ch.CHPump,
- host=ligier_ip,
- port=ligier_port,
- tags='IO_EVT, IO_SUM',
- timeout=60 * 60 * 24 * 7,
- max_queue=2000
- )
+ pipe.attach(LocalDBService)
+ pipe.attach(kp.io.ch.CHPump,
+ host=ligier_ip,
+ port=ligier_port,
+ tags='IO_EVT, IO_SUM',
+ timeout=60 * 60 * 24 * 7,
+ max_queue=2000)
pipe.attach(kp.io.daq.DAQProcessor)
pipe.attach(ZTPlot, det_id=det_id, plots_path=plots_path)
pipe.drain()
--
GitLab