SiStrip/test/SiStripBadChannelPatcher_cfg.py

0001 import FWCore.ParameterSet.Config as cms
0002 import FWCore.ParameterSet.VarParsing as VarParsing
0003
0004 process = cms.Process("Test")
0005
0006 options = VarParsing.VarParsing("analysis")
0007
0008 options.register ('isUnitTest',
0009                   False,
0010                   VarParsing.VarParsing.multiplicity.singleton,  # singleton or list
0011                   VarParsing.VarParsing.varType.bool,            # string, int, or float
0012                   "GlobalTag")
0013
0014 options.register ('runNumber',
0015                   359334,
0016                   VarParsing.VarParsing.multiplicity.singleton, # singleton or list
0017                   VarParsing.VarParsing.varType.int,            # string, int, or float
0018                   "run number")
0019
0020 options.register ('FEDsToAdd',
0021                   [],
0022                   VarParsing.VarParsing.multiplicity.list,  # singleton or list
0023                   VarParsing.VarParsing.varType.int,        # string, int, or float
0024                   "list of FEDs to Add")
0025
0026 options.register ('FEDsToRemove',
0027                   [],
0028                   VarParsing.VarParsing.multiplicity.list,  # singleton or list
0029                   VarParsing.VarParsing.varType.int,        # string, int, or float
0030                   "list of FEDs to Remove")
0031
0032 options.register ('DetIdsToAdd',
0033                   [],
0034                   VarParsing.VarParsing.multiplicity.list,  # singleton or list
0035                   VarParsing.VarParsing.varType.int,        # string, int, or float
0036                   "list of DetIds to Add")
0037
0038 options.register ('DetIdsToRemove',
0039                   [],
0040                   VarParsing.VarParsing.multiplicity.list,  # singleton or list
0041                   VarParsing.VarParsing.varType.int,        # string, int, or float
0042                   "list of DetIds to Remove")
0043
0044
0045 options.register ('inputConnection',
0046                   "frontier://FrontierProd/CMS_CONDITIONS",
0047                   VarParsing.VarParsing.multiplicity.singleton,  # singleton or list
0048                   VarParsing.VarParsing.varType.string,            # string, int, or float
0049                   "input DB connection")
0050
0051 options.register ('inputTag',
0052                   "SiStripBadChannel_Ideal_31X_v2",
0053                   VarParsing.VarParsing.multiplicity.singleton,  # singleton or list
0054                   VarParsing.VarParsing.varType.string,            # string, int, or float
0055                   "input DB tag")
0056
0057 options.register ('outputTag',
0058                   "output",
0059                   VarParsing.VarParsing.multiplicity.singleton,  # singleton or list
0060                   VarParsing.VarParsing.varType.string,            # string, int, or float
0061                   "output DB tag")
0062
0063
0064 options.parseArguments()
0065
0066 process.maxEvents = cms.untracked.PSet(
0067     input = cms.untracked.int32(1)
0068 )
0069
0070 process.source = cms.Source("EmptySource",
0071                             firstRun = cms.untracked.uint32(options.runNumber),
0072                             numberEventsInRun = cms.untracked.uint32(1),
0073                             )
0074
0075 process.MessageLogger = cms.Service("MessageLogger",
0076     cout = cms.untracked.PSet(
0077         threshold = cms.untracked.string('WARNING') if options.isUnitTest else cms.untracked.string('INFO')
0078     ),
0079     destinations = cms.untracked.vstring('cout')
0080 )
0081
0082 ##
0083 ## Database services
0084 ##
0085 process.load("CondCore.CondDB.CondDB_cfi")
0086
0087 ##
0088 ## Neeeded for the cabling
0089 ##
0090 process.CondDB.connect='frontier://FrontierProd/CMS_CONDITIONS'
0091 process.CablingESSource = cms.ESSource('PoolDBESSource',
0092                                        process.CondDB,
0093                                        toGet = cms.VPSet( cms.PSet(record = cms.string('SiStripFedCablingRcd'),
0094                                                                    tag    = cms.string('SiStripFedCabling_GR10_v1_hlt')   # real data cabling map
0095                                                                    #tag     = cms.string('SiStripFedCabling_Ideal_31X_v2')  # ideal cabling map
0096                                                                )))
0097
0098 process.load("Configuration.Geometry.GeometryExtended2017_cff")
0099 process.load("Geometry.TrackerGeometryBuilder.trackerParameters_cfi")
0100 process.TrackerTopologyEP = cms.ESProducer("TrackerTopologyEP")
0101 process.load("CalibTracker.SiStripESProducers.SiStripConnectivity_cfi")
0102
0103 ##
0104 ## Input bad components
0105 ##
0106 process.PoolDBESSource = cms.ESSource("PoolDBESSource",
0107                                       DBParameters = cms.PSet(
0108                                           messageLevel = cms.untracked.int32(2),
0109                                           authenticationPath = cms.untracked.string('/afs/cern.ch/cms/DB/conddb')
0110                                       ),
0111                                       toGet = cms.VPSet(cms.PSet(
0112                                           record = cms.string('SiStripBadStripRcd'),
0113                                           tag = cms.string(options.inputTag)
0114                                       )),
0115                                       connect = cms.string(options.inputConnection)
0116                                       )
0117
0118 ##
0119 ## Output database (in this case local sqlite file)
0120 ##
0121 process.CondDB.connect = 'sqlite_file:outputDB.db'
0122 process.PoolDBOutputService = cms.Service("PoolDBOutputService",
0123                                           process.CondDB,
0124                                           timetype = cms.untracked.string('runnumber'),
0125                                           toPut = cms.VPSet(cms.PSet(record = cms.string('SiStripBadStripRcd'),
0126                                                                      tag = cms.string(options.outputTag)
0127                                                                      )
0128                                                             )
0129                                           )
0130
0131 process.prod = cms.EDAnalyzer("SiStripBadChannelPatcher",
0132                               printDebug = cms.bool(True) if options.isUnitTest else cms.bool(False),
0133                               Record = cms.string("SiStripBadStripRcd"),
0134                               FEDsToExclude = cms.vuint32(options.FEDsToRemove),
0135                               FEDsToInclude = cms.vuint32(options.FEDsToAdd),
0136                               detIdsToExclude = cms.vuint32(options.DetIdsToAdd),
0137                               detIdsToInclude = cms.vuint32(options.DetIdsToRemove))
0138
0139 process.Timing = cms.Service("Timing")
0140 process.print = cms.OutputModule("AsciiOutputModule")
0141
0142 process.p = cms.Path(process.prod)
0143 #process.ep = cms.EndPath(process.print)