vetFrameYUV420.cpp

Go to the documentation of this file.

#include "vetFrameYUV420.h"

#include "vetUtility.h"


vetFrameYUV420::vetFrameYUV420()  : vetFrame()
 {
        height = 0;
        width = 0;
        data = NULL;
        Y = NULL; // = data[0]
        U = NULL; // = data[ width*height ]
        V = NULL; // = data[ width*height*2 ]

        autoFreeData = true;
 }


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

        if ( width != 0 && height != 0 )
         {
                memcpy(data, img.data, (int)(width * height * 1.5 * sizeof(unsigned char)));
         }
        autoFreeData = true;
 }


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


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


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

        Y = NULL; // = data[0]
        U = NULL; // = data[ width*height ]
        V = NULL; // = data[ width*height*2 ]
        width = w;
        height = h;

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

        data = new unsigned char[ (int)(width * height * 1.5) ];
        Y = data;
        U = data + width * height;
        V = data + (int)(width * height * 1.25);

        return VETRET_OK;
 }


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

        memset(data, '\0', (int)(width * height * 1.5) );

        return VETRET_OK;
 }

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

        memset(data, 255, width * height );
        memset(U, '\0', (int)(width * height * 0.5) );

        return VETRET_OK;
 }

VETRESULT vetFrameYUV420::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;
         }

        memcpy (data, buffer, width*height);
        return VETRET_OK;
 }



VETRESULT vetFrameYUV420::getPixel(unsigned int x, unsigned int y, unsigned char& value, ChannelYUV channel)
 {
        #ifdef _VETFRAMECACHE24_SLOWMODE
                if ( x >= width || y >= height )
                        throw "Invalid Coordinates in method vetFrameYUV420::getPixel(unsigned int x, unsigned int y, PixelRGB& p)";
        #endif //_VETFRAMECACHE24_SLOWMODE

        if ( channel == vetFrameYUV420::Lum)
                value = data[y * width + x];
        else if ( channel == vetFrameYUV420::Cb )
         {

//BUG
                value = data[y * width + x];

         }
        else if ( channel == vetFrameYUV420::Cr )
         {

//BUG
                value = data[y * width + x];

         }

        return VETRET_OK;
 }

VETRESULT vetFrameYUV420::setPixel(unsigned int x, unsigned int y, unsigned char& value, ChannelYUV channel)
 {
        #ifdef _VETFRAMECACHE24_SLOWMODE
                if ( x >= width || y >= height )
                        throw "Invalid Coordinates in method vetFrameYUV420::setPixel(unsigned int x, unsigned int y, PixelRGB p)";
        #endif //_VETFRAMECACHE24_SLOWMODE

        if ( channel == vetFrameYUV420::Lum)
                data[y * width + x] = value;
        else if ( channel == vetFrameYUV420::Cb )
         {

//BUG
                data[y * width + x] = value;

         }
        else if ( channel == vetFrameYUV420::Cr )
         {

//BUG
                data[y * width + x] = value;

         }

        return VETRET_OK;
 }


vetFrameYUV420& vetFrameYUV420::clearWith(unsigned char* value, ChannelYUV channel)
 {
        if ( data == NULL )
                throw "Image is empty.";

        unsigned int start = 0;
        unsigned int end = 0;

        if ( channel == vetFrameYUV420::Lum )
         {
                start = 0;
                end = width*height;
         }
        else if ( channel == vetFrameYUV420::Cb )
         {
                start = width*height;
                end = (int)(width*height*1.25);
         }
        else if ( channel == vetFrameYUV420::Cr )
         {
                start = (int)(width*height*1.25);
                end = (int)(width*height*1.5);
         }


        for (unsigned int i=start; i<end; i++)
         {
                data[i] = *value;
         }
        return *this;
 }

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

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

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

        memcpy(data, img.data, (int)(width * height * 1.5 * sizeof(unsigned char)) );

        return *this;
 }


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

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

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

    return *this;
}

vetFrameYUV420& vetFrameYUV420::operator *= (vetFrameYUV420& img)
{
        INFO("vetFrameYUV420& vetFrameYUV420::operator += (vetFrameYUV420& img)")

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

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

    return *this;
}







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

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

        vetUtility::conv_PlanarYUVtoPixelRGB(data, (unsigned char*)img.data[0], width, height);

        return *this;
 }


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

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

        vetUtility::conv_PlanarYUVtoPixelRGB(data, (unsigned char*)img.data[0], width, height);
 }

---