diff options
| -rwxr-xr-x | analyze_genie_mc.py | 55 |
1 files changed, 39 insertions, 16 deletions
diff --git a/analyze_genie_mc.py b/analyze_genie_mc.py index cdc839c..0a83fbb 100755 --- a/analyze_genie_mc.py +++ b/analyze_genie_mc.py @@ -58,18 +58,34 @@ def get_reaction(event): else: print("unknown nucleon %i" % event.hitnuc) - if event.neu == 12: - products.append('e-') - elif event.neu == -12: - products.append('e+') - elif event.neu == 14: - products.append('u-') - elif event.neu == -14: - products.append('u+') - elif event.neu == 16: - products.append('t-') - elif event.neu == -16: - products.append('t+') + if event.cc: + if event.neu == 12: + products.append('e-') + elif event.neu == -12: + products.append('e+') + elif event.neu == 14: + products.append('u-') + elif event.neu == -14: + products.append('u+') + elif event.neu == 16: + products.append('t-') + elif event.neu == -16: + products.append('t+') + elif event.nc: + if event.neu == 12: + products.append('ve') + elif event.neu == -12: + products.append('vebar') + elif event.neu == 14: + products.append('vu') + elif event.neu == -14: + products.append('vubar') + elif event.neu == 16: + products.append('vt') + elif event.neu == -16: + products.append('vtbar') + else: + products.append("???") for pdg in event.pdgf: if pdg == 2112: @@ -145,14 +161,21 @@ if __name__ == '__main__': for event in T: neutrons = 0 - nrings = 1 + nrings = 0 e_like_rings = 0 u_like_rings = 0 - if abs(event.neu) == 12: - e_like_rings = 1 + if event.cc: + if abs(event.neu) == 12: + e_like_rings = 1 + else: + u_like_rings = 1 + nrings = 1 + elif event.nc: + pass else: - u_like_rings = 1 + print("event is not cc or nc!") + continue for i, pdg in enumerate(event.pdgf): if pdg == 2112: |