/********************************************************
*
* 北京四维卓信电子有限公司
* http://www.openadsp.com
*
* 【OpenADSP开源社区】
* http://www.openadsp.com/bbs/
*
********************************************************/
#include<stdio.h>
#include "bmp.h"
/********************************************************
* 函数名 : OpenBmpFile
*
* 函数功能 : 以二进制形式打开计算机硬盘BMP图像文件
*
* 函数参数 : const char * 图像文件名称
* BMPIMAGE * 图像结构体指针
*
* 函数返回值 :FALSE 打开图像文件失败
* TRUE 打开图像文件成功
********************************************************/
int OpenBmpFile(const char* filename, BMPIMAGE* img) {
if((img->bmpfile = fopen(filename, "rb")) == NULL)
{
printf("open file is Failure\n\r");
return FALSE;
}
return TRUE;
}
/********************************************************
* 函数名 : writeBmpFile
*
* 函数功能 : 将二进制图像数据保存到计算机硬盘中
*
* 函数参数 : const char * 图像文件名称
* BMPIMAGE * 图像结构体指针
*
* 函数返回值 :FALSE 保存图像文件失败
* TRUE 保存图像文件成功
********************************************************/
int writeBmpFile(const char* filename,const BMPIMAGE* img) {
FILE *fp;
unsigned int i,j;
unsigned char tempData, *pData;
unsigned int tempHeight;
unsigned int tempWidth;
tempHeight = img->infohead.BiHeight>>1;
tempWidth = img->infohead.BiWidth*3;
pData = img->imgbuf;
for(i = 0; i < img->imagesize; i = i + 3) {
tempData = pData[i];
pData[i] = pData[i + 2];
pData[i + 2] = tempData;
}
for(i=0;i<tempHeight;i++)
{
for(j=0;j<tempWidth;j++)
{
tempData = pData[(img->infohead.BiHeight-1-i)*tempWidth+j];
pData[(img->infohead.BiHeight-1-i)*tempWidth+j] = pData[i*tempWidth+j];
pData[i*tempWidth+j] = tempData;
}
}
if((fp = fopen(filename, "wb")) == NULL)
{
printf("seek file is Failure\n\r");
return FALSE;
}
fwrite(&img->filehead, 1, 2, fp);
fwrite(&img->filehead.bfSize, 1, 12, fp);
fwrite(&img->infohead, 1, 40, fp);
fwrite(img->imgbuf, 1, img->imagesize, fp);
fclose(fp);
fclose(img->bmpfile);
return TRUE;
}
/********************************************************
* 函数名 : GetBmpHeader
*
* 函数功能 : 读取BMP文件头
*
* 函数参数 : BMPIMAGE * 图像结构体指针
*
* 函数返回值 :FALSE 读取BMP文件头失败
* TRUE 读取BMP文件头成功
********************************************************/
int GetBmpHeader(BMPIMAGE* img) {
unsigned char headbuffer[INFOHEADSIZE];
BMPFILEHEAD *filehead = &img->filehead;
BMPINFOHEAD *infohead = &img->infohead;
if (fread(headbuffer, 1, FILEHEADSIZE, img->bmpfile) != FILEHEADSIZE) {
return FALSE;
}
img->filehead.bfType[0] = headbuffer[0];
img->filehead.bfType[1] = headbuffer[1];
if (*(unsigned short *)&filehead->bfType[0] != (0x4D42)) { /* 'BM' */
printf("file is not bmp\n\r");
return FALSE; /* not bmp image*/
}
memcpy(&filehead->bfSize, &headbuffer[2], 4);
memcpy(&filehead->bfOffBits, &headbuffer[10], 4);
if(fseek(img->bmpfile, FILEHEADSIZE, SEEK_SET)) {
return FALSE;
}
if (fread(headbuffer, 1, INFOHEADSIZE, img->bmpfile) != INFOHEADSIZE) {
return FALSE;
}
memcpy(&infohead->BiSize, &headbuffer[0], 4);
memcpy(&infohead->BiWidth, &headbuffer[4], 4);
memcpy(&infohead->BiHeight, &headbuffer[8], 4);
memcpy(&infohead->BiPlanes, &headbuffer[12], 2);
memcpy(&infohead->BiBitCount, &headbuffer[14], 2);
memcpy(&infohead->BiCompression, &headbuffer[16], 4);
memcpy(&infohead->BiSizeImage, &headbuffer[20], 4);
memcpy(&infohead->BiXpelsPerMeter, &headbuffer[24], 4);
memcpy(&infohead->BiYpelsPerMeter, &headbuffer[28], 4);
memcpy(&infohead->BiClrUsed, &headbuffer[32], 4);
memcpy(&infohead->BiClrImportant, &headbuffer[36], 4);
if(infohead->BiPlanes != 1) {
return FALSE;
}
if(infohead->BiBitCount !=24) {
return FALSE;
}
if(infohead->BiCompression != BI_RGB) {
return FALSE;
}
GetImageSize(img);
return TRUE;
}
/********************************************************
* 函数名 : GetImageSize
*
* 函数功能 : 获取BMP文件大小
*
* 函数参数 : BMPIMAGE * 图像结构体指针
*
* 函数返回值 :void
********************************************************/
void GetImageSize(BMPIMAGE *img) {
img->imagesize = img->infohead.BiHeight * WIDTHBYTES(img->infohead.BiWidth * img->infohead.BiBitCount);
}
/********************************************************
* 函数名 : ReadBMPData
*
* 函数功能 : 读取BMP数据缓冲区
*
* 函数参数 : BMPIMAGE * 图像结构体指针
*
* 函数返回值 :FALSE 读取BMP数据失败
* TRUE 读取BMP数据成功
********************************************************/
int ReadBMPData(const BMPIMAGE* img) {
unsigned int pitch = WIDTHBYTES(img->infohead.BiWidth * img->infohead.BiBitCount);
unsigned int readnum = 0;
int i, j, k;
unsigned char m_temp;
unsigned char *p_tmp, *p_tmp_cur;
unsigned char tempData, *pData;
unsigned int index_data;
pData = img->imgbuf;
if(fseek(img->bmpfile, img->filehead.bfOffBits, SEEK_SET)) {
return FALSE;
}
p_tmp = NULL;
p_tmp = (unsigned char *)malloc(pitch * img->infohead.BiHeight);
if(p_tmp != NULL) {
readnum = fread(p_tmp, 1, pitch * img->infohead.BiHeight, img->bmpfile);
p_tmp_cur = p_tmp;
for(i = img->infohead.BiHeight - 1; i >= 0; i--, p_tmp_cur += pitch) {
memcpy(&img->imgbuf[i*pitch], p_tmp_cur, pitch);
}
}
else {
for(i = img->infohead.BiHeight - 1; i >= 0; i--) {
readnum += fread(&img->imgbuf[i*pitch], 1, pitch, img->bmpfile);
}
}
for(index_data = 0; index_data < img->imagesize; index_data = index_data + 3) {
tempData = pData[index_data];
pData[index_data] = pData[index_data + 2];
pData[index_data + 2] = tempData;
}
if(p_tmp != NULL) {
free(p_tmp);
}
return (readnum == img->imagesize);
}
/********************************************************
* 函数名 : Allocbuf
*
* 函数功能 : 分配图像缓冲区
*
* 函数参数 : BMPIMAGE * 图像结构体指针
*
* 函数返回值 :FALSE 分配图像缓冲区失败
* TRUE 分配图像缓冲区成功
********************************************************/
int Allocbuf(BMPIMAGE* img) {
if((img->imgbuf = malloc(img->imagesize)) == NULL)
return FALSE;
return TRUE;
}
/********************************************************
* 函数名 : Freebuf
*
* 函数功能 : 释放图像缓冲区
*
* 函数参数 : BMPIMAGE * 图像结构体指针
*
* 函数返回值 :void
********************************************************/
void Freebuf(BMPIMAGE* img) {
if(img->imgbuf != NULL)
free(img->imgbuf);
img->imgbuf = NULL;
}