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 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;
 
 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