MCTruth.cpp

Go to the documentation of this file.

#include "fmt/format.h"
#include <bitset>
 
#include "Containers/MCTruth.h"
 
namespace NAIA {
const std::string MCTruthBase::BranchName = "MCTruthBaseData";
const std::string MCTruthPlus::BranchName = "MCTruthPlusData";
 
void MCTruthBaseData::Clear() { Primary = MCParticle{}; }
 
void MCTruthBaseData::Dump() const {
  fmt::print("{:=^120} \n", " MCTruthBaseData ");
 
  fmt::print("{:<32}\n", "Primary particle");
  fmt::print("   {:>12} {:>12} {:>12} {:>12} {:>12} {:>20}\n", "Z", "A", "Mass", "PDG", "Gen. Momentum",
             "Gen. Position");
  int charge = Primary.Z;
  for (unsigned int i = 0; i < Primary.Position.size(); ++i) {
    fmt::print("   {:>12} {:>12} {:>12.3f} {:>12} {:>12.3f} ({:6.2f}, {:6.2f}, {:6.2f})\n", charge, Primary.A,
               Primary.Mass, Primary.PDGCode, Primary.GetGenMomentum(), Primary.Position[i][0], Primary.Position[i][1],
               Primary.Position[i][2]);
  }
  fmt::print("{:-^120}\n", "");
}
 
void MCTruthPlusData::Dump() const {
  fmt::print("{:=^120} \n", " MCTruthPlus ");
 
  fmt::print("{:<32}\n", "Secondary particles");
  fmt::print("   {:>12} {:>12} {:>12} {:>12} {:>12} {:>12} {:>12} {:>12} {:>20}\n", "Z", "A", "Mass", "PDG",
             "ID (parent)", "Proc1", "Proc2", "Gen. Momentum", "Gen. Position");
  for (const auto &sec : Secondaries) {
    int charge = sec.Z;
    fmt::print("   {:>12} {:>12} {:>12.3f} {:>12} {:>12} {:>12} {:>12} {:>12.3f} ({:6.2f}, {:6.2f}, {:6.2f})\n", charge,
               sec.A, sec.Mass, sec.PDGCode, fmt::format("{} ({})", sec.partID, sec.parentID), sec.Proc1, sec.Proc2,
               sec.GetGenMomentum(), sec.Position[0][0], sec.Position[0][1], sec.Position[0][2]);
  }
  fmt::print("{:-^120}\n", "");
}
 
void MCTruthPlusData::Clear() {
  Secondaries.clear();
  TrackMCHits.clear();
}
} // namespace NAIA