egammaMiddleShape.cxx

Go to the documentation of this file.

/*
   Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 */
 
#include "egammaCaloUtils/egammaMiddleShape.h"
#include "egammaCaloUtils/egammaEnergyPositionAllSamples.h"
#include "egammaCaloUtils/egammaqweta2c.h"
 
#include "CaloDetDescr/CaloDetDescrManager.h"
#include "CaloEvent/CaloCluster.h"
#include "CaloUtils/CaloCellList.h"
#include "CaloUtils/CaloLayerCalculator.h"
#include "xAODCaloEvent/CaloCluster.h"
 
 
StatusCode
egammaMiddleShape::execute(const xAOD::CaloCluster& cluster,
                           const CaloDetDescrManager& cmgr,
                           const CaloCellContainer& cell_container,
                           Info& info,
                           bool doRetaOnly)
{
  //
  // Estimate shower shapes from 2nd compartment
  // based on hottest cell and deta,dphi
  // with eta = cluster->eta(sam)
  //      phi = cluster->phi(sam)
  // and search for hottest cell in a 7X7 window
  // in the second sampling
  //
  //
  StatusCode sc = StatusCode::SUCCESS;
  // check if cluster is in barrel or in the end-cap
  if (!cluster.inBarrel() && !cluster.inEndcap()) {
    return sc;
  }
 
  // check if cluster is in barrel or end-cap
  const bool in_barrel = egammaEnergyPositionAllSamples::inBarrel(cluster, 2);
 
  // define accordingly position of CaloSampling
  const CaloSampling::CaloSample sam = in_barrel ? CaloSampling::EMB2 : CaloSampling::EME2;
 
  // granularity in (eta,phi) in the pre sampler
  // CaloCellList needs both enums: subCalo and CaloSample
  bool barrel = false;
  CaloCell_ID::SUBCALO subcalo = CaloCell_ID::LAREM;
  int sampling_or_module = 0;
 
  float etacell = cluster.etamax(sam);
  double eta = cluster.etaSample(sam);
  double phi = cluster.phiSample(sam);
  if ((eta == 0. && phi == 0.) || std::abs(eta) > 100) {
    return sc;
  }
 
  CaloDetDescrManager::decode_sample(subcalo, barrel, sampling_or_module, sam);
 
  // Get the corresponding grannularities : needs to know where you are
  //                  the easiest is to look for the CaloDetDescrElement
  const CaloDetDescrElement* dde =
    cmgr.get_element(subcalo, sampling_or_module, barrel, eta, phi);
  // if object does not exist then return
  if (!dde) {
    return sc;
  }
  // local granularity
  double deta = dde->deta();
  double dphi = dde->dphi();
 
  // Find the hottest cell
  // in a 7x7 (7deta,7*dphi) window
  // opened in the eta/phi of the
  // cluster in this sampling
  CaloLayerCalculator calc;
  sc = calc.fill(cmgr,
                 &cell_container,
                 cluster.etaSample(sam),
                 cluster.phiSample(sam),
                 7. * deta,
                 7. * dphi,
                 sam);
  double etamax = calc.etarmax();
  double phimax = calc.phirmax();
 
  //select the widest cell list we will use
  //i.e a 7x7 around the found hottest cell
  CaloCellList cell_list(&cmgr, &cell_container);
  cell_list.select(etamax, phimax, 7. * deta, 7. * dphi, sam);
 
  // Using that list do the filling
  // of all the sub windows.
 
  // 7x7
  calc.fill(cell_list.begin(),
            cell_list.end(),
            etamax,
            phimax,
            7. * deta,
            7. * dphi,
            sam);
  info.e277 = calc.em();
  // 3X7
  calc.fill(cell_list.begin(),
            cell_list.end(),
            etamax,
            phimax,
            3. * deta,
            7. * dphi,
            sam);
  info.e237 = calc.em();
 
  if (!doRetaOnly) {
    // estimate the relevant quantities around the hottest cell
    // in the following eta X phi windows
    // 3X3
    calc.fill(cell_list.begin(),
              cell_list.end(),
              etamax,
              phimax,
              3. * deta,
              3. * dphi,
              sam);
    info.e233 = calc.em();
    // 3X5
    calc.fill(cell_list.begin(),
              cell_list.end(),
              etamax,
              phimax,
              3. * deta,
              5. * dphi,
              sam);
    info.e235 = calc.em();
    double etaw = calc.etas();
    info.phiw = calc.phis();
    info.etaw = egammaqweta2c::Correct(eta, etacell, etaw);
    info.width = etaw;
    info.poscs2 = egammaqweta2c::RelPosition(eta, etacell);
    // 5X5
    calc.fill(cell_list.begin(),
              cell_list.end(),
              etamax,
              phimax,
              5. * deta,
              5. * dphi,
              sam);
    info.e255 = calc.em();
  }
  return sc;
}