HiRegitMuonMixedTripletStep_cff.py

Go to the documentation of this file.

import FWCore.ParameterSet.Config as cms
 
 
 
from RecoHI.HiTracking.HITrackingRegionProducer_cfi import *
# Are the following values set to the same in every iteration? If yes,
# why not making the change in HITrackingRegionProducer_cfi directly
# once for all?
hiRegitMuMixedTripletStepTrackingRegionsA = HiTrackingRegionFactoryFromSTAMuonsEDProducer.clone(
    MuonSrc = "standAloneMuons:UpdatedAtVtx", # this is the same as default, why repeat?
    MuonTrackingRegionBuilder = dict(
        vertexCollection = "hiSelectedPixelVertex",
        UseVertex     = True,
        Phi_fixed     = True,
        Eta_fixed     = True,
        DeltaPhi      = 0.3,
        DeltaEta      = 0.2,
        # Ok, the following ones are specific to MixedTripletStep
        Pt_min        = 1.3,
        DeltaR        = 0.5, # default = 0.2
        DeltaZ        = 0.5, # this give you the length
        Rescale_Dz    = 4.,  # max(DeltaZ_Region,Rescale_Dz*vtx->zError())
    )
)
hiRegitMuMixedTripletStepTrackingRegionsB = hiRegitMuMixedTripletStepTrackingRegionsA.clone(
    MuonTrackingRegionBuilder = dict(Pt_min = 1.5)
)
 
 
from RecoTracker.IterativeTracking.MixedTripletStep_cff import *
 
# NEW CLUSTERS (remove previously used clusters)
hiRegitMuMixedTripletStepClusters = RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepClusters.clone(
    oldClusterRemovalInfo = "hiRegitMuPixelPairStepClusters",
    trajectories          = "hiRegitMuPixelPairStepTracks",
    overrideTrkQuals      = 'hiRegitMuPixelPairStepSelector:hiRegitMuPixelPairStep',
    trackClassifier       = '',
    TrackQuality          = 'tight'
)
 
 
# SEEDING LAYERS A
hiRegitMuMixedTripletStepSeedLayersA =  RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepSeedLayersA.clone(
    BPix = dict(skipClusters = 'hiRegitMuMixedTripletStepClusters'),
    FPix = dict(skipClusters = 'hiRegitMuMixedTripletStepClusters'),
    TEC  = dict(skipClusters = 'hiRegitMuMixedTripletStepClusters')
)
# SEEDS A
hiRegitMuMixedTripletStepHitDoubletsA = RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepHitDoubletsA.clone(
    seedingLayers   = "hiRegitMuMixedTripletStepSeedLayersA",
    trackingRegions = "hiRegitMuMixedTripletStepTrackingRegionsA",
    clusterCheck    = "hiRegitMuClusterCheck",
)
hiRegitMuMixedTripletStepHitTripletsA = RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepHitTripletsA.clone(
    doublets = "hiRegitMuMixedTripletStepHitDoubletsA"
)
hiRegitMuMixedTripletStepSeedsA = RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepSeedsA.clone(
    seedingHitSets = "hiRegitMuMixedTripletStepHitTripletsA"
)
 
# SEEDING LAYERS B
hiRegitMuMixedTripletStepSeedLayersB =  RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepSeedLayersB.clone(
    BPix = dict(skipClusters = 'hiRegitMuMixedTripletStepClusters'),
    TIB  = dict(skipClusters = 'hiRegitMuMixedTripletStepClusters')
)
 
hiRegitMuMixedTripletStepHitDoubletsB = RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepHitDoubletsB.clone(
    seedingLayers   = "hiRegitMuMixedTripletStepSeedLayersB",
    trackingRegions = "hiRegitMuMixedTripletStepTrackingRegionsB",
    clusterCheck    = "hiRegitMuClusterCheck",
)
hiRegitMuMixedTripletStepHitTripletsB = RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepHitTripletsB.clone(
    doublets = "hiRegitMuMixedTripletStepHitDoubletsB"
)
hiRegitMuMixedTripletStepSeedsB = RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepSeedsA.clone(
    seedingHitSets = "hiRegitMuMixedTripletStepHitTripletsB"
)
 
# combine seeds
hiRegitMuMixedTripletStepSeeds = RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepSeeds.clone(
    seedCollections = [
        'hiRegitMuMixedTripletStepSeedsA',
        'hiRegitMuMixedTripletStepSeedsB',
        ]
)
 
# track building
hiRegitMuMixedTripletStepTrajectoryFilter = RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepTrajectoryFilter.clone(
    minPt               = 1.,
    minimumNumberOfHits = 6,
    minHitsMinPt        = 4
)
 
 # after each new hit, apply pT cut for traj w/ at least minHitsMinPt = cms.int32(3),
 
hiRegitMuMixedTripletStepTrajectoryBuilder = RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepTrajectoryBuilder.clone(
    trajectoryFilter = cms.PSet(
       refToPSet_ = cms.string('hiRegitMuMixedTripletStepTrajectoryFilter')
       ),
    minNrOfHitsForRebuild = 6 #change from default 4
)
 
hiRegitMuMixedTripletStepTrackCandidates = RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepTrackCandidates.clone(
    src               = 'hiRegitMuMixedTripletStepSeeds',
    TrajectoryBuilderPSet = cms.PSet(
       refToPSet_ = cms.string('hiRegitMuMixedTripletStepTrajectoryBuilder')
       ),
    clustersToSkip    = 'hiRegitMuMixedTripletStepClusters', 
    maxNSeeds         = 1000000
)
 
# fitting: feed new-names
hiRegitMuMixedTripletStepTracks  = RecoTracker.IterativeTracking.MixedTripletStep_cff.mixedTripletStepTracks.clone(
    AlgorithmName = 'hiRegitMuMixedTripletStep',
    src           = 'hiRegitMuMixedTripletStepTrackCandidates',
)
 
# TRACK SELECTION AND QUALITY FLAG SETTING.
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
import RecoHI.HiTracking.hiMultiTrackSelector_cfi
hiRegitMuMixedTripletStepSelector = RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiMultiTrackSelector.clone(
    src            = 'hiRegitMuMixedTripletStepTracks',
    vertices       = "hiSelectedPixelVertex",
    useAnyMVA      = True,
    GBRForestLabel = 'HIMVASelectorIter7',
    GBRForestVars  = ['chi2perdofperlayer', 'nhits', 'nlayers', 'eta'],
    trackSelectors = cms.VPSet(
        RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
           name      = 'hiRegitMuMixedTripletStepLoose',
           min_nhits = 8
            ),
        RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
            name      = 'hiRegitMuMixedTripletStepTight',
            preFilterName = 'hiRegitMuMixedTripletStepLoose',
            min_nhits = 8,
            useMVA    = True,
            minMVA    = -0.2
            ),
        RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
            name      = 'hiRegitMuMixedTripletStep',
            preFilterName = 'hiRegitMuMixedTripletStepTight',
            min_nhits = 8,
            useMVA    = True,
            minMVA    = -0.09
            )
        ) #end of vpset
) #end of clone
from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
trackingPhase1.toModify(hiRegitMuMixedTripletStepSelector, useAnyMVA = False)
trackingPhase1.toModify(hiRegitMuMixedTripletStepSelector, trackSelectors= cms.VPSet(
        RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
           name      = 'hiRegitMuMixedTripletStepLoose',
           min_nhits = 8
            ),
        RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
            name      = 'hiRegitMuMixedTripletStepTight',
            preFilterName = 'hiRegitMuMixedTripletStepLoose',
            min_nhits = 8,
            useMVA    = False,
            minMVA    = -0.2
            ),
        RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
            name      = 'hiRegitMuMixedTripletStep',
            preFilterName = 'hiRegitMuMixedTripletStepTight',
            min_nhits = 8,
            useMVA    = False,
            minMVA    = -0.09
            )
        ) #end of vpset
)
 
hiRegitMuonMixedTripletStepTask = cms.Task(hiRegitMuMixedTripletStepClusters,
                                         hiRegitMuMixedTripletStepSeedLayersA,
                                         hiRegitMuMixedTripletStepTrackingRegionsA,
                                         hiRegitMuMixedTripletStepHitDoubletsA,
                                         hiRegitMuMixedTripletStepHitTripletsA,
                                         hiRegitMuMixedTripletStepSeedsA,
                                         hiRegitMuMixedTripletStepSeedLayersB,
                                         hiRegitMuMixedTripletStepTrackingRegionsB,
                                         hiRegitMuMixedTripletStepHitDoubletsB,
                                         hiRegitMuMixedTripletStepHitTripletsB,
                                         hiRegitMuMixedTripletStepSeedsB,
                                         hiRegitMuMixedTripletStepSeeds,
                                         hiRegitMuMixedTripletStepTrackCandidates,
                                         hiRegitMuMixedTripletStepTracks,
                                         hiRegitMuMixedTripletStepSelector)
hiRegitMuonMixedTripletStep = cms.Sequence(hiRegitMuonMixedTripletStepTask)