UtilityEvaluatorSnapshot.h

// Copyright QED Software 2023.
 
#ifndef GRAIL_UTILITY_EVALUATOR_SNAPSHOT_H
#define GRAIL_UTILITY_EVALUATOR_SNAPSHOT_H
 
#include "EvaluationDebugData.h"
#include "Flatbuffers/UtilityEvaluatorSnapshot_generated.h"
 
#include <string>
#include <vector>
#include <map>
 
namespace YAML
{
 template <typename DataType>
 struct convert;
}
 
namespace grail
{
namespace data
{
 class UtilityEvaluatorSnapshot
 {
 friend struct YAML::convert<UtilityEvaluatorSnapshot>;
 public:
 UtilityEvaluatorSnapshot() = default;
 UtilityEvaluatorSnapshot(const std::string& evaluatedObjectName, const std::string& metadata, int rank = 0);
 
 void AddEvaluation(EvaluationDebugData&& debugData);
 
 unsigned int AddNode(const void* nodePointer);
 
 const std::vector<EvaluationDebugData>& GetEvaluations() const;
 
 const std::string& GetEvaluatedObjectName() const;
 
 const std::string& GetEvaluatedObjectMetadata() const;
 
 const int& GetRank() const;
 
 const std::map<const void*, unsigned int>& GetNodeMapping() const;
 
 private:
 std::vector<EvaluationDebugData> evaluations{};
 std::map<const void*, unsigned int> nodeMapping{};
 unsigned int nextIndex{0};
 std::string evaluatedObjectName{};
 std::string evaluatedObjectMetadata{};
 int rank{0};
 };
 
 flatbuffers::Offset<generated::UtilityEvaluatorSnapshot> Pack(flatbuffers::FlatBufferBuilder& builder, const UtilityEvaluatorSnapshot& snapshot);
 UtilityEvaluatorSnapshot Unpack(const generated::UtilityEvaluatorSnapshot& snapshot);
}
}
 
#endif