【算法与数据结构】海康相机接收数据转Mat结构

1.前提条件

安装好对应相机的sdk，设置相机roi视野以及接受的颜色空间信息，并通过网口ping通。

2.数据转Mat结构

主要思想是设置设备的存储数据用char * pData型数据进行接收，由于该相机获取的彩色数据以RGB形式存储，需要转成BGR格式。整个过程见如下代码：

//图像颜色通道转换函数
int RGB2BGR(unsigned char* pRgbData, unsigned int nWidth, unsigned int nHeight)
{
	if (NULL == pRgbData)
	{
		return MV_E_PARAMETER;  //0x80000004：错误的参数
	}
	//将R通道与B通道元素交换
	for (unsigned int j = 0; j < nHeight; j++)
	{
		for (unsigned int i = 0; i < nWidth; i++)
		{
			unsigned char red = pRgbData[j * (nWidth * 3) + i * 3];//每个位置第一个通道值
			pRgbData[j * (nWidth * 3) + i * 3] = pRgbData[j * (nWidth * 3) + i * 3 + 2];//每个位置第三个通道值
			pRgbData[j * (nWidth * 3) + i * 3 + 2] = red;
		}
	}

	return MV_OK;
}

// convert data stream in Mat format and save image 
bool Convert2Mat(MV_FRAME_OUT_INFO_EX* pstImageInfo, unsigned char * pData,int i)
{
	cv::Mat srcImage;
	if (pstImageInfo->enPixelType == PixelType_Gvsp_Mono8)//获取当前采集到的图像格式
	{	//获取的是单通道灰度图
		srcImage = cv::Mat(pstImageInfo->nHeight, pstImageInfo->nWidth, CV_8UC1, pData);
	}
	else if (pstImageInfo->enPixelType == PixelType_Gvsp_RGB8_Packed)
	{	//三通道RGB图像
		RGB2BGR(pData, pstImageInfo->nWidth, pstImageInfo->nHeight);
		srcImage = cv::Mat(pstImageInfo->nHeight, pstImageInfo->nWidth, CV_8UC3, pData);

		imshow("srcImage", srcImage);
		waitKey(1000);

	}
	else
	{
		printf("unsupported pixel format\n");
		return false;
	}

	if (NULL == srcImage.data)
	{
		return false;
	}
	char ch[256] = { 0 };
	sprintf(ch, "..//imageCap//00000%d.jpg", i);
	imwrite(ch, srcImage);
}