From 792150eaf6e96c587007851d87af1b608911fb5d Mon Sep 17 00:00:00 2001 From: Dave Casper <dcasper@uci.edu> Date: Fri, 14 Jan 2022 05:32:45 -0800 Subject: [PATCH] Fix critical units error --- Generators/GenieReader/python/GenieReaderAlg.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/Generators/GenieReader/python/GenieReaderAlg.py b/Generators/GenieReader/python/GenieReaderAlg.py index fc1cb184..c930ae66 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 -- GitLab