Python toReplaceWith 예제들

예제 #1

파일: MixedTripletStep_cff.py 프로젝트: jkunkle/cmssw

    TIB = dict(clusterChargeCut = dict(refToPSet_ = 'SiStripClusterChargeCutTiny')),
)
trackingPhase1.toModify(mixedTripletStepSeedLayersB, layerList = ['BPix3+BPix4+TIB1'])

# TrackingRegion
mixedTripletStepTrackingRegionsB = _mixedTripletStepTrackingRegionsCommon.clone(RegionPSet = dict(ptMin=0.6, originHalfLength=10.0))
for e in [pp_on_XeXe_2017, pp_on_AA_2018]:
    e.toReplaceWith(mixedTripletStepTrackingRegionsB, 
                    _globalTrackingRegionWithVertices.clone(RegionPSet=dict(
                fixedError = 2.5,
                ptMin = 0.6,
                originRadius = 1.5
                )
                                                                      )
)
highBetaStar_2018.toReplaceWith(mixedTripletStepTrackingRegionsB, _mixedTripletStepTrackingRegionsCommon.clone())

# seeding
mixedTripletStepHitDoubletsB = mixedTripletStepHitDoubletsA.clone(
    seedingLayers = "mixedTripletStepSeedLayersB",
    trackingRegions = "mixedTripletStepTrackingRegionsB",
)
mixedTripletStepHitTripletsB = mixedTripletStepHitTripletsA.clone(doublets = "mixedTripletStepHitDoubletsB")
mixedTripletStepSeedsB = _mixedTripletStepSeedsACommon.clone(seedingHitSets = "mixedTripletStepHitTripletsB")
#fastsim
_fastSim_mixedTripletStepSeedsB = FastSimulation.Tracking.TrajectorySeedProducer_cfi.trajectorySeedProducer.clone(
    layerList = mixedTripletStepSeedLayersB.layerList.value(),
    trackingRegions = "mixedTripletStepTrackingRegionsB",
    hitMasks = cms.InputTag("mixedTripletStepMasks"),
    seedFinderSelector = dict(pixelTripletGeneratorFactory = _hitSetProducerToFactoryPSet(mixedTripletStepHitTripletsB))
)

예제 #2

    layerList=['BPix2+BPix3+TIB1', 'BPix2+BPix3+TIB2'],
    TIB=dict(clusterChargeCut=dict(refToPSet_='SiStripClusterChargeCutTiny')),
)
trackingPhase1.toModify(mixedTripletStepSeedLayersB,
                        layerList=['BPix3+BPix4+TIB1'])

# TrackingRegion
mixedTripletStepTrackingRegionsB = _mixedTripletStepTrackingRegionsCommon.clone(
    RegionPSet=dict(ptMin=0.6, originHalfLength=10.0))
(pp_on_XeXe_2017 | pp_on_AA).toReplaceWith(
    mixedTripletStepTrackingRegionsB,
    _mixedTripletStepTrackingRegionsCommon_pp_on_HI.clone(RegionPSet=dict(
        fixedError=2.5,
        ptMin=0.6,
    )))
highBetaStar_2018.toReplaceWith(mixedTripletStepTrackingRegionsB,
                                _mixedTripletStepTrackingRegionsCommon.clone())

# seeding
mixedTripletStepHitDoubletsB = mixedTripletStepHitDoubletsA.clone(
    seedingLayers='mixedTripletStepSeedLayersB',
    trackingRegions='mixedTripletStepTrackingRegionsB',
)
mixedTripletStepHitTripletsB = mixedTripletStepHitTripletsA.clone(
    doublets='mixedTripletStepHitDoubletsB')
mixedTripletStepSeedsB = _mixedTripletStepSeedsACommon.clone(
    seedingHitSets='mixedTripletStepHitTripletsB')
#fastsim
_fastSim_mixedTripletStepSeedsB = FastSimulation.Tracking.TrajectorySeedProducer_cfi.trajectorySeedProducer.clone(
    trackingRegions='mixedTripletStepTrackingRegionsB',
    hitMasks=cms.InputTag('mixedTripletStepMasks'),
    seedFinderSelector=dict(

예제 #3

# Final selection
from RecoTracker.FinalTrackSelectors.TrackLwtnnClassifier_cfi import *
from RecoTracker.FinalTrackSelectors.trackSelectionLwtnn_cfi import *
highPtTripletStep = TrackLwtnnClassifier.clone(
    src='highPtTripletStepTracks',
    qualityCuts=[0.75, 0.775, 0.8],
)
fastSim.toModify(highPtTripletStep,
                 vertices="firstStepPrimaryVerticesBeforeMixing")

from RecoTracker.FinalTrackSelectors.TrackMVAClassifierPrompt_cfi import *

highBetaStar_2018.toReplaceWith(
    highPtTripletStep,
    TrackMVAClassifierPrompt.clone(
        qualityCuts=[-0.2, 0.3, 0.4],
        src='highPtTripletStepTracks',
        mva=dict(GBRForestLabel='MVASelectorDetachedTripletStep_Phase1')))

pp_on_AA_2018.toReplaceWith(
    highPtTripletStep,
    TrackMVAClassifierPrompt.clone(
        qualityCuts=[-0.9, -0.3, 0.85],
        src='highPtTripletStepTracks',
        mva=dict(GBRForestLabel='HIMVASelectorHighPtTripletStep_Phase1')))

# For Phase2PU140
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
highPtTripletStepSelector = RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
    src='highPtTripletStepTracks',
    trackSelectors=[

예제 #4

# Final selection
from RecoTracker.FinalTrackSelectors.TrackLwtnnClassifier_cfi import *
from RecoTracker.FinalTrackSelectors.trackSelectionLwtnn_cfi import *
lowPtQuadStep = TrackLwtnnClassifier.clone(
    src = 'lowPtQuadStepTracks',
    qualityCuts = [0.2, 0.425, 0.75],
)
fastSim.toModify(lowPtQuadStep,vertices = "firstStepPrimaryVerticesBeforeMixing")

from RecoTracker.FinalTrackSelectors.TrackMVAClassifierPrompt_cfi import *

highBetaStar_2018.toReplaceWith(lowPtQuadStep, TrackMVAClassifierPrompt.clone(
    src = 'lowPtQuadStepTracks',
    qualityCuts = [-0.9,-0.35,-0.15],
    mva = dict(GBRForestLabel = 'MVASelectorDetachedTripletStep_Phase1') 
))

pp_on_AA_2018.toReplaceWith(lowPtQuadStep, TrackMVAClassifierPrompt.clone(
    src = 'lowPtQuadStepTracks',
    qualityCuts = [-0.9, -0.4, 0.3],
    mva = dict(GBRForestLabel = 'HIMVASelectorLowPtQuadStep_Phase1')
))

# For Phase2PU140
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
lowPtQuadStepSelector = RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
    src = 'lowPtQuadStepTracks',
    trackSelectors = [
        RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(

예제 #5

pixelPairStep =  TrackMVAClassifierPrompt.clone()
pixelPairStep.src = 'pixelPairStepTracks'
pixelPairStep.mva.GBRForestLabel = 'MVASelectorIter2_13TeV'
pixelPairStep.qualityCuts = [-0.2,0.0,0.3]

#LWTNN selector
from RecoTracker.FinalTrackSelectors.TrackLwtnnClassifier_cfi import *
from RecoTracker.FinalTrackSelectors.trackSelectionLwtnn_cfi import *
trackingPhase1.toReplaceWith(pixelPairStep, TrackLwtnnClassifier.clone(
    src='pixelPairStepTracks',
    qualityCuts=[-0.6, -0.1, 0.4],
))

highBetaStar_2018.toReplaceWith(pixelPairStep, TrackMVAClassifierPrompt.clone(
    src='pixelPairStepTracks',
    qualityCuts = [-0.95,0.0,0.3],
    mva=dict(GBRForestLabel = 'MVASelectorPixelPairStep_Phase1')
))

pp_on_AA_2018.toReplaceWith(pixelPairStep, TrackMVAClassifierPrompt.clone( 
    src='pixelPairStepTracks',
    qualityCuts = [-0.2, 0.0, 0.98],
    mva=dict(GBRForestLabel = 'MVASelectorPixelPairStep_Phase1') 
))

fastSim.toModify(pixelPairStep, vertices = "firstStepPrimaryVerticesBeforeMixing")



# For LowPU and Phase2PU140
import RecoTracker.IterativeTracking.LowPtTripletStep_cff

예제 #6

    layerList=['BPix2+BPix3+TIB1', 'BPix2+BPix3+TIB2'],
    TIB=dict(clusterChargeCut=dict(refToPSet_='SiStripClusterChargeCutTiny')),
)
trackingPhase1.toModify(mixedTripletStepSeedLayersB,
                        layerList=['BPix3+BPix4+TIB1'])

# TrackingRegion
mixedTripletStepTrackingRegionsB = _mixedTripletStepTrackingRegionsCommon.clone(
    RegionPSet=dict(ptMin=0.6, originHalfLength=10.0))
(pp_on_XeXe_2017 | pp_on_AA_2018).toReplaceWith(
    mixedTripletStepTrackingRegionsB,
    _mixedTripletStepTrackingRegionsCommon_pp_on_HI.clone(RegionPSet=dict(
        fixedError=2.5,
        ptMin=0.6,
    )))
highBetaStar_2018.toReplaceWith(mixedTripletStepTrackingRegionsB,
                                _mixedTripletStepTrackingRegionsCommon.clone())

# seeding
mixedTripletStepHitDoubletsB = mixedTripletStepHitDoubletsA.clone(
    seedingLayers="mixedTripletStepSeedLayersB",
    trackingRegions="mixedTripletStepTrackingRegionsB",
)
mixedTripletStepHitTripletsB = mixedTripletStepHitTripletsA.clone(
    doublets="mixedTripletStepHitDoubletsB")
mixedTripletStepSeedsB = _mixedTripletStepSeedsACommon.clone(
    seedingHitSets="mixedTripletStepHitTripletsB")
#fastsim
_fastSim_mixedTripletStepSeedsB = FastSimulation.Tracking.TrajectorySeedProducer_cfi.trajectorySeedProducer.clone(
    trackingRegions="mixedTripletStepTrackingRegionsB",
    hitMasks=cms.InputTag("mixedTripletStepMasks"),
    seedFinderSelector=dict(

예제 #7

#LWTNN selector
from RecoTracker.FinalTrackSelectors.TrackLwtnnClassifier_cfi import *
from RecoTracker.FinalTrackSelectors.trackSelectionLwtnn_cfi import *
trackingPhase1.toReplaceWith(
    detachedTripletStep,
    TrackLwtnnClassifier.clone(
        src='detachedTripletStepTracks',
        qualityCuts=[0.0, 0.4, 0.8],
    ))
(trackingPhase1 & fastSim).toModify(
    detachedTripletStep, vertices="firstStepPrimaryVerticesBeforeMixing")

highBetaStar_2018.toReplaceWith(
    detachedTripletStep,
    detachedTripletStepClassifier1.clone(
        qualityCuts=[-0.5, 0.0, 0.5],
        mva=dict(GBRForestLabel='MVASelectorDetachedTripletStep_Phase1')))

pp_on_AA_2018.toReplaceWith(
    detachedTripletStep,
    detachedTripletStepClassifier1.clone(
        qualityCuts=[-0.2, 0.4, 0.85],
        mva=dict(GBRForestLabel='HIMVASelectorDetachedTripletStep_Phase1')))

# For LowPU
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
detachedTripletStepSelector = RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
    src='detachedTripletStepTracks',
    useAnyMVA=cms.bool(False),
    GBRForestLabel=cms.string('MVASelectorIter3'),

예제 #8

fastSim.toModify(detachedQuadStepTracks, TTRHBuilder='WithoutRefit')

# TRACK SELECTION AND QUALITY FLAG SETTING.
from RecoTracker.FinalTrackSelectors.TrackLwtnnClassifier_cfi import *
from RecoTracker.FinalTrackSelectors.trackSelectionLwtnn_cfi import *
detachedQuadStep = TrackLwtnnClassifier.clone(
    src='detachedQuadStepTracks',
    qualityCuts=[-0.6, 0.05, 0.7],
)
fastSim.toModify(detachedQuadStep,
                 vertices="firstStepPrimaryVerticesBeforeMixing")

from RecoTracker.FinalTrackSelectors.TrackMVAClassifierDetached_cfi import *
highBetaStar_2018.toReplaceWith(
    detachedQuadStep,
    TrackMVAClassifierDetached.clone(
        src='detachedQuadStepTracks',
        qualityCuts=[-0.7, 0.0, 0.5],
        mva=dict(GBRForestLabel='MVASelectorDetachedQuadStep_Phase1')))

pp_on_AA_2018.toReplaceWith(
    detachedQuadStep,
    TrackMVAClassifierDetached.clone(
        src='detachedQuadStepTracks',
        qualityCuts=[-0.2, 0.2, 0.5],
        mva=dict(GBRForestLabel='HIMVASelectorDetachedQuadStep_Phase1')))

# For Phase2PU140
import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
detachedQuadStepSelector = RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
    src='detachedQuadStepTracks',
    trackSelectors=[