TSNaiaChain.cpp

Go to the documentation of this file.

 
#include "Chain/TSNaiaChain.h"
#include "Utility/Logging/Logging.h"
 
using namespace NAIA;
ClassImp(TSNAIAChain);
 
TSNAIAChain::TSNAIAChain() : TChain("NAIAChain"), ev(nullptr), rti(nullptr), finfo(nullptr), mcfinfo(nullptr) {
  rti_chain.SetName("RTI");
  file_chain.SetName("FileInfo");
}
 
TSNAIAChain::~TSNAIAChain() {
  if (ev)
    delete ev;
  if (rti)
    delete rti;
  if (finfo)
    delete finfo;
  if (mcfinfo)
    delete mcfinfo;
}
 
Int_t TSNAIAChain::Add(const char *name, Long64_t nentries) {
  static constexpr std::string_view routineName{"TSNAIAChain::Add"};
  auto logger = getLogger(routineName);
 
  Long64_t ntrees[3] = {0, 0, 0};
 
  ntrees[0] += TChain::Add(name, nentries);
 
  if (ntrees[0] > 0)
    GetEntry(0);
 
  if (ev == nullptr) {
    ev = new Event;
    if (IsMC())
      ev->SetMC(true);
    ev->SetAllBranchAddress(this);
  }
 
  ntrees[1] += rti_chain.Add(name, nentries);
  if (ntrees[1] > 0) {
    if (rti == nullptr && !IsMC()) {
      rti = new RTIInfo;
      rti_chain.SetBranchAddress("RTIInfo", &rti);
    }
    if (ntrees[1] != ntrees[0]) {
      logger->warn("I found {} NAIAChain trees but only {} RTIInfo trees", ntrees[0], ntrees[1]);
    }
    rti_chain.BuildIndex("UTCTime");
    AddFriend(&rti_chain);
  }
 
  ntrees[2] += file_chain.Add(name, nentries);
  if (finfo == nullptr) {
    finfo = new FileInfo;
    file_chain.SetBranchAddress("FileInfo", &finfo);
    if (IsMC()) {
      mcfinfo = new MCFileInfo;
      file_chain.SetBranchAddress("MCFileInfo", &mcfinfo);
    }
  }
  if (ntrees[2] != ntrees[0]) {
    logger->warn("I found {} NAIAChain trees but only {} FileInfo trees", ntrees[0], ntrees[2]);
  }
 
  GetEntry(0);
 
  return ntrees[0];
}
 
bool TSNAIAChain::IsMC() { return GetBranch(MCTruthBase::BranchName.c_str()); }
 
RTIInfo *TSNAIAChain::UpdateEventRTIInfo(unsigned int UT) {
  if (IsMC())
    throw(std::runtime_error("UpdateEventRTIInfo called on a MC chain"));
 
  if (rti && rti->UTCTime == UT) {
    return rti;
  } else {
    int ret = rti_chain.GetEntryWithIndex(UT);
    return (ret > 0) ? rti : nullptr;
  }
}
 
Int_t TSNAIAChain::GetEntry(Long64_t entry, Int_t getall) {
  int ret = TChain::GetEntry(entry, getall);
  // if (ret > 0 && !IsMC() && ev) UpdateEventRTIInfo(ev->header.UTCTime);
  return ret;
}
 
Int_t TSNAIAChain::GetNextEntry(bool reset) {
  Long64_t current = GetReadEntry();
  if (reset)
    current = -1;
  return GetEntry(++current);
}
 
Event *TSNAIAChain::GetEventWithRunEvent(int run, int event) {
  Long64_t Entry_num = GetEntryNumberWithIndex(run, event);
  if (Entry_num < 0)
    return nullptr;
 
  int ret = GetEntry(Entry_num);
 
  if (ret > 0)
    return ev;
  else
    return nullptr;
}