MIVAS/modules/inference/include/inference.h

/*
 * @Description: 推理模块
 * @Version: 1.0
 * @Autor: lishengyin
 * @Date: 2021-10-13 09:37:51
 * @LastEditors: lishengyin
 * @LastEditTime: 2021-10-13 09:40:21
 */
#pragma once
#include <iostream>
#include <stdio.h>
#include <string.h>

#include <gst/gst.h>
#include <glib.h>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <sys/time.h>
#include "gstnvdsmeta.h"
#include "gst-nvmessage.h"

#include "DataSource.h"

// ZLMediaKit
#include "Util/logger.h"
#include "Util/NoticeCenter.h"
#include "Poller/EventPoller.h"
#include "Player/PlayerProxy.h"
#include "Rtmp/RtmpPusher.h"
#include "Common/config.h"
#include "Pusher/MediaPusher.h"
#include "Extension/Frame.h"
#include "Util/SqlPool.h"
#include "Network/TcpClient.h"
#include "Poller/Timer.h"
#include "TCPClient.h"
#include <mutex>

#include <list>

using namespace std;

namespace MIVA{
    // 基于Deepstream
    class Inference
    {
    private:
        // Deepstream
        GMainLoop *loop = NULL;
        GstElement *pipeline = NULL,*streammux =NULL, *sink = NULL, *pgie = NULL,
        *queue1, *queue2, *queue3, *queue4, *queue5, *nvvidconv = NULL,
        *nvosd = NULL, *tiler = NULL;

        GstElement *transform = NULL;

        GstBus *bus = NULL;
        guint bus_watch_id;
        GstPad *tiler_src_pad = NULL;
        guint tiler_rows, tiler_columns;
        guint pgie_batch_size;

        std::vector<DataSource> m_DataList;

    public:
        static std::shared_ptr<Inference> CreateNew();
        Inference();
        ~Inference();
    public:
        
        int32_t Init(std::vector<DataSource> DataList);
        void Destory();
        void ReadyTask();
        void StartTask();
        void PauseTask();
        void StopTask();
        void RestartTask();
        

    private:
        // static 
        static GstPadProbeReturn tiler_src_pad_buffer_probe(GstPad * pad, GstPadProbeInfo * info, gpointer u_data);
        static gboolean bus_call (GstBus * bus, GstMessage * msg, gpointer data);
        static void decodebin_child_added (GstChildProxy * child_proxy, GObject * object,
            gchar * name, gpointer user_data);

        static GstElement * create_uridecode_bin (guint index, gchar * filename);
        static void cb_newpad (GstElement * decodebin, GstPad * pad, gpointer data);

        // 动态管理源
        gboolean add_sources (int source_Id, std::string uri);
        void stop_release_source (gint source_id);
    };
}