Source code for CaloHitBuilder.CaloHitBuilder

__all__ = ["CaloHitBuilder"]

from GaugiKernel import Logger, LoggingLevel
from GaugiKernel.macros import MSG_INFO
from CaloHitBuilder import CaloHitMaker
from CaloHitBuilder import CaloHitMerge
from G4Kernel import ComponentAccumulator


#
# Calo cell builder
#


class CaloHitBuilder(Logger):

    def __init__(self, 
                 name : str,
                 OutputHitsKey: str,
                 HistogramPath: str = "Expert",
                 OutputLevel: int = LoggingLevel.toC('INFO'),
                 ):

        Logger.__init__(self, name)
        self.__recoAlgs = []
        self.HistogramPath = HistogramPath
        self.OutputLevel = OutputLevel
        self.OutputHitsKey = OutputHitsKey
        self.OutputCollectionKeys = []



    def configure(self):

        MSG_INFO(self, "Configure CaloHitBuilder.")

        for samp in self.__detector.samplings:

            MSG_INFO(
                self,
                "Create new CaloHitMaker and dump all hits into %s collection",
                samp.CollectionKey
            )
            histogramPath = self.HistogramPath + '/' + samp.name()
            alg = CaloHitMaker("CaloHitMaker", samp,
                               OutputCollectionKey=samp.CollectionKey,
                               SamplingNoiseStd=samp.Noise,  # TOF selection
                               HistogramPath=histogramPath,
                               OutputLevel=self.OutputLevel,
                               # Use True when debug with only one thread
                               DetailedHistograms=False
                               )

            self.__recoAlgs.append(alg)
            self.OutputCollectionKeys.append(samp.CollectionKey)

        MSG_INFO(
            self,
            "Create CaloHitMerge and dump all hit collections into"
            " %s container",
            "Hits")
        # Merge all collection into a container
        # and split between truth and reco
        mergeAlg = CaloHitMerge("CaloHitMerge",
                                InputCollectionKeys=self.OutputCollectionKeys,
                                OutputHitsKey=self.OutputHitsKey,
                                OutputLevel=self.OutputLevel)
        self.__recoAlgs.append(mergeAlg)




    def merge(self, acc: ComponentAccumulator):
        """
        Obtains the detector from the ComponentAccumulator and appends
        all the hit makers required by the accumulator's detector.

        Parameters
        ----------
        acc : ComponentAccumulator
            Accumulator to merge with
        """

        self.__detector = acc.detector()
        self.configure()
        for reco in self.__recoAlgs:
            acc += reco