|| || || || || || || ||

---

Bitmap.h

Go to the documentation of this file.

#ifndef Impala_Core_Feature_Bitmap_h
#define Impala_Core_Feature_Bitmap_h

#include "Basis/String.h"
#include "Core/Database/RawDataSet.h"
#include "Core/Feature/FeatureDefinition.h"

namespace Impala
{
namespace Core
{
namespace Feature
{

class Bitmap
{
public:

    Bitmap()
    {
    }
   
    ~Bitmap()
    {
    }

    void
    SaveRgb2BitmapFile(const char* file, BYTE* rgb_data, int width, int height)
    {
        char filename[256];
        strcpy(filename,file);
        int bufsize = sizeof(BYTE)* width * height * 3;
        BYTE* buf = new BYTE [bufsize];
        memset(buf,0,bufsize);
        RGB2BGR(rgb_data,buf,width,height);

        WriteRgbBufferToBitmap(buf,width,height, filename);
        delete []buf;
    }

    void 
    SaveGray2BitmapFile(char* file, BYTE* gray_data, int width, int height)
    {
        char filename[256];
        strcpy(filename,file);
        int bufsize = sizeof(BYTE)* width * height * 3;
        BYTE* buf = new BYTE [bufsize];
        memset(buf,0,bufsize);
        
        for (int i=0; i<width; i++)
        {
            for (int j=0; j<height; j++)
            {
                buf[(i+j*width)*3+0] = gray_data[i+j*width]; // Blue
                buf[(i+j*width)*3+1] = gray_data[i+j*width]; // Green
                buf[(i+j*width)*3+2] = gray_data[i+j*width]; // Red

            }
        }
        
        WriteRgbBufferToBitmap(buf,width,height, filename);
        delete []buf;
    }

    //  This buffer *buf should be bottom-up mode.
    void  
    WriteRgbBufferToBitmap(BYTE *buf, const int Width, const int Height, const char* filename)
    {
        
        // if it is NOT the multiple of 4,just full it to.
        int    byteWidth=(3*Width+3)/4*4; 
        long   m_ImageSize = byteWidth*Height;

        BITMAPFILEHEADER  bmfHdr; 
        BITMAPINFO *pInfo;

        bmfHdr.bfType=0x4d42; //bmfHdr.bfType='MB';
        bmfHdr.bfSize=m_ImageSize+sizeof(BITMAPFILEHEADER)+sizeof(BITMAPINFOHEADER); 
        bmfHdr.bfReserved1=0; 
        bmfHdr.bfReserved2=0; 
        bmfHdr.bfOffBits=sizeof(BITMAPFILEHEADER)+sizeof(BITMAPINFOHEADER); 

        // note:
        // ------------------------------------------------------------
        // If biHeight is positive, the bitmap is a bottom-up DIB and
        //           its origin is the lower-left corner. 
        // If biHeight is negative, the bitmap is a top-down DIB and 
        //           its origin is the upper-left corner. 
        // ------------------------------------------------------------
        pInfo = (BITMAPINFO*)malloc(sizeof(BITMAPINFOHEADER));
        pInfo->bmiHeader.biSize=sizeof(BITMAPINFOHEADER); 
        pInfo->bmiHeader.biWidth=Width; 
        pInfo->bmiHeader.biHeight=Height; 
        pInfo->bmiHeader.biPlanes=1; 
        pInfo->bmiHeader.biBitCount=24; 
        pInfo->bmiHeader.biCompression=BI_RGB; 
        pInfo->bmiHeader.biSizeImage=m_ImageSize; 
        pInfo->bmiHeader.biXPelsPerMeter=0; 
        pInfo->bmiHeader.biYPelsPerMeter=0; 
        pInfo->bmiHeader.biClrUsed=0; 
        pInfo->bmiHeader.biClrImportant=0;

        FILE* fp = fopen(filename, "wb");
        if (fp==NULL)
        {
            printf("Can not open the file: %s\n",filename);
            return;
        }
        
        fwrite((LPSTR)&bmfHdr,   sizeof(BITMAPFILEHEADER), 1, fp);
        fwrite(pInfo,    sizeof(BITMAPINFOHEADER), 1, fp);
        fwrite(buf, sizeof(BYTE), m_ImageSize,   fp);

        fclose(fp);
        free(pInfo);
    }

    // Array2dScalarUInt8 is the order of RGBRGBRGB...
    // The Bitmap buffer is the order of BGRBGRBGR...
    void 
    RGB2BGR(BYTE *src, BYTE *dst, const int Width, const int Height)
    {
        for (int i=0; i<Width; i++)
        {
            for (int j=0; j<Height; j++)
            {
                // R G B --> B G R
                dst[(j*Width+i)*3+2] = src[(j*Width+i)*3+0];
                dst[(j*Width+i)*3+1] = src[(j*Width+i)*3+1];
                dst[(j*Width+i)*3+0] = src[(j*Width+i)*3+2];
            }
        }
    }

private:

    int mWidth;
    int mHeight;

};

} // namespace Feature
} // namespace Core
} // namespace Impala

#endif

---