arbol.h

#ifndef ARBOL_H
#define ARBOL_H
 
#include <iostream>
#include <functional>
#include <string>
#include <vector>
#include <map>
 
#include "TString.h"
#include "TTree.h"
#include "TFile.h"
#include "TObject.h"
 
#include "hepcli.h"
#include "utilities.h"
 
template<typename Type>
class Branch : public Utilities::Variable<Type>
{
private:
    TBranch* branch;
public:
    Branch();
    Branch(TTree* ttree, TString new_branch_name);
};
 
class Arbol
{
protected:
    std::map<TString, Utilities::Dynamic*> branches;
    std::map<TString, std::function<void()>> branch_resetters;
    template<typename Type>
    Branch<Type>* getBranch(TString branch_name);
public:
    TTree* ttree;
    TFile* tfile;
 
    Arbol();
    Arbol(TFile* tfile, TString ttree_name = "tree");
    Arbol(TString tfile_name, TString ttree_name = "tree");
    Arbol(HEPCLI& cli);
    virtual ~Arbol();
    template<typename Type>
    void newBranch(TString new_branch_name);
    template<typename Type>
    void newBranch(TString new_branch_name, Type new_reset_value);
    template<typename Type>
    void setBranchResetValue(TString branch_name, Type new_reset_value);
    template<typename Type>
    Type getLeaf(TString branch_name);
    template<typename Type>
    void setLeaf(TString branch_name, Type new_value);
    template<typename Type>
    void newVecBranch(TString new_branch_name);
    template<typename Type>
    void newVecBranch(TString new_branch_name, std::vector<Type> new_reset_vector);
    template<typename Type>
    void setVecBranchResetValue(TString branch_name, std::vector<Type> new_reset_vector);
    template<typename Type>
    std::vector<Type> getVecLeaf(TString branch_name);
    template<typename Type>
    void setVecLeaf(TString branch_name, std::vector<Type> new_vector);
 
    template<typename Type>
    void appendToVecLeaf(TString branch_name, Type new_value);
    template<typename Type>
    void prependToVecLeaf(TString branch_name, Type new_value);
    template<typename Type>
    void insertIntoVecLeaf(TString branch_name, Type new_value, int index);
    template<typename Type>
    void sortVecLeaf(TString branch_name, std::function<bool(Type, Type)> &lambda);
    void resetBranches();
    virtual void fill();
    virtual void write(bool close = true);
};
 
#include "arbol.icc"
 
#endif