File gt_base_evaluator.h

Location: autonomy/evaluator/gt_base_evaluator/gt_base_evaluator.h

Classes

• simulation_framework::evaluator::GtBaseEvaluator

Namespaces

• simulation_framework
• evaluator
• simulation_framework::evaluator

Includes

• core/kpi/i_evaluator/i_evaluator.h
• <osi_groundtruth.pb.h>

Source

#ifndef SIMFWK_AUTONOMY_AUTONOMY_EVALUATOR_GTBASEEVALUATOR_GTBASEEVALUATOR_H
#define SIMFWK_AUTONOMY_AUTONOMY_EVALUATOR_GTBASEEVALUATOR_GTBASEEVALUATOR_H

#include "core/kpi/i_evaluator/i_evaluator.h"
#include <osi_groundtruth.pb.h>
#include <vector>

namespace simulation_framework
{

namespace evaluator
{

class GtBaseEvaluator : public core::IEvaluator<osi3::GroundTruth>
{
  public:
    GtBaseEvaluator();
    virtual ~GtBaseEvaluator() = default;

    void Init();

    void Reset();

    void Evaluate();

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

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

    void SetInput(const osi3::GroundTruth& ground_truth);

  private:
    std::vector<std::unique_ptr<core::kpi::IKpi<osi3::GroundTruth>>> kpis_;

    osi3::GroundTruth ground_truth_;

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

}  // namespace evaluator
}  // namespace simulation_framework

#endif  // SIMFWK_AUTONOMY_AUTONOMY_EVALUATOR_GTBASEEVALUATOR_GTBASEEVALUATOR_H