From 75fe3593dd487771962ba242ffaa82def21552e0 Mon Sep 17 00:00:00 2001
From: Giuliano Maggi <gmaggi@km3net.de>
Date: Sat, 18 May 2019 15:48:42 +0200
Subject: [PATCH] online_reco.py produces now gandalf_zenith.png and
gandalf_quality.png
---
scripts/online_reco.py | 75 ++++++++++++++++++++++++++----------------
1 file changed, 47 insertions(+), 28 deletions(-)
diff --git a/scripts/online_reco.py b/scripts/online_reco.py
index cf8144a..b3b6aca 100644
--- a/scripts/online_reco.py
+++ b/scripts/online_reco.py
@@ -32,62 +32,81 @@ import km3pipe.style
km3pipe.style.use('km3pipe')
-class ZenithDistribution(kp.Module):
+class IO_OLINEDistributions(kp.Module):
def configure(self):
+ self.fontsize = 16
+
self.plots_path = self.require('plots_path')
self.max_events = self.get('max_events', default=5000)
- self.zeniths = deque(maxlen=self.max_events)
- self.interval = 60
+ self.zeniths = deque(maxlen=self.max_events)
+ self.qualities = deque(maxlen=self.max_events)
+ self.interval = 60
threading.Thread(target=self.plot).start()
def process(self, blob):
track = blob['RecoTrack']
- zenith = np.cos(
- kp.math.angle_between([0, 0, -1], [track.dx, track.dy, track.dz]))
- self.zeniths.append(zenith)
+
+ if track.status==1:
+ zenith = np.cos(
+ kp.math.angle_between([0, 0, -1], [track.dx, track.dy, track.dz]))
+ self.zeniths.append(zenith)
+
+ self.qualities.append(track.Q)
+
return blob
def plot(self):
while True:
time.sleep(self.interval)
- self.create_plot()
-
- def create_plot(self):
- n = len(self.zeniths)
- n_ok = n - np.count_nonzero(np.isnan(self.zeniths))
- fontsize = 16
+ self.create_zenith_plot()
+ self.create_quality_plot()
- reco = np.loadtxt("reco2.txt", delimiter=',')[-self.max_events:]
- roy_zeniths, roy_Q = -reco[:, 0], reco[:, 1]
- roy_zeniths = roy_zeniths[roy_Q < 40]
- n_roy = len(roy_zeniths)
+ def create_quality_plot(self):
+ n = len(self.qualities)
+
+ plt.clf()
fig, ax = plt.subplots(figsize=(16, 8))
ax.hist(
- self.zeniths,
- bins=180,
+ self.qualities,
+ bins=100,
label="JGandalf (last %d events)" % n,
- # label="JGandalf",
histtype="step",
normed=True,
lw=3)
+ ax.set_title(
+ "Quality distribution of online track reconstructions\n%s UTC" %
+ datetime.utcnow().strftime("%c"))
+ ax.set_xlabel(r"Quality", fontsize=self.fontsize)
+ ax.set_ylabel("normed count", fontsize=self.fontsize)
+ ax.tick_params(labelsize=self.fontsize)
+ ax.set_yscale("log")
+ plt.legend(fontsize=self.fontsize, loc=2)
+ filename = os.path.join(self.plots_path, 'gandalf_quality.png')
+ plt.savefig(filename, dpi=120, bbox_inches="tight")
+ plt.close('all')
+
+ def create_zenith_plot(self):
+ n = len(self.zeniths)
+
+ plt.clf()
+ fig, ax = plt.subplots(figsize=(16, 8))
ax.hist(
- roy_zeniths,
+ self.zeniths,
bins=180,
- label="ROy (last %d events)" % n_roy,
- # label="ROy",
+ label="JGandalf (last %d events)" % n,
histtype="step",
normed=True,
lw=3)
ax.set_title(
"Zenith distribution of online track reconstructions\n%s UTC" %
datetime.utcnow().strftime("%c"))
- ax.set_xlabel(r"cos(zenith)", fontsize=fontsize)
- ax.set_ylabel("normed count", fontsize=fontsize)
- ax.tick_params(labelsize=fontsize)
+ ax.set_xlabel(r"cos(zenith)", fontsize=self.fontsize)
+ ax.set_ylabel("normed count", fontsize=self.fontsize)
+ ax.tick_params(labelsize=self.fontsize)
ax.set_yscale("log")
- plt.legend(fontsize=fontsize, loc=2)
- filename = os.path.join(self.plots_path, 'track_reco.png')
+ plt.legend(fontsize=self.fontsize, loc=2)
+ filename = os.path.join(self.plots_path, 'gandalf_zenith.png')
plt.savefig(filename, dpi=120, bbox_inches="tight")
plt.close('all')
@@ -109,7 +128,7 @@ def main():
timeout=60 * 60 * 24 * 7,
max_queue=2000)
pipe.attach(kp.io.daq.DAQProcessor)
- pipe.attach(ZenithDistribution, plots_path=plots_path)
+ pipe.attach(IO_OLINEDistributions, plots_path=plots_path)
pipe.drain()
--
GitLab