TRT_ConditionsAlgsConfig.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
 
 
def TRTAlignCondAlgCfg(flags, name="TRTAlignCondAlg", **kwargs):
    """Return a ComponentAccumulator for TRTAlignCondAlg algorithm"""
    from TRT_GeoModel.TRT_GeoModelConfig import TRT_GeoModelCfg
    acc = TRT_GeoModelCfg(flags)
    acc.merge(addFoldersSplitOnline(flags, "TRT", "/TRT/Onl/Calib/DX", "/TRT/Calib/DX", className="TRTCond::StrawDxContainer"))
 
    if flags.GeoModel.Align.Dynamic:
        acc.merge(addFoldersSplitOnline(flags, "TRT", "/TRT/Onl/AlignL1/TRT", "/TRT/AlignL1/TRT", className="CondAttrListCollection"))
        acc.merge(addFoldersSplitOnline(flags, "TRT", "/TRT/Onl/AlignL2", "/TRT/AlignL2", className="AlignableTransformContainer"))
        kwargs.setdefault("ReadKeyDynamicGlobal", "/TRT/AlignL1/TRT")
        kwargs.setdefault("ReadKeyDynamicRegular", "/TRT/AlignL2")
    else:
        acc.merge(addFoldersSplitOnline(flags, "TRT", "/TRT/Onl/Align", "/TRT/Align", className="AlignableTransformContainer"))
 
    kwargs.setdefault("UseDynamicFolders", flags.GeoModel.Align.Dynamic)
    kwargs.setdefault("WriteKeyAlignStore", "")  # currently unused
 
    acc.addCondAlgo(CompFactory.TRTAlignCondAlg(name, **kwargs))
    return acc
 
 
def TRTStrawStatusCondAlgCfg(flags, name="TRTStrawStatusCondAlg", **kwargs):
    """Return a ComponentAccumulator for TRTStrawStatusCondAlg algorithm"""
    acc = ComponentAccumulator()
    acc.merge(addFoldersSplitOnline(flags, "TRT",
                                    onlineFolders  = ["/TRT/Onl/Cond/Status",
                                                      "/TRT/Onl/Cond/StatusPermanent",
                                                      "/TRT/Onl/Cond/StatusHT"], # Argon straw list
                                    offlineFolders = ["/TRT/Cond/Status",
                                                      "/TRT/Cond/StatusPermanent",
                                                      "/TRT/Cond/StatusHT"], # Argon straw list
                                    className = "TRTCond::StrawStatusMultChanContainer"))
 
    acc.addCondAlgo(CompFactory.TRTStrawStatusCondAlg(name, **kwargs))
    return acc
 
 
@AccumulatorCache
def TRTStrawCondAlgCfg(flags, name="TRTStrawCondAlg", **kwargs):
    """Return a ComponentAccumulator for TRTStrawCondAlg algorithm"""
    acc = TRTAlignCondAlgCfg(flags)
    acc.merge(TRTStrawStatusCondAlgCfg(flags))
    # Alive straws algorithm
    acc.addCondAlgo(CompFactory.TRTStrawCondAlg(name, **kwargs))
    return acc
 
 
@AccumulatorCache
def TRTActiveCondAlgCfg(flags, name="TRTActiveCondAlg", **kwargs):
    """Return a ComponentAccumulator for TRTActiveCondAlg algorithm"""
    acc = TRTAlignCondAlgCfg(flags)
    from TRT_ConditionsServices.TRT_ConditionsServicesConfig import TRT_StrawStatusSummaryToolCfg
    acc.merge(addFoldersSplitOnline(flags, "TRT", "/TRT/Onl/Cond/Status", "/TRT/Cond/Status", className="TRTCond::StrawStatusMultChanContainer"))
    kwargs.setdefault("TRTStrawStatusSummaryTool", acc.popToolsAndMerge(TRT_StrawStatusSummaryToolCfg(flags)))
    acc.addCondAlgo(CompFactory.TRTActiveCondAlg(name, **kwargs))
    return acc
 
 
def TRTPhaseCondCfg(flags, name="TRTPhaseCondAlg", **kwargs):
    """Return a ComponentAccumulator for TRTPhaseCondAlg algorithm"""
    acc = ComponentAccumulator()
    from TRT_ConditionsServices.TRT_ConditionsServicesConfig import TRT_CalDbToolCfg
    CalDbTool = acc.popToolsAndMerge(TRT_CalDbToolCfg(flags))
    acc.addPublicTool(CalDbTool)  # public as it is has many clients to save some memory
    kwargs.setdefault("TRTCalDbTool", CalDbTool)
    # Average T0 CondAlg
    acc.merge(addFoldersSplitOnline(flags, "TRT", "/TRT/Onl/Calib/T0", "/TRT/Calib/T0",
                                    className='TRTCond::StrawT0MultChanContainer'))
    acc.addCondAlgo(CompFactory.TRTPhaseCondAlg(name, **kwargs))
    return acc
 
 
def TRTToTCondAlgCfg(flags, name="TRTToTCondAlg", **kwargs):
    """Return a ComponentAccumulator for TRTToTCondAlg algorithm"""
    acc = ComponentAccumulator()
    acc.merge(addFoldersSplitOnline(flags, "TRT", "/TRT/Onl/Calib/ToT/ToTVectors", "/TRT/Calib/ToT/ToTVectors", className="CondAttrListVec"))
    acc.merge(addFoldersSplitOnline(flags, "TRT", "/TRT/Onl/Calib/ToT/ToTValue", "/TRT/Calib/ToT/ToTValue", className="CondAttrListCollection"))
 
    kwargs.setdefault("ToTWriteKey", "Dedxcorrection")
 
    acc.addCondAlgo(CompFactory.TRTToTCondAlg(name, **kwargs))
    return acc
 
 
def TRTHTCondAlgCfg(flags, name="TRTHTCondAlg", **kwargs):
    """Return a ComponentAccumulator for TRTHTCondAlg algorithm"""
    acc = ComponentAccumulator()
    acc.merge(addFoldersSplitOnline(flags, "TRT", "/TRT/Onl/Calib/PID_vector", "/TRT/Calib/PID_vector", className="CondAttrListVec"))
    kwargs.setdefault("HTWriteKey", "HTcalculator")
    acc.addCondAlgo(CompFactory.TRTHTCondAlg(name, **kwargs))
    return acc
 
 
def TRTPIDNNCondAlgCfg(flags, name="TRTPIDNNCondAlg", **kwargs):
    """Return a ComponentAccumulator for TRTHTCondAlg algorithm"""
    acc = ComponentAccumulator()
    acc.merge(addFoldersSplitOnline(flags, "TRT", "/TRT/Onl/Calib/PID_NN", "/TRT/Calib/PID_NN", className="CondAttrListCollection"))
 
    from AthenaConfiguration.Enums import LHCPeriod
    if flags.GeoModel.Run is LHCPeriod.Run1:
        from IOVDbSvc.IOVDbSvcConfig import addOverride
        acc.merge(addOverride(flags,"/TRT/Calib/PID_NN", "TRTCalibPID_NN_v1"))
        acc.merge(addOverride(flags,"/TRT/Onl/Calib/PID_NN", "TRTCalibPID_NN_v1"))
 
    kwargs.setdefault("TRTPIDNNWriteKey", "TRTPIDNN")
    acc.addCondAlgo(CompFactory.TRTPIDNNCondAlg(name, **kwargs))
    return acc
 
 
def TRTCondWriteCfg(flags, name="TRTCondWriteAlg", **kwargs):
    
    # This needs more implementations. Skeleton for now - @serodrig
    if "CalibInputFile" not in kwargs:
        kwargs.setdefault("CalibInputFile", "dummy.txt")
    
    acc = ComponentAccumulator()
    acc.addCondAlgo(CompFactory.TRTCondWrite(name, **kwargs))
    return acc