def exec_event(self):
# calculate nvetex info
nvtx_prim = 0
nvtx_pile = 0
for i in range(0,self.ch.vxp_n):
if self.ch.vxp_type[i]==1 and self.ch.vxp_nTracks[i]>=4: nvtx_prim+=1
elif self.ch.vxp_type[i]==3 and self.ch.vxp_nTracks[i]>=2: nvtx_pile+=1
# Select Taus
taus = self.tau_selector.select()
for tau in taus:
self.resetBranches()
# set the event properties agains, since we clear for each tau
self.setBranch( 'averageIntPerXing', self.ch.averageIntPerXing )
self.setBranch( 'actualIntPerXing', self.ch.actualIntPerXing )
self.setBranch( 'NPrimaryVtx', nvtx_prim )
self.setBranch( 'NPileupVtx', nvtx_pile )
index = tau.index
# Writeout Variables
self.setBranch( 'tau_pt', self.ch.tau_pt[index] )
self.setBranch( 'tau_m', self.ch.tau_m[index] )
self.setBranch( 'tau_eta', self.ch.tau_eta[index] )
self.setBranch( 'tau_phi', self.ch.tau_phi[index] )
self.setBranch( 'tau_charge', self.ch.tau_charge[index] )
self.setBranch( 'tau_BDTEleScore', self.ch.tau_BDTEleScore[index] )
self.setBranch( 'tau_BDTJetScore', self.ch.tau_BDTJetScore[index] )
self.setBranch( 'tau_likelihood', self.ch.tau_likelihood[index] )
self.setBranch( 'tau_electronVetoLoose', self.ch.tau_electronVetoLoose[index] )
self.setBranch( 'tau_electronVetoMedium', self.ch.tau_electronVetoMedium[index] )
self.setBranch( 'tau_electronVetoTight', self.ch.tau_electronVetoTight[index] )
self.setBranch( 'tau_muonVeto', self.ch.tau_muonVeto[index] )
self.setBranch( 'tau_tauCutLoose', self.ch.tau_tauCutLoose[index] )
self.setBranch( 'tau_tauCutMedium', self.ch.tau_tauCutMedium[index] )
self.setBranch( 'tau_tauCutTight', self.ch.tau_tauCutTight[index] )
self.setBranch( 'tau_tauLlhLoose', self.ch.tau_tauLlhLoose[index] )
self.setBranch( 'tau_tauLlhMedium', self.ch.tau_tauLlhMedium[index] )
self.setBranch( 'tau_tauLlhTight', self.ch.tau_tauLlhTight[index] )
self.setBranch( 'tau_JetBDTSigLoose', self.ch.tau_JetBDTSigLoose[index] )
self.setBranch( 'tau_JetBDTSigMedium', self.ch.tau_JetBDTSigMedium[index] )
self.setBranch( 'tau_JetBDTSigTight', self.ch.tau_JetBDTSigTight[index] )
self.setBranch( 'tau_EleBDTLoose', self.ch.tau_EleBDTLoose[index] )
self.setBranch( 'tau_EleBDTMedium', self.ch.tau_EleBDTMedium[index] )
self.setBranch( 'tau_EleBDTTight', self.ch.tau_EleBDTTight[index] )
self.setBranch( 'tau_author', self.ch.tau_author[index] )
self.setBranch( 'tau_numTrack', self.ch.tau_numTrack[index] )
# do trigger matching
for trigger in self.triggers:
# check trigger configured
if not hasattr( self.ch, trigger): continue
# get event decision
passed_event = getattr(self.ch,trigger)
if not passed_event: continue
self.setBranch( trigger, 1 )
# get trigger match --> maybe make a common function to construct trigger containers
if not hasattr(self.ch,'trig_EF_tau_%s'%trigger):
print 'WARNING trying to access branch that doesnt exist: ', self.ch,'trig_EF_tau_%s'%trigger
continue
trigger_objects = getattr(self.ch,'trig_EF_tau_%s'%trigger)
matched = 0
for i in range(0,self.ch.trig_EF_tau_n):
if not trigger_objects[i]: continue
eftau = ParticleBase( i,
_pt = self.ch.trig_EF_tau_pt[i],
_eta = self.ch.trig_EF_tau_eta[i],
_phi = self.ch.trig_EF_tau_phi[i],
_m = self.ch.trig_EF_tau_m[i]
)
if eftau.DeltaR(tau)>self.max_dr: continue
matched = 1
break
self.setBranch( '%s_match'%trigger, matched )
# do trigger matching for emulated 1 triggers
for basename in self.emulated_1trigger_basenames:
# check trigger configured
if not hasattr( self.ch, basename): continue
indicies = self.getIndiciesPassedEmulated1Trigger( basename )
if len(indicies)<=0: continue
self.setBranch( '%s1'%basename, 1 )
# get trigger match --> maybe make a common function to construct trigger containers
matched = 0
for i in range(0,len(indicies)):
trig_index = indicies[i]
eftau = ParticleBase( trig_index,
_pt = self.ch.trig_EF_tau_pt[trig_index],
_eta = self.ch.trig_EF_tau_eta[trig_index],
_phi = self.ch.trig_EF_tau_phi[trig_index],
_m = self.ch.trig_EF_tau_m[trig_index]
)
if eftau.DeltaR(tau)>self.max_dr: continue
matched = 1
break
self.setBranch( '%s1_match'%basename, matched )
self.fill()
def select(self):
self.clear_particles()
# require TTree to be loaded
if not self.ch:
print 'TauSelector. Warning! No input TTree'
return
# construct loose muons
loose_muons = []
if self.veto_loose_muon:
for imu in range( 0,self.ch.mu_staco_n):
if not self.ch.mu_staco_loose[imu]: continue
loose_muons.append( ParticleBase( _index = imu,
_pt = self.ch.mu_staco_pt[imu],
_eta = self.ch.mu_staco_eta[imu],
_phi = self.ch.mu_staco_phi[imu],
_m = self.ch.mu_staco_m[imu]
) )
# Loop over taus and select candidates
for i in range(0,self.ch.tau_n):
p = ParticleBase( _index = i,
_pt = self.ch.tau_pt[i],
_eta = self.ch.tau_eta[i],
_phi = self.ch.tau_phi[i],
_m = self.ch.tau_m[i] )
if p.Pt() < self.min_pt : continue
if abs(p.Eta()) > self.max_eta: continue
if self.allowed_authors and not self.ch.tau_author[i] in self.allowed_authors: continue
if self.allowed_tracks and not self.ch.tau_numTrack[i] in self.allowed_tracks: continue
if self.nonzero_tracks and not self.ch.tau_numTrack[i] > 0 : continue
if self.req_unit_charge and not abs( self.ch_tau_charge[i] )==1: continue
if self.req_truth and not self.ch.tau_trueTauAssoc_matched[i]: continue
if self.req_cut_l and not self.ch.tau_tauCutLoose[i]: continue
if self.req_cut_m and not self.ch.tau_tauCutMedium[i]: continue
if self.req_cut_t and not self.ch.tau_tauCutTight[i]: continue
if self.req_llh_l and not self.ch.tau_tauLlhLoose[i]: continue
if self.req_llh_m and not self.ch.tau_tauLlhMedium[i]: continue
if self.req_llh_t and not self.ch.tau_tauLlhTight[i]: continue
if self.req_ecut_l and self.ch.tau_electronVetoLoose[i]: continue
if self.req_ecut_m and self.ch.tau_electronVetoMedium[i]: continue
if self.req_ecut_t and self.ch.tau_electronVetoTight[i]: continue
if self.req_ebdt_l and self.ch.tau_EleBDTLoose[i]: continue
if self.req_ebdt_m and self.ch.tau_EleBDTMedium[i]: continue
if self.req_ebdt_t and self.ch.tau_EleBDTTight[i]: continue
if self.min_bdt_ele_score and self.ch.tau_BDTEleScore[i] < self.min_bdt_ele_score: continue
if self.req_muon_veto and self.ch.tau_muonVeto[i]: continue
if self.min_bdt_jet_score and self.ch.tau_BDTJetScore[i] < self.min_bdt_jet_score: continue
if self.veto_loose_muon:
veto_tau = False
for muon in loose_muons:
if p.DeltaR(muon) < 0.2:
veto_tau = True
break
if veto_tau: continue
#added/modified to adapt the tau id patch
if self.recalculate_tauID:
# BDT is recalculated
patch = TauIDpatch(self.year)
pt = self.ch.tau_pt[i]
tracks = self.ch.tau_numTrack[i]
bdtscore = self.ch.tau_BDTJetScore[i]
if self.year == 2012:
#print 'recalculating year 2012 pT= %f MeV, Ntrk = %d, bdt = %f'% (pt, tracks, bdtscore)
self.decision_tau_JetBDTSigLoose, \
self.decision_tau_JetBDTSigMedium, \
self.decision_tau_JetBDTSigTight, \
= patch.passes_2012(pt, tracks, bdtscore)
#print self.ch.tau_JetBDTSigLoose[i], self.ch.tau_JetBDTSigMedium[i], self.ch.tau_JetBDTSigTight[i], \
#' vs ', self.decision_tau_JetBDTSigLoose, self.decision_tau_JetBDTSigMedium, self.decision_tau_JetBDTSigTight
elif self.year == 2011:
print "2011 is under contruction"
continue
else:
raise ValueError("TauSelector: No tauid defined for year %d" % year)
else:
self.decision_tau_JetBDTSigLoose = self.ch.tau_JetBDTSigLoose[i]
self.decision_tau_JetBDTSigMedium = self.ch.tau_JetBDTSigMedium[i]
self.decision_tau_JetBDTSigTight = self.ch.tau_JetBDTSigTight[i]
##
if self.req_bdt_l and not self.decision_tau_JetBDTSigLoose: continue
if self.req_bdt_m and not self.decision_tau_JetBDTSigMedium: continue
if self.req_bdt_t and not self.decision_tau_JetBDTSigTight: continue
self.add_particle(p)
#end of loop
self.sort_particles()
return self.get_particles()
