FastSimulation/Tracking/python/iterativeTk_cff.py - master - githubmirror/cmssw

FastSimulation/Tracking/python/iterativeTk_cff.py
70 строк3 KB
Kirutheeka Namasivayam
added phase2 tracking steps and changed simhits to input tags
14 мар 2026, 02:28
14 мар 2026, 02:282bfd4e1

##############################
# FastSim equivalent of RecoTracker/IterativeTracking/python/iterativeTk_cff.py
##############################

import FWCore.ParameterSet.Config as cms
from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
from TrackingTools.MaterialEffects.MaterialPropagatorParabolicMf_cff import *
#importing iterations directly from RecoTracker
from RecoTracker.IterativeTracking.InitialStep_cff import *
from RecoTracker.IterativeTracking.DetachedQuadStep_cff import *
from RecoTracker.IterativeTracking.HighPtTripletStep_cff import *
from RecoTracker.IterativeTracking.LowPtQuadStep_cff import *
from RecoTracker.IterativeTracking.DetachedTripletStep_cff import *
from RecoTracker.IterativeTracking.LowPtTripletStep_cff import *
from RecoTracker.IterativeTracking.PixelPairStep_cff import *
from RecoTracker.IterativeTracking.MixedTripletStep_cff import *
from RecoTracker.IterativeTracking.PixelLessStep_cff import *
from RecoTracker.IterativeTracking.TobTecStep_cff import *
# the following loads a dummy empty track collection
# such that FastSim can import earlyGeneralTracks_cfi from full tracking
# todo: actual implementation of JetCore iteration  
from RecoTracker.IterativeTracking.JetCoreRegionalStep_cff import *

import RecoTracker.FinalTrackSelectors.earlyGeneralTracks_cfi
# todo, import MuonSeededStep_cff, preDuplicateMergingGeneralTracks_cfi, MergeTrackCollections_cff, ConversionStep_cff

generalTracksBeforeMixing = RecoTracker.FinalTrackSelectors.earlyGeneralTracks_cfi.earlyGeneralTracks.clone()

from RecoTracker.FinalTrackSelectors.trackTfClassifier_cfi import *
trackdnn_essource = cms.ESSource("EmptyESSource", recordName = cms.string("TfGraphRecord"), firstValid = cms.vuint32(1), iovIsRunNotTime = cms.bool(True) )

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)


_iterations_trackingPhase2PU140 = cms.Sequence(
    InitialStep
    +HighPtTripletStep
    +LowPtQuadStep
    +LowPtTripletStep
    +LowPtTripletStep
    +DetachedQuadStep
    +PixelPairStep
    +generalTracksBeforeMixing
)

trackingPhase1.toReplaceWith(iterTracking, _iterTracking_Phase1)
trackingPhase2PU140.toReplaceWith(iterTracking,_iterations_trackingPhase2PU140)