From 0826e245d6dabfac7fd1ddf9f66d82692e1b6adf Mon Sep 17 00:00:00 2001
From: Johannes Schumann <johannes.schumann@fau.de>
Date: Tue, 24 Nov 2020 10:41:50 +0100
Subject: [PATCH] Fixes for the AANET part
---
km3buu/output.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/km3buu/output.py b/km3buu/output.py
index 4ef5389..8e67bf8 100644
--- a/km3buu/output.py
+++ b/km3buu/output.py
@@ -280,7 +280,7 @@ def write_detector_file(gibuu_output,
mc_event_id = 0
mc_trk_id = 0
- def add_particles(particle_data, rotation):
+ def add_particles(particle_data, pos_offset, rotation):
for i in range(len(particle_data.E)):
trk = ROOT.Trk()
trk.id = mc_trk_id
@@ -289,9 +289,9 @@ def write_detector_file(gibuu_output,
particle_data.Px[i], particle_data.Py[i], particle_data.Pz[i]
])
p_dir = rotation.apply(mom / np.linalg.norm(mom))
- prtcl_pos_offset = np.array(
+ ptrcl_pos = np.array(
[particle_data.x[i], particle_data.y[i], particle_data.z[i]])
- trk.pos.set(*np.add(vtx_pos, prtcl_pos_offset))
+ trk.pos.set(*np.add(pos_offset, prtcl_pos))
trk.dir.set(*p_dir)
trk.mother_id = 0
trk.type = int(particle_data.barcode[i])
@@ -389,9 +389,9 @@ def write_detector_file(gibuu_output,
if tau_secondaries is not None:
event_tau_sec = tau_secondaries[i]
- add_particles(event_tau_sec, R)
+ add_particles(event_tau_sec, vtx_pos, R)
- add_particles(event, R)
+ add_particles(event, vtx_pos, R)
aafile.write()
--
GitLab