diff options
| -rw-r--r-- | chroma/generator/g4gen.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/chroma/generator/g4gen.py b/chroma/generator/g4gen.py index 27f1454..3e185a9 100644 --- a/chroma/generator/g4gen.py +++ b/chroma/generator/g4gen.py @@ -130,9 +130,8 @@ class G4Generator(object): try: for vertex in vertices: self.particle_gun.SetParticleByName(vertex.particle_name) - mass = G4ParticleTable.GetParticleTable().FindParticle(vertex.particle_name).GetPDGMass() - total_energy = vertex.ke*MeV + mass - self.particle_gun.SetParticleEnergy(total_energy) + #mass = G4ParticleTable.GetParticleTable().FindParticle(vertex.particle_name).GetPDGMass() + self.particle_gun.SetParticleEnergy(vertex.ke*MeV) # Must be float type to call GEANT4 code pos = np.asarray(vertex.pos, dtype=np.float64) |