From 1c2db72a001fc39472d1db3b7b6474756cb13f09 Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Thu, 10 Dec 2020 23:51:12 +0100
Subject: [PATCH] Propagate branch map corrections
---
km3io/rootio.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/km3io/rootio.py b/km3io/rootio.py
index 77926e0..f0b9e3f 100644
--- a/km3io/rootio.py
+++ b/km3io/rootio.py
@@ -28,6 +28,7 @@ class EventReader:
step_size=2000,
keys=None,
aliases=None,
+ nested_branches=None,
event_ctor=None,
):
"""EventReader base class
@@ -65,6 +66,11 @@ class EventReader:
self._event_ctor = event_ctor
self._index_chain = [] if index_chain is None else index_chain
+ if aliases is not None:
+ self.aliases = aliases
+ if nested_branches is not None:
+ self.nested_branches = nested_branches
+
if self._keys is None:
self._initialise_keys()
@@ -141,6 +147,7 @@ class EventReader:
index_chain=self._index_chain + [key],
step_size=self._step_size,
aliases=self.aliases,
+ nested_branches=self.nested_branches,
keys=self.keys(),
event_ctor=self._event_ctor,
)
--
GitLab