HGCalValidation/scripts/testHGCalDigi_cfg.py

0001 ###############################################################################
0002 # Way to use this:
0003 #   cmsRun testHGCalDigi_cfg.py geometry=D92 type=DDD data=mu noise=none
0004 #                               threshold=none
0005 #
0006 #   Options for geometry: D88, D92, D93, V17Shift, V18
0007 #               type: DDD, DD4hep
0008 #               data: mu, tt
0009 #               noise: none, ok
0010 #               threshold: none, ok
0011 #
0012 ###############################################################################
0013 import FWCore.ParameterSet.Config as cms
0014 import os, sys, imp, re, random
0015 import FWCore.ParameterSet.VarParsing as VarParsing
0016
0017 ####################################################################
0018 ### SETUP OPTIONS
0019 options = VarParsing.VarParsing('standard')
0020 options.register('geometry',
0021                  "D92",
0022                   VarParsing.VarParsing.multiplicity.singleton,
0023                   VarParsing.VarParsing.varType.string,
0024                   "geometry of operations: D88, D92, D93, V17Shift, V18")
0025 options.register('type',
0026                  "DDD",
0027                   VarParsing.VarParsing.multiplicity.singleton,
0028                   VarParsing.VarParsing.varType.string,
0029                   "type of operations: DDD, DD4hep")
0030 options.register('data',
0031                  "mu",
0032                  VarParsing.VarParsing.multiplicity.singleton,
0033                  VarParsing.VarParsing.varType.string,
0034                  "data of operations: mu, tt")
0035 options.register('noise',
0036                  "ok",
0037                   VarParsing.VarParsing.multiplicity.singleton,
0038                   VarParsing.VarParsing.varType.string,
0039                  "noise of operations: none, ok")
0040 options.register('threshold',
0041                  "ok",
0042                   VarParsing.VarParsing.multiplicity.singleton,
0043                   VarParsing.VarParsing.varType.string,
0044                  "threshold of operations: none, ok")
0045
0046 ### get and parse the command line arguments
0047 options.parseArguments()
0048
0049 print(options)
0050
0051 ####################################################################
0052 # Use the options
0053 from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
0054 if (options.type == "DD4hep"):
0055     from Configuration.ProcessModifiers.dd4hep_cff import dd4hep
0056     process = cms.Process('SingleMuonSim',Phase2C17I13M9,dd4hep)
0057     if (options.geometry == "V17Shift"):
0058         geomFile = "Geometry.HGCalCommonData.testHGCal" + options.type + options.geometry + "Reco_cff"
0059     elif (options.geometry == "V18"):
0060         geomFile = "Geometry.HGCalCommonData.testHGCal" + options.type + options.geometry + "Reco_cff"
0061     else:
0062         geomFile = "Configuration.Geometry.Geometry" + options.type +"Extended2026" + options.geometry + "Reco_cff"
0063 else:
0064     process = cms.Process('SingleMuonSim',Phase2C17I13M9)
0065     if (options.geometry == "V17Shift"):
0066         geomFile = "Geometry.HGCalCommonData.testHGCal" + options.geometry + "Reco_cff"
0067     elif (options.geometry == "V18"):
0068         geomFile = "Geometry.HGCalCommonData.testHGCal" + options.geometry + "Reco_cff"
0069     else:
0070         geomFile = "Configuration.Geometry.GeometryExtended2026" + options.geometry + "Reco_cff"
0071
0072 globalTag = "auto:phase2_realistic_T21"
0073 inFile = "file:step1" + options.type + options.geometry + options.data + ".root"
0074 outFile = "file:step2" + options.type + options.geometry + options.data + ".root"
0075
0076 print("Geometry file: ", geomFile)
0077 print("Global Tag:    ", globalTag)
0078 print("Input file:    ", inFile)
0079 print("Output file:   ", outFile)
0080
0081 from SimCalorimetry.HGCalSimProducers.hgcalDigitizer_cfi import hgceeDigitizer
0082 from SimCalorimetry.HGCalSimProducers.hgcalDigitizer_cfi import hgchefrontDigitizer
0083 from SimCalorimetry.HGCalSimProducers.hgcalDigitizer_cfi import hgchebackDigitizer
0084 from SimCalorimetry.HGCalSimProducers.hgcalDigitizer_cfi import hfnoseDigitizer
0085 from SimCalorimetry.HGCalSimProducers.hgcROCParameters_cfi import hgcROCSettings
0086 # import of standard configurations
0087 process.load('Configuration.StandardSequences.Services_cff')
0088 process.load('SimGeneral.HepPDTESSource.pythiapdt_cfi')
0089 process.load('FWCore.MessageService.MessageLogger_cfi')
0090 process.load('Configuration.EventContent.EventContent_cff')
0091 process.load('SimGeneral.MixingModule.mixNoPU_cfi')
0092 process.load(geomFile)
0093 process.load('Configuration.StandardSequences.MagneticField_cff')
0094 process.load('Configuration.StandardSequences.Digi_cff')
0095 process.load('Configuration.StandardSequences.L1TrackTrigger_cff')
0096 process.load('Configuration.StandardSequences.SimL1Emulator_cff')
0097 process.load('Configuration.StandardSequences.DigiToRaw_cff')
0098 process.load('HLTrigger.Configuration.HLT_Fake2_cff')
0099 process.load('Configuration.StandardSequences.EndOfProcess_cff')
0100 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0101
0102 process.MessageLogger.cerr.FwkReport.reportEvery = 1
0103
0104 process.maxEvents = cms.untracked.PSet(
0105     input = cms.untracked.int32(-1),
0106     output = cms.optional.untracked.allowed(cms.int32,cms.PSet)
0107 )
0108
0109 # Input source
0110 process.source = cms.Source("PoolSource",
0111     dropDescendantsOfDroppedBranches = cms.untracked.bool(False),
0112     fileNames = cms.untracked.vstring(inFile),
0113     inputCommands = cms.untracked.vstring(
0114         'keep *',
0115         'drop *_genParticles_*_*',
0116         'drop *_genParticlesForJets_*_*',
0117         'drop *_kt4GenJets_*_*',
0118         'drop *_kt6GenJets_*_*',
0119         'drop *_iterativeCone5GenJets_*_*',
0120         'drop *_ak4GenJets_*_*',
0121         'drop *_ak7GenJets_*_*',
0122         'drop *_ak8GenJets_*_*',
0123         'drop *_ak4GenJetsNoNu_*_*',
0124         'drop *_ak8GenJetsNoNu_*_*',
0125         'drop *_genCandidatesForMET_*_*',
0126         'drop *_genParticlesForMETAllVisible_*_*',
0127         'drop *_genMetCalo_*_*',
0128         'drop *_genMetCaloAndNonPrompt_*_*',
0129         'drop *_genMetTrue_*_*',
0130         'drop *_genMetIC5GenJs_*_*'
0131     ),
0132     secondaryFileNames = cms.untracked.vstring()
0133 )
0134
0135 process.options = cms.untracked.PSet(
0136     IgnoreCompletely = cms.untracked.vstring(),
0137     Rethrow = cms.untracked.vstring(),
0138     TryToContinue = cms.untracked.vstring(),
0139     accelerators = cms.untracked.vstring('*'),
0140     allowUnscheduled = cms.obsolete.untracked.bool,
0141     canDeleteEarly = cms.untracked.vstring(),
0142     deleteNonConsumedUnscheduledModules = cms.untracked.bool(True),
0143     dumpOptions = cms.untracked.bool(False),
0144     emptyRunLumiMode = cms.obsolete.untracked.string,
0145     eventSetup = cms.untracked.PSet(
0146         forceNumberOfConcurrentIOVs = cms.untracked.PSet(
0147             allowAnyLabel_=cms.required.untracked.uint32
0148         ),
0149         numberOfConcurrentIOVs = cms.untracked.uint32(0)
0150     ),
0151     fileMode = cms.untracked.string('FULLMERGE'),
0152     forceEventSetupCacheClearOnNewRun = cms.untracked.bool(False),
0153     makeTriggerResults = cms.obsolete.untracked.bool,
0154     numberOfConcurrentLuminosityBlocks = cms.untracked.uint32(0),
0155     numberOfConcurrentRuns = cms.untracked.uint32(1),
0156     numberOfStreams = cms.untracked.uint32(0),
0157     numberOfThreads = cms.untracked.uint32(1),
0158     printDependencies = cms.untracked.bool(False),
0159     sizeOfStackForThreadsInKB = cms.optional.untracked.uint32,
0160     throwIfIllegalParameter = cms.untracked.bool(True),
0161     wantSummary = cms.untracked.bool(False)
0162 )
0163
0164 # Production Info
0165 process.configurationMetadata = cms.untracked.PSet(
0166     annotation = cms.untracked.string('step2 nevts:1000'),
0167     name = cms.untracked.string('Applications'),
0168     version = cms.untracked.string('$Revision: 1.19 $')
0169 )
0170
0171 # Output definition
0172
0173 process.FEVTDEBUGHLToutput = cms.OutputModule("PoolOutputModule",
0174     dataset = cms.untracked.PSet(
0175         dataTier = cms.untracked.string('GEN-SIM-DIGI-RAW'),
0176         filterName = cms.untracked.string('')
0177     ),
0178     fileName = cms.untracked.string(outFile),
0179     outputCommands = process.FEVTDEBUGHLTEventContent.outputCommands,
0180     splitLevel = cms.untracked.int32(0)
0181 )
0182
0183 # Additional output definition
0184
0185 # Other statements
0186 process.mix.digitizers = cms.PSet(process.theDigitizersValid)
0187 from Configuration.AlCa.GlobalTag import GlobalTag
0188 process.GlobalTag = GlobalTag(process.GlobalTag, globalTag, '')
0189
0190 # Path and EndPath definitions
0191 process.digitisation_step = cms.Path(process.pdigi_valid)
0192 process.L1TrackTrigger_step = cms.Path(process.L1TrackTrigger)
0193 process.L1simulation_step = cms.Path(process.SimL1Emulator)
0194 process.digi2raw_step = cms.Path(process.DigiToRaw)
0195 process.endjob_step = cms.EndPath(process.endOfProcess)
0196 process.FEVTDEBUGHLToutput_step = cms.EndPath(process.FEVTDEBUGHLToutput)
0197
0198 if (options.noise == "none"):
0199     process.HGCAL_noise_fC.values = [0,0,0]
0200
0201 if (options.threshold == "none"):
0202     process.mix.digitizers.hgceeDigitizer.digiCfg.feCfg.adcThreshold_fC = 0.0
0203     process.mix.digitizers.hgchefrontDigitizer.digiCfg.feCfg.adcThreshold_fC = 0.0
0204     process.mix.digitizers.hgchebackDigitizer.digiCfg.feCfg.adcThreshold_fC = 0.0
0205
0206 # Schedule definition
0207 # process.schedule imported from cff in HLTrigger.Configuration
0208 process.schedule.insert(0, process.digitisation_step)
0209 process.schedule.insert(1, process.L1TrackTrigger_step)
0210 process.schedule.insert(2, process.L1simulation_step)
0211 process.schedule.insert(3, process.digi2raw_step)
0212 process.schedule.extend([process.endjob_step,process.FEVTDEBUGHLToutput_step])
0213
0214 from PhysicsTools.PatAlgos.tools.helpers import associatePatAlgosToolsTask
0215 associatePatAlgosToolsTask(process)
0216
0217 # customisation of the process.
0218
0219 # Automatic addition of the customisation function from HLTrigger.Configuration.customizeHLTforMC
0220 from HLTrigger.Configuration.customizeHLTforMC import customizeHLTforMC
0221
0222 #call to customisation function customizeHLTforMC imported from HLTrigger.Configuration.customizeHLTforMC
0223 process = customizeHLTforMC(process)
0224
0225 # End of customisation functions
0226
0227
0228 # Customisation from command line
0229
0230 # Add early deletion of temporary data products to reduce peak memory need
0231 from Configuration.StandardSequences.earlyDeleteSettings_cff import customiseEarlyDelete
0232 process = customiseEarlyDelete(process)
0233 # End adding early deletion