testReadNtp.cpp

Go to the documentation of this file.

// dependencies headers
#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(SingleTreeChain::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];

    // if (event.trTrackBase->FitIDExists(TrTrack::Fit::Kalman, TrTrack::Span::InnerOnly))
    //   rigidity = event.trTrackBase->RigidityCorr[TrTrack::Fit::Kalman][TrTrack::Span::InnerOnly];

    // chain.GetEventFileInfo().Dump();
    // chain.GetEventRTIInfo().Dump();

    // 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();

    // event.ecalBase->Dump();
    // event.ecalPlus->Dump();

    // event.richBase->Dump();
    // event.richPlus->Dump();

    fmt::print("{:<32}: {:>15} {:>15} {:>15} {:>15}\n", "", "Rigidity", "InvRigErr", "TrChiSqX", "TrChiSqY");
    for (unsigned int ijl = 0; ijl < 9; ijl++) {
      if (!KeyExists(ijl, event.trTrackPlus->PartialRigidity))
        continue;

      //   for (auto &fitIter : event.trTrackPlus->PartialRigidity[ijl]) {
      //     auto fit = fitIter.first;

      //     fmt::print("{:<32}: {:>15} {:>15} {:>15} {:>15}\n",
      //                fmt::format("Layer {} - Fit {}", ijl + 1, TrTrack::fitNames[fit]),
      //                event.trTrackPlus->PartialRigidity[ijl][fit], event.trTrackPlus->PartialInvRigErr[ijl][fit],
      //                event.trTrackPlus->PartialTrChiSq[ijl][fit][TrTrack::Side::X],
      //                event.trTrackPlus->PartialTrChiSq[ijl][fit][TrTrack::Side::Y]);
      //   }
    }

    if (event.CheckMask(NAIA::Category::ChargeGT2_Trk)) {
      handle1.Fill();
      skimmedEntries++;
    }

    std::cin.get();
  }

  spdlog::info("We skimmed {} entries", skimmedEntries);

  handle1.Write();

  return 0;
}