HION5.py

Go to the documentation of this file.

# Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
# HION5.py  
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.Enums import MetadataCategory
from AthenaCommon.CFElements import seqAND
 
def HION5SkimmingToolCfg(flags):
    """Configure the example skimming tool"""
    acc = ComponentAccumulator()
    
    ExtraData  = []
    ExtraData += ['xAOD::MuonContainer/Muons']
    ExtraData += ['xAOD::ElectronContainer/Electrons']
    ExtraData += ['xAOD::PhotonContainer/Photons']
    ExtraData += ['xAOD::TrackParticleContainer/InDetTrackParticles']
    
    acc.addSequence( seqAND("HION5Sequence") )
    acc.getSequence("HION5Sequence").ExtraDataForDynamicConsumers = ExtraData
    acc.getSequence("HION5Sequence").ProcessDynamicDataDependencies = True
    
    from DerivationFrameworkHI import ListTriggers
    
    triggers = ListTriggers.HION5SkimmingTriggers()
    
    req_electrons = 'count( ( Electrons.pt > 15*GeV ) && ( abs(Electrons.eta) < 2.5) )>0'
    req_muons     = 'count( Muons.DFCommonMuonPassPreselection && (Muons.pt > 15*GeV) && ( abs(Muons.eta) < 2.7))>0'
    req_photons = 'count( Photons.DFCommonPhotonsIsEMLoose && (Photons.pt > 30*GeV) ) > 0'
    req_total = '(' + req_electrons + ' || ' + req_muons + ' || ' + req_photons + ')'
 
    expression = ' ( ' +' || '.join(triggers) + ' )  && ' + req_total
    
    from DerivationFrameworkTools.DerivationFrameworkToolsConfig import (
        xAODStringSkimmingToolCfg)
    acc.addPublicTool(acc.getPrimaryAndMerge(xAODStringSkimmingToolCfg(
        flags, name = "HION5StringSkimmingTool", expression = expression)), primary = True)
    
    return acc
 
def HION5Thinning(flags):    
    from DerivationFrameworkInDet.InDetToolsConfig import TrackParticleThinningCfg,JetTrackParticleThinningCfg
    acc = ComponentAccumulator()
 
    # find collision type
    from CoolConvUtilities.ParticleTypeUtil import getTypeForRun
    info=getTypeForRun(flags.Input.RunNumbers[0])
    isOxygenOxygenCollision = False
    if (info.getBeam1Type() == 8) or (info.getBeam2Type() == 8):
        isOxygenOxygenCollision = True
 
    pTCut = 0.9
    if isOxygenOxygenCollision:
        pTCut = 0.5
 
    track_thinning_expression  = f"InDetTrackParticles.pt > {pTCut}*GeV"
    TrackParticleThinningTool  = acc.getPrimaryAndMerge(TrackParticleThinningCfg(
         flags,
         name                    = "PHYSTrackParticleThinningTool",
         StreamName              = "StreamDAOD_HION5", 
         SelectionString         = track_thinning_expression,
         InDetTrackParticlesKey  = "InDetTrackParticles"))
 
    AntiKt2HIJetsThinningTool  = acc.getPrimaryAndMerge(JetTrackParticleThinningCfg(
         flags,
         name                    = "AntiKt2HIJetsThinningTool",
         StreamName              = "StreamDAOD_HION5",
         JetKey                  = "AntiKt2HIJets",
         SelectionString         = "AntiKt2HIJets.pt > 15*GeV",
         InDetTrackParticlesKey  = "InDetTrackParticles"))
    
    AntiKt4HIJetsThinningTool  = acc.getPrimaryAndMerge(JetTrackParticleThinningCfg(
         flags,
         name                    = "AntiKt4HIJetsThinningTool",
         StreamName              = "StreamDAOD_HION5",
         JetKey                  = "AntiKt4HIJets",
         SelectionString         = "AntiKt4HIJets.pt > 15*GeV",
         InDetTrackParticlesKey  = "InDetTrackParticles"))
    
    acc.addPublicTool(TrackParticleThinningTool,primary = True)
    acc.addPublicTool(AntiKt2HIJetsThinningTool)
    acc.addPublicTool(AntiKt4HIJetsThinningTool)
    
    return acc
 
def HION5KernelCfg(flags, name="HION5Kernel", **kwargs):
    """Configure the derivation framework driving algorithm (kernel)
    for HION5"""
    acc = ComponentAccumulator()
 
    # Schedule extra jets collections
    from JetRecConfig.StandardSmallRJets import AntiKt4PV0Track
    from JetRecConfig.JetRecConfig import JetRecCfg
 
    jetList = [AntiKt4PV0Track]
    for jd in jetList:
        acc.merge(JetRecCfg(flags, jd))
 
    # Common augmentations
    # cannot use PhysCommon sequence because
    # - no triggers
    # - no TauJets
    # so we have to use a modified version here
    from DerivationFrameworkMuons.MuonsCommonConfig import MuonsCommonCfg
    from DerivationFrameworkEGamma.EGammaCommonConfig import EGammaCommonCfg
 
    acc.merge(MuonsCommonCfg(flags))
    acc.merge(EGammaCommonCfg(flags))
 
    # jet cleaning
    # standard way in PhysCommon is
    # - calculate tau ID (needed for default jet OR)
    # - decorate jets with overlap removal
    # - do event cleaning
    # but taus are missing in HI derivations so need to do differently
 
    # NO JVT criteria in HI data (see pp config for details)
 
    # Decorate if jet passes OR and save decoration DFCommonJets_passOR
    # Use modified OR that does not check overlaps with taus
    from AssociationUtils.AssociationUtilsConfig import OverlapRemovalToolCfg
 
    outputLabel = "DFCommonJets_passOR_HI"
    bJetLabel = ""  # default
    tauLabel = ""  # workaround for missing taus
    tauKey = ""  # workaround for missing taus
    orTool = acc.popToolsAndMerge(
        OverlapRemovalToolCfg(
            flags, outputLabel=outputLabel, bJetLabel=bJetLabel, doTaus=False
        )
    )
    algOR = CompFactory.OverlapRemovalGenUseAlg(
        "OverlapRemovalGenUseAlg",
        OverlapLabel=outputLabel,
        OverlapRemovalTool=orTool,
        JetKey = 'AntiKt4HIJets',
        TauKey=tauKey,
        TauLabel=tauLabel,
        BJetLabel=bJetLabel,
    )
    acc.addEventAlgo(algOR)
 
    # skimming
    skimmingTool = acc.getPrimaryAndMerge(HION5SkimmingToolCfg(flags))
    
    # Thinning
    thinningTool= acc.getPrimaryAndMerge(HION5Thinning(flags))
 
    # setup the kernel
    acc.addEventAlgo(
        CompFactory.DerivationFramework.DerivationKernel(
            name,
            SkimmingTools = [skimmingTool],
            ThinningTools = [thinningTool],
            AugmentationTools = [],
            ))
 
    return acc
 
def HION5Cfg(flags):
    acc = ComponentAccumulator()
 
    from DerivationFrameworkEGamma.PhotonsCPDetailedContent import PhotonsCPDetailedContent
    from DerivationFrameworkEGamma.ElectronsCPDetailedContent import ExtraElectronShowerShapes,ExtraElectronGSFVar
    
    from DerivationFrameworkPhys.TriggerListsHelper import TriggerListsHelper
    HION5TriggerListsHelper = TriggerListsHelper(flags)
        
    acc.merge(HION5KernelCfg(flags, name="HION5Kernel", StreamName="StreamDAOD_HION5", TriggerListsHelper = HION5TriggerListsHelper,))
    
    # configure slimming
    from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
    from xAODMetaDataCnv.InfileMetaDataConfig import SetupMetaDataForStreamCfg
    from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
    from DerivationFrameworkHI import ListSlimming
    
    HION5SlimmingHelper = SlimmingHelper("HION5SlimmingHelper", NamesAndTypes = flags.Input.TypedCollections, flags = flags)
    HION5SlimmingHelper.AppendToDictionary = {'EventInfo':'xAOD::EventInfo','EventInfoAux':'xAOD::EventAuxInfo',
                                               'TruthEvents':'xAOD::TruthEventContainer','TruthEventsAux':'xAOD::TruthEventAuxContainer',
                                               'MET_Truth':'xAOD::MissingETContainer','MET_TruthAux':'xAOD::MissingETAuxContainer',
                                               'TruthLHEParticles':'xAOD::TruthParticleContainer', 'TruthLHEParticlesAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthElectrons':'xAOD::TruthParticleContainer','TruthElectronsAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthMuons':'xAOD::TruthParticleContainer','TruthMuonsAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthPhotons':'xAOD::TruthParticleContainer','TruthPhotonsAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthTaus':'xAOD::TruthParticleContainer','TruthTausAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthNeutrinos':'xAOD::TruthParticleContainer','TruthNeutrinosAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthBSM':'xAOD::TruthParticleContainer','TruthBSMAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthBoson':'xAOD::TruthParticleContainer','TruthBosonAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthBottom':'xAOD::TruthParticleContainer','TruthBottomAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthTop':'xAOD::TruthParticleContainer','TruthTopAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthForwardProtons':'xAOD::TruthParticleContainer','TruthForwardProtonsAux':'xAOD::TruthParticleAuxContainer',
                                               'BornLeptons':'xAOD::TruthParticleContainer','BornLeptonsAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthBosonsWithDecayParticles':'xAOD::TruthParticleContainer','TruthBosonsWithDecayParticlesAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthBosonsWithDecayVertices':'xAOD::TruthVertexContainer','TruthBosonsWithDecayVerticesAux':'xAOD::TruthVertexAuxContainer',
                                               'TruthBSMWithDecayParticles':'xAOD::TruthParticleContainer','TruthBSMWithDecayParticlesAux':'xAOD::TruthParticleAuxContainer',
                                               'TruthBSMWithDecayVertices':'xAOD::TruthVertexContainer','TruthBSMWithDecayVerticesAux':'xAOD::TruthVertexAuxContainer',
                                               'AntiKt4TruthDressedWZJets':'xAOD::JetContainer','AntiKt4TruthDressedWZJetsAux':'xAOD::JetAuxContainer',
                                               'AntiKt10TruthSoftDropBeta100Zcut10Jets':'xAOD::JetContainer','AntiKt10TruthSoftDropBeta100Zcut10JetsAux':'xAOD::JetAuxContainer',
                                               'MET_Track1000':'xAOD::MissingETContainer', 'MET_Track1000Aux':'xAOD::MissingETAuxContainer',
                                               'MET_Track2000':'xAOD::MissingETContainer', 'MET_Track2000Aux':'xAOD::MissingETAuxContainer',
                                               'MET_Track3000':'xAOD::MissingETContainer', 'MET_Track3000Aux':'xAOD::MissingETAuxContainer',
                                               'MET_Track4000':'xAOD::MissingETContainer', 'MET_Track4000Aux':'xAOD::MissingETAuxContainer',
                                               'MET_Track5000':'xAOD::MissingETContainer', 'MET_Track5000Aux':'xAOD::MissingETAuxContainer',
                                              }
     # Build track MET with ptCut in MeV and HItight Tracks
    from DerivationFrameworkHI.TrackMET_config import Cfg_METTrack
    met_ptCutList = [1000,2000,3000,4000,5000]
 
    for ptCut in met_ptCutList:
        acc.merge(Cfg_METTrack(flags, ptCut))
 
    AllVariables  = []    
    AllVariables += ListSlimming.HION5AllVariables(flags.Input.RunNumbers[0])
    AllVariables += ListSlimming.HION5ExtraContainersTrigger()
 
    if flags.Input.isMC:
        from DerivationFrameworkMCTruth.MCTruthCommonConfig import AddStandardTruthContentsCfg
        acc.merge(AddStandardTruthContentsCfg(flags))
        AllVariables += ListSlimming.HION5AllTruthVariables()
 
    HION5SlimmingHelper.SmartCollections = ListSlimming.HION5SmartCollections()
    HION5SlimmingHelper.ExtraVariables   = ListSlimming.HION5ExtraVariables()
    HION5SlimmingHelper.ExtraVariables   += PhotonsCPDetailedContent
    HION5SlimmingHelper.ExtraVariables   += ExtraElectronShowerShapes
    HION5SlimmingHelper.ExtraVariables   += ExtraElectronGSFVar
    HION5SlimmingHelper.AllVariables     = AllVariables
 
    # Add egamma trigger objects
    HION5SlimmingHelper.IncludeEGammaTriggerContent = True
    # Add muon trigger objects
    HION5SlimmingHelper.IncludeMuonTriggerContent = True
    
    HION5ItemList = HION5SlimmingHelper.GetItemList()
 
    acc.merge(OutputStreamCfg(flags, "DAOD_HION5", ItemList=HION5ItemList, AcceptAlgs=["HION5Kernel"]))
    acc.merge(SetupMetaDataForStreamCfg(flags, "DAOD_HION5", AcceptAlgs=["HION5Kernel"], createMetadata=[MetadataCategory.CutFlowMetaData]))
 
    return acc