gstreamerROOTFactory.cc

Go to the documentation of this file.

// gstreamer
#include "gRootTree.h"
#include "gstreamerROOTFactory.h"
 
// keeps the include independent of TROOT
#include <ROOT/RConfig.hxx>
 
// Implementation summary:
// Enable ROOT thread safety at plugin load time and provide lazy tree-creation helpers.
 
namespace {
    struct EnableRootTS
    {
        EnableRootTS() {
            ROOT::EnableThreadSafety();
            std::cout << "GstreamerRootFactory: ROOT thread safety enabled" << std::endl;
        }
    };
 
    static EnableRootTS _enableROOTLocks;
}
 
// Implementation summary:
// Return the event-header tree, creating it on first use.
const std::unique_ptr<GRootTree>& GstreamerRootFactory::getOrInstantiateHeaderTree(
    [[maybe_unused]] const std::unique_ptr<GEventHeader>& event_header) {
    rootfile->cd();
 
    if (!log) {
        std::cerr << "FATAL: log is null in GstreamerRootFactory::getOrInstantiateHeaderTree" << std::endl;
        std::terminate();
    }
 
    if (!event_header) {
        log->error(
            gstreamer::ERR_PUBLISH_ERROR,
            "event header is null in GstreamerRootFactory::getOrInstantiateHeaderTree");
    }
 
    // operator[] either returns the existing entry or creates an empty one.
    auto& treePtr = gRootTrees[gstreamer::root::EVENTHEADERTREENAME];
    if (!treePtr) {
        log->info(2, "GstreamerRootFactoryCreating ROOT%1 tree", gstreamer::root::EVENTHEADERTREENAME);
        treePtr = std::make_unique<GRootTree>(event_header, log);
    }
 
    return treePtr;
}
 
// Implementation summary:
// Return the run-header tree, creating it on first use.
const std::unique_ptr<GRootTree>& GstreamerRootFactory::getOrInstantiateHeaderTree(
    [[maybe_unused]] const std::unique_ptr<GRunHeader>& run_header) {
    rootfile->cd();
 
    if (!log) {
        std::cerr << "FATAL: log is null in GstreamerRootFactory::getOrInstantiateHeaderTree" << std::endl;
        std::terminate();
    }
 
    if (!run_header) {
        log->error(
            gstreamer::ERR_PUBLISH_ERROR,
            "run header is null in GstreamerRootFactory::getOrInstantiateHeaderTree");
    }
 
    auto& treePtr = gRootTrees[gstreamer::root::RUNHEADERTREENAME];
    if (!treePtr) {
        log->info(2, "GstreamerRootFactoryCreating ROOT%1 tree", gstreamer::root::RUNHEADERTREENAME);
        treePtr = std::make_unique<GRootTree>(run_header, log);
    }
 
    return treePtr;
}
 
// Implementation summary:
// Return the true-information detector tree, creating it lazily from the first sample hit.
const std::unique_ptr<GRootTree>& GstreamerRootFactory::getOrInstantiateTrueInfoDataTree(
    const std::string&   detectorName,
    const GTrueInfoData* gdata) {
    rootfile->cd();
    std::string treeName = gstreamer::root::TRUEINFONAMEPREFIX + detectorName;
 
    auto& treePtr = gRootTrees[treeName];
    if (!treePtr) {
        log->info(2, "GstreamerRootFactoryCreating GTrueInfoData ROOT tree for %1", detectorName);
        treePtr = std::make_unique<GRootTree>(treeName, gdata, log);
    }
 
    return treePtr;
}
 
// Implementation summary:
// Return the digitized detector tree, creating it lazily from the first sample hit.
const std::unique_ptr<GRootTree>& GstreamerRootFactory::getOrInstantiateDigitizedDataTree(
    const std::string&    detectorName,
    const GDigitizedData* gdata) {
    rootfile->cd();
    std::string treeName = gstreamer::root::DIGITIZEDNAMEPREFIX + detectorName;
 
    auto& treePtr = gRootTrees[treeName];
    if (!treePtr) {
        log->info(2, "GstreamerRootFactoryCreating GDigitizedData ROOT tree for %1", detectorName);
        treePtr = std::make_unique<GRootTree>(treeName, gdata, log);
    }
 
    return treePtr;
}
 
const std::unique_ptr<GRootTree>& GstreamerRootFactory::getOrInstantiateGeneratedParticleTree(
    const std::string& treeName,
    const GGeneratedParticleBank& particles) {
    rootfile->cd();
    auto& treePtr = gRootTrees[treeName];
    if (!treePtr) {
        log->info(2, "GstreamerRootFactoryCreating generated-particle ROOT tree for %1", treeName);
        treePtr = std::make_unique<GRootTree>(treeName, particles, log);
    }
 
    return treePtr;
}
 
const std::unique_ptr<GRootTree>& GstreamerRootFactory::getOrInstantiateAncestorTree(
    const GAncestorBank& ancestors) {
    rootfile->cd();
    auto& tree_ptr = gRootTrees[gstreamer::root::ANCESTORTREENAME];
    if (!tree_ptr) {
        log->info(2, "GstreamerRootFactoryCreating ancestor ROOT tree");
        tree_ptr = std::make_unique<GRootTree>(ancestors, log);
    }
    return tree_ptr;
}
 
 
// Implementation summary:
// Export the factory symbol required by the plugin loader.
extern "C" GStreamer* GStreamerFactory(const std::shared_ptr<GOptions>& g) {
    return static_cast<GStreamer*>(new GstreamerRootFactory(g));
}