SkimTreeHandle.h

Go to the documentation of this file.

#ifndef NAIA_SKIMTREEHANDLE_H
#define NAIA_SKIMTREEHANDLE_H

#include <memory>

#include <TFile.h>

#include "NAIAVersion.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() {

    if (!m_originalChain->IsMC()) {
      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() {
    NAIA::Version versionHeader;
    m_outFile->WriteTObject(&versionHeader, "VersionHeader");
    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