def test_sequence_access():
evt = hep.GenEvent()
evt.add_particle(hep.GenParticle())
evt.particles[0].momentum = (1, 2, 3, 4)
evt.particles[0].pid = 5
evt.add_vertex(hep.GenVertex())
evt.vertices[0].position = (1, 2, 3, 4)
assert len(evt.particles) == 1
assert evt.particles[0].id == 1
assert evt.particles[0].pid == 5
assert evt.particles[0].momentum == (1, 2, 3, 4)
assert len(evt.vertices) == 1
assert evt.vertices[0].id == -1
assert evt.vertices[0].position == (1, 2, 3, 4)
def test_sequence_access():
evt = hep.GenEvent()
evt.add_particle(hep.GenParticle())
evt.particles[0].momentum = (1, 2, 3, 4)
evt.particles[0].pid = 5
evt.add_vertex(hep.GenVertex())
evt.vertices[0].position = (1, 2, 3, 4)
assert len(evt.particles) == 1
assert evt.particles[0].id == 1
assert evt.particles[0].pid == 5
assert evt.particles[0].momentum == (1, 2, 3, 4)
assert len(evt.vertices) == 1
assert evt.vertices[0].id == -1
assert evt.vertices[0].position == (1, 2, 3, 4)
assert repr(
evt
) == "GenEvent(momentum_unit=1, length_unit=0, event_number=0, particles=[GenParticle(FourVector(1, 2, 3, 4), status=0, id=1, production_vertex=0, end_vertex=-1)], vertices=[GenVertex(FourVector(1, 2, 3, 4), status=0, id=-1, particles_in=[], particles_out=[])], run_info=None)"
def particles_parsed(self, event, min_track_pt=0.):
particles = []
for particle in event:
pid = int(particle['particle_ID'])
status = int(particle['status'])
e = particle['E']
px = particle['px']
py = particle['py']
pz = particle['pz']
four_vector = pyhepmc_ng.FourVector(px, py, pz, e)
particle = pyhepmc_ng.GenParticle(four_vector, pid, status)
pt = particle.momentum.pt()
if pid != 12 and pid != 14 and pid != 16: # Remove neutrinos
if pt > min_track_pt:
particles.append(particle)
return particles
def particles(self, event, min_track_pt=0.):
particles = []
for particle in event:
pid = int(particle[1])
status = int(particle[2])
e = particle[3]
px = particle[4]
py = particle[5]
pz = particle[6]
four_vector = pyhepmc_ng.FourVector(px, py, pz, e)
particle = pyhepmc_ng.GenParticle(four_vector, pid, status)
pt = particle.momentum.pt()
if pid != 12 and pid != 14 and pid != 16: # Remove neutrinos
if pt > min_track_pt:
particles.append(particle)
return particles
def add_hadron(self, event, line):
self.string_tokenizer.set(line)
vars = []
while self.string_tokenizer.done():
token = self.string_tokenizer.next()
if token != 'H':
vars.append(token)
label = int(vars[1])
id = int(vars[2])
stat = int(vars[3])
pt = float(vars[4])
eta = float(vars[5])
phi = float(vars[6])
e = float(vars[7])
px = pt * np.cos(phi)
py = pt * np.sin(phi)
pz = pt * np.sinh(eta)
four_vector = pyhepmc_ng.FourVector(px, py, pz, e)
particle = pyhepmc_ng.GenParticle(four_vector, id, status)
event.hadrons.append(particle)
def prepare_event():
#
# In this example we will place the following event into HepMC "by hand"
#
# name status pdg_id parent Px Py Pz Energy Mass
# 1 !p+! 3 2212 0,0 0.000 0.000 7000.000 7000.000 0.938
# 2 !p+! 3 2212 0,0 0.000 0.000-7000.000 7000.000 0.938
#=========================================================================
# 3 !d! 3 1 1,1 0.750 -1.569 32.191 32.238 0.000
# 4 !u~! 3 -2 2,2 -3.047 -19.000 -54.629 57.920 0.000
# 5 !W-! 3 -24 3,4 1.517 -20.68 -20.605 85.925 80.799
# 6 !gamma! 1 22 3,4 -3.813 0.113 -1.833 4.233 0.000
# 7 !d! 1 1 5,5 -2.445 28.816 6.082 29.552 0.010
# 8 !u~! 1 -2 5,5 3.962 -49.498 -26.687 56.373 0.006
# now we build the graph, which will looks like
# p7 #
# p1 / #
# \v1__p3 p5---v4 #
# \_v3_/ \ #
# / \ p8 #
# v2__p4 \ #
# / p6 #
# p2 #
# #
evt = hep.GenEvent(hep.Units.GEV, hep.Units.MM)
evt.event_number = 1
# px py pz e pdgid status
p1 = hep.GenParticle((0.0, 0.0, 7000.0, 7000.0), 2212, 1)
p1.generated_mass = 0.938
p2 = hep.GenParticle((0.0, 0.0, -7000.0, 7000.0), 2212, 2)
p2.generated_mass = 0.938
p3 = hep.GenParticle((0.750, -1.569, 32.191, 32.238), 1, 3)
p3.generated_mass = 0
p4 = hep.GenParticle((-3.047, -19.0, -54.629, 57.920), -2, 4)
p4.generated_mass = 0
p5 = hep.GenParticle((1.517, -20.68, -20.605, 85.925), -24, 5)
p5.generated_mass = 80.799
p6 = hep.GenParticle((-3.813, 0.113, -1.833, 4.233), 22, 6)
p6.generated_mass = 0
p7 = hep.GenParticle((-2.445, 28.816, 6.082, 29.552), 1, 7)
p7.generated_mass = 0.01
p8 = hep.GenParticle((3.962, -49.498, -26.687, 56.373), -2, 8)
p8.generated_mass = 0.006
evt.add_particle(p1)
evt.add_particle(p2)
evt.add_particle(p3)
evt.add_particle(p4)
evt.add_particle(p5)
evt.add_particle(p6)
evt.add_particle(p7)
evt.add_particle(p8)
# make sure vertex is not optimized away by WriterAscii
v1 = hep.GenVertex((1.0, 1.0, 1.0, 1.0))
v1.add_particle_in(p1)
v1.add_particle_out(p3)
evt.add_vertex(v1)
# make sure vertex is not optimized away by WriterAscii
v2 = hep.GenVertex((2.0, 2.0, 2.0, 2.0))
v2.add_particle_in(p2)
v2.add_particle_out(p4)
evt.add_vertex(v2)
# make sure vertex is not optimized away by WriterAscii
v3 = hep.GenVertex((3.0, 3.0, 3.0, 3.0))
v3.add_particle_in(p3)
v3.add_particle_in(p4)
v3.add_particle_out(p5)
v3.add_particle_out(p6)
evt.add_vertex(v3)
# make sure vertex is not optimized away by WriterAscii
v4 = hep.GenVertex((4.0, 4.0, 4.0, 4.0))
v4.add_particle_in(p5)
v4.add_particle_out(p7)
v4.add_particle_out(p8)
evt.add_vertex(v4)
return evt