NtpSelector.h

Go to the documentation of this file.

#ifndef NAIA_NtpSelector_h
#define NAIA_NtpSelector_h

// c++ headers
#include <string>

// dependencies headers
#include "spdlog/sinks/stdout_color_sinks.h"
#include "spdlog/spdlog.h"

// AMS headers
#include "root.h"
#include "root_setup.h"

// NAIA headers
#include "Chain/NAIAChain.h"

#include "Utility/Benchmark/Benchmark.h"

namespace NAIA {
class NtpSelector : public AMSEventR {
public:
  NtpSelector(bool isMC) : m_isMC{isMC}, m_logger{spdlog::stdout_color_st("NtpSelector")}, m_benchmark{} {};

  bool m_isMC = false;
  unsigned long long firstentry = 0;
  unsigned long long lastentry = 0;

  void ReprocessRTI(bool reprocess) { m_reprocess_rti = reprocess; }
  void SetOutputFilename(std::string outFilename) { m_outputFilename = std::move(outFilename); }

  // AMS TSelector hooks
  virtual void UBegin() override;       
  virtual bool UProcessCut() override;  
  virtual void UProcessFill() override; 
  virtual void UTerminate() override;   

  virtual void Init(TTree *tree) override {
    m_tree = tree;
    AMSEventR::Init(tree);
  }
  virtual bool Notify() override; 

  void ProcessFileInfo();

private:
  void DefineStandAlone(); 

  // Reprocess only RTIInfo tree
  bool m_reprocess_rti = false;

  // pointer to currently open input tree
  TTree *m_tree;

  // output file
  std::string m_outputFilename;
  std::unique_ptr<TFile> m_outputFile;

  // trees
  std::unique_ptr<NAIAChain> m_chain;

  // standalone quantities
  BetaHR *m_betahStPtr;
  TrdTrackR *m_trdTrackStPtr;
  AMSSetupR::RTI m_rti;

  // logging and benchmarking
  Benchmark m_benchmark;
  std::shared_ptr<spdlog::logger> m_logger;
};
} // namespace NAIA
#endif