STDM13.py

Go to the documentation of this file.

# Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
#====================================================================
# DAOD_STDM13.py for W+D analysises
# * 1L skimming
# * InDetTrackParticles with |z0|*sinTheta < 6.0
#====================================================================
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.Enums import MetadataCategory
 
# Main algorithm config
def STDM13KernelCfg(flags, name='STDM13Kernel', **kwargs):
    """Configure the derivation framework driving algorithm (kernel) for STDM13"""
    acc = ComponentAccumulator()
 
    # Common augmentations
    from DerivationFrameworkPhys.PhysCommonConfig import PhysCommonAugmentationsCfg
    acc.merge(PhysCommonAugmentationsCfg(flags, TriggerListsHelper = kwargs['TriggerListsHelper']))
 
    
    from BTagging.FlavorTaggingConfig import FlavorTaggingCfg
    acc.merge(FlavorTaggingCfg(flags, "AntiKt4EMPFlowJets"))
 
    if flags.Input.isMC:
        
        from JetRecConfig.JetRecConfig import getModifier
        from JetRecConfig.StandardJetMods import stdJetModifiers
        from JetRecConfig.StandardSmallRJets import AntiKt4EMPFlow
        JetIRCSafeLabelTool = getModifier(AntiKt4EMPFlow, stdJetModifiers['JetIRCSafeLabel'], stdJetModifiers['JetIRCSafeLabel'].modspec, flags=flags)
        acc.addEventAlgo(CompFactory.JetDecorationAlg(name='JetIRCSafeLabelAlg',
                                                    JetContainer='AntiKt4EMPFlowJets', 
                                                    Decorators=[JetIRCSafeLabelTool]))
 
    #===================================================
    # HEAVY FLAVOR CLASSIFICATION FOR ttbar+jets EVENTS
    #===================================================
    from DerivationFrameworkMCTruth.HFClassificationCommonConfig import HFClassificationCommonCfg
    acc.merge(HFClassificationCommonCfg(flags))
 
    from DerivationFrameworkInDet.InDetToolsConfig import TrackParticleThinningCfg
    
    # filter leptons
    lepton_skimming_expression = 'count( (Muons.pt > 25*GeV) && (0 == Muons.muonType || 1 == Muons.muonType || 4 == Muons.muonType) ) + count(( Electrons.pt > 25*GeV) && ((Electrons.Loose) || (Electrons.DFCommonElectronsLHLoose))) >= 1'
 
    from DerivationFrameworkTools.DerivationFrameworkToolsConfig import (
        xAODStringSkimmingToolCfg)
    STDM13StringSkimmingTool = acc.getPrimaryAndMerge(xAODStringSkimmingToolCfg(
        flags, name = "STDM13StringSkimmingTool", expression = lepton_skimming_expression))
 
    tp_thinning_expression = "abs(InDetTrackParticles.DFCommonInDetTrackZ0AtPV*sin(InDetTrackParticles.theta)) < 6.0*mm"
    STDM13TPThinningTool =  acc.getPrimaryAndMerge(TrackParticleThinningCfg(
        flags,
        name                    = "STDM13TPThinningTool",
        StreamName              = kwargs['StreamName'],
        SelectionString         = tp_thinning_expression,
        InDetTrackParticlesKey  = "InDetTrackParticles"))
 
    # Truth thinning
    if flags.Input.isMC:
        truth_stable_expr = "TruthParticles.isGenStable"   
        STDM13TruthThinningTool = CompFactory.DerivationFramework.GenericTruthThinning(
            name                     = "STDM13TruthThin_stable",
            StreamName               = kwargs['StreamName'],
            ParticleSelectionString  = truth_stable_expr,   
        )
        acc.addPublicTool(STDM13TruthThinningTool)
 
    thinningTools = [STDM13TPThinningTool]
    if flags.Input.isMC:
        thinningTools += [STDM13TruthThinningTool]
    skimmingTools = [STDM13StringSkimmingTool]
    
    DerivationKernel = CompFactory.DerivationFramework.DerivationKernel
    acc.addEventAlgo(DerivationKernel(name, SkimmingTools = skimmingTools, ThinningTools = thinningTools))
 
    return acc
 
 
def STDM13Cfg(flags):
    acc = ComponentAccumulator()
 
    # Get the lists of triggers needed for trigger matching.
    # This is needed at this scope (for the slimming) and further down in the config chain
    # for actually configuring the matching, so we create it here and pass it down
    # TODO: this should ideally be called higher up to avoid it being run multiple times in a train
    from DerivationFrameworkPhys.TriggerListsHelper import TriggerListsHelper
    STDM13TriggerListsHelper = TriggerListsHelper(flags)
 
    # Common augmentations
    acc.merge(STDM13KernelCfg(flags, name="STDM13Kernel", StreamName = 'StreamDAOD_STDM13', TriggerListsHelper = STDM13TriggerListsHelper))
 
    # ============================
    # Define contents of the format
    # =============================
    from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
    from xAODMetaDataCnv.InfileMetaDataConfig import SetupMetaDataForStreamCfg
    from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
    
    STDM13SlimmingHelper = SlimmingHelper("STDM13SlimmingHelper", NamesAndTypes = flags.Input.TypedCollections, flags = flags)
 
    from DerivationFrameworkFlavourTag import FtagBaseContent
    FtagBaseContent.add_common_augmentation(flags, acc, STDM13SlimmingHelper)
    
    STDM13SlimmingHelper.SmartCollections += [
        "Electrons",
        "Muons",
        "AntiKt4EMPFlowJets",
        "MET_Baseline_AntiKt4EMPFlow",
        "PrimaryVertices",
        "InDetTrackParticles",
        "AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets",
    ]
 
    STDM13SlimmingHelper.ExtraVariables += [
        "AntiKt4EMPFlowJets.GhostTrack",
        "InDetTrackParticles.btagIp_d0.btagIp_d0Uncertainty.btagIp_z0SinTheta.btagIp_z0SinThetaUncertainty.btagIp_trackMomentum.btagIp_trackDisplacement.btagIp_invalidIp",
        "InDetTrackParticles.TTVA_AMVFVertices.TTVA_AMVFWeights",
        "InDetTrackParticles.truthMatchProbability.truthParticleLink",
        "AntiKt4EMPFlowJets.GN2v01_TrackOrigin.GN2v01_TrackLinks",
    ]
 
    STDM13SlimmingHelper.AllVariables += [
        "EventInfo",
        "PrimaryVertices",
        "InDetTrackParticles",
        "TruthParticles",
        "AntiKt4TruthDressedWZJets",
        "TruthEvents","TruthHFWithDecayParticles","TruthBoson","TruthBottom", "TruthCharm","TruthElectrons","TruthMuons","TruthTop","TruthTaus","MET_Truth",
        "TruthPrimaryVertices","TruthHFWithDecayVertices",
    ]
 
    STDM13SlimmingHelper.IncludeTriggerNavigation = False
    STDM13SlimmingHelper.IncludeJetTriggerContent = False
    STDM13SlimmingHelper.IncludeMuonTriggerContent = False
    STDM13SlimmingHelper.IncludeEGammaTriggerContent = False
    STDM13SlimmingHelper.IncludeJetTauEtMissTriggerContent = False
    STDM13SlimmingHelper.IncludeTauTriggerContent = False
    STDM13SlimmingHelper.IncludeEtMissTriggerContent = False
    STDM13SlimmingHelper.IncludeBJetTriggerContent = False
    STDM13SlimmingHelper.IncludeBPhysTriggerContent = False
    STDM13SlimmingHelper.IncludeMinBiasTriggerContent = False
 
   # Truth extra content
    if flags.Input.isMC:
 
        from DerivationFrameworkMCTruth.MCTruthCommonConfig import addTruth3ContentToSlimmerTool
        addTruth3ContentToSlimmerTool(STDM13SlimmingHelper)
        STDM13SlimmingHelper.AllVariables += ['TruthLHEParticles', 'TruthHFWithDecayParticles','TruthHFWithDecayVertices','TruthCharm','TruthPileupParticles','InTimeAntiKt4TruthJets','OutOfTimeAntiKt4TruthJets']
        STDM13SlimmingHelper.ExtraVariables += ["Electrons.TruthLink",
                                          "Muons.TruthLink",
                                          "Photons.TruthLink",
                                          "AntiKt4TruthDressedWZJets.IsoFixedCone5Pt"]
        
        from DerivationFrameworkMCTruth.MCTruthCommonConfig import AddTauAndDownstreamParticlesCfg
        acc.merge(AddTauAndDownstreamParticlesCfg(flags))
        STDM13SlimmingHelper.AllVariables += ['TruthTausWithDecayParticles','TruthTausWithDecayVertices']
 
        
        STDM13SlimmingHelper.ExtraVariables += ["AntiKt4EMPFlowJets.IRCSafeLabelIFN.IRCSafeLabelCMP.IRCSafeLabelGHS.IRCSafeLabelSDF.IRCSafeLabelAKT" ]
 
    # Extra content
    STDM13SlimmingHelper.ExtraVariables += ["AntiKt4EMPFlowJets.QGTransformer_ConstScore.DFCommonJets_QGTagger_truthjet_nCharged.DFCommonJets_QGTagger_truthjet_pt.DFCommonJets_QGTagger_truthjet_eta.DFCommonJets_QGTagger_NTracks.DFCommonJets_QGTagger_TracksWidth.DFCommonJets_QGTagger_TracksC1.ConeExclBHadronsFinal.ConeExclCHadronsFinal.GhostBHadronsFinal.GhostCHadronsFinal.GhostBHadronsFinalCount.GhostBHadronsFinalPt.GhostCHadronsFinalCount.GhostCHadronsFinalPt.GhostPartons.isJvtHS.isJvtPU.IsoFixedCone5PtPUsub" ]
 
    # Trigger matching
    # Run 2
    if flags.Trigger.EDMVersion == 2:
        from DerivationFrameworkPhys.TriggerMatchingCommonConfig import AddRun2TriggerMatchingToSlimmingHelper
        AddRun2TriggerMatchingToSlimmingHelper(SlimmingHelper = STDM13SlimmingHelper, 
                                         OutputContainerPrefix = "TrigMatch_", 
                                         TriggerList = STDM13TriggerListsHelper.Run2TriggerNamesTau)
        AddRun2TriggerMatchingToSlimmingHelper(SlimmingHelper = STDM13SlimmingHelper, 
                                         OutputContainerPrefix = "TrigMatch_",
                                         TriggerList = STDM13TriggerListsHelper.Run2TriggerNamesNoTau)
    # Run 3, or Run 2 with navigation conversion
    if flags.Trigger.EDMVersion == 3 or (flags.Trigger.EDMVersion == 2 and flags.Trigger.doEDMVersionConversion):
        from TrigNavSlimmingMT.TrigNavSlimmingMTConfig import AddRun3TrigNavSlimmingCollectionsToSlimmingHelper
        AddRun3TrigNavSlimmingCollectionsToSlimmingHelper(STDM13SlimmingHelper)
 
    # Output stream    
    STDM13ItemList = STDM13SlimmingHelper.GetItemList()
    acc.merge(OutputStreamCfg(flags, "DAOD_STDM13", ItemList=STDM13ItemList, AcceptAlgs=["STDM13Kernel"]))
    acc.merge(SetupMetaDataForStreamCfg(flags, "DAOD_STDM13", AcceptAlgs=["STDM13Kernel"], createMetadata=[MetadataCategory.CutFlowMetaData, MetadataCategory.TruthMetaData]))
 
    return acc