ImageMix.cpp

Go to the documentation of this file.

/*
-----------------------------------------------------------------------------
This source file is part of VideoTexture library

Copyright (c) 2007 The Zdeno Ash Miklas

This program is free software; you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License as published by the Free Software
Foundation; either version 2 of the License, or (at your option) any later
version.

This program is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License along with
this program; if not, write to the Free Software Foundation, Inc., 59 Temple
Place - Suite 330, Boston, MA 02111-1307, USA, or go to
http://www.gnu.org/copyleft/lesser.txt.
-----------------------------------------------------------------------------
*/

// implementation

#include <PyObjectPlus.h>
#include <structmember.h>

#include "ImageMix.h"

#include "ImageBase.h"

#include "Exception.h"


// cast ImageSource pointer to ImageSourceMix
inline ImageSourceMix * getImageSourceMix (ImageSource * src)
{ return static_cast<ImageSourceMix*>(src); }


// get weight
short ImageMix::getWeight (const char * id)
{
    // find source
    ImageSourceList::iterator src = findSource(id);
    // if found, return its weight
    return src != m_sources.end() ? getImageSourceMix(*src)->getWeight() : 0;
}

// set weight
bool ImageMix::setWeight (const char * id, short weight)
{
    // find source
    ImageSourceList::iterator src = findSource(id);
    // if source isn't found, report it
    if (src == m_sources.end()) return false;
    // set its weight
    getImageSourceMix(*src)->setWeight(weight);
    return true;
}

ExceptionID ImageSizesNotMatch;

ExpDesc ImageSizesNotMatchDesc (ImageSizesNotMatch, "Image sizes of sources are different");

// calculate image from sources and set its availability
void ImageMix::calcImage (unsigned int texId, double ts)
{
    // check source sizes
    if (!checkSourceSizes()) THRWEXCP(ImageSizesNotMatch, S_OK);
    // set offsets to image buffers
    for (ImageSourceList::iterator it = m_sources.begin(); it != m_sources.end(); ++it)
        // if image buffer is available
        if ((*it)->getImageBuf() != NULL)
            // set its offset
            getImageSourceMix(*it)->setOffset(m_sources[0]->getImageBuf());
        // otherwise don't calculate image
        else 
            return;
    // if there is only single source
    if (m_sources.size() == 1)
    {
        // use single filter
        FilterBase mixFilt;
        // fiter and convert image
        filterImage(mixFilt, m_sources[0]->getImageBuf(), m_sources[0]->getSize());
    }
    // otherwise use mix filter to merge source images
    else
    {
        FilterImageMix mixFilt (m_sources);
        // fiter and convert image
        filterImage(mixFilt, m_sources[0]->getImageBuf(), m_sources[0]->getSize());
    }
}



// cast Image pointer to ImageMix
inline ImageMix * getImageMix (PyImage * self)
{ return static_cast<ImageMix*>(self->m_image); }


// python methods

// get source weight
PyObject * getWeight (PyImage * self, PyObject * args)
{
    // weight
    short weight = 0;
    // get arguments
    char * id;
    if (!PyArg_ParseTuple(args, "s:getWeight", &id))
        return NULL;
    if (self->m_image != NULL)
        // get weight
        weight = getImageMix(self)->getWeight(id);
    // return weight
    return Py_BuildValue("h", weight);
}


// set source weight
PyObject * setWeight (PyImage * self, PyObject * args)
{
    // get arguments
    char * id;
    short weight = 0;
    if (!PyArg_ParseTuple(args, "sh:setWeight", &id, &weight))
        return NULL;
    if (self->m_image != NULL)
        // set weight
        if (!getImageMix(self)->setWeight(id, weight))
        {
            // if not set, report error
            PyErr_SetString(PyExc_RuntimeError, "Invalid id of source");
            return NULL;
        }
    // return none
    Py_RETURN_NONE;
}


// methods structure
static PyMethodDef imageMixMethods[] =
{ 
    {"getSource", (PyCFunction)Image_getSource, METH_VARARGS, "get image source"},
    {"setSource", (PyCFunction)Image_setSource, METH_VARARGS, "set image source"},
    {"getWeight", (PyCFunction)getWeight, METH_VARARGS, "get image source weight"},
    {"setWeight", (PyCFunction)setWeight, METH_VARARGS, "set image source weight"},
    // methods from ImageBase class
    {"refresh", (PyCFunction)Image_refresh, METH_NOARGS, "Refresh image - invalidate its current content"},
    {NULL}
};
// attributes structure
static PyGetSetDef imageMixGetSets[] =
{ // attributes from ImageBase class
    {(char*)"valid", (getter)Image_valid, NULL, (char*)"bool to tell if an image is available", NULL},
    {(char*)"image", (getter)Image_getImage, NULL, (char*)"image data", NULL},
    {(char*)"size", (getter)Image_getSize, NULL, (char*)"image size", NULL},
    {(char*)"scale", (getter)Image_getScale, (setter)Image_setScale, (char*)"fast scale of image (near neighbour)", NULL},
    {(char*)"flip", (getter)Image_getFlip, (setter)Image_setFlip, (char*)"flip image vertically", NULL},
    {(char*)"filter", (getter)Image_getFilter, (setter)Image_setFilter, (char*)"pixel filter", NULL},
    {NULL}
};


// define python type
PyTypeObject ImageMixType =
{ 
    PyVarObject_HEAD_INIT(NULL, 0)
    "VideoTexture.ImageMix",   /*tp_name*/
    sizeof(PyImage),          /*tp_basicsize*/
    0,                         /*tp_itemsize*/
    (destructor)Image_dealloc, /*tp_dealloc*/
    0,                         /*tp_print*/
    0,                         /*tp_getattr*/
    0,                         /*tp_setattr*/
    0,                         /*tp_compare*/
    0,                         /*tp_repr*/
    0,                         /*tp_as_number*/
    0,                         /*tp_as_sequence*/
    0,                         /*tp_as_mapping*/
    0,                         /*tp_hash */
    0,                         /*tp_call*/
    0,                         /*tp_str*/
    0,                         /*tp_getattro*/
    0,                         /*tp_setattro*/
    &imageBufferProcs,         /*tp_as_buffer*/
    Py_TPFLAGS_DEFAULT,        /*tp_flags*/
    "Image mixer",       /* tp_doc */
    0,                     /* tp_traverse */
    0,                     /* tp_clear */
    0,                     /* tp_richcompare */
    0,                     /* tp_weaklistoffset */
    0,                     /* tp_iter */
    0,                     /* tp_iternext */
    imageMixMethods,    /* tp_methods */
    0,                   /* tp_members */
    imageMixGetSets,          /* tp_getset */
    0,                         /* tp_base */
    0,                         /* tp_dict */
    0,                         /* tp_descr_get */
    0,                         /* tp_descr_set */
    0,                         /* tp_dictoffset */
    (initproc)Image_init<ImageMix>,     /* tp_init */
    0,                         /* tp_alloc */
    Image_allocNew,           /* tp_new */
};