/* $Id: VideoBase.cpp 21229 2009-06-29 02:25:54Z campbellbarton $ ----------------------------------------------------------------------------- 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. ----------------------------------------------------------------------------- */ #if defined WIN32 #define WINDOWS_LEAN_AND_MEAN #include #endif #include "VideoBase.h" #include "FilterSource.h" // VideoBase implementation // initialize image data void VideoBase::init(short width, short height) { // save original sizes m_orgSize[0] = width; m_orgSize[1] = height; // call base class initialization ImageBase::init(width, height); } // process video frame void VideoBase::process (BYTE * sample) { // if scale was changed if (m_scaleChange) // reset image init(m_orgSize[0], m_orgSize[1]); // if image is allocated and is able to store new image if (m_image != NULL && !m_avail) { // filters used // convert video format to image switch (m_format) { case RGBA32: { FilterRGBA32 filtRGBA; // use filter object for format to convert image filterImage(filtRGBA, sample, m_orgSize); // finish break; } case RGB24: { FilterRGB24 filtRGB; // use filter object for format to convert image filterImage(filtRGB, sample, m_orgSize); // finish break; } case YV12: { // use filter object for format to convert image FilterYV12 filtYUV; filtYUV.setBuffs(sample, m_orgSize); filterImage(filtYUV, sample, m_orgSize); // finish break; } } } } // python functions // exceptions for video source initialization ExceptionID SourceVideoEmpty, SourceVideoCreation; ExpDesc SourceVideoEmptyDesc (SourceVideoEmpty, "Source Video is empty"); ExpDesc SourceVideoCreationDesc (SourceVideoCreation, "SourceVideo object was not created"); // open video source void Video_open (VideoBase * self, char * file, short captureID) { // if file is empty, throw exception if (file == NULL) THRWEXCP(SourceVideoEmpty, S_OK); // open video file or capture device if (captureID >= 0) self->openCam(file, captureID); else self->openFile(file); } // play video PyObject * Video_play (PyImage * self) { if (getVideo(self)->play()) Py_RETURN_TRUE; else Py_RETURN_FALSE; } // pause video PyObject * Video_pause (PyImage * self) { if (getVideo(self)->pause()) Py_RETURN_TRUE; else Py_RETURN_FALSE; } PyObject * Video_stop (PyImage * self) { if (getVideo(self)->stop()) Py_RETURN_TRUE; else Py_RETURN_FALSE; } // get status PyObject * Video_getStatus (PyImage * self, void * closure) { return Py_BuildValue("h", getVideo(self)->getStatus()); } // refresh video PyObject * Video_refresh (PyImage * self) { getVideo(self)->refresh(); return Video_getStatus(self, NULL); } // get range PyObject * Video_getRange (PyImage * self, void * closure) { return Py_BuildValue("[ff]", getVideo(self)->getRange()[0], getVideo(self)->getRange()[1]); } // set range int Video_setRange (PyImage * self, PyObject * value, void * closure) { // check validity of parameter if (value == NULL || !PySequence_Check(value) || PySequence_Length(value) != 2 || !PyFloat_Check(PySequence_Fast_GET_ITEM(value, 0)) || !PyFloat_Check(PySequence_Fast_GET_ITEM(value, 1))) { PyErr_SetString(PyExc_TypeError, "The value must be a sequence of 2 float"); return -1; } // set range getVideo(self)->setRange(PyFloat_AsDouble(PySequence_Fast_GET_ITEM(value, 0)), PyFloat_AsDouble(PySequence_Fast_GET_ITEM(value, 1))); // success return 0; } // get repeat PyObject * Video_getRepeat (PyImage * self, void * closure) { return Py_BuildValue("h", getVideo(self)->getRepeat()); } // set repeat int Video_setRepeat (PyImage * self, PyObject * value, void * closure) { // check validity of parameter if (value == NULL || !PyLong_Check(value)) { PyErr_SetString(PyExc_TypeError, "The value must be an int"); return -1; } // set repeat getVideo(self)->setRepeat(int(PyLong_AsSsize_t(value))); // success return 0; } // get frame rate PyObject * Video_getFrameRate (PyImage * self, void * closure) { return Py_BuildValue("f", double(getVideo(self)->getFrameRate())); } // set frame rate int Video_setFrameRate (PyImage * self, PyObject * value, void * closure) { // check validity of parameter if (value == NULL || !PyFloat_Check(value)) { PyErr_SetString(PyExc_TypeError, "The value must be a float"); return -1; } // set repeat getVideo(self)->setFrameRate(float(PyFloat_AsDouble(value))); // success return 0; }