예제 #1
0
파일: skim.py 프로젝트: dabercro/MonoX
    print tree.GetEntries()

    if sname == snames[0]:
        if obj == 'muon':
            treeType = ROOT.kDimuon
        elif obj == 'electron':
            treeType = ROOT.kDielectron
        elif obj == 'met':
            treeType = ROOT.kSingleElectron
        elif sname.startswith('sph-'):
            treeType = ROOT.kDiphoton
        elif sname.startswith('sel-'):
            treeType = ROOT.kElectronPhoton
        elif sname.startswith('smu-'):
            treeType = ROOT.kMuonPhoton
        elif sname.startswith('jht-'):
            treeType = ROOT.kJetHT
            
#        suffix = sname[:sname.find('-')]
        suffix = sname

        # TEMPORARY
        st19 = (sample.book == 'filefi/045')

outDir = config.histDir + '/trigger'
if not os.path.exists(outDir):
    os.makedirs(outDir)

ROOT.skim(tree, treeType, outDir + '/trigger_' + suffix + '_' + obj + '.root', st19, 50000000)
예제 #2
0
파일: skim.py 프로젝트: blallen/MitMonoX
ROOT.gROOT.LoadMacro(thisdir + '/Skimmer.cc+')

# available object - sample - skim type configurations
configs = {
    'photon': {'sph': ROOT.kDiphoton, 'sel': ROOT.kElectronPhoton, 'smu': ROOT.kMuonPhoton, 'jht': ROOT.kJetHT},
    'muon': {'smu': ROOT.kDimuon},
    'electron': {'sel': ROOT.kDielectron},
    'met': {'sel': ROOT.kElectronMET}
}

tree = ROOT.TChain('events')

for sample in samples:
    try:
        treeType = configs[obj][sample.name[:sample.name.find('-')]]
    except KeyError:
        print 'treeType for', obj, sample.name, 'not found'
        continue

    treePath = config.ntuplesDir + '/' + sample.book + '/' + sample.fullname + '/*.root'
    print sample.name, "adding files from", treePath
    tree.Add(treePath)

    print tree.GetEntries(), 'entries'
            
    ROOT.skim(tree, treeType, '/tmp/trigger_' + sample.name + '_' + obj + '.root')

    shutil.move('/tmp/trigger_' + sample.name + '_' + obj + '.root', config.histDir + '/trigger')