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(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];

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

    // 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;
}