예제 #1
0
    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)
예제 #2
0
 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()