gDataCollection.h

Go to the documentation of this file.

#pragma once
 
#include "gTrueInfoData.h"
#include "gDigitizedData.h"
 
// c++
#include <memory>
#include <vector>
 
class GDataCollection
{
public:
    explicit GDataCollection() = default;
 
    ~GDataCollection() = default;
 
    void collectTrueInfosData(const std::unique_ptr<GTrueInfoData>& data) {
        // first event
        if (trueInfosData.empty()) {
            trueInfosData.push_back(std::make_unique<GTrueInfoData>(*data));
        }
        else {
            for (const auto& [varName, value] : data->getDoubleVariablesMap()) {
                trueInfosData.front()->accumulateVariable(varName, value);
            }
        }
    }
 
    void collectDigitizedData(const std::unique_ptr<GDigitizedData>& data) {
        // first event
        if (digitizedData.empty()) {
            digitizedData.push_back(std::make_unique<GDigitizedData>(*data));
        }
        else {
            // argument passed: 0: do not get SRO var
            for (const auto& [varName, value] : data->getIntObservablesMap(0)) {
                digitizedData.front()->accumulateVariable(varName, value);
            }
            for (const auto& [varName, value] : data->getDblObservablesMap(0)) {
                digitizedData.front()->accumulateVariable(varName, value);
            }
        }
    }
 
    void addDigitizedData(std::unique_ptr<GDigitizedData> data) {
        digitizedData.push_back(std::move(data)); // taking ownership of the unique_ptr
    }
 
    void addTrueInfoData(std::unique_ptr<GTrueInfoData> data) {
        trueInfosData.push_back(std::move(data)); // taking ownership of the unique_ptr
    }
 
    [[nodiscard]] auto getTrueInfoData() const -> const std::vector<std::unique_ptr<GTrueInfoData>>& {
        return trueInfosData;
    }
 
    [[nodiscard]] auto getDigitizedData() const -> const std::vector<std::unique_ptr<GDigitizedData>>& {
        return digitizedData;
    }
 
private:
    std::vector<std::unique_ptr<GTrueInfoData>> trueInfosData;
 
    std::vector<std::unique_ptr<GDigitizedData>> digitizedData;
};