#include
#include
#include "bmp2rgb.h"
u_int8_t BMP2RGB(BITMAPFILEHEADER file_header,BITMAPINFOHEADER info_header, FILE* bmpFile, u_int8_t* rgbBuf);//24bit RGB
u_int8_t RGB24ToYUV420(int Width,int Height,u_int8_t* rgbBuf,u_int8_t*YuvBuffer);
#define max(a,b) (((a)>(b))?(a):(b))
#define min(a,b) (((a) < (b)) ? (a) : (b))
int main(int argc, char** argv)
{
//设置命令行参数
argv[1]= "boot_logo.bmp";
argv[2]="boot_logo.yuv"; //相当于设置文件名
char* bmpFileName = argv[1];
char* yuvFileName = argv[2];
//打开文件
FILE* bmpFile = fopen(bmpFileName, "rb");
if (bmpFile == NULL)
{
printf(" Open the BMP file.\n");
exit(1);
}
else
{
printf("The BMP file is %s\n", bmpFileName);
}
FILE* yuvFile = fopen(yuvFileName, "wb");
if (yuvFile == NULL)
{
printf("Cannot open the YUV file.\n");
exit(1);
}
else
{
printf("The YUV file is %s\n", yuvFileName);
}
//读取BMP文件头,信息头,读取错误时的处理代码
BITMAPFILEHEADER file_header;
BITMAPINFOHEADER info_header;
if (fread(&file_header, sizeof(BITMAPFILEHEADER), 1, bmpFile) != 1)
if (file_header.bfType != 0x4D42)
{
printf("Not BMP file.\n");
exit(1);
}
if (fread(&info_header, sizeof(BITMAPINFOHEADER), 1, bmpFile) != 1)
{
printf("read info header error!");
exit(1);
}//结束读取BMP文件头
//读取图像尺寸
int width = info_header.biWidth;
int height = info_header.biHeight;
//开辟缓冲区 buf
u_int8_t* yBuf = (u_int8_t*)malloc(height*width);
u_int8_t* uBuf = (u_int8_t*)malloc(height*width / 4);
u_int8_t* vBuf = (u_int8_t*)malloc(height*width / 4);
u_int8_t* rgbBuf = (u_int8_t*)malloc(height*width * 3);
u_int8_t*YuvBuffer =(u_int8_t*)malloc(height*width * 5);
if (yBuf == NULL || uBuf == NULL || vBuf == NULL || rgbBuf == NULL || YuvBuffer==NULL)
{
printf("Not enough memory\n");
exit(1);
}
//BMP与RGB的转换,得到RGB数据
if (BMP2RGB(file_header, info_header, bmpFile, rgbBuf))
{
printf("BMP2RGB error\n");
exit(1);
}
//RGB与YUV的转换,得到YUV数据
// int flip = 0;
/*读取到的图像数据是倒序存放的,flip=0保证了RGB2YUV可以正确地对其转换*/
/* if (RGB2YUV(width, height, rgbBuf, yBuf, uBuf, vBuf, flip))
{
printf("RGB2YUV error\n");
exit(1);
}
//将yuv按顺序写入yuvfile文件
fwrite(yBuf, 1, width * height, yuvFile);
fwrite(uBuf, 1, (width * height) / 4, yuvFile);
fwrite(vBuf, 1, (width * height) / 4, yuvFile);*/
if( RGB24ToYUV420( width, height, rgbBuf,YuvBuffer))
{
printf("RGB24ToYUV420 error\n");
exit(1);
}
int len=0;
len= fwrite(YuvBuffer, 1,sizeof(YuvBuffer), yuvFile);
printf("len ==%d byte\n",len);
//打印宽高,方便yuv观看程序打开
printf("width is %d", width);
printf("\n");
printf("height is %d", height);
printf("\n");
//清理内存
free(rgbBuf);
free(YuvBuffer);
free(yBuf); free(uBuf); free(vBuf);
fclose(bmpFile);
fclose(yuvFile);
return 0;
}
u_int8_t BMP2RGB(BITMAPFILEHEADER file_header,BITMAPINFOHEADER info_header, FILE* bmpFile, u_int8_t* rgbBuf)
{
BITMAPFILEHEADER file_h=file_header;
BITMAPINFOHEADER info_h=info_header;
FILE* pFile =bmpFile;
int w=0,h=0;
//确定像素的实际点阵数
w = (info_h.biWidth*info_h.biBitCount + 31) / 32 * 4;//w为实际一行的字节数
h = info_h.biHeight;//h为列数
// printf("w==%d,h==%d\n",w,h);
//开辟实际字节数量的缓冲区,读数据,一次读取一个字节
u_int8_t* dataBuf = (u_int8_t*)malloc(w*h);
/*使用文件头的字节偏移属性bfOffBits
直接把文件指针定位到像素值数据的起始 /
fseek(pFile, file_h.bfOffBits, 0);
fread(dataBuf, 1, w*h, pFile);
unsigned char data = dataBuf;
u_int8_t* rgb = rgbBuf;
//开始写入rgb
int i, j;
for (j = 0; j < h; j++)//j控制行循环
{
for (i = 0; i < w; i += 3)//i控制列循环
{
*rgb = data[i + w*j];//B
*(rgb + 1) = data[i + w*j + 1];//G
*(rgb + 2) = data[i + w*j + 2];//R
rgb += 3;
}
}
//释放内存
free(dataBuf);
return 0;
}
/***************************************************************************************************************/
u_int8_t RGB24ToYUV420(int Width,int Height,u_int8_t* rgbBuf,u_int8_t*YuvBuffer)
{
u_int8_t* yuvBuf=YuvBuffer;//YUV空间
int nWidth=Width;
int nHeight=Height;
/////////////////////下面转换算法是网上查到的
int i, j;
u_int8_t*bufY = yuvBuf;
u_int8_t*bufU = yuvBuf + nWidth * nHeight;
u_int8_t*bufV = bufU + (nWidth* nHeight* 1/4);
u_int8_t*Y=bufY;
u_int8_t*U=bufU;
u_int8_t*V=bufV;
u_int8_t*bufRGB;
unsigned char y, u, v, r, g, b;
if (NULL==rgbBuf)
{
printf("NULL==rgbBuf\n");
return 1 ;
}
for (j = 0; j<nHeight;j++)
{
bufRGB = rgbBuf + nWidth * (nHeight - 1-j) * 3 ;
for (i = 0;i<nWidth;i++)
{
int pos = nWidth * i + j;
r= *(bufRGB++);
g = *(bufRGB++);
b = *(bufRGB++);
y =(unsigned char)(( 66 * r + 129 * g + 25 * b + 128) >>8) + 16;//16
v = (unsigned char)((-38 * r - 74 * g + 112 * b + 128) >>8) +128 ; //128
u = (unsigned char)((112 * r - 94 * g - 18 * b + 128) >> 8) + 128 ;
*(bufY++)=max(0,min(y, 255 ));
if (j%2==0&&i%2 ==0)
{
if (u>255)
{
u=255;
}
if (u<0)
{
u = 0;
}
*(bufU++) =u;
//存u分量
}
else
{
//存v分量
if (i%2==0)
{
if (v>255)
{
v = 255;
}
if (v<0)
{
v = 0;
}
*(bufV++) =v;
}
}
}
}
return 0;
}
#include <stdio.h>
#include "sys/types.h"
#include <stdlib.h>
typedef unsigned long DWORD;//32bit
typedef unsigned short WORD;//16bit
typedef unsigned long LONG; //32bit
typedef struct tagBITMAPFILEHEADER {
//0x00~0x01,说明文件的类型
WORD bfType;
//0x02~0x05,说明文件的大小,用字节B为单位
DWORD bfSize;
//0x06~0x07,保留,设置为0
WORD bfReserved1;
//0x08~0x09,保留,设置为0
WORD bfReserved2;
//0x0a~0x0d,说明从BITMAP_FILE_HEADER结构开始到实际的图像数据之间的字节偏移量
DWORD bfOffBits;
} BITMAPFILEHEADER;
typedef struct tagBITMAPINFOHEADER {
//0x0e~0x11,说明当前结构体所需字节数
DWORD biSize;
//0x12~0x15,以像素为单位说明图像的宽度
LONG biWidth;
//0x16~0x19,以像素为单位说明图像的高度
LONG biHeight;
//0x1a~0x1b,说明位面数,必须为1
WORD biPlanes;
//0x1c~0x1d,说明图像的位深度
WORD biBitCount;
//0x1e~0x21,说明图像是否压缩及压缩类型
DWORD biCompression;
//0x22~0x25,以字节为单位说明图像大小,必须是4的整数倍
DWORD biSizeImage;
//0x26~0x29,目标设备的水平分辨率,像素/米
LONG biXPelsPerMeter;
//0x2a~0x2d,目标设备的垂直分辨率,像素/米
LONG biYPelsPerMeter;
//0x2e~0x31,说明图像实际用到的颜色数,如果为0,则颜色数为2的biBitCount次方
DWORD biClrUsed;
//0x32~0x35,说明对图像显示有重要影响的颜色索引的数目,如果是0,表示都重要。
DWORD biClrImportant;
} BITMAPINFOHEADER;
运行环境linux,bmp图片24位
运行结果:
The BMP file is boot_logo.bmp
The YUV file is boot_logo.yuv
len ==8 byte
width is 185729024
height is 0