#pragma once

#include <QObject>
#include <QString>
#include <opencv2/core.hpp>


class AiEngineInferenceResult {
public:
    cv::Mat frame;
    int objects;
};


class AiEngineInference : public QObject
{
    Q_OBJECT
public:
    explicit AiEngineInference(QString modelPath, QObject *parent = nullptr);
    bool isActive(void);

protected:
    QString mModelPath;
    bool mActive;

public slots:
    virtual void performInferenceSlot(cv::Mat frame) = 0;

signals:
    void resultsReady(AiEngineInferenceResult results);
};