pixelLessStepClassifier1.qualityCuts = [-0.4,0.0,0.4]
fastSim.toModify(pixelLessStepClassifier1, vertices = "firstStepPrimaryVerticesBeforeMixing" )
pixelLessStepClassifier2 = TrackMVAClassifierPrompt.clone()
pixelLessStepClassifier2.src = 'pixelLessStepTracks'
pixelLessStepClassifier2.mva.GBRForestLabel = 'MVASelectorIter0_13TeV'
pixelLessStepClassifier2.qualityCuts = [-0.0,0.0,0.0]
fastSim.toModify(pixelLessStepClassifier2, vertices = "firstStepPrimaryVerticesBeforeMixing" )
from RecoTracker.FinalTrackSelectors.ClassifierMerger_cfi import *
pixelLessStep = ClassifierMerger.clone()
pixelLessStep.inputClassifiers=['pixelLessStepClassifier1','pixelLessStepClassifier2']
from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
trackingPhase1.toReplaceWith(pixelLessStep, pixelLessStepClassifier1.clone(
mva = dict(GBRForestLabel = 'MVASelectorPixelLessStep_Phase1'),
qualityCuts = [-0.4,0.0,0.4],
))
# For LowPU
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
pixelLessStepSelector = RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
src='pixelLessStepTracks',
useAnyMVA = cms.bool(False),
GBRForestLabel = cms.string('MVASelectorIter5'),
trackSelectors= cms.VPSet(
RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
name = 'pixelLessStepLoose',
chi2n_par = 0.5,
res_par = ( 0.003, 0.001 ),
minNumberLayers = 4,
maxNumberLostLayers = 1,
ignoreSingleFPixPanelModules = True,
)
pixelPairStepHitDoubletsB = pixelPairStepHitDoublets.clone(
seedingLayers = "",
trackingRegions = "",
trackingRegionsSeedingLayers = "pixelPairStepTrackingRegionsSeedLayersB",
)
pixelPairStepSeedsB = pixelPairStepSeedsA.clone(seedingHitSets = "pixelPairStepHitDoubletsB")
# Merge
from RecoTracker.TkSeedGenerator.GlobalCombinedSeeds_cfi import globalCombinedSeeds as _globalCombinedSeeds
_pixelPairStepSeedsMerged = _globalCombinedSeeds.clone(
seedCollections = ["pixelPairStepSeedsA", "pixelPairStepSeedsB"],
)
trackingPhase1.toReplaceWith(pixelPairStepSeeds, _pixelPairStepSeedsMerged)
# QUALITY CUTS DURING TRACK BUILDING
import TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff
_pixelPairStepTrajectoryFilterBase = TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff.CkfBaseTrajectoryFilter_block.clone(
minimumNumberOfHits = 3,
minPt = 0.1,
)
pixelPairStepTrajectoryFilterBase = _pixelPairStepTrajectoryFilterBase.clone(
seedPairPenalty =0,
maxCCCLostHits = 0,
minGoodStripCharge = cms.PSet(refToPSet_ = cms.string('SiStripClusterChargeCutLoose'))
)
from Configuration.Eras.Modifier_tracker_apv_vfp30_2016_cff import tracker_apv_vfp30_2016
_tracker_apv_vfp30_2016.toModify(pixelPairStepTrajectoryFilterBase, maxCCCLostHits = 2)
name = 'hiPixelPairStepLoose',
useMVA = cms.bool(False)
), #end of pset
RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
name = 'hiPixelPairStepTight',
preFilterName = 'hiPixelPairStepLoose',
useMVA = cms.bool(False),
minMVA = cms.double(-0.58)
),
RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
name = 'hiPixelPairStep',
preFilterName = 'hiPixelPairStepTight',
useMVA = cms.bool(False),
minMVA = cms.double(0.77)
),
) #end of vpset
)
# Final sequence
hiPixelPairStep = cms.Sequence(hiPixelPairClusters*
hiPixelPairSeedLayers*
hiPixelPairSeeds*
hiPixelPairTrackCandidates*
hiPixelPairGlobalPrimTracks*
hiPixelPairStepSelector)
hiPixelPairStep_Phase1 = hiPixelPairStep.copy()
hiPixelPairStep_Phase1.replace(hiPixelPairSeeds,hiPixelPairStepTrackingRegionPhase1*hiPixelPairStepHitDoubletsPhase1*hiPixelPairStepSeedsPhase1)
trackingPhase1.toReplaceWith(hiPixelPairStep, hiPixelPairStep_Phase1)
)
import RecoTracker.FinalTrackSelectors.trackListMerger_cfi
hiDetachedTripletStepQual = RecoTracker.FinalTrackSelectors.trackListMerger_cfi.trackListMerger.clone(
TrackProducers=cms.VInputTag(cms.InputTag('hiDetachedTripletStepTracks')),
hasSelector=cms.vint32(1),
selectedTrackQuals = cms.VInputTag(cms.InputTag("hiDetachedTripletStepSelector","hiDetachedTripletStep")),
copyExtras = True,
makeReKeyedSeeds = cms.untracked.bool(False),
)
hiDetachedTripletStep = cms.Sequence(hiDetachedTripletStepClusters*
hiDetachedTripletStepSeedLayers*
hiDetachedTripletStepTrackingRegions*
hiDetachedTripletStepTracksHitDoublets*
hiDetachedTripletStepTracksHitTriplets*
pixelFitterByHelixProjections*
hiDetachedTripletStepPixelTracksFilter*
hiDetachedTripletStepPixelTracks*
hiDetachedTripletStepSeeds*
hiDetachedTripletStepTrackCandidates*
hiDetachedTripletStepTracks*
hiDetachedTripletStepSelector*
hiDetachedTripletStepQual)
hiDetachedTripletStep_Phase1 = hiDetachedTripletStep.copy()
hiDetachedTripletStep_Phase1.replace(hiDetachedTripletStepTracksHitDoublets, hiDetachedTripletStepTracksHitDoubletsCA)
hiDetachedTripletStep_Phase1.replace(hiDetachedTripletStepTracksHitTriplets, hiDetachedTripletStepTracksHitTripletsCA)
trackingPhase1.toReplaceWith(hiDetachedTripletStep, hiDetachedTripletStep_Phase1)
initialStepSeedsPreSplitting = _seedCreatorFromRegionConsecutiveHitsEDProducer.clone(
seedingHitSets="initialStepHitTripletsPreSplitting", )
from RecoPixelVertexing.PixelTriplets.caHitQuadrupletEDProducer_cfi import caHitQuadrupletEDProducer as _caHitQuadrupletEDProducer
trackingPhase1.toReplaceWith(
initialStepHitQuadrupletsPreSplitting,
_caHitQuadrupletEDProducer.clone(
doublets="initialStepHitDoubletsPreSplitting",
extraHitRPhitolerance=initialStepHitTripletsPreSplitting.
extraHitRPhitolerance,
SeedComparitorPSet=initialStepHitTripletsPreSplitting.
SeedComparitorPSet,
maxChi2=dict(
pt1=0.7,
pt2=2,
value1=200,
value2=50,
),
useBendingCorrection=True,
fitFastCircle=True,
fitFastCircleChi2Cut=True,
CAThetaCut=0.0012,
CAPhiCut=0.2,
))
trackingPhase1QuadProp.toModify(
initialStepHitTripletsPreSplitting,
produceSeedingHitSets=False,
produceIntermediateHitTriplets=True,
)
tobTecStepClassifier2.src = 'tobTecStepTracks'
tobTecStepClassifier2.mva.GBRForestLabel = 'MVASelectorIter0_13TeV'
tobTecStepClassifier2.qualityCuts = [0.0, 0.0, 0.0]
fastSim.toModify(tobTecStepClassifier2,
vertices="firstStepPrimaryVerticesBeforeMixing")
from RecoTracker.FinalTrackSelectors.ClassifierMerger_cfi import *
tobTecStep = ClassifierMerger.clone()
tobTecStep.inputClassifiers = [
'tobTecStepClassifier1', 'tobTecStepClassifier2'
]
from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
trackingPhase1.toReplaceWith(
tobTecStep,
tobTecStepClassifier1.clone(
mva=dict(GBRForestLabel='MVASelectorTobTecStep_Phase1'),
qualityCuts=[-0.6, -0.45, -0.3]))
from RecoTracker.FinalTrackSelectors.TrackLwtnnClassifier_cfi import *
from RecoTracker.FinalTrackSelectors.trackSelectionLwtnn_cfi import *
trackdnn.toReplaceWith(
tobTecStep,
TrackLwtnnClassifier.clone(src='tobTecStepTracks',
qualityCuts=[-0.4, -0.25, -0.1]))
(trackdnn & fastSim).toModify(tobTecStep,
vertices="firstStepPrimaryVerticesBeforeMixing")
pp_on_AA_2018.toModify(tobTecStep, qualityCuts=[-0.6, -0.3, 0.7])
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
initialStepSeedsPreSplitting,
initialStepTrackCandidatesPreSplitting,
initialStepTracksPreSplitting,
firstStepPrimaryVerticesPreSplitting,
initialStepTrackRefsForJetsPreSplitting,
caloTowerForTrkPreSplitting,
ak4CaloJetsForTrkPreSplitting,
jetsForCoreTrackingPreSplitting,
siPixelClusters,
siPixelRecHits,
MeasurementTrackerEvent,
siPixelClusterShapeCache)
InitialStepPreSplitting = cms.Sequence(InitialStepPreSplittingTask)
_InitialStepPreSplittingTask_trackingPhase1 = InitialStepPreSplittingTask.copy()
_InitialStepPreSplittingTask_trackingPhase1.replace(initialStepHitTripletsPreSplitting, cms.Task(initialStepHitTripletsPreSplitting,initialStepHitQuadrupletsPreSplitting))
trackingPhase1.toReplaceWith(InitialStepPreSplittingTask, _InitialStepPreSplittingTask_trackingPhase1.copyAndExclude([initialStepHitTripletsPreSplitting]))
# Although InitialStepPreSplitting is not really part of LowPU/Run1/Phase2PU140
# tracking, we use it to get siPixelClusters and siPixelRecHits
# collections for non-splitted pixel clusters. All modules before
# iterTracking sequence use siPixelClustersPreSplitting and
# siPixelRecHitsPreSplitting for that purpose.
#
# If siPixelClusters would be defined in
# RecoLocalTracker.Configuration.RecoLocalTracker_cff, we would have a
# situation where
# - LowPU/Phase2PU140 has siPixelClusters defined in RecoLocalTracker_cff
# - everything else has siPixelClusters defined here
# and this leads to a mess. The way it is done here we have only
# one place (within Reconstruction_cff) where siPixelClusters
tobTecStepClassifier1.qualityCuts = [-0.6,-0.45,-0.3]
fastSim.toModify(tobTecStepClassifier1, vertices = "firstStepPrimaryVerticesBeforeMixing")
tobTecStepClassifier2 = TrackMVAClassifierPrompt.clone()
tobTecStepClassifier2.src = 'tobTecStepTracks'
tobTecStepClassifier2.mva.GBRForestLabel = 'MVASelectorIter0_13TeV'
tobTecStepClassifier2.qualityCuts = [0.0,0.0,0.0]
fastSim.toModify(tobTecStepClassifier2,vertices = "firstStepPrimaryVerticesBeforeMixing")
from RecoTracker.FinalTrackSelectors.ClassifierMerger_cfi import *
tobTecStep = ClassifierMerger.clone()
tobTecStep.inputClassifiers=['tobTecStepClassifier1','tobTecStepClassifier2']
from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
trackingPhase1.toReplaceWith(tobTecStep, tobTecStepClassifier1.clone(
mva = dict(GBRForestLabel = 'MVASelectorTobTecStep_Phase1'),
qualityCuts = [-0.6,-0.45,-0.3]
))
from RecoTracker.FinalTrackSelectors.TrackLwtnnClassifier_cfi import *
from RecoTracker.FinalTrackSelectors.trackSelectionLwtnn_cfi import *
trackdnn.toReplaceWith(tobTecStep, TrackLwtnnClassifier.clone(
src = 'tobTecStepTracks',
qualityCuts = [-0.4, -0.25, -0.1]
))
(trackdnn & fastSim).toModify(tobTecStep,vertices = "firstStepPrimaryVerticesBeforeMixing")
pp_on_AA_2018.toModify(tobTecStep, qualityCuts = [-0.6,-0.3,0.7])
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
trackingLowPU.toReplaceWith(tobTecStep, RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
src = 'tobTecStepTracks',
trackingPhase1.toModify(
hiPixelPairStepSelector,
trackSelectors=cms.VPSet(
RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiLooseMTS.clone(
name='hiPixelPairStepLoose', useMVA=False), #end of pset
RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
name='hiPixelPairStepTight',
preFilterName='hiPixelPairStepLoose',
useMVA=False,
minMVA=-0.58),
RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
name='hiPixelPairStep',
preFilterName='hiPixelPairStepTight',
useMVA=False,
minMVA=0.77),
) #end of vpset
)
# Final sequence
hiPixelPairStepTask = cms.Task(hiPixelPairClusters, hiPixelPairSeedLayers,
hiPixelPairSeeds, hiPixelPairTrackCandidates,
hiPixelPairGlobalPrimTracks,
hiPixelPairStepSelector)
hiPixelPairStep = cms.Sequence(hiPixelPairStepTask)
hiPixelPairStep_Phase1 = hiPixelPairStepTask.copy()
hiPixelPairStep_Phase1.replace(
hiPixelPairSeeds,
cms.Task(hiPixelPairStepTrackingRegionPhase1,
hiPixelPairStepHitDoubletsPhase1, hiPixelPairStepSeedsPhase1))
trackingPhase1.toReplaceWith(hiPixelPairStepTask, hiPixelPairStep_Phase1)
Fitter = cms.string('FlexibleKFFittingSmoother')
)
fastSim.toModify(pixelPairStepTracks, TTRHBuilder = 'WithoutRefit')
# Final selection
from RecoTracker.FinalTrackSelectors.TrackMVAClassifierPrompt_cfi import *
pixelPairStep = TrackMVAClassifierPrompt.clone()
pixelPairStep.src = 'pixelPairStepTracks'
pixelPairStep.mva.GBRForestLabel = 'MVASelectorIter2_13TeV'
pixelPairStep.qualityCuts = [-0.2,0.0,0.3]
#LWTNN selector
from RecoTracker.FinalTrackSelectors.TrackLwtnnClassifier_cfi import *
from RecoTracker.FinalTrackSelectors.trackSelectionLwtnn_cfi import *
trackingPhase1.toReplaceWith(pixelPairStep, TrackLwtnnClassifier.clone(
src='pixelPairStepTracks',
qualityCuts=[-0.6, -0.1, 0.4],
))
highBetaStar_2018.toReplaceWith(pixelPairStep, TrackMVAClassifierPrompt.clone(
src='pixelPairStepTracks',
qualityCuts = [-0.95,0.0,0.3],
mva=dict(GBRForestLabel = 'MVASelectorPixelPairStep_Phase1')
))
pp_on_AA_2018.toReplaceWith(pixelPairStep, TrackMVAClassifierPrompt.clone(
src='pixelPairStepTracks',
qualityCuts = [-0.2, 0.0, 0.98],
mva=dict(GBRForestLabel = 'MVASelectorPixelPairStep_Phase1')
))
fastSim.toModify(pixelPairStep, vertices = "firstStepPrimaryVerticesBeforeMixing")
pixelPairStepTracks,
pixelPairStep)
PixelPairStep = cms.Sequence(PixelPairStepTask)
_PixelPairStepTask_LowPU_Phase2PU140 = PixelPairStepTask.copy()
_PixelPairStepTask_LowPU_Phase2PU140.replace(pixelPairStep, pixelPairStepSelector)
trackingLowPU.toReplaceWith(PixelPairStepTask, _PixelPairStepTask_LowPU_Phase2PU140)
trackingPhase2PU140.toReplaceWith(PixelPairStepTask, _PixelPairStepTask_LowPU_Phase2PU140)
_PixelPairStepTask_Phase1 = PixelPairStepTask.copy()
_PixelPairStepTask_pp_on_AA = PixelPairStepTask.copy()
_PixelPairStepTask_Phase1.replace(pixelPairStepSeeds,cms.Task(
pixelPairStepSeedsA ,
pixelPairStepTrackingRegionsSeedLayersB,pixelPairStepHitDoubletsB,pixelPairStepSeedsB,
pixelPairStepSeeds))
trackingPhase1.toReplaceWith(PixelPairStepTask, _PixelPairStepTask_Phase1)
_PixelPairStepTask_pp_on_AA.replace(pixelPairStepHitDoublets, cms.Task(pixelPairStepTrackingRegionsSeedLayersB,pixelPairStepHitDoubletsB))
(pp_on_XeXe_2017 | pp_on_AA_2018).toReplaceWith(PixelPairStepTask, _PixelPairStepTask_pp_on_AA)
#fastSim
import FastSimulation.Tracking.FastTrackerRecHitMaskProducer_cfi
pixelPairStepMasks = FastSimulation.Tracking.FastTrackerRecHitMaskProducer_cfi.maskProducerFromClusterRemover(pixelPairStepClusters)
fastSim.toReplaceWith(PixelPairStepTask,
cms.Task(pixelPairStepMasks
,pixelPairStepTrackingRegions
,pixelPairStepSeeds
,pixelPairStepTrackCandidates
,pixelPairStepTracks
,pixelPairStep
)
ret.append(item.replace("SeedsPair", "Seeds"))
continue
if "SeedsTripl" in item:
continue
ret.append(item)
return ret
_seedProducers = _filterForNtuple(_TrackValidation_cff._seedProducers)
_seedProducers_trackingPhase1 = _filterForNtuple(_TrackValidation_cff._seedProducers_trackingPhase1)
_seedProducers_trackingPhase2PU140 = _filterForNtuple(_TrackValidation_cff._seedProducers_trackingPhase2PU140)
(_seedSelectors, trackingNtupleSeedSelectors) = _TrackValidation_cff._addSeedToTrackProducers(_seedProducers, globals())
(_seedSelectors_trackingPhase1, _trackingNtupleSeedSelectors_trackingPhase1) = _TrackValidation_cff._addSeedToTrackProducers(_seedProducers_trackingPhase1, globals())
(_seedSelectors_trackingPhase2PU140, _trackingNtupleSeedSelectors_trackingPhase2PU140) = _TrackValidation_cff._addSeedToTrackProducers(_seedProducers_trackingPhase2PU140, globals())
from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
from Configuration.Eras.Modifier_trackingPhase2PU140_cff import trackingPhase2PU140
trackingPhase1.toReplaceWith(trackingNtupleSeedSelectors, _trackingNtupleSeedSelectors_trackingPhase1)
trackingPhase2PU140.toReplaceWith(trackingNtupleSeedSelectors, _trackingNtupleSeedSelectors_trackingPhase2PU140)
trackingNtuple.seedTracks = _seedSelectors
trackingPhase1.toModify(trackingNtuple, seedTracks = _seedSelectors_trackingPhase1)
trackingPhase2PU140.toModify(trackingNtuple, seedTracks = _seedSelectors_trackingPhase2PU140)
def _seedProdToTrackCands(name):
return name.replace("seedTracks", "").replace("Seeds", "TrackCandidates")
trackingNtuple.trackCandidates = [_seedProdToTrackCands(i) for i in _seedProducers]
trackingPhase1.toModify(trackingNtuple, trackCandidates=[_seedProdToTrackCands(i) for i in _seedProducers_trackingPhase1])
trackingPhase2PU140.toModify(trackingNtuple, trackCandidates=[_seedProdToTrackCands(i) for i in _seedProducers_trackingPhase2PU140])
trackingNtupleTask = cms.Task()
# reproduce hits because they're not stored in RECO
if _includeHits:
# Ordered Hits
SeedingHitSets = cms.InputTag("hiPixel3PrimTracksHitTriplets"),
# Fitter
Fitter = cms.InputTag("pixelFitterByHelixProjections"),
# Filter
Filter = cms.InputTag("hiFilter"),
# Cleaner
Cleaner = cms.string("trackCleaner")
)
trackingPhase1.toModify(hiPixel3PrimTracks,
SeedingHitSets = cms.InputTag("hiPixel3PrimTracksHitQuadrupletsCA"),
)
hiPixel3PrimTracksSequence = cms.Sequence(
hiTrackingRegionWithVertex +
hiPixel3PrimTracksHitDoublets +
hiPixel3PrimTracksHitTriplets +
pixelFitterByHelixProjections +
hiFilter +
hiPixel3PrimTracks
)
#phase 1 changes
hiPixel3PrimTracksSequence_Phase1 = hiPixel3PrimTracksSequence.copy()
hiPixel3PrimTracksSequence_Phase1.replace(hiPixel3PrimTracksHitDoublets,hiPixelLayerQuadruplets+hiPixel3PrimTracksHitDoubletsCA)
hiPixel3PrimTracksSequence_Phase1.replace(hiPixel3PrimTracksHitTriplets,hiPixel3PrimTracksHitQuadrupletsCA)
trackingPhase1.toReplaceWith(hiPixel3PrimTracksSequence,hiPixel3PrimTracksSequence_Phase1)
value1 = 0.2, value2 = 0.05,
enabled = True,
),
useBendingCorrection = True,
fitFastCircle = True,
fitFastCircleChi2Cut = True,
)
from RecoPixelVertexing.PixelTriplets.caHitQuadrupletEDProducer_cfi import caHitQuadrupletEDProducer as _caHitQuadrupletEDProducer
trackingPhase1.toModify(detachedQuadStepHitDoublets, layerPairs = [0,1,2]) # layer pairs (0,1), (1,2), (2,3)
trackingPhase1.toReplaceWith(detachedQuadStepHitQuadruplets, _caHitQuadrupletEDProducer.clone(
doublets = "detachedQuadStepHitDoublets",
extraHitRPhitolerance = detachedQuadStepHitTriplets.extraHitRPhitolerance,
maxChi2 = dict(
pt1 = 0.8, pt2 = 2,
value1 = 500, value2 = 100,
),
useBendingCorrection = True,
fitFastCircle = True,
fitFastCircleChi2Cut = True,
CAThetaCut = 0.0011,
CAPhiCut = 0,
))
from RecoPixelVertexing.PixelTriplets.pixelQuadrupletMergerEDProducer_cfi import pixelQuadrupletMergerEDProducer as _pixelQuadrupletMergerEDProducer
from RecoPixelVertexing.PixelTriplets.quadrupletseedmerging_cff import *
_detachedQuadStepHitQuadrupletsMerging = _pixelQuadrupletMergerEDProducer.clone(
triplets = "detachedQuadStepHitTriplets",
layerList = dict(refToPSet_ = cms.string("PixelSeedMergerQuadruplets")),
)
from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer_cff import seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer as _seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer
detachedQuadStepSeeds = _seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer.clone(
tripletElectronHitDoublets*
tripletElectronHitTriplets*
tripletElectronSeeds*
tripletElectronClusterMask*
pixelPairElectronSeedLayers*
pixelPairElectronTrackingRegions*
pixelPairElectronHitDoublets*
pixelPairElectronSeeds*
stripPairElectronSeedLayers*
stripPairElectronTrackingRegions*
stripPairElectronHitDoublets*
stripPairElectronSeeds*
newCombinedSeeds)
_electronSeedsSeq_Phase1 = electronSeedsSeq.copy()
_electronSeedsSeq_Phase1.replace(pixelPairStepSeedClusterMask, detachedTripletStepSeedClusterMask)
trackingPhase1.toReplaceWith(electronSeedsSeq, _electronSeedsSeq_Phase1)
trackingPhase1QuadProp.toReplaceWith(electronSeedsSeq, _electronSeedsSeq_Phase1)
trackingPhase1PU70.toReplaceWith(electronSeedsSeq, cms.Sequence(
initialStepSeedClusterMask*
highPtTripletStepSeedClusterMask*
pixelPairStepSeedClusterMask*
tripletElectronSeedLayers*
tripletElectronTrackingRegions*
tripletElectronHitDoublets*
tripletElectronHitTriplets*
tripletElectronSeeds*
newCombinedSeeds
))
trackingPhase2PU140.toReplaceWith(electronSeedsSeq, cms.Sequence(
initialStepSeedClusterMask*
highPtTripletStepSeedClusterMask*
fitFastCircle=True,
fitFastCircleChi2Cut=True,
SeedComparitorPSet=initialStepHitTriplets.SeedComparitorPSet)
from RecoPixelVertexing.PixelTriplets.caHitQuadrupletEDProducer_cfi import caHitQuadrupletEDProducer as _caHitQuadrupletEDProducer
trackingPhase1.toModify(initialStepHitDoublets,
layerPairs=[0, 1,
2]) # layer pairs (0,1), (1,2), (2,3)
trackingPhase1.toReplaceWith(
initialStepHitQuadruplets,
_caHitQuadrupletEDProducer.clone(
doublets="initialStepHitDoublets",
extraHitRPhitolerance=initialStepHitTriplets.extraHitRPhitolerance,
SeedComparitorPSet=initialStepHitTriplets.SeedComparitorPSet,
maxChi2=dict(
pt1=0.7,
pt2=2,
value1=200,
value2=50,
),
useBendingCorrection=True,
fitFastCircle=True,
fitFastCircleChi2Cut=True,
CAThetaCut=0.0012,
CAPhiCut=0.2,
))
trackingPhase2PU140.toModify(
initialStepHitTriplets,
produceSeedingHitSets=False,
produceIntermediateHitTriplets=True,
)
from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer_cff import seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer as _seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer
_initialStepSeedsConsecutiveHitsTripletOnly = _seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer.clone(
generalTracksBeforeMixing = RecoTracker.FinalTrackSelectors.earlyGeneralTracks_cfi.earlyGeneralTracks.clone()
iterTracking = cms.Sequence(
InitialStep
+DetachedTripletStep
+LowPtTripletStep
+PixelPairStep
+MixedTripletStep
+PixelLessStep
+TobTecStep
+JetCoreRegionalStep
+generalTracksBeforeMixing)
_iterTracking_Phase1 = cms.Sequence(
InitialStep
+LowPtQuadStep
+HighPtTripletStep
+LowPtTripletStep
+DetachedQuadStep
+DetachedTripletStep
+PixelPairStep
+MixedTripletStep
+PixelLessStep
+TobTecStep
+JetCoreRegionalStep
+generalTracksBeforeMixing)
trackingPhase1.toReplaceWith(iterTracking, _iterTracking_Phase1)
tripletElectronSeedLayers,
tripletElectronTrackingRegions,
tripletElectronHitDoublets,
tripletElectronHitTriplets,
tripletElectronSeeds,
tripletElectronClusterMask,
pixelPairElectronSeedLayers,
pixelPairElectronTrackingRegions,
pixelPairElectronHitDoublets,
pixelPairElectronSeeds,
stripPairElectronSeedLayers,
stripPairElectronTrackingRegions,
stripPairElectronHitDoublets,
stripPairElectronSeeds,
newCombinedSeeds)
electronSeedsSeq = cms.Sequence(electronSeedsSeqTask)
_electronSeedsSeqTask_Phase1 = electronSeedsSeqTask.copy()
_electronSeedsSeqTask_Phase1.replace(pixelPairStepSeedClusterMask, detachedTripletStepSeedClusterMask)
trackingPhase1.toReplaceWith(electronSeedsSeqTask, _electronSeedsSeqTask_Phase1 )
trackingPhase2PU140.toReplaceWith(electronSeedsSeqTask, cms.Task(
initialStepSeedClusterMask,
highPtTripletStepSeedClusterMask,
pixelPairStepSeedClusterMask,
tripletElectronSeedLayers,
tripletElectronTrackingRegions,
tripletElectronHitDoublets,
tripletElectronHitTriplets,
tripletElectronSeeds,
newCombinedSeeds
))
fitFastCircleChi2Cut=True,
)
from RecoPixelVertexing.PixelTriplets.caHitQuadrupletEDProducer_cfi import caHitQuadrupletEDProducer as _caHitQuadrupletEDProducer
trackingPhase1.toModify(detachedQuadStepHitDoublets,
layerPairs=[0, 1,
2]) # layer pairs (0,1), (1,2), (2,3)
trackingPhase1.toReplaceWith(
detachedQuadStepHitQuadruplets,
_caHitQuadrupletEDProducer.clone(
doublets="detachedQuadStepHitDoublets",
extraHitRPhitolerance=detachedQuadStepHitTriplets.
extraHitRPhitolerance,
maxChi2=dict(
pt1=0.8,
pt2=2,
value1=500,
value2=100,
),
useBendingCorrection=True,
fitFastCircle=True,
fitFastCircleChi2Cut=True,
CAThetaCut=0.0011,
CAPhiCut=0,
))
from RecoPixelVertexing.PixelTriplets.pixelQuadrupletMergerEDProducer_cfi import pixelQuadrupletMergerEDProducer as _pixelQuadrupletMergerEDProducer
from RecoPixelVertexing.PixelTriplets.quadrupletseedmerging_cff import *
_detachedQuadStepHitQuadrupletsMerging = _pixelQuadrupletMergerEDProducer.clone(
triplets="detachedQuadStepHitTriplets",
layerList=dict(refToPSet_=cms.string("PixelSeedMergerQuadruplets")),
from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsEDProducer_cff import seedCreatorFromRegionConsecutiveHitsEDProducer as _seedCreatorFromRegionConsecutiveHitsEDProducer
lowPtTripletStepSeeds = _seedCreatorFromRegionConsecutiveHitsEDProducer.clone(
seedingHitSets="lowPtTripletStepHitTriplets", )
from RecoPixelVertexing.PixelTriplets.caHitTripletEDProducer_cfi import caHitTripletEDProducer as _caHitTripletEDProducer
trackingPhase1.toModify(lowPtTripletStepHitDoublets,
layerPairs=[0, 1]) # layer pairs (0,1), (1,2)
trackingPhase1.toReplaceWith(
lowPtTripletStepHitTriplets,
_caHitTripletEDProducer.clone(
doublets="lowPtTripletStepHitDoublets",
extraHitRPhitolerance=lowPtTripletStepHitTriplets.
extraHitRPhitolerance,
SeedComparitorPSet=lowPtTripletStepHitTriplets.SeedComparitorPSet,
maxChi2=dict(
pt1=0.8,
pt2=2,
value1=70,
value2=8,
),
useBendingCorrection=True,
CAThetaCut=0.002,
CAPhiCut=0.05,
))
# QUALITY CUTS DURING TRACK BUILDING
import TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff as _TrajectoryFilter_cff
_lowPtTripletStepStandardTrajectoryFilterBase = _TrajectoryFilter_cff.CkfBaseTrajectoryFilter_block.clone(
minimumNumberOfHits=3,
minPt=0.075,
)
, hiCentrality
#, centralityBin # temporarily removed
, hiClusterCompatibility
, hiEvtPlane
, hcalnoise
, muonRecoHighLevelPbPbTask
, particleFlowLinksTask
, hiRecoPFJetsTask
, reducedRecHitsTask
)
globalRecoPbPb = cms.Sequence(globalRecoPbPbTask)
globalRecoPbPb_wPhase1Task = globalRecoPbPbTask.copy()
globalRecoPbPb_wPhase1Task.replace(hiTracking_wSplittingTask, hiTracking_wSplitting_Phase1Task)
from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
trackingPhase1.toReplaceWith(globalRecoPbPbTask, globalRecoPbPb_wPhase1Task)
globalRecoPbPb_wConformalPixelTask = cms.Task(hiTracking_wConformalPixelTask
, hiParticleFlowLocalRecoTask
, hiEcalClustersTask
, hiRecoJetsTask
, muonRecoPbPbTask
, hiElectronTask
, hiEgammaTask
, hiParticleFlowRecoTask
, egammaHighLevelRecoPostPFTask
, hiCentrality
, hiClusterCompatibility
, hiEvtPlane
, hcalnoise
)
from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer_cff import seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer as _seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer
_initialStepSeedsConsecutiveHitsTripletOnly = _seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer.clone(
seedingHitSets="initialStepHitTriplets",
SeedComparitorPSet=
dict( # FIXME: is this defined in any cfi that could be imported instead of copy-paste?
ComponentName='PixelClusterShapeSeedComparitor',
FilterAtHelixStage=cms.bool(False),
FilterPixelHits=cms.bool(True),
FilterStripHits=cms.bool(False),
ClusterShapeHitFilterName=cms.string('ClusterShapeHitFilter'),
ClusterShapeCacheSrc=cms.InputTag('siPixelClusterShapeCache')),
)
trackingPhase1.toReplaceWith(
initialStepSeeds,
_initialStepSeedsConsecutiveHitsTripletOnly.clone(
seedingHitSets="initialStepHitQuadruplets"))
trackingPhase2PU140.toReplaceWith(
initialStepSeeds,
_initialStepSeedsConsecutiveHitsTripletOnly.clone(
seedingHitSets="initialStepHitQuadruplets"))
import FastSimulation.Tracking.TrajectorySeedProducer_cfi
from FastSimulation.Tracking.SeedingMigration import _hitSetProducerToFactoryPSet
_fastSim_initialStepSeeds = FastSimulation.Tracking.TrajectorySeedProducer_cfi.trajectorySeedProducer.clone(
trackingRegions="initialStepTrackingRegions",
seedFinderSelector=dict(
pixelTripletGeneratorFactory=_hitSetProducerToFactoryPSet(
initialStepHitTriplets),
layerList=initialStepSeedLayers.layerList.value()))
_fastSim_initialStepSeeds.seedFinderSelector.pixelTripletGeneratorFactory.SeedComparitorPSet.ComponentName = "none"
#new for phase1
trackingRegions = "lowPtTripletStepTrackingRegions",
hitMasks = cms.InputTag("lowPtTripletStepMasks"),
)
from FastSimulation.Tracking.SeedingMigration import _hitSetProducerToFactoryPSet
_fastSim_lowPtTripletStepSeeds.seedFinderSelector.pixelTripletGeneratorFactory = _hitSetProducerToFactoryPSet(lowPtTripletStepHitTriplets)
_fastSim_lowPtTripletStepSeeds.seedFinderSelector.pixelTripletGeneratorFactory.SeedComparitorPSet.ComponentName = "none"
fastSim.toReplaceWith(lowPtTripletStepSeeds,_fastSim_lowPtTripletStepSeeds)
from RecoPixelVertexing.PixelTriplets.caHitTripletEDProducer_cfi import caHitTripletEDProducer as _caHitTripletEDProducer
trackingPhase1.toModify(lowPtTripletStepHitDoublets, layerPairs = [0,1]) # layer pairs (0,1), (1,2)
trackingPhase1.toReplaceWith(lowPtTripletStepHitTriplets, _caHitTripletEDProducer.clone(
doublets = "lowPtTripletStepHitDoublets",
extraHitRPhitolerance = lowPtTripletStepHitTriplets.extraHitRPhitolerance,
SeedComparitorPSet = lowPtTripletStepHitTriplets.SeedComparitorPSet,
maxChi2 = dict(
pt1 = 0.8, pt2 = 2,
value1 = 70 , value2 = 8,
),
useBendingCorrection = True,
CAThetaCut = 0.002,
CAPhiCut = 0.05,
))
trackingPhase2PU140.toModify(lowPtTripletStepHitDoublets, layerPairs = [0,1]) # layer pairs (0,1), (1,2)
trackingPhase2PU140.toReplaceWith(lowPtTripletStepHitTriplets, _caHitTripletEDProducer.clone(
doublets = "lowPtTripletStepHitDoublets",
extraHitRPhitolerance = lowPtTripletStepHitTriplets.extraHitRPhitolerance,
SeedComparitorPSet = lowPtTripletStepHitTriplets.SeedComparitorPSet,
maxChi2 = dict(
pt1 = 0.8, pt2 = 2,
value1 = 70 , value2 = 8,
),
)
from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer_cff import seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer as _seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer
_initialStepSeedsConsecutiveHitsTripletOnly = _seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer.clone(
seedingHitSets = "initialStepHitTriplets",
SeedComparitorPSet = dict(# FIXME: is this defined in any cfi that could be imported instead of copy-paste?
ComponentName = 'PixelClusterShapeSeedComparitor',
FilterAtHelixStage = cms.bool(False),
FilterPixelHits = cms.bool(True),
FilterStripHits = cms.bool(False),
ClusterShapeHitFilterName = cms.string('ClusterShapeHitFilter'),
ClusterShapeCacheSrc = cms.InputTag('siPixelClusterShapeCache')
),
)
trackingPhase1.toReplaceWith(initialStepSeeds, _initialStepSeedsConsecutiveHitsTripletOnly.clone(
seedingHitSets = "initialStepHitQuadruplets"
))
trackingPhase2PU140.toReplaceWith(initialStepSeeds, _initialStepSeedsConsecutiveHitsTripletOnly.clone(
seedingHitSets = "initialStepHitQuadruplets"
))
import FastSimulation.Tracking.TrajectorySeedProducer_cfi
from FastSimulation.Tracking.SeedingMigration import _hitSetProducerToFactoryPSet
_fastSim_initialStepSeeds = FastSimulation.Tracking.TrajectorySeedProducer_cfi.trajectorySeedProducer.clone(
trackingRegions = "initialStepTrackingRegions",
seedFinderSelector = dict( pixelTripletGeneratorFactory = _hitSetProducerToFactoryPSet(initialStepHitTriplets),
layerList = initialStepSeedLayers.layerList.value())
)
_fastSim_initialStepSeeds.seedFinderSelector.pixelTripletGeneratorFactory.SeedComparitorPSet.ComponentName = "none"
#new for phase1
trackingPhase1.toModify(_fastSim_initialStepSeeds, seedFinderSelector = dict(
pixelTripletGeneratorFactory = None,
* hiRecoJets
* muonRecoPbPb
* hiElectronSequence
* hiEgammaSequence
* hiParticleFlowReco
* hiCentrality
#* centralityBin # temporarily removed
* hiClusterCompatibility
* hiEvtPlane
* hcalnoise
* muonRecoHighLevelPbPb
)
globalRecoPbPb_wPhase1 = globalRecoPbPb.copy()
globalRecoPbPb_wPhase1.replace(hiTracking_wSplitting, hiTracking_wSplitting_Phase1)
from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
trackingPhase1.toReplaceWith(globalRecoPbPb, globalRecoPbPb_wPhase1)
globalRecoPbPb_wConformalPixel = cms.Sequence(hiTracking_wConformalPixel
* hiParticleFlowLocalReco
* hiEcalClusters
* hiRecoJets
* muonRecoPbPb
* hiElectronSequence
* hiEgammaSequence
* hiParticleFlowReco
* hiCentrality
#* centralityBin # temporarily removed
* hiClusterCompatibility
* hiEvtPlane
* hcalnoise
FilterAtHelixStage = cms.bool(False),
FilterPixelHits = cms.bool(True),
FilterStripHits = cms.bool(False),
ClusterShapeHitFilterName = cms.string('ClusterShapeHitFilter'),
ClusterShapeCacheSrc = cms.InputTag('siPixelClusterShapeCache')
),
)
from RecoPixelVertexing.PixelTriplets.caHitTripletEDProducer_cfi import caHitTripletEDProducer as _caHitTripletEDProducer
trackingPhase1.toModify(detachedTripletStepHitDoublets, layerPairs = [0,1]) # layer pairs (0,1), (1,2)
trackingPhase1.toReplaceWith(detachedTripletStepHitTriplets, _caHitTripletEDProducer.clone(
doublets = "detachedTripletStepHitDoublets",
extraHitRPhitolerance = detachedTripletStepHitTriplets.extraHitRPhitolerance,
maxChi2 = dict(
pt1 = 0.8, pt2 = 2,
value1 = 300 , value2 = 10,
),
useBendingCorrection = True,
CAThetaCut = 0.001,
CAPhiCut = 0,
CAHardPtCut = 0.2,
))
highBetaStar_2018.toModify(detachedTripletStepHitTriplets,CAThetaCut = 0.002,CAPhiCut = 0.1,CAHardPtCut = 0)
import FastSimulation.Tracking.TrajectorySeedProducer_cfi
_fastSim_detachedTripletStepSeeds = FastSimulation.Tracking.TrajectorySeedProducer_cfi.trajectorySeedProducer.clone(
trackingRegions = "detachedTripletStepTrackingRegions",
hitMasks = cms.InputTag("detachedTripletStepMasks"),
seedFinderSelector = dict( pixelTripletGeneratorFactory = _hitSetProducerToFactoryPSet(detachedTripletStepHitTriplets),
layerList = detachedTripletStepSeedLayers.layerList.value())
)
#new for phase1
SeedingHitSets=cms.InputTag("hiPixel3PrimTracksHitTriplets"),
# Fitter
Fitter=cms.InputTag("pixelFitterByHelixProjections"),
# Filter
Filter=cms.InputTag("hiFilter"),
# Cleaner
Cleaner=cms.string("trackCleaner"))
trackingPhase1.toModify(
hiPixel3PrimTracks,
SeedingHitSets="hiPixel3PrimTracksHitQuadrupletsCA",
)
hiPixel3PrimTracksTask = cms.Task(hiTrackingRegionWithVertex,
hiPixel3PrimTracksHitDoublets,
hiPixel3PrimTracksHitTriplets,
pixelFitterByHelixProjections, hiFilter,
hiPixel3PrimTracks)
hiPixel3PrimTracksSequence = cms.Sequence(hiPixel3PrimTracksTask)
#phase 1 changes
hiPixel3PrimTracksTask_Phase1 = hiPixel3PrimTracksTask.copy()
hiPixel3PrimTracksTask_Phase1.replace(
hiPixel3PrimTracksHitDoublets,
cms.Task(hiPixelLayerQuadruplets, hiPixel3PrimTracksHitDoubletsCA))
hiPixel3PrimTracksTask_Phase1.replace(hiPixel3PrimTracksHitTriplets,
hiPixel3PrimTracksHitQuadrupletsCA)
trackingPhase1.toReplaceWith(hiPixel3PrimTracksTask,
hiPixel3PrimTracksTask_Phase1)
import RecoTracker.FinalTrackSelectors.trackListMerger_cfi
hiLowPtTripletStepQual = RecoTracker.FinalTrackSelectors.trackListMerger_cfi.trackListMerger.clone(
TrackProducers = cms.VInputTag(cms.InputTag('hiLowPtTripletStepTracks')),
hasSelector=cms.vint32(1),
selectedTrackQuals = cms.VInputTag(cms.InputTag("hiLowPtTripletStepSelector","hiLowPtTripletStep")),
copyExtras = True,
makeReKeyedSeeds = cms.untracked.bool(False),
#writeOnlyTrkQuals = True
)
# Final sequence
hiLowPtTripletStep = cms.Sequence(hiLowPtTripletStepClusters*
hiLowPtTripletStepSeedLayers*
hiLowPtTripletStepTrackingRegions*
hiLowPtTripletStepTracksHitDoublets*
hiLowPtTripletStepTracksHitTriplets*
pixelFitterByHelixProjections*
hiLowPtTripletStepPixelTracksFilter*
hiLowPtTripletStepPixelTracks*hiLowPtTripletStepSeeds*
hiLowPtTripletStepTrackCandidates*
hiLowPtTripletStepTracks*
hiLowPtTripletStepSelector*
hiLowPtTripletStepQual
)
hiLowPtTripletStep_Phase1 = hiLowPtTripletStep.copy()
hiLowPtTripletStep_Phase1.replace(hiLowPtTripletStepTracksHitDoublets, hiLowPtTripletStepTracksHitDoubletsCA)
hiLowPtTripletStep_Phase1.replace(hiLowPtTripletStepTracksHitTriplets, hiLowPtTripletStepTracksHitTripletsCA)
trackingPhase1.toReplaceWith(hiLowPtTripletStep, hiLowPtTripletStep_Phase1)
from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsEDProducer_cff import seedCreatorFromRegionConsecutiveHitsEDProducer as _seedCreatorFromRegionConsecutiveHitsEDProducer
highPtTripletStepSeeds = _seedCreatorFromRegionConsecutiveHitsEDProducer.clone(
seedingHitSets="highPtTripletStepHitTriplets", )
from RecoPixelVertexing.PixelTriplets.caHitTripletEDProducer_cfi import caHitTripletEDProducer as _caHitTripletEDProducer
trackingPhase1.toModify(highPtTripletStepHitDoublets,
layerPairs=[0, 1]) # layer pairs (0,1), (1,2)
trackingPhase1.toReplaceWith(
highPtTripletStepHitTriplets,
_caHitTripletEDProducer.clone(
doublets="highPtTripletStepHitDoublets",
extraHitRPhitolerance=highPtTripletStepHitTriplets.
extraHitRPhitolerance,
SeedComparitorPSet=highPtTripletStepHitTriplets.SeedComparitorPSet,
maxChi2=dict(
pt1=0.8,
pt2=8,
value1=100,
value2=6,
),
useBendingCorrection=True,
CAThetaCut=0.004,
CAPhiCut=0.07,
CAHardPtCut=0.3,
))
# QUALITY CUTS DURING TRACK BUILDING
import TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff as _TrajectoryFilter_cff
_highPtTripletStepTrajectoryFilterBase = _TrajectoryFilter_cff.CkfBaseTrajectoryFilter_block.clone(
minimumNumberOfHits=3,
minPt=0.2,
)
phi = [3.0],
)
)
)
pixelPairStepHitDoubletsB = pixelPairStepHitDoublets.clone(
seedingLayers = "pixelPairStepSeedLayersB",
trackingRegions = "pixelPairStepTrackingRegionsB",
)
pixelPairStepSeedsB = pixelPairStepSeedsA.clone(seedingHitSets = "pixelPairStepHitDoubletsB")
# Merge
from RecoTracker.TkSeedGenerator.GlobalCombinedSeeds_cfi import globalCombinedSeeds as _globalCombinedSeeds
_pixelPairStepSeedsMerged = _globalCombinedSeeds.clone(
seedCollections = ["pixelPairStepSeedsA", "pixelPairStepSeedsB"],
)
trackingPhase1.toReplaceWith(pixelPairStepSeeds, _pixelPairStepSeedsMerged)
trackingPhase1QuadProp.toReplaceWith(pixelPairStepSeeds, _pixelPairStepSeedsMerged)
# QUALITY CUTS DURING TRACK BUILDING
import TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff
_pixelPairStepTrajectoryFilterBase = TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff.CkfBaseTrajectoryFilter_block.clone(
minimumNumberOfHits = 3,
minPt = 0.1,
)
pixelPairStepTrajectoryFilterBase = _pixelPairStepTrajectoryFilterBase.clone(
seedPairPenalty =0,
maxCCCLostHits = 0,
minGoodStripCharge = cms.PSet(refToPSet_ = cms.string('SiStripClusterChargeCutLoose'))
)
enabled = True,
),
useBendingCorrection = True,
fitFastCircle = True,
fitFastCircleChi2Cut = True,
SeedComparitorPSet = lowPtQuadStepHitTriplets.SeedComparitorPSet,
)
from RecoPixelVertexing.PixelTriplets.caHitQuadrupletEDProducer_cfi import caHitQuadrupletEDProducer as _caHitQuadrupletEDProducer
trackingPhase1.toModify(lowPtQuadStepHitDoublets, layerPairs = [0,1,2]) # layer pairs (0,1), (1,2), (2,3)
trackingPhase1.toReplaceWith(lowPtQuadStepHitQuadruplets, _caHitQuadrupletEDProducer.clone(
doublets = "lowPtQuadStepHitDoublets",
extraHitRPhitolerance = lowPtQuadStepHitTriplets.extraHitRPhitolerance,
SeedComparitorPSet = lowPtQuadStepHitTriplets.SeedComparitorPSet,
maxChi2 = dict(
pt1 = 0.7, pt2 = 2,
value1 = 1000, value2 = 150,
),
useBendingCorrection = True,
fitFastCircle = True,
fitFastCircleChi2Cut = True,
CAThetaCut = 0.0017,
CAPhiCut = 0.3,
))
from RecoPixelVertexing.PixelTriplets.pixelQuadrupletMergerEDProducer_cfi import pixelQuadrupletMergerEDProducer as _pixelQuadrupletMergerEDProducer
from RecoPixelVertexing.PixelTriplets.quadrupletseedmerging_cff import *
_lowPtQuadStepHitQuadrupletsMerging = _pixelQuadrupletMergerEDProducer.clone(
triplets = "lowPtQuadStepHitTriplets",
layerList = dict(refToPSet_ = cms.string("PixelSeedMergerQuadruplets")),
)
from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsEDProducer_cff import seedCreatorFromRegionConsecutiveHitsEDProducer as _seedCreatorFromRegionConsecutiveHitsEDProducer
lowPtQuadStepSeeds = _seedCreatorFromRegionConsecutiveHitsEDProducer.clone(
_PixelPairStepTask_LowPU_Phase2PU140 = PixelPairStepTask.copy()
_PixelPairStepTask_LowPU_Phase2PU140.replace(pixelPairStep,
pixelPairStepSelector)
trackingLowPU.toReplaceWith(PixelPairStepTask,
_PixelPairStepTask_LowPU_Phase2PU140)
trackingPhase2PU140.toReplaceWith(PixelPairStepTask,
_PixelPairStepTask_LowPU_Phase2PU140)
_PixelPairStepTask_Phase1 = PixelPairStepTask.copy()
_PixelPairStepTask_pp_on_AA = PixelPairStepTask.copy()
_PixelPairStepTask_Phase1.replace(
pixelPairStepSeeds,
cms.Task(pixelPairStepSeedsA, pixelPairStepTrackingRegionsSeedLayersB,
pixelPairStepHitDoubletsB, pixelPairStepSeedsB,
pixelPairStepSeeds))
trackingPhase1.toReplaceWith(PixelPairStepTask, _PixelPairStepTask_Phase1)
_PixelPairStepTask_pp_on_AA.replace(
pixelPairStepHitDoublets,
cms.Task(pixelPairStepTrackingRegionsSeedLayersB,
pixelPairStepHitDoubletsB))
(pp_on_XeXe_2017 | pp_on_AA).toReplaceWith(PixelPairStepTask,
_PixelPairStepTask_pp_on_AA)
#fastSim
import FastSimulation.Tracking.FastTrackerRecHitMaskProducer_cfi
pixelPairStepMasks = FastSimulation.Tracking.FastTrackerRecHitMaskProducer_cfi.maskProducerFromClusterRemover(
pixelPairStepClusters)
fastSim.toReplaceWith(
PixelPairStepTask,
cms.Task(pixelPairStepMasks, pixelPairStepTrackingRegions,
enabled = True,
),
useBendingCorrection = True,
fitFastCircle = True,
fitFastCircleChi2Cut = True,
SeedComparitorPSet = initialStepHitTriplets.SeedComparitorPSet
)
from RecoPixelVertexing.PixelTriplets.caHitQuadrupletEDProducer_cfi import caHitQuadrupletEDProducer as _caHitQuadrupletEDProducer
trackingPhase1.toModify(initialStepHitDoublets, layerPairs = [0,1,2]) # layer pairs (0,1), (1,2), (2,3)
trackingPhase1.toReplaceWith(initialStepHitQuadruplets, _caHitQuadrupletEDProducer.clone(
doublets = "initialStepHitDoublets",
extraHitRPhitolerance = initialStepHitTriplets.extraHitRPhitolerance,
SeedComparitorPSet = initialStepHitTriplets.SeedComparitorPSet,
maxChi2 = dict(
pt1 = 0.7, pt2 = 2,
value1 = 200, value2 = 50,
),
useBendingCorrection = True,
fitFastCircle = True,
fitFastCircleChi2Cut = True,
CAThetaCut = 0.0012,
CAPhiCut = 0.2,
))
trackingPhase2PU140.toModify(initialStepHitTriplets,
produceSeedingHitSets = False,
produceIntermediateHitTriplets = True,
)
from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer_cff import seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer as _seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer
_initialStepSeedsConsecutiveHitsTripletOnly = _seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer.clone(
seedingHitSets = "initialStepHitTriplets",
SeedComparitorPSet = dict(# FIXME: is this defined in any cfi that could be imported instead of copy-paste?
ComponentName = 'PixelClusterShapeSeedComparitor',
mixedTripletStepClassifier1.mva.GBRForestLabel = 'MVASelectorIter4_13TeV'
mixedTripletStepClassifier1.qualityCuts = [-0.5,0.0,0.5]
fastSim.toModify(mixedTripletStepClassifier1, vertices = "firstStepPrimaryVerticesBeforeMixing")
mixedTripletStepClassifier2 = TrackMVAClassifierPrompt.clone()
mixedTripletStepClassifier2.src = 'mixedTripletStepTracks'
mixedTripletStepClassifier2.mva.GBRForestLabel = 'MVASelectorIter0_13TeV'
mixedTripletStepClassifier2.qualityCuts = [-0.2,-0.2,-0.2]
fastSim.toModify(mixedTripletStepClassifier2,vertices = "firstStepPrimaryVerticesBeforeMixing")
from RecoTracker.FinalTrackSelectors.ClassifierMerger_cfi import *
mixedTripletStep = ClassifierMerger.clone()
mixedTripletStep.inputClassifiers=['mixedTripletStepClassifier1','mixedTripletStepClassifier2']
trackingPhase1.toReplaceWith(mixedTripletStep, mixedTripletStepClassifier1.clone(
mva = dict(GBRForestLabel = 'MVASelectorMixedTripletStep_Phase1'),
qualityCuts = [-0.5,0.0,0.5],
))
highBetaStar_2018.toModify(mixedTripletStep,qualityCuts = [-0.7,0.0,0.5])
# For LowPU
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
mixedTripletStepSelector = RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
src = 'mixedTripletStepTracks',
useAnyMVA = cms.bool(False),
GBRForestLabel = cms.string('MVASelectorIter4'),
trackSelectors = [
RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
name = 'mixedTripletStepVtxLoose',
chi2n_par = 1.2,
res_par = ( 0.003, 0.001 ),
minNumberLayers = 3,
initialStepHitTripletsPreSplitting, initialStepSeedsPreSplitting,
initialStepTrackCandidatesPreSplitting, initialStepTracksPreSplitting,
firstStepPrimaryVerticesPreSplitting,
initialStepTrackRefsForJetsPreSplitting, caloTowerForTrkPreSplitting,
ak4CaloJetsForTrkPreSplitting, jetsForCoreTrackingPreSplitting,
siPixelClusters, siPixelRecHits, MeasurementTrackerEvent,
siPixelClusterShapeCache)
InitialStepPreSplitting = cms.Sequence(InitialStepPreSplittingTask)
_InitialStepPreSplittingTask_trackingPhase1 = InitialStepPreSplittingTask.copy(
)
_InitialStepPreSplittingTask_trackingPhase1.replace(
initialStepHitTripletsPreSplitting,
cms.Task(initialStepHitTripletsPreSplitting,
initialStepHitQuadrupletsPreSplitting))
trackingPhase1.toReplaceWith(
InitialStepPreSplittingTask,
_InitialStepPreSplittingTask_trackingPhase1.copyAndExclude(
[initialStepHitTripletsPreSplitting]))
# Although InitialStepPreSplitting is not really part of LowPU/Run1/Phase2PU140
# tracking, we use it to get siPixelClusters and siPixelRecHits
# collections for non-splitted pixel clusters. All modules before
# iterTracking sequence use siPixelClustersPreSplitting and
# siPixelRecHitsPreSplitting for that purpose.
#
# If siPixelClusters would be defined in
# RecoLocalTracker.Configuration.RecoLocalTracker_cff, we would have a
# situation where
# - LowPU/Phase2PU140 has siPixelClusters defined in RecoLocalTracker_cff
# - everything else has siPixelClusters defined here
# and this leads to a mess. The way it is done here we have only
# one place (within Reconstruction_cff) where siPixelClusters
mixedTripletStepClassifier2 = TrackMVAClassifierPrompt.clone()
mixedTripletStepClassifier2.src = 'mixedTripletStepTracks'
mixedTripletStepClassifier2.mva.GBRForestLabel = 'MVASelectorIter0_13TeV'
mixedTripletStepClassifier2.qualityCuts = [-0.2, -0.2, -0.2]
fastSim.toModify(mixedTripletStepClassifier2,
vertices="firstStepPrimaryVerticesBeforeMixing")
from RecoTracker.FinalTrackSelectors.ClassifierMerger_cfi import *
mixedTripletStep = ClassifierMerger.clone()
mixedTripletStep.inputClassifiers = [
'mixedTripletStepClassifier1', 'mixedTripletStepClassifier2'
]
trackingPhase1.toReplaceWith(
mixedTripletStep,
mixedTripletStepClassifier1.clone(
mva=dict(GBRForestLabel='MVASelectorMixedTripletStep_Phase1'),
qualityCuts=[-0.5, 0.0, 0.5],
))
trackingPhase1QuadProp.toReplaceWith(
mixedTripletStep,
mixedTripletStepClassifier1.clone(
mva=dict(GBRForestLabel='MVASelectorMixedTripletStep_Phase1'),
qualityCuts=[-0.5, 0.0, 0.5],
))
# For LowPU
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
mixedTripletStepSelector = RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
src='mixedTripletStepTracks',
useAnyMVA=cms.bool(False),
GBRForestLabel=cms.string('MVASelectorIter4'),
# For Phase1
# MVA selection to be enabled after re-training, for time being we go with cut-based selector
from RecoTracker.FinalTrackSelectors.TrackCutClassifier_cfi import TrackCutClassifier as _TrackCutClassifier
trackingPhase1.toReplaceWith(
lowPtTripletStep,
_TrackCutClassifier.clone(src="lowPtTripletStepTracks",
vertices="firstStepPrimaryVertices",
mva=dict(
minPixelHits=[1, 1, 1],
maxChi2=[9999., 9999., 9999.],
maxChi2n=[2.0, 0.9, 0.5],
minLayers=[3, 3, 3],
min3DLayers=[3, 3, 3],
maxLostLayers=[2, 2, 2],
dz_par=dict(
dz_par1=[0.7, 0.6, 0.45],
dz_par2=[0.5, 0.4, 0.4],
dz_exp=[4, 4, 4],
),
dr_par=dict(
dr_par1=[0.8, 0.7, 0.6],
dr_par2=[0.5, 0.4, 0.3],
dr_exp=[4, 4, 4],
d0err_par=[0.002, 0.002, 0.001],
),
)))
# For LowPU and Phase1PU70
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
lowPtTripletStepSelector = RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
fitFastCircleChi2Cut = True,
SeedComparitorPSet = initialStepHitTripletsPreSplitting.SeedComparitorPSet
)
from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsEDProducer_cff import seedCreatorFromRegionConsecutiveHitsEDProducer as _seedCreatorFromRegionConsecutiveHitsEDProducer
initialStepSeedsPreSplitting = _seedCreatorFromRegionConsecutiveHitsEDProducer.clone(
seedingHitSets = "initialStepHitTripletsPreSplitting",
)
from RecoPixelVertexing.PixelTriplets.caHitQuadrupletEDProducer_cfi import caHitQuadrupletEDProducer as _caHitQuadrupletEDProducer
trackingPhase1.toReplaceWith(initialStepHitQuadrupletsPreSplitting, _caHitQuadrupletEDProducer.clone(
doublets = "initialStepHitDoubletsPreSplitting",
extraHitRPhitolerance = initialStepHitTripletsPreSplitting.extraHitRPhitolerance,
SeedComparitorPSet = initialStepHitTripletsPreSplitting.SeedComparitorPSet,
maxChi2 = dict(
pt1 = 0.7, pt2 = 2,
value1 = 200, value2 = 50,
),
useBendingCorrection = True,
fitFastCircle = True,
fitFastCircleChi2Cut = True,
CAThetaCut = 0.0012,
CAPhiCut = 0.2,
))
trackingPhase1QuadProp.toModify(initialStepHitTripletsPreSplitting,
produceSeedingHitSets = False,
produceIntermediateHitTriplets = True,
)
trackingPhase1.toModify(initialStepSeedsPreSplitting, seedingHitSets = "initialStepHitQuadrupletsPreSplitting")
trackingPhase1QuadProp.toModify(initialStepSeedsPreSplitting, seedingHitSets = "initialStepHitQuadrupletsPreSplitting")
from RecoTracker.FinalTrackSelectors.ClassifierMerger_cfi import *
pixelLessStep = ClassifierMerger.clone()
pixelLessStep.inputClassifiers = [
'pixelLessStepClassifier1', 'pixelLessStepClassifier2'
]
from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
#LWTNN selector
from RecoTracker.FinalTrackSelectors.TrackLwtnnClassifier_cfi import *
from RecoTracker.FinalTrackSelectors.trackSelectionLwtnn_cfi import *
trackingPhase1.toReplaceWith(
pixelLessStep,
TrackLwtnnClassifier.clone(
src='pixelLessStepTracks',
qualityCuts=[-0.6, -0.05, 0.5],
))
(trackingPhase1 & fastSim).toModify(
pixelLessStep, vertices="firstStepPrimaryVerticesBeforeMixing")
pp_on_AA_2018.toReplaceWith(
pixelLessStep,
pixelLessStepClassifier1.clone(
qualityCuts=[-0.4, 0.0, 0.8],
mva=dict(GBRForestLabel='MVASelectorPixelLessStep_Phase1')))
# For LowPU
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
pixelLessStepSelector = RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
FilterAtHelixStage = cms.bool(False),
FilterPixelHits = cms.bool(True),
FilterStripHits = cms.bool(False),
ClusterShapeHitFilterName = cms.string('ClusterShapeHitFilter'),
ClusterShapeCacheSrc = cms.InputTag('siPixelClusterShapeCache')
),
)
from RecoPixelVertexing.PixelTriplets.caHitTripletEDProducer_cfi import caHitTripletEDProducer as _caHitTripletEDProducer
trackingPhase1.toModify(detachedTripletStepHitDoublets, layerPairs = [0,1]) # layer pairs (0,1), (1,2)
trackingPhase1.toReplaceWith(detachedTripletStepHitTriplets, _caHitTripletEDProducer.clone(
doublets = "detachedTripletStepHitDoublets",
extraHitRPhitolerance = detachedTripletStepHitTriplets.extraHitRPhitolerance,
maxChi2 = dict(
pt1 = 0.8, pt2 = 2,
value1 = 300 , value2 = 10,
),
useBendingCorrection = True,
CAThetaCut = 0.001,
CAPhiCut = 0,
CAHardPtCut = 0.2,
))
# QUALITY CUTS DURING TRACK BUILDING
import TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff
_detachedTripletStepTrajectoryFilterBase = TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff.CkfBaseTrajectoryFilter_block.clone(
# maxLostHitsFraction = cms.double(1./10.),
# constantValueForLostHitsFractionFilter = cms.double(0.701),
minimumNumberOfHits = 3,
minPt = 0.075,
)
detachedTripletStepTrajectoryFilterBase = _detachedTripletStepTrajectoryFilterBase.clone(
SeedComparitorPSet = RecoPixelVertexing.PixelLowPtUtilities.LowPtClusterShapeSeedComparitor_cfi.LowPtClusterShapeSeedComparitor
)
from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsEDProducer_cff import seedCreatorFromRegionConsecutiveHitsEDProducer as _seedCreatorFromRegionConsecutiveHitsEDProducer
highPtTripletStepSeeds = _seedCreatorFromRegionConsecutiveHitsEDProducer.clone(
seedingHitSets = "highPtTripletStepHitTriplets",
)
from RecoPixelVertexing.PixelTriplets.caHitTripletEDProducer_cfi import caHitTripletEDProducer as _caHitTripletEDProducer
trackingPhase1.toModify(highPtTripletStepHitDoublets, layerPairs = [0,1]) # layer pairs (0,1), (1,2)
trackingPhase1.toReplaceWith(highPtTripletStepHitTriplets, _caHitTripletEDProducer.clone(
doublets = "highPtTripletStepHitDoublets",
extraHitRPhitolerance = highPtTripletStepHitTriplets.extraHitRPhitolerance,
SeedComparitorPSet = highPtTripletStepHitTriplets.SeedComparitorPSet,
maxChi2 = dict(
pt1 = 0.8, pt2 = 8,
value1 = 100, value2 = 6,
),
useBendingCorrection = True,
CAThetaCut = 0.004,
CAPhiCut = 0.07,
CAHardPtCut = 0.3,
))
# QUALITY CUTS DURING TRACK BUILDING
import TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff as _TrajectoryFilter_cff
_highPtTripletStepTrajectoryFilterBase = _TrajectoryFilter_cff.CkfBaseTrajectoryFilter_block.clone(
minimumNumberOfHits = 3,
minPt = 0.2,
)
highPtTripletStepTrajectoryFilterBase = _highPtTripletStepTrajectoryFilterBase.clone(
maxCCCLostHits = 0,
pixelLessStepClassifier1.qualityCuts = [-0.4,0.0,0.4]
fastSim.toModify(pixelLessStepClassifier1, vertices = "firstStepPrimaryVerticesBeforeMixing" )
pixelLessStepClassifier2 = TrackMVAClassifierPrompt.clone()
pixelLessStepClassifier2.src = 'pixelLessStepTracks'
pixelLessStepClassifier2.mva.GBRForestLabel = 'MVASelectorIter0_13TeV'
pixelLessStepClassifier2.qualityCuts = [-0.0,0.0,0.0]
fastSim.toModify(pixelLessStepClassifier2, vertices = "firstStepPrimaryVerticesBeforeMixing" )
from RecoTracker.FinalTrackSelectors.ClassifierMerger_cfi import *
pixelLessStep = ClassifierMerger.clone()
pixelLessStep.inputClassifiers=['pixelLessStepClassifier1','pixelLessStepClassifier2']
from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
trackingPhase1.toReplaceWith(pixelLessStep, pixelLessStepClassifier1.clone(
mva = dict(GBRForestLabel = 'MVASelectorPixelLessStep_Phase1'),
qualityCuts = [-0.4,0.0,0.4],
))
# For LowPU
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
pixelLessStepSelector = RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
src='pixelLessStepTracks',
useAnyMVA = cms.bool(False),
GBRForestLabel = cms.string('MVASelectorIter5'),
trackSelectors= cms.VPSet(
RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
name = 'pixelLessStepLoose',
chi2n_par = 0.5,
res_par = ( 0.003, 0.001 ),
minNumberLayers = 4,
maxNumberLostLayers = 1,
_seedProducers, globals())
(_seedSelectors_trackingPhase1, _trackingNtupleSeedSelectors_trackingPhase1
) = _TrackValidation_cff._addSeedToTrackProducers(
_seedProducers_trackingPhase1, globals())
(_seedSelectors_trackingPhase1QuadProp,
_trackingNtupleSeedSelectors_trackingPhase1QuadProp
) = _TrackValidation_cff._addSeedToTrackProducers(
_seedProducers_trackingPhase1QuadProp, globals())
(_seedSelectors_trackingPhase2PU140,
_trackingNtupleSeedSelectors_trackingPhase2PU140
) = _TrackValidation_cff._addSeedToTrackProducers(
_seedProducers_trackingPhase2PU140, globals())
from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
from Configuration.Eras.Modifier_trackingPhase1QuadProp_cff import trackingPhase1QuadProp
from Configuration.Eras.Modifier_trackingPhase2PU140_cff import trackingPhase2PU140
trackingPhase1.toReplaceWith(trackingNtupleSeedSelectors,
_trackingNtupleSeedSelectors_trackingPhase1)
trackingPhase1QuadProp.toReplaceWith(
trackingNtupleSeedSelectors,
_trackingNtupleSeedSelectors_trackingPhase1QuadProp)
trackingPhase2PU140.toReplaceWith(
trackingNtupleSeedSelectors,
_trackingNtupleSeedSelectors_trackingPhase2PU140)
trackingNtuple.seedTracks = _seedSelectors
trackingPhase1.toModify(trackingNtuple,
seedTracks=_seedSelectors_trackingPhase1)
trackingPhase1QuadProp.toModify(trackingNtuple,
seedTracks=_seedSelectors_trackingPhase1)
trackingPhase2PU140.toModify(trackingNtuple,
seedTracks=_seedSelectors_trackingPhase2PU140)
min3DLayers=[1, 2, 3],
maxLostLayers=[4, 3, 2],
maxDz=[0.5, 0.35, 0.2],
maxDr=[0.3, 0.2, 0.1]),
vertices='firstStepGoodPrimaryVertices')
jetCoreRegionalStepBarrel = jetCoreRegionalStep.clone(
src='jetCoreRegionalStepBarrelTracks',
mva=dict(
# minPixelHits = [1,1,1], # they could be easily increased to at least 2 or 3 !
min3DLayers=[1, 2, 2], ),
)
from RecoTracker.FinalTrackSelectors.TrackMVAClassifierPrompt_cfi import *
trackingPhase1.toReplaceWith(
jetCoreRegionalStep,
TrackMVAClassifierPrompt.clone(
mva=dict(GBRForestLabel='MVASelectorJetCoreRegionalStep_Phase1'),
src='jetCoreRegionalStepTracks',
qualityCuts=[-0.2, 0.0, 0.4]))
trackingPhase1.toReplaceWith(
jetCoreRegionalStepBarrel,
jetCoreRegionalStep.clone(src='jetCoreRegionalStepBarrelTracks', ))
from RecoTracker.FinalTrackSelectors.trackTfClassifier_cfi import *
from RecoTracker.FinalTrackSelectors.trackSelectionTf_cfi import *
from RecoTracker.FinalTrackSelectors.trackSelectionTf_CKF_cfi import *
trackdnn.toReplaceWith(
jetCoreRegionalStep,
trackTfClassifier.clone(
src='jetCoreRegionalStepTracks',
qualityCuts=qualityCutDictionary.JetCoreRegionalStep.value()))
jetCoreRegionalStep = TrackCutClassifier.clone()
jetCoreRegionalStep.src='jetCoreRegionalStepTracks'
jetCoreRegionalStep.mva.minPixelHits = [1,1,1]
jetCoreRegionalStep.mva.maxChi2 = [9999.,9999.,9999.]
jetCoreRegionalStep.mva.maxChi2n = [1.6,1.0,0.7]
jetCoreRegionalStep.mva.minLayers = [3,5,5]
jetCoreRegionalStep.mva.min3DLayers = [1,2,3]
jetCoreRegionalStep.mva.maxLostLayers = [4,3,2]
jetCoreRegionalStep.mva.maxDz = [0.5,0.35,0.2];
jetCoreRegionalStep.mva.maxDr = [0.3,0.2,0.1];
jetCoreRegionalStep.vertices = 'firstStepGoodPrimaryVertices'
from RecoTracker.FinalTrackSelectors.TrackMVAClassifierPrompt_cfi import *
trackingPhase1.toReplaceWith(jetCoreRegionalStep, TrackMVAClassifierPrompt.clone(
src = 'jetCoreRegionalStepTracks',
mva = dict(GBRForestLabel = 'MVASelectorJetCoreRegionalStep_Phase1'),
qualityCuts = [-0.2,0.0,0.4],
))
fastSim.toModify(jetCoreRegionalStep,vertices = "firstStepPrimaryVerticesBeforeMixing")
# Final sequence
JetCoreRegionalStepTask = cms.Task(jetsForCoreTracking,
firstStepGoodPrimaryVertices,
#jetCoreRegionalStepClusters,
jetCoreRegionalStepSeedLayers,
jetCoreRegionalStepTrackingRegions,
jetCoreRegionalStepHitDoublets,
jetCoreRegionalStepSeeds,
jetCoreRegionalStepTrackCandidates,
jetCoreRegionalStepTracks,
# jetCoreRegionalStepClassifier1,jetCoreRegionalStepClassifier2,
lowPtTripletStepSeeds = _seedCreatorFromRegionConsecutiveHitsEDProducer.clone(
seedingHitSets="lowPtTripletStepHitTriplets", )
from RecoPixelVertexing.PixelTriplets.caHitTripletEDProducer_cfi import caHitTripletEDProducer as _caHitTripletEDProducer
trackingPhase1.toModify(lowPtTripletStepHitDoublets,
layerPairs=[0, 1]) # layer pairs (0,1), (1,2)
trackingPhase1.toReplaceWith(
lowPtTripletStepHitTriplets,
_caHitTripletEDProducer.clone(
doublets="lowPtTripletStepHitDoublets",
extraHitRPhitolerance=lowPtTripletStepHitTriplets.
extraHitRPhitolerance,
SeedComparitorPSet=lowPtTripletStepHitTriplets.SeedComparitorPSet,
maxChi2=dict(
pt1=0.8,
pt2=2,
value1=70,
value2=8,
),
useBendingCorrection=True,
CAThetaCut=0.002,
CAPhiCut=0.05,
))
trackingPhase2PU140.toModify(lowPtTripletStepHitDoublets,
layerPairs=[0, 1]) # layer pairs (0,1), (1,2)
trackingPhase2PU140.toReplaceWith(
lowPtTripletStepHitTriplets,
_caHitTripletEDProducer.clone(
doublets="lowPtTripletStepHitDoublets",
enabled = True,
),
useBendingCorrection = True,
fitFastCircle = True,
fitFastCircleChi2Cut = True,
SeedComparitorPSet = lowPtQuadStepHitTriplets.SeedComparitorPSet,
)
from RecoPixelVertexing.PixelTriplets.caHitQuadrupletEDProducer_cfi import caHitQuadrupletEDProducer as _caHitQuadrupletEDProducer
trackingPhase1.toModify(lowPtQuadStepHitDoublets, layerPairs = [0,1,2]) # layer pairs (0,1), (1,2), (2,3)
trackingPhase1.toReplaceWith(lowPtQuadStepHitQuadruplets, _caHitQuadrupletEDProducer.clone(
doublets = "lowPtQuadStepHitDoublets",
extraHitRPhitolerance = lowPtQuadStepHitTriplets.extraHitRPhitolerance,
SeedComparitorPSet = lowPtQuadStepHitTriplets.SeedComparitorPSet,
maxChi2 = dict(
pt1 = 0.7, pt2 = 2,
value1 = 1000, value2 = 150,
),
useBendingCorrection = True,
fitFastCircle = True,
fitFastCircleChi2Cut = True,
CAThetaCut = 0.0017,
CAPhiCut = 0.3,
))
from RecoPixelVertexing.PixelTriplets.pixelQuadrupletMergerEDProducer_cfi import pixelQuadrupletMergerEDProducer as _pixelQuadrupletMergerEDProducer
from RecoPixelVertexing.PixelTriplets.quadrupletseedmerging_cff import *
_lowPtQuadStepHitQuadrupletsMerging = _pixelQuadrupletMergerEDProducer.clone(
triplets = "lowPtQuadStepHitTriplets",
layerList = dict(refToPSet_ = cms.string("PixelSeedMergerQuadruplets")),
)
from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsEDProducer_cff import seedCreatorFromRegionConsecutiveHitsEDProducer as _seedCreatorFromRegionConsecutiveHitsEDProducer
lowPtQuadStepSeeds = _seedCreatorFromRegionConsecutiveHitsEDProducer.clone(
