VFWImage.cpp

Go to the documentation of this file.

// ************************************************************************
//      class CVFWImage
//      Can capture bitmap from camera, display and save it.
//      Camera functionality provided by separate CVFWCapture class.
//  Author:     Audrey J. W. Mbogho
//      Email:  walegwa AT yahoo DOT com
//      Last Modified: January 2005
//      Previously, I tried to give credit here to all the
//      other people from whom I "borrowed" code. But that list
//      has grown to be too long for that. Apologies to all!
//      Environment: Visual C++, Windows 98, Logitech Webcam 4000 Pro.
// ************************************************************************

#include <fstream.h>
#include "VFWImage.h"
#include "VFWCapture.h"
#include "wx/wx.h"

// Constructor
// Initializes object data
CVFWImage::CVFWImage()
{
        Width = 0;
        Height = 0;             
        bmpData = '\0';         
        pbmi = NULL;    
        BitmapSize = 0;
        
}

CVFWImage::~CVFWImage()
{
}


wxImage CVFWImage::getWxImage() {
    m_wximage.Create(Width, Height);


        int rows=Height;
        int cols=Width;
        char *dibdata=bmpData;

    unsigned char *wximagedata = m_wximage.GetData();
        
        
        int r, colpos,rowpos, rowsize = 3*Width;        
        for(r=rows-1;r>=0;r--)
        {
                rowpos=r*rowsize;
                for(colpos=0;colpos<cols*3;) 
                {                       
                *wximagedata++=dibdata[rowpos+2+colpos]; //BGR in BMP
                        *wximagedata++=dibdata[rowpos+1+colpos]; //BGR in BMP
                        *wximagedata++=dibdata[rowpos+0+colpos]; //BGR in BMP
                        colpos +=3;
                }
        }       
        return m_wximage;
        
}


wxImage CVIP2WX(Image *cvipImage) {
    



        int rows=240; //240; //Height;
        int cols=320; //320; //Width;

                wxImage wx=wxImage(cols, rows,true);
        //char *dibdata=bmpData;

    unsigned char *wximagedata = wx.GetData();
        
        char **image_red;
        char **image_blue;
        char **image_green;

        image_red = (char **)getData_Image(cvipImage,0);
        image_blue = (char **)getData_Image(cvipImage,1);
        image_green = (char **)getData_Image(cvipImage,2);
        
        int r,c, colpos,rowpos ; // , rowsize = 3*Width;        
        for(r=0;r<rows;r++)
        {
                //rowpos=r*rowsize;
                //for(colpos=0;colpos<cols*3;) 
                for(c=0;c<cols;c++)
                {                       
                *wximagedata++=image_red[r][c]; //RGB in wximagedata
                        *wximagedata++=image_blue[r][c]; 
                        *wximagedata++=image_green[r][c]; 
                        //colpos +=3;
                }
        }       
        return wx;
        
}


Image *CVFWImage::getCVIPImage() {
   

        int rows=Height;
        int cols=Width;
        char *data=bmpData;

    Image *cvipImage = NULL;
    //LPVOID result;
        int r, c, b;
        char **image_red;
        char **image_blue;
        char **image_green;
        
 
        cvipImage = new_Image(PPM,RGB,3,rows,cols,CVIP_BYTE,REAL);


        image_red = (char **)getData_Image(cvipImage,0);
        image_blue = (char **)getData_Image(cvipImage,1);
        image_green = (char **)getData_Image(cvipImage,2);
                
        for(r=rows-1;r>=0;r--)
        {
                for(c=0;c<cols;c++) 
                {
                        image_green[r][c]=data[0]; //BGR in BMP
                        image_blue[r][c]=data[1]; //BGR in BMP
                        image_red[r][c]=data[2]; //BGR in BMP
                        data +=3;
                }
        }       
        return cvipImage;
        
}


// Captures a DIB captured from video camera into a CVFWImage object
int CVFWImage::Capture()
{
        BITMAPINFOHEADER bmih;  // Contained in BITMAPINFO structure

        CVFWCapture cap;
        cap.Initialize();       // Initialize first found VFW device

        pbmi = NULL;    // CaptureDIB will automatically allocate this if set to NULL

        // Capture an image from the capture device.
        if (cap.CaptureDIB(&pbmi, 0, &BitmapSize))
        {
                // Obtain args for SetDIBitsToDevice
                bmih = pbmi->bmiHeader;
                Width = bmih.biWidth;
                Height = bmih.biHeight;
                Height = (bmih.biHeight>0) ? bmih.biHeight : -bmih.biHeight; // absolute value
                bmpData = (char *)pbmi;

                bmpData += cap.CalcBitmapInfoSize(bmih);
        }
        cap.Destroy();  // Done using VFW object
        return 0;
}


// CVFWImage::GDIPaint (hdc,x,y);
// Paints bitmap to a Windows DC
int CVFWImage::GDIPaint (HDC hdc,int x=0,int y=0)
{
        int ret = 0;
        // Paint the image to the device.
        ret = SetDIBitsToDevice (hdc,x,y,Width,Height,0,0,
                0,Height,(LPVOID)bmpData,pbmi,0);
        return ret;
}


// SaveBMP()
// Saves current bitmap to file
int CVFWImage::SaveToFile(LPCSTR fileName)
{
        BITMAPFILEHEADER bfh;

        if (!pbmi)
                return 1;                       // bitmap is empty

        ofstream bmpFile(fileName, ios::out | ios::binary);
        if (bmpFile.is_open())
        {
                bfh.bfType = 0x4d42;    // 0x42 = "B" 0x4d = "M"
                bfh.bfSize = (DWORD) BitmapSize + sizeof(BITMAPFILEHEADER);
                bfh.bfOffBits = (DWORD)   sizeof(BITMAPFILEHEADER) +
                                   sizeof(BITMAPINFOHEADER) +
                                    pbmi->bmiHeader.biClrUsed * 
                sizeof (RGBQUAD);
                bfh.bfReserved1 = 0;
                bfh.bfReserved2 = 0;

                bmpFile.write(reinterpret_cast<const char *>(&bfh),sizeof(bfh));
                bmpFile.write(reinterpret_cast<const char *>(pbmi),BitmapSize);
                bmpFile.close();
        }
        return 0;
}

---