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 is_mc) : isMC{is_mc}, m_logger{spdlog::stdout_color_st("NtpSelector")}, m_benchmark{} {};
 
  bool isMC = false;
  unsigned long long firstentry = 0;
  unsigned long long lastentry = 0;
  bool useGetMCBeta = false;
 
  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