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)
)


import RecoTracker.IterativeTracking.MixedTripletStep_cff
from RecoTracker.IterativeTracking.MixedTripletStep_cff import mixedTripletStepTrajectoryBuilder,mixedTripletStepClusterShapeHitFilter,mixedTripletStepTrajectoryCleanerBySharedHits,mixedTripletStepChi2Est

# 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._mixedTripletStepTrackCandidatesCkf.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,
    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)