FirstStep_cff.py

Go to the documentation of this file.

####################################################################
# Does two tracking iterations and then merges them.               #
# Step 0 uses pixel-triplet seeding.                               #
# Step 1 uses mixed-pair seeding.                                  #
####################################################################

### STEP 0 ###

# hit building
from RecoLocalTracker.SiPixelRecHits.PixelCPEESProducers_cff import *
from RecoTracker.TransientTrackingRecHit.TTRHBuilders_cff import *

# These 3 are not used anywhere. Can they be removed ?
from RecoTracker.TkSeedGenerator.GlobalCombinedSeeds_cff import *
from RecoTracker.TkSeedGenerator.GlobalMixedSeeds_cff import *
from RecoTracker.TkSeedGenerator.GlobalPixelSeeds_cff import *

# seeding
from RecoTracker.TkSeedGenerator.GlobalSeedsFromTriplets_cff import *
from RecoTracker.TkTrackingRegions.GlobalTrackingRegionFromBeamSpot_cfi import RegionPsetFomBeamSpotBlock
newSeedFromTriplets = RecoTracker.TkSeedGenerator.GlobalSeedsFromTriplets_cff.globalSeedsFromTriplets.clone(
    RegionFactoryPSet = RegionPsetFomBeamSpotBlock.clone(
    ComponentName = cms.string('GlobalRegionProducerFromBeamSpot'),
    RegionPSet = RegionPsetFomBeamSpotBlock.RegionPSet.clone(
    ptMin = 0.8
    )
    )
    )
from RecoPixelVertexing.PixelLowPtUtilities.ClusterShapeHitFilterESProducer_cfi import *
newSeedFromTriplets.SeedComparitorPSet.ComponentName = 'LowPtClusterShapeSeedComparitor'

# building
import TrackingTools.TrajectoryFiltering.TrajectoryFilterESProducer_cfi
newTrajectoryFilter = TrackingTools.TrajectoryFiltering.TrajectoryFilterESProducer_cfi.trajectoryFilterESProducer.clone(
    ComponentName = 'newTrajectoryFilter',
    filterPset = TrackingTools.TrajectoryFiltering.TrajectoryFilterESProducer_cfi.trajectoryFilterESProducer.filterPset.clone(
    minimumNumberOfHits = 3,
    minPt = 0.6
    )
    )

import RecoTracker.CkfPattern.GroupedCkfTrajectoryBuilderESProducer_cfi
newTrajectoryBuilder = RecoTracker.CkfPattern.GroupedCkfTrajectoryBuilderESProducer_cfi.GroupedCkfTrajectoryBuilder.clone(
    ComponentName = 'newTrajectoryBuilder',
    trajectoryFilterName = 'newTrajectoryFilter'
    )

from RecoTracker.CkfPattern.CkfTrackCandidates_cff import *
newTrackCandidateMaker = RecoTracker.CkfPattern.CkfTrackCandidates_cfi.ckfTrackCandidates.clone(
    src = cms.InputTag('newSeedFromTriplets'),
    TrajectoryBuilder = 'newTrajectoryBuilder'
    )

# fitting
from RecoTracker.TrackProducer.CTFFinalFitWithMaterial_cff import *
preFilterZeroStepTracks = RecoTracker.TrackProducer.TrackProducer_cfi.TrackProducer.clone(
    src = 'newTrackCandidateMaker',
    AlgorithmName = 'iter0'
    )

### STEP 1 ###

# Lock hits from step0 tracks
zeroStepFilter = cms.EDProducer("QualityFilter",
    TrackQuality = cms.string('highPurity'),
    recTracks = cms.InputTag("zeroStepTracksWithQuality:")
)

# new hit collection
newClusters = cms.EDProducer("TrackClusterRemover",
    trajectories = cms.InputTag("zeroStepFilter"), 
    pixelClusters = cms.InputTag("siPixelClusters"),
    stripClusters = cms.InputTag("siStripClusters"),
    Common = cms.PSet(
        maxChi2 = cms.double(30.0)
    )
)

# make corresponding rechits
import RecoLocalTracker.SiPixelRecHits.SiPixelRecHits_cfi
import RecoLocalTracker.SiStripRecHitConverter.SiStripRecHitConverter_cfi
newPixelRecHits = RecoLocalTracker.SiPixelRecHits.SiPixelRecHits_cfi.siPixelRecHits.clone(
    src = cms.InputTag("newClusters")
    )
newStripRecHits = RecoLocalTracker.SiStripRecHitConverter.SiStripRecHitConverter_cfi.siStripMatchedRecHits.clone(
    ClusterProducer = 'newClusters'
    )


# seeding 
newMixedLayerPairs = RecoTracker.TkSeedingLayers.MixedLayerPairs_cfi.mixedlayerpairs.clone(
    ComponentName = 'newMixedLayerPairs'
    )
newMixedLayerPairs.BPix.HitProducer = 'newPixelRecHits'
newMixedLayerPairs.FPix.HitProducer = 'newPixelRecHits'
newMixedLayerPairs.TEC.matchedRecHits = cms.InputTag("newStripRecHits","matchedRecHit")

from RecoTracker.TkSeedGenerator.GlobalSeedsFromPairsWithVertices_cff import *
newSeedFromPairs = RecoTracker.TkSeedGenerator.GlobalSeedsFromPairsWithVertices_cff.globalSeedsFromPairsWithVertices.clone()
newSeedFromPairs.RegionFactoryPSet.RegionPSet.ptMin = 0.6
newSeedFromPairs.RegionFactoryPSet.RegionPSet.originRadius = 0.05
newSeedFromPairs.OrderedHitsFactoryPSet.SeedingLayers = cms.string('newMixedLayerPairs')
newSeedFromPairs.ClusterCheckPSet.PixelClusterCollectionLabel = 'newClusters'
newSeedFromPairs.ClusterCheckPSet.ClusterCollectionLabel = 'newClusters'
   


# building 
import RecoTracker.MeasurementDet.MeasurementTrackerESProducer_cfi
newMeasurementTracker = RecoTracker.MeasurementDet.MeasurementTrackerESProducer_cfi.MeasurementTracker.clone(
    ComponentName = 'newMeasurementTracker',
    pixelClusterProducer = 'newClusters',
    stripClusterProducer = 'newClusters'
    )

import TrackingTools.TrajectoryFiltering.TrajectoryFilterESProducer_cfi
stepOneTrajectoryFilter = TrackingTools.TrajectoryFiltering.TrajectoryFilterESProducer_cfi.trajectoryFilterESProducer.clone(
    ComponentName = 'stepOneTrajectoryFilter',
    filterPset = TrackingTools.TrajectoryFiltering.TrajectoryFilterESProducer_cfi.trajectoryFilterESProducer.filterPset.clone(
    minimumNumberOfHits = 3,
    minPt = 0.4
    )
    )

stepOneCkfTrajectoryBuilder = RecoTracker.CkfPattern.GroupedCkfTrajectoryBuilderESProducer_cfi.GroupedCkfTrajectoryBuilder.clone(
    ComponentName = 'stepOneCkfTrajectoryBuilder',
    MeasurementTrackerName = 'newMeasurementTracker',
    trajectoryFilterName = 'stepOneTrajectoryFilter'
    )

stepOneTrackCandidateMaker = RecoTracker.CkfPattern.CkfTrackCandidates_cfi.ckfTrackCandidates.clone(
    src = cms.InputTag('newSeedFromPairs'),
    TrajectoryBuilder = 'stepOneCkfTrajectoryBuilder'
    )


# fitting
preFilterStepOneTracks = RecoTracker.TrackProducer.TrackProducer_cfi.TrackProducer.clone(
    AlgorithmName = 'iter1',
    src = 'stepOneTrackCandidateMaker',
    clusterRemovalInfo = 'newClusters'
    )

### BOTH STEPS TOGETHER ###

# Set track quality flags for both steps
from RecoTracker.FinalTrackSelectors.TracksWithQuality_cff import *

# Merge step 0 and step 1
from RecoTracker.FinalTrackSelectors.MergeTrackCollections_cff import *

# Final sequence
firstStep = cms.Sequence(newSeedFromTriplets*newTrackCandidateMaker*preFilterZeroStepTracks*tracksWithQualityZeroStep*
                         zeroStepFilter*newClusters*newPixelRecHits*newStripRecHits*
                         newSeedFromPairs*stepOneTrackCandidateMaker*preFilterStepOneTracks*tracksWithQualityStepOne*
                         firstStepTracksWithQuality)