IDTR2.py

Go to the documentation of this file.

# Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
 
# ====================================================================
# IDTR2.py
# ====================================================================
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.Enums import MetadataCategory
 
streamName = "StreamDAOD_IDTR2"
 
 
def IDTR2Cfg(flags):
 
    acc = ComponentAccumulator()
 
    # LRT merge
    TrackParticleContainer = "InDetTrackParticles"
    if flags.Tracking.doLargeD0:
        from DerivationFrameworkInDet.InDetToolsConfig import InDetLRTMergeCfg
        acc.merge(InDetLRTMergeCfg(flags))
        TrackParticleContainer = "InDetWithLRTTrackParticles"
 
    # VrtSecInclusive
    from VrtSecInclusive.VrtSecInclusiveConfig import VrtSecInclusiveCfg
    acc.merge(VrtSecInclusiveCfg(
        flags,
        name="VrtSecInclusive",
        AugmentingVersionString="",
        FillIntermediateVertices=False,
        TrackLocation=TrackParticleContainer))
 
    # NewVrtSecInclusive
    from NewVrtSecInclusiveTool.NewVrtSecInclusiveConfig import (
        MaterialSVFinderToolCfg, DVFinderToolCfg, KsFinderToolCfg)
    MaterialSVFinderTool = acc.popToolsAndMerge(
        MaterialSVFinderToolCfg(flags, AugmentingVersionString="_Material"))
    acc.addEventAlgo(CompFactory.Rec.NewVrtSecInclusiveAlg(
        name="NewVrtSecInclusive_Material",
        TrackParticleContainer=TrackParticleContainer,
        PrimaryVertexContainer="PrimaryVertices",
        BVertexContainerName="NewVrtSecInclusive_SecondaryVertices_Material",
        BVertexTool=MaterialSVFinderTool))
 
    DVFinderTool = acc.popToolsAndMerge(
        DVFinderToolCfg(flags, AugmentingVersionString="_DV"))
    acc.addEventAlgo(CompFactory.Rec.NewVrtSecInclusiveAlg(
        name="NewVrtSecInclusive_DV",
        TrackParticleContainer=TrackParticleContainer,
        PrimaryVertexContainer="PrimaryVertices",
        BVertexContainerName="NewVrtSecInclusive_SecondaryVertices_DV",
        BVertexTool=DVFinderTool))
 
    KsFinderTool = acc.popToolsAndMerge(
        KsFinderToolCfg(flags, AugmentingVersionString="_Ks"))
    acc.addEventAlgo(CompFactory.Rec.NewVrtSecInclusiveAlg(
        name="NewVrtSecInclusive_Ks",
        TrackParticleContainer=TrackParticleContainer,
        PrimaryVertexContainer="PrimaryVertices",
        BVertexContainerName="NewVrtSecInclusive_SecondaryVertices_Ks",
        BVertexTool=KsFinderTool))
 
    # V0Finder
    IDTR2V0ContainerName = "IDTR2RecoV0Candidates"
    IDTR2KshortContainerName = "IDTR2RecoKshortCandidates"
    IDTR2LambdaContainerName = "IDTR2RecoLambdaCandidates"
    IDTR2LambdabarContainerName = "IDTR2RecoLambdabarCandidates"
 
    from InDetConfig.InDetV0FinderConfig import IDTR2_V0FinderToolCfg
    V0FinderTool = acc.popToolsAndMerge(IDTR2_V0FinderToolCfg(
        flags,
        name="IDTR2_V0FinderTool",
        TrackParticleCollection=TrackParticleContainer,
        V0ContainerName=IDTR2V0ContainerName,
        KshortContainerName=IDTR2KshortContainerName,
        LambdaContainerName=IDTR2LambdaContainerName,
        LambdabarContainerName=IDTR2LambdabarContainerName))
 
    from DerivationFrameworkBPhys.V0ToolConfig import BPHY_Reco_V0FinderCfg
    IDTR2_Reco_V0Finder = acc.popToolsAndMerge(BPHY_Reco_V0FinderCfg(
        flags, derivation = "IDTR2",
        V0ContainerName = IDTR2V0ContainerName,
        KshortContainerName = IDTR2KshortContainerName,
        LambdaContainerName = IDTR2LambdaContainerName,
        LambdabarContainerName = IDTR2LambdabarContainerName,
        CheckVertexContainers = ['PrimaryVertices'],
        V0FinderTool = V0FinderTool))
 
    skimmingTools = []
    augmentationTools = [IDTR2_Reco_V0Finder]
 
    if flags.Derivation.InDet.doTrackSystematics:
        from InDetTrackSystematicsTools.InDetTrackSystematicsToolsConfig import TrackSystematicsAlgCfg
        acc.merge(TrackSystematicsAlgCfg(
            flags,
            InputTrackContainer=TrackParticleContainer,
            OutputTrackContainer=(
                TrackParticleContainer+"_TRK_EFF_LARGED0_GLOBAL__1down")))
 
        V0FinderToolSyst = acc.popToolsAndMerge(IDTR2_V0FinderToolCfg(
            flags,
            name="IDTR2_V0FinderTool_Syst",
            TrackParticleCollection=(
                TrackParticleContainer+"_TRK_EFF_LARGED0_GLOBAL__1down"),
            V0ContainerName=IDTR2V0ContainerName + "Syst",
            KshortContainerName=IDTR2KshortContainerName + "Syst",
            LambdaContainerName=IDTR2LambdaContainerName + "Syst",
            LambdabarContainerName=IDTR2LambdabarContainerName + "Syst"))
 
        IDTR2_Reco_V0FinderSyst = acc.popToolsAndMerge(BPHY_Reco_V0FinderCfg(
            flags, derivation = "IDTR2", suffix = "_Syst",
            V0ContainerName = IDTR2V0ContainerName + "Syst",
            KshortContainerName = IDTR2KshortContainerName + "Syst",
            LambdaContainerName = IDTR2LambdaContainerName + "Syst",
            LambdabarContainerName = IDTR2LambdabarContainerName + "Syst",
            CheckVertexContainers = ['PrimaryVertices'],
            V0FinderTool = V0FinderToolSyst))
 
        augmentationTools += [IDTR2_Reco_V0FinderSyst]
 
    for t in augmentationTools:
        acc.addPublicTool(t)
 
    # Define the main kernel
    DerivationKernel = CompFactory.DerivationFramework.DerivationKernel
    acc.addEventAlgo(DerivationKernel("IDTR2Kernel",
                                      AugmentationTools=augmentationTools,
                                      SkimmingTools=skimmingTools))
 
    # ============================
    # Define contents of the format
    # =============================
    from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
    from xAODMetaDataCnv.InfileMetaDataConfig import SetupMetaDataForStreamCfg
    from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
 
    IDTR2SlimmingHelper = SlimmingHelper(
        "IDTR2SlimmingHelper",
        NamesAndTypes=flags.Input.TypedCollections,
        flags=flags)
 
    IDTR2SlimmingHelper.AllVariables = [
        "EventInfo",
        "PrimaryVertices",
        "InDetTrackParticles",
        "TruthParticles",
        "TruthVertices"
    ]
    if flags.Tracking.doLargeD0:
        IDTR2SlimmingHelper.AllVariables += ["InDetLargeD0TrackParticles"]
 
    StaticContent = []
    StaticContent += ["xAOD::VertexContainer#VrtSecInclusive_SecondaryVertices"]
    StaticContent += ["xAOD::VertexAuxContainer#VrtSecInclusive_SecondaryVerticesAux."]
 
    for vertexContainer in ["NewVrtSecInclusive_SecondaryVertices_DV",
                            "NewVrtSecInclusive_SecondaryVertices_Material",
                            "NewVrtSecInclusive_SecondaryVertices_Ks",
                            "IDTR2RecoV0Candidates",
                            "IDTR2RecoKshortCandidates",
                            "IDTR2RecoLambdaCandidates",
                            "IDTR2RecoLambdabarCandidates",
                            "IDTR2RecoV0CandidatesSyst",
                            "IDTR2RecoKshortCandidatesSyst",
                            "IDTR2RecoLambdaCandidatesSyst",
                            "IDTR2RecoLambdabarCandidatesSyst"]:
        StaticContent += ["xAOD::VertexContainer#%s" % vertexContainer]
        StaticContent += ["xAOD::VertexAuxContainer#%sAux.-vxTrackAtVertex" %
                          vertexContainer]
 
    IDTR2SlimmingHelper.StaticContent = StaticContent
 
    IDTR2ItemList = IDTR2SlimmingHelper.GetItemList()
    acc.merge(OutputStreamCfg(flags, "DAOD_IDTR2",
                              ItemList=IDTR2ItemList,
                              AcceptAlgs=["IDTR2Kernel"]))
    acc.merge(SetupMetaDataForStreamCfg(flags, "DAOD_IDTR2",
                                AcceptAlgs=["IDTR2Kernel"],
                                createMetadata=[MetadataCategory.CutFlowMetaData]))
 
    return acc