histflow.h

#ifndef HISTOS_H
#define HISTOS_H
 
#include <functional>
#include <map>
 
#include "cutflow.h"
#include "utilities.h"
 
class Histflow : public Cutflow
{
protected:
    std::map<std::string, std::vector<std::function<void(double)>>> fill_schedule;
    std::map<TString, std::function<void()>> hist_writers;
 
    bool recursiveEvaluate(Cut* cut) override;
 
    template<typename THist>
    THist* bookHist(std::string target_cut_name, THist* hist);
 
public:
    Histflow(std::string new_name);
 
    Histflow(std::string new_name, Cut* new_root);
 
    ~Histflow();
 
    template<typename THist1D>
    void bookHist1D(std::string target_cut_name, THist1D* hist, std::function<bool()> eval_lambda, 
                    std::function<double()> fill_lambda);
 
    template<typename THist1D>
    void bookHist1D(Cut* target_cut, THist1D* hist, std::function<bool()> eval_lambda, 
                    std::function<double()> fill_lambda);
 
    template<typename THist1D>
    void bookHist1D(std::string target_cut_name, THist1D* hist, std::function<double()> fill_lambda);
 
    template<typename THist1D>
    void bookHist1D(Cut* target_cut, THist1D* hist, std::function<double()> fill_lambda);
 
    template<typename THist2D>
    void bookHist2D(std::string target_cut_name, THist2D* hist, 
                    std::function<bool()> eval_lambda, 
                    std::function<std::pair<double, double>()> fill_lambda);
 
    template<typename THist2D>
    void bookHist2D(Cut* target_cut, THist2D* hist, 
                    std::function<bool()> eval_lambda, 
                    std::function<std::pair<double, double>()> fill_lambda);
 
    template<typename THist2D>
    void bookHist2D(std::string target_cut_name, THist2D* hist, 
                    std::function<std::pair<double, double>()> fill_lambda);
 
    template<typename THist2D>
    void bookHist2D(Cut* target_cut, THist2D* hist, std::function<std::pair<double, double>()> fill_lambda);
 
    template<typename THist3D>
    void bookHist3D(std::string target_cut_name, THist3D* hist, 
                    std::function<bool()> eval_lambda, 
                    std::function<std::tuple<double, double, double>()> fill_lambda);
 
    template<typename THist3D>
    void bookHist3D(Cut* target_cut, THist3D* hist, 
                    std::function<bool()> eval_lambda, 
                    std::function<std::tuple<double, double, double>()> fill_lambda);
 
    template<typename THist3D>
    void bookHist3D(std::string target_cut_name, THist3D* hist, 
                    std::function<std::tuple<double, double, double>()> fill_lambda);
 
    template<typename THist3D>
    void bookHist3D(Cut* target_cut, THist3D* hist, 
                    std::function<std::tuple<double, double, double>()> fill_lambda);
 
    void writeHists(TFile* tfile);
};
 
#include "histflow.icc"
 
#endif