SCT_ConditionsAlgorithmsConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
from AthenaConfiguration.AccumulatorCache import AccumulatorCache
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
from IOVDbSvc.IOVDbSvcConfig import addFoldersSplitOnline
 
 
@AccumulatorCache
def SCT_AlignCondAlgCfg(flags, name="SCT_AlignCondAlg", **kwargs):
    """Return a configured SCT_AlignCondAlg"""
    from SCT_GeoModel.SCT_GeoModelConfig import SCT_GeoModelCfg
    acc = SCT_GeoModelCfg(flags)
    if flags.GeoModel.Align.Dynamic:
        acc.merge(addFoldersSplitOnline(flags, "INDET",
                                        ["/Indet/Onl/AlignL1/ID", "/Indet/Onl/AlignL2/SCT"],
                                        ["/Indet/AlignL1/ID", "/Indet/AlignL2/SCT"],
                                        className="CondAttrListCollection"))
        acc.merge(addFoldersSplitOnline(flags, "INDET", "/Indet/Onl/AlignL3", "/Indet/AlignL3", className="AlignableTransformContainer"))
    else:
        acc.merge(addFoldersSplitOnline(flags, "INDET", "/Indet/Onl/Align", "/Indet/Align", className="AlignableTransformContainer"))
 
    kwargs.setdefault("UseDynamicAlignFolders", flags.GeoModel.Align.Dynamic)
 
    acc.addCondAlgo(CompFactory.SCT_AlignCondAlg(name, **kwargs))
    return acc
 
 
def getSCTDAQConfigFolder(flags) :
    if flags.InDet.ForceCoolVectorPayload and flags.InDet.ForceCoraCool:
        raise Exception("SCT DB CONFIGURATION FLAG CONFLICT: Both CVP and CoraCool selected")  
    return "/SCT/DAQ/Config/" if (flags.InDet.ForceCoolVectorPayload
                                  or (flags.IOVDb.DatabaseInstance != "COMP200"
                                  and not flags.InDet.ForceCoraCool)) else "/SCT/DAQ/Configuration/"
 
 
def SCT_ConfigurationCondAlgCfg(flags, name="SCT_ConfigurationCondAlg", **kwargs):
    acc = ComponentAccumulator()
    config_folder_prefix = getSCTDAQConfigFolder(flags)
    channelFolder = config_folder_prefix + ("Chip" if flags.IOVDb.DatabaseInstance == "COMP200" else "ChipSlim")
    kwargs.setdefault("ReadKeyChannel", channelFolder)
    kwargs.setdefault("ReadKeyModule", f"{config_folder_prefix}Module")
    kwargs.setdefault("ReadKeyMur", f"{config_folder_prefix}MUR")
 
    acc.merge(addFoldersSplitOnline(flags,
                                    detDb="SCT",
                                    onlineFolders=[channelFolder,
                                                   f"{config_folder_prefix}Module",
                                                   f"{config_folder_prefix}MUR"],
                                    offlineFolders=[channelFolder,
                                                    f"{config_folder_prefix}Module",
                                                    f"{config_folder_prefix}MUR"],
                                    className="CondAttrListVec",
                                    splitMC=True))
 
    from SCT_Cabling.SCT_CablingConfig import SCT_CablingToolCfg
    kwargs.setdefault("SCT_CablingTool", acc.popToolsAndMerge(SCT_CablingToolCfg(flags)))
 
    from SCT_ConditionsTools.SCT_ConditionsToolsConfig import SCT_ReadoutToolCfg
    kwargs.setdefault("SCT_ReadoutTool", acc.popToolsAndMerge(SCT_ReadoutToolCfg(flags)))
 
    acc.addCondAlgo(CompFactory.SCT_ConfigurationCondAlg(name, **kwargs))
    return acc
 
 
@AccumulatorCache
def SCT_DetectorElementCondAlgCfg(flags, name="SCT_DetectorElementCondAlg", **kwargs):
    acc = SCT_AlignCondAlgCfg(flags)
 
    # FIXME
    # add artifical dependencies to SCT, TRT and Muon
    # conditions algs to ensure that the IOV
    # is identical to the IOV of the tracking geometry
    if flags.Detector.GeometryMuon and flags.Muon.enableAlignment:
        from MuonConfig.MuonGeometryConfig import MuonGeoModelCfg
        acc.merge(MuonGeoModelCfg(flags))
        kwargs.setdefault("MuonManagerKey", "MuonDetectorManager")
    if flags.Detector.GeometryTRT:
        from TRT_GeoModel.TRT_GeoModelConfig import TRT_ReadoutGeometryCfg
        acc.merge(TRT_ReadoutGeometryCfg(flags))
        kwargs.setdefault("TRT_DetEltContKey", "TRT_DetElementContainer")
    if not flags.GeoModel.Align.LegacyConditionsAccess and flags.Detector.GeometryPixel:
        from PixelGeoModel.PixelGeoModelConfig import PixelAlignmentCfg
        acc.merge(PixelAlignmentCfg(flags))
        kwargs.setdefault("PixelAlignmentStore", "PixelAlignmentStore")
    # end of hack
 
    acc.addCondAlgo(CompFactory.SCT_DetectorElementCondAlg(name, **kwargs))
    return acc
 
# SCTDetectorElementStatusAlg which creates the status data to be used in the SCT_Clusterization
def SCT_DetectorElementStatusCondAlgCfg(flags, name="SCTDetectorElementStatusCondAlg",**kwargs) :
    acc = ComponentAccumulator()
    if 'ConditionsSummaryTool' not in kwargs :
        from SCT_ConditionsTools.SCT_ConditionsToolsConfig import SCT_ConditionsSummaryToolCfg
        kwargs.setdefault("ConditionsSummaryTool", acc.popToolsAndMerge(SCT_ConditionsSummaryToolCfg(flags, withFlaggedCondTool=False, withByteStreamErrorsTool=False)) )
    kwargs.setdefault("WriteKey", "SCTDetectorElementStatusCondData")
 
    # not a conditions algorithm since it combines conditions data and data from the bytestream
    acc.addCondAlgo( CompFactory.InDet.SiDetectorElementStatusCondAlg(name, **kwargs) )
    return acc
 
def SCT_DetectorElementStatusAlgWithoutFlaggedCfg(flags, name="SCTDetectorElementStatusAlgWithoutFlagged",**kwargs) :
    '''
    Algorithm which just creates event data from conditions data.
    '''
    acc = SCT_DetectorElementStatusCondAlgCfg(flags)
    if 'ConditionsSummaryTool' not in kwargs :
        from SCT_ConditionsTools.SCT_ConditionsToolsConfig import SCT_DetectorElementStatusAddByteStreamErrorsToolCfg
        kwargs.setdefault("ConditionsSummaryTool", acc.popToolsAndMerge(SCT_DetectorElementStatusAddByteStreamErrorsToolCfg(
                                                                           flags,
                                                                           SCTDetElStatusCondDataBaseKey  = "SCTDetectorElementStatusCondData",
                                                                           SCTDetElStatusEventDataBaseKey = "")) )
    kwargs.setdefault("WriteKey",                      "SCTDetectorElementStatusWithoutFlagged")
 
    # not a conditions algorithm since it combines conditions data and data from the bytestream
    acc.addEventAlgo( CompFactory.InDet.SiDetectorElementStatusAlg(name, **kwargs) )
    return acc
 
# SCTDetectorElementStatusAlg which creates the status data to be used everywhere but the SCT_Clusterization
def SCT_DetectorElementStatusAlgCfg(flags, name = "SCTDetectorElementStatusAlg", **kwargs) :
    acc = SCT_DetectorElementStatusAlgWithoutFlaggedCfg(flags)
    if 'ConditionsSummaryTool' not in kwargs :
        from SCT_ConditionsTools.SCT_ConditionsToolsConfig import SCT_DetectorElementStatusAddFlaggedToolCfg
        kwargs.setdefault("ConditionsSummaryTool", acc.popToolsAndMerge(SCT_DetectorElementStatusAddFlaggedToolCfg(
                                                                           flags,
                                                                           SCTDetElStatusCondDataBaseKey  = "",
                                                                           SCTDetElStatusEventDataBaseKey = "SCTDetectorElementStatusWithoutFlagged")) )
 
    kwargs.setdefault("WriteKey",                      "SCTDetectorElementStatus")
    acc.merge( SCT_DetectorElementStatusAlgWithoutFlaggedCfg(flags, name, **kwargs) )
    return acc
 
if __name__=="__main__":
    from AthenaCommon.Logging import log
    from AthenaCommon.Constants import INFO
    log.setLevel(INFO)
 
    algorithm = ""
 
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    flags = initConfigFlags()
    flags.Input.isMC = False
    flags.Input.Files = ["./myESD.pool.root"]
    flags.Input.ProjectName = "data17_13TeV" # q431 input
    flags.Input.RunNumbers = [330470] # q431 input
    flags.IOVDb.GlobalTag = "CONDBR2-BLKPA-2018-03" # q431 setup
    from AthenaConfiguration.TestDefaults import defaultGeometryTags
    flags.GeoModel.AtlasVersion = defaultGeometryTags.RUN2
    flags.lock()
 
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg
    cfg = MainServicesCfg(flags)
 
    from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
    cfg.merge(PoolReadCfg(flags))
 
    algkwargs = {}
    algkwargs["OutputLevel"] = INFO
    if algorithm == "AlignCondAlg":
        cfg.merge(SCT_AlignCondAlgCfg(flags, **algkwargs))
    elif algorithm == "ConfigurationCondAlg":
        cfg.merge(SCT_ConfigurationCondAlgCfg(flags, **algkwargs))
    elif algorithm == "DetectorElementCondAlg":
        cfg.merge(SCT_DetectorElementCondAlgCfg(flags, **algkwargs))
 
    cfg.run(maxEvents=25)