EvaluationDebugData.h

// Copyright QED Software 2023.
 
#ifndef GRAIL_EVALUATION_DEBUG_DATA_H
#define GRAIL_EVALUATION_DEBUG_DATA_H
 
#include <string>
#include <vector>
#include "../UtilityModel/EvaluatorType.h"
 
namespace YAML
{
    template <typename DataType>
    struct convert;
}
 
namespace grail
{
namespace data
{
    class EvaluationDebugData
    {
        friend struct YAML::convert<EvaluationDebugData>;
    public:
        EvaluationDebugData(EvaluatorType evaluatorType, float score, std::size_t nodeIndex);
        EvaluationDebugData() = default;
 
        void AddChildNode(std::size_t childNodeIndex);
 
        EvaluatorType GetEvaluatorType() const;
 
        float GetScore() const;
 
        std::size_t GetNodeIndex() const;
 
        const std::vector<std::size_t>& GetChildNodes() const;
 
        const std::string& GetMetadata() const;
 
        void SetMetadata(const std::string& newMetadata);
 
    private:
        EvaluatorType evaluatorType{};
        float score{};
        std::size_t nodeIndex{};
        std::vector<std::size_t> childNodes{};
        std::string metadata{};
    };
}
}
 
#endif