gEventDataCollection.h

Go to the documentation of this file.

#pragma once
 
#include "gEventHeader.h"
#include "gDataCollection.h"
 
// C++
#include <map>
#include <string>
 
constexpr const char* GEVENTDATA_LOGGER = "gevent_data";
 
namespace gevent_data {
inline GOptions defineOptions() {
    auto goptions = GOptions(GEVENTDATA_LOGGER);
    goptions += gheader::defineOptions();
    goptions += gtrue_data::defineOptions();
    goptions += gdigi_data::defineOptions();
    goptions += gtouchable::defineOptions();
    return goptions;
}
}
 
 
class GEventDataCollection : public GBase<GEventDataCollection> {
public:
    GEventDataCollection(const std::shared_ptr<GOptions>& gopts, std::unique_ptr<GEventHeader> header)
        : GBase(gopts, GDATAHEADER_LOGGER), gheader(std::move(header)) { }
 
    void addDetectorTrueInfoData(const std::string& sdName, std::unique_ptr<GTrueInfoData> data);
 
    void addDetectorDigitizedData(const std::string& sdName, std::unique_ptr<GDigitizedData> data);
 
    [[nodiscard]] inline const std::unique_ptr<GEventHeader>& getHeader() const { return gheader; }
 
    [[nodiscard]] inline const std::map<std::string, std::unique_ptr<GDataCollection>>& getDataCollectionMap() const { return gdataCollectionMap; }
 
    [[nodiscard]] inline int getEventNumber() const { return gheader->getG4LocalEvn(); }
 
 
    // returning shared here as GEventDataCollection may be used by multiple streams and also collected in a runData vector
    static std::shared_ptr<GEventDataCollection> create(const std::shared_ptr<GOptions>& gopts) {
        auto header = GEventHeader::create(gopts);
        auto edc    = std::make_shared<GEventDataCollection>(gopts, std::move(header));
 
        auto digi_data = GDigitizedData::create(gopts);
        auto true_data = GTrueInfoData::create(gopts);
 
        edc->addDetectorDigitizedData("ctof", std::move(digi_data));
        edc->addDetectorTrueInfoData("ctof", std::move(true_data));
 
        return edc;
    }
 
private:
    std::unique_ptr<GEventHeader>                           gheader;
    std::map<std::string, std::unique_ptr<GDataCollection>> gdataCollectionMap;
 
    static std::atomic<int> globalEventDataCollectionCounter;
 
};