Included antineutrons in Ekin cut. Fixed bug with cutting on Ekin bef…

…ore reading the MARS values
MuonColliderSoft · Oct 9, 2020 · fa9c4bc · fa9c4bc
1 parent 24f1b89
commit fa9c4bc
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/utils/mars_to_slcio.py b/utils/mars_to_slcio.py
@@ -102,17 +102,17 @@
             elif not args.invert_pdgs and pdg not in args.pdgs:
                 continue
 
-        # Skipping if it's a neutron with too low kinetic energy
-        if args.ne_min is not None and pdg == 2112 and sqrt(px*px + py*py + pz*pz) < args.ne_min:
-            continue
-
         # Converting the rest of string values to floats
         x,y,z,px,py,pz,toff,w = map(float, ss[2:10])
 
+        # Skipping if it's a neutron with too low kinetic energy
+        if args.ne_min is not None and abs(pdg) == 2112 and sqrt(px*px + py*py + pz*pz) < args.ne_min:
+            continue
+
         # Converting time: s -> ns
         toff *= 1e9
 
-        # Skipping if its time is greater than allowed
+        # Skipping if particle's time is greater than allowed
         if args.t_max is not None and toff > args.t_max:
             continue