CommonTools/ParticleFlow/python/PFBRECO_cff.py - master - githubmirror/cmssw

CommonTools/ParticleFlow/python/PFBRECO_cff.py
187 строк9 KB
Andreas Hinzmann
rename Enable to enable
15 янв 2022, 19:12
15 янв 2022, 19:1238aecdb
import FWCore.ParameterSet.Config as cms

# getting the ptrs
from RecoParticleFlow.PFProducer.pfLinker_cff import particleFlowPtrs

from CommonTools.ParticleFlow.pfPileUp_cfi import *
from CommonTools.ParticleFlow.TopProjectors.pfNoPileUp_cfi import *
pfPileUpIsoPFBRECO = pfPileUp.clone( PFCandidates = 'particleFlowPtrs' )
from Configuration.ProcessModifiers.pp_on_AA_cff import pp_on_AA
pp_on_AA.toModify(pfPileUpIsoPFBRECO, enable = False)

pfNoPileUpIsoPFBRECO = pfNoPileUp.clone( topCollection = 'pfPileUpIsoPFBRECO',
                                         bottomCollection = 'particleFlowPtrs')
pfNoPileUpIsoPFBRECOTask = cms.Task(
    pfPileUpIsoPFBRECO,
    pfNoPileUpIsoPFBRECO
    )
pfNoPileUpIsoPFBRECOSequence = cms.Sequence(pfNoPileUpIsoPFBRECOTask)

from CommonTools.ParticleFlow.pfNoPileUpJME_cff import *

pfPileUpPFBRECO = pfPileUp.clone( PFCandidates = 'particleFlowPtrs' )
pfNoPileUpPFBRECO = pfNoPileUp.clone( topCollection = 'pfPileUpPFBRECO',
                                      bottomCollection = 'particleFlowPtrs')
pfNoPileUpPFBRECOTask = cms.Task(
    pfPileUpPFBRECO,
    pfNoPileUpPFBRECO
    )
pfNoPileUpPFBRECOSequence = cms.Sequence(pfNoPileUpPFBRECOTask)

from CommonTools.ParticleFlow.ParticleSelectors.pfAllNeutralHadrons_cfi import *
pfAllNeutralHadronsPFBRECO = pfAllNeutralHadrons.clone( src = 'pfNoPileUpIsoPFBRECO' )
from CommonTools.ParticleFlow.ParticleSelectors.pfAllChargedHadrons_cfi import *
pfAllChargedHadronsPFBRECO = pfAllChargedHadrons.clone( src = 'pfNoPileUpIsoPFBRECO' )
from CommonTools.ParticleFlow.ParticleSelectors.pfAllPhotons_cfi import *
pfAllPhotonsPFBRECO = pfAllPhotons.clone( src = 'pfNoPileUpIsoPFBRECO' )
from CommonTools.ParticleFlow.ParticleSelectors.pfAllMuons_cfi import *
pfAllMuonsPFBRECO = pfAllMuons.clone( src = 'pfNoPileUpPFBRECO' )
pfAllMuonsClonesPFBRECO = pfAllMuonsClones.clone( src = 'pfAllMuonsPFBRECO' )
from CommonTools.ParticleFlow.ParticleSelectors.pfAllElectrons_cfi import *
pfAllElectronsPFBRECO = pfAllElectrons.clone( src = 'pfNoMuonPFBRECO' )
pfAllElectronsClonesPFBRECO = pfAllElectronsClones.clone( src = 'pfAllElectronsPFBRECO' )
from CommonTools.ParticleFlow.ParticleSelectors.pfAllChargedParticles_cfi import *
pfAllChargedParticlesPFBRECO = pfAllChargedParticles.clone( src = 'pfNoPileUpIsoPFBRECO' )
from CommonTools.ParticleFlow.ParticleSelectors.pfAllNeutralHadronsAndPhotons_cfi import *
pfAllNeutralHadronsAndPhotonsPFBRECO = pfAllNeutralHadronsAndPhotons.clone( src = 'pfNoPileUpIsoPFBRECO' )
pfPileUpAllChargedParticlesPFBRECO = pfAllChargedParticles.clone( src = 'pfPileUpIsoPFBRECO' )
pfSortByTypePFBRECOTask = cms.Task(
    pfAllNeutralHadronsPFBRECO,
    pfAllChargedHadronsPFBRECO,
    pfAllPhotonsPFBRECO,
    # charged hadrons , electrons , muons
    pfAllChargedParticlesPFBRECO,
    # same, but from pile up
    pfPileUpAllChargedParticlesPFBRECO,
    pfAllNeutralHadronsAndPhotonsPFBRECO
#    ,
#    pfAllElectronsPFBRECO,
#    pfAllMuonsPFBRECO
    )
pfSortByTypePFBRECOSequence = cms.Sequence(pfSortByTypePFBRECOTask)

pfParticleSelectionPFBRECOTask = cms.Task(
    pfNoPileUpIsoPFBRECOTask,
    # In principle JME sequence should go here, but this is used in RECO
    # in addition to here, and is used in the "first-step" PF process
    # so needs to go later.
    #pfNoPileUpJMETask ,
    pfNoPileUpPFBRECOTask,
    pfSortByTypePFBRECOTask
    )
pfParticleSelectionPFBRECOSequence = cms.Sequence(pfParticleSelectionPFBRECOTask)

from CommonTools.ParticleFlow.ParticleSelectors.pfSelectedPhotons_cfi import *
pfSelectedPhotonsPFBRECO = pfSelectedPhotons.clone( src = 'pfAllPhotonsPFBRECO' )
from CommonTools.ParticleFlow.Isolation.pfPhotonIsolationPFBRECO_cff import *
from CommonTools.ParticleFlow.Isolation.pfIsolatedPhotons_cfi import *
pfIsolatedPhotonsPFBRECO = pfIsolatedPhotons.clone( 
    src = 'pfSelectedPhotonsPFBRECO',
    isolationValueMapsCharged = ["phPFIsoValueCharged04PFIdPFBRECO"],
    isolationValueMapsNeutral = ["phPFIsoValueNeutral04PFIdPFBRECO",
                                 "phPFIsoValueGamma04PFIdPFBRECO"],
    deltaBetaIsolationValueMap = 'phPFIsoValuePU04PFIdPFBRECO' 
)
pfPhotonPFBRECOTask = cms.Task(
    pfSelectedPhotonsPFBRECO ,
    pfPhotonIsolationPFBRECOTask ,
    # selecting isolated photons:
    pfIsolatedPhotonsPFBRECO
    )
pfPhotonPFBRECOSequence = cms.Sequence(pfPhotonPFBRECOTask)

from CommonTools.ParticleFlow.ParticleSelectors.pfMuonsFromVertex_cfi import *
pfMuonsFromVertexPFBRECO = pfMuonsFromVertex.clone( src = 'pfAllMuonsPFBRECO' )
from CommonTools.ParticleFlow.Isolation.pfIsolatedMuons_cfi import *
pfIsolatedMuonsPFBRECO = pfIsolatedMuons.clone( src = 'pfMuonsFromVertexPFBRECO' )
pfMuonsPFBRECO = pfIsolatedMuonsPFBRECO.clone(cut = "pt > 5 & muonRef.isAvailable()")
pfMuonPFBRECOTask = cms.Task(
    pfAllMuonsPFBRECO ,
    pfMuonsFromVertexPFBRECO ,
    pfIsolatedMuonsPFBRECO,
    pfMuonsPFBRECO
    )
pfMuonPFBRECOSequence = cms.Sequence(pfMuonPFBRECOTask)

from CommonTools.ParticleFlow.ParticleSelectors.pfElectronsFromVertex_cfi import *
pfElectronsFromVertexPFBRECO = pfElectronsFromVertex.clone( src = 'pfAllElectronsPFBRECO' )
from CommonTools.ParticleFlow.Isolation.pfIsolatedElectrons_cfi import *
pfIsolatedElectronsPFBRECO = pfIsolatedElectrons.clone( src = 'pfElectronsFromVertexPFBRECO' )
pfElectronsPFBRECO = pfIsolatedElectronsPFBRECO.clone( cut = " pt > 5 & gsfElectronRef.isAvailable() & gsfTrackRef.hitPattern().numberOfLostHits('MISSING_INNER_HITS')<2")
pfElectronPFBRECOTask = cms.Task(
    pfAllElectronsPFBRECO ,
    pfElectronsFromVertexPFBRECO ,
    pfIsolatedElectronsPFBRECO ,
    pfElectronsPFBRECO
    )
pfElectronPFBRECOSequence = cms.Sequence(pfElectronPFBRECOTask)

from CommonTools.ParticleFlow.Tools.jetTools import jetAlgo
pfJetsPFBRECO = jetAlgo('AK4')
pfJetsPFBRECO.src = 'pfNoElectronJMEPFBRECO'
pfJetsPtrsPFBRECO = cms.EDProducer("PFJetFwdPtrProducer",
                                   src=cms.InputTag("pfJetsPFBRECO")
                                   )
pfJetPFBRECOTask = cms.Task(
    pfJetsPFBRECO ,
    pfJetsPtrsPFBRECO
    )
pfJetPFBRECOSequence = cms.Sequence(pfJetPFBRECOTask)

from CommonTools.ParticleFlow.pfTaus_cff import *

from CommonTools.ParticleFlow.pfMET_cfi import *
pfMETPFBRECO = pfMET.clone( srcJets = 'pfJetsPFBRECO' )

##delta beta weighting
#from CommonTools.ParticleFlow.deltaBetaWeights_cfi import *
#pfWeightedPhotonsPFBRECO = pfWeightedPhotons.clone( src = 'pfAllPhotonsPFBRECO',
                                                    #chargedFromPV = 'pfAllChargedParticlesPFBRECO',
                                                    #chargedFromPU = 'pfPileUpAllChargedParticlesPFBRECO' )
#pfWeightedNeutralHadronsPFBRECO = pfWeightedNeutralHadrons.clone( src = 'pfAllNeutralHadronsPFBRECO',
                                                                  #chargedFromPV = 'pfAllChargedParticlesPFBRECO',
                                                                  #chargedFromPU = 'pfPileUpAllChargedParticlesPFBRECO' )
#pfDeltaBetaWeightingPFBRECOTask = cms.Task(pfWeightedPhotonsPFBRECO+pfWeightedNeutralHadronsPFBRECO)
#pfDeltaBetaWeightingPFBRECOSequence = cms.Sequence(pfDeltaBetaWeightingPFBRECOTask)

# sequential top projection cleaning
from CommonTools.ParticleFlow.TopProjectors.pfNoMuon_cfi import *
pfNoMuonPFBRECO = pfNoMuon.clone( topCollection = 'pfIsolatedMuonsPFBRECO',
                                  bottomCollection = 'pfNoPileUpPFBRECO' )
pfNoMuonJMEPFBRECO = pfNoMuonJME.clone( topCollection = 'pfIsolatedMuonsPFBRECO' )
from CommonTools.ParticleFlow.TopProjectors.pfNoElectron_cfi import *
pfNoElectronPFBRECO = pfNoElectron.clone( topCollection = 'pfIsolatedElectronsPFBRECO',
                                          bottomCollection = 'pfNoMuonPFBRECO' )
pfNoElectronJMEPFBRECO = pfNoElectronJME.clone( topCollection = 'pfIsolatedElectronsPFBRECO',
                                                bottomCollection = 'pfNoMuonJMEPFBRECO' )
pfNoElectronJMEClonesPFBRECO = pfNoElectronJMEClones.clone( src = 'pfNoElectronJMEPFBRECO' )
from CommonTools.ParticleFlow.TopProjectors.pfNoJet_cff import *
pfNoJetPFBRECO = pfNoJet.clone( topCollection = 'pfJetsPtrsPFBRECO',
                                bottomCollection = 'pfNoElectronJMEPFBRECO' )
from CommonTools.ParticleFlow.TopProjectors.pfNoTau_cff import *
pfNoTauPFBRECO = pfNoTau.clone ( bottomCollection = 'pfJetsPtrsPFBRECO' )
pfNoTauClonesPFBRECO = pfNoTauClones.clone ( src = 'pfNoTauPFBRECO' )

# generator tools
from CommonTools.ParticleFlow.genForPF2PAT_cff import *

PFBRECOTask = cms.Task(
    particleFlowPtrs ,
    pfParticleSelectionPFBRECOTask ,
    pfNoPileUpJMETask ,
#    pfDeltaBetaWeightingPFBRECOTask ,
    pfPhotonPFBRECOTask ,
    pfMuonPFBRECOTask ,
    pfNoMuonPFBRECO ,
    pfNoMuonJMEPFBRECO ,
    pfElectronPFBRECOTask ,
    pfNoElectronPFBRECO ,
    pfNoElectronJMEPFBRECO ,
    pfNoElectronJMEClonesPFBRECO,
    pfJetPFBRECOTask ,
    pfNoJetPFBRECO ,
    pfTauTask ,
    pfNoTauPFBRECO ,
    pfMETPFBRECO
    )
PFBRECO = cms.Sequence(PFBRECOTask)