testReadNtp.cpp

Go to the documentation of this file.

// dependencies headers
#include "fmt/ranges.h"
#include "spdlog/sinks/stdout_color_sinks.h"
#include "spdlog/spdlog.h"
 
// NAIA headers
#include "Chain/NAIAChain.h"
 
using namespace NAIA;
 
int main(int argc, char const *argv[]) {
  NAIAChain chain(NAIAChain::AccessMode::Read);
  chain.Add("test.root");
  // chain.Add("skimmed.root");
 
  chain.SetupBranches();
 
  spdlog::info("{} entries in the chain", chain.GetEntries());
 
  auto handle1 = chain.CreateSkimTree("skimmed.root", "RichBase;RichPlus");
 
  unsigned long long nEntries = chain.GetEntries();
  // unsigned long long nEntries = 1;
 
  unsigned long long skimmedEntries = 0;
 
  // auto firstEvent = chain.GetEvent(0);
 
  // std::vector<unsigned long long> evList = {0, 1, 0, 1, 2, 1, 3, 1};
 
  // for (auto iEv : evList) {
  // for (unsigned long long iEv = 0; iEv < nEntries; iEv++) {
  //   auto &event = chain.GetEvent(iEv);
  for (auto &event : chain) {
 
    // if (!event.CheckMask(NAIA::Category::HasEcal))
    //   continue;
 
    spdlog::info("Entry {} - Mask {}", event.header->EventNo, to_string_binary<32>(event.header->Mask()));
 
    // const RTIInfo &rtiInfo = chain.GetEventRTIInfo();
    // if (rtiInfo.IsInSAA())
    //   continue;
 
    // float tofCharge, innerCharge, rigidity;
 
    // if (KeyExists(Tof::ChargeType::Upper, event.tofBase->Charge))
    //   tofCharge = event.tofBase->Charge[Tof::ChargeType::Upper];
 
    // innerCharge = event.trTrackBase->Charge[TrTrack::ChargeRecoType::YJ];
 
    auto pattern = event.trTrackBase->GetTrackPattern();
    for (auto hit_type : pattern) {
      switch (hit_type.second) {
      case HitClusterAssociation::None:
        fmt::print("{} None\n", hit_type.first);
        break;
      case HitClusterAssociation::Y:
        fmt::print("{} Y-only\n", hit_type.first);
        break;
      case HitClusterAssociation::XY:
        fmt::print("{} XY\n", hit_type.first);
        break;
      }
    }
 
    event.trTrackPlus->Dump();
 
    // if (event.trTrackBase->FitIDExists(TrTrack::Fit::Kalman, TrTrack::Span::InnerOnly))
    //   rigidity = event.trTrackBase->Rigidity[TrTrack::Fit::Kalman][TrTrack::Span::InnerOnly];
 
    // chain.GetEventFileInfo().Dump();
    // chain.GetEventRTIInfo().Dump();
    // auto rtiInfo = chain.GetEventRTIInfo();
    // fmt::print("{}\n{}\n", rtiInfo.nEventInnerL1L9, rtiInfo.nEventInnerL1L9Int90);
 
    // event.header->Dump();
    // event.evSummary->Dump();
 
    // event.mcTruthBase->Dump();
    // event.mcTruthPlus->Dump();
 
    // event.tofBaseSt->Dump();
    // event.tofPlusSt->Dump();
    // event.trdKBaseSt->Dump();
    // event.extHitBase->Dump();
 
    // event.tofBase->Dump();
    // event.tofPlus->Dump();
 
    // event.trdKBase->Dump();
 
    // event.trTrackBase->Dump();
    // event.trTrackPlus->Dump();
    // auto x_pattern = event.trTrackNase->GetTrackPattern(TrTrack::Side::X);
    // auto y_pattern = event.trTrackNase->GetTrackPattern(TrTrack::Side::Y);
    //
    // spdlog::info("X: {}  Y: {}", x_pattern.to_string(), y_pattern.to_string());
 
    // event.ecalBase->Dump();
    // event.ecalPlus->Dump();
 
    // event.richBase->Dump();
    // event.richPlus->Dump();
 
    // if (event.CheckMask(NAIA::Category::ChargeGT2_Trk)) {
    //   handle1.Fill();
    //   skimmedEntries++;
    // }
 
    //    std::cin.get();
  }
 
  spdlog::info("We skimmed {} entries", skimmedEntries);
 
  // handle1.Write();
 
  return 0;
}