EMTrackMatchBuilderConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
""" Instantiate EMTrackMatchBuilder with default configuration """
 
from AthenaCommon.Logging import logging
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.Enums import BeamType
from egammaTrackTools.egammaTrackToolsConfig import (
    EMExtrapolationToolsCfg)
 
 
def EMTrackMatchBuilderCfg(flags, name='EMTrackMatchBuilder', **kwargs):
 
    mlog = logging.getLogger(name)
    mlog.debug('Start configuration')
 
    acc = ComponentAccumulator()
 
    if "ExtrapolationTool" not in kwargs:
        extrap = EMExtrapolationToolsCfg(flags)
        kwargs["ExtrapolationTool"] = acc.popToolsAndMerge(extrap)
 
    if "TrackParticlesName" not in kwargs:
      kwargs.setdefault("TrackParticlesName",
                        flags.Egamma.Keys.Output.GSFTrackParticles)
 
    # candidate match is done in 2 times this  so +- 0.2
    kwargs.setdefault("broadDeltaEta",      0.1)
    # candidate match is done in 2 times this  so +- 0.3
    kwargs.setdefault("broadDeltaPhi",      0.15)
    kwargs.setdefault("useCandidateMatch",  True)
    kwargs.setdefault("useScoring",         True)
    kwargs.setdefault("SecondPassRescale",  True)
    kwargs.setdefault("UseRescaleMetric",   True)
    kwargs.setdefault("isCosmics",          flags.Beam.Type is BeamType.Cosmics)
 
    tool = CompFactory.EMTrackMatchBuilder(name, **kwargs)
 
    acc.setPrivateTools(tool)
    return acc
 
 
if __name__ == "__main__":
 
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    from AthenaConfiguration.ComponentAccumulator import printProperties
    from AthenaConfiguration.TestDefaults import defaultTestFiles
    flags = initConfigFlags()
    flags.Input.Files = defaultTestFiles.RDO_RUN2
    flags.fillFromArgs()
    flags.lock()
    flags.dump()
 
    cfg = ComponentAccumulator()
    mlog = logging.getLogger("EMTrackMatchBuilderConfigTest")
    mlog.info("Configuring  EMTrackMatchBuilder: ")
    printProperties(mlog, cfg.popToolsAndMerge(
        EMTrackMatchBuilderCfg(flags)),
        nestLevel=1,
        printDefaults=True)
 
    f = open("emtrackmatchbuilder.pkl", "wb")
    cfg.store(f)
    f.close()