gRootTree.h

Go to the documentation of this file.

#pragma once
 
// c++
#include <map>
 
// ROOT
#include "TTree.h"
#include "ROOT/TThreadedObject.hxx"
 
// gemc
#include "event/gEventDataCollection.h"
#include "run/gRunDataCollection.h"
 
#define EVENTHEADERTREENAME "event_header"
#define RUNHEADERTREENAME   "run_header"
#define TRUEINFONAMEPREFIX  "true_info_"
#define DIGITIZEDNAMEPREFIX "digitized_"
#define GENERATEDTREENAMEDESC "Generated Particles"
#define ANCESTORTREENAME "ancestors"
#define ANCESTORTREENAMEDESC "Geant4 Ancestors"
 
#define ERR_GSTREAMERROOTTREENOTFOUND 850
 
#define EVENTHEADERTREENAMEDESC "Event Header"
#define RUNHEADERTREENAMEDESC   "Run Header"
#define TRUEINFOTREENAMEDESC    "True Info Data"
#define DIGITIZEDTREENAMEDESC   "Digitized Data"
 
class GRootTree
{
public:
    ~GRootTree() { if (log) log->debug(NORMAL, "~GRootTree"); }
 
    GRootTree([[maybe_unused]] const std::unique_ptr<GEventHeader>& gevent_header, std::shared_ptr<GLogger>& log);
 
    GRootTree([[maybe_unused]] const std::unique_ptr<GRunHeader>& grun_header, std::shared_ptr<GLogger>& log);
 
    GRootTree(const std::string& detectorName, const GTrueInfoData* gdata, std::shared_ptr<GLogger>& log);
 
    GRootTree(const std::string& detectorName, const GDigitizedData* gdata, std::shared_ptr<GLogger>& log);
 
    GRootTree(const std::string& treeName, const GGeneratedParticleBank& particles, std::shared_ptr<GLogger>& log);
 
    GRootTree(const GAncestorBank& ancestors, std::shared_ptr<GLogger>& log);
 
    bool fillTree(const std::unique_ptr<GEventHeader>& gevent_header);
 
    bool fillTree(const std::unique_ptr<GRunHeader>& run_header);
 
    bool fillTree(const std::vector<const GTrueInfoData*>& trueInfoData);
 
    bool fillTree(const std::vector<const GDigitizedData*>& digitizedData);
 
    bool fillTree(const GGeneratedParticleBank& particles);
 
    bool fillTree(const GAncestorBank& ancestors);
 
private:
    std::unique_ptr<TTree> root_tree;
 
    std::map<std::string, std::vector<int>> intVarsMap;
 
    std::map<std::string, std::vector<double>> doubleVarsMap;
 
    std::map<std::string, std::vector<std::string>>stringVarsMap;
 
    void registerVariable(const std::string& varname, int value, bool can_ignore_duplicates = false);
 
    void registerVariable(const std::string& varname, double value);
 
    void registerVariable(const std::string& varname, const std::string& value);
 
    std::shared_ptr<GLogger>& log;
};