Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
K
km3io
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Package registry
Container registry
Model registry
Operate
Environments
Terraform modules
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
km3py
km3io
Merge requests
!21
Resolve "Reduce the amount of uproot.open (to one)"
Code
Review changes
Check out branch
Download
Patches
Plain diff
Merged
Resolve "Reduce the amount of uproot.open (to one)"
36-reduce-the-amount-of-uproot-open-to-one
into
master
Overview
0
Commits
14
Pipelines
13
Changes
1
Merged
Tamas Gal
requested to merge
36-reduce-the-amount-of-uproot-open-to-one
into
master
5 years ago
Overview
0
Commits
14
Pipelines
13
Changes
1
Expand
Closes
#36 (closed)
Edited
5 years ago
by
Tamas Gal
0
0
Merge request reports
Viewing commit
3a913f81
Prev
Next
Show latest version
1 file
+
14
−
11
Inline
Compare changes
Side-by-side
Inline
Show whitespace changes
Show one file at a time
3a913f81
Cleanup OfflineReader interface
· 3a913f81
Tamas Gal
authored
5 years ago
km3io/offline.py
+
14
−
11
Options
@@ -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
,
i
tem
):
return
OfflineReader
.
from_
tree
(
tree
=
self
.
_tree
,
data
=
self
.
_data
[
item
]
)
def
__getitem__
(
self
,
i
ndex
):
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
:
Loading