MediaAi.hpp

//
// Created by xiong on 2022/4/27.
//
 
#ifndef EMX_MediaAi_HPP
#define EMX_MediaAi_HPP
 
#include "media/client/MediaParamCBase.hpp"
#include "media/common/MediaInfAiDataCrossingDetection.hpp"
#include "media/common/MediaInfAiDataFaceRecognition.hpp"
#include "media/common/MediaInfAiData.hpp"
#include "media/common/MediaInfAiDataInvasionDetection.hpp"
#include "media/common/MediaInfAiDataLicensePlateRecognition.hpp"
#include "media/common/MediaInfAiDataMotionDetection.hpp"
#include "media/common/MediaInfAiDataPassengerFlowStatistics.hpp"
#include "media/common/MediaInfAiDataPersonDetection.hpp"
#include "media/common/MediaInfAiDataPersonVehicleNonDetection.hpp"
#include "media/common/MediaInfAiDataVehicleRecognition.hpp"
#include "media/common/MediaInfAiDataFaceDetection.hpp"
#include "media/common/MediaInfAiDataElectricBicycleDetection.hpp"
#include "media/common/MediaInfAiDataOffDutyDetection.hpp"
#include "media/common/MediaInfAiDataRegionalPeopleStatistics.hpp"
#include "media/common/MediaInfAiDataBabyCryDetection.hpp"
#include "media/common/MediaInfAiDataPtzTargetTrack.hpp"
#include "media/common/MediaInfAiDataClassify.hpp"
#include "media/common/MediaInfAiDataKeywordSpotting.hpp"
#include "media/common/MediaInfAiDataFlameSmokeDetection.hpp"
#include "media/common/MediaInfAiDataPersonRecognition.hpp"
#include "media/common/MediaInfAi.hpp"
#include "media/common/MediaRingBuffer.hpp"
 
 
namespace Emx {
    class MediaAi : public MediaInfAi, public MediaParamCBase<Json::Value> {
    public:
        explicit MediaAi(int32_t chn) : MediaParamCBase(MediaModE::Ai, chn) {}
 
        ErrCodeE SetParam(const Json::Value &param) override;
 
        ErrCodeE GetParam(Json::Value &param) override;
 
        ErrCodeE GetVersion(std::string &version) override;
 
        ErrCodeE Ctrl(const Json::Value &in, Json::Value &out) override;
    };
 
    class MediaAiDataRecvBase {
    public:
        MediaAiDataRecvBase() {}
 
        virtual ~MediaAiDataRecvBase() {}
 
        using OnGetInfoCallback = std::function<void(MediaInfAiData::Header &header)>;
 
        ErrCodeE GetDataByInfo(const MediaInfAiData::Header &header, MediaInfAiData *data);
 
    protected:
        void OnMsgRecv(uint32_t topic, const char *data, int32_t size);
 
    protected:
        OnGetInfoCallback m_cbInfo;
        MediaRingBufferReader m_ring;
 
        ErrCodeE OnGetBuffer(ErrCodeE e, const uint8_t *buffer, void *usr, int32_t off, int32_t size, uint32_t magicId);
    };
 
    class MediaAiDataRecvAsync : public MediaAiDataRecvBase {
    public:
        MediaAiDataRecvAsync() : MediaAiDataRecvBase() {}
 
        virtual ~MediaAiDataRecvAsync() {}
 
        ErrCodeE Start(EuvLoop *loop, char *buffer, int size, OnGetInfoCallback cbInfo);
 
        void Stop();
 
        void RecvFramePause() { m_sub.PauseRecv(); }
 
        void RecvFrameResume() { m_sub.ResumeRecv(); }
 
        bool IsReceivingFrame() { return m_sub.IsReceiving(); }
 
    private:
        MsgTopicSubscriberAsync m_sub;
    };
 
    class MediaAiDataRecvSync : public MediaAiDataRecvBase {
    public:
        MediaAiDataRecvSync() : MediaAiDataRecvBase() {}
 
        ErrCodeE Open(int32_t timeoutMs);
 
        void Close();
 
        ErrCodeE GetInfo(MediaInfAiData::Header &header, int32_t timeoutMs);
 
    private:
        MsgTopicSubscriberSync m_sub;
    };
 
}// namespace Emx
#endif//EMX_MediaAi_HPP