test/python/testHGCalGeometryCheck_cfg.py

0001 ###############################################################################
0002 # Way to use this:
0003 #   cmsRun testHGCalGeometryCheck_cfg.py geometry=D88
0004 #
0005 #   Options for geometry D88, D92, D93
0006 #
0007 ###############################################################################
0008 import FWCore.ParameterSet.Config as cms
0009 import os, sys, imp, re
0010 import FWCore.ParameterSet.VarParsing as VarParsing
0011
0012 ####################################################################
0013 ### SETUP OPTIONS
0014 options = VarParsing.VarParsing('standard')
0015 options.register('geometry',
0016                  "D88",
0017                   VarParsing.VarParsing.multiplicity.singleton,
0018                   VarParsing.VarParsing.varType.string,
0019                   "geometry of operations: D88, D92, D93")
0020
0021 ### get and parse the command line arguments
0022 options.parseArguments()
0023
0024 print(options)
0025
0026 ####################################################################
0027 # Use the options
0028
0029 if (options.geometry == "D93"):
0030     from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
0031     process = cms.Process('HGCGeomCheck',Phase2C17I13M9)
0032     process.load('Configuration.Geometry.GeometryExtended2026D93Reco_cff')
0033     fileName = 'HGCGeomStudyV17n.root'
0034 elif (options.geometry == "D92"):
0035     from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
0036     process = cms.Process('HGCGeomCheck',Phase2C17I13M9)
0037     process.load('Configuration.Geometry.GeometryExtended2026D92Reco_cff')
0038     fileName = 'HGCGeomStudyV17.root'
0039 else:
0040     from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
0041     process = cms.Process('HGCGeomCheck',Phase2C17I13M9)
0042     process.load('Configuration.Geometry.GeometryExtended2026D88Reco_cff')
0043     fileName = 'HGCGeomStudyV16.root'
0044
0045 print("Output file: ", fileName)
0046
0047 process.load("SimGeneral.HepPDTESSource.pdt_cfi")
0048 process.load('Geometry.HGCalGeometry.hgcalGeometryCheck_cfi')
0049 process.load('FWCore.MessageService.MessageLogger_cfi')
0050
0051 if hasattr(process,'MessageLogger'):
0052     process.MessageLogger.HGCalGeom=dict()
0053
0054 process.load("IOMC.RandomEngine.IOMC_cff")
0055 process.RandomNumberGeneratorService.generator.initialSeed = 456789
0056
0057 process.source = cms.Source("EmptySource")
0058
0059 process.generator = cms.EDProducer("FlatRandomEGunProducer",
0060     PGunParameters = cms.PSet(
0061         PartID = cms.vint32(14),
0062         MinEta = cms.double(-3.5),
0063         MaxEta = cms.double(3.5),
0064         MinPhi = cms.double(-3.14159265359),
0065         MaxPhi = cms.double(3.14159265359),
0066         MinE   = cms.double(9.99),
0067         MaxE   = cms.double(10.01)
0068     ),
0069     AddAntiParticle = cms.bool(False),
0070     Verbosity       = cms.untracked.int32(0),
0071     firstRun        = cms.untracked.uint32(1)
0072 )
0073
0074 process.maxEvents = cms.untracked.PSet(
0075     input = cms.untracked.int32(1)
0076 )
0077
0078 process.TFileService = cms.Service("TFileService",
0079                                    fileName = cms.string(fileName),
0080                                    closeFileFast = cms.untracked.bool(True)
0081                                    )
0082
0083 #process.hgcalGeometryCheck.verbosity = True
0084
0085 process.p1 = cms.Path(process.generator*process.hgcalGeometryCheck)