CaloEstimatedGainTool_test.py

Go to the documentation of this file.

#
# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration.
#
# File: CaloTools/python/CaloEstimatedGainTool_test.py
# Author: scott snyder
# Date: Aug, 2019
# Brief: Test for CaloEstimatedGainTool.
#
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaPython.PyAthenaComps import Alg, StatusCode
from AthenaCommon.Logging import logging
import ROOT
 
 
def i32(x):
    if x >= (1<<31):
        x -= (1<<32)
    return x
 
 
class TestAlg (Alg):
    def __init__ (self, name):
        Alg.__init__ (self, name)
        return
 
    def initialize (self):
        ROOT.ICaloEstimatedGainTool
 
        self.tool = ROOT.ToolHandle(ROOT.ICaloEstimatedGainTool)('CaloEstimatedGainTool')
        if not self.tool.retrieve():
            assert 0
        return StatusCode.Success
 
 
    def testcell (self, elt, elist):
        ctx = self.getContext()
        for (e, gain) in elist:
            if isinstance(e, type(())):
                cell = ROOT.TileCell(elt, e[0])
                cell.setEnergy (e[0], e[1], 0, 0)
            else:
                cell = ROOT.CaloCell (elt, e, 0, 0, 0)
                g = i32(self.tool.estimatedGain (ctx, elt, e, 0))
                assert g == gain, (e, g, gain)
                g = i32(self.tool.estimatedGain (ctx, elt, e, 1))
                assert g == gain, (e, g, gain)
 
            g = i32(self.tool.estimatedGain (ctx, cell, 0))
            assert g == gain, (e, g, gain)
            g = i32(self.tool.estimatedGain (ctx, cell, elt, 0))
            assert g == gain, (e, g, gain)
            g = i32(self.tool.estimatedGain (ctx, cell, 1))
            assert g == gain, (e, g, gain)
            g = i32(self.tool.estimatedGain (ctx, cell, elt, 1))
            assert g == gain, (e, g, gain)
        return
 
    def execute (self):
        log = logging.getLogger ('TestAlg')
        ctx = self.getContext()
        mgr = self.condStore['CaloDetDescrManager'].find (ctx.eventID())
        elt1 = mgr.get_element (ROOT.CaloCell_ID.LAREM,
                                2, True, 0.5, 0.1)
        self.testcell (elt1, [(1000, 0), (50000, 1), (300000, 2)])
 
        elt2 = mgr.get_element (ROOT.CaloCell_ID.LARHEC,
                                2, False, 2.1, 0.1)
        self.testcell (elt2, [(1000, 1), (50000, 1), (300000, 2)])
 
        elt3 = mgr.get_element (ROOT.CaloCell_ID.LARFCAL,
                                1, False, 4.9, 0.1)
        self.testcell (elt3, [(1000, 0), (50000, 1), (800000, 2)])
 
        elt4 = mgr.get_element (ROOT.CaloCell_ID.TileBar1, 0.3, 0.1)
        self.testcell (elt4, [((1000, 1000), i32(ROOT.CaloGain.TILEHIGHHIGH)),
                              ((1000, 50000), i32(ROOT.CaloGain.TILEHIGHLOW)),
                              ((50000, 1000), i32(ROOT.CaloGain.TILEHIGHLOW)),
                              ((50000, 50000), i32(ROOT.CaloGain.TILELOWLOW)),
                             ])
        log.info ('finished')
        return StatusCode.Success
 
 
def testCfg (flags):
    result = ComponentAccumulator()
 
    from LArGeoAlgsNV.LArGMConfig import LArGMCfg
    from TileGeoModel.TileGMConfig import TileGMCfg
    result.merge(LArGMCfg(flags))
    result.merge(TileGMCfg(flags))
 
    from CaloTools.CaloEstimatedGainToolConfig import CaloEstimatedGainToolCfg
    acc = CaloEstimatedGainToolCfg (flags)
    acc.popPrivateTools()
    result.merge (acc)
 
    result.addEventAlgo (TestAlg ('TestAlg'))
    return result
 
 
from AthenaConfiguration.AllConfigFlags import initConfigFlags
from AthenaConfiguration.TestDefaults import defaultTestFiles
flags = initConfigFlags()
flags.Input.Files = defaultTestFiles.RDO_RUN2
flags.Detector.GeometryLAr = True
flags.Detector.GeometryTile = True
flags.needFlagsCategory('Tile')
flags.needFlagsCategory('LAr')
 
flags.lock()
from AthenaConfiguration.MainServicesConfig import MainServicesCfg
acc = MainServicesCfg(flags)
 
acc.merge (testCfg (flags))
acc.run(1)