SkimTreeHandle.h

Go to the documentation of this file.

#ifndef NAIA_SKIMTREEHANDLE_H
#define NAIA_SKIMTREEHANDLE_H

#include <memory>

#include "TFile.h"

namespace NAIA {

template <class T> class SkimTreeHandle {
public:
  SkimTreeHandle(std::unique_ptr<TFile> outFile, std::unique_ptr<T> chain, T *originalChain)
      : m_outFile{std::move(outFile)}, m_chain{std::move(chain)}, m_originalChain{originalChain} {};

  int Fill() {
    static unsigned int lastSec = 0;
    if (m_originalChain->GetEventRTIInfo().UTCTime != lastSec) {
      lastSec = m_originalChain->GetEventRTIInfo().UTCTime;
      m_chain->FillRTI();
    }

    static unsigned int lastRun = 0;
    if (m_originalChain->GetEventFileInfo().Run != lastRun) {
      lastRun = m_originalChain->GetEventFileInfo().Run;
      m_chain->FillFileInfo();
    }

    m_originalChain->m_event.ForceReadAllBranches();
    return m_chain->Fill();
  }

  int Write() { return m_chain->Write(); }

  TFile *RootFile() { return m_outFile.get(); }

private:
  std::unique_ptr<TFile> m_outFile;
  std::unique_ptr<T> m_chain;
  T *m_originalChain;
};
} // namespace NAIA

#endif // NAIA_SKIMTREEHANDLE_H