gstreamerCSVFactory.h

Go to the documentation of this file.

#pragma once
 
// gstreamer
#include "gstreamer.h"
 
// c++
#include <fstream>
 
class GstreamerCsvFactory : public GStreamer
{
public:
    using GStreamer::GStreamer;
 
    GstreamerCsvFactory(const GstreamerCsvFactory&)            = delete;
    GstreamerCsvFactory& operator=(const GstreamerCsvFactory&) = delete;
    GstreamerCsvFactory(GstreamerCsvFactory&&)                 = delete;
    GstreamerCsvFactory& operator=(GstreamerCsvFactory&&)      = delete;
 
private:
    bool openConnection() override;
 
    bool closeConnectionImpl() override;
 
    bool startEventImpl([[maybe_unused]] const std::shared_ptr<GEventDataCollection>& event_data) override;
 
    bool endEventImpl([[maybe_unused]] const std::shared_ptr<GEventDataCollection>& event_data) override;
 
    bool publishEventHeaderImpl([[maybe_unused]] const std::unique_ptr<GEventHeader>& gevent_header) override;
 
    bool publishEventTrueInfoDataImpl(const std::string& detectorName,
                                      const std::vector<const GTrueInfoData*>& trueInfoData) override;
 
    bool publishEventDigitizedDataImpl(const std::string& detectorName,
                                       const std::vector<const GDigitizedData*>& digitizedData) override;
 
    bool publishEventGeneratedParticlesImpl(const std::string& bankName,
                                            const GGeneratedParticleBank& particles) override;
 
    bool publishEventAncestorsImpl(const GAncestorBank& ancestors) override;
 
    bool startRunImpl([[maybe_unused]] const std::shared_ptr<GRunDataCollection>& run_data) override;
 
    bool endRunImpl([[maybe_unused]] const std::shared_ptr<GRunDataCollection>& run_data) override;
 
    bool publishRunDigitizedDataImpl(const std::string& detectorName,
                                     const std::vector<const GDigitizedData*>& digitizedData) override;
 
    bool startStreamImpl(const GFrameDataCollection* frameRunData) override;
 
    bool endStreamImpl(const GFrameDataCollection* frameRunData) override;
 
    bool publishFrameHeaderImpl(const GFrameHeader* gframeHeader) override;
 
    bool publishPayloadImpl(const std::vector<GIntegralPayload*>* payload) override;
 
    std::ofstream ofile_true_info;
 
    std::ofstream ofile_digitized;
 
    std::ofstream ofile_generated;
 
    std::ofstream ofile_generated_tracked;
 
    std::ofstream ofile_ancestors;
 
    [[nodiscard]] std::string filename() const override { return gstreamer_definitions.rootname + ".csv"; }
 
    [[nodiscard]] std::string filename_true_info() const { return gstreamer_definitions.rootname + "_true_info.csv"; }
 
    [[nodiscard]] std::string filename_digitized() const { return gstreamer_definitions.rootname + "_digitized.csv"; }
 
    [[nodiscard]] std::string filename_generated() const { return gstreamer_definitions.rootname + "_generated.csv"; }
 
    [[nodiscard]] std::string filename_generated_tracked() const {
        return gstreamer_definitions.rootname + "_generated_tracked.csv";
    }
 
    [[nodiscard]] std::string filename_ancestors() const {
        return gstreamer_definitions.rootname + "_ancestors.csv";
    }
 
    std::ofstream& generated_stream_for_bank(const std::string& bankName);
 
    bool is_first_event_with_truedata = false;
 
    bool is_first_event_with_digidata = false;
 
    std::string timestamp;
 
    int event_number = -1;
 
    int runId = -1;
 
    int thread_id = -1;
};