YUV和RGB相互转换

#include <iostream> #include <highgui.h> #include <cv.h> #include <fstream> #include <sstream> using namespace cv; IplImage* YUV420_To_IplImage(unsigned char* pYUV420, int width, int height); void YUV420_To_BGR24(unsigned char *puc_y, unsigned char *puc_u, unsigned char *puc_v, unsigned char *puc_rgb, int width_y, int height_y); void BGR_To_YUV(unsigned char *puc_y, unsigned char *puc_u, unsigned char *puc_v, unsigned char *puc_rgb, int width_y, int height_y); int main(int argc,char *argv[]) { int iFrameCount = atoi(argv[1]) ;//字符串转换成长整型数的一种函数，检测到非数字(包括结束符 \0) 字符时停止转换，返回整型数 int iPicWidth = atoi(argv[2]) ; int iPicHeight = atoi(argv[3]) ; FILE *fp=fopen(argv[4],"rb"); FILE *fp_new = fopen(" new.yuv","wb"); unsigned char *buff = new unsigned char[iPicWidth*iPicHeight*3/2]; for (int i=0; i<iFrameCount; i++) { fseek(fp,i*(iPicWidth*iPicHeight*3/2),SEEK_SET); fread(buff,1,iPicWidth*iPicHeight*3/2,fp); IplImage *rgbimg = YUV420_To_IplImage(buff, iPicWidth, iPicHeight);//将buf转为rgb，存入rgbimg unsigned char *tmpY = new unsigned char[iPicWidth*iPicHeight]; unsigned char *tmpCb = new unsigned char[iPicWidth*iPicHeight/4]; unsigned char *tmpCr = new unsigned char[iPicWidth*iPicHeight/4]; BGR_To_YUV(tmpY,tmpCb,tmpCr,(unsigned char *)(rgbimg->imageData),iPicWidth,iPicHeight);//将rgb转为yuv输出 fwrite(tmpY,1,iPicWidth*iPicHeight,fp_new); fwrite(tmpCb,1,iPicWidth*iPicHeight/4,fp_new); fwrite(tmpCr,1,iPicWidth*iPicHeight/4,fp_new); delete[] tmpY; delete[] tmpCb; delete[] tmpCr; // cvNamedWindow( "yuv" ,1); // cvShowImage( "yuv", rgbimg); // cvWaitKey(10); cvReleaseImage(&rgbimg); } // cvDestroyAllWindows(); fclose(fp); fclose(fp_new); } IplImage* YUV420_To_IplImage(unsigned char* pYUV420, int width, int height) { //初始化变量 int baseSize = width*height; int imgSize = baseSize*3; unsigned char* pRGB24 = new unsigned char[imgSize]; memset(pRGB24, 0, imgSize); /* 变量声明 */ unsigned char* yData = pYUV420; //y分量地址 unsigned char* uData = pYUV420 + baseSize; //u分量地址 unsigned char* vData = uData + (baseSize>>2); //v分量地址 //将YUV转为RGB YUV420_To_BGR24(yData, uData, vData, pRGB24, width, height); IplImage *image = cvCreateImage(cvSize(width, height), 8,3); memcpy(image->imageData, pRGB24, imgSize); delete [] pRGB24; return image; } void YUV420_To_BGR24(unsigned char *puc_y, unsigned char *puc_u, unsigned char *puc_v, unsigned char *puc_rgb, int width_y, int height_y) { //初始化变量 int baseSize = width_y * height_y; int rgbSize = baseSize * 3; unsigned char* rgbData = new unsigned char[rgbSize]; memset(rgbData, 0, rgbSize); /* 变量声明 */ int temp = 0; unsigned char* rData = rgbData; //r分量地址 unsigned char* gData = rgbData + baseSize; //g分量地址 unsigned char* bData = gData + baseSize; //b分量地址 int uvIndex =0, yIndex =0; //YUV->RGB 的转换矩阵 //double Yuv2Rgb[3][3] = {1, 0, 1.4022, // 1, -0.3456, -0.7145, // 1, 1.771, 0}; for(int y=0; y < height_y; y++) { for(int x=0; x < width_y; x++) { uvIndex = (y>>1) * (width_y>>1) + (x>>1); yIndex = y * width_y + x; /* r分量 */ temp = (int)(puc_y[yIndex] + (puc_v[uvIndex] - 128) * 1.4022); rData[yIndex] = temp<0 ? 0 : (temp > 255 ? 255 : temp); /* g分量 */ temp = (int)(puc_y[yIndex] + (puc_u[uvIndex] - 128) * (-0.3456) + (puc_v[uvIndex] - 128) * (-0.7145)); gData[yIndex] = temp < 0 ? 0 : (temp > 255 ? 255 : temp); /* b分量 */ temp = (int)(puc_y[yIndex] + (puc_u[uvIndex] - 128) * 1.771); bData[yIndex] = temp < 0 ? 0 : (temp > 255 ? 255 : temp); } } //将R,G,B三个分量赋给img_data int widthStep = width_y*3; for (int y = 0; y < height_y; y++) { for (int x = 0; x < width_y; x++)//yuv按存完全部Y分量，再存全部U，最后存全部V；rgb按每个像素BGR存储 { puc_rgb[y * widthStep + x * 3 + 2] = rData[y * width_y + x]; //R puc_rgb[y * widthStep + x * 3 + 1] = gData[y * width_y + x]; //G puc_rgb[y * widthStep + x * 3 + 0] = bData[y * width_y + x]; //B } } delete [] rgbData; } void BGR_To_YUV(unsigned char *puc_y, unsigned char *puc_u, unsigned char *puc_v, unsigned char *puc_rgb, int width_y, int height_y) { int baseSize = width_y * height_y; int rgbSize = baseSize * 3; /* 变量声明 */ int tempY = 0,tempU = 0,tempV = 0; int index =0, yIndex =0,uvIndex=0; //Y'= 0.299*R' + 0.587*G' + 0.114*B' ， U'= -0.147*R' - 0.289*G' + 0.436*B' = 0.492*(B'- Y') ， V'= 0.615*R' - 0.515*G' - 0.100*B' = 0.877*(R'- Y') for(int y=0; y < height_y; y++) { for(int x=0; x < width_y; x++) { yIndex = y * width_y + x; index = 3*yIndex; //Y tempY = (int)(puc_rgb[index+2]*0.299 + puc_rgb[index+1]*0.587+ puc_rgb[index]*0.11); puc_y[yIndex] = tempY<0 ? 0 : (tempY > 255 ? 255 : tempY); if ((x%2==0)&&(y%2==0)) { uvIndex = (y>>1) * (width_y>>1) + (x>>1); //U tempU = (int)(puc_rgb[index+2]*(-0.147) + puc_rgb[index+1]*(-0.289)+ puc_rgb[index]*0.436+128); puc_u[uvIndex] = tempU<0 ? 0 : (tempU > 255 ? 255 : tempU); //V tempV = (int)(puc_rgb[index+2]*(0.615) + puc_rgb[index+1]*(-0.5159)+ puc_rgb[index]*(-0.100)+128); puc_v[uvIndex] = tempV<0 ? 0 : (tempV > 255 ? 255 : tempV); } } } }