Skip to content
Snippets Groups Projects
Commit 769c8aba authored by Tooba Ali's avatar Tooba Ali
Browse files

Update benchmarks/dis/analysis/truth_reconstruction.py

parent f4496805
No related branches found
No related tags found
No related merge requests found
This commit is part of merge request !186. Comments created here will be created in the context of that merge request.
#!/usr/bin/env python
# coding: utf-8
# In[1]:
import os import os
import numpy as np import numpy as np
import uproot as ur import uproot as ur
...@@ -32,9 +26,6 @@ p = int(args.pbeam) ...@@ -32,9 +26,6 @@ p = int(args.pbeam)
Nevents = int(args.nevents) Nevents = int(args.nevents)
# In[16]:
for array in ur.iterate(rec_file + ':events',['MCParticles/MCParticles.generatorStatus', for array in ur.iterate(rec_file + ':events',['MCParticles/MCParticles.generatorStatus',
'MCParticles/MCParticles.mass', 'MCParticles/MCParticles.mass',
'MCParticles/MCParticles.PDG', 'MCParticles/MCParticles.PDG',
...@@ -71,7 +62,6 @@ theta_rc = np.arctan2(np.sqrt(px_rc**2+py_rc**2), pz_rc) ...@@ -71,7 +62,6 @@ theta_rc = np.arctan2(np.sqrt(px_rc**2+py_rc**2), pz_rc)
phi_rc = np.arctan2(py_rc, px_rc) phi_rc = np.arctan2(py_rc, px_rc)
# In[ ]:
booll = (PDG_mc[simID])==(PDG_rc[recID]) booll = (PDG_mc[simID])==(PDG_rc[recID])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment