gstreamerROOTFactory.h

Go to the documentation of this file.

#pragma once
 
// gstreamer
#include "gstreamer.h"
#include "gRootTree.h"
 
// ROOT
#include "TFile.h"
 
class GstreamerRootFactory : public GStreamer
{
public:
    using GStreamer::GStreamer;
 
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 publishRunHeaderImpl([[maybe_unused]] const std::unique_ptr<GRunHeader>& run_header) 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;
 
    // Private helper methods used to lazily create trees on first use.
    const std::unique_ptr<GRootTree>& getOrInstantiateHeaderTree(
        [[maybe_unused]] const std::unique_ptr<GEventHeader>& event_header);
    const std::unique_ptr<GRootTree>& getOrInstantiateHeaderTree(
        [[maybe_unused]] const std::unique_ptr<GRunHeader>& run_header);
    const std::unique_ptr<GRootTree>& getOrInstantiateTrueInfoDataTree(const std::string& treeName,
                                                                       const GTrueInfoData* gdata);
    const std::unique_ptr<GRootTree>& getOrInstantiateDigitizedDataTree(const std::string& treeName,
                                                                        const GDigitizedData* gdata);
    const std::unique_ptr<GRootTree>& getOrInstantiateGeneratedParticleTree(const std::string& treeName,
                                                                            const GGeneratedParticleBank& particles);
    const std::unique_ptr<GRootTree>& getOrInstantiateAncestorTree(const GAncestorBank& ancestors);
 
    std::unordered_map<std::string, std::unique_ptr<GRootTree>> gRootTrees;
 
    std::unique_ptr<TFile> rootfile;
 
    [[nodiscard]] std::string filename() const override { return gstreamer_definitions.rootname + ".root"; }
};