/* $Id: ImageMix.cpp 27066 2010-02-21 22:20:00Z ben2610 $ ----------------------------------------------------------------------------- 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 #include #include "ImageMix.h" #include "ImageBase.h" #include "Exception.h" // cast ImageSource pointer to ImageSourceMix inline ImageSourceMix * getImageSourceMix (ImageSource * src) { return static_cast(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(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, /* tp_init */ 0, /* tp_alloc */ Image_allocNew, /* tp_new */ };