HiTracking/python/HISelectedTracks_cfi.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 #from RecoHI.HiTracking.hiMultiTrackSelector_cfi import *
0004
0005 # Track selection
0006 import RecoHI.HiTracking.hiMultiTrackSelector_cfi
0007 hiInitialStepSelector = RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiMultiTrackSelector.clone(
0008     src ='hiGlobalPrimTracks',
0009     useAnyMVA = True,
0010     GBRForestLabel = 'HIMVASelectorIter4',
0011     GBRForestVars = ['chi2perdofperlayer', 'dxyperdxyerror', 'dzperdzerror', 'nhits', 'nlayers', 'eta'],
0012     trackSelectors= cms.VPSet(
0013         RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiLooseMTS.clone(
0014             name = 'hiInitialStepLoose',
0015             useMVA = False
0016         ), #end of pset
0017         RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
0018             name = 'hiInitialStepTight',
0019             preFilterName = 'hiInitialStepLoose',
0020             useMVA = True,
0021             minMVA = -0.77
0022         ),
0023         RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
0024             name = 'hiInitialStep',
0025             preFilterName = 'hiInitialStepTight',
0026             useMVA = True,
0027             minMVA = -0.77
0028         ),
0029     ) #end of vpset
0030 ) #end of clone
0031 from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
0032 trackingPhase1.toModify(hiInitialStepSelector,
0033     useAnyMVA = False,
0034     trackSelectors= cms.VPSet(
0035         RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiLooseMTS.clone(
0036             name = 'hiInitialStepLoose',
0037             useMVA = False
0038         ), #end of pset
0039         RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
0040             name = 'hiInitialStepTight',
0041             preFilterName = 'hiInitialStepLoose',
0042             useMVA = False,
0043             minMVA = -0.77
0044         ),
0045         RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
0046             name = 'hiInitialStep',
0047             preFilterName = 'hiInitialStepTight',
0048             useMVA = False,
0049             minMVA = -0.77
0050
0051         ),
0052     ) #end of vpset
0053 )
0054
0055
0056
0057 # using the tracklist merger with one collection simply applies the quality flags
0058 from RecoTracker.FinalTrackSelectors.trackAlgoPriorityOrder_cfi import trackAlgoPriorityOrder
0059 import RecoTracker.FinalTrackSelectors.trackListMerger_cfi
0060 hiSelectedTracks = RecoTracker.FinalTrackSelectors.trackListMerger_cfi.trackListMerger.clone(
0061     TrackProducers     = ['hiGlobalPrimTracks'],
0062     hasSelector        = [1],
0063     selectedTrackQuals = ["hiInitialStepSelector:hiInitialStep"],
0064     copyExtras         = True,
0065     copyMVA            = True,
0066     makeReKeyedSeeds   = cms.untracked.bool(False)
0067 )
0068
0069 #complete sequence
0070 hiTracksWithQualityTask = cms.Task(hiInitialStepSelector
0071                                    #* hiSelectedTracks
0072 )