test/writehelpers/geometryExtended2025_writer.py

0001 import FWCore.ParameterSet.Config as cms
0002 import argparse
0003 import sys
0004
0005 parser = argparse.ArgumentParser(prog=sys.argv[0], description='Generate XML geometry.')
0006 parser.add_argument("--tag", help="global tag to use", type=str)
0007 args = parser.parse_args()
0008
0009
0010 import FWCore.ParameterSet.Config as cms
0011 from Configuration.Eras.Era_Run3_DDD_cff import Run3_DDD
0012
0013 process = cms.Process("GeometryWriter", Run3_DDD)
0014
0015 process.load('CondCore.CondDB.CondDB_cfi')
0016
0017 process.load('Configuration.Geometry.GeometryExtended2025_cff')
0018 process.load('Geometry.CaloEventSetup.CaloGeometryDBWriter_cfi')
0019 process.load('CondTools.Geometry.HcalParametersWriter_cff')
0020 process.load("Geometry.MuonNumbering.muonGeometryConstants_cff")
0021 process.load("Geometry.ForwardGeometry.ZdcGeometry_cfi")
0022
0023 process.CaloGeometryBuilder = cms.ESProducer("CaloGeometryBuilder",
0024     SelectedCalos = cms.vstring(
0025         'HCAL',
0026         'ZDC',
0027         'EcalBarrel',
0028         'EcalEndcap',
0029         'EcalPreshower',
0030         'TOWER'
0031     )
0032 )
0033
0034 process.source = cms.Source("EmptyIOVSource",
0035                             lastValue = cms.uint64(1),
0036                             timetype = cms.string('runnumber'),
0037                             firstValue = cms.uint64(1),
0038                             interval = cms.uint64(1)
0039                             )
0040
0041 # This reads the big XML file and the only way to fill the
0042 # nonreco part of the database is to read this file.
0043 process.XMLGeometryWriter = cms.EDAnalyzer("XMLGeometryBuilder",
0044                                            XMLFileName = cms.untracked.string("./geSingleBigFile.xml"),
0045                                            ZIP = cms.untracked.bool(True)
0046                                            )
0047
0048 process.TrackerGeometryWriter = cms.EDAnalyzer("PGeometricDetBuilder",fromDD4hep=cms.bool(False))
0049 process.TrackerParametersWriter = cms.EDAnalyzer("PTrackerParametersDBBuilder",fromDD4hep=cms.bool(False))
0050
0051 process.CaloGeometryWriter = cms.EDAnalyzer("PCaloGeometryBuilder",fromDD4hep = cms.untracked.bool(False))
0052
0053 process.CSCGeometryWriter = cms.EDAnalyzer("CSCRecoIdealDBLoader",fromDD4hep = cms.untracked.bool(False))
0054
0055 process.DTGeometryWriter = cms.EDAnalyzer("DTRecoIdealDBLoader",fromDD4hep = cms.untracked.bool(False))
0056
0057 process.RPCGeometryWriter = cms.EDAnalyzer("RPCRecoIdealDBLoader",fromDD4hep = cms.untracked.bool(False))
0058
0059 process.GEMGeometryWriter = cms.EDAnalyzer("GEMRecoIdealDBLoader",fromDD4hep = cms.untracked.bool(False))
0060
0061 process.CondDB.timetype = cms.untracked.string('runnumber')
0062 process.CondDB.connect = cms.string('sqlite_file:myfile.db')
0063 process.PoolDBOutputService = cms.Service("PoolDBOutputService",
0064                                           process.CondDB,
0065                                           toPut = cms.VPSet(cms.PSet(record = cms.string('GeometryFileRcd'),tag = cms.string('XMLFILE_Geometry_'+args.tag+'_Extended2025_mc')),
0066                                                             cms.PSet(record = cms.string('IdealGeometryRecord'),tag = cms.string('TKRECO_Geometry_'+args.tag)),
0067                                                             cms.PSet(record = cms.string('PTrackerParametersRcd'),tag = cms.string('TKParameters_Geometry_'+args.tag)),
0068                                                             cms.PSet(record = cms.string('PEcalBarrelRcd'),   tag = cms.string('EBRECO_Geometry_'+args.tag)),
0069                                                             cms.PSet(record = cms.string('PEcalEndcapRcd'),   tag = cms.string('EERECO_Geometry_'+args.tag)),
0070                                                             cms.PSet(record = cms.string('PEcalPreshowerRcd'),tag = cms.string('EPRECO_Geometry_'+args.tag)),
0071                                                             cms.PSet(record = cms.string('PHcalRcd'),         tag = cms.string('HCALRECO_Geometry_'+args.tag)),
0072                                                             cms.PSet(record = cms.string('HcalParametersRcd'), tag = cms.string('HCALParameters_Geometry_'+args.tag)),
0073                                                             cms.PSet(record = cms.string('PCaloTowerRcd'),    tag = cms.string('CTRECO_Geometry_'+args.tag)),
0074                                                             cms.PSet(record = cms.string('PZdcRcd'),          tag = cms.string('ZDCRECO_Geometry_'+args.tag)),
0075                                                             cms.PSet(record = cms.string('PCastorRcd'),       tag = cms.string('CASTORRECO_Geometry_'+args.tag)),
0076                                                             cms.PSet(record = cms.string('CSCRecoGeometryRcd'),tag = cms.string('CSCRECO_Geometry_'+args.tag)),
0077                                                             cms.PSet(record = cms.string('CSCRecoDigiParametersRcd'),tag = cms.string('CSCRECODIGI_Geometry_'+args.tag)),
0078                                                             cms.PSet(record = cms.string('DTRecoGeometryRcd'),tag = cms.string('DTRECO_Geometry_'+args.tag)),
0079                                                             cms.PSet(record = cms.string('RPCRecoGeometryRcd'),tag = cms.string('RPCRECO_Geometry_'+args.tag)),
0080                                                             cms.PSet(record = cms.string('GEMRecoGeometryRcd'),tag = cms.string('GEMRECO_Geometry_'+args.tag))
0081                                                         )
0082                                     )
0083
0084 process.maxEvents = cms.untracked.PSet(
0085     input = cms.untracked.int32(1)
0086     )
0087
0088 process.p1 = cms.Path(process.XMLGeometryWriter+process.TrackerGeometryWriter+process.TrackerParametersWriter+process.CaloGeometryWriter+process.HcalParametersWriter+process.CSCGeometryWriter+process.DTGeometryWriter+process.RPCGeometryWriter+process.GEMGeometryWriter)