vetFrameRGB24.cpp

Go to the documentation of this file.

#include "vetFrameRGB24.h"

vetFrameRGB24::vetFrameRGB24()  : vetFrame()
 {
        height = 0;
        width = 0;
        data = NULL;
        autoFreeData = true;
 }

vetFrameRGB24::vetFrameRGB24(unsigned int w, unsigned int h)
 {
        data = NULL;
        reAllocCanvas(w, h);
        autoFreeData = true;
 }


vetFrameRGB24::vetFrameRGB24(vetFrameRGB24& img)
 {
        data = NULL;
        reAllocCanvas(img.width, img.height);

        *this = img;

        autoFreeData = true;
 }

vetFrameRGB24::vetFrameRGB24(vetFrameRGB96& img)
 {
        data = NULL;
        reAllocCanvas(img.width, img.height);

        *this << img;

        autoFreeData = true;
 }
vetFrameRGB24::vetFrameRGB24(vetFrameRGBA32& img)
 {
        data = NULL;
        reAllocCanvas(img.width, img.height);

        *this << img;

        autoFreeData = true;
 }


vetFrameRGB24::~vetFrameRGB24()
 {
        if ( autoFreeData && data != NULL )
            delete [] data;
 }


VETRESULT vetFrameRGB24::reAllocCanvas(unsigned int w, unsigned int h)
 {
        if (data != NULL)
                delete [] data;

        height = h;
        width = w;
        data = NULL;

        if ( width == 0 || height == 0)
                return VETRET_PARAM_ERR;

        data = new PixelRGB24[width * height];

        return VETRET_OK;
 }

VETRESULT vetFrameRGB24::setBlack()
 {
        if (width == 0 || height == 0 || data == NULL)
                return VETRET_ILLEGAL_USE;

        memset(data, '\0', width * height * sizeof(PixelRGB24) );

        return VETRET_OK;
 }

VETRESULT vetFrameRGB24::setWhite()
 {
        if (width == 0 || height == 0 || data == NULL)
                return VETRET_ILLEGAL_USE;

        memset(data, 255, width * height  * sizeof(PixelRGB24) );

        return VETRET_OK;
 }

VETRESULT vetFrameRGB24::extractBrightness(unsigned char* buffer, unsigned int* size)
 {
        if (width == 0 || height == 0 || data == NULL)
                return VETRET_ILLEGAL_USE;

        if (buffer == NULL)
         {
                 if ( size == NULL)
                        return VETRET_PARAM_ERR;

                 *size = width*height;
                 return VETRET_OK;
         }

        return VETRET_NOT_IMPLEMENTED;
 }


VETRESULT vetFrameRGB24::getPixel(unsigned int x, unsigned int y, PixelRGB24& p)
 {
        #ifdef _vetFrameRGB24_SLOWMODE
                if ( x >= width || y >= height )
                        throw "Invalid Coordinates in method vetFrameRGB24::getPixel(unsigned int x, unsigned int y, PixelRGB& p)";
        #endif //_vetFrameRGB24_SLOWMODE

        unsigned int offset = y * width + x;

        p[RED]  = data[offset][RED];
        p[GREEN]= data[offset][GREEN];
        p[BLUE] = data[offset][BLUE];

        return VETRET_OK;
 }

VETRESULT vetFrameRGB24::setPixel(unsigned int x, unsigned int y, PixelRGB24 p)
 {
        #ifdef _vetFrameRGB24_SLOWMODE
                if ( x >= width || y >= height )
                        throw "Invalid Coordinates in method vetFrameRGB24::setPixel(unsigned int x, unsigned int y, PixelRGB p)";
        #endif //_vetFrameRGB24_SLOWMODE

        data[y * width + x] = p;

        return VETRET_OK;
 }


vetFrameRGB24& vetFrameRGB24::clearWith(PixelRGB24& p)
 {
        if ( data == NULL )
                throw "Image is empty.";

        for ( unsigned int i=0; i < width * height; i++)
                data[i] = p;

        return *this;
 }

vetFrameRGB24& vetFrameRGB24::operator = (vetFrameRGB24& img)
 {
        /* check we're not trying to copy ourself */
        if (this == &img)
                return *this;

        /* First check images are the same size and valid */
        if (width != img.width || height != img.height)
                reAllocCanvas(img.width, img.height);

        if ( width == 0 || height == 0 )
                throw "Cannot do that with empty image (no size)";

        /* Perform copy */
        memcpy(data, img.data, width * height * sizeof(PixelRGB24));

        return *this;
 }


vetFrameRGB24& vetFrameRGB24::operator += (vetFrameRGB24& img)
{
        INFO("vetFrameRGB24& vetFrameRGB24::operator += (vetFrameRGB24& img)")

        if (width != img.width || height != img.height)
                throw "Difference in vetFrameYUV420 Dimensions";

    for(unsigned int i=0; i < width * height; i++)
        data[i] += img.data[i];

    return *this;
}



vetFrameRGB24& vetFrameRGB24::operator -= (vetFrameRGB24& img)
{
        INFO("vetFrameRGB24& vetFrameRGB24::operator += (vetFrameRGB24& img)")

        if (width != img.width || height != img.height)
                throw "Difference in vetFrameYUV420 Dimensions";

    for(unsigned int i=0; i < width * height; i++)
        data[i] -= img.data[i];

    return *this;
}






vetFrameRGB24& vetFrameRGB24::operator >> (vetFrameYUV420& img)
 {
        INFO("vetFrameRGB24& vetFrameRGB24::operator >> (vetFrameYUV420& img) [pushing data]")

        if ( width == 0 || height == 0 )
                throw "Cannot do that with empty image (no size)";

        if ( width != img.width || height != img.height )
                img.reAllocCanvas(width, height);

//BUG

        return *this;
 }

vetFrameRGB24& vetFrameRGB24::operator >> (vetFrameRGB96& img)
 {
        INFO("vetFrameRGB24& vetFrameRGB24::operator >> (vetFrameRGB96& img) [pushing data]")

        if ( width == 0 || height == 0 )
                throw "Cannot do that with empty image (no size)";

        if ( width != img.width || height != img.height )
                img.reAllocCanvas(width, height);

        unsigned char* dataPtr = static_cast<unsigned char*>( data[0] );
        int* destPtr = static_cast<int*>( img.data[0] );

    for(unsigned int i=0; i < width * height * 3; i++)
        *(destPtr++) = (int) (*dataPtr++);

        return *this;
 }

vetFrameRGB24& vetFrameRGB24::operator >> (vetFrameRGBA32& img)
 {
        INFO("vetFrameRGB24& vetFrameRGB24::operator >> (vetFrameRGBA32& img) [pushing data]")

        if ( width == 0 || height == 0 )
                throw "Cannot do that with empty image (no size)";

        if ( width != img.width || height != img.height )
                img.reAllocCanvas(width, height);

        memset(img.data + width * height * 3, '\0', width * height );

        unsigned char* R = img.data;
        unsigned char* G = img.data + width*height;
        unsigned char* B = img.data + width*height*2;

        for (unsigned int i=0; i< width*height; i++)
         {
                *(R++) = data[i][0];
                *(G++) = data[i][1];
                *(B++) = data[i][2];
         }

//CHECKBUG

        return *this;
 }




void vetFrameRGB24::operator << (const vetFrameYUV420& img)
{
        INFO("void vetFrameRGB24::operator << (const vetFrameYUV420& img) [importing data]")

        if ( img.width == 0 || img.height == 0 )
                throw "Cannot do that with empty image (no size)";

        if ( width != img.width || height != img.height )
                reAllocCanvas(img.width, img.height);

//BUG

 }


void vetFrameRGB24::operator << (const vetFrameRGB96& img)
{
        INFO("void vetFrameRGB24::operator << (const vetFrameRGB32& img) [importing data]")

        if ( img.width == 0 || img.height == 0 )
                throw "Cannot do that with empty image (no size)";

        if ( width != img.width || height != img.height )
                reAllocCanvas(img.width, img.height);

        unsigned char* dataPtr = static_cast<unsigned char*>( data[0] );
        int* srcPtr = static_cast<int*>( img.data[0] );

    for(unsigned int i=0; i < width * height * 3; i++)
        *(dataPtr++) = (unsigned char) (*srcPtr++);

 }


void vetFrameRGB24::operator << (const vetFrameRGBA32& img)
{
        INFO("void vetFrameRGB24::operator << (const vetFrameRGBA32& img) [importing data]")

        if ( img.width == 0 || img.height == 0 )
                throw "Cannot do that with empty image (no size)";

        if ( width != img.width || height != img.height )
                reAllocCanvas(img.width, img.height);

        unsigned char* R = img.data;
        unsigned char* G = img.data + width*height;
        unsigned char* B = img.data + width*height*2;

        for (unsigned int i=0; i< width*height; i++)
         {
                data[i][0] = *(R++);
                data[i][1] = *(G++);
                data[i][2] = *(B++);
         }
//CHECKBUG

 }

---