Geometry/HGCalGeometry/test/python/testHGCalGeomLocator_cfg.py - master - githubmirror/cmssw

Geometry/HGCalGeometry/test/python/testHGCalGeomLocator_cfg.py

84 строки3 KB

Sunanda

Update all test codes in Geometry/HGCalGeometry for the existing versions of HGCal geometry within CMSSW

03 май 2026, 09:05

03 май 2026, 09:051a15266

###############################################################################
# Way to use this:
#   cmsRun testHGCalGeomLocator_cfg.py geometry=D120 step=2
#
#   Options for geometry D104, D110, D111, D112, D113, D114, D115, D120, D121
#                        D122, D123, D124, D125
#
###############################################################################
import FWCore.ParameterSet.Config as cms
import os, sys, importlib, re
import FWCore.ParameterSet.VarParsing as VarParsing

####################################################################
### SETUP OPTIONS
options = VarParsing.VarParsing('standard')
options.register('geometry',
                 "D121",
                  VarParsing.VarParsing.multiplicity.singleton,
                  VarParsing.VarParsing.varType.string,
                  "geometry of operations: D104, D110, D111, D112, D113, D114, D115, D120, D121, D122, D123, D124, D125")

options.register('step',
                 10,
                  VarParsing.VarParsing.multiplicity.singleton,
                  VarParsing.VarParsing.varType.int,
                  "geometry of operations: 1, 2, 10")

### get and parse the command line arguments
options.parseArguments()
print(options)

####################################################################
# Use the options

geomName = "Run4" + options.geometry
step = options.step
geomFile = "Configuration.Geometry.GeometryExtended" + geomName + "Reco_cff"
import Configuration.Geometry.defaultPhase2ConditionsEra_cff as _settings
GLOBAL_TAG, ERA = _settings.get_era_and_conditions(geomName)

print("Geometry Name:   ", geomName)
print("Geom file Name:  ", geomFile)
print("Global Tag Name: ", GLOBAL_TAG)
print("Era Name:        ", ERA)
print("Step             ", step)

process = cms.Process("HGCalGeomLocator",ERA)

process.load(geomFile)
process.load("SimGeneral.HepPDTESSource.pdt_cfi")
process.load('FWCore.MessageService.MessageLogger_cfi')

if hasattr(process,'MessageLogger'):
    process.MessageLogger.HGCalGeom=dict()
    process.MessageLogger.HGCalGeomX=dict()

process.load("IOMC.RandomEngine.IOMC_cff")
process.RandomNumberGeneratorService.generator.initialSeed = 456789

process.source = cms.Source("EmptySource")

process.generator = cms.EDProducer("FlatRandomEGunProducer",
    PGunParameters = cms.PSet(
        PartID = cms.vint32(14),
        MinEta = cms.double(-3.5),
        MaxEta = cms.double(3.5),
        MinPhi = cms.double(-3.14159265359),
        MaxPhi = cms.double(3.14159265359),
        MinE   = cms.double(9.99),
        MaxE   = cms.double(10.01)
    ),
    AddAntiParticle = cms.bool(False),
    Verbosity       = cms.untracked.int32(0),
    firstRun        = cms.untracked.uint32(1)
)

process.maxEvents = cms.untracked.PSet(
    input = cms.untracked.int32(1)
)

process.load("Geometry.HGCalGeometry.hgcalGeomLocatorTesterEE_cfi")
process.hgcalGeomLocatorTesterEE.step = step

process.p1 = cms.Path(process.generator*process.hgcalGeomLocatorTesterEE)