CollectionMergerConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaCommon.Logging import logging
from AthenaConfiguration.AccumulatorCache import AccumulatorCache
 
 
@AccumulatorCache
def ISFCollectionMergerCfg(flags,name="ISF_CollectionMerger", **kwargs):
   
    hardscatterSG=""
    from AthenaConfiguration.Enums import Project
    if flags.Sim.DoFullChain:
        if flags.Common.Project is Project.Athena:
            if flags.Digitization.PileUp:
                hardscatterSG = "OriginalEvent_SG+"
        else:
            msg = logging.getLogger(name)
            msg.warning("Fast Chain running only supported in the Athena project.")
    if flags.Detector.EnableBCM:
        kwargs.setdefault( "OutputBCMHits",             hardscatterSG+"BCMHits"             )
        kwargs.setdefault( "OutputBLMHits",             hardscatterSG+"BLMHits"             )
    if flags.Detector.EnablePixel:
        kwargs.setdefault( "OutputPixelHits",           hardscatterSG+"PixelHits"           )
    if flags.Detector.EnableSCT:
        kwargs.setdefault( "OutputSCTHits",             hardscatterSG+"SCT_Hits"            )
    if flags.Detector.EnableTRT:
        kwargs.setdefault( "OutputTRTUncompressedHits", hardscatterSG+"TRTUncompressedHits" )
 
    if flags.Detector.EnableITkPixel:
        kwargs.setdefault( "OutputITkPixelHits",        hardscatterSG+"ITkPixelHits"        )
    if flags.Detector.EnableITkStrip:
        kwargs.setdefault( "OutputITkStripHits",        hardscatterSG+"ITkStripHits"        )
    if flags.Detector.EnablePLR:
        kwargs.setdefault( "OutputPLRHits",             hardscatterSG+"PLR_Hits"            )
    if flags.Detector.EnableHGTD:
        kwargs.setdefault( "OutputHGTDHits",            hardscatterSG+"HGTD_Hits"           )
 
    if flags.Detector.EnableLAr:
        kwargs.setdefault( "OutputLArEMBHits",          hardscatterSG+"LArHitEMB"           )
        kwargs.setdefault( "OutputLArEMECHits",         hardscatterSG+"LArHitEMEC"          )
        kwargs.setdefault( "OutputLArFCALHits",         hardscatterSG+"LArHitFCAL"          )
        kwargs.setdefault( "OutputLArHECHits",          hardscatterSG+"LArHitHEC"           )
 
    if flags.Detector.EnableTile:
        kwargs.setdefault( "OutputTileHits",            hardscatterSG+"TileHitVec"          )
    if flags.Detector.EnableMBTS:
        kwargs.setdefault( "OutputMBTSHits",            hardscatterSG+"MBTSHits"            )
 
    if flags.Detector.EnableCSC:
        kwargs.setdefault( "OutputCSCHits",             hardscatterSG+"CSC_Hits"            )
    if flags.Detector.EnableMDT:
        kwargs.setdefault( "OutputMDTHits",             hardscatterSG+"MDT_Hits"            )
    if flags.Detector.EnableRPC:
        kwargs.setdefault( "OutputRPCHits",             hardscatterSG+"RPC_Hits"            )
    if flags.Detector.EnableTGC:
        kwargs.setdefault( "OutputTGCHits",             hardscatterSG+"TGC_Hits"            )
    if flags.Detector.EnablesTGC:
        kwargs.setdefault( "OutputsTGCHits",            hardscatterSG+"sTGC_Hits"           )
    if flags.Detector.EnableMM:
        kwargs.setdefault( "OutputMMHits",              hardscatterSG+"MM_Hits"             )
    return CompFactory.ISF.CollectionMerger(name, **kwargs)
 
 
def CollectionMergerCfg(flags,
                        bare_collection_name,
                        mergeable_collection_suffix,
                        merger_input_property,
                        region):
    """Generates and returns a collection name that is also registered to
       the ISF CollectionMerger algorithm.
 
       :param bare_collection_name: name of the collection if no merging
                                    is taking place.
       :param mergeable_collection_suffix: suffix to the collection in
                                           case merging is taking place.
       :param merger_input_property: name of the Input* property in the
                                     CollectionMerger algorithm to add the
                                     mergeable collection to."""
 
    result = ComponentAccumulator()
    from SimulationConfig.SimEnums import LArParameterization
    if (flags.Sim.ISFRun or flags.Sim.LArParameterization is LArParameterization.FastCaloSim) and flags.Sim.ISF.HITSMergingRequired.get(region, True):
        mergeable_collection = f'{bare_collection_name}{mergeable_collection_suffix}'
 
        from ISF_Algorithms.CollectionMergerConfig import ISFCollectionMergerCfg
        algo = ISFCollectionMergerCfg(flags)
        if flags.Sim.ISF.ReSimulation:
            result.addEventAlgo(algo,'SimSequence') # TODO ideally this would be configurable
        else:
            result.addEventAlgo(algo)
 
        input_attribute_name = f'Input{merger_input_property}'
        merger_input_collections = getattr(algo, input_attribute_name)
        if mergeable_collection not in merger_input_collections:
            merger_input_collections.append(mergeable_collection)
    else:
        mergeable_collection = bare_collection_name
    return result, mergeable_collection