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"
 
namespace gstreamer::root {
inline constexpr char EVENTHEADERTREENAME[] = "event_header";
inline constexpr char RUNHEADERTREENAME[] = "run_header";
inline constexpr char TRUEINFONAMEPREFIX[] = "true_info_";
inline constexpr char DIGITIZEDNAMEPREFIX[] = "digitized_";
inline constexpr char GENERATEDTREENAMEDESC[] = "Generated Particles";
inline constexpr char ANCESTORTREENAME[] = "ancestors";
inline constexpr char ANCESTORTREENAMEDESC[] = "Geant4 Ancestors";
inline constexpr int ERR_GSTREAMERROOTTREENOTFOUND = 850;
inline constexpr char EVENTHEADERTREENAMEDESC[] = "Event Header";
inline constexpr char RUNHEADERTREENAMEDESC[] = "Run Header";
inline constexpr char TRUEINFOTREENAMEDESC[] = "True Info Data";
inline constexpr char DIGITIZEDTREENAMEDESC[] = "Digitized Data";
} // namespace gstreamer::root


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;
};