SiTrackMakerConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
# Configuration of SiTrackMakerTool_xk package
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.Enums import BeamType
 
 
def SiTrackMaker_xkCfg(flags, name="InDetSiTrackMaker", **kwargs):
    # To produce BeamSpotData and AtlasFieldCacheCondObj
    from BeamSpotConditions.BeamSpotConditionsConfig import BeamSpotCondAlgCfg
    from MagFieldServices.MagFieldServicesConfig import (
        AtlasFieldCacheCondAlgCfg)
    acc = BeamSpotCondAlgCfg(flags)
    acc.merge(AtlasFieldCacheCondAlgCfg(flags))
 
    if "RoadTool" not in kwargs:
        from InDetConfig.SiDetElementsRoadToolConfig import (
            SiDetElementsRoadMaker_xkCfg)
        kwargs.setdefault("RoadTool", acc.popToolsAndMerge(
            SiDetElementsRoadMaker_xkCfg(flags)))
 
    if "CombinatorialTrackFinder" not in kwargs:
        from InDetConfig.SiCombinatorialTrackFinderToolConfig import (
            SiCombinatorialTrackFinder_xkCfg)
        kwargs.setdefault("CombinatorialTrackFinder", acc.popToolsAndMerge(
            SiCombinatorialTrackFinder_xkCfg(flags)))
 
    kwargs.setdefault("useSCT",    flags.Tracking.ActiveConfig.useSCT)
    kwargs.setdefault("usePixel",  flags.Tracking.ActiveConfig.usePixel)
 
    kwargs.setdefault("pTmin",     flags.Tracking.ActiveConfig.minPT)
    kwargs.setdefault("pTminBrem", flags.Tracking.ActiveConfig.minPTBrem)
    kwargs.setdefault("nClustersMin", flags.Tracking.ActiveConfig.minClusters)
    kwargs.setdefault("nHolesMax", flags.Tracking.ActiveConfig.nHolesMax)
    kwargs.setdefault("nHolesGapMax", flags.Tracking.ActiveConfig.nHolesGapMax)
    kwargs.setdefault("UseSeedFilter",
                      flags.Tracking.ActiveConfig.useSeedFilter)
    kwargs.setdefault("Xi2max", flags.Tracking.ActiveConfig.Xi2max)
    kwargs.setdefault("Xi2maxNoAdd", flags.Tracking.ActiveConfig.Xi2maxNoAdd)
    kwargs.setdefault("nWeightedClustersMin",
                      flags.Tracking.ActiveConfig.nWeightedClustersMin)
 
    kwargs.setdefault("CosmicTrack", flags.Beam.Type is BeamType.Cosmics)
    kwargs.setdefault("Xi2maxMultiTracks", flags.Tracking.ActiveConfig.Xi2max)
    kwargs.setdefault("useSSSseedsFilter", True)
    kwargs.setdefault("doMultiTracksProd", True)
 
    kwargs.setdefault("useBremModel",
                      flags.Tracking.ActiveConfig.doBremRecoverySi)
    kwargs.setdefault("doCaloSeededBrem", flags.Tracking.doCaloSeededBrem)
 
    if kwargs["useBremModel"] and kwargs["doCaloSeededBrem"]:
        from InDetConfig.InDetCaloClusterROISelectorConfig import (
            CaloClusterROIPhiRZContainerMakerCfg)
        acc.merge(CaloClusterROIPhiRZContainerMakerCfg(flags))
 
    kwargs.setdefault("doHadCaloSeedSSS", flags.Tracking.doHadCaloSeededSSS)
 
    if kwargs["doHadCaloSeedSSS"]:
        from InDetConfig.InDetCaloClusterROISelectorConfig import (
            HadCaloClusterROIPhiRZContainerMakerCfg)
        acc.merge(HadCaloClusterROIPhiRZContainerMakerCfg(flags))
 
    kwargs.setdefault("phiWidth", flags.Tracking.phiWidthBrem)
    kwargs.setdefault("etaWidth", flags.Tracking.etaWidthBrem)
    kwargs.setdefault("EMROIPhiRZContainer", "InDetCaloClusterROIPhiRZ0GeV")
    kwargs.setdefault("HadROIPhiRZContainer", "InDetHadCaloClusterROIPhiRZ")
    kwargs.setdefault("UseAssociationTool",
                      flags.Tracking.ActiveConfig.usePrdAssociationTool)
 
    if flags.Beam.Type is BeamType.Cosmics:
        kwargs.setdefault("TrackPatternRecoInfo",
                          'SiSpacePointsSeedMaker_Cosmic')
 
    elif flags.Tracking.ActiveConfig.extension == "LowPt":
        kwargs.setdefault("TrackPatternRecoInfo",
                          'SiSpacePointsSeedMaker_LowMomentum')
 
    elif (flags.Tracking.ActiveConfig.extension == "VeryLowPt" or
          (flags.Tracking.ActiveConfig.extension == "Pixel" and
           flags.Tracking.doMinBias)):
        kwargs.setdefault("TrackPatternRecoInfo",
                          'SiSpacePointsSeedMaker_VeryLowMomentum')
 
    elif flags.Tracking.ActiveConfig.extension == "HeavyIon":
        kwargs.setdefault("TrackPatternRecoInfo",
                          'SiSpacePointsSeedMaker_HeavyIon')
 
    elif flags.Tracking.ActiveConfig.extension == "BeamGas":
        kwargs.setdefault("TrackPatternRecoInfo",
                          'SiSpacePointsSeedMaker_BeamGas')
 
    elif flags.Tracking.ActiveConfig.extension == "Forward":
        kwargs.setdefault("TrackPatternRecoInfo",
                          'SiSpacePointsSeedMaker_ForwardTracks')
 
    elif "LargeD0" in flags.Tracking.ActiveConfig.extension:
        kwargs.setdefault("TrackPatternRecoInfo",
                          'SiSpacePointsSeedMaker_LargeD0')
 
    else:
        kwargs.setdefault("TrackPatternRecoInfo", 'SiSPSeededFinder')
 
    # Enable storing seeds if requested.
    # Always for primary pass. Only if appropriate flag is set for other passes.
    if (flags.Tracking.doStoreTrackSeeds and 
            (flags.Tracking.ActiveConfig.extension == "" or flags.Tracking.ActiveConfig.storeTrackSeeds) and
            "SeedToTrackConversion" not in kwargs):
        from InDetConfig.SeedToTrackConversionToolConfig import (
            SeedToTrackConversionToolCfg)
        kwargs.setdefault("SeedToTrackConversion", acc.popToolsAndMerge(
            SeedToTrackConversionToolCfg(flags)))
        kwargs.setdefault("SeedSegmentsWrite", True)
 
    acc.setPrivateTools(CompFactory.InDet.SiTrackMaker_xk(
        name+flags.Tracking.ActiveConfig.extension, **kwargs))
    return acc
 
 
def TrigSiTrackMaker_xkCfg(flags, name="TrigSiTrackMaker", **kwargs):
    # To produce BeamSpotData and AtlasFieldCacheCondObj
    from BeamSpotConditions.BeamSpotConditionsConfig import BeamSpotCondAlgCfg
    from MagFieldServices.MagFieldServicesConfig import (
        AtlasFieldCacheCondAlgCfg)
    acc = BeamSpotCondAlgCfg(flags)
    acc.merge(AtlasFieldCacheCondAlgCfg(flags))
 
    if "RoadTool" not in kwargs:
        from InDetConfig.SiDetElementsRoadToolConfig import (
            SiDetElementsRoadMaker_xkCfg)
        kwargs.setdefault("RoadTool", acc.popToolsAndMerge(
            SiDetElementsRoadMaker_xkCfg(flags)))
 
    if "CombinatorialTrackFinder" not in kwargs:
        from InDetConfig.SiCombinatorialTrackFinderToolConfig import (
            SiCombinatorialTrackFinder_xk_Trig_Cfg)
        kwargs.setdefault("CombinatorialTrackFinder", acc.popToolsAndMerge(
            SiCombinatorialTrackFinder_xk_Trig_Cfg(flags)))
 
    kwargs.setdefault("pTmin", flags.Tracking.ActiveConfig.minPT)
    kwargs.setdefault("nClustersMin", flags.Tracking.ActiveConfig.nClustersMin)
    kwargs.setdefault("nHolesMax", flags.Tracking.ActiveConfig.nHolesMax)
    kwargs.setdefault("nHolesGapMax",
                      flags.Tracking.ActiveConfig.nHolesGapMax)
    kwargs.setdefault("UseSeedFilter",
                      flags.Tracking.ActiveConfig.useSeedFilter)
    kwargs.setdefault("Xi2max", flags.Tracking.ActiveConfig.Xi2max)
    kwargs.setdefault("Xi2maxNoAdd", flags.Tracking.ActiveConfig.Xi2maxNoAdd)
    kwargs.setdefault("nWeightedClustersMin",
                      flags.Tracking.ActiveConfig.nWeightedClustersMin)
    kwargs.setdefault("Xi2maxMultiTracks", flags.Tracking.ActiveConfig.Xi2max)
    kwargs.setdefault("UseAssociationTool", False)
 
    kwargs.setdefault("useBremModel",
                      flags.Tracking.ActiveConfig.doBremRecoverySi)
 
    kwargs.setdefault("CosmicTrack",
                      flags.Tracking.ActiveConfig.input_name == "cosmics")
 
    acc.setPrivateTools(CompFactory.InDet.SiTrackMaker_xk(name, **kwargs))
    return acc
 
 
def ITkSiTrackMaker_xkCfg(flags, name="ITkSiTrackMaker", **kwargs):
    # To produce BeamSpotData and AtlasFieldCacheCondObj
    from BeamSpotConditions.BeamSpotConditionsConfig import BeamSpotCondAlgCfg
    from MagFieldServices.MagFieldServicesConfig import (
        AtlasFieldCacheCondAlgCfg)
    acc = BeamSpotCondAlgCfg(flags)
    acc.merge(AtlasFieldCacheCondAlgCfg(flags))
 
    if "RoadTool" not in kwargs:
        from InDetConfig.SiDetElementsRoadToolConfig import (
            ITkSiDetElementsRoadMaker_xkCfg)
        kwargs.setdefault("RoadTool", acc.popToolsAndMerge(
            ITkSiDetElementsRoadMaker_xkCfg(flags)))
 
    if "CombinatorialTrackFinder" not in kwargs:
        from InDetConfig.SiCombinatorialTrackFinderToolConfig import (
            ITkSiCombinatorialTrackFinder_xkCfg)
        kwargs.setdefault("CombinatorialTrackFinder", acc.popToolsAndMerge(
            ITkSiCombinatorialTrackFinder_xkCfg(flags)))
 
    kwargs.setdefault("useSCT", flags.Tracking.ActiveConfig.useITkStrip)
    kwargs.setdefault("usePixel", flags.Tracking.ActiveConfig.useITkPixel)
    kwargs.setdefault("etaBins", flags.Tracking.ActiveConfig.etaBins)
    kwargs.setdefault("pTBins", flags.Tracking.ActiveConfig.minPT)
    kwargs.setdefault("pTmin", flags.Tracking.ActiveConfig.minPT[0])
    kwargs.setdefault("pTminBrem", flags.Tracking.ActiveConfig.minPTBrem[0])
    kwargs.setdefault("nClustersMin",
                      min(flags.Tracking.ActiveConfig.minClusters))
    kwargs.setdefault("nHolesMax", flags.Tracking.ActiveConfig.nHolesMax[0])
    kwargs.setdefault("nHolesGapMax",
                      flags.Tracking.ActiveConfig.nHolesGapMax[0])
    kwargs.setdefault("UseSeedFilter",
                      flags.Tracking.ActiveConfig.useSeedFilter)
    kwargs.setdefault("Xi2max", flags.Tracking.ActiveConfig.Xi2max[0])
    kwargs.setdefault("Xi2maxNoAdd", flags.Tracking.ActiveConfig.Xi2maxNoAdd[0])
    kwargs.setdefault("nWeightedClustersMin",
                      flags.Tracking.ActiveConfig.nWeightedClustersMin[0])
    kwargs.setdefault("CosmicTrack", flags.Beam.Type is BeamType.Cosmics)
    kwargs.setdefault("Xi2maxMultiTracks",
                      flags.Tracking.ActiveConfig.Xi2max[0])
    kwargs.setdefault("doMultiTracksProd", True)
 
    # Disabled for second passes in reco
    kwargs.setdefault("useBremModel",
                      flags.Tracking.ActiveConfig.doBremRecoverySi)
    kwargs.setdefault("doCaloSeededBrem", flags.Tracking.doCaloSeededBrem)
 
    if kwargs["useBremModel"] and kwargs["doCaloSeededBrem"]:
        from InDetConfig.InDetCaloClusterROISelectorConfig import (
            ITkCaloClusterROIPhiRZContainerMakerCfg)
        acc.merge(ITkCaloClusterROIPhiRZContainerMakerCfg(flags))
 
    kwargs.setdefault("doHadCaloSeedSSS", flags.Tracking.doHadCaloSeededSSS)
 
    if kwargs["doHadCaloSeedSSS"]:
        from InDetConfig.InDetCaloClusterROISelectorConfig import (
            ITkHadCaloClusterROIPhiRZContainerMakerCfg)
        acc.merge(ITkHadCaloClusterROIPhiRZContainerMakerCfg(flags))
 
    kwargs.setdefault("phiWidth", flags.Tracking.phiWidthBrem)
    kwargs.setdefault("etaWidth", flags.Tracking.etaWidthBrem)
    kwargs.setdefault("EMROIPhiRZContainer", "ITkCaloClusterROIPhiRZ0GeV")
    kwargs.setdefault("HadROIPhiRZContainer", "ITkHadCaloClusterROIPhiRZ")
    kwargs.setdefault("UseAssociationTool",
                      flags.Tracking.ActiveConfig.usePrdAssociationTool)
    kwargs.setdefault("ITKGeometry", True)
 
    if flags.Beam.Type is BeamType.Cosmics:
        kwargs.setdefault("TrackPatternRecoInfo",
                          'SiSpacePointsSeedMaker_Cosmic')
 
    elif flags.Tracking.ActiveConfig.extension == "Conversion":
        kwargs.setdefault("TrackPatternRecoInfo",
                          'SiSpacePointsSeedMaker_ITkConversionTracks')
 
    elif flags.Tracking.ActiveConfig.extension == "LargeD0":
        kwargs.setdefault("TrackPatternRecoInfo",
                          'SiSpacePointsSeedMaker_LargeD0')
 
    elif flags.Tracking.ActiveConfig.extension == "LowPt":
        kwargs.setdefault("TrackPatternRecoInfo",
                          'SiSpacePointsSeedMaker_LowMomentum')
 
    else:
        kwargs.setdefault("TrackPatternRecoInfo", 'SiSPSeededFinder')
 
    if (flags.Tracking.doStoreTrackSeeds and
            "SeedToTrackConversion" not in kwargs):
        from InDetConfig.SeedToTrackConversionToolConfig import (
            ITkSeedToTrackConversionToolCfg)
        kwargs.setdefault("SeedToTrackConversion", acc.popToolsAndMerge(
            ITkSeedToTrackConversionToolCfg(flags)))
        kwargs.setdefault("SeedSegmentsWrite", True)
 
    acc.setPrivateTools(CompFactory.InDet.SiTrackMaker_xk(
        name+flags.Tracking.ActiveConfig.extension, **kwargs))
    return acc