test/python/runPrintG4Solids2_cfg.py

0001 ###############################################################################
0002 # Way to use this:
0003 #   cmsRun grunPrintG4Solids_cfg.py geometry=D98 dd4hep=False
0004 #
0005 #   Options for geometry D88, D91, D92, D93, D94, D95, D96, D98, D99, D100,
0006 #                        D101, D102, D103
0007 #   Options for type DDD, DD4hep
0008 #
0009 ###############################################################################
0010 import FWCore.ParameterSet.Config as cms
0011 import os, sys, importlib, re
0012 import FWCore.ParameterSet.VarParsing as VarParsing
0013
0014 ####################################################################
0015 ### SETUP OPTIONS
0016 options = VarParsing.VarParsing('standard')
0017 options.register('geometry',
0018                  "D88",
0019                   VarParsing.VarParsing.multiplicity.singleton,
0020                   VarParsing.VarParsing.varType.string,
0021                   "geometry of operations: D88, D91, D92, D93, D94, D95, D96, D98, D99, D100, D101, D102, D103")
0022 options.register('type',
0023                  "DDD",
0024                   VarParsing.VarParsing.multiplicity.singleton,
0025                   VarParsing.VarParsing.varType.string,
0026                   "type of operations: DDD, DD4hep")
0027
0028 ### get and parse the command line arguments
0029 options.parseArguments()
0030
0031 print(options)
0032
0033 ####################################################################
0034 # Use the options
0035 from Configuration.ProcessModifiers.dd4hep_cff import dd4hep
0036
0037 if (options.type == "DD4hep"):
0038     geomFile = "Configuration.Geometry.GeometryDD4hepExtended2026" + options.geometry + "Reco_cff"
0039     if (options.geometry == "D94"):
0040         from Configuration.Eras.Era_Phase2C20I13M9_cff import Phase2C20I13M9
0041         process = cms.Process('PrintG4Solids',Phase2C20I13M9,dd4hep)
0042     else:
0043         from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
0044         process = cms.Process('PrintG4Solids',Phase2C17I13M9,dd4hep)
0045 else:
0046     geomFile = "Configuration.Geometry.GeometryExtended2026" + options.geometry + "Reco_cff"
0047     if (options.geometry == "D94"):
0048         from Configuration.Eras.Era_Phase2C20I13M9_cff import Phase2C20I13M9
0049         process = cms.Process('PrintG4Solids',Phase2C20I13M9)
0050     else:
0051         from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
0052         process = cms.Process('PrintG4Solids',Phase2C17I13M9)
0053
0054 print("Geometry file Name: ", geomFile)
0055
0056 process.load(geomFile)
0057 process.load('FWCore.MessageService.MessageLogger_cfi')
0058 process.load('SimGeneral.HepPDTESSource.pdt_cfi')
0059 process.load('IOMC.RandomEngine.IOMC_cff')
0060 process.load('IOMC.EventVertexGenerators.VtxSmearedFlat_cfi')
0061 process.load('GeneratorInterface.Core.generatorSmeared_cfi')
0062 process.load('FWCore.MessageService.MessageLogger_cfi')
0063 process.load('SimG4Core.Application.g4SimHits_cfi')
0064
0065 if hasattr(process,'MessageLogger'):
0066     process.MessageLogger.G4cerr=dict()
0067     process.MessageLogger.G4cout=dict()
0068
0069 process.source = cms.Source("EmptySource")
0070
0071 process.generator = cms.EDProducer("FlatRandomEGunProducer",
0072                                    PGunParameters = cms.PSet(
0073                                        PartID = cms.vint32(14),
0074                                        MinEta = cms.double(-3.5),
0075                                        MaxEta = cms.double(3.5),
0076                                        MinPhi = cms.double(-3.14159265359),
0077                                        MaxPhi = cms.double(3.14159265359),
0078                                        MinE   = cms.double(9.99),
0079                                        MaxE   = cms.double(10.01)
0080                                    ),
0081                                    AddAntiParticle = cms.bool(False),
0082                                    Verbosity       = cms.untracked.int32(0),
0083                                    firstRun        = cms.untracked.uint32(1)
0084                                )
0085
0086 process.maxEvents = cms.untracked.PSet(
0087     input = cms.untracked.int32(1)
0088 )
0089
0090
0091 process.g4SimHits.UseMagneticField = False
0092 process.g4SimHits.Physics.type = 'SimG4Core/Physics/DummyPhysics'
0093 process.g4SimHits.Physics.DummyEMPhysics = True
0094 process.g4SimHits.Physics.DefaultCutValue = 10.
0095 process.g4SimHits.LHCTransport = False
0096
0097 if (options.type == "DD4hep"):
0098     dd4hep = True
0099 else:
0100     dd4hep = False
0101
0102 process.g4SimHits.Watchers = cms.VPSet(cms.PSet(
0103     dd4hep         = cms.untracked.bool(dd4hep),
0104     dumpVolumes    = cms.untracked.vstring(),
0105     dumpShapes     = cms.untracked.vstring("G4ExtrudedSolid"),
0106     type           = cms.string('PrintG4Solids')
0107 ))
0108
0109 process.p1 = cms.Path(process.generator*process.VtxSmeared*process.generatorSmeared*process.g4SimHits)