src/modules/CorryvreckanWriter/CorryvreckanWriterModule.hpp

#include <string>

#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"

// Local includes
#include "corryvreckan/Event.hpp"
#include "corryvreckan/MCParticle.hpp"
#include "corryvreckan/Object.hpp"
#include "corryvreckan/Pixel.hpp"
#include "objects/PixelHit.hpp"

// ROOT includes
#include "TFile.h"
#include "TTree.h"

namespace allpix {
    class CorryvreckanWriterModule : public SequentialModule {
    public:
        CorryvreckanWriterModule(Configuration& config, Messenger* messenger, GeometryManager* geo_manager);

        void initialize() override;

        void run(Event* event) override;

        void finalize() override;

    private:
        // General module members
        Messenger* messenger_;
        GeometryManager* geometryManager_;

        // Parameters for output writing
        std::string fileName_;               // Output filename
        std::string geometryFileName_;       // Output geometry filename
        std::unique_ptr<TFile> output_file_; // Output file
        double time_{};                      // Event time being written
        bool output_mc_truth_{};             // Decision to write out MC
        bool timing_global_{};               // Decision to write global or local timestamps

        std::string reference_;
        std::vector<std::string> dut_;

        std::unique_ptr<TTree> event_tree_;
        corryvreckan::Event* event_{};

        std::unique_ptr<TTree> pixel_tree_;
        std::unique_ptr<TTree> mcparticle_tree_;
        std::map<std::string, std::vector<corryvreckan::Pixel*>*> write_list_px_;
        std::map<std::string, std::vector<corryvreckan::MCParticle*>*> write_list_mcp_;
    };
} // namespace allpix