TrdK.cpp

Go to the documentation of this file.

#include "fmt/format.h"
 
#include "Containers/TrdK.h"
 
namespace NAIA {
const std::string TrdKBase::BranchName = "TrdKBaseData";
const std::string TrdKBaseStandalone::BranchName = "TrdKBaseStandaloneData";
 
using namespace TrdK;
 
void TrdKBaseData::Dump() const {
  fmt::print("{:=^120} \n", " TrdKBaseData ");
  fmt::print("{:^15} {:^15} {:^15}\n", "AlignStatus", "CalibStatus", "ChargeStatus");
  fmt::print("{:^15} {:^15} {:^15}\n", AlignStatus, CalibStatus, ChargeStatus);
 
  if (!(AlignStatus || CalibStatus || ChargeStatus))
    return;
 
  if (NHits.size() > 0) {
    fmt::print("  {:<25}   {:>15} {:>15}\n", "Type", "NHits", "Amps");
    for (auto qualType : TrdK::qualTypes) {
      fmt::print("  - {:<23}   {:>15} {:>15}\n", TrdK::qualNames[qualType], NHits[qualType], Amps[qualType]);
    }
  }
 
  fmt::print("{:-^120}\n", "");
 
  if (Charge.size() > 0) {
    fmt::print("  {:^25}", " ");
    for (auto chargeType : TrdK::recoTypes)
      fmt::print("   {:>15}", TrdK::recoNames[chargeType]);
    fmt::print("\n");
 
    fmt::print("  {:<25}", "Charge");
    for (auto chargeType : TrdK::recoTypes)
      fmt::print("   {:>15f}", Charge[chargeType]);
    fmt::print("\n");
  }
 
  fmt::print("{:-^120}\n", "");
 
  if (Likelihood.size() > 0) {
    fmt::print("  {:^25}", " ");
    for (auto likeType : TrdK::likelihoodTypes)
      fmt::print("   {:>15}", TrdK::likelihoodNames[likeType]);
    fmt::print("\n");
 
    fmt::print("  {:<25}", "Likelihood");
    for (size_t likeType = 0; likeType < Likelihood.size(); likeType++)
      fmt::print("   {:>15f}", Likelihood[likeType]);
    fmt::print("\n");
  }
 
  if (LikelihoodRatio.size() > 0) {
    fmt::print("  {:^25}", " ");
    for (auto likeType : TrdK::likelihoodRTypes)
      fmt::print("   {:>15}", TrdK::likelihoodRNames[likeType]);
    fmt::print("\n");
 
    fmt::print("  {:<25}", "Likelihood ratio");
    for (auto likeType : TrdK::likelihoodRTypes)
      fmt::print("   {:>15f}", LikelihoodRatio[likeType]);
    fmt::print("\n");
  }
 
  fmt::print("{:-^120}\n", "");
 
  if (Edep.size() > 0) {
    fmt::print("{:-^120}\n", "");
    fmt::print("{:<32}   {:>15} {:>15}\n", " ", "Edep", "Pathlength");
    for (auto layerIter : Edep) {
      fmt::print("  - {:<28} = {:> 15.4f} {:> 15.4f}\n", fmt::format("Layer {}", layerIter.first),
                 Edep.at(layerIter.first), Pathlength.at(layerIter.first));
    }
  }
  fmt::print("{:-^120}\n", "");
}
 
void TrdKBaseData::Clear() {
  AlignStatus = false;
  CalibStatus = false;
  ChargeStatus = false;
 
  NHits.clear();
  Amps.clear();
  Charge.clear();
  Likelihood.clear();
  LikelihoodRatio.clear();
  Edep.clear();
  Pathlength.clear();
 
  Cleanliness = IPChi2 = 0;
  DirectionalStoermerCutoff = {0, 0};
 
  m_pointAndDir.clear();
}
 
TVector3 TrdKBaseData::InterpolateAtZ(float z) const {
  // not elegant, but who cares
  float dxdz = m_pointAndDir.at(1)[0] / m_pointAndDir.at(1)[2];
  float dydz = m_pointAndDir.at(1)[1] / m_pointAndDir.at(1)[2];
 
  TVector3 result{m_pointAndDir.at(0)};
  result[0] += dxdz * (z - result[2]);
  result[1] += dydz * (z - result[2]);
  result[2] = z;
 
  return result;
}
} // namespace NAIA