#pragma once

#include <QObject>
#include <opencv2/core.hpp>
#include <opencv2/videoio.hpp>
#include "aienginertsplistener.h"
#include "aiengineinference.h"
#include "aienginegimbalcontrol.h"

class AiEngine : public QObject
{
    Q_OBJECT
public:
    explicit AiEngine(QString modelPath, QObject *parent = nullptr);
    void start(void);
    void stop(void);

public slots:
    void frameReceivedSlot(cv::Mat frame);
    void inferenceResultsReceivedSlot(AiEngineInferenceResult result);

signals:
    void inferenceFrame(cv::Mat frame);

private:
    AiEngineRtspListener *mRtspListener;
    AiEngineInference *mInference;
    AiEngineGimbalControl *mGimbalControl;
};