File indexing completed on 2024-11-06 06:06:53
0001 import FWCore.ParameterSet.Config as cms
0002 import os, glob
0003
0004 def getHeterogeneousRecHitsSource(pu):
0005 indir = '/eos/user/b/bfontana/Samples/'
0006 filename_suff = 'step3_ttbar_PU' + str(pu)
0007 fNames = [ 'file:' + x for x in glob.glob(os.path.join(indir, filename_suff + '*.root')) ]
0008 print(indir, filename_suff, pu, fNames)
0009 for _ in range(4):
0010 fNames.extend(fNames)
0011 if len(fNames)==0:
0012 print('Used globbing: ', glob.glob(os.path.join(indir, filename_suff + '*.root')))
0013 raise ValueError('No input files!')
0014
0015 keep = 'keep *'
0016 drop1 = 'drop CSCDetIdCSCALCTPreTriggerDigiMuonDigiCollection_simCscTriggerPrimitiveDigis__HLT'
0017 drop2 = 'drop HGCRecHitsSorted_HGCalRecHit_HGC*E*RecHits_*'
0018 return cms.Source("PoolSource",
0019 fileNames = cms.untracked.vstring(fNames),
0020 inputCommands = cms.untracked.vstring(keep, drop1, drop2),
0021 duplicateCheckMode = cms.untracked.string("noDuplicateCheck"))
0022
0023 enableGPU = True
0024 from Configuration.ProcessModifiers.gpu_cff import gpu
0025
0026 from RecoLocalCalo.HGCalRecProducers.HGCalRecHit_cfi import HGCalRecHit
0027 from SimCalorimetry.HGCalSimProducers.hgcalDigitizer_cfi import HGCAL_noise_fC, HGCAL_chargeCollectionEfficiencies
0028
0029
0030 from FWCore.ParameterSet.VarParsing import VarParsing
0031 F = VarParsing('analysis')
0032 F.register('PU',
0033 -1,
0034 F.multiplicity.singleton,
0035 F.varType.int,
0036 "Pileup to consider.")
0037 F.parseArguments()
0038
0039 process = cms.Process("TESTgpu", gpu) if enableGPU else cms.Process("TESTnongpu")
0040 process.load('FWCore.MessageService.MessageLogger_cfi')
0041 process.load('Configuration.StandardSequences.Services_cff')
0042 process.load('SimGeneral.HepPDTESSource.pythiapdt_cfi')
0043 process.load('Configuration.StandardSequences.MagneticField_cff')
0044
0045 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0046 process.load('Configuration.Geometry.GeometryExtendedRun4D49Reco_cff')
0047 process.load('HeterogeneousCore.CUDACore.ProcessAcceleratorCUDA_cfi')
0048 process.load('RecoLocalCalo.HGCalRecProducers.HGCalRecHit_cfi')
0049 process.load('SimCalorimetry.HGCalSimProducers.hgcalDigitizer_cfi')
0050
0051 process.TFileService = cms.Service("TFileService",
0052 fileName = cms.string("histo.root"),
0053 closeFileFast = cms.untracked.bool(True)
0054 )
0055
0056
0057 process.source = getHeterogeneousRecHitsSource(F.PU)
0058 process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(20) )
0059 process.options = cms.untracked.PSet(
0060 wantSummary = cms.untracked.bool( False ))
0061
0062 process.load('RecoLocalCalo.HGCalRecProducers.HeterogeneousEERecHitGPU_cfi')
0063 process.load('RecoLocalCalo.HGCalRecProducers.HeterogeneousEERecHitGPUtoSoA_cfi')
0064 process.load('RecoLocalCalo.HGCalRecProducers.HeterogeneousEERecHitFromSoA_cfi')
0065
0066 process.load('RecoLocalCalo.HGCalRecProducers.HeterogeneousHEFRecHitGPU_cfi')
0067 process.load('RecoLocalCalo.HGCalRecProducers.HeterogeneousHEFRecHitGPUtoSoA_cfi')
0068 process.load('RecoLocalCalo.HGCalRecProducers.HeterogeneousHEFRecHitFromSoA_cfi')
0069
0070 process.load('RecoLocalCalo.HGCalRecProducers.HeterogeneousHEBRecHitGPU_cfi')
0071 process.load('RecoLocalCalo.HGCalRecProducers.HeterogeneousHEBRecHitGPUtoSoA_cfi')
0072 process.load('RecoLocalCalo.HGCalRecProducers.HeterogeneousHEBRecHitFromSoA_cfi')
0073
0074
0075
0076
0077
0078
0079
0080
0081
0082
0083 process.ee_task = cms.Task( process.EERecHitGPUProd, process.EERecHitGPUtoSoAProd, process.EERecHitFromSoAProd )
0084 process.hef_task = cms.Task( process.HEFRecHitGPUProd, process.HEFRecHitGPUtoSoAProd, process.HEFRecHitFromSoAProd )
0085 process.heb_task = cms.Task( process.HEBRecHitGPUProd, process.HEBRecHitGPUtoSoAProd, process.HEBRecHitFromSoAProd )
0086 process.recHits_task = cms.Task( process.ee_task, process.hef_task, process.heb_task )
0087 process.path = cms.Path( process.recHits_task )
0088
0089 outkeeps = ['keep *_EERecHitFromSoAProd_*_*',
0090 'keep *_HEFRecHitFromSoAProd_*_*',
0091 'keep *_HEBRecHitFromSoAProd_*_*']
0092
0093 process.out = cms.OutputModule( "PoolOutputModule",
0094 fileName = cms.untracked.string( '/eos/user/b/bfontana/GPUs/GPUs_' + str(F.PU) + '.root'),
0095 outputCommands = cms.untracked.vstring(outkeeps[0], outkeeps[1], outkeeps[2])
0096 )
0097 process.outpath = cms.EndPath(process.out)