FlatTupleMaker.h

Go to the documentation of this file.

/*
 * @file FlatTupleMaker.h
 * @author Omar Moreno
 * @date January 18, 2016
 * @author PF
 * @date Jan, 2020
 * @brief 
 *
 */
 
#ifndef __FLAT_TUPLE_MAKER_H__
#define __FLAT_TUPLE_MAKER_H__
 
//----------------//   
//   C++ StdLib   //
//----------------//   
#include <string>
#include <map>
#include <iostream>
 
//----------//
//   ROOT   //
//----------//
#include <TFile.h>
#include <TTree.h>
 
class FlatTupleMaker {
 
    public:
        FlatTupleMaker(std::string file_name, std::string tree_name);
        
        FlatTupleMaker(std::string tree_name);
 
        ~FlatTupleMaker();
 
        void addVariable(std::string variable_name);
 
        void addString(std::string variable_name);
 
        void addVector(std::string vector_name);
 
        void setVariableValue(std::string variable_name, double value) { variables[variable_name] = value; };
 
        void setVariableValue(std::string variable_name, std::string value) { string_variables[variable_name] = value; };
 
        void addToVector(std::string variable_name, double value); 
 
        bool hasVariable(std::string variable_name); 
 
        void close();
 
        std::vector<double> getVector(std::string variable_name);
 
        void writeTree() {
            tree->Write();
        }
        void writeTree() {…}
        
        void fill();
 
    private: 
    
        TFile* file{nullptr};
 
        TTree* tree{nullptr}; 
 
        std::map <std::string, double> variables;
 
        std::map <std::string, std::string> string_variables; 
 
        std::map <std::string, std::vector<double>> vectors; 
        
 
};
class FlatTupleMaker {…};
 
#endif // __FLAT_TUPLE_MAKER_H__