Skip to content

Commit d010f49

Browse files
Add mcsteps branch to ntuplehelper
1 parent 77102eb commit d010f49

2 files changed

Lines changed: 6 additions & 5 deletions

File tree

helper/ntuplehelper.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,14 @@
33
class nthelper:
44

55
single_object_branches = ['evtinfo', 'evtinfomc', 'hitcount', 'tcnt', 'crvsummary', 'crvsummarymc']
6-
vector_object_branches = ['trk', 'trkmc', 'trkcalohit', 'trkcalohitmc', 'caloclusters', 'calohits', 'calorecodigis', 'calodigis', 'crvcoincs', 'crvcoincsmc', 'crvcoincsmcplane', 'trkqual', 'trkpid']
6+
vector_object_branches = ['trk', 'trkmc', 'trkcalohit', 'trkcalohitmc', 'caloclusters', 'calohits', 'calorecodigis', 'calodigis', 'crvcoincs', 'crvcoincsmc', 'crvcoincsmcplane', 'trkqual', 'trkpid', 'mcsteps']
77
vector_vector_object_branches = ['trksegs', 'trksegpars_lh', 'trksegpars_ch', 'trksegpars_kl', 'trkmcsim', 'trkhits', 'trkhitsmc', 'trkmats', 'trkhitcalibs', 'trkmcsci', 'trkmcssi', 'trksegsmc' ]
88

99
evt_branches = ['evtinfo','evtinfomc','hitcount','tcnt']
1010
trk_branches = ['trk', 'trkmc', 'trkcalohit', 'trkcalohitmc', 'trkqual', 'trkpid']
1111
trksegs_branches = ['trksegs', 'trksegpars_lh', 'trksegpars_ch', 'trksegpars_kl', 'trksegsmc']
1212
straw_branches = ['trkhits', 'trkmats', 'trkhitsmc', 'trkhitcalibs']
13-
mc_branches = ['trkmcsim']
13+
mc_branches = ['trkmcsim', 'mcsteps']
1414
calo_branches = ['caloclusters', 'calohits', 'calorecodigis', 'calodigis']
1515
crv_branches = ['crvsummary','crvsummarymc','crvcoincs','crvcoincsmc','crvcoincsmcplane']
1616
deprecated_branches = ['trkmcsci','trkmcssi']
@@ -58,7 +58,8 @@ class nthelper:
5858
"trkqual" : "MVAResultInfo",
5959
"trkpid" : "MVAResultInfo",
6060
"helices" : "HelixInfo",
61-
"trksegsmc" : "SurfaceStepInfo"
61+
"trksegsmc" : "SurfaceStepInfo",
62+
"mcsteps" : "MCStepInfo"
6263
}
6364

6465
#

inc/MCStepInfo.hh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,8 @@ namespace mu2e {
1414
float time = 0; // time of this step WRT MC primary proton (ns)
1515
float de = 0; // energy deposit through this step (MeV)
1616
float dp = 0; // momentum magnitude change throw this step (MeV/c)
17-
bool early = false;
18-
bool late = false; // flag if this is the earliest or latest step
17+
bool early = false; // flag if this is the earliest step
18+
bool late = false; // flag if this is the latest step
1919
XYZVectorF mom; // particle momentum at the start of this step
2020
XYZVectorF pos; // particle position at the start of this step
2121
void reset() {*this = MCStepInfo(); }

0 commit comments

Comments
 (0)