Пример #1
0
 def run(self):
     queueLock.acquire()
     (sampleName, inputFilePath) = workQueue.get()
     queueLock.release()
     print("%s: %s" % (sampleName, time.ctime(time.time())))
     print("%s: Searching for input files in path = '%s'" %
           (sampleName, inputFilePath))
     numEvents = 0
     start_time = time.time()
     end_time = None
     if os.path.exists(inputFilePath):
         inputFileNames = getInputFileNames(inputFilePath)
         print("%s: Found %i input files." %
               (sampleName, len(inputFileNames)))
         for idx, inputFileName in enumerate(inputFileNames):
             if (idx + 1) % 100 == 0:
                 print("%s: Processing %ith input file..." % (sampleName,
                                                              (idx + 1)))
             inputFile = ROOT.TFile.Open(inputFileName, 'read')
             if not inputFile or inputFile.IsZombie():
                 print(
                     "%s: Failed to open input file = '%s' --> skipping !!"
                     % (sampleName, inputFileName))
                 continue
             events_tree = inputFile.Get(events_treeName)
             numEvents += events_tree.GetEntries()
         end_time = time.time()
         print("%s: %s" % (sampleName, time.ctime(time.time())))
         print("%s: Finished processing %i input files." %
               (sampleName, len(inputFileNames)))
         print(" (numEvents = %i, run-time = %s seconds)" %
               (numEvents, end_time - start_time))
     else:
         print("%s: Path = '%s' does not exist --> skipping !!" %
               (sampleName, inputFilePath))
         end_time = time.time()
     self.numEvents[sampleName] = numEvents
     self.start_time[sampleName] = start_time
     self.end_time[sampleName] = end_time
     workQueue.task_done()
inputFileNames = [
    'file:/hdfs/cms/store/user/rdewanje/VBFHToTauTau_M125_14TeV_powheg_pythia8_correctedGridpack_tuneCP5/HLTConfig_VBFHToTauTau_M125_14TeV_powheg_pythia8_correctedGridpack_tuneCP5_wOfflineVtx_wDeepTau4/200826_185528/0000/step3_RAW2DIGI_RECO_1.root'
]
processName = "qqH_htt"
hlt_srcVertices = 'offlinePrimaryVertices'
hlt_algorithm = "hps"
hlt_isolation_maxDeltaZOption = "primaryVertex"
hlt_isolation_minTrackHits = 8
outputFileName = "deepTauTest_%s.root" % processName

#--------------------------------------------------------------------------------
# set input files
if inputFilePath:
    from HLTrigger.TallinnHLTPFTauAnalyzer.tools.jobTools import getInputFileNames
    print("Searching for input files in path = '%s'" % inputFilePath)
    inputFileNames = getInputFileNames(inputFilePath)
    print("Found %i input files." % len(inputFileNames))
    process.source.fileNames = cms.untracked.vstring(inputFileNames)
else:
    print("Processing %i input files: %s" %
          (len(inputFileNames), inputFileNames))
    process.source.fileNames = cms.untracked.vstring(inputFileNames)
#--------------------------------------------------------------------------------

from Configuration.AlCa.GlobalTag import GlobalTag
process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:phase2_realistic', '')

process.analysisSequence = cms.Sequence()

process.load("PhysicsTools.HepMCCandAlgos.genParticles_cfi")
process.analysisSequence += process.genParticles