EMShowerBuilder.cxx

Go to the documentation of this file.

/*
   Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
 */
 
// INCLUDE HEADER FILES:
#include "GaudiKernel/EventContext.h"
#include "GaudiKernel/IChronoStatSvc.h"
#include "StoreGate/ReadHandle.h"
 
#include "EMShowerBuilder.h"
#include "egammaInterfaces/IegammaIso.h"
#include "egammaInterfaces/IegammaShowerShape.h"
 
#include "CaloDetDescr/CaloDetDescrManager.h"
#include "CaloEvent/CaloCellContainer.h"
#include "CaloUtils/CaloCellList.h"
 
#include "xAODCaloEvent/CaloCluster.h"
#include "xAODEgamma/Egamma.h"
 
#include <algorithm>
 
//  END OF HEADER FILES INCLUDE
 
EMShowerBuilder::EMShowerBuilder(const std::string& type,
                                 const std::string& name,
                                 const IInterface* parent)
  : AthAlgTool(type, name, parent)
{
  // declare interface
  declareInterface<IEMShowerBuilder>(this);
}
 
 
StatusCode
EMShowerBuilder::initialize()
{
  ATH_MSG_DEBUG(
    " Initializing EMShowerBuilder, m_cellKey = " << m_cellsKey.key());
 
  ATH_CHECK(m_cellsKey.initialize((m_UseShowerShapeTool || m_UseCaloIsoTool) &&
                                  !m_cellsKey.key().empty()));
 
  ATH_CHECK(RetrieveTool<IegammaShowerShape>(m_ShowerShapeTool, m_UseShowerShapeTool));
  ATH_CHECK(RetrieveTool<IegammaIso>(m_HadronicLeakageTool, m_UseCaloIsoTool));
 
  return StatusCode::SUCCESS;
}
 
template <typename T>
StatusCode
EMShowerBuilder::RetrieveTool(ToolHandle<T> &tool, bool tool_requested)
{
  if (!tool_requested) {
    tool.disable();
    return StatusCode::SUCCESS;
  }
  if (tool.empty()) {
    ATH_MSG_INFO(tool.type() << " is empty");
    return StatusCode::FAILURE;
  }
  ATH_CHECK(tool.retrieve());
  return StatusCode::SUCCESS;
}
 
StatusCode
EMShowerBuilder::finalize()
{
  return StatusCode::SUCCESS;
}
 
StatusCode
EMShowerBuilder::execute(const EventContext& ctx,
                         const CaloDetDescrManager& cmgr,
                         xAOD::Egamma* eg) const
{
 
  ATH_MSG_DEBUG("Executing EMShowerBuilder::execute");
  // protection against bad pointers
  if (eg == nullptr)
    return StatusCode::SUCCESS;
 
  // retrieve the cell containers
  SG::ReadHandle<CaloCellContainer> cellcoll(m_cellsKey, ctx);
  // check is only used for serial running; remove when MT scheduler used
  ATH_CHECK(cellcoll.isValid());
 
  ATH_CHECK(executeWithCells(cellcoll.cptr(), cmgr, eg));
  return StatusCode::SUCCESS;
}
 
StatusCode
EMShowerBuilder::executeWithCells(const CaloCellContainer* cellcoll,
                                  const CaloDetDescrManager& cmgr,
                                  xAOD::Egamma* eg) const
{
  ATH_CHECK(CalcShowerShape(eg, cmgr, cellcoll));
  return StatusCode::SUCCESS;
}
 
StatusCode
EMShowerBuilder::CalcShowerShape(xAOD::Egamma* eg,
                                 const CaloDetDescrManager& cmgr,
                                 const CaloCellContainer* cellcoll) const
{
  //
  // Estimate shower shapes and fill the EMShower object associated to eg
  //
  ATH_MSG_DEBUG("Executing CalcShowerShape");
  // protection against bad pointers
  if (eg == nullptr) {
    return StatusCode::SUCCESS;
  }
  // retrieve the cluster
  const xAOD::CaloCluster* clus = eg->caloCluster();
  // Protect against non-existent structures.
  if (clus == nullptr) {
    ATH_MSG_WARNING("No Cluster");
    return StatusCode::SUCCESS;
  }
  if (cellcoll == nullptr) {
    ATH_MSG_WARNING("No cells");
    return StatusCode::SUCCESS;
  }
  // call calorimeter isolation tool only if needed
  if (m_UseCaloIsoTool) {
    ATH_CHECK(CalcHadronicLeakage(eg, cmgr, clus, cellcoll));
  }
 
  // Calculate shower shapes in all samplings
  if (m_UseShowerShapeTool) {
    // protection in case tool does not exist
    IegammaShowerShape::Info info;
    ATH_CHECK(m_ShowerShapeTool->execute(*clus, cmgr, *cellcoll, info));
    ATH_CHECK(FillEMShowerShape(eg, info));
  }
  return StatusCode::SUCCESS;
}
 
StatusCode
EMShowerBuilder::CalcHadronicLeakage(xAOD::Egamma* eg,
                                     const CaloDetDescrManager& cmgr,
                                     const xAOD::CaloCluster* clus,
                                     const CaloCellContainer* cellcoll) const
{
  //
  // Call calorimeter isolation tool
  //
  ATH_MSG_DEBUG("Executing CalcHadronicLeakage");
  if (m_caloNums.size() < 3) {
    ATH_MSG_DEBUG("Less than 3 subCalos, skipping");
    return StatusCode::SUCCESS;
  }
  // protection in case tool does not exist
  if (m_HadronicLeakageTool.empty()) {
    return StatusCode::SUCCESS;
  }
  // calculate information concerning just the hadronic leakage
  IegammaIso::Info info;
  StatusCode sc = m_HadronicLeakageTool->execute(*clus, cmgr, *cellcoll, info);
  if (sc.isFailure()) {
    ATH_MSG_WARNING("call to Iso returns failure for execute");
    return sc;
  }
  float value = 0;
  const double et = eg->caloCluster()->et();
  value = static_cast<float>(info.ethad1);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::ethad1);
  eg->setShowerShapeValue(et != 0. ? value / et : 0.,
                          xAOD::EgammaParameters::Rhad1);
  value = static_cast<float>(info.ethad);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::ethad);
  eg->setShowerShapeValue(et != 0. ? value / et : 0.,
                          xAOD::EgammaParameters::Rhad);
  value = static_cast<float>(info.ehad1);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::ehad1);
  return StatusCode::SUCCESS;
 
  unsigned int iflag = 0;
  if (info.nBadT0 > 0)
    iflag |= (0x1 << xAOD::EgammaParameters::DeadCellTileS0);
  if (info.nBadT0 > 0)
    iflag |= (0x1 << xAOD::EgammaParameters::DeadCellTileS1S2);
  eg->setOQ(iflag);
}
 
StatusCode
EMShowerBuilder::FillEMShowerShape(xAOD::Egamma* eg,
                                   const IegammaShowerShape::Info& info) const
{
 
  // protection in case Tool does not exist
  if (m_ShowerShapeTool.empty()) {
    return StatusCode::SUCCESS;
  }
  // information in the presampler
 
  // E in 1x1 cells in pre sampler
  float value = 0;
  // E in 1x1 cells in pre sampler
  value = static_cast<float>(info.e011);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e011);
  // E in 3x3 cells in pre sampler
  value = static_cast<float>(info.e033);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e033);
  // E in 3x2 cells in S1
  value = static_cast<float>(info.e132);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e132);
  // E in 15x2 cells in S1
  value = static_cast<float>(info.e1152);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e1152);
  // fraction of E in S1
  value = static_cast<float>(info.f1);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::f1);
  // fraction of E in the core(e132) in S1
  value = static_cast<float>(info.f1core);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::f1core);
  // corr width with 3 strips
  value = static_cast<float>(info.ws3c);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::weta1);
  // energy in second max
  value = static_cast<float>(info.esec);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e2ts1);
  // energy strip of second max
  value = static_cast<float>(info.esec1);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e2tsts1);
  // fraction of E outside core in S1
  value = static_cast<float>(info.fside);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::fracs1);
  // width with 5 strips
  value = static_cast<float>(info.widths5);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::widths1);
  // eta pos within cell in S1
  value = static_cast<float>(info.poscs1);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::poscs1);
  // asymmetry with 3 strips
  value = static_cast<float>(info.asymmetrys3);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::asy1);
  // diff position +/- 1 cells
  value = static_cast<float>(info.deltaEtaTrackShower);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::pos);
  // diff position +/- 7 cells
  value = static_cast<float>(info.deltaEtaTrackShower7);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::pos7);
  // E of strip with min E
  value = static_cast<float>(info.emins1);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::emins1);
  // E of strip with max E
  value = static_cast<float>(info.emaxs1);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::emaxs1);
  // barycentre in eta in S1
  value = static_cast<float>(info.etas3);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::barys1);
  // total width in strips
  value = static_cast<float>(info.wstot);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::wtots1);
  //
  // information in the 2nd sampling
  //
  // E in 3x3 cells in S2
  value = static_cast<float>(info.e233);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e233);
  // E in 3x5 cells in S2
  value = static_cast<float>(info.e235);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e235);
  // E in 3x7 cells in S2
  value = static_cast<float>(info.e237);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e237);
  // E in 5x5 cells in S2
  value = static_cast<float>(info.e255);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e255);
  // E in 7x7 cells in S2
  value = static_cast<float>(info.e277);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e277);
  // corr width in S2
  value = static_cast<float>(info.etaw);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::weta2);
  // uncorr width in S2
  value = static_cast<float>(info.width);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::widths2);
  // position in eta within cell in S2
  value = static_cast<float>(info.poscs2);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::poscs2);
  //
  // information in the 3rd sampling
  //
  // fraction of E in S3
  value = static_cast<float>(info.f3);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::f3);
  // fraction of E in the core (e333) in S3
  value = static_cast<float>(info.f3core);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::f3core);
  // E in 3x3 cells in S3
  value = static_cast<float>(info.e333);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e333);
  // E in 3x5 cells in S3
  value = static_cast<float>(info.e335);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e335);
  // E in 3x7 cells in S3
  value = static_cast<float>(info.e337);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e337);
  // E in 7x7 cells in S3
  value = static_cast<float>(info.e377);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::e377);
  //
  // information combining all samplings
  //
  // ratio of energy in 3x3/3x7 cells
  value = static_cast<float>(info.reta3337_allcalo);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::r33over37allcalo);
  // core energy
  value = static_cast<float>(info.ecore);
  eg->setShowerShapeValue(value, xAOD::EgammaParameters::ecore);
  //
  // information combining different shower shape
  //
  float valueSecond = 0;
  value = static_cast<float>(info.e277);
  valueSecond = static_cast<float>(info.e237);
  eg->setShowerShapeValue(value != 0 ? valueSecond / value : 0.,
                          xAOD::EgammaParameters::Reta);
  value = static_cast<float>(info.e233);
  valueSecond = static_cast<float>(info.e237);
  eg->setShowerShapeValue(valueSecond != 0 ? value / valueSecond : 0.,
                          xAOD::EgammaParameters::Rphi);
  value = static_cast<float>(info.emaxs1);
  valueSecond = static_cast<float>(info.esec1);
  eg->setShowerShapeValue(std::abs(value + valueSecond) > 0.
                            ? (value - valueSecond) / (value + valueSecond)
                            : 0.,
                          xAOD::EgammaParameters::Eratio);
 
  value = static_cast<float>(info.emins1);
  eg->setShowerShapeValue((valueSecond - value),
                          xAOD::EgammaParameters::DeltaE);
  return StatusCode::SUCCESS;
}