def beginLoop(self, setup):
super(TreeProducerBDT, self).beginLoop(setup)
self.rootfileS = TFile('/'.join([self.dirName,
'treeS.root']),
'recreate')
self.treeS = Tree( 'events', '')
self.treeS.var('weight', float)
bookParticle(self.treeS, 'jet')
bookParticle(self.treeS, 'softDropped_jet')
self.treeS.var('jet_tau1', float)
self.treeS.var('jet_tau2', float)
self.treeS.var('jet_tau3', float)
self.treeS.var('jet_tau32', float)
self.treeS.var('jet_tau31', float)
self.treeS.var('jet_tau21', float)
self.treeS.var('jet_flow15', float)
self.treeS.var('jet_flow25', float)
self.treeS.var('jet_flow35', float)
self.treeS.var('jet_flow45', float)
self.treeS.var('jet_flow55', float)
self.treeS.var('jet_nbs', float)
self.rootfileB = TFile('/'.join([self.dirName,
'treeB.root']),
'recreate')
self.treeB = Tree( 'events', '')
bookParticle(self.treeB, 'jet')
bookParticle(self.treeB, 'softDropped_jet')
self.treeB.var('jet_tau1', float)
self.treeB.var('jet_tau2', float)
self.treeB.var('jet_tau3', float)
self.treeB.var('jet_tau32', float)
self.treeB.var('jet_tau31', float)
self.treeB.var('jet_tau21', float)
self.treeB.var('jet_flow15', float)
self.treeB.var('jet_flow25', float)
self.treeB.var('jet_flow35', float)
self.treeB.var('jet_flow45', float)
self.treeB.var('jet_flow55', float)
self.treeB.var('jet_nbs', float)
def beginLoop(self, setup):
super(HiggsTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree(self.cfg_ana.tree_name, self.cfg_ana.tree_title)
bookZed(self.tree, 'ztomumu')
bookZed(self.tree, 'higgstojj')
def beginLoop(self, setup):
super(TreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName,
'tree.root']),
'recreate')
self.tree = Tree( 'events', '')
self.tree.var('weight', float)
#self.tree.var('met', float)
bookParticle(self.tree, 'higgs')
bookParticle(self.tree, 'zed1')
bookParticle(self.tree, 'zed2')
bookParticle(self.tree, 'l1')
bookParticle(self.tree, 'l2')
bookParticle(self.tree, 'l3')
bookParticle(self.tree, 'l4')
bookMet(self.tree, 'met')
self.tree.var('4mu', int)
self.tree.var('4e', int)
self.tree.var('2mu2e', int)
self.tree.var('nljets', float)
self.tree.var('nbjets', float)
self.tree.var('njets', float)
self.tree.var('nleptons', float)
self.tree.var('nextraleptons', float)
def beginLoop(self, setup):
super(TreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
self.tree.var('weight', float)
self.tree.var('njets', float)
self.tree.var('nleptons', float)
bookParticle(self.tree, 'l1')
bookParticle(self.tree, 'l2')
bookParticle(self.tree, 'l3')
bookParticle(self.tree, 'l4')
bookParticle(self.tree, 'j1')
bookParticle(self.tree, 'j2')
bookParticle(self.tree, 'j3')
bookParticle(self.tree, 'j4')
bookParticle(self.tree, 'gj1')
bookParticle(self.tree, 'gj2')
bookParticle(self.tree, 'gl1')
bookParticle(self.tree, 'gl2')
bookParticle(self.tree, 'gv1')
bookParticle(self.tree, 'gv2')
bookMet(self.tree, 'met')
def beginLoop(self, setup):
super(LeptonTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree(self.cfg_ana.tree_name, self.cfg_ana.tree_title)
bookLepton(self.tree, 'lep1')
bookLepton(self.tree, 'lep2')
def beginLoop(self, setup):
super(TreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
self.tree.var('weight', float)
float_vars = []
int_vars = []
self.book_container_jet("pfjets02", 10, True)
# self.book_container_jet("pfjets04")
self.book_container_jet("pfjets08", 5, False)
self.book_container_jet("pfjets04_btag", 8, False)
self.book_container("leptons")
for var in float_vars:
self.tree.var(var, float)
for var in int_vars:
self.tree.var(var, float)
bookMet(self.tree, 'met')
bookParticle(self.tree, 'gen_fcnc_t')
bookParticle(self.tree, 'gen_fcnc_higgs')
bookParticle(self.tree, 'gen_fcnc_u')
bookParticle(self.tree, 'gen_fcnc_b1')
bookParticle(self.tree, 'gen_fcnc_b2')
bookParticle(self.tree, 'gen_sm_t')
self.tree.var('delta_R', float)
def beginLoop(self, setup):
super(TreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
self.tree.var('weight', float)
float_vars = [
'bJets_N', 'lJets_N', "fatJets_N", "bFatJets_N", "bFatJets_N_Parts"
]
int_vars = []
bjets_vars = ["bfatjet_1", "bfatjet_2"]
for part in bjets_vars:
self.book_bfatjet(part)
for var in float_vars:
self.tree.var(var, float)
for var in int_vars:
self.tree.var(var, float)
bookMet(self.tree, 'met')
bookParticle(self.tree, 'gen_fcnc_t')
bookParticle(self.tree, 'gen_fcnc_higgs')
bookParticle(self.tree, 'gen_fcnc_u')
bookParticle(self.tree, 'gen_fcnc_b1')
bookParticle(self.tree, 'gen_fcnc_b2')
bookParticle(self.tree, 'gen_sm_t')
self.tree.var('delta_R', float)
def beginLoop(self, setup):
super(TreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
self.tree.var('weight', float)
self.tree.var('weight_1tagex', float)
self.tree.var('weight_2tagex', float)
self.tree.var('weight_1tagin', float)
bookParticle(self.tree, 'Jet1_pf04')
bookParticle(self.tree, 'Jet2_pf04')
self.tree.var('Mj1j2_pf04', float)
self.tree.var('Mj1j2_pf04_MetCorr', float)
self.tree.var('Mj1j2_pf04_MetCorr2', float)
self.tree.var('mt', float)
self.tree.var('mr', float)
self.tree.var('mr2', float)
self.tree.var('mr3', float)
self.tree.var('dr', float)
self.tree.var('dphi', float)
self.tree.var('dphi_met', float)
self.tree.var('ntau', int)
bookMet(self.tree, 'met')
def beginLoop(self, setup):
super(SimpleTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'simple_tree.root']),
'recreate')
self.tree = Tree(self.cfg_ana.tree_name, self.cfg_ana.tree_title)
self.tree.var('test_variable')
self.tree.var('test_variable_random')
def beginLoop(self, setup):
super(TreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
self.tree.var('weight', float)
self.tree.var('zprime_y', float)
bookParticle(self.tree, 'zprime_ele')
bookParticle(self.tree, 'zprime_muon')
bookParticle(self.tree, 'jet1')
bookParticle(self.tree, 'jet2')
bookParticle(self.tree, 'jet3')
bookLepton(self.tree, 'lep1', pflow=False)
bookLepton(self.tree, 'lep2', pflow=False)
bookLepton(self.tree, 'lep3', pflow=False)
bookLepton(self.tree, 'lep1_gen_1', pflow=False)
bookLepton(self.tree, 'lep2_gen_1', pflow=False)
bookLepton(self.tree, 'lep1_gen_23', pflow=False)
bookLepton(self.tree, 'lep2_gen_23', pflow=False)
bookMet(self.tree, 'met')
def beginLoop(self, setup):
super(GlobalEventTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
bookJet(self.tree, 'sum_all')
bookJet(self.tree, 'sum_all_gen')
def beginLoop(self, setup):
super(TreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
self.tree.var('weight', float)
#self.tree.var('met', float)
self.tree.var('nljets', float)
self.tree.var('nbjets', float)
self.tree.var('njets', float)
self.tree.var('is_sf', float)
self.tree.var('is_of', float)
self.tree.var('mt', float)
self.tree.var('mr', float)
self.tree.var('higgs_pt', float)
self.tree.var('dphi_ll', float)
self.tree.var('dphi_llmet', float)
bookParticle(self.tree, 'l1')
bookParticle(self.tree, 'l2')
bookParticle(self.tree, 'll')
bookMet(self.tree, 'met')
def beginLoop(self, setup):
super(TreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
self.tree.var('weight', float)
self.tree.var('missingET', float)
self.tree.var('rapiditySeparation_calo04', float)
self.tree.var('pseudorapiditySeparation_calo04', float)
self.tree.var('transverseMomentumAsymmetry_calo04', float)
self.tree.var('rapiditySeparation_pf04', float)
self.tree.var('pseudorapiditySeparation_pf04', float)
self.tree.var('transverseMomentumAsymmetry_pf04', float)
bookParticle(self.tree, 'Jet1_pf04')
bookParticle(self.tree, 'Jet2_pf04')
bookParticle(self.tree, 'Jet1_calo04')
bookParticle(self.tree, 'Jet2_calo04')
self.tree.var('Mj1j2_pf04', float)
self.tree.var('Mj1j2_calo04', float)
def beginLoop(self, setup):
super(QQTauTauTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
if hasattr(self.cfg_ana, 'recoil'):
bookParticle(self.tree, 'recoil')
if hasattr(self.cfg_ana, 'zeds'):
bookZed(self.tree, 'zed')
self.taggers = ['b', 'bmatch', 'bfrac']
## for label in self.cfg_ana.leptons:
## bookIsoParticle(self.tree, '{}_1'.format(label), pflow=True)
## bookIsoParticle(self.tree, '{}_2'.format(label), pflow=True)
for label in self.cfg_ana.particles:
bookParticle(self.tree, label)
var(self.tree, 'n_' + label)
for label in self.cfg_ana.jet_collections:
bookJet(self.tree, '{}_1'.format(label), self.taggers)
bookJet(self.tree, '{}_2'.format(label), self.taggers)
for label in self.cfg_ana.resonances:
iso = False
if 'zed' in label and not 'qq' in label:
iso = True
bookResonanceWithLegs(self.tree, label, iso)
bookIso(self.tree, 'besttaus_1_iso')
bookIso(self.tree, 'besttaus_2_iso')
bookResonanceWithLegs(self.tree, 'genboson1')
bookResonanceWithLegs(self.tree, 'genboson2')
for label in self.cfg_ana.misenergy:
bookParticle(self.tree, label)
var(self.tree, 'n_nu')
var(self.tree, 'beta4_chi2')
def beginLoop(self, setup):
super(TreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName,
'tree.root']),
'recreate')
self.tree = Tree( 'events', '')
self.tree.var('weight', float)
self.tree.var('missingET', float)
self.tree.var('numberOfElectrons', int)
self.tree.var('numberOfMuons', int)
self.tree.var('numberOfFatJets', int)
self.tree.var('Jet1_tau1', float)
self.tree.var('Jet1_tau2', float)
self.tree.var('Jet1_tau3', float)
self.tree.var('Jet2_tau1', float)
self.tree.var('Jet2_tau2', float)
self.tree.var('Jet2_tau3', float)
self.tree.var('Jet1_tau32', float)
self.tree.var('Jet1_tau31', float)
self.tree.var('Jet1_tau21', float)
self.tree.var('Jet2_tau32', float)
self.tree.var('Jet2_tau31', float)
self.tree.var('Jet2_tau21', float)
bookParticle(self.tree, 'Jet1')
bookParticle(self.tree, 'Jet2')
bookParticle(self.tree, 'Jet3')
bookParticle(self.tree, 'Jet4')
self.tree.var('rapiditySeparation', float)
self.tree.var('transverseMomentumAsymmetry', float)
self.tree.var('topJetMassDifference', float)
self.tree.var('Jet1_trimmedProngMaxPtRatio', float)
self.tree.var('Jet1_trimmedProngMinPtRatio', float)
self.tree.var('Jet2_trimmedProngMaxPtRatio', float)
self.tree.var('Jet2_trimmedProngMinPtRatio', float)
bookParticle(self.tree, 'softDroppedJet1')
bookParticle(self.tree, 'softDroppedJet2')
bookParticle(self.tree, 'softDroppedJet3')
bookParticle(self.tree, 'softDroppedJet4')
bookParticle(self.tree, 'smallJet1')
bookParticle(self.tree, 'smallJet2')
bookParticle(self.tree, 'smallJet3')
bookParticle(self.tree, 'smallJet4')
bookParticle(self.tree, 'Electron1')
bookParticle(self.tree, 'Electron2')
bookParticle(self.tree, 'Muon1')
bookParticle(self.tree, 'Muon2')
self.tree.var('BDTvariable_qcd', float)
self.tree.var('zPrimeReconstructedMass', float)
def beginLoop(self, setup):
super(TauTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName,
'taus.root']),
'recreate')
self.tree = Tree( 'events', '')
bookJet(self.tree, 'tau')
bookJet(self.tree, 'tau_match')
def beginLoop(self, setup):
super(IsoParticleTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName,
'tree.root']),
'recreate')
self.tree = Tree( self.cfg_ana.tree_name,
self.cfg_ana.tree_title )
bookIsoParticle(self.tree, 'ptc')
def beginLoop(self, setup):
super(SimpleTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
bookParticle(self.tree, 'recoil_gen')
bookParticle(self.tree, 'recoil_visible_gen')
bookParticle(self.tree, 'recoil_papas')
bookParticle(self.tree, 'recoil_visible_papas')
def beginLoop(self, setup):
super(ZTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'mytree.root']),
'recreate')
self.tree = Tree('events', '')
bookParticle(self.tree, 'zed')
#bookParticle(self.tree, 'recoil')
bookJet(self.tree, 'zed_1')
bookJet(self.tree, 'zed_2')
def beginLoop(self, setup):
super(ConeTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'jet_tree.root']),
'recreate')
self.tree = Tree(self.cfg_ana.tree_name, self.cfg_ana.tree_title)
bookParticle(self.tree, 'particle')
var(self.tree, 'dR')
var(self.tree, 'is_gen_matched')
var(self.tree, 'iEv')
def beginLoop(self, setup):
super(TreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
self.taggers = 'mu'
bookParticle(self.tree, 'higgs')
bookParticle(self.tree, 'muon1')
bookParticle(self.tree, 'muon2')
var(self.tree, 'lenmu')
def beginLoop(self, setup):
super(TreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
self.taggers = [
'b', 'bmatch', 'bfrac', 'eff', 'btag', 'ctag', 'udsgtag',
'b_pvprob'
]
bookJet(self.tree, 'jet1', self.taggers)
bookJet(self.tree, 'jet2', self.taggers)
bookJet(self.tree, 'scaledjet1', self.taggers)
bookJet(self.tree, 'scaledjet2', self.taggers)
# bookJet(self.tree, 'jet3', self.taggers)
# bookJet(self.tree, 'jet4', self.taggers)
bookJet(self.tree, 'genjet1', self.taggers)
bookJet(self.tree, 'genjet2', self.taggers)
# bookJet(self.tree, 'genjet3', self.taggers)
# bookJet(self.tree, 'genjet4', self.taggers)
bookParticle(self.tree, 'genmisenergy')
bookParticle(self.tree, 'misenergy')
bookParticle(self.tree, 'higgs')
bookParticle(self.tree, 'higgsnosf')
bookParticle(self.tree, 'zed')
bookParticle(self.tree, 'zed_parton1')
bookParticle(self.tree, 'zed_parton2')
bookParticle(self.tree, 'higgs_parton1')
bookParticle(self.tree, 'higgs_parton2')
var(self.tree, 'mmiss_rescaled')
var(self.tree, 'acop')
var(self.tree, 'mmiss_rescaled2')
var(self.tree, 'jet1res')
var(self.tree, 'jet2res')
var(self.tree, 'absjet1res')
var(self.tree, 'absjet2res')
var(self.tree, 'cross')
var(self.tree, 'ctracks')
var(self.tree, 'mvis')
var(self.tree, 'genmvis')
var(self.tree, 'emratio')
# var(self.tree, 'quarkdR')
# var(self.tree, 'jetdR')
var(self.tree, 'alpha')
var(self.tree, 'beta')
var(self.tree, 'pLges')
var(self.tree, 'pTges')
var(self.tree, 'scaling_fac')
var(self.tree, 'scaling_fac2')
var(self.tree, 'n_jets')
var(self.tree, 'n_genjets')
var(self.tree, 'bquarks')
var(self.tree, 'n_genptc_23')
def beginLoop(self, setup):
super(PapasTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName,'jet_tree.root']),
'recreate')
self.tree = Tree( self.cfg_ana.tree_name,
self.cfg_ana.tree_title )
bookJet(self.tree, 'jet1')
bookJet(self.tree, 'jet1_rec')
bookJet(self.tree, 'jet2')
bookJet(self.tree, 'jet2_rec')
var(self.tree, 'iEv')
def test_fill(self):
fi = TFile('tree.root', 'RECREATE')
tr = Tree('test_tree', 'A test tree')
tr.var('a')
tr.var('b')
tr.fill('a', 3)
tr.fill('a', 4)
tr.fill('b', 5)
tr.tree.Fill()
fi.Write()
fi.Close()
def create_tree(filename="test_tree.root"):
outfile = TFile(filename, 'recreate')
tree = Tree('test_tree', 'A test tree')
tree.var('var1')
for i in range(100):
tree.fill('var1', i)
tree.tree.Fill()
print 'creating a tree', tree.tree.GetName(),\
tree.tree.GetEntries(), 'entries in',\
outfile.GetName()
outfile.Write()
def beginLoop(self, setup):
super(TTbarTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName,
'tree.root']),
'recreate')
self.tree = Tree( 'events', '')
bookParticle(self.tree, 'jet1')
bookParticle(self.tree, 'jet2')
bookParticle(self.tree, 'jet3')
bookParticle(self.tree, 'm3')
bookMet(self.tree, 'met')
bookLepton(self.tree, 'lepton')
def beginLoop(self, setup):
super(GenTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
self.tree.var('weight', float)
self.tree.var('nh', float)
self.tree.var('drhh', float)
bookParticle(self.tree, 'h1')
bookParticle(self.tree, 'h2')
bookParticle(self.tree, 'hh')
def beginLoop(self, setup):
super(JetTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'jet_tree.root']),
'recreate')
self.tree = Tree(self.cfg_ana.tree_name, self.cfg_ana.tree_title)
bookJet(self.tree, 'jet1')
bookJet(self.tree, 'jet1_gen')
bookJet(self.tree, 'jet2')
bookJet(self.tree, 'jet2_gen')
var(self.tree, 'event')
var(self.tree, 'lumi')
var(self.tree, 'run')
def beginLoop(self, setup):
super(HTo4lTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('events', '')
bookParticle(self.tree, 'zed1')
bookParticle(self.tree, 'zed2')
bookLepton(self.tree, 'zed1_lep1')
bookLepton(self.tree, 'zed1_lep2')
bookLepton(self.tree, 'zed2_lep1')
bookLepton(self.tree, 'zed2_lep2')
bookParticle(self.tree, 'higgs')
def beginLoop(self, setup):
super(GenRecTreeProducer, self).beginLoop(setup)
self.rootfile = TFile('/'.join([self.dirName, 'tree.root']),
'recreate')
self.tree = Tree('RecGen', '')
for i in range(5):
bookParticle(self.tree, 'gen_ptc{j}'.format(j=i))
bookParticle(self.tree, 'rec_ptc{j}'.format(j=i))
var(self.tree, 'nb_gen_ptc')
var(self.tree, 'nb_rec_ptc')
var(self.tree, 'event_gen_E')
var(self.tree, 'event_rec_E')
var(self.tree, 'iev')
