Cleanup OfflineReader interface

km3io/offline.py

@@ -201,7 +201,7 @@ class OfflineKeys:
 
 class OfflineReader:
     """reader for offline ROOT files"""
-    def __init__(self, file_path=None):
+    def __init__(self, file_path=None, fobj=None, data=None):
         """ OfflineReader class is an offline ROOT file wrapper
 
         Parameters
@@ -209,34 +209,37 @@ class OfflineReader:
         file_path : path-like object
             Path to the file of interest. It can be a str or any python
             path-like object that points to the file.
+
         """
         self._file_path = file_path
 
         if file_path is not None:
-            self._tree = uproot.open(self._file_path)[MAIN_TREE_NAME]
+            self._fobj = uproot.open(self._file_path)
+            self._tree = self._fobj[MAIN_TREE_NAME]
             self._data = self._tree.lazyarrays(
                 basketcache=uproot.cache.ThreadSafeArrayCache(
                     BASKET_CACHE_SIZE))
+        else:
+            self._fobj = fobj
+            self._tree = self._fobj[MAIN_TREE_NAME]
+            self._data = data
 
     @classmethod
-    def from_tree(cls, tree, data):
-        instance = cls()
-        instance._tree = tree
-        instance._data = data
+    def from_index(cls, source, index):
+        instance = cls(fobj=source._fobj, data=source._data[index])
         return instance
 
-    def __getitem__(self, item):
-        return OfflineReader.from_tree(tree=self._tree, data=self._data[item])
+    def __getitem__(self, index):
+        return OfflineReader.from_index(source=self, index=index)
 
     def __len__(self):
         return len(self._data)
 
     @cached_property
     def header(self):
-        fobj = uproot.open(self._file_path)
-        if 'Head' in fobj:
+        if 'Head' in self._fobj:
             header = {}
-            for n, x in fobj['Head']._map_3c_string_2c_string_3e_.items():
+            for n, x in self._fobj['Head']._map_3c_string_2c_string_3e_.items():
                 header[n.decode("utf-8")] = x.decode("utf-8").strip()
             return header
         else: