MIVAS/modules/userApp/include/user_app.h

/*
 * @Description: USERAPP
 * @Version: 1.0
 * @Autor: lishengyin
 * @Date: 2021-10-13 09:41:50
 * @LastEditors: lishengyin
 * @LastEditTime: 2021-10-15 15:57:17
 */
#pragma once

#include <iostream>
#include <signal.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 "inifile.h"

#include <gst/gst.h>
#include <vector>
#include <map>
#include <time.h>
#include "inference.h"
#include "DataSource.h"

// dataType
#include "SendLogin.h"
#include "NettyServerCommandEnum.h"
#include "NettyServerResultMsg.h"
#include "NettyClientResultMsg.h"
#include "SendLogin.h"
#include "NettyClientCommandEnum.h"
#include "InferData.h"
#include "PIDSClientResultMsg.h"
#include "PIDSServerResultMsg.h"

#include "Notices.h"

using namespace toolkit;
using namespace mediakit;
using namespace inifile;

using namespace std;


namespace MIVA
{  
    class UserApp
    {
    public:
        /**
         * @description: 创建对象
         * @param {*}
         * @return {std::shared_ptr<UserApp>} 智能指针
         * @author: lishengyin
         */    
        static std::shared_ptr<UserApp> CreateNew();
        UserApp();
        ~UserApp();
        /**
         * @description: 模块初始化
         * @param {string} appName 应用名
         * @return {int} 是否成功
         * @author: lishengyin
         */        
        int32_t Init(std::string appName);
        /**
         * @description: 释放数据
         * @param {*}
         * @return {*}
         * @author: lishengyin
         */        
        void Destroy();
        /**
         * @description: 启动任务
         * @param {*}
         * @return {*}
         * @author: lishengyin
         */        
        void StartTask();
        /**
         * @description: 赋值struct sockaddr
         * @param {sockaddr} *out
         * @param {char} *ip
         * @param {uint16_t} port
         * @return {*}
         * @author: lishengyin
         */        
        static void makeAddr(struct sockaddr *out,const char *ip,uint16_t port);
        /**
         * @description: 最终的结果运算
         * @param {uint32_t} noLoad 空载车厢载荷
         * @param {uint32_t} dynamicLoad 动态车厢载荷
         * @param {int} num 推理结果
         * @return {uint32_t} 运算结构
         * @author: lishengyin
         */        
        static uint32_t DataCalculation(uint32_t noLoad, uint32_t dynamicLoad, int num);
        /**
         * @description: 拥挤成都判定
         * @param {int} num 识别结果
         * @return {uint8_t} 拥挤程度 1-4
         * @author: lishengyin
         */        
        static uint8_t GradeDetermination(int num);
    private:
        IniFile m_ini;
        std::string m_appName;
        const std::string m_configSrc = "../config/config.ini";
    private:
        // Natty
        std::string Netty_ip;
        int32_t Netty_port;
        std::string Netty_idName;

        // PIS
        std::string PIS_ip;
        int32_t PIS_port;
        std::string PIS_IdName;

        // Mysql
        std::string sql_ip;
        int32_t sql_port;
        std::string user;
        std::string password;
        std::string character;

        int device_id;
        bool play = false;
        bool start = true;
    private:
        std::shared_ptr<TCPClient> m_tcpClient = NULL;
        std::shared_ptr<Inference> m_Infer = NULL;
        Socket::Ptr m_udpClient = NULL;

        std::vector<DataSource> DataList;
        Timer::Ptr m_timer0;
        Timer::Ptr m_timer1;
        Timer::Ptr m_timer2;
        
    private:
        /**
         * @description: 监听推理模块广播
         * @param {int} Source_id 数据源ID
         * @param {int} num 推理结果
         * @return {*}
         * @author: lishengyin
         */        
        void ListenInfer(int Source_id, int num);
        /**
         * @description: 监听Netty下发的数据
         * @param {Ptr} &pBuf 接受到的数据
         * @return {*}
         * @author: lishengyin
         */        
        void ListenNettyData(const Buffer::Ptr &pBuf);
        /**
         * @description: 监听PIS系统的数据
         * @param {Ptr} &buf 接受到的数据
         * @param {sockaddr} *addr socker信息
         * @return {*}
         * @author: lishengyin
         */        
        void ListenPISData(const Buffer::Ptr &buf, struct sockaddr *addr);
        /**
         * @description: 监听车厢关门广播
         * @param {*}
         * @return {*}
         * @author: lishengyin
         */        
        void ListenClosed();
        /**
         * @description: 监听重启识别广播
         * @param {*}
         * @return {*}
         * @author: lishengyin
         */        
        void ListenRelease();
        /**
         * @description: 监听推理数据广播
         * @param {*}
         * @return {*}
         * @author: lishengyin
         */        
        void ListenInferData();
        /**
         * @description: 连接Netty
         * @param {*}
         * @return {*}
         * @author: lishengyin
         */        
        void ConnectNetty();
        /**
         * @description: 上报数据
         * @param {*}
         * @return {*}
         * @author: lishengyin
         */        
        void ReportData();
    };
}