gTrueInfoData.h

Go to the documentation of this file.

#pragma once
 
#include <atomic>
#include <map>
#include <ostream>
#include <string>
#include <vector>
 
// gemc
#include "gbase.h"
#include "ghit.h"
 
constexpr const char* GTRUEDATA_LOGGER = "true_data";
 
namespace gtrue_data {
 
inline GOptions defineOptions() {
    auto goptions = GOptions(GTRUEDATA_LOGGER);
    return goptions;
}
 
} // namespace gtrue_data
 
class GTrueInfoData : public GBase<GTrueInfoData>
{
public:
    GTrueInfoData(const std::shared_ptr<GOptions>& gopts, const GHit* ghit);
 
    void includeVariable(const std::string& varName, double var);
 
    void includeVariable(const std::string& varName, std::string var);
 
    void accumulateVariable(const std::string& vname, double value);
 
    [[nodiscard]] inline std::map<std::string, double> getDoubleVariablesMap() const {
        return doubleObservablesMap;
    }
 
    [[nodiscard]] inline std::map<std::string, std::string> getStringVariablesMap() const {
        return stringVariablesMap;
    }
 
    static std::unique_ptr<GTrueInfoData> create(const std::shared_ptr<GOptions>& gopts) {
        auto hit            = GHit::create(gopts);
        auto true_info_data = std::make_unique<GTrueInfoData>(gopts, hit);
        auto counter        = globalTrueInfoDataCounter.fetch_add(1, std::memory_order_relaxed);
 
        true_info_data->includeVariable("totalEDeposited", counter * 0.1);
        true_info_data->includeVariable("avgTime", counter * 1.0);
        true_info_data->includeVariable("avgx", counter * 0.01);
        true_info_data->includeVariable("avgy", counter * 0.02);
        true_info_data->includeVariable("avgz", counter * 0.03);
        true_info_data->includeVariable("hitn", counter);
 
        return true_info_data;
    }
 
    [[nodiscard]] inline const std::vector<GIdentifier>& getIdentity() const { return gidentity; }
 
private:
    std::map<std::string, double> doubleObservablesMap;
 
    std::map<std::string, std::string> stringVariablesMap;
 
    std::vector<GIdentifier> gidentity;
 
    static std::atomic<int> globalTrueInfoDataCounter;
 
protected:
    friend std::ostream& operator<<(std::ostream& os, const GTrueInfoData& data);
};