gt_base_evaluator

1 

5 

6 #pragma once

7 

8 #include "core/kpi/i_evaluator/i_evaluator.h"

9 #include <avx_osi_groundtruth.pb.h>

10 #include <vector>

11 

12 namespace simulation_framework

13 {

14 namespace evaluator

15 {

122 class GtBaseEvaluator : public core::evaluator::IEvaluator<avx_osi3::GroundTruth>

123 {

124  public:

125  GtBaseEvaluator();

126  virtual ~GtBaseEvaluator() = default;

127 

131  void Init();

132 

136  void Reset();

137 

141  void Evaluate();

142 

147  void AddKpi(std::unique_ptr<core::kpi::IKpi<avx_osi3::GroundTruth>> kpi) override;

148 

153  core::kpi::KpiMessage GetKpiMessage() const;

154 

159  void SetInput(const avx_osi3::GroundTruth& ground_truth);

160 

161  private:

164  std::vector<std::unique_ptr<core::kpi::IKpi<avx_osi3::GroundTruth>>> kpis_;

165 

167  avx_osi3::GroundTruth ground_truth_;

168 

170  core::kpi::KpiMessage kpi_message_{};

171 };

172 

173 } // namespace evaluator

174 } // namespace simulation_framework