src/modules/ROOTObjectReader/ROOTObjectReaderModule.hpp

#include <functional>
#include <map>
#include <string>

#include <TFile.h>
#include <TTree.h>

#include "core/config/Configuration.hpp"
#include "core/geometry/GeometryManager.hpp"
#include "core/messenger/Messenger.hpp"
#include "core/module/Event.hpp"
#include "core/module/Module.hpp"

// Contains tuple of all defined objects
#include "objects/objects.h"

namespace allpix {
    class ROOTObjectReaderModule : public Module {
    public:
        using MessageCreatorMap =
            std::map<std::type_index,
                     std::function<std::shared_ptr<BaseMessage>(std::vector<Object*>, std::shared_ptr<Detector>)>>;

        ROOTObjectReaderModule(Configuration& config, Messenger* messenger, GeometryManager* geo_mgr);
        ~ROOTObjectReaderModule() override;

        void initialize() override;

        void run(Event*) override;

        void finalize() override;

    private:
        Messenger* messenger_;
        GeometryManager* geo_mgr_;

        struct message_info {
            std::vector<Object*>* objects;
            std::shared_ptr<Detector> detector;
            std::string name;
            std::shared_ptr<BaseMessage> message;
        };

        // Object names to include or exclude from reading
        std::set<std::string> include_;
        std::set<std::string> exclude_;

        // File containing the objects
        std::unique_ptr<TFile> input_file_;

        // Object trees in the file
        std::vector<TTree*> trees_;

        // List of objects and message information converted from the trees
        std::list<message_info> message_info_array_;

        // Statistics for total amount of objects stored
        std::atomic<unsigned long> read_cnt_{};

        // Internal map to construct an object from it's type index
        MessageCreatorMap message_creator_map_;
    };
} // namespace allpix