CaloSwClcon_v2.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
#include "CaloSwClcon_v2.h"
#include "CaloClusterCorrection/interpolate.h"
#include <cmath>
 
 
using xAOD::CaloCluster;
using CaloClusterCorr::interpolate;
 
 
void CaloSwClcon_v2::makeTheCorrection (const Context& myctx,
                                        CaloCluster* cluster,
                                        const CaloDetDescrElement* /*elt*/,
                                        float eta,
                                        float adj_eta,
                                        float /*phi*/,
                                        float /*adj_phi*/,
                                        CaloSampling::CaloSample /*samp*/)
  const
{
  // ??? In principle, we should use adj_eta for the interpolation
  //     and range checks.  However, the v2 corrections were derived
  //     using regular eta instead.
  float the_aeta;
  if (m_use_raw_eta (myctx))
    the_aeta = std::abs (adj_eta);
  else
    the_aeta = std::abs (eta);
 
  if (the_aeta >= m_etamax (myctx))
    return;
 
  const CxxUtils::Array<2> correction = m_correction (myctx);
  const CxxUtils::Array<1> energies = m_energies (myctx);
 
  // Calculate the correction for each energy.
  unsigned int n_energies = energies.size();
  unsigned int shape[] = {n_energies, 2};
  CaloRec::WritableArrayData<2> offstab (shape);
 
  for (unsigned int i=0; i<n_energies; i++) {
    offstab[i][0] = energies[i];
    offstab[i][1] = interpolate (correction,
                                 the_aeta,
                                 m_degree (myctx),
                                 i+1);
  }
 
  // Find the median correction.
  float energy = cluster->e() * offstab[n_energies/2][1];
 
  // Now interpolate in energy.
  float corr;
  if (energy <= offstab[0][0])
    corr = offstab[0][1];
  else if (energy >= offstab[n_energies-1][0])
    corr = offstab[n_energies-1][1];
  else
    corr = interpolate (offstab, energy, m_energy_degree (myctx));
 
  // Do the correction
  setenergy (cluster, corr * cluster->e());
}