File indexing completed on 2025-06-12 23:29:24
0001 import FWCore.ParameterSet.Config as cms
0002 import sys
0003 from enum import Enum
0004 from Alignment.OfflineValidation.TkAlAllInOneTool.defaultInputFiles_cff import filesDefaultMC_TTbarPhase2RECO
0005
0006 class RefitType(Enum):
0007 STANDARD = 1
0008 COMMON = 2
0009
0010 isDA = True
0011 isMC = True
0012 allFromGT = True
0013 applyBows = True
0014 applyExtraConditions = True
0015 theRefitter = RefitType.COMMON
0016 _theTrackCollection = "generalTracks"
0017
0018 from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
0019 process = cms.Process("Demo",Phase2C17I13M9)
0020
0021
0022
0023
0024 process.options.numberOfThreads = 8
0025
0026
0027
0028
0029 process.source = cms.Source("PoolSource",
0030 fileNames = filesDefaultMC_TTbarPhase2RECO,
0031 duplicateCheckMode = cms.untracked.string('checkAllFilesOpened')
0032 )
0033
0034 runboundary = 1
0035 process.source.firstRun = cms.untracked.uint32(int(runboundary))
0036 process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(100) )
0037
0038
0039
0040
0041 if isMC:
0042 print("############ testPVValidation_cfg.py: msg%-i: This is Simulation!")
0043 runboundary = 1
0044 else:
0045 print("############ testPVValidation_cfg.py: msg%-i: This is DATA!")
0046 import FWCore.PythonUtilities.LumiList as LumiList
0047 process.source.lumisToProcess = LumiList.LumiList(filename ='None').getVLuminosityBlockRange()
0048
0049
0050
0051
0052 process.load('FWCore.MessageService.MessageLogger_cfi')
0053 process.MessageLogger.cerr.enable = False
0054 process.MessageLogger.PrimaryVertexValidation=dict()
0055 process.MessageLogger.SplitVertexResolution=dict()
0056 process.MessageLogger.FilterOutLowPt=dict()
0057 process.MessageLogger.cout = cms.untracked.PSet(
0058 enable = cms.untracked.bool(True),
0059 threshold = cms.untracked.string("INFO"),
0060 default = cms.untracked.PSet(limit = cms.untracked.int32(0)),
0061 FwkReport = cms.untracked.PSet(limit = cms.untracked.int32(-1),
0062 reportEvery = cms.untracked.int32(1)
0063 ),
0064 PrimaryVertexValidation = cms.untracked.PSet( limit = cms.untracked.int32(-1)),
0065 SplitVertexResolution = cms.untracked.PSet( limit = cms.untracked.int32(-1)),
0066 FilterOutLowPt = cms.untracked.PSet( limit = cms.untracked.int32(-1)),
0067 enableStatistics = cms.untracked.bool(True)
0068 )
0069
0070
0071
0072
0073 process.load("TrackingTools.TransientTrack.TransientTrackBuilder_cfi")
0074
0075
0076
0077
0078 process.load('Configuration.StandardSequences.MagneticField_cff')
0079
0080
0081
0082
0083
0084 process.load('Configuration.Geometry.GeometryExtendedRun4D110Reco_cff')
0085
0086
0087
0088
0089 process.load("RecoVertex.BeamSpotProducer.BeamSpot_cff")
0090
0091
0092
0093
0094 process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
0095 from Configuration.AlCa.GlobalTag import GlobalTag
0096 process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:phase2_realistic', '')
0097
0098 if allFromGT:
0099 print("############ testPVValidation_cfg.py: msg%-i: All is taken from GT")
0100 else:
0101
0102
0103
0104 from CondCore.DBCommon.CondDBSetup_cfi import *
0105 process.trackerAlignment = cms.ESSource("PoolDBESSource",CondDBSetup,
0106 connect = cms.string('frontier://FrontierProd/CMS_CONDITIONS'),
0107 toGet = cms.VPSet(cms.PSet(record = cms.string('TrackerAlignmentRcd'),
0108 tag = cms.string('TrackerAlignment_Upgrade2017_design_v4')
0109 )
0110 )
0111 )
0112 process.es_prefer_trackerAlignment = cms.ESPrefer("PoolDBESSource", "trackerAlignment")
0113
0114
0115
0116
0117 process.setAPE = cms.ESSource("PoolDBESSource",CondDBSetup,
0118 connect = cms.string('frontier://FrontierProd/CMS_CONDITIONS'),
0119 toGet = cms.VPSet(cms.PSet(record = cms.string('TrackerAlignmentErrorExtendedRcd'),
0120 tag = cms.string('TrackerAlignmentErrorsExtended_Upgrade2017_design_v0')
0121 )
0122 )
0123 )
0124 process.es_prefer_setAPE = cms.ESPrefer("PoolDBESSource", "setAPE")
0125
0126
0127
0128
0129 if applyBows:
0130 print("############ testPVValidation_cfg.py: msg%-i: Applying TrackerSurfaceDeformations!")
0131 process.trackerBows = cms.ESSource("PoolDBESSource",CondDBSetup,
0132 connect = cms.string('frontier://FrontierProd/CMS_CONDITIONS'),
0133 toGet = cms.VPSet(cms.PSet(record = cms.string('TrackerSurfaceDeformationRcd'),
0134 tag = cms.string('TrackerSurfaceDeformations_zero')
0135 )
0136 )
0137 )
0138 process.es_prefer_Bows = cms.ESPrefer("PoolDBESSource", "trackerBows")
0139 else:
0140 print("############ testPVValidation_cfg.py: msg%-i: MultiPVValidation: Not applying TrackerSurfaceDeformations!")
0141
0142
0143
0144
0145 if applyExtraConditions:
0146
0147 import CalibTracker.Configuration.Common.PoolDBESSource_cfi
0148
0149
0150 else:
0151 print("############ testPVValidation_cfg.py: msg%-i: Not applying extra calibration constants!")
0152
0153
0154
0155
0156 process.primaryVertexFilter = cms.EDFilter("VertexSelector",
0157 src = cms.InputTag("offlinePrimaryVertices"),
0158 cut = cms.string("!isFake && ndof > 4 && abs(z) <= 24 && position.Rho <= 2"),
0159 filter = cms.bool(True)
0160 )
0161
0162 process.noscraping = cms.EDFilter("FilterOutScraping",
0163 applyfilter = cms.untracked.bool(True),
0164 src = cms.untracked.InputTag(_theTrackCollection),
0165 debugOn = cms.untracked.bool(False),
0166 numtrack = cms.untracked.uint32(10),
0167 thresh = cms.untracked.double(0.25)
0168 )
0169
0170 process.noslowpt = cms.EDFilter("FilterOutLowPt",
0171 applyfilter = cms.untracked.bool(True),
0172 src = cms.untracked.InputTag(_theTrackCollection),
0173 debugOn = cms.untracked.bool(False),
0174 numtrack = cms.untracked.uint32(0),
0175 thresh = cms.untracked.int32(1),
0176 ptmin = cms.untracked.double(3.),
0177 runControl = cms.untracked.bool(True),
0178 runControlNumber = cms.untracked.vuint32(int(runboundary))
0179 )
0180
0181
0182
0183
0184 from RecoVertex.BeamSpotProducer.beamSpotCompatibilityChecker_cfi import beamSpotCompatibilityChecker
0185 process.BeamSpotChecker = beamSpotCompatibilityChecker.clone(
0186 bsFromFile = "offlineBeamSpot::RECO",
0187 bsFromDB = "offlineBeamSpot::@currentProcess",
0188 dbFromEvent = True,
0189 warningThr = 3,
0190 errorThr = 5,
0191 )
0192
0193 if isMC:
0194 process.goodvertexSkim = cms.Sequence(process.noscraping)
0195 else:
0196 process.goodvertexSkim = cms.Sequence(process.primaryVertexFilter + process.noscraping + process.noslowpt)
0197
0198
0199 if(theRefitter == RefitType.COMMON):
0200
0201 print("############ testPVValidation_cfg.py: msg%-i: using the common track selection and refit sequence!")
0202
0203
0204
0205 import Alignment.CommonAlignment.tools.trackselectionRefitting as trackselRefit
0206 process.seqTrackselRefit = trackselRefit.getSequence(process, _theTrackCollection ,
0207 isPVValidation=True,
0208 TTRHBuilder='WithAngleAndTemplate',
0209 usePixelQualityFlag=True,
0210 openMassWindow=False,
0211 cosmicsDecoMode=True,
0212 cosmicsZeroTesla=False,
0213 momentumConstraint=None,
0214 cosmicTrackSplitting=False,
0215 use_d0cut=False,
0216 )
0217
0218 elif (theRefitter == RefitType.STANDARD):
0219
0220 print("############ testPVValidation_cfg.py: msg%-i: using the standard single refit sequence!")
0221
0222
0223
0224
0225
0226
0227
0228
0229
0230
0231
0232
0233
0234 process.load("RecoTracker.TrackProducer.TrackRefitters_cff")
0235 import RecoTracker.TrackProducer.TrackRefitters_cff
0236 process.FinalTrackRefitter = RecoTracker.TrackProducer.TrackRefitter_cfi.TrackRefitter.clone()
0237 process.FinalTrackRefitter.src = _theTrackCollection
0238 process.FinalTrackRefitter.TrajectoryInEvent = True
0239 process.FinalTrackRefitter.NavigationSchool = ''
0240 process.FinalTrackRefitter.TTRHBuilder = "WithAngleAndTemplate"
0241
0242
0243
0244
0245 process.load("RecoLocalTracker.SiPixelRecHits.SiPixelTemplateStoreESProducer_cfi")
0246 process.seqTrackselRefit = cms.Sequence(process.offlineBeamSpot*
0247
0248
0249 process.FinalTrackRefitter,
0250 cms.Task(process.SiPixelTemplateStoreESProducer))
0251
0252
0253
0254
0255 process.TFileService = cms.Service("TFileService",
0256 fileName=cms.string("PVValidation_test_0.root")
0257 )
0258
0259
0260
0261
0262 from RecoVertex.PrimaryVertexProducer.OfflinePrimaryVertices_cfi import offlinePrimaryVertices
0263
0264 FilteringParams = offlinePrimaryVertices.TkFilterParameters.clone(
0265 maxNormalizedChi2 = 5.0,
0266 maxD0Significance = 5.0,
0267 maxEta = 5.0,
0268 )
0269
0270
0271 from RecoVertex.PrimaryVertexProducer.OfflinePrimaryVertices_cfi import DA_vectParameters
0272 DAClusterizationParams = DA_vectParameters.clone()
0273
0274 GapClusterizationParams = cms.PSet(algorithm = cms.string('gap'),
0275 TkGapClusParameters = cms.PSet(zSeparation = cms.double(0.2))
0276 )
0277
0278
0279
0280
0281 def switchClusterizerParameters(da):
0282 if da:
0283 print("############ testPVValidation_cfg.py: msg%-i: Running DA Algorithm!")
0284 return DAClusterizationParams
0285 else:
0286 print("############ testPVValidation_cfg.py: msg%-i: Running GAP Algorithm!")
0287 return GapClusterizationParams
0288
0289
0290
0291
0292 process.PVValidation = cms.EDAnalyzer("PrimaryVertexValidation",
0293 TrackCollectionTag = cms.InputTag("FinalTrackRefitter"),
0294 VertexCollectionTag = cms.InputTag("offlinePrimaryVertices"),
0295 Debug = cms.bool(False),
0296 storeNtuple = cms.bool(False),
0297 useTracksFromRecoVtx = cms.bool(False),
0298 isLightNtuple = cms.bool(True),
0299 askFirstLayerHit = cms.bool(False),
0300 forceBeamSpot = cms.untracked.bool(False),
0301 probePt = cms.untracked.double(3.),
0302 minPt = cms.untracked.double(1.),
0303 maxPt = cms.untracked.double(30.),
0304 runControl = cms.untracked.bool(True),
0305 runControlNumber = cms.untracked.vuint32(int(runboundary)),
0306 TkFilterParameters = FilteringParams,
0307 TkClusParameters = switchClusterizerParameters(isDA)
0308 )
0309
0310
0311
0312
0313
0314
0315
0316
0317
0318 process.p = cms.Path(process.goodvertexSkim*
0319 process.seqTrackselRefit*
0320 process.BeamSpotChecker*
0321 process.PVValidation)
0322
0323
0324 process.load("TrackingTools.TransientTrack.TransientTrackBuilder_cfi")
0325 process.offlinePrimaryVerticesFromRefittedTrks = offlinePrimaryVertices.clone()
0326 process.offlinePrimaryVerticesFromRefittedTrks.TrackLabel = cms.InputTag("FinalTrackRefitter")
0327 process.offlinePrimaryVerticesFromRefittedTrks.vertexCollections.maxDistanceToBeam = 1
0328 process.offlinePrimaryVerticesFromRefittedTrks.TkFilterParameters.maxNormalizedChi2 = 20
0329 process.offlinePrimaryVerticesFromRefittedTrks.TkFilterParameters.minSiliconLayersWithHits = 5
0330 process.offlinePrimaryVerticesFromRefittedTrks.TkFilterParameters.maxD0Significance = 5.0
0331
0332 process.offlinePrimaryVerticesFromRefittedTrks.TkFilterParameters.maxD0Error = 1.0
0333 process.offlinePrimaryVerticesFromRefittedTrks.TkFilterParameters.maxDzError = 1.0
0334 process.offlinePrimaryVerticesFromRefittedTrks.TkFilterParameters.minPixelLayersWithHits = 2
0335
0336
0337
0338
0339 from HLTrigger.HLTfilters.triggerResultsFilter_cfi import *
0340 process.HLTFilter = triggerResultsFilter.clone(
0341 triggerConditions = cms.vstring("HLT_ZeroBias_*"),
0342
0343 hltResults = cms.InputTag( "TriggerResults", "", "HLT" ),
0344 l1tResults = cms.InputTag( "" ),
0345 throw = cms.bool(False)
0346 )
0347
0348
0349
0350
0351 process.myanalysis = cms.EDAnalyzer("GeneralPurposeTrackAnalyzer",
0352 TkTag = cms.InputTag('FinalTrackRefitter'),
0353 doLatencyAnalysis = cms.bool(False),
0354 isCosmics = cms.bool(False)
0355 )
0356
0357
0358
0359
0360 process.PrimaryVertexResolution = cms.EDAnalyzer('SplitVertexResolution',
0361 storeNtuple = cms.bool(True),
0362 vtxCollection = cms.InputTag("offlinePrimaryVerticesFromRefittedTrks"),
0363 trackCollection = cms.InputTag("FinalTrackRefitter"),
0364 minVertexNdf = cms.untracked.double(10.),
0365 minVertexMeanWeight = cms.untracked.double(0.5),
0366 runControl = cms.untracked.bool(True),
0367 runControlNumber = cms.untracked.vuint32(int(runboundary))
0368 )
0369
0370
0371 process.p2 = cms.Path(process.HLTFilter +
0372 process.seqTrackselRefit +
0373 process.BeamSpotChecker +
0374 process.offlinePrimaryVerticesFromRefittedTrks +
0375 process.PrimaryVertexResolution +
0376 process.myanalysis)