StdTracker_Board_App/sophon_bm1684x/NeoTracker/tests/utils.h

#pragma once
#include <opencv2/opencv.hpp>
#include "NeoArithStandardDll.h"
class SimTargetImage_Y16
{

public:
	SimTargetImage_Y16(int w,int h);
	~SimTargetImage_Y16();


public:
	void setBackGround(int gray,int std);
	void addTarget(int x, int y, int w, int h, int gray);

	// 获取Y16数据
	unsigned short* getImageData();

	cv::Mat getMatRGB();
private:
	cv::Mat Y8Mat;//显示图像
	unsigned short* pSrc;
	int nImageWidth;
	int nImageHeight;
};




// 目标结构体
struct Target {
    float x, y;            	// 目标位置
    float width, height;   	// 目标尺寸
    float vx, vy;          	// 水平和垂直速度
	float vw, vh;          	// 尺寸变化速度
    cv::Scalar color;    	// 目标颜色（适用于 RGB）
	bool useTexture;			// 使用贴图

	cv::Mat texture_cur;			// 贴图
	cv::Mat alaph_cur;				// 贴图alaph

	// 保存原始贴图状态
	cv::Mat texture;			// 贴图
	cv::Mat alaph;				// 贴图alaph

	Target()
	{

	};
	

    void updatePosition(int frameWidth, int frameHeight) 
	{
        // 更新位置
        x += vx;
        y += vy;

		width += vw;
		height += vh;

        // 检测边界并反弹
        if (x <= width || x + width >= frameWidth) vx = -vx;
        if (y <= height || y + height >= frameHeight) vy = -vy;

        // 保持目标在图像范围内
        x = std::max(0.0f, std::min(x, frameWidth - width));
        y = std::max(0.0f, std::min(y, frameHeight - height));
    }

	void updateTexture()
	{
		if(useTexture && (texture_cur.cols != height || texture_cur.rows != width))
		{
			texture_cur = cv::Mat(height,width,CV_8UC3);
			alaph_cur = cv::Mat(height,width,CV_8UC1);
			cv::resize(texture,texture_cur,cv::Size(width,height));
			cv::resize(alaph,alaph_cur,cv::Size(width,height));
		}
	}


	void addTexture(unsigned char* png_data,int Len)
	{
		std::vector<uchar> img_data(png_data, png_data + Len);
		cv::Mat pic = cv::imdecode(img_data, cv::IMREAD_UNCHANGED);

		// cv::imshow("pic",pic);
		// cv::waitKey(0);

		// 分离前景的 RGB 和 Alpha 通道
		cv::Mat bgr, alpha;
		std::vector<cv::Mat> channels(4);
		cv::split(pic, channels);
		cv::merge(std::vector<cv::Mat>{channels[0], channels[1], channels[2]}, bgr);


		useTexture = true;
		alaph = channels[3]; // Alpha 通道
		texture = bgr;

		alaph_cur = channels[3]; 
		texture_cur = bgr;
	}

};


class SimTargetImage
{
public:
	SimTargetImage(int w,int h);
	~SimTargetImage(){};

public:

	// 状态更新
	void update();

	// 设置背景
	void setBackGround(int gray,int std);

	// 增加一个目标
	void addTarget(Target t);

	// 增加一个遮挡
	void addOcc(Target t);

public:
	cv::Mat getImageRGB();
	cv::Mat getImageY16();
	cv::Mat getImageY8();
	cv::Mat getImageNV12();

	Target* getTarget(int id);


private:
	void drawObj(Target t);
	void drawOcc(Target t);
	cv::Mat m_BaseMat;//图像帧(RGB)
	cv::Mat m_CurMat;//图像帧(RGB)

	std::vector<Target> m_targetList;
	std::vector<Target> m_OccList;

	int m_imgW;
	int m_imgH;

// 各类型输出结果
private:
	cv::Mat matY8;
	cv::Mat matY16;
	cv::Mat matNV12;

};




// 16位调光
void Map16BitTo8Bit(unsigned short* psh16BitData, long lDataLen, unsigned char* pby8BitData);


void showArithInfo(cv::Mat src,ARIDLL_OUTPUT* output);