You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
965 lines
20 KiB
965 lines
20 KiB
4 years ago
|
/* SV module -- interface to the Indigo video board */
|
||
|
|
||
|
/* WARNING! This module is for hardware that we don't have any more,
|
||
|
so it hasn't been tested. It has been converted to the new coding
|
||
|
style, and it is possible that this conversion has broken something
|
||
|
-- user beware! */
|
||
|
|
||
|
#include <sys/time.h>
|
||
|
#include <svideo.h>
|
||
|
#include "Python.h"
|
||
|
#include "compile.h"
|
||
|
#include "yuv.h" /* for YUV conversion functions */
|
||
|
|
||
|
typedef struct {
|
||
|
PyObject_HEAD
|
||
|
SV_nodeP ob_svideo;
|
||
|
svCaptureInfo ob_info;
|
||
|
} svobject;
|
||
|
|
||
|
typedef struct {
|
||
|
PyObject_HEAD
|
||
|
void *ob_capture;
|
||
|
int ob_mustunlock;
|
||
|
svCaptureInfo ob_info;
|
||
|
svobject *ob_svideo;
|
||
|
} captureobject;
|
||
|
|
||
|
static PyObject *SvError; /* exception sv.error */
|
||
|
|
||
|
static PyObject *newcaptureobject(svobject *, void *, int);
|
||
|
|
||
|
/* Set a SV-specific error from svideo_errno and return NULL */
|
||
|
static PyObject *
|
||
|
sv_error(void)
|
||
|
{
|
||
|
PyErr_SetString(SvError, svStrerror(svideo_errno));
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
svc_conversion(captureobject *self, PyObject *args, void (*function)(), float factor)
|
||
|
{
|
||
|
PyObject *output;
|
||
|
int invert;
|
||
|
char* outstr;
|
||
|
|
||
|
if (!PyArg_Parse(args, "i", &invert))
|
||
|
return NULL;
|
||
|
|
||
|
if (!(output = PyString_FromStringAndSize(
|
||
|
NULL,
|
||
|
(int)(self->ob_info.width * self->ob_info.height * factor))))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
if (!(outstr = PyString_AsString(output))) {
|
||
|
Py_DECREF(output);
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
(*function)((boolean)invert, self->ob_capture,
|
||
|
outstr,
|
||
|
self->ob_info.width, self->ob_info.height);
|
||
|
|
||
|
return output;
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* 3 functions to convert from Starter Video YUV 4:1:1 format to
|
||
|
* Compression Library 4:2:2 Duplicate Chroma format.
|
||
|
*/
|
||
|
static PyObject *
|
||
|
svc_YUVtoYUV422DC(captureobject *self, PyObject *args)
|
||
|
{
|
||
|
if (self->ob_info.format != SV_YUV411_FRAMES) {
|
||
|
PyErr_SetString(SvError, "data has bad format");
|
||
|
return NULL;
|
||
|
}
|
||
|
return svc_conversion(self, args, yuv_sv411_to_cl422dc, 2.0);
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
svc_YUVtoYUV422DC_quarter(captureobject *self, PyObject *args)
|
||
|
{
|
||
|
if (self->ob_info.format != SV_YUV411_FRAMES) {
|
||
|
PyErr_SetString(SvError, "data has bad format");
|
||
|
return NULL;
|
||
|
}
|
||
|
return svc_conversion(self, args,
|
||
|
yuv_sv411_to_cl422dc_quartersize, 0.5);
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
svc_YUVtoYUV422DC_sixteenth(captureobject *self, PyObject *args)
|
||
|
{
|
||
|
if (self->ob_info.format != SV_YUV411_FRAMES) {
|
||
|
PyErr_SetString(SvError, "data has bad format");
|
||
|
return NULL;
|
||
|
}
|
||
|
return svc_conversion(self, args,
|
||
|
yuv_sv411_to_cl422dc_sixteenthsize, 0.125);
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
svc_YUVtoRGB(captureobject *self, PyObject *args)
|
||
|
{
|
||
|
switch (self->ob_info.format) {
|
||
|
case SV_YUV411_FRAMES:
|
||
|
case SV_YUV411_FRAMES_AND_BLANKING_BUFFER:
|
||
|
break;
|
||
|
default:
|
||
|
PyErr_SetString(SvError, "data had bad format");
|
||
|
return NULL;
|
||
|
}
|
||
|
return svc_conversion(self, args, svYUVtoRGB, (float) sizeof(long));
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
svc_RGB8toRGB32(captureobject *self, PyObject *args)
|
||
|
{
|
||
|
if (self->ob_info.format != SV_RGB8_FRAMES) {
|
||
|
PyErr_SetString(SvError, "data has bad format");
|
||
|
return NULL;
|
||
|
}
|
||
|
return svc_conversion(self, args, svRGB8toRGB32, (float) sizeof(long));
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
svc_InterleaveFields(captureobject *self, PyObject *args)
|
||
|
{
|
||
|
if (self->ob_info.format != SV_RGB8_FRAMES) {
|
||
|
PyErr_SetString(SvError, "data has bad format");
|
||
|
return NULL;
|
||
|
}
|
||
|
return svc_conversion(self, args, svInterleaveFields, 1.0);
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
svc_GetFields(captureobject *self, PyObject *args)
|
||
|
{
|
||
|
PyObject *f1 = NULL;
|
||
|
PyObject *f2 = NULL;
|
||
|
PyObject *ret = NULL;
|
||
|
int fieldsize;
|
||
|
char* obcapture;
|
||
|
|
||
|
if (self->ob_info.format != SV_RGB8_FRAMES) {
|
||
|
PyErr_SetString(SvError, "data has bad format");
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
fieldsize = self->ob_info.width * self->ob_info.height / 2;
|
||
|
obcapture = (char*)self->ob_capture;
|
||
|
|
||
|
if (!(f1 = PyString_FromStringAndSize(obcapture, fieldsize)))
|
||
|
goto finally;
|
||
|
if (!(f2 = PyString_FromStringAndSize(obcapture + fieldsize,
|
||
|
fieldsize)))
|
||
|
goto finally;
|
||
|
ret = Py_BuildValue("(OO)", f1, f2);
|
||
|
|
||
|
finally:
|
||
|
Py_XDECREF(f1);
|
||
|
Py_XDECREF(f2);
|
||
|
return ret;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
svc_UnlockCaptureData(captureobject *self, PyObject *args)
|
||
|
{
|
||
|
if (!PyArg_Parse(args, ""))
|
||
|
return NULL;
|
||
|
|
||
|
if (!self->ob_mustunlock) {
|
||
|
PyErr_SetString(SvError, "buffer should not be unlocked");
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
if (svUnlockCaptureData(self->ob_svideo->ob_svideo, self->ob_capture))
|
||
|
return sv_error();
|
||
|
|
||
|
self->ob_mustunlock = 0;
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
return Py_None;
|
||
|
}
|
||
|
|
||
|
#ifdef USE_GL
|
||
|
#include <gl.h>
|
||
|
|
||
|
static PyObject *
|
||
|
svc_lrectwrite(captureobject *self, PyObject *args)
|
||
|
{
|
||
|
Screencoord x1, x2, y1, y2;
|
||
|
|
||
|
if (!PyArg_Parse(args, "(hhhh)", &x1, &x2, &y1, &y2))
|
||
|
return NULL;
|
||
|
|
||
|
lrectwrite(x1, x2, y1, y2, (unsigned long *) self->ob_capture);
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
return Py_None;
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
static PyObject *
|
||
|
svc_writefile(captureobject *self, PyObject *args)
|
||
|
{
|
||
|
PyObject *file;
|
||
|
int size;
|
||
|
FILE* fp;
|
||
|
|
||
|
if (!PyArg_Parse(args, "O", &file))
|
||
|
return NULL;
|
||
|
|
||
|
if (!PyFile_Check(file)) {
|
||
|
PyErr_SetString(SvError, "not a file object");
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
if (!(fp = PyFile_AsFile(file)))
|
||
|
return NULL;
|
||
|
|
||
|
size = self->ob_info.width * self->ob_info.height;
|
||
|
|
||
|
if (fwrite(self->ob_capture, sizeof(long), size, fp) != size) {
|
||
|
PyErr_SetString(SvError, "writing failed");
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
return Py_None;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
svc_FindVisibleRegion(captureobject *self, PyObject *args)
|
||
|
{
|
||
|
void *visible;
|
||
|
int width;
|
||
|
|
||
|
if (!PyArg_Parse(args, ""))
|
||
|
return NULL;
|
||
|
|
||
|
if (svFindVisibleRegion(self->ob_svideo->ob_svideo,
|
||
|
self->ob_capture, &visible,
|
||
|
self->ob_info.width))
|
||
|
return sv_error();
|
||
|
|
||
|
if (visible == NULL) {
|
||
|
PyErr_SetString(SvError, "data in wrong format");
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
return newcaptureobject(self->ob_svideo, visible, 0);
|
||
|
}
|
||
|
|
||
|
static PyMethodDef capture_methods[] = {
|
||
|
{"YUVtoRGB", (PyCFunction)svc_YUVtoRGB, METH_OLDARGS},
|
||
|
{"RGB8toRGB32", (PyCFunction)svc_RGB8toRGB32, METH_OLDARGS},
|
||
|
{"InterleaveFields", (PyCFunction)svc_InterleaveFields, METH_OLDARGS},
|
||
|
{"UnlockCaptureData", (PyCFunction)svc_UnlockCaptureData, METH_OLDARGS},
|
||
|
{"FindVisibleRegion", (PyCFunction)svc_FindVisibleRegion, METH_OLDARGS},
|
||
|
{"GetFields", (PyCFunction)svc_GetFields, METH_OLDARGS},
|
||
|
{"YUVtoYUV422DC", (PyCFunction)svc_YUVtoYUV422DC, METH_OLDARGS},
|
||
|
{"YUVtoYUV422DC_quarter",(PyCFunction)svc_YUVtoYUV422DC_quarter, METH_OLDARGS},
|
||
|
{"YUVtoYUV422DC_sixteenth",(PyCFunction)svc_YUVtoYUV422DC_sixteenth, METH_OLDARGS},
|
||
|
#ifdef USE_GL
|
||
|
{"lrectwrite", (PyCFunction)svc_lrectwrite, METH_OLDARGS},
|
||
|
#endif
|
||
|
{"writefile", (PyCFunction)svc_writefile, METH_OLDARGS},
|
||
|
{NULL, NULL} /* sentinel */
|
||
|
};
|
||
|
|
||
|
static void
|
||
|
capture_dealloc(captureobject *self)
|
||
|
{
|
||
|
if (self->ob_capture != NULL) {
|
||
|
if (self->ob_mustunlock)
|
||
|
(void)svUnlockCaptureData(self->ob_svideo->ob_svideo,
|
||
|
self->ob_capture);
|
||
|
self->ob_capture = NULL;
|
||
|
Py_DECREF(self->ob_svideo);
|
||
|
self->ob_svideo = NULL;
|
||
|
}
|
||
|
PyObject_Del(self);
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
capture_getattr(svobject *self, char *name)
|
||
|
{
|
||
|
return Py_FindMethod(capture_methods, (PyObject *)self, name);
|
||
|
}
|
||
|
|
||
|
PyTypeObject Capturetype = {
|
||
|
PyObject_HEAD_INIT(&PyType_Type)
|
||
|
0, /*ob_size*/
|
||
|
"sv.capture", /*tp_name*/
|
||
|
sizeof(captureobject), /*tp_size*/
|
||
|
0, /*tp_itemsize*/
|
||
|
/* methods */
|
||
|
(destructor)capture_dealloc, /*tp_dealloc*/
|
||
|
0, /*tp_print*/
|
||
|
(getattrfunc)capture_getattr, /*tp_getattr*/
|
||
|
0, /*tp_setattr*/
|
||
|
0, /*tp_compare*/
|
||
|
0, /*tp_repr*/
|
||
|
};
|
||
|
|
||
|
static PyObject *
|
||
|
newcaptureobject(svobject *self, void *ptr, int mustunlock)
|
||
|
{
|
||
|
captureobject *p;
|
||
|
|
||
|
p = PyObject_New(captureobject, &Capturetype);
|
||
|
if (p == NULL)
|
||
|
return NULL;
|
||
|
p->ob_svideo = self;
|
||
|
Py_INCREF(self);
|
||
|
p->ob_capture = ptr;
|
||
|
p->ob_mustunlock = mustunlock;
|
||
|
p->ob_info = self->ob_info;
|
||
|
return (PyObject *) p;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_GetCaptureData(svobject *self, PyObject *args)
|
||
|
{
|
||
|
void *ptr;
|
||
|
long fieldID;
|
||
|
PyObject *res, *c;
|
||
|
|
||
|
if (!PyArg_Parse(args, ""))
|
||
|
return NULL;
|
||
|
|
||
|
if (svGetCaptureData(self->ob_svideo, &ptr, &fieldID))
|
||
|
return sv_error();
|
||
|
|
||
|
if (ptr == NULL) {
|
||
|
PyErr_SetString(SvError, "no data available");
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
c = newcaptureobject(self, ptr, 1);
|
||
|
if (c == NULL)
|
||
|
return NULL;
|
||
|
res = Py_BuildValue("(Oi)", c, fieldID);
|
||
|
Py_DECREF(c);
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_BindGLWindow(svobject *self, PyObject *args)
|
||
|
{
|
||
|
long wid;
|
||
|
int mode;
|
||
|
|
||
|
if (!PyArg_Parse(args, "(ii)", &wid, &mode))
|
||
|
return NULL;
|
||
|
|
||
|
if (svBindGLWindow(self->ob_svideo, wid, mode))
|
||
|
return sv_error();
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
return Py_None;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_EndContinuousCapture(svobject *self, PyObject *args)
|
||
|
{
|
||
|
|
||
|
if (!PyArg_Parse(args, ""))
|
||
|
return NULL;
|
||
|
|
||
|
if (svEndContinuousCapture(self->ob_svideo))
|
||
|
return sv_error();
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
return Py_None;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_IsVideoDisplayed(svobject *self, PyObject *args)
|
||
|
{
|
||
|
int v;
|
||
|
|
||
|
if (!PyArg_Parse(args, ""))
|
||
|
return NULL;
|
||
|
|
||
|
v = svIsVideoDisplayed(self->ob_svideo);
|
||
|
if (v == -1)
|
||
|
return sv_error();
|
||
|
|
||
|
return PyInt_FromLong((long) v);
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_OutputOffset(svobject *self, PyObject *args)
|
||
|
{
|
||
|
int x_offset;
|
||
|
int y_offset;
|
||
|
|
||
|
if (!PyArg_Parse(args, "(ii)", &x_offset, &y_offset))
|
||
|
return NULL;
|
||
|
|
||
|
if (svOutputOffset(self->ob_svideo, x_offset, y_offset))
|
||
|
return sv_error();
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
return Py_None;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_PutFrame(svobject *self, PyObject *args)
|
||
|
{
|
||
|
char *buffer;
|
||
|
|
||
|
if (!PyArg_Parse(args, "s", &buffer))
|
||
|
return NULL;
|
||
|
|
||
|
if (svPutFrame(self->ob_svideo, buffer))
|
||
|
return sv_error();
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
return Py_None;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_QuerySize(svobject *self, PyObject *args)
|
||
|
{
|
||
|
int w;
|
||
|
int h;
|
||
|
int rw;
|
||
|
int rh;
|
||
|
|
||
|
if (!PyArg_Parse(args, "(ii)", &w, &h))
|
||
|
return NULL;
|
||
|
|
||
|
if (svQuerySize(self->ob_svideo, w, h, &rw, &rh))
|
||
|
return sv_error();
|
||
|
|
||
|
return Py_BuildValue("(ii)", (long) rw, (long) rh);
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_SetSize(svobject *self, PyObject *args)
|
||
|
{
|
||
|
int w;
|
||
|
int h;
|
||
|
|
||
|
if (!PyArg_Parse(args, "(ii)", &w, &h))
|
||
|
return NULL;
|
||
|
|
||
|
if (svSetSize(self->ob_svideo, w, h))
|
||
|
return sv_error();
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
return Py_None;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_SetStdDefaults(svobject *self, PyObject *args)
|
||
|
{
|
||
|
|
||
|
if (!PyArg_Parse(args, ""))
|
||
|
return NULL;
|
||
|
|
||
|
if (svSetStdDefaults(self->ob_svideo))
|
||
|
return sv_error();
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
return Py_None;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_UseExclusive(svobject *self, PyObject *args)
|
||
|
{
|
||
|
boolean onoff;
|
||
|
int mode;
|
||
|
|
||
|
if (!PyArg_Parse(args, "(ii)", &onoff, &mode))
|
||
|
return NULL;
|
||
|
|
||
|
if (svUseExclusive(self->ob_svideo, onoff, mode))
|
||
|
return sv_error();
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
return Py_None;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_WindowOffset(svobject *self, PyObject *args)
|
||
|
{
|
||
|
int x_offset;
|
||
|
int y_offset;
|
||
|
|
||
|
if (!PyArg_Parse(args, "(ii)", &x_offset, &y_offset))
|
||
|
return NULL;
|
||
|
|
||
|
if (svWindowOffset(self->ob_svideo, x_offset, y_offset))
|
||
|
return sv_error();
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
return Py_None;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_CaptureBurst(svobject *self, PyObject *args)
|
||
|
{
|
||
|
int bytes, i;
|
||
|
svCaptureInfo info;
|
||
|
void *bitvector = NULL;
|
||
|
PyObject *videodata = NULL;
|
||
|
PyObject *bitvecobj = NULL;
|
||
|
PyObject *res = NULL;
|
||
|
static PyObject *evenitem, *odditem;
|
||
|
|
||
|
if (!PyArg_Parse(args, "(iiiii)", &info.format,
|
||
|
&info.width, &info.height,
|
||
|
&info.size, &info.samplingrate))
|
||
|
return NULL;
|
||
|
|
||
|
switch (info.format) {
|
||
|
case SV_RGB8_FRAMES:
|
||
|
bitvector = malloc(SV_BITVEC_SIZE(info.size));
|
||
|
break;
|
||
|
case SV_YUV411_FRAMES_AND_BLANKING_BUFFER:
|
||
|
break;
|
||
|
default:
|
||
|
PyErr_SetString(SvError, "illegal format specified");
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
if (svQueryCaptureBufferSize(self->ob_svideo, &info, &bytes)) {
|
||
|
res = sv_error();
|
||
|
goto finally;
|
||
|
}
|
||
|
|
||
|
if (!(videodata = PyString_FromStringAndSize(NULL, bytes)))
|
||
|
goto finally;
|
||
|
|
||
|
/* XXX -- need to do something about the bitvector */
|
||
|
{
|
||
|
char* str = PyString_AsString(videodata);
|
||
|
if (!str)
|
||
|
goto finally;
|
||
|
|
||
|
if (svCaptureBurst(self->ob_svideo, &info, str, bitvector)) {
|
||
|
res = sv_error();
|
||
|
goto finally;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (bitvector) {
|
||
|
if (evenitem == NULL) {
|
||
|
if (!(evenitem = PyInt_FromLong(0)))
|
||
|
goto finally;
|
||
|
}
|
||
|
if (odditem == NULL) {
|
||
|
if (!(odditem = PyInt_FromLong(1)))
|
||
|
goto finally;
|
||
|
}
|
||
|
if (!(bitvecobj = PyTuple_New(2 * info.size)))
|
||
|
goto finally;
|
||
|
|
||
|
for (i = 0; i < 2 * info.size; i++) {
|
||
|
int sts;
|
||
|
|
||
|
if (SV_GET_FIELD(bitvector, i) == SV_EVEN_FIELD) {
|
||
|
Py_INCREF(evenitem);
|
||
|
sts = PyTuple_SetItem(bitvecobj, i, evenitem);
|
||
|
} else {
|
||
|
Py_INCREF(odditem);
|
||
|
sts = PyTuple_SetItem(bitvecobj, i, odditem);
|
||
|
}
|
||
|
if (sts < 0)
|
||
|
goto finally;
|
||
|
}
|
||
|
} else {
|
||
|
bitvecobj = Py_None;
|
||
|
Py_INCREF(Py_None);
|
||
|
}
|
||
|
|
||
|
res = Py_BuildValue("((iiiii)OO)", info.format,
|
||
|
info.width, info.height,
|
||
|
info.size, info.samplingrate,
|
||
|
videodata, bitvecobj);
|
||
|
|
||
|
finally:
|
||
|
if (bitvector)
|
||
|
free(bitvector);
|
||
|
|
||
|
Py_XDECREF(videodata);
|
||
|
Py_XDECREF(bitvecobj);
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_CaptureOneFrame(svobject *self, PyObject *args)
|
||
|
{
|
||
|
svCaptureInfo info;
|
||
|
int format, width, height;
|
||
|
int bytes;
|
||
|
PyObject *videodata = NULL;
|
||
|
PyObject *res = NULL;
|
||
|
char *str;
|
||
|
|
||
|
if (!PyArg_Parse(args, "(iii)", &format, &width, &height))
|
||
|
return NULL;
|
||
|
|
||
|
info.format = format;
|
||
|
info.width = width;
|
||
|
info.height = height;
|
||
|
info.size = 0;
|
||
|
info.samplingrate = 0;
|
||
|
if (svQueryCaptureBufferSize(self->ob_svideo, &info, &bytes))
|
||
|
return sv_error();
|
||
|
|
||
|
if (!(videodata = PyString_FromStringAndSize(NULL, bytes)))
|
||
|
return NULL;
|
||
|
|
||
|
str = PyString_AsString(videodata);
|
||
|
if (!str)
|
||
|
goto finally;
|
||
|
|
||
|
if (svCaptureOneFrame(self->ob_svideo, format, &width, &height, str)) {
|
||
|
res = sv_error();
|
||
|
goto finally;
|
||
|
}
|
||
|
|
||
|
res = Py_BuildValue("(iiO)", width, height, videodata);
|
||
|
|
||
|
finally:
|
||
|
Py_XDECREF(videodata);
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_InitContinuousCapture(svobject *self, PyObject *args)
|
||
|
{
|
||
|
svCaptureInfo info;
|
||
|
|
||
|
if (!PyArg_Parse(args, "(iiiii)", &info.format,
|
||
|
&info.width, &info.height,
|
||
|
&info.size, &info.samplingrate))
|
||
|
return NULL;
|
||
|
|
||
|
if (svInitContinuousCapture(self->ob_svideo, &info))
|
||
|
return sv_error();
|
||
|
|
||
|
self->ob_info = info;
|
||
|
|
||
|
return Py_BuildValue("(iiiii)", info.format, info.width, info.height,
|
||
|
info.size, info.samplingrate);
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_LoadMap(svobject *self, PyObject *args)
|
||
|
{
|
||
|
PyObject *rgb;
|
||
|
PyObject *res = NULL;
|
||
|
rgb_tuple *mapp = NULL;
|
||
|
int maptype;
|
||
|
int i, j; /* indices */
|
||
|
|
||
|
if (!PyArg_Parse(args, "(iO)", &maptype, &rgb))
|
||
|
return NULL;
|
||
|
|
||
|
if (!PyList_Check(rgb) || PyList_Size(rgb) != 256) {
|
||
|
PyErr_BadArgument();
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
if (!(mapp = PyMem_NEW(rgb_tuple, 256)))
|
||
|
return PyErr_NoMemory();
|
||
|
|
||
|
for (i = 0; i < 256; i++) {
|
||
|
PyObject* v = PyList_GetItem(rgb, i);
|
||
|
if (!v)
|
||
|
goto finally;
|
||
|
|
||
|
if (!PyTuple_Check(v) || PyTuple_Size(v) != 3) {
|
||
|
PyErr_BadArgument();
|
||
|
goto finally;
|
||
|
}
|
||
|
for (j = 0; j < 3; j++) {
|
||
|
PyObject* cell = PyTuple_GetItem(v, j);
|
||
|
if (!cell)
|
||
|
goto finally;
|
||
|
|
||
|
if (!PyInt_Check(cell)) {
|
||
|
PyErr_BadArgument();
|
||
|
goto finally;
|
||
|
}
|
||
|
switch (j) {
|
||
|
case 0: mapp[i].red = PyInt_AsLong(cell); break;
|
||
|
case 1: mapp[i].blue = PyInt_AsLong(cell); break;
|
||
|
case 2: mapp[i].green = PyInt_AsLong(cell); break;
|
||
|
}
|
||
|
if (PyErr_Occurred())
|
||
|
goto finally;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (svLoadMap(self->ob_svideo, maptype, mapp)) {
|
||
|
res = sv_error();
|
||
|
goto finally;
|
||
|
}
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
res = Py_None;
|
||
|
|
||
|
finally:
|
||
|
PyMem_DEL(mapp);
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_CloseVideo(svobject *self, PyObject *args)
|
||
|
{
|
||
|
if (!PyArg_Parse(args, ""))
|
||
|
return NULL;
|
||
|
|
||
|
if (svCloseVideo(self->ob_svideo))
|
||
|
return sv_error();
|
||
|
|
||
|
self->ob_svideo = NULL;
|
||
|
Py_INCREF(Py_None);
|
||
|
return Py_None;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
doParams(svobject *self, PyObject *args,
|
||
|
int (*func)(SV_nodeP, long *, int), int modified)
|
||
|
{
|
||
|
PyObject *list;
|
||
|
PyObject *res = NULL;
|
||
|
long *PVbuffer = NULL;
|
||
|
long length;
|
||
|
int i;
|
||
|
|
||
|
if (!PyArg_Parse(args, "O", &list))
|
||
|
return NULL;
|
||
|
|
||
|
if (!PyList_Check(list)) {
|
||
|
PyErr_BadArgument();
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
if ((length = PyList_Size(list)) < 0)
|
||
|
return NULL;
|
||
|
|
||
|
PVbuffer = PyMem_NEW(long, length);
|
||
|
if (PVbuffer == NULL)
|
||
|
return PyErr_NoMemory();
|
||
|
|
||
|
for (i = 0; i < length; i++) {
|
||
|
PyObject *v = PyList_GetItem(list, i);
|
||
|
if (!v)
|
||
|
goto finally;
|
||
|
|
||
|
if (!PyInt_Check(v)) {
|
||
|
PyErr_BadArgument();
|
||
|
goto finally;
|
||
|
}
|
||
|
PVbuffer[i] = PyInt_AsLong(v);
|
||
|
/* can't just test the return value, because what if the
|
||
|
value was -1?!
|
||
|
*/
|
||
|
if (PVbuffer[i] == -1 && PyErr_Occurred())
|
||
|
goto finally;
|
||
|
}
|
||
|
|
||
|
if ((*func)(self->ob_svideo, PVbuffer, length)) {
|
||
|
res = sv_error();
|
||
|
goto finally;
|
||
|
}
|
||
|
|
||
|
if (modified) {
|
||
|
for (i = 0; i < length; i++) {
|
||
|
PyObject* v = PyInt_FromLong(PVbuffer[i]);
|
||
|
if (!v || PyList_SetItem(list, i, v) < 0)
|
||
|
goto finally;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
Py_INCREF(Py_None);
|
||
|
res = Py_None;
|
||
|
|
||
|
finally:
|
||
|
PyMem_DEL(PVbuffer);
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_GetParam(PyObject *self, PyObject *args)
|
||
|
{
|
||
|
return doParams(self, args, svGetParam, 1);
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_GetParamRange(PyObject *self, PyObject *args)
|
||
|
{
|
||
|
return doParams(self, args, svGetParamRange, 1);
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_SetParam(PyObject *self, PyObject *args)
|
||
|
{
|
||
|
return doParams(self, args, svSetParam, 0);
|
||
|
}
|
||
|
|
||
|
static PyMethodDef svideo_methods[] = {
|
||
|
{"BindGLWindow", (PyCFunction)sv_BindGLWindow, METH_OLDARGS},
|
||
|
{"EndContinuousCapture",(PyCFunction)sv_EndContinuousCapture, METH_OLDARGS},
|
||
|
{"IsVideoDisplayed", (PyCFunction)sv_IsVideoDisplayed, METH_OLDARGS},
|
||
|
{"OutputOffset", (PyCFunction)sv_OutputOffset, METH_OLDARGS},
|
||
|
{"PutFrame", (PyCFunction)sv_PutFrame, METH_OLDARGS},
|
||
|
{"QuerySize", (PyCFunction)sv_QuerySize, METH_OLDARGS},
|
||
|
{"SetSize", (PyCFunction)sv_SetSize, METH_OLDARGS},
|
||
|
{"SetStdDefaults", (PyCFunction)sv_SetStdDefaults, METH_OLDARGS},
|
||
|
{"UseExclusive", (PyCFunction)sv_UseExclusive, METH_OLDARGS},
|
||
|
{"WindowOffset", (PyCFunction)sv_WindowOffset, METH_OLDARGS},
|
||
|
{"InitContinuousCapture",(PyCFunction)sv_InitContinuousCapture, METH_OLDARGS},
|
||
|
{"CaptureBurst", (PyCFunction)sv_CaptureBurst, METH_OLDARGS},
|
||
|
{"CaptureOneFrame", (PyCFunction)sv_CaptureOneFrame, METH_OLDARGS},
|
||
|
{"GetCaptureData", (PyCFunction)sv_GetCaptureData, METH_OLDARGS},
|
||
|
{"CloseVideo", (PyCFunction)sv_CloseVideo, METH_OLDARGS},
|
||
|
{"LoadMap", (PyCFunction)sv_LoadMap, METH_OLDARGS},
|
||
|
{"GetParam", (PyCFunction)sv_GetParam, METH_OLDARGS},
|
||
|
{"GetParamRange", (PyCFunction)sv_GetParamRange, METH_OLDARGS},
|
||
|
{"SetParam", (PyCFunction)sv_SetParam, METH_OLDARGS},
|
||
|
{NULL, NULL} /* sentinel */
|
||
|
};
|
||
|
|
||
|
static PyObject *
|
||
|
sv_conversion(PyObject *self, PyObject *args, void (*function)(),
|
||
|
int inputfactor, float factor)
|
||
|
{
|
||
|
int invert, width, height, inputlength;
|
||
|
char *input, *str;
|
||
|
PyObject *output;
|
||
|
|
||
|
if (!PyArg_Parse(args, "(is#ii)", &invert,
|
||
|
&input, &inputlength, &width, &height))
|
||
|
return NULL;
|
||
|
|
||
|
if (width * height * inputfactor > inputlength) {
|
||
|
PyErr_SetString(SvError, "input buffer not long enough");
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
if (!(output = PyString_FromStringAndSize(NULL,
|
||
|
(int)(width * height * factor))))
|
||
|
return NULL;
|
||
|
|
||
|
str = PyString_AsString(output);
|
||
|
if (!str) {
|
||
|
Py_DECREF(output);
|
||
|
return NULL;
|
||
|
}
|
||
|
(*function)(invert, input, str, width, height);
|
||
|
|
||
|
return output;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_InterleaveFields(PyObject *self, PyObject *args)
|
||
|
{
|
||
|
return sv_conversion(self, args, svInterleaveFields, 1, 1.0);
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_RGB8toRGB32(PyObject *self, PyObject *args)
|
||
|
{
|
||
|
return sv_conversion(self, args, svRGB8toRGB32, 1, (float) sizeof(long));
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_YUVtoRGB(PyObject *self, PyObject *args)
|
||
|
{
|
||
|
return sv_conversion(self, args, svYUVtoRGB, 2, (float) sizeof(long));
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
svideo_dealloc(svobject *self)
|
||
|
{
|
||
|
if (self->ob_svideo != NULL)
|
||
|
(void) svCloseVideo(self->ob_svideo);
|
||
|
PyObject_Del(self);
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
svideo_getattr(svobject *self, char *name)
|
||
|
{
|
||
|
return Py_FindMethod(svideo_methods, (PyObject *)self, name);
|
||
|
}
|
||
|
|
||
|
PyTypeObject Svtype = {
|
||
|
PyObject_HEAD_INIT(&PyType_Type)
|
||
|
0, /*ob_size*/
|
||
|
"sv.sv", /*tp_name*/
|
||
|
sizeof(svobject), /*tp_size*/
|
||
|
0, /*tp_itemsize*/
|
||
|
/* methods */
|
||
|
(destructor)svideo_dealloc, /*tp_dealloc*/
|
||
|
0, /*tp_print*/
|
||
|
(getattrfunc)svideo_getattr, /*tp_getattr*/
|
||
|
0, /*tp_setattr*/
|
||
|
0, /*tp_compare*/
|
||
|
0, /*tp_repr*/
|
||
|
};
|
||
|
|
||
|
static PyObject *
|
||
|
newsvobject(SV_nodeP svp)
|
||
|
{
|
||
|
svobject *p;
|
||
|
|
||
|
p = PyObject_New(svobject, &Svtype);
|
||
|
if (p == NULL)
|
||
|
return NULL;
|
||
|
p->ob_svideo = svp;
|
||
|
p->ob_info.format = 0;
|
||
|
p->ob_info.size = 0;
|
||
|
p->ob_info.width = 0;
|
||
|
p->ob_info.height = 0;
|
||
|
p->ob_info.samplingrate = 0;
|
||
|
return (PyObject *) p;
|
||
|
}
|
||
|
|
||
|
static PyObject *
|
||
|
sv_OpenVideo(PyObject *self, PyObject *args)
|
||
|
{
|
||
|
SV_nodeP svp;
|
||
|
|
||
|
if (!PyArg_Parse(args, ""))
|
||
|
return NULL;
|
||
|
|
||
|
svp = svOpenVideo();
|
||
|
if (svp == NULL)
|
||
|
return sv_error();
|
||
|
|
||
|
return newsvobject(svp);
|
||
|
}
|
||
|
|
||
|
static PyMethodDef sv_methods[] = {
|
||
|
{"InterleaveFields", (PyCFunction)sv_InterleaveFields, METH_OLDARGS},
|
||
|
{"RGB8toRGB32", (PyCFunction)sv_RGB8toRGB32, METH_OLDARGS},
|
||
|
{"YUVtoRGB", (PyCFunction)sv_YUVtoRGB, METH_OLDARGS},
|
||
|
{"OpenVideo", (PyCFunction)sv_OpenVideo, METH_OLDARGS},
|
||
|
{NULL, NULL} /* Sentinel */
|
||
|
};
|
||
|
|
||
|
void
|
||
|
initsv(void)
|
||
|
{
|
||
|
PyObject *m, *d;
|
||
|
|
||
|
m = Py_InitModule("sv", sv_methods);
|
||
|
d = PyModule_GetDict(m);
|
||
|
|
||
|
SvError = PyErr_NewException("sv.error", NULL, NULL);
|
||
|
if (SvError == NULL || PyDict_SetItemString(d, "error", SvError) != 0)
|
||
|
return;
|
||
|
}
|