diff --git a/Generators/GenieReader/python/GenieReaderAlg.py b/Generators/GenieReader/python/GenieReaderAlg.py index fc1cb184f0016fe4f73599a9f1eb1edab209f5f1..c930ae66ae2aa8ef4eddf7f31873acc598d31a5c 100644 --- a/Generators/GenieReader/python/GenieReaderAlg.py +++ b/Generators/GenieReader/python/GenieReaderAlg.py @@ -3,6 +3,8 @@ from AthenaCommon.AppMgr import ServiceMgr as svcMgr from GeneratorModules.EvgenAlg import EvgenAlg from AthenaPython.PyAthena import StatusCode +from AthenaCommon.SystemOfUnits import GeV, m + import ROOT __author__ = "Dave Caser <dcasper@uci.edu>" @@ -20,7 +22,7 @@ class GenieReader(EvgenAlg): from AthenaPython.PyAthena import HepMC as HepMC evt.weights().push_back(1.0) - pos = HepMC.FourVector(self.evtStore["vx"]*1000, self.evtStore["vy"]*1000, self.evtStore["vz"]*1000, 0) + pos = HepMC.FourVector(self.evtStore["vx"]*m, self.evtStore["vy"]*m, self.evtStore["vz"]*m, 0) gv = HepMC.GenVertex(pos) ROOT.SetOwnership(gv, False) evt.add_vertex(gv) @@ -36,9 +38,9 @@ class GenieReader(EvgenAlg): for i in range(nParticles): gp = HepMC.GenParticle() - mom = HepMC.FourVector(px[i], py[i], pz[i], E[i]) + mom = HepMC.FourVector(px[i]*GeV, py[i]*GeV, pz[i]*GeV, E[i]*GeV) gp.set_momentum(mom) - gp.set_generated_mass(M[i]) + gp.set_generated_mass(M[i]*GeV) gp.set_pdg_id(pdgc[i]) genie_status = status[i] if (genie_status == 0): # initial particle