Ecal.cpp

Go to the documentation of this file.

#include "fmt/format.h"

#include "Containers/Ecal.h"

namespace NAIA {
const std::string EcalBase::BranchName = "EcalBaseData";
const std::string EcalPlus::BranchName = "EcalPlusData";

void EcalBaseData::Clear() {
  TotalHits = ShowerHits = 0;

  BDT.clear();
  Energy.clear();

  m_pointAndDir.clear();
};

void EcalPlusData::Clear() {
  EcalKNShowers = 0;
  EcalKHadronicFlag = 0;
  Likelihood.clear();
  MaxEnergyFraction.clear();
};

void EcalBaseData::Dump() const {
  fmt::print("{:=^120} \n", " EcalBaseData ");
  fmt::print("{:^15} {:^15}\n", "TotalHits", "ShowerHits");
  fmt::print("{:^15} {:^15}\n", TotalHits, ShowerHits);

  if (Energy.size() > 0) {
    fmt::print("  {:^15}", " ");
    for (auto recoType : Ecal::recoTypes)
      fmt::print("   {:>20}", Ecal::recoNames[recoType]);
    fmt::print("\n");

    fmt::print("  {:<15}", "Energy");
    for (auto recoType : Ecal::recoTypes)
      fmt::print("   {:>20f}", Energy[recoType]);
    fmt::print("\n");
  }

  if (BDT.size() > 0) {
    fmt::print("  {:^15}", " ");
    for (auto bdtType : Ecal::BDTTypes)
      fmt::print("   {:>20}", Ecal::BDTNames[bdtType]);
    fmt::print("\n");

    fmt::print("  {:<15}", "BDT");
    for (auto bdtType : Ecal::BDTTypes)
      fmt::print("   {:>20f}", BDT[bdtType]);
    fmt::print("\n");
  }

  if (m_pointAndDir.size() > 0) {
    fmt::print("\n  COG: ({:5.2f}, {:5.2f}, {:5.2f})   Direction: ({:5.2f}, {:5.2f}, {:5.2f})\n", GetCOG().x(),
               GetCOG().y(), GetCOG().z(), GetDirection().x(), GetDirection().y(), GetDirection().z());
  }

  fmt::print("{:-^120}\n", "");
}

void EcalPlusData::Dump() const {
  fmt::print("{:=^120} \n", " EcalPlusData ");
  fmt::print("{:^15} {:^15}\n", "EcalKNShowers", "EcalKHadronicFlag");
  fmt::print("{:^15} {:^15}\n", EcalKNShowers, EcalKHadronicFlag);

  if (Likelihood.size() > 0) {
    fmt::print("  Likelihood: ({:5.2f}, {:5.2f})\n", Likelihood[Ecal::LikelihoodType::Integral],
               Likelihood[Ecal::LikelihoodType::LayerCombined]);
  }

  if (MaxEnergyFraction.size() > 0) {
    fmt::print("  {:<15}", "MaxEnergyFraction:");
    for (unsigned int iLayer = 0; iLayer < 18; iLayer++) {
      if (ContainsKeys(MaxEnergyFraction, iLayer))
        fmt::print(" {:>5.2f}", MaxEnergyFraction.at(iLayer));
    }
    fmt::print("\n");
  }

  fmt::print("{:-^120}\n", "");
}

} // namespace NAIA