Skip to content
Snippets Groups Projects

B PV association

Merged Abhijit Mathad requested to merge pv into master
3 unresolved threads
1 file
+ 79
79
Compare changes
  • Side-by-side
  • Inline
+ 79
79
@@ -166,89 +166,89 @@ ALLVELOTRACKS_BPV_IPCHI2 = lambda tracks, pvs: F.MAP(F.VALUE_OR(F.NaN) @ F.BPVIP
#ALLVELOTRACKS_ENDVELO_Z = lambda tracks: F.MAP(F.VALUE_OR(F.NaN) @ F.Z_COORDINATE @ F.POSITION @ F.STATE_AT("EndVelo")) @ F.TES(tracks)
#ALLVELOTRACKS_ENDVELO_POSERR = lambda tracks, indx_i: F.MAP(F.VALUE_OR(F.NaN) @ F.CALL(indx_i, indx_i) @ F.TRACK_COVARIANCE @ F.STATE_AT("EndVelo")) @ F.TES(tracks)
def get_all_part_vars(evt_vars, particles, pvs, MCTRUTH_parts, suffix = 'PARTS', is_long = False):
evt_vars[f'ALL{suffix}_ID[part_indx]'] = ALLPART_ID(particles)
evt_vars[f'ALL{suffix}_KEY[part_indx]'] = ALLPART_KEY(particles)
evt_vars[f'ALL{suffix}_Q[part_indx]'] = ALLPART_Q(particles)
evt_vars[f'ALL{suffix}_PX[part_indx]'] = ALLPART_PX(particles)
evt_vars[f'ALL{suffix}_PY[part_indx]'] = ALLPART_PY(particles)
evt_vars[f'ALL{suffix}_PZ[part_indx]'] = ALLPART_PZ(particles)
evt_vars[f'ALL{suffix}_PXERR[part_indx]'] = ALLPART_MOMERR(particles, 0)
evt_vars[f'ALL{suffix}_PYERR[part_indx]'] = ALLPART_MOMERR(particles, 1)
evt_vars[f'ALL{suffix}_PZERR[part_indx]'] = ALLPART_MOMERR(particles, 2)
evt_vars[f'ALL{suffix}_MASS[part_indx]'] = ALLPART_MASS(particles)
evt_vars[f'ALL{suffix}_ENERGY[part_indx]'] = ALLPART_ENERGY(particles)
evt_vars[f'ALL{suffix}_ETA[part_indx]'] = ALLPART_ETA(particles)
evt_vars[f'ALL{suffix}_PHI[part_indx]'] = ALLPART_PHI(particles)
evt_vars[f'ALL{suffix}_PIDPI[part_indx]'] = ALLPART_PIDPI(particles)
evt_vars[f'ALL{suffix}_PIDK[part_indx]'] = ALLPART_PIDK(particles)
evt_vars[f'ALL{suffix}_PIDP[part_indx]'] = ALLPART_PIDP(particles)
evt_vars[f'ALL{suffix}_PIDE[part_indx]'] = ALLPART_PIDE(particles)
evt_vars[f'ALL{suffix}_PIDMU[part_indx]'] = ALLPART_PIDMU(particles)
def get_all_part_vars(evt_vars, particles, pvs, MCTRUTH_parts, index_name, suffix = 'PARTS', is_long = False):
evt_vars[f'ALL{suffix}_ID[{index_name}]'] = ALLPART_ID(particles)
evt_vars[f'ALL{suffix}_KEY[{index_name}]'] = ALLPART_KEY(particles)
evt_vars[f'ALL{suffix}_Q[{index_name}]'] = ALLPART_Q(particles)
evt_vars[f'ALL{suffix}_PX[{index_name}]'] = ALLPART_PX(particles)
evt_vars[f'ALL{suffix}_PY[{index_name}]'] = ALLPART_PY(particles)
evt_vars[f'ALL{suffix}_PZ[{index_name}]'] = ALLPART_PZ(particles)
evt_vars[f'ALL{suffix}_PXERR[{index_name}]'] = ALLPART_MOMERR(particles, 0)
evt_vars[f'ALL{suffix}_PYERR[{index_name}]'] = ALLPART_MOMERR(particles, 1)
evt_vars[f'ALL{suffix}_PZERR[{index_name}]'] = ALLPART_MOMERR(particles, 2)
evt_vars[f'ALL{suffix}_MASS[{index_name}]'] = ALLPART_MASS(particles)
evt_vars[f'ALL{suffix}_ENERGY[{index_name}]'] = ALLPART_ENERGY(particles)
evt_vars[f'ALL{suffix}_ETA[{index_name}]'] = ALLPART_ETA(particles)
evt_vars[f'ALL{suffix}_PHI[{index_name}]'] = ALLPART_PHI(particles)
evt_vars[f'ALL{suffix}_PIDPI[{index_name}]'] = ALLPART_PIDPI(particles)
evt_vars[f'ALL{suffix}_PIDK[{index_name}]'] = ALLPART_PIDK(particles)
evt_vars[f'ALL{suffix}_PIDP[{index_name}]'] = ALLPART_PIDP(particles)
evt_vars[f'ALL{suffix}_PIDE[{index_name}]'] = ALLPART_PIDE(particles)
evt_vars[f'ALL{suffix}_PIDMU[{index_name}]'] = ALLPART_PIDMU(particles)
#P -> Proto -> Track
evt_vars[f'ALL{suffix}_TRACK_CHI2[part_indx]'] = ALLPART_TRACK_CHI2(particles)
evt_vars[f'ALL{suffix}_TRACK_NDOF[part_indx]'] = ALLPART_TRACK_NDOF(particles)
evt_vars[f'ALL{suffix}_TRACK_TYPE[part_indx]'] = ALLPART_TRACK_TYPE(particles)
evt_vars[f'ALL{suffix}_TRACK_GHOSTPROB[part_indx]'] = ALLPART_TRACK_GHOSTPROB(particles)
evt_vars[f'ALL{suffix}_TRACK_CHI2[{index_name}]'] = ALLPART_TRACK_CHI2(particles)
evt_vars[f'ALL{suffix}_TRACK_NDOF[{index_name}]'] = ALLPART_TRACK_NDOF(particles)
evt_vars[f'ALL{suffix}_TRACK_TYPE[{index_name}]'] = ALLPART_TRACK_TYPE(particles)
evt_vars[f'ALL{suffix}_TRACK_GHOSTPROB[{index_name}]'] = ALLPART_TRACK_GHOSTPROB(particles)
# BPV information
evt_vars[f'ALL{suffix}_BPV_KEY[part_indx]'] = ALLPART_BPV_KEY(particles, pvs)
evt_vars[f'ALL{suffix}_BPV_X[part_indx]'] = ALLPART_BPV_X(particles, pvs)
evt_vars[f'ALL{suffix}_BPV_Y[part_indx]'] = ALLPART_BPV_Y(particles, pvs)
evt_vars[f'ALL{suffix}_BPV_Z[part_indx]'] = ALLPART_BPV_Z(particles, pvs)
evt_vars[f'ALL{suffix}_BPV_IP[part_indx]'] = ALLPART_BPV_IP(particles, pvs)
evt_vars[f'ALL{suffix}_BPV_IPCHI2[part_indx]'] = ALLPART_BPV_IPCHI2(particles, pvs)
evt_vars[f'ALL{suffix}_BPV_KEY[{index_name}]'] = ALLPART_BPV_KEY(particles, pvs)
evt_vars[f'ALL{suffix}_BPV_X[{index_name}]'] = ALLPART_BPV_X(particles, pvs)
evt_vars[f'ALL{suffix}_BPV_Y[{index_name}]'] = ALLPART_BPV_Y(particles, pvs)
evt_vars[f'ALL{suffix}_BPV_Z[{index_name}]'] = ALLPART_BPV_Z(particles, pvs)
evt_vars[f'ALL{suffix}_BPV_IP[{index_name}]'] = ALLPART_BPV_IP(particles, pvs)
evt_vars[f'ALL{suffix}_BPV_IPCHI2[{index_name}]'] = ALLPART_BPV_IPCHI2(particles, pvs)
# particle has track but not a closest to beam state, so throws an error (Should this be an error at all?)
if is_long:
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_X[part_indx]'] = ALLPART_CLOSESTTOBEAM_X(particles)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_Y[part_indx]'] = ALLPART_CLOSESTTOBEAM_Y(particles)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_TX[part_indx]'] = ALLPART_CLOSESTTOBEAM_TX(particles)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_TY[part_indx]'] = ALLPART_CLOSESTTOBEAM_TY(particles)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_QOP[part_indx]'] = ALLPART_CLOSESTTOBEAM_QOP(particles)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_XERR[part_indx]'] = ALLPART_CLOSESTTOBEAM_STATEERR(particles, 0)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_YERR[part_indx]'] = ALLPART_CLOSESTTOBEAM_STATEERR(particles, 1)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_TXERR[part_indx]'] = ALLPART_CLOSESTTOBEAM_STATEERR(particles, 2)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_TYERR[part_indx]'] = ALLPART_CLOSESTTOBEAM_STATEERR(particles, 3)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_QOPERR[part_indx]'] = ALLPART_CLOSESTTOBEAM_STATEERR(particles, 4)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_X[{index_name}]'] = ALLPART_CLOSESTTOBEAM_X(particles)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_Y[{index_name}]'] = ALLPART_CLOSESTTOBEAM_Y(particles)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_TX[{index_name}]'] = ALLPART_CLOSESTTOBEAM_TX(particles)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_TY[{index_name}]'] = ALLPART_CLOSESTTOBEAM_TY(particles)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_QOP[{index_name}]'] = ALLPART_CLOSESTTOBEAM_QOP(particles)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_XERR[{index_name}]'] = ALLPART_CLOSESTTOBEAM_STATEERR(particles, 0)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_YERR[{index_name}]'] = ALLPART_CLOSESTTOBEAM_STATEERR(particles, 1)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_TXERR[{index_name}]'] = ALLPART_CLOSESTTOBEAM_STATEERR(particles, 2)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_TYERR[{index_name}]'] = ALLPART_CLOSESTTOBEAM_STATEERR(particles, 3)
evt_vars[f'ALL{suffix}_CLOSESTTOBEAM_QOPERR[{index_name}]'] = ALLPART_CLOSESTTOBEAM_STATEERR(particles, 4)
# store first measurement
evt_vars[f'ALL{suffix}_FIRSTMEAS_X[part_indx]'] = ALLPART_FIRSTMEAS_X(particles)
evt_vars[f'ALL{suffix}_FIRSTMEAS_Y[part_indx]'] = ALLPART_FIRSTMEAS_Y(particles)
evt_vars[f'ALL{suffix}_FIRSTMEAS_Z[part_indx]'] = ALLPART_FIRSTMEAS_Z(particles)
evt_vars[f'ALL{suffix}_FIRSTMEAS_TX[part_indx]'] = ALLPART_FIRSTMEAS_TX(particles)
evt_vars[f'ALL{suffix}_FIRSTMEAS_TY[part_indx]'] = ALLPART_FIRSTMEAS_TY(particles)
evt_vars[f'ALL{suffix}_FIRSTMEAS_QOP[part_indx]'] = ALLPART_FIRSTMEAS_QOP(particles)
evt_vars[f'ALL{suffix}_FIRSTMEAS_XERR[part_indx]'] = ALLPART_FIRSTMEAS_STATEERR(particles, 0)
evt_vars[f'ALL{suffix}_FIRSTMEAS_YERR[part_indx]'] = ALLPART_FIRSTMEAS_STATEERR(particles, 1)
evt_vars[f'ALL{suffix}_FIRSTMEAS_TXERR[part_indx]'] = ALLPART_FIRSTMEAS_STATEERR(particles, 2)
evt_vars[f'ALL{suffix}_FIRSTMEAS_TYERR[part_indx]'] = ALLPART_FIRSTMEAS_STATEERR(particles, 3)
evt_vars[f'ALL{suffix}_FIRSTMEAS_QOPERR[part_indx]'] = ALLPART_FIRSTMEAS_STATEERR(particles, 4)
evt_vars[f'ALL{suffix}_FIRSTMEAS_X[{index_name}]'] = ALLPART_FIRSTMEAS_X(particles)
evt_vars[f'ALL{suffix}_FIRSTMEAS_Y[{index_name}]'] = ALLPART_FIRSTMEAS_Y(particles)
evt_vars[f'ALL{suffix}_FIRSTMEAS_Z[{index_name}]'] = ALLPART_FIRSTMEAS_Z(particles)
evt_vars[f'ALL{suffix}_FIRSTMEAS_TX[{index_name}]'] = ALLPART_FIRSTMEAS_TX(particles)
evt_vars[f'ALL{suffix}_FIRSTMEAS_TY[{index_name}]'] = ALLPART_FIRSTMEAS_TY(particles)
evt_vars[f'ALL{suffix}_FIRSTMEAS_QOP[{index_name}]'] = ALLPART_FIRSTMEAS_QOP(particles)
evt_vars[f'ALL{suffix}_FIRSTMEAS_XERR[{index_name}]'] = ALLPART_FIRSTMEAS_STATEERR(particles, 0)
evt_vars[f'ALL{suffix}_FIRSTMEAS_YERR[{index_name}]'] = ALLPART_FIRSTMEAS_STATEERR(particles, 1)
evt_vars[f'ALL{suffix}_FIRSTMEAS_TXERR[{index_name}]'] = ALLPART_FIRSTMEAS_STATEERR(particles, 2)
evt_vars[f'ALL{suffix}_FIRSTMEAS_TYERR[{index_name}]'] = ALLPART_FIRSTMEAS_STATEERR(particles, 3)
evt_vars[f'ALL{suffix}_FIRSTMEAS_QOPERR[{index_name}]'] = ALLPART_FIRSTMEAS_STATEERR(particles, 4)
#P -> MCP
evt_vars[f'ALL{suffix}_TRUE_ID[part_indx]'] = ALLPART_TRUE_ID(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_KEY[part_indx]'] = ALLPART_TRUE_KEY(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_PX[part_indx]'] = ALLPART_TRUE_PX(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_PY[part_indx]'] = ALLPART_TRUE_PY(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_PZ[part_indx]'] = ALLPART_TRUE_PZ(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_ENERGY[part_indx]'] = ALLPART_TRUE_ENERGY(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_MASS[part_indx]'] = ALLPART_TRUE_MASS(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MC_ISPROMPT[part_indx]'] = ALLPART_MC_ISPROMPT(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_PV_X[part_indx]'] = ALLPART_TRUE_PV_X(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_PV_Y[part_indx]'] = ALLPART_TRUE_PV_Y(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_PV_Z[part_indx]'] = ALLPART_TRUE_PV_Z(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_ID[{index_name}]'] = ALLPART_TRUE_ID(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_KEY[{index_name}]'] = ALLPART_TRUE_KEY(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_PX[{index_name}]'] = ALLPART_TRUE_PX(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_PY[{index_name}]'] = ALLPART_TRUE_PY(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_PZ[{index_name}]'] = ALLPART_TRUE_PZ(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_ENERGY[{index_name}]'] = ALLPART_TRUE_ENERGY(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_MASS[{index_name}]'] = ALLPART_TRUE_MASS(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MC_ISPROMPT[{index_name}]'] = ALLPART_MC_ISPROMPT(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_PV_X[{index_name}]'] = ALLPART_TRUE_PV_X(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_PV_Y[{index_name}]'] = ALLPART_TRUE_PV_Y(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_PV_Z[{index_name}]'] = ALLPART_TRUE_PV_Z(particles, MCTRUTH_parts)
#P -> LONG LIVED ANCESTOR
evt_vars[f'ALL{suffix}_LONGLIVED_ANCESTOR_ID[part_indx]'] = ALLPART_LONGLIVED_ANCESTOR_ID(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_LONGLIVED_ANCESTOR_KEY[part_indx]'] = ALLPART_LONGLIVED_ANCESTOR_KEY(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_ORGIN_KEY[part_indx]'] = ALLPART_TRUE_ORGIN_KEY(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_ORGIN_X[part_indx]'] = ALLPART_TRUE_ORGIN_X(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_ORGIN_Y[part_indx]'] = ALLPART_TRUE_ORGIN_Y(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_ORGIN_Z[part_indx]'] = ALLPART_TRUE_ORGIN_Z(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MOTHER_1_ID[part_indx]'] = ALLPART_MOTHER_ID(particles, 1, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MOTHER_1_KEY[part_indx]'] = ALLPART_MOTHER_KEY(particles, 1, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MOTHER_2_ID[part_indx]'] = ALLPART_MOTHER_ID(particles, 2, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MOTHER_2_KEY[part_indx]'] = ALLPART_MOTHER_KEY(particles, 2, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MOTHER_3_ID[part_indx]'] = ALLPART_MOTHER_ID(particles, 3, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MOTHER_3_KEY[part_indx]'] = ALLPART_MOTHER_KEY(particles, 3, MCTRUTH_parts)
#evt_vars[f'ALL{suffix}_HISTORY[part_indx]'] = ALLPART_HISTORY(particles)
#evt_vars[f'ALL{suffix}_FLAG[part_indx]'] = ALLPART_FLAG(particles)
evt_vars[f'ALL{suffix}_LONGLIVED_ANCESTOR_ID[{index_name}]'] = ALLPART_LONGLIVED_ANCESTOR_ID(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_LONGLIVED_ANCESTOR_KEY[{index_name}]'] = ALLPART_LONGLIVED_ANCESTOR_KEY(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_ORGIN_KEY[{index_name}]'] = ALLPART_TRUE_ORGIN_KEY(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_ORGIN_X[{index_name}]'] = ALLPART_TRUE_ORGIN_X(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_ORGIN_Y[{index_name}]'] = ALLPART_TRUE_ORGIN_Y(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_TRUE_ORGIN_Z[{index_name}]'] = ALLPART_TRUE_ORGIN_Z(particles, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MOTHER_1_ID[{index_name}]'] = ALLPART_MOTHER_ID(particles, 1, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MOTHER_1_KEY[{index_name}]'] = ALLPART_MOTHER_KEY(particles, 1, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MOTHER_2_ID[{index_name}]'] = ALLPART_MOTHER_ID(particles, 2, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MOTHER_2_KEY[{index_name}]'] = ALLPART_MOTHER_KEY(particles, 2, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MOTHER_3_ID[{index_name}]'] = ALLPART_MOTHER_ID(particles, 3, MCTRUTH_parts)
evt_vars[f'ALL{suffix}_MOTHER_3_KEY[{index_name}]'] = ALLPART_MOTHER_KEY(particles, 3, MCTRUTH_parts)
#evt_vars[f'ALL{suffix}_HISTORY[{index_name}]'] = ALLPART_HISTORY(particles)
#evt_vars[f'ALL{suffix}_FLAG[{index_name}]'] = ALLPART_FLAG(particles)
def get_user_algs(decay_channel):
"""
@@ -317,9 +317,9 @@ def get_user_algs(decay_channel):
evt_vars['ALLPV_ZERR[pv_indx]'] = ALLPV_POSERR(pvs, 2)
#get long/down/up type particle variables
get_all_part_vars(evt_vars, long_particles, pvs, MCTRUTH_long_parts, suffix="LONGPART", is_long=True)
get_all_part_vars(evt_vars, down_particles, pvs, MCTRUTH_down_parts, suffix="DOWNPART")
get_all_part_vars(evt_vars, up_particles, pvs, MCTRUTH_up_parts, suffix="UPPART")
get_all_part_vars(evt_vars, long_particles, pvs, MCTRUTH_long_parts, index_name='long_indx', suffix="LONGPART", is_long=True)
get_all_part_vars(evt_vars, down_particles, pvs, MCTRUTH_down_parts, index_name='down_indx', suffix="DOWNPART")
get_all_part_vars(evt_vars, up_particles, pvs, MCTRUTH_up_parts, index_name='up_indx', suffix="UPPART")
#get Bcand variables
evt_vars['ALLB_ID[bcand_indx]'] = ALLB_ID(Bcands)
@@ -357,7 +357,7 @@ def get_user_algs(decay_channel):
##evt_vars['ALLB_TRUE_PI2[bcand_indx]'] = ALLB_TRUE_CHILD112_ID(Bcands, MCTRUTH_Bs) #pi
##all basic particles of Bcands
get_all_part_vars(evt_vars, all_basics, pvs, MCTRUTH_Bs, suffix="BASICS")
get_all_part_vars(evt_vars, all_basics, pvs, MCTRUTH_Bs, index_name='basics_indx', suffix="BASICS")
#get velo tracks variables
evt_vars['ALLVELOTRACKS_ETA[velo_trk_indx]'] = ALLVELOTRACKS_ETA(velo_tracks)
Loading