MuonRIO_OnTrackCreatorToolConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.Enums import BeamType
from MuonConfig.MuonCalibrationConfig import MdtCalibrationToolCfg
 

 
__mdtCalibWindows = ['User',
                     'Default', 
                     'Collision_G4', 
                     'Collision_data', 
                     'Collision_t0fit', 
                     ]
 
def MdtCalibWindowNumber(name):
    """Returns index number corresponding to the calibration window name.
    This will throw a ValueError if name is not in the list.
    """
    return __mdtCalibWindows.index(name)
 
def TriggerChamberClusterOnTrackCreatorCfg(flags, name="TriggerChamberClusterOnTrackCreator", **kwargs):
    result=ComponentAccumulator()
    acc =  MuonClusterOnTrackCreatorCfg(flags)
    muon_cluster_creator=acc.getPrimary()
    result.merge(acc)
    kwargs.setdefault("ClusterCreator", muon_cluster_creator)
    result.setPrivateTools(CompFactory.Muon.TriggerChamberClusterOnTrackCreator(name, **kwargs))
    return result
 
def CscClusterOnTrackCreatorCfg(flags,name="CscClusterOnTrackCreator", **kwargs):
    from MuonConfig.MuonSegmentFindingConfig import QratCscClusterFitterCfg, CscClusterUtilToolCfg, CalibCscStripFitterCfg
 
    result=ComponentAccumulator()    
    acc = QratCscClusterFitterCfg(flags)
    qrat = acc.popPrivateTools()
    result.merge(acc)
    kwargs.setdefault("CscClusterFitter", qrat )
    
    acc = CalibCscStripFitterCfg(flags)
    strip_fitter = acc.popPrivateTools()
    result.merge(acc)
    kwargs.setdefault("CscStripFitter", strip_fitter)
    
    acc = CscClusterUtilToolCfg(flags)
    cluster_util_tool = acc.popPrivateTools()
    kwargs.setdefault("CscClusterUtilTool", cluster_util_tool )
    result.merge(acc)
    
    if not flags.Input.isMC: # collisions real data or simulated first data
        # scale CSC and hit errors 
        kwargs.setdefault("ErrorScalerBeta", 0.070 )
 
    result.setPrivateTools(CompFactory.Muon.CscClusterOnTrackCreator(name,**kwargs))
    
    return result
 
 
def MdtCalibToolForRotsCfg(flags, name ="MdtCalibrationTool", **kwargs):
    kwargs.setdefault("DoMagneticFieldCorrection", flags.Muon.Calib.correctMdtRtForBField)
 
    kwargs.setdefault("DoSlewingCorrection", flags.Muon.Calib.correctMdtRtForTimeSlewing)
    if flags.Beam.Type in [BeamType.Cosmics, BeamType.SingleBeam]:
        kwargs.setdefault("DoTofCorrection", False)
    else:
        kwargs.setdefault("DoTofCorrection", True)
        kwargs.setdefault("TimeWindowSetting", MdtCalibWindowNumber('Collision_data'))
    return MdtCalibrationToolCfg(flags, name = name, **kwargs)
 
def MdtDriftCircleOnTrackCreatorCfg(flags,name="MdtDriftCircleOnTrackCreator", **kwargs):
    result = ComponentAccumulator()
    if flags.Beam.Type in [BeamType.Cosmics, BeamType.SingleBeam]:
        kwargs.setdefault("DoFixedError", True)
        kwargs.setdefault("TimingMode", 1)
        kwargs.setdefault("UseParametrisedError", True)
        kwargs.setdefault("ApplyToF", False)
 
    else: # collisions simulation/data settings
        kwargs.setdefault("UseParametrisedError", False)
        kwargs.setdefault("DoFixedError", False)
        kwargs.setdefault("DoErrorScaling", False)
 
        if not flags.Input.isMC : 
            kwargs.setdefault("CreateTubeHit", True)  # BroadErrors
            kwargs.setdefault("UseLooseErrors", flags.Muon.useLooseErrorTuning)  # LooseErrors on data                          
 
    kwargs.setdefault("CalibrationTool", result.popToolsAndMerge( MdtCalibToolForRotsCfg(flags)) )
    kwargs.setdefault("IsMC", flags.Input.isMC)
 
    result.setPrivateTools(CompFactory.Muon.MdtDriftCircleOnTrackCreator(name, WasConfigured=True, **kwargs))
    return result
    
def MuonClusterOnTrackCreatorCfg(flags, name="MuonClusterOnTrackCreator", **kwargs):
    result=ComponentAccumulator()
    if not flags.Input.isMC: # collisions real data or simulated first data
        # scale TGC eta hit errors as long as TGC eta are not well aligned
        kwargs.setdefault("DoFixedErrorTgcEta", True)
        kwargs.setdefault("FixedErrorTgcEta", 15.)
    else:
        kwargs.setdefault("DoFixedErrorTgcEta", False) # This is ONLY to make the tool configured. Real solution is to use default name.
 
    if flags.Detector.EnablesTGC or flags.Detector.EnableMM:
        from MuonConfig.MuonCalibrationConfig import NSWCalibToolCfg
        kwargs.setdefault("NSWCalibTool", result.popToolsAndMerge(NSWCalibToolCfg(flags)))
 
        from MuonConfig.MuonConfigFlags import MMClusterBuilderEnum
        if flags.Muon.MMClusterCalibRecoTool == MMClusterBuilderEnum.Centroid:
            from MuonConfig.MuonRecToolsConfig import SimpleMMClusterBuilderToolCfg
            kwargs.setdefault("MMClusterBuilder", result.popToolsAndMerge(SimpleMMClusterBuilderToolCfg(flags))) 
        elif flags.Muon.MMClusterCalibRecoTool == MMClusterBuilderEnum.ClusterTimeProjection:
            #Turn on the calibrated drift velocities only for data (by default 2022 is also switched to centroid!)
            if not flags.Input.isMC:
                kwargs["NSWCalibTool"].calibrateDriftVelocityFromData = True
 
            from MuonConfig.MuonRecToolsConfig import  ClusterTimeProjectionMMClusterBuilderToolCfg
            kwargs.setdefault("MMClusterBuilder", result.popToolsAndMerge(ClusterTimeProjectionMMClusterBuilderToolCfg(flags))) 
            
    muon_cluster_rot_creator = CompFactory.Muon.MuonClusterOnTrackCreator(name, **kwargs)
    result.setPrivateTools(muon_cluster_rot_creator)
    return result