EcalFill.cpp

Go to the documentation of this file.

#include "EcalKXRec.h"

#include "Containers/Ecal.h"

namespace NAIA {
using namespace Ecal;

void EcalBaseData::ResizeMembers() {
  Energy.resize(Ecal::numEnergyRecos);
  BDT.resize(Ecal::numBDTTypes);
}

bool EcalBaseData::Fill(EcalShowerR *showerPtr) {
  ResizeMembers();
  TotalHits = AMSEventR::Head()->nEcalHit();
  ShowerHits = showerPtr->Nhits;

  Energy[EnergyRecoType::EnergyD] = showerPtr->EnergyD / 1000.0f;
  Energy[EnergyRecoType::EnergyE] = showerPtr->EnergyE;
  Energy[EnergyRecoType::EnergyCorrEneE_E] = showerPtr->GetCorrectedEnergy(2, 2);
  Energy[EnergyRecoType::EnergyCorrEne2017] = showerPtr->ElectronEnergy2017();

  auto ecalKRec = EcalKXRec::GetHead();
  EcalKXRec::ProcessAMSEvent(AMSEventR::Head());
  ecalKRec->GetTotalEnergy(Energy[EnergyRecoType::EcalKEle]);

  BDT[BDTType::v7std] = showerPtr->GetEcalBDT(AMSEventR::Head(), 7, 0, 1);
  BDT[BDTType::v7ada] = showerPtr->GetEcalBDT(AMSEventR::Head(), 7, 1, 1);
  BDT[BDTType::v5std] = showerPtr->GetEcalBDT(AMSEventR::Head(), 5, 0, 1);
  BDT[BDTType::v5ada] = showerPtr->GetEcalBDT(AMSEventR::Head(), 5, 1, 1);

  m_pointAndDir.emplace_back(showerPtr->CofG[0], showerPtr->CofG[1], showerPtr->CofG[2]);
  m_pointAndDir.emplace_back(showerPtr->Dir[0], showerPtr->Dir[1], showerPtr->Dir[2]);

  return true;
}

void EcalPlusData::ResizeMembers() { Likelihood.resize(Ecal::numLikelihoodTypes); }

bool EcalPlusData::Fill(EcalShowerR *showerPtr) {
  ResizeMembers();
  auto ecalKRec = EcalKXRec::GetHead();
  ecalKRec->GetHadronFlag(EcalKHadronicFlag);
  ecalKRec->GetNumberOfShowers(EcalKNShowers);
  ecalKRec->GetLikelihood(Likelihood[LikelihoodType::Integral], Likelihood[LikelihoodType::LayerCombined]);

  std::array<float, 18> energy_total_planes{0};
  std::array<float, 18> energy_max_planes{0};

  auto event = AMSEventR::Head();
  for (int ihit = 0; ihit < event->nEcalHit(); ihit++) {
    energy_total_planes[event->pEcalHit(ihit)->Plane] += event->pEcalHit(ihit)->Edep;
    energy_max_planes[event->pEcalHit(ihit)->Plane] =
        std::max(energy_max_planes[event->pEcalHit(ihit)->Plane], event->pEcalHit(ihit)->Edep);
  }

  for (int ipl = 0; ipl < 18; ipl++) {
    if (energy_total_planes[ipl] > 0)
      MaxEnergyFraction[ipl] = energy_max_planes[ipl] / energy_total_planes[ipl];
  }

  return true;
}
} // namespace NAIA