Evaluator.hh

// Copyright QED Software 2023.
 
#ifndef GRAIL_EVALUATOR_H
#define GRAIL_EVALUATOR_H
 
#include "../GrailData/DebugInfo/UtilityEvaluatorSnapshot.h"
#include "../GrailData/UtilityModel/EvaluatorType.h"
 
#include <cassert>
#include <limits>
#include <memory>
#include <vector>
 
namespace grail
{
namespace evaluator
{
    template<typename ContextType>
    class Consideration;
 
    template <typename ContextType>
    class Evaluator
    {
    public:
        float EvaluateContext(const ContextType& context, data::UtilityEvaluatorSnapshot* const snapshot) const
        {
            float score = Evaluate(context, snapshot);
            if(snapshot != nullptr)
            {
                unsigned int nodeIndex = snapshot->AddNode(this);
                data::EvaluationDebugData debugData{GetEvaluatorType(), score, nodeIndex};
                DebugDump(snapshot->GetNodeMapping(), debugData);
                snapshot->AddEvaluation(std::move(debugData));
            }
            return score;
        }
 
        virtual data::EvaluatorType GetEvaluatorType() const =0;
        virtual void GetDependentConsiderations(std::vector<Consideration<ContextType>*>& considerations) = 0;
 
    protected:
        virtual float Evaluate(const ContextType& context, data::UtilityEvaluatorSnapshot* const snapshot) const =0;
 
        virtual void DebugDump(const std::map<const void*, unsigned int>& nodeMapping,
                               data::EvaluationDebugData& debugData) const =0;
    };
}
}
#endif  //GRAIL_EVALUATOR_H