vetFilter.cpp

Go to the documentation of this file.

#include "vetFilter.h"


vetFilter::vetFilter(float fps) : vetInput(fps), vetObject()
 {
        bufferYUV = NULL;
        bufferRGB = NULL;
        bufferTuC = NULL;

        setName("Abstract Filter");
        setDescription("Abstract Class");
        setVersion(0.0);
 };


vetFilter::~vetFilter()
 {
        if (bufferYUV != NULL)
                delete bufferYUV;

        if (bufferRGB != NULL)
                delete bufferRGB;

        if (bufferTuC != NULL)
                delete bufferTuC;

        bufferYUV = NULL;
        bufferRGB = NULL;
        bufferTuC = NULL;
 };



void vetFilter::allocateBuffer(vetFilterParameters::BUFFER_TYPE bType)
 {
        switch (bType)
         {
                 case vetFilterParameters::YUV:
                        useBufferYUV(0,0);
                        break;
                 case vetFilterParameters::RGB:
                        useBufferRGB(0,0);
                        break;
                 case vetFilterParameters::TuC:
                         useBufferTuC(0,0,vetFrame::VETFRAME_NONE);
                        break;
                 case vetFilterParameters::NONE:
                 default:
                        releaseBuffers();
         }
 };


void vetFilter::releaseBuffers()
 {
        if (bufferYUV != NULL)
                delete bufferYUV;

        if (bufferRGB != NULL)
                delete bufferRGB;

        if (bufferTuC != NULL)
                delete bufferTuC;

        bufferYUV = NULL;
        bufferRGB = NULL;
        bufferTuC = NULL;

        printf("getFilterParameters()->currentBuffer");

        if ( getFilterParameters() != NULL )
                getFilterParameters()->currentBuffer = vetFilterParameters::NONE;
 };


void vetFilter::useBufferYUV(unsigned int width, unsigned int height)
 {
        if ( bufferYUV == NULL )
                bufferYUV = new vetFrameYUV420(width, height);
        else if ( bufferYUV->width != width || bufferYUV->height != height )
                bufferYUV->reAllocCanvas(width, height);

        if ( getFilterParameters() != NULL )
                getFilterParameters()->currentBuffer = vetFilterParameters::YUV;

        if ( bufferRGB != NULL )
         {
                delete bufferRGB;
                bufferRGB = NULL;
         }
        if ( bufferTuC != NULL )
         {
                delete bufferTuC;
                bufferTuC = NULL;
         }
 };

void vetFilter::useBufferRGB(unsigned int width, unsigned int height)
 {
        if ( bufferRGB == NULL )
                bufferRGB = new vetFrameRGB24(width, height);
        else if ( bufferRGB->width != width || bufferRGB->height != height )
                bufferRGB->reAllocCanvas(width, height);

        if ( getFilterParameters() != NULL )
                getFilterParameters()->currentBuffer = vetFilterParameters::RGB;

        if ( bufferYUV != NULL )
         {
                delete bufferYUV;
                bufferYUV = NULL;
         }
        if ( bufferTuC != NULL )
         {
                delete bufferTuC;
                bufferTuC = NULL;
         }
 };

void vetFilter::useBufferTuC(unsigned int width, unsigned int height, vetFrame::VETFRAME_PROFILE profile)
 {
        if ( bufferTuC == NULL )
                bufferTuC = new vetFrameT<unsigned char>(width, height, profile);
        else if ( bufferTuC->width != width || bufferTuC->height != height )
                bufferTuC->reAllocCanvas(width, height);

        if ( getFilterParameters() != NULL )
                getFilterParameters()->currentBuffer = vetFilterParameters::TuC;

        if ( bufferYUV != NULL )
         {
                delete bufferYUV;
                bufferYUV = NULL;
         }
        if ( bufferRGB != NULL )
         {
                delete bufferRGB;
                bufferRGB = NULL;
         }
 };


VETRESULT vetFilter::setHeight(unsigned int value)
 {
        if (bufferYUV != NULL)
         {
                bufferYUV->reAllocCanvas(bufferYUV->width, value);
                return VETRET_OK;
         }
        else if (bufferRGB != NULL)
         {
                bufferRGB->reAllocCanvas(bufferRGB->width, value);
                return VETRET_OK;
         }
        else if (bufferTuC != NULL)
         {
                bufferTuC->reAllocCanvas(bufferTuC->width, value);
                return VETRET_OK;
         }

        return VETRET_NOT_IMPLEMENTED;
 };

VETRESULT vetFilter::setWidth(unsigned int value)
 {
        if (bufferYUV != NULL)
         {
                bufferYUV->reAllocCanvas(value, bufferYUV->height);
                return VETRET_OK;
         }
        else if (bufferRGB != NULL)
         {
                bufferRGB->reAllocCanvas(value, bufferRGB->height);
                return VETRET_OK;
         }
        else if (bufferTuC != NULL)
         {
                bufferTuC->reAllocCanvas(value, bufferTuC->height);
                return VETRET_OK;
         }

        return VETRET_NOT_IMPLEMENTED;
 };

unsigned int vetFilter::getWidth() const
 {
        if (bufferYUV != NULL)
                return bufferYUV->width;

        else if (bufferRGB != NULL)
                return bufferRGB->width;

        else if (bufferTuC != NULL)
                return bufferTuC->width;

        return 0;
 };

unsigned int vetFilter::getHeight() const
 {
        if (bufferYUV != NULL)
                return bufferYUV->height;

        else if (bufferRGB != NULL)
                return bufferRGB->height;

        else if (bufferTuC != NULL)
                return bufferTuC->height;

        return 0;
 };


bool vetFilter::EoF()
 {
        if (bufferYUV != NULL || bufferRGB != NULL || bufferTuC != NULL)
                return false;

        return true;
 };





VETRESULT vetFilter::extractTo(vetFrameYUV420& img)
 {
        INFO("VETRESULT vetFilter::extractTo(vetFrameYUV420& img) [pushing data]")

        if ( !isBufferYUV() )
                return VETRET_ILLEGAL_USE;

        img = *bufferYUV;

        return VETRET_OK;
 };

VETRESULT vetFilter::extractTo(vetFrameRGB24& img)
 {
        INFO("VETRESULT vetFilter::extractTo(vetFrameRGB24& img) [pushing data]")

        if ( !isBufferRGB() )
                return VETRET_ILLEGAL_USE;

        img = *bufferRGB;

        return VETRET_OK;
 };

VETRESULT vetFilter::extractTo(vetFrameT<unsigned char>& img)
 {
        INFO("VETRESULT vetFilter::extractTo(vetFrameT<unsigned char>& img) [pushing data]")

        if ( !isBufferTuC() )
                return VETRET_ILLEGAL_USE;

        img = *bufferTuC;

        return VETRET_OK;
 };









int vetFilterParameters::saveToXML(const char* filename)
 {
        FILE *fp;
        int ret = VETRET_OK;

        if ( (fp = fopen(filename, "w")) == NULL )
                return VETRET_PARAM_ERR;

        if( fprintf(fp, "<?xml version=\"1.0\" ?>\n\n") == EOF )
         {
                fclose(fp);
                return VETRET_INTERNAL_ERR;
         }

        ret = saveToStreamXML(fp);

        fclose(fp);

        return ret;
 };


int vetFilterParameters::loadFromXML(const char* filename)
 {
    FILE *fp;
        int ret = VETRET_OK;

    if ( (fp=fopen(filename,"r")) == NULL )
        return VETRET_PARAM_ERR;

    float xmlversion = 1.0;

    fscanf(fp, "<?xml version=\"%f\" ?>\n\n", &xmlversion);

    if (xmlversion == 0)
        throw "Incompatible XML file format";

        ret = loadFromStreamXML(fp);

        // closing tags reading is omitted

    fclose(fp);

    return ret;
 };

---