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("  - {:<20}   {:>15} {:>15} {:>15} {:>15} {:>20}\n", "", "Z", "A", "PDG", "Gen. Momentum",
             "Gen. Position");
  fmt::print("  - {:<20}   {:>15} {:>15} {:>15} {:>15.3f} ({:4.2f},{:4.2f},{:4.2f})\n", "", Primary.Z, Primary.A,
             Primary.PDGCode, Primary.GetGenMomentum(), Primary.Position[0][0], Primary.Position[0][1],
             Primary.Position[0][2]);
  fmt::print("{:-^120}\n", "");
}

void MCTruthPlusData::Dump() const {
  fmt::print("{:=^120} \n", " MCTruthPlus ");

  fmt::print("{:<32}\n", "Secondary particles");
  fmt::print("  - {:<20}   {:>15} {:>15} {:>15} {:>15} {:>15} {:>20}\n", "", "Z", "A", "PDG", "ID (parent)",
             "Gen. Momentum", "Gen. Position");
  for (const auto &sec : Secondaries) {
    fmt::print("    {:<20}   {:>15} {:>15} {:>15} {:>15} {:>15.3f} ({:4.2f},{:4.2f},{:4.2f})\n", "", sec.Z, sec.A,
               sec.PDGCode, fmt::format("{} ({})", sec.partID, sec.parentID), 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