From aa59719a33a6df0334b82898b1de950b35cb822f Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Sat, 12 Jan 2019 21:20:07 +0100
Subject: [PATCH] Determine n_doms and n_dus from the detector file
---
scripts/live_triggermap.py | 27 +++++++++++++--------------
1 file changed, 13 insertions(+), 14 deletions(-)
diff --git a/scripts/live_triggermap.py b/scripts/live_triggermap.py
index 58116ed..7adaac2 100755
--- a/scripts/live_triggermap.py
+++ b/scripts/live_triggermap.py
@@ -50,9 +50,6 @@ from km3pipe.logger import logging
# print("Setting log level to debug for '{0}'".format(logger_name))
# logger.setLevel("DEBUG")
-N_DOMS = 18
-N_DUS = 2
-
# xfmt = md.DateFormatter('%Y-%m-%d %H:%M')
lock = threading.Lock()
@@ -61,7 +58,7 @@ class DOMHits(Module):
def configure(self):
self.plots_path = self.require('plots_path')
det_id = self.require('det_id')
- self.detector = kp.hardware.Detector(det_id=det_id)
+ self.det = kp.hardware.Detector(det_id=det_id)
self.run = True
self.max_events = 1000
@@ -77,15 +74,16 @@ class DOMHits(Module):
event_hits = blob['Hits']
with lock:
- hits = np.zeros(N_DOMS * N_DUS)
+ hits = np.zeros(self.det.n_doms * self.det.n_dus)
for dom_id in event_hits.dom_id:
- du, floor, _ = self.detector.doms[dom_id]
- hits[(du - 1) * N_DOMS + floor - 1] += 1
+ du, floor, _ = self.det.doms[dom_id]
+ hits[(du - 1) * self.detecotor.n_doms + floor - 1] += 1
self.hits.append(hits)
- triggered_hits = np.zeros(N_DOMS * N_DUS)
- for dom_id in event_hits.dom_id[event_hits.triggered.astype('bool')]:
- du, floor, _ = self.detector.doms[dom_id]
- triggered_hits[(du - 1) * N_DOMS + floor - 1] += 1
+ triggered_hits = np.zeros(self.det.n_doms * self.det.n_dus)
+ for dom_id in event_hits.dom_id[event_hits.triggered.astype(
+ 'bool')]:
+ du, floor, _ = self.det.doms[dom_id]
+ triggered_hits[(du - 1) * self.det.n_doms + floor - 1] += 1
self.triggered_hits.append(triggered_hits)
return blob
@@ -116,9 +114,10 @@ class DOMHits(Module):
origin='lower',
zorder=3,
norm=LogNorm(vmin=1, vmax=np.amax(hit_matrix)))
- yticks = np.arange(N_DOMS * N_DUS)
+ yticks = np.arange(self.det.n_doms * self.det.n_dus)
ytick_labels = ["DU{0:0.0f}-DOM{1:02d}"
- .format(np.ceil((y+1)/N_DOMS), y % (N_DOMS) + 1) \
+ .format(np.ceil((y+1)/self.det.n_doms),
+ y % (self.det.n_doms) + 1) \
for y in yticks]
ax.set_yticks(yticks)
ax.set_yticklabels(ytick_labels)
@@ -166,6 +165,6 @@ def main():
pipe.attach(DOMHits, det_id=det_id, plots_path=plots_path)
pipe.drain()
+
if __name__ == '__main__':
main()
-
--
GitLab