gEventDataCollection.h

Go to the documentation of this file.

#pragma once
 
#include "gDataCollection.h"
#include "gEventHeader.h"
 
// C++
#include <map>
#include <string>
 
constexpr const char* GEVENTDATA_LOGGER = "gevent_data";
 
namespace gevent_data {
 
inline auto defineOptions() -> GOptions {
    auto goptions = GOptions(GEVENTDATA_LOGGER);
    goptions      += geventheader::defineOptions();
    goptions      += gtrue_data::defineOptions();
    goptions      += gdigi_data::defineOptions();
    goptions      += gtouchable::defineOptions();
    return goptions;
}
 
} // namespace gevent_data
 
class GEventDataCollection : public GBase<GEventDataCollection>
{
public:
    GEventDataCollection(const std::shared_ptr<GOptions>& gopts, std::unique_ptr<GEventHeader> header)
        : GBase(gopts, GEVENTDATA_LOGGER), gevent_header(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]] auto getHeader() const -> const std::unique_ptr<GEventHeader>& { return gevent_header; }
 
    [[nodiscard]] auto getDataCollectionMap() const -> const std::map<std::string, std::unique_ptr<GDataCollection>>& {
        return gdataCollectionMap;
    }
 
    [[nodiscard]] auto getEventNumber() const -> int { return gevent_header->getG4LocalEvn(); }
 
    static auto create(const std::shared_ptr<GOptions>& gopts) -> std::shared_ptr<GEventDataCollection> {
        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> gevent_header;
 
    std::map<std::string, std::unique_ptr<GDataCollection>> gdataCollectionMap;
 
    static std::atomic<int> globalEventDataCollectionCounter;
};