HiTracking/python/hiRegitInitialStep_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 # pp iterative tracking modified for hiOffline reco (the vertex is the one reconstructed in HI)
0004 ################################### 0st step:pixel-triplet seeding, high-pT;
0005 from RecoHI.HiTracking.HITrackingRegionProducer_cfi import *
0006
0007 ###################################
0008 import RecoTracker.IterativeTracking.InitialStep_cff
0009 from RecoTracker.IterativeTracking.InitialStep_cff import initialStepTrajectoryFilterShape
0010
0011 #################################
0012 # Filter on quality tracks
0013 hiGeneralTrackFilter = cms.EDProducer("QualityFilter",
0014                                       TrackQuality = cms.string('highPurity'),
0015                                       recTracks = cms.InputTag("hiGeneralTracks")
0016                                       )
0017
0018 # NEW CLUSTERS (remove previously used clusters)
0019 hiRegitInitialStepClusters = cms.EDProducer("HITrackClusterRemover",
0020                                             clusterLessSolution= cms.bool(True),
0021                                             oldClusterRemovalInfo = cms.InputTag("hiPixelPairClusters"),
0022                                             trajectories = cms.InputTag("hiGeneralTrackFilter"),
0023                                             TrackQuality = cms.string('highPurity'),
0024                                             pixelClusters = cms.InputTag("siPixelClusters"),
0025                                             stripClusters = cms.InputTag("siStripClusters"),
0026                                             Common = cms.PSet(
0027                             maxChi2 = cms.double(9.0),
0028                             ),
0029                         Strip = cms.PSet(
0030                             maxChi2 = cms.double(9.0),
0031                             )
0032                                             )
0033
0034
0035
0036 # seeding
0037 hiRegitInitialStepSeeds     = RecoTracker.IterativeTracking.InitialStep_cff.initialStepSeeds.clone(
0038     RegionFactoryPSet = HiTrackingRegionFactoryFromJetsBlock.clone(
0039     RegionPSet = dict(ptMin = 1.2)
0040     ),
0041     ClusterCheckPSet = dict(doClusterCheck = False), # do not check for max number of clusters pixel or strips
0042     skipClusters = cms.InputTag('hiRegitInitialStepClusters')
0043 )
0044 # building: feed the new-named seeds
0045 hiRegitInitialStepTrajectoryFilter = RecoTracker.IterativeTracking.InitialStep_cff.initialStepTrajectoryFilterBase.clone()
0046
0047
0048 hiRegitInitialStepTrajectoryBuilder = RecoTracker.IterativeTracking.InitialStep_cff.initialStepTrajectoryBuilder.clone(
0049     trajectoryFilter = dict(refToPSet_ = 'hiRegitInitialStepTrajectoryFilter'),
0050     clustersToSkip = 'hiRegitInitialStepClusters'
0051 )
0052
0053 # track candidates
0054 hiRegitInitialStepTrackCandidates        =  RecoTracker.IterativeTracking.InitialStep_cff._initialStepTrackCandidatesCkf.clone(
0055     src               = 'hiRegitInitialStepSeeds',
0056     TrajectoryBuilderPSet = cms.PSet(refToPSet_ = cms.string('hiRegitInitialStepTrajectoryBuilder')),
0057     maxNSeeds = 100000
0058 )
0059
0060 # fitting: feed new-names
0061 hiRegitInitialStepTracks                 = RecoTracker.IterativeTracking.InitialStep_cff.initialStepTracks.clone(
0062     src                 = 'hiRegitInitialStepTrackCandidates',
0063     AlgorithmName = 'initialStep'
0064 )
0065
0066
0067 # Track selection
0068 import RecoHI.HiTracking.hiMultiTrackSelector_cfi
0069 hiRegitInitialStepSelector = RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiMultiTrackSelector.clone(
0070     src = 'hiRegitInitialStepTracks',
0071     trackSelectors = cms.VPSet(
0072        RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiLooseMTS.clone(
0073            name = 'hiRegitInitialStepLoose',
0074            d0_par2 = [9999.0, 0.0],
0075            dz_par2 = [9999.0, 0.0],
0076            applyAdaptedPVCuts = False
0077        ), #end of pset
0078        RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
0079            name = 'hiRegitInitialStepTight',
0080            preFilterName = 'hiRegitInitialStepLoose',
0081            d0_par2 = [9999.0, 0.0],
0082            dz_par2 = [9999.0, 0.0],
0083            applyAdaptedPVCuts = False
0084        ),
0085        RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
0086            name = 'hiRegitInitialStep',
0087            preFilterName = 'hiRegitInitialStepTight',
0088            d0_par2 = [9999.0, 0.0],
0089            dz_par2 = [9999.0, 0.0],
0090            applyAdaptedPVCuts = False
0091        ),
0092     ) #end of vpset
0093 ) #end of clone
0094
0095
0096 hiRegitInitialStepTask = cms.Task(hiGeneralTrackFilter,
0097                                   hiRegitInitialStepClusters,
0098                                   hiRegitInitialStepSeeds,
0099                                   hiRegitInitialStepTrackCandidates,
0100                                   hiRegitInitialStepTracks,
0101                                   hiRegitInitialStepSelector)
0102 hiRegitInitialStep = cms.Sequence(hiRegitInitialStepTask)