Evaluator.hh

#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 utility
    {
        template <typename ContextType>
        class Evaluator
        {
        public:
            float EvaluateContext(const ContextType& context, UtilityEvaluatorSnapshot* const snapshot) const
            {
                float score = Evaluate(context, snapshot);
                if(snapshot != nullptr)
                {
                    std::size_t nodeIndex = snapshot->AddNode(this);
                    EvaluationDebugData debugData{GetEvaluatorType(), score, nodeIndex};
                    DebugDump(snapshot->GetNodeMapping(), debugData);
                    snapshot->AddEvaluation(std::move(debugData));
                }
                return score;
            }
 
            virtual EvaluatorType GetEvaluatorType() const =0;
 
        protected:
            virtual float Evaluate(const ContextType& context, UtilityEvaluatorSnapshot* const snapshot) const =0;
 
            virtual void DebugDump(const std::map<const void*, std::size_t>& nodeMapping,
                                   EvaluationDebugData& debugData) const =0;
        };
    }
}
#endif  //GRAIL_EVALUATOR_H