【机器视觉】相机获取图像数据

HVSTATUS status = STATUS_OK;
	HHV hhv = NULL;
	int i,j;

	/*
	 *	初始化所有成员变量，同时打开数字摄像机
	 */
	status = BeginHVDevice(DeviceNum, &hhv);//	打开数字摄像机 1
	HV_VERIFY(status);//	检验状态
	
	/*
	*	初始化数字摄像机硬件状态，用户也可以在其他位置初始化数字摄像机，
	*	但应保证数字摄像机已经打开，建议用户在应用程序初始化时，
	*	同时初始化数字摄像机硬件。
	 */

	
	HVSetResolution(hhv, Resolution);//设置分辨率			
	HVSetSnapMode(hhv, SnapMode);//	采集模式，包括 CONTINUATION(连续)、TRIGGER(外触发)	
	for (i = 0; i < 4; i++)
	{
		HVAGCControl(hhv, RED_CHANNEL + i, Gain);//  设置各个分量的增益
	}		
	SetExposureTime(hhv,Width,ExposureTint_Upper,ExposureTint_Lower);//	设置曝光时间
	HVADCControl(hhv, ADC_BITS, ADCLevel);//  设置ADC的级别
	
	/*
	 *	视频输出窗口，即视频输出范围，输出窗口取值范围必须在输入窗口范围以内，
	 *  视频窗口左上角X坐标和窗口宽度应为4的倍数，左上角Y坐标和窗口高度应为2的倍数
	 *	输出窗口的起始位置一般设置为(0, 0)即可。
	 */
	HVSetOutputWindow(hhv, XStart, YStart, Width, Height);

	BYTE *pRawBuffer = new BYTE[Width * Height];//分配原始图像缓冲区,一般用来存储采集图像原始数据
	ASSERT(pRawBuffer);                         //一般图像缓冲区大小由输出窗口大小和视频格式确定。


	BYTE *pImageBuffer = new BYTE[Width * Height * 3];//分配Bayer转换后图像数据缓冲
	ASSERT(pImageBuffer);

	 
	BYTE *pInfoBuffer = new BYTE[sizeof(BITMAPINFO)];//	分配BITMAPINFO缓冲
	ASSERT(pInfoBuffer);

	BYTE* pTargetBuffer = NULL;
	DWORD dwTargetSize = 0;
	dwTargetSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD)*256 + Height * Width;
	pTargetBuffer = (BYTE*)VirtualAlloc(NULL, dwTargetSize,  MEM_COMMIT,  PAGE_READWRITE);
	memset(pTargetBuffer, 0, dwTargetSize);

	/*
	 *	初始化BITMAPINFO 结构，此结构在保存bmp文件、显示采集图像时使用
	 */
	//	m_pBmpInfo即指向m_chBmpBuf缓冲区，用户可以自己分配BTIMAPINFO缓冲区	
	BITMAPINFO *pBmpInfo				= (BITMAPINFO *)pInfoBuffer;
	
	pBmpInfo->bmiHeader.biSize			= sizeof(BITMAPINFOHEADER);	
	pBmpInfo->bmiHeader.biWidth			= Width;// 	图像宽度，一般为输出窗口宽度	
	pBmpInfo->bmiHeader.biHeight		= Height;//	图像宽度，一般为输出窗口高度	
	pBmpInfo->bmiHeader.biBitCount		= 24;//	图像位深度，数字摄像机采集的原始数据为8位，Bayer转换后为24位
	//	以下设置一般相同，对于低于8位的位图，还应设置相应的位图调色板
	pBmpInfo->bmiHeader.biPlanes		= 1;
	pBmpInfo->bmiHeader.biCompression	= BI_RGB;
	pBmpInfo->bmiHeader.biSizeImage		= 0;
	pBmpInfo->bmiHeader.biXPelsPerMeter	= 0;
	pBmpInfo->bmiHeader.biYPelsPerMeter	= 0;
	pBmpInfo->bmiHeader.biClrUsed		= 0;
	pBmpInfo->bmiHeader.biClrImportant	= 0;
		
	
	BYTE *ppBuf[1];  
	ppBuf[0] = pRawBuffer;
	status = HVSnapShot(hhv, ppBuf,1);// 采集1帧图像到内存，采集完成后停止
	HV_VERIFY(status);

	if (HV_SUCCESS(status)) 
	{
		//	将原始图像数据进行Bayer转换，转换后为24位。
		//同时将原始数据进行上下翻转

		//颜色查找表
		BYTE pLutR[256] ;
		BYTE pLutG[256] ;
		BYTE pLutB[256] ;
		for(int i=0;i<256;i++)
		{
			pLutR[i] = i;
			pLutG[i] = i;
			pLutB[i] = i;
		}		
		ConvertBayer2Rgb(pImageBuffer,pRawBuffer,Width,Height,ConvertType,pLutR,pLutG,pLutB,true,Layout);

//
		

		BITMAPFILEHEADER *pTargetFileHeader = (BITMAPFILEHEADER *)pTargetBuffer;
		BITMAPINFOHEADER *pTargetInfoHeader = (BITMAPINFOHEADER *)(pTargetBuffer + sizeof(BITMAPFILEHEADER));
		
		pTargetFileHeader->bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER)+ sizeof(RGBQUAD)*256;
		pTargetFileHeader->bfReserved1 = 0;
		pTargetFileHeader->bfReserved2 = 0;
		pTargetFileHeader->bfSize = dwTargetSize;
		pTargetFileHeader->bfType = 0x4d42;
		
		
		pTargetInfoHeader->biBitCount = 8;
		pTargetInfoHeader->biClrImportant = 0;
		pTargetInfoHeader->biClrUsed = 256;
		pTargetInfoHeader->biCompression = BI_RGB;
		pTargetInfoHeader->biHeight = Height;
		pTargetInfoHeader->biPlanes = 1;
		pTargetInfoHeader->biSize =sizeof(BITMAPINFOHEADER);
		pTargetInfoHeader->biSizeImage = Height * nTargetWidth;
		pTargetInfoHeader->biWidth = Width;
		pTargetInfoHeader->biXPelsPerMeter = 0;
		pTargetInfoHeader->biYPelsPerMeter = 0;
		
		RGBQUAD *pRGBQUAD;
		for (i=0; i<256; i++) //初始化8位灰度图的调色板信息
		{
			pRGBQUAD = (RGBQUAD*)(pTargetBuffer + sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER)+ i*sizeof(RGBQUAD));
			pRGBQUAD->rgbBlue		= i;
			pRGBQUAD->rgbGreen		= i;
			pRGBQUAD->rgbRed		= i;
			pRGBQUAD->rgbReserved   = 0;
		}
		
		
		for (i=0;i<Height;i++)
		{
			for (j=0; j<Width; j++)
			{
				pTargetBuffer[54+sizeof(RGBQUAD)*256 + i*nTargetWidth + j] = 
					(pImageBuffer[54+i*nSourceWidth +j*3] * 11
					+pImageBuffer[54+i*nSourceWidth +j*3+1] * 59
					+pImageBuffer[54+i*nSourceWidth +j*3+2] * 30)/100;
			}
		}

		for (i=0;i<Height;i++)
		{
			for (j=0; j<Width; j++)
			{
				if (pTargetBuffer[54+sizeof(RGBQUAD)*256 + i*nTargetWidth + j] > 220)
				{
					pTargetBuffer[54+sizeof(RGBQUAD)*256 + i*nTargetWidth + j] = 255;
				}
				else
					pTargetBuffer[54+sizeof(RGBQUAD)*256 + i*nTargetWidth + j] = 0;
				
			}
		}


		

//
		
		/*	
		 *	以下保存BMP文件设置基本相同
		 */
		CFileDialog dlg(FALSE , "*.bmp", NULL ,OFN_OVERWRITEPROMPT|OFN_HIDEREADONLY, "Bitmap Files(*.bmp)|*.bmp", this);
		if (dlg.DoModal() == IDOK) {

			DWORD dwImageSize		= 0;
			DWORD dwBytesRead		= 0;
			BOOL bRVal				= TRUE;


			HANDLE hFile = ::CreateFile(dlg.GetPathName(),
										GENERIC_WRITE ,
										0,
										NULL,
										CREATE_ALWAYS,														
										FILE_ATTRIBUTE_NORMAL,
										NULL
										);
			if (hFile == INVALID_HANDLE_VALUE) 
			{
				bRVal = FALSE;
			}
			else
			{
				::WriteFile(hFile,pTargetBuffer, dwTargetSize, &dwBytesRead, NULL);			
				CloseHandle(hFile);
			}
		}
	}

	delete []pRawBuffer;	
	delete []pImageBuffer;	
	delete []pInfoBuffer;
	VirtualFree(pTargetBuffer,NULL,MEM_RELEASE);

	//	关闭数字摄像机1
	status = EndHVDevice(hhv);
	HV_VERIFY(status);