S3315_RK3588/NeoTracker/src/Arith_AITTracker.h

#ifndef ARITH_AITRACKER_INTERFACE_H
#define ARITH_AITRACKER_INTERFACE_H

#pragma once
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fstream>
#include "json.hpp"
#include "gd_alg_type.h"
#include "Arith_Common.hpp"

// ai跟踪器在传统算法框架的接口，与其他跟踪器一样提供统一的控制接口。
// AIT_OUTPUT m_Ai_TkOut 是AI跟踪推理的结果，作为本模块的输入
// AIT_Command m_Ai_TkCmd 是本模块决策后对AI跟踪器的控制指令。



class AIT_Interface
{
public:
    explicit AIT_Interface(const char* jsonConf);
    ~AIT_Interface();
    // 建议使用如下接口，反序列化形式接口尽量在上层调用
    AIT_Interface(Param_AITracker para);
    // ai跟踪器初始化
    BBOOL AIT_Init(GD_VIDEO_FRAME_S img, PIPE* pLockPipe, GLB_INPUT* g_GLB_Input);

    // ai跟踪处理流程
    int AIT_Run_Nano(GD_VIDEO_FRAME_S img, CENTERRECT32F srBbox, GLB_INPUT* g_GLB_Input);

    // 输出标准跟踪结果
    void AIT_UpdateTracker(SINT16 nWidth, SINT16 nHeight, OBJECTSTATUS *pObjStatus, GLB_INPUT* g_GLB_Input);

    // 记忆跟踪
    UBYTE8 AIT_MemTracker(OBJECTSTATUS *pObjStatus, SINT32 nWidth, SINT32 nHeight, GLB_INPUT* g_GLB_Input);

    // 外部导入AI跟踪结果,并同步到当前时刻(推理设备->cpu)
    void AIT_SyncInfo(AIT_OUTPUT* out);

    void parse_json_to_struct(const std::string& json_file_path, gud_nano_config_t& config);

    //获取跟踪结果
    OBJECTSTATUS* GetTrackeStatus();

	// 获取跟踪参数
    AIT_OUTPUT* GetAIPara();

    // 解锁
    void AIT_Cancle();

    // 封装Ait运行与管道填充
    BBOOL AIT_Run(GD_VIDEO_FRAME_S img, GLB_INPUT* p_GLB_Input, PIPE* m_LockingPipe);


    bool bAssignflag;  // 已被分配标记

public:
    AIT_OUTPUT m_Ai_TkOut;//ai跟踪器的通用输出,也是本模块的输入

    AIT_Command m_Ai_TkCmd;//本模块的输出，也是外部AI跟踪器的输入、

    PIPE* m_LockingPipe;//锁定管道指针，指向PIPE队列锁定目标，初始化赋值

    OBJECTSTATUS m_ObjStatus;//跟踪结果输出，保持与其他跟踪器一致性

    UBYTE8 m_OccCnt;    //连续遮挡计数

    UBYTE8 m_ArrestCnt;    //连续捕获计数

    unsigned char responeMap[127 * 127] = {0};

 private:
     ArithHandle nTracker;
};

#endif