def __init__(self, cfg_ana, cfg_comp, looperName): super(H2TauTauEventSorter, self).__init__(cfg_ana, cfg_comp, looperName) self.regions = H2TauTauRegions(self.cfg_ana) self.output = H2TauTauOutput(self.dirName, self.regions, self.cfg_ana.leg1, self.cfg_ana.leg2) self.treeFile = TFile('/'.join([self.dirName, 'tree.root']), 'recreate') self.tree = Tree('tree', 'flat ntuple') self.tree.addVar('float', 'vismass') self.tree.addVar('float', 'svfitmass') self.tree.addVar('float', 'l1_pt') self.tree.addVar('float', 'l1_eta') self.tree.addVar('float', 'l2_pt') self.tree.addVar('float', 'l2_eta') self.tree.addVar('float', 'j1_pt') self.tree.addVar('float', 'j1_eta') self.tree.addVar('float', 'j2_pt') self.tree.addVar('float', 'j2_eta') self.tree.book() self.fakes = False if hasattr(self.cfg_comp, 'fakes') and \ self.cfg_comp.fakes is True : self.fakes = True # fakedir = self.dirName.replace('DYJets', 'DYJets_Fakes') fakedir = '/'.join([self.looperName, 'Fakes', self.name]) mkdir_p(fakedir) self.outputFakes = H2TauTauOutput(fakedir, self.regions, self.cfg_ana.leg1, self.cfg_ana.leg2)
def __init__(self, cfg_ana, cfg_comp, looperName): super(TreeAnalyzer, self).__init__(cfg_ana, cfg_comp, looperName) fileName = '/'.join([self.dirName, self.name + '_tree.root']) self.file = TFile(fileName, 'recreate') self.tree = Tree(self.name, self.name) self.declareVariables()