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
!47
Resolve "uproot4 integration"
Code
Review changes
Check out branch
Download
Patches
Plain diff
Merged
Resolve "uproot4 integration"
58-uproot4-integration-2
into
master
Overview
0
Commits
47
Pipelines
43
Changes
4
Merged
Tamas Gal
requested to merge
58-uproot4-integration-2
into
master
4 years ago
Overview
0
Commits
47
Pipelines
43
Changes
4
Expand
Closes
#58 (closed)
Edited
4 years ago
by
Tamas Gal
0
0
Merge request reports
Viewing commit
3cce5598
Prev
Next
Show latest version
4 files
+
231
−
467
Inline
Compare changes
Side-by-side
Inline
Show whitespace changes
Show one file at a time
Files
4
Search (e.g. *.vue) (Ctrl+P)
3cce5598
Restructure and adapt GSGReader
· 3cce5598
Tamas Gal
authored
4 years ago
km3io/gseagen.py
+
4
−
21
Options
@@ -3,28 +3,15 @@
# Filename: gseagen.py
# Author: Johannes Schumann <jschumann@km3net.de>
import
uproot3
import
numpy
as
np
import
warnings
from
.rootio
import
Branch
,
BranchMapp
er
from
.rootio
import
EventRead
er
from
.tools
import
cached_property
MAIN_TREE_NAME
=
"
Events
"
class
GSGReader
:
class
GSGReader
(
EventReader
):
"""
reader for gSeaGen ROOT files
"""
def
__init__
(
self
,
file_path
=
None
,
fobj
=
None
):
"""
GSGReader class is a gSeaGen ROOT file wrapper
Parameters
----------
file_path : file path or file-like object
The file handler. It can be a str or any python path-like object
that points to the file.
"""
self
.
_fobj
=
uproot3
.
open
(
file_path
)
event_path
=
"
Events
"
skip_keys
=
[
"
Header
"
]
@cached_property
def
header
(
self
):
@@ -42,7 +29,3 @@ class GSGReader:
return
header
else
:
warnings
.
warn
(
"
Your file header has an unsupported format
"
)
@cached_property
def
events
(
self
):
return
Branch
(
self
.
_fobj
,
BranchMapper
(
name
=
"
Events
"
,
key
=
"
Events
"
))
Loading