test/python/minbiasRun4_cfg.py

0001 ###############################################################################
0002 # Way to use this:
0003 #   cmsRun testHGCalParametersDDD_cfg.py type=V18
0004 #
0005 #   Options for type V16, V17, V17Shift, V18, V19
0006 #
0007 ###############################################################################
0008 import FWCore.ParameterSet.Config as cms
0009 import os, sys, importlib, re
0010 import FWCore.ParameterSet.VarParsing as VarParsing
0011
0012 ####################################################################
0013 ### SETUP OPTIONS
0014 options = VarParsing.VarParsing('standard')
0015 options.register('type',
0016                  "V18",
0017                   VarParsing.VarParsing.multiplicity.singleton,
0018                   VarParsing.VarParsing.varType.string,
0019                   "type of operations: V16, V17, V17Shift, V18, V19")
0020
0021 ### get and parse the command line arguments
0022 options.parseArguments()
0023 print(options)
0024
0025
0026 if (options.type == "V18"):
0027     from Configuration.Eras.Era_Phase2C22I13M9_cff import Phase2C22I13M9
0028     process = cms.Process("HGCalParametersTest",Phase2C22I13M9)
0029     geomFile = "Geometry.HGCalCommonData.testHGCal" + options.type + "Reco_cff"
0030 elif (options.type == "V19"):
0031     from Configuration.Eras.Era_Phase2C22I13M9_cff import Phase2C22I13M9
0032     process = cms.Process("HGCalParametersTest",Phase2C22I13M9)
0033     geomFile = "Geometry.HGCalCommonData.testHGCal" + options.type + "Reco_cff"
0034 elif (options.type == "V17Shift"):
0035     from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
0036     process = cms.Process("HGCalParametersTest",Phase2C17I13M9)
0037     geomFile = "Geometry.HGCalCommonData.testHGCal" + options.type + "Reco_cff"
0038 elif (options.type == "V17"):
0039     from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
0040     process = cms.Process("HGCalParametersTest",Phase2C17I13M9)
0041     geomFile = "Configuration.Geometry.GeometryExtendedRun4D110Reco_cff"
0042 else:
0043     from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
0044     process = cms.Process("HGCalParametersTest",Phase2C17I13M9)
0045     geomFile = "Configuration.Geometry.GeometryExtendedRun4D100Reco_cff"
0046
0047 outFile = "minbias" + options.type + ".root"
0048
0049 print("Geometry file: ", geomFile)
0050 print("Output file: ", outFile)
0051
0052 process.load("SimG4CMS.Calo.PythiaMinBias_cfi")
0053 process.load("SimGeneral.HepPDTESSource.pythiapdt_cfi")
0054 process.load('FWCore.MessageService.MessageLogger_cfi')
0055 process.load("IOMC.EventVertexGenerators.VtxSmearedGauss_cfi")
0056 process.load(geomFile)
0057 process.load("Configuration.StandardSequences.MagneticField_cff")
0058 process.load("Configuration.EventContent.EventContent_cff")
0059 process.load('Configuration.StandardSequences.Generator_cff')
0060 process.load('Configuration.StandardSequences.SimIdeal_cff')
0061
0062 process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
0063 from Configuration.AlCa.GlobalTag import GlobalTag
0064 process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:phase2_realistic_T21', '')
0065
0066 if 'MessageLogger' in process.__dict__:
0067     process.MessageLogger.G4cerr=dict()
0068     process.MessageLogger.HGCSim=dict()
0069
0070 process.source = cms.Source("EmptySource")
0071
0072 process.maxEvents = cms.untracked.PSet(
0073     input = cms.untracked.int32(10)
0074 )
0075
0076 process.Timing = cms.Service("Timing")
0077
0078 process.SimpleMemoryCheck = cms.Service("SimpleMemoryCheck",
0079     oncePerEventMode = cms.untracked.bool(True),
0080     showMallocInfo = cms.untracked.bool(True),
0081     dump = cms.untracked.bool(True),
0082     ignoreTotal = cms.untracked.int32(1)
0083 )
0084
0085 process.load("IOMC.RandomEngine.IOMC_cff")
0086 process.RandomNumberGeneratorService.generator.initialSeed = 456789
0087 process.RandomNumberGeneratorService.g4SimHits.initialSeed = 9876
0088 process.RandomNumberGeneratorService.VtxSmeared.initialSeed = 123456789
0089 process.rndmStore = cms.EDProducer("RandomEngineStateProducer")
0090
0091 # Event output
0092 process.output = cms.OutputModule("PoolOutputModule",
0093     process.FEVTSIMEventContent,
0094     fileName = cms.untracked.string(outFile)
0095 )
0096
0097 process.generation_step = cms.Path(process.pgen)
0098 process.simulation_step = cms.Path(process.psim)
0099 process.out_step = cms.EndPath(process.output)
0100
0101 process.generator.pythiaHepMCVerbosity = False
0102 process.generator.pythiaPylistVerbosity = 0
0103 process.g4SimHits.Physics.type = 'SimG4Core/Physics/FTFP_BERT_EMM'
0104
0105
0106 # Schedule definition
0107 process.schedule = cms.Schedule(process.generation_step,
0108                                 process.simulation_step,
0109                                 process.out_step
0110                                 )
0111
0112 # filter all path with the production filter sequence
0113 for path in process.paths:
0114         getattr(process,path)._seq = process.generator * getattr(process,path)._seq
0115