EgammaClusterProducers/test/DRNTest_cfg.py

0001 import FWCore.ParameterSet.Config as cms
0002 import FWCore.ParameterSet.VarParsing as VarParsing
0003
0004 from Configuration.ProcessModifiers.enableSonicTriton_cff import enableSonicTriton
0005 process = cms.Process("Demo",enableSonicTriton)
0006
0007 process.load("HeterogeneousCore.SonicTriton.TritonService_cff")
0008 process.load("FWCore.MessageService.MessageLogger_cfi")
0009 process.load("Configuration.StandardSequences.GeometryRecoDB_cff")
0010
0011 process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(1000) )
0012 process.options.numberOfThreads = cms.untracked.uint32(4)
0013 process.options.numberOfStreams = cms.untracked.uint32(4)
0014 process.options.TryToContinue = cms.untracked.vstring('ProductNotFound')
0015
0016 process.TritonService.verbose = False
0017 process.TritonService.fallback.verbose = False
0018 process.TritonService.fallback.container = "apptainer"
0019 process.TritonService.fallback.device = "cpu"
0020
0021 process.MessageLogger.cerr.FwkReport.reportEvery = cms.untracked.int32(100)
0022 #process.MessageLogger.suppressWarning = cms.untracked.vstring('DRNProducerEB', 'DRNProducerEE')
0023
0024 process.source = cms.Source("PoolSource",
0025                                 # replace 'myfile.root' with the source file you want to use
0026                                 fileNames = cms.untracked.vstring(
0027 '/store/mc/RunIISummer20UL18RECO/DoubleElectron_Pt-1To300-gun/AODSIM/FlatPU0to70EdalIdealGT_EdalIdealGT_106X_upgrade2018_realistic_v11_L1v1_EcalIdealIC-v2/270000/4CDD9457-E14C-D84A-9BD4-3140CB6AEEB6.root',
0028 '/store/mc/RunIISummer20UL18RECO/DoubleElectron_Pt-1To300-gun/AODSIM/FlatPU0to70EdalIdealGT_EdalIdealGT_106X_upgrade2018_realistic_v11_L1v1_EcalIdealIC-v2/270000/D3A06456-7F7D-C940-9E56-0DCA06B3ECC9.root',
0029 '/store/mc/RunIISummer20UL18RECO/DoubleElectron_Pt-1To300-gun/AODSIM/FlatPU0to70EdalIdealGT_EdalIdealGT_106X_upgrade2018_realistic_v11_L1v1_EcalIdealIC-v2/270000/477810A1-AE5C-6A49-8067-35776F0C78B6.root',
0030 '/store/mc/RunIISummer20UL18RECO/DoubleElectron_Pt-1To300-gun/AODSIM/FlatPU0to70EdalIdealGT_EdalIdealGT_106X_upgrade2018_realistic_v11_L1v1_EcalIdealIC-v2/270000/99F8FA99-B120-BF40-BD67-7825696B9E78.root'
0031
0032                 )
0033                             )
0034
0035 process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
0036
0037 from Configuration.AlCa.GlobalTag import GlobalTag
0038 process.GlobalTag = GlobalTag(process.GlobalTag, '106X_upgrade2018_realistic_v11_Ecal5', '')
0039 #from RecoEcal.EgammaClusterProducers.SCEnergyCorrectorDRNProducer_cfi import *
0040 from RecoEcal.EgammaClusterProducers.SCEnergyCorrectorDRNProducer_cfi import *
0041
0042 process.DRNProducerEB = DRNProducerEB
0043 process.DRNProducerEE = DRNProducerEE
0044
0045 from PhysicsTools.SelectorUtils.tools.vid_id_tools import *
0046 dataFormat = DataFormat.AOD
0047 switchOnVIDElectronIdProducer(process, dataFormat)
0048
0049 # define which IDs we want to produce
0050 my_id_modules = ['RecoEgamma.ElectronIdentification.Identification.cutBasedElectronID_Fall17_94X_V2_cff']
0051
0052 for idmod in my_id_modules:
0053         setupAllVIDIdsInModule(process, idmod, setupVIDElectronSelection)
0054
0055 process.nTuplelize = cms.EDAnalyzer('DRNTestNTuplizer',
0056         vertexCollection = cms.InputTag('offlinePrimaryVertices'),
0057         rhoFastJet = cms.InputTag("fixedGridRhoFastjetAll"),
0058         pileupInfo = cms.InputTag("addPileupInfo"),
0059         electrons = cms.InputTag("gedGsfElectrons"),
0060         genParticles = cms.InputTag("genParticles"),
0061         #Cut Based Id
0062         eleLooseIdMap = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-loose"),
0063         eleMediumIdMap = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-medium"),
0064         eleTightIdMap = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-tight"),
0065
0066         tracks = cms.InputTag("globalTracks"),
0067         SkipEvent = cms.untracked.vstring('ProductNotFound')
0068         )
0069
0070 process.TFileService = cms.Service("TFileService",
0071      fileName = cms.string("nTupleMC.root"),
0072       closeFileFast = cms.untracked.bool(True)
0073   )
0074
0075 process.load( "HLTrigger.Timer.FastTimerService_cfi" )
0076
0077 #process.p = cms.Path(process.DRNProducerEB*process.egmGsfElectronIDSequence*process.nTuplelize)
0078
0079 #process.p = cms.Path(process.DRNProducerEB*process.DRNProducerEE*process.egmGsfElectronIDSequence*process.nTuplelize)
0080 process.p = cms.Path(process.DRNProducerEB*process.DRNProducerEE)