From 1c484b75e817f78fc97111467d39ef583194b70b Mon Sep 17 00:00:00 2001
From: Dmitry Kalinkin <dmitry.kalinkin@gmail.com>
Date: Wed, 5 Jun 2024 19:13:11 -0400
Subject: [PATCH] ecal_backwards.org: use pyHepMC3 SWIG bindings that support
ROOT formats
---
benchmarks/backgrounds/ecal_backwards.org | 74 +++++++++++------------
1 file changed, 37 insertions(+), 37 deletions(-)
diff --git a/benchmarks/backgrounds/ecal_backwards.org b/benchmarks/backgrounds/ecal_backwards.org
index a420358e..e35387ab 100644
--- a/benchmarks/backgrounds/ecal_backwards.org
+++ b/benchmarks/backgrounds/ecal_backwards.org
@@ -15,7 +15,7 @@ import dask_awkward as dak
import dask_histogram as dh
import numpy as np
import uproot
-import pyhepmc
+from pyHepMC3 import HepMC3
#+end_src
#+begin_src jupyter-python :results silent
@@ -66,47 +66,47 @@ output_dir.mkdir(parents=True, exist_ok=True)
#+begin_src jupyter-python :results silent
builder = ak.ArrayBuilder()
n = 0
-with pyhepmc.open(PROTON_BEAM_GAS_GEN) as f:
- it = iter(f)
- for event in f:
- builder.begin_list()
- for vertex in event.vertices:
- assert event.length_unit.name == "MM"
- with builder.record("Vector4D"):
- builder.field("x")
- builder.real(vertex.position.x)
- builder.field("y")
- builder.real(vertex.position.y)
- builder.field("z")
- builder.real(vertex.position.z)
- builder.field("t")
- builder.real(vertex.position.t)
- builder.end_list()
- n += 1
- if n > 10000: break
+f = HepMC3.ReaderPlugin(PROTON_BEAM_GAS_GEN, "libHepMC3rootIO.so", "newReaderRootTreefile")
+event = HepMC3.GenEvent()
+while f.read_event(event):
+ builder.begin_list()
+ assert event.length_unit().name == "MM"
+ for vertex in event.vertices():
+ with builder.record("Vector4D"):
+ builder.field("x")
+ builder.real(vertex.position().x())
+ builder.field("y")
+ builder.real(vertex.position().y())
+ builder.field("z")
+ builder.real(vertex.position().z())
+ builder.field("t")
+ builder.real(vertex.position().t())
+ builder.end_list()
+ n += 1
+ if n > 10000: break
vertices_proton_beam_gas = builder.snapshot()
builder = ak.ArrayBuilder()
n = 0
-with pyhepmc.open(ELECTRON_BEAM_GAS_GEN) as f:
- it = iter(f)
- for event in f:
- builder.begin_list()
- for vertex in event.vertices:
- assert event.length_unit.name == "MM"
- with builder.record("Vector3D"):
- builder.field("x")
- builder.real(vertex.position.x)
- builder.field("y")
- builder.real(vertex.position.y)
- builder.field("z")
- builder.real(vertex.position.z)
- builder.field("t")
- builder.real(vertex.position.t)
- builder.end_list()
- n += 1
- if n > 10000: break
+f = HepMC3.ReaderPlugin(ELECTRON_BEAM_GAS_GEN, "libHepMC3rootIO.so", "newReaderRootTreefile")
+event = HepMC3.GenEvent()
+while f.read_event(event):
+ builder.begin_list()
+ assert event.length_unit().name == "MM"
+ for vertex in event.vertices():
+ with builder.record("Vector4D"):
+ builder.field("x")
+ builder.real(vertex.position().x())
+ builder.field("y")
+ builder.real(vertex.position().y())
+ builder.field("z")
+ builder.real(vertex.position().z())
+ builder.field("t")
+ builder.real(vertex.position().t())
+ builder.end_list()
+ n += 1
+ if n > 10000: break
vertices_electron_beam_gas = builder.snapshot()
#+end_src
--
GitLab