From 12daabe232f9cd59657e80e0c3a7f46a022b4285 Mon Sep 17 00:00:00 2001
From: Johannes Schumann <johannes.schumann@fau.de>
Date: Tue, 24 Nov 2020 12:29:51 +0100
Subject: [PATCH] Fixes; declaring nonlocal variables
---
km3buu/output.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/km3buu/output.py b/km3buu/output.py
index 8e67bf8..fe00988 100644
--- a/km3buu/output.py
+++ b/km3buu/output.py
@@ -281,6 +281,9 @@ def write_detector_file(gibuu_output,
mc_trk_id = 0
def add_particles(particle_data, pos_offset, rotation):
+ nonlocal aafile
+ nonlocal mc_event_id
+ nonlocal mc_trk_id
for i in range(len(particle_data.E)):
trk = ROOT.Trk()
trk.id = mc_trk_id
@@ -289,9 +292,12 @@ 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))
- ptrcl_pos = np.array(
+ try:
+ prtcl_pos = np.array(
[particle_data.x[i], particle_data.y[i], particle_data.z[i]])
- trk.pos.set(*np.add(pos_offset, prtcl_pos))
+ trk.pos.set(*np.add(pos_offset, prtcl_pos))
+ except AttributeError:
+ trk.pos.set(*pos_offset)
trk.dir.set(*p_dir)
trk.mother_id = 0
trk.type = int(particle_data.barcode[i])
--
GitLab