CHeTReader.hh

Go to the documentation of this file.

#ifndef CHETREADER_HH
#define CHETREADER_HH
 
#include <algorithm>
#include <iostream>
#include <limits>
#include <map>
#include <string>
#include <vector>
 
#include <ROOT/RDataFrame.hxx>
#include <ROOT/RVec.hxx>
 
#include "CHeT/CHeTGlobalSettings.hh"
 
namespace CHeT
{
namespace Data
{
 
class Reader
{
  public:
    Reader(const std::string &filename, const std::string &treeName = "auto");
 
    // Disable copy (RDataFrame has unique pointers)
    Reader(const Reader &) = delete;
    Reader &operator=(const Reader &) = delete;
 
    void SetCuts(double toaMin, double toaMax, unsigned int totMin, unsigned int totMax);
 
    void SetSingleEntry(long entry);
 
    void SetEventByID(int eventID);
 
    void SetEnabledBoards(const std::vector<int> &boards);
 
    void SetEnabledCylinders(const std::vector<int> &cylinders);
 
    void SetEnabledLayers(const std::vector<int> &layers);
 
    void SetEnabledGeometries(const std::vector<std::pair<int, int>> &geometries);
 
    ROOT::RDF::RNode GetRaw();
 
    void SaveToTree(const std::string &filename, const std::string &treeName = "chet");
 
    ROOT::RDF::RNode GetCHeTTree();
 
  private:
    std::string fFilename;
    std::string fTreeName;
    ROOT::RDataFrame fDF;
    ROOT::RDF::RNode fHeadNode; // The active head node (potentially filtered)
 
    // Cut parameters (default: open)
    double fToaMin = -1e9;
    double fToaMax = 1e9;
    unsigned int fTotMin = 0;
    unsigned int fTotMax = 99999;
 
    // Filters
    std::vector<int> fEnabledBoards;
    std::vector<int> fEnabledCylinders;
    std::vector<int> fEnabledLayers;
    std::vector<std::pair<int, int>> fEnabledGeometries;
};
 
} // namespace Data
} // namespace CHeT
 
#endif