test/python/runPrintG4Solids_cfg.py

0001 ###############################################################################
0002 # Way to use this:
0003 #   cmsRun grunPrintG4Solids_cfg.py geometry=D110 dd4hep=False
0004 #
0005 #   Options for geometry D95, D96, D98, D99, D100, D101, D102, D103, D104,
0006 #                        D105, D106, D107, D108, D109, D110, D111, D112, D113,
0007 #                        D114, D115, D116, D117, D118, D119, D120, D121, D122,
0008 #                        D123
0009 #   Options for type DDD, DD4hep
0010 #
0011 ###############################################################################
0012 import FWCore.ParameterSet.Config as cms
0013 import os, sys, importlib, re
0014 import FWCore.ParameterSet.VarParsing as VarParsing
0015
0016 ####################################################################
0017 ### SETUP OPTIONS
0018 options = VarParsing.VarParsing('standard')
0019 options.register('geometry',
0020                  "D110",
0021                   VarParsing.VarParsing.multiplicity.singleton,
0022                   VarParsing.VarParsing.varType.string,
0023                   "geometry of operations: D95, D96, D98, D99, D100, D101, D102, D103, D104, D105, D106, D107, D108, D109, D110, D111, D112, D113, D114, D115, D116, D117, D118, D119, D120, D121, D122, D123")
0024 options.register('type',
0025                  "DDD",
0026                   VarParsing.VarParsing.multiplicity.singleton,
0027                   VarParsing.VarParsing.varType.string,
0028                   "type of operations: DDD, DD4hep")
0029
0030 ### get and parse the command line arguments
0031 options.parseArguments()
0032
0033 print(options)
0034
0035 ####################################################################
0036 # Use the options
0037
0038 geomName = "Run4" + options.geometry
0039 import Configuration.Geometry.defaultPhase2ConditionsEra_cff as _settings
0040 GLOBAL_TAG, ERA = _settings.get_era_and_conditions(geomName)
0041 print("Geometry Name:   ", geomName)
0042 print("Global Tag Name: ", GLOBAL_TAG)
0043 print("Era Name:        ", ERA)
0044
0045 if (options.type == "DD4hep"):
0046     from Configuration.ProcessModifiers.dd4hep_cff import dd4hep
0047     process = cms.Process('G4PrintGeometry',ERA,dd4hep)
0048     geomFile = "Configuration.Geometry.GeometryDD4hepExtended" + geomName + "Reco_cff"
0049     dd4hep = True
0050 else:
0051     process = cms.Process('G4PrintGeometry',ERA)
0052     geomFile = "Configuration.Geometry.GeometryExtended" + geomName + "Reco_cff"
0053     dd4hep = False
0054
0055 print("Geometry file Name: ", geomFile)
0056 print("dd4hep Flag:        ", dd4hep)
0057
0058 if hasattr(process,'MessageLogger'):
0059     process.MessageLogger.G4cerr=dict()
0060     process.MessageLogger.G4cout=dict()
0061
0062 process.load('SimGeneral.HepPDTESSource.pdt_cfi')
0063 process.load(geomFile)
0064 process.load('IOMC.RandomEngine.IOMC_cff')
0065 process.load('IOMC.EventVertexGenerators.VtxSmearedFlat_cfi')
0066 process.load('GeneratorInterface.Core.generatorSmeared_cfi')
0067 process.load('FWCore.MessageService.MessageLogger_cfi')
0068 process.load("Configuration.StandardSequences.MagneticField_38T_cff")
0069 process.load('SimG4Core.Application.g4SimHits_cfi')
0070 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0071 from Configuration.AlCa.GlobalTag import GlobalTag
0072 process.GlobalTag = GlobalTag(process.GlobalTag, GLOBAL_TAG, '')
0073
0074 process.source = cms.Source("EmptySource")
0075
0076 process.generator = cms.EDProducer("FlatRandomEGunProducer",
0077     PGunParameters = cms.PSet(
0078         PartID = cms.vint32(14),
0079         MinEta = cms.double(-3.5),
0080         MaxEta = cms.double(3.5),
0081         MinPhi = cms.double(-3.14159265359),
0082         MaxPhi = cms.double(3.14159265359),
0083         MinE   = cms.double(9.99),
0084         MaxE   = cms.double(10.01)
0085     ),
0086     AddAntiParticle = cms.bool(False),
0087     Verbosity       = cms.untracked.int32(0),
0088     firstRun        = cms.untracked.uint32(1)
0089 )
0090
0091 process.maxEvents = cms.untracked.PSet(
0092     input = cms.untracked.int32(1)
0093 )
0094
0095 process.p1 = cms.Path(process.generator*process.VtxSmeared*process.generatorSmeared*process.g4SimHits)