nano_cff Namespace Reference

Functions
def	nanoAOD_activateVID (process)
def	nanoAOD_addBoostedTauIds (process, idsToRun=[])
def	nanoAOD_addTauIds (process, idsToRun=[])
def	nanoAOD_addUTagToTaus (process, addUTagInfo=False, usePUPPIjets=False)
def	nanoAOD_customizeCommon (process)
def	nanoWmassGenCustomize (process)
	increasing the precision of selected GenParticles. More...

Variables
	_nanoTableTaskCommonRun2
	boostedTaus
	electrons
	jets
	lhcInfoTable
	linkedObjects
	lowPtElectrons
	muons
	nanoMetadata
	nanoSequence
	nanoSequenceCommon
	nanoSequenceFS
	nanoSequenceMC
	nanoSequenceOnlyData
	nanoSequenceOnlyFullSim
	nanoTableTaskCommon
	nanoTableTaskFS
	photons
	strings
	tag
	taus
	useNewLHCInfo
	vertices

Function Documentation

nanoAOD_activateVID()

def nano_cff.nanoAOD_activateVID

(

process

)

Definition at line 190 of file nano_cff.py.

References vid_id_tools.setupAllVIDIdsInModule(), vid_id_tools.switchOnVIDElectronIdProducer(), and vid_id_tools.switchOnVIDPhotonIdProducer().

Referenced by nanoAOD_customizeCommon().

def nanoAOD_activateVID(process):

    switchOnVIDElectronIdProducer(process,DataFormat.MiniAOD,electronTask)
    for modname in electron_id_modules_WorkingPoints_nanoAOD.modules:
        setupAllVIDIdsInModule(process,modname,setupVIDElectronSelection)

    process.electronTask.add( process.egmGsfElectronIDTask )

    switchOnVIDPhotonIdProducer(process,DataFormat.MiniAOD,photonTask) # do not call this to avoid resetting photon IDs in VID, if called before inside makePuppiesFromMiniAOD
    for modname in photon_id_modules_WorkingPoints_nanoAOD.modules:
        setupAllVIDIdsInModule(process,modname,setupVIDPhotonSelection)

    process.photonTask.add( process.egmPhotonIDTask )

    return process

nanoAOD_addBoostedTauIds()

def nano_cff.nanoAOD_addBoostedTauIds	(		process,
			idsToRun = []
	)

Definition at line 125 of file nano_cff.py.

Referenced by nanoAOD_customizeCommon().

def nanoAOD_addBoostedTauIds(process, idsToRun=[]):
    if idsToRun: #no-empty list of tauIDs to run
        updatedBoostedTauName = "slimmedTausBoostedNewID"
        boostedTauIdEmbedder = tauIdConfig.TauIDEmbedder(process, debug = False,
                                                         originalTauName = "slimmedTausBoosted",
                                                         updatedTauName = updatedBoostedTauName,
                                                         postfix = "BoostedForNano",
                                                         toKeep = idsToRun)
        boostedTauIdEmbedder.runTauID()
        process.finalBoostedTaus.src = updatedBoostedTauName
        #remember to adjust the selection and tables with added IDs

        process.boostedTauTask.add( process.rerunMvaIsolationTaskBoostedForNano, getattr(process, updatedBoostedTauName))

    return process

nanoAOD_addTauIds()

def nano_cff.nanoAOD_addTauIds	(		process,
			idsToRun = []
	)

Definition at line 110 of file nano_cff.py.

Referenced by nanoAOD_customizeCommon().

def nanoAOD_addTauIds(process, idsToRun=[]):
    if idsToRun: #no-empty list of tauIDs to run
        updatedTauName = "slimmedTausUpdated"
        tauIdEmbedder = tauIdConfig.TauIDEmbedder(process, debug = False,
                                                  updatedTauName = updatedTauName,
                                                  postfix = "ForNano",
            toKeep = idsToRun)
        tauIdEmbedder.runTauID()
        process.finalTaus.src = updatedTauName
        #remember to adjust the selection and tables with added IDs

        process.tauTask.add( process.rerunMvaIsolationTaskForNano, getattr(process, updatedTauName) )

    return process

nanoAOD_addUTagToTaus()

def nano_cff.nanoAOD_addUTagToTaus	(		process,
			addUTagInfo = False,
			usePUPPIjets = False
	)

Definition at line 141 of file nano_cff.py.

Referenced by nanoAOD_customizeCommon().

def nanoAOD_addUTagToTaus(process, addUTagInfo=False, usePUPPIjets=False):
    
    if addUTagInfo:
        originalTauName = process.finalTaus.src.value()
        
        if usePUPPIjets: # option to use PUPPI jets   
            jetCollection = "updatedJetsPuppi"
            TagName = "pfUnifiedParticleTransformerAK4JetTags"
            tag_prefix = "byUTagPUPPI"
            updatedTauName = originalTauName+'WithUTagPUPPI'
            # Unified ParT Tagger used for PUPPI jets
            from RecoBTag.ONNXRuntime.pfUnifiedParticleTransformerAK4JetTags_cfi import pfUnifiedParticleTransformerAK4JetTags
            Discriminators = [TagName+":"+tag for tag in pfUnifiedParticleTransformerAK4JetTags.flav_names.value()]
        else: # use CHS jets by default
            jetCollection = "updatedJets"
            TagName = "pfParticleNetFromMiniAODAK4CHSCentralJetTags"
            tag_prefix = "byUTagCHS"
            updatedTauName = originalTauName+'WithUTagCHS'
            # PNet tagger used for CHS jets
            from RecoBTag.ONNXRuntime.pfParticleNetFromMiniAODAK4_cff import pfParticleNetFromMiniAODAK4CHSCentralJetTags
            Discriminators = [TagName+":"+tag for tag in pfParticleNetFromMiniAODAK4CHSCentralJetTags.flav_names.value()]

        # Define "hybridTau" producer
        from PhysicsTools.PatAlgos.patTauHybridProducer_cfi import patTauHybridProducer
        setattr(process, updatedTauName, patTauHybridProducer.clone(
            src = originalTauName,
            jetSource = jetCollection,
            dRMax = 0.4,
            jetPtMin = 15,
            jetEtaMax = 2.5,
            UTagLabel = TagName,
            UTagScoreNames = Discriminators,
            tagPrefix = tag_prefix,
            tauScoreMin = -1,
            vsJetMin = 0.05,
            checkTauScoreIsBest = False,
            chargeAssignmentProbMin = 0.2,
            addGenJetMatch = False,
            genJetMatch = ""
        ))
        process.finalTaus.src = updatedTauName

        #remember to adjust the selection and tables with added IDs

        process.tauTask.add(process.jetTask, getattr(process, updatedTauName))

    return process

nanoAOD_customizeCommon()

def nano_cff.nanoAOD_customizeCommon

(

process

)

Definition at line 206 of file nano_cff.py.

References leptonTimeLifeInfo_common_cff.addTimeLifeInfoBase(), nanoAOD_activateVID(), nanoAOD_addBoostedTauIds(), jetsAK4_Puppi_cff.nanoAOD_addDeepInfoAK4(), jetsAK4_CHS_cff.nanoAOD_addDeepInfoAK4CHS(), jetsAK8_cff.nanoAOD_addDeepInfoAK8(), nanoAOD_addTauIds(), and nanoAOD_addUTagToTaus().

def nanoAOD_customizeCommon(process):

    process = nanoAOD_activateVID(process)

    run2_nanoAOD_106Xv2.toModify(
        nanoAOD_addDeepInfoAK4CHS_switch, nanoAOD_addParticleNet_switch=True,
        nanoAOD_addRobustParTAK4Tag_switch=False,
        nanoAOD_addUnifiedParTAK4Tag_switch=True,
    )
  
    # enable rerun of PNet for CHS jets for early run3 eras
    # (it is rerun for run2 within jet tasks while is not needed for newer
    # run3 eras as it is present in miniAOD)
    (run3_nanoAOD_122 | run3_nanoAOD_124).toModify(
        nanoAOD_addDeepInfoAK4CHS_switch, nanoAOD_addParticleNet_switch = True
    )
    
    # This function is defined in jetsAK4_Puppi_cff.py
    process = nanoAOD_addDeepInfoAK4(process,
        addParticleNet=nanoAOD_addDeepInfoAK4_switch.nanoAOD_addParticleNet_switch,
        addRobustParTAK4=nanoAOD_addDeepInfoAK4_switch.nanoAOD_addRobustParTAK4Tag_switch,
        addUnifiedParTAK4=nanoAOD_addDeepInfoAK4_switch.nanoAOD_addUnifiedParTAK4Tag_switch
    )

    # This function is defined in jetsAK4_CHS_cff.py
    process = nanoAOD_addDeepInfoAK4CHS(process,
        addDeepBTag=nanoAOD_addDeepInfoAK4CHS_switch.nanoAOD_addDeepBTag_switch,
        addDeepFlavour=nanoAOD_addDeepInfoAK4CHS_switch.nanoAOD_addDeepFlavourTag_switch,
        addParticleNet=nanoAOD_addDeepInfoAK4CHS_switch.nanoAOD_addParticleNet_switch,
        addRobustParTAK4=nanoAOD_addDeepInfoAK4CHS_switch.nanoAOD_addRobustParTAK4Tag_switch,
        addUnifiedParTAK4=nanoAOD_addDeepInfoAK4CHS_switch.nanoAOD_addUnifiedParTAK4Tag_switch
    )

    # This function is defined in jetsAK8_cff.py
    process = nanoAOD_addDeepInfoAK8(process,
        addDeepBTag=nanoAOD_addDeepInfoAK8_switch.nanoAOD_addDeepBTag_switch,
        addDeepBoostedJet=nanoAOD_addDeepInfoAK8_switch.nanoAOD_addDeepBoostedJet_switch,
        addDeepDoubleX=nanoAOD_addDeepInfoAK8_switch.nanoAOD_addDeepDoubleX_switch,
        addDeepDoubleXV2=nanoAOD_addDeepInfoAK8_switch.nanoAOD_addDeepDoubleXV2_switch,
        addParticleNetMassLegacy=nanoAOD_addDeepInfoAK8_switch.nanoAOD_addParticleNetMassLegacy_switch,
        addParticleNet=nanoAOD_addDeepInfoAK8_switch.nanoAOD_addParticleNet_switch,
        jecPayload=nanoAOD_addDeepInfoAK8_switch.jecPayload
    )

    nanoAOD_tau_switch = cms.PSet(
        idsToAdd = cms.vstring(),
        addUParTInfo = cms.bool(True),
        addPNet = cms.bool(True)
    )
    (run2_nanoAOD_106Xv2 | run3_nanoAOD_122).toModify(
        nanoAOD_tau_switch, idsToAdd = ["deepTau2018v2p5"]
    ).toModify(
        process, lambda p : nanoAOD_addTauIds(p, nanoAOD_tau_switch.idsToAdd.value())
    )
    
    # Don't add Unified Tagger for PUPPI jets for Run 2 (as different PUPPI tune
    # and base jet algorithm) or early Run 3 eras
    (run3_nanoAOD_122 | run3_nanoAOD_124 | run2_nanoAOD_106Xv2).toModify(
        nanoAOD_tau_switch, addUParTInfo = False
    )
    
    # Add Unified Tagger For CHS Jets (PNet 2023)
    nanoAOD_addUTagToTaus(process,
                          addUTagInfo = nanoAOD_tau_switch.addPNet.value(),
                          usePUPPIjets = False
    )

    # Add Unified Tagger For PUPPI Jets (UParT 2024)
    nanoAOD_addUTagToTaus(process,
                        addUTagInfo = nanoAOD_tau_switch.addUParTInfo.value(),
                        usePUPPIjets = True
    )
    
    nanoAOD_boostedTau_switch = cms.PSet(
        idsToAdd = cms.vstring()
    )
    run2_nanoAOD_106Xv2.toModify(
        nanoAOD_boostedTau_switch, idsToAdd = ["mvaIso", "mvaIsoNewDM", "mvaIsoDR0p3", "againstEle"]
    ).toModify(
        process, lambda p : nanoAOD_addBoostedTauIds(p, nanoAOD_boostedTau_switch.idsToAdd.value())
    )

    # Add lepton time-life info
    from PhysicsTools.NanoAOD.leptonTimeLifeInfo_common_cff import addTimeLifeInfoBase
    process = addTimeLifeInfoBase(process)

    return process

nanoWmassGenCustomize()

def nano_cff.nanoWmassGenCustomize

(

process

)

increasing the precision of selected GenParticles.

Definition at line 295 of file nano_cff.py.

def nanoWmassGenCustomize(process):
    pdgSelection="?(abs(pdgId) == 11|| abs(pdgId)==13 || abs(pdgId)==15 ||abs(pdgId)== 12 || abs(pdgId)== 14 || abs(pdgId)== 16|| abs(pdgId)== 24|| pdgId== 23)"
    # Keep precision same as default RECO for selected particles
    ptPrecision="{}?{}:{}".format(pdgSelection, CandVars.pt.precision.value(),genParticleTable.variables.pt.precision.value())
    process.genParticleTable.variables.pt.precision=cms.string(ptPrecision)
    phiPrecision="{} ? {} : {}".format(pdgSelection, CandVars.phi.precision.value(), genParticleTable.variables.phi.precision.value())
    process.genParticleTable.variables.phi.precision=cms.string(phiPrecision)
    etaPrecision="{} ? {} : {}".format(pdgSelection, CandVars.eta.precision.value(), genParticleTable.variables.eta.precision.value())
    process.genParticleTable.variables.eta.precision=cms.string(etaPrecision)
    return process

Variable Documentation

_nanoTableTaskCommonRun2

nano_cff._nanoTableTaskCommonRun2

private

Definition at line 80 of file nano_cff.py.

boostedTaus

nano_cff.boostedTaus

Definition at line 43 of file nano_cff.py.

Referenced by PATObjectCrossLinker.produce().

electrons

nano_cff.electrons

Definition at line 40 of file nano_cff.py.

jets

nano_cff.jets

Definition at line 38 of file nano_cff.py.

lhcInfoTable

nano_cff.lhcInfoTable

Definition at line 59 of file nano_cff.py.

linkedObjects

nano_cff.linkedObjects

Definition at line 37 of file nano_cff.py.

lowPtElectrons

nano_cff.lowPtElectrons

Definition at line 41 of file nano_cff.py.

Referenced by PATObjectCrossLinker.produce().

muons

nano_cff.muons

Definition at line 39 of file nano_cff.py.

nanoMetadata

nano_cff.nanoMetadata

Definition at line 31 of file nano_cff.py.

nanoSequence

nano_cff.nanoSequence

Definition at line 93 of file nano_cff.py.

nanoSequenceCommon

nano_cff.nanoSequenceCommon

Definition at line 88 of file nano_cff.py.

nanoSequenceFS

nano_cff.nanoSequenceFS

Definition at line 102 of file nano_cff.py.

nanoSequenceMC

nano_cff.nanoSequenceMC

Definition at line 105 of file nano_cff.py.

nanoSequenceOnlyData

nano_cff.nanoSequenceOnlyData

Definition at line 91 of file nano_cff.py.

nanoSequenceOnlyFullSim

nano_cff.nanoSequenceOnlyFullSim

Definition at line 90 of file nano_cff.py.

nanoTableTaskCommon

nano_cff.nanoTableTaskCommon

Definition at line 64 of file nano_cff.py.

nanoTableTaskFS

nano_cff.nanoTableTaskFS

Definition at line 95 of file nano_cff.py.

photons

nano_cff.photons

Definition at line 44 of file nano_cff.py.

strings

nano_cff.strings

Definition at line 32 of file nano_cff.py.

tag

nano_cff.tag

Definition at line 33 of file nano_cff.py.

taus

nano_cff.taus

Definition at line 42 of file nano_cff.py.

useNewLHCInfo

nano_cff.useNewLHCInfo

Definition at line 61 of file nano_cff.py.

vertices

nano_cff.vertices

Definition at line 45 of file nano_cff.py.

---

Generated for CMSSW Reference Manual by  1.8.14