MockCupDetector.hpp

#pragma once

#include <Vision/ICupDetector.hpp>

namespace HLR
{
namespace Vision
{
class MockCupDetector : public ICupDetector
{
public:
    MockCupDetector();

    ~MockCupDetector() override = default;

    MockCupDetector(MockCupDetector&& cup_detector) = default;

    MockCupDetector(const MockCupDetector& cup_detector) = default;

    MockCupDetector& operator=(const MockCupDetector& cup_detector) = default;

    MockCupDetector& operator=(MockCupDetector&& cup_detector) = default;

    CupDetection detect_cups() override;

    std::shared_ptr<cv::Mat> get_detected_cups_image() override;

    std::shared_ptr<cv::Mat> get_latest_depth_frame() override;

    std::shared_ptr<cv::Mat> get_latest_ir_frame() override;

    std::shared_ptr<cv::Mat> get_latest_color_frame() override;

private:
    CupDetection detection;
    std::shared_ptr<cv::Mat> depth;
    std::shared_ptr<cv::Mat> ir;
    std::shared_ptr<cv::Mat> color;
};
} // namespace Vision
} // namespace HLR