blob: 6e419ce9e563b002ea89a844496ead59c4ef08a0 [file] [log] [blame]
Sjoerd Mullendered59d201993-01-06 13:36:38 +00001/* SV module -- interface to the Indigo video board */
2
Guido van Rossumc88c9cb1996-12-17 20:43:55 +00003/* WARNING! This module is for hardware that we don't have any more,
4 so it hasn't been tested. It has been converted to the new coding
5 style, and it is possible that this conversion has broken something
6 -- user beware! */
7
Sjoerd Mullendered59d201993-01-06 13:36:38 +00008#include <sys/time.h>
9#include <svideo.h>
Barry Warsawf630f6b1996-12-13 01:24:29 +000010#include "Python.h"
Sjoerd Mullendered59d201993-01-06 13:36:38 +000011#include "compile.h"
Antoine Pitrouc83ea132010-05-09 14:46:46 +000012#include "yuv.h" /* for YUV conversion functions */
Sjoerd Mullendered59d201993-01-06 13:36:38 +000013
14typedef struct {
Antoine Pitrouc83ea132010-05-09 14:46:46 +000015 PyObject_HEAD
16 SV_nodeP ob_svideo;
17 svCaptureInfo ob_info;
Sjoerd Mullendered59d201993-01-06 13:36:38 +000018} svobject;
19
20typedef struct {
Antoine Pitrouc83ea132010-05-09 14:46:46 +000021 PyObject_HEAD
22 void *ob_capture;
23 int ob_mustunlock;
24 svCaptureInfo ob_info;
25 svobject *ob_svideo;
Sjoerd Mullendered59d201993-01-06 13:36:38 +000026} captureobject;
27
Antoine Pitrouc83ea132010-05-09 14:46:46 +000028static PyObject *SvError; /* exception sv.error */
Sjoerd Mullendered59d201993-01-06 13:36:38 +000029
Tim Petersdbd9ba62000-07-09 03:09:57 +000030static PyObject *newcaptureobject(svobject *, void *, int);
Sjoerd Mullendered59d201993-01-06 13:36:38 +000031
32/* Set a SV-specific error from svideo_errno and return NULL */
Barry Warsawf630f6b1996-12-13 01:24:29 +000033static PyObject *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +000034sv_error(void)
Sjoerd Mullendered59d201993-01-06 13:36:38 +000035{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000036 PyErr_SetString(SvError, svStrerror(svideo_errno));
37 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +000038}
39
Barry Warsawf630f6b1996-12-13 01:24:29 +000040static PyObject *
Antoine Pitrouc83ea132010-05-09 14:46:46 +000041svc_conversion(captureobject *self, PyObject *args, void (*function)(), float factor)
Sjoerd Mullendered59d201993-01-06 13:36:38 +000042{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000043 PyObject *output;
44 int invert;
45 char* outstr;
Sjoerd Mullendered59d201993-01-06 13:36:38 +000046
Antoine Pitrouc83ea132010-05-09 14:46:46 +000047 if (!PyArg_Parse(args, "i", &invert))
48 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +000049
Antoine Pitrouc83ea132010-05-09 14:46:46 +000050 if (!(output = PyString_FromStringAndSize(
51 NULL,
52 (int)(self->ob_info.width * self->ob_info.height * factor))))
53 {
54 return NULL;
55 }
56 if (!(outstr = PyString_AsString(output))) {
57 Py_DECREF(output);
58 return NULL;
59 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +000060
Antoine Pitrouc83ea132010-05-09 14:46:46 +000061 (*function)((boolean)invert, self->ob_capture,
62 outstr,
63 self->ob_info.width, self->ob_info.height);
Sjoerd Mullendered59d201993-01-06 13:36:38 +000064
Antoine Pitrouc83ea132010-05-09 14:46:46 +000065 return output;
Sjoerd Mullendered59d201993-01-06 13:36:38 +000066}
67
Sjoerd Mullender6b517fd1993-03-16 12:25:30 +000068/*
69 * 3 functions to convert from Starter Video YUV 4:1:1 format to
70 * Compression Library 4:2:2 Duplicate Chroma format.
71 */
Barry Warsawf630f6b1996-12-13 01:24:29 +000072static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +000073svc_YUVtoYUV422DC(captureobject *self, PyObject *args)
Sjoerd Mullender6b517fd1993-03-16 12:25:30 +000074{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000075 if (self->ob_info.format != SV_YUV411_FRAMES) {
76 PyErr_SetString(SvError, "data has bad format");
77 return NULL;
78 }
79 return svc_conversion(self, args, yuv_sv411_to_cl422dc, 2.0);
Sjoerd Mullender6b517fd1993-03-16 12:25:30 +000080}
81
Barry Warsawf630f6b1996-12-13 01:24:29 +000082static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +000083svc_YUVtoYUV422DC_quarter(captureobject *self, PyObject *args)
Sjoerd Mullender6b517fd1993-03-16 12:25:30 +000084{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000085 if (self->ob_info.format != SV_YUV411_FRAMES) {
86 PyErr_SetString(SvError, "data has bad format");
87 return NULL;
88 }
89 return svc_conversion(self, args,
90 yuv_sv411_to_cl422dc_quartersize, 0.5);
Sjoerd Mullender6b517fd1993-03-16 12:25:30 +000091}
92
Barry Warsawf630f6b1996-12-13 01:24:29 +000093static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +000094svc_YUVtoYUV422DC_sixteenth(captureobject *self, PyObject *args)
Sjoerd Mullender6b517fd1993-03-16 12:25:30 +000095{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000096 if (self->ob_info.format != SV_YUV411_FRAMES) {
97 PyErr_SetString(SvError, "data has bad format");
98 return NULL;
99 }
100 return svc_conversion(self, args,
101 yuv_sv411_to_cl422dc_sixteenthsize, 0.125);
Sjoerd Mullender6b517fd1993-03-16 12:25:30 +0000102}
103
Barry Warsawf630f6b1996-12-13 01:24:29 +0000104static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000105svc_YUVtoRGB(captureobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000106{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000107 switch (self->ob_info.format) {
108 case SV_YUV411_FRAMES:
109 case SV_YUV411_FRAMES_AND_BLANKING_BUFFER:
110 break;
111 default:
112 PyErr_SetString(SvError, "data had bad format");
113 return NULL;
114 }
115 return svc_conversion(self, args, svYUVtoRGB, (float) sizeof(long));
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000116}
117
Barry Warsawf630f6b1996-12-13 01:24:29 +0000118static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000119svc_RGB8toRGB32(captureobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000120{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000121 if (self->ob_info.format != SV_RGB8_FRAMES) {
122 PyErr_SetString(SvError, "data has bad format");
123 return NULL;
124 }
125 return svc_conversion(self, args, svRGB8toRGB32, (float) sizeof(long));
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000126}
127
Barry Warsawf630f6b1996-12-13 01:24:29 +0000128static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000129svc_InterleaveFields(captureobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000130{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000131 if (self->ob_info.format != SV_RGB8_FRAMES) {
132 PyErr_SetString(SvError, "data has bad format");
133 return NULL;
134 }
135 return svc_conversion(self, args, svInterleaveFields, 1.0);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000136}
137
Barry Warsawf630f6b1996-12-13 01:24:29 +0000138static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000139svc_GetFields(captureobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000140{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000141 PyObject *f1 = NULL;
142 PyObject *f2 = NULL;
143 PyObject *ret = NULL;
144 int fieldsize;
145 char* obcapture;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000146
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000147 if (self->ob_info.format != SV_RGB8_FRAMES) {
148 PyErr_SetString(SvError, "data has bad format");
149 return NULL;
150 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000151
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000152 fieldsize = self->ob_info.width * self->ob_info.height / 2;
153 obcapture = (char*)self->ob_capture;
154
155 if (!(f1 = PyString_FromStringAndSize(obcapture, fieldsize)))
156 goto finally;
157 if (!(f2 = PyString_FromStringAndSize(obcapture + fieldsize,
158 fieldsize)))
159 goto finally;
160 ret = PyTuple_Pack(2, f1, f2);
Barry Warsawf630f6b1996-12-13 01:24:29 +0000161
162 finally:
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000163 Py_XDECREF(f1);
164 Py_XDECREF(f2);
165 return ret;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000166}
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000167
Barry Warsawf630f6b1996-12-13 01:24:29 +0000168static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000169svc_UnlockCaptureData(captureobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000170{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000171 if (!PyArg_Parse(args, ""))
172 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000173
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000174 if (!self->ob_mustunlock) {
175 PyErr_SetString(SvError, "buffer should not be unlocked");
176 return NULL;
177 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000178
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000179 if (svUnlockCaptureData(self->ob_svideo->ob_svideo, self->ob_capture))
180 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000181
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000182 self->ob_mustunlock = 0;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000183
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000184 Py_INCREF(Py_None);
185 return Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000186}
187
188#ifdef USE_GL
189#include <gl.h>
190
Barry Warsawf630f6b1996-12-13 01:24:29 +0000191static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000192svc_lrectwrite(captureobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000193{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000194 Screencoord x1, x2, y1, y2;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000195
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000196 if (!PyArg_Parse(args, "(hhhh)", &x1, &x2, &y1, &y2))
197 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000198
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000199 lrectwrite(x1, x2, y1, y2, (unsigned long *) self->ob_capture);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000200
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000201 Py_INCREF(Py_None);
202 return Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000203}
204#endif
205
Barry Warsawf630f6b1996-12-13 01:24:29 +0000206static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000207svc_writefile(captureobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000208{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000209 PyObject *file;
210 int size;
211 FILE* fp;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000212
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000213 if (!PyArg_Parse(args, "O", &file))
214 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000215
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000216 if (!PyFile_Check(file)) {
217 PyErr_SetString(SvError, "not a file object");
218 return NULL;
219 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000220
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000221 if (!(fp = PyFile_AsFile(file)))
222 return NULL;
Barry Warsawf630f6b1996-12-13 01:24:29 +0000223
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000224 size = self->ob_info.width * self->ob_info.height;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000225
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000226 if (fwrite(self->ob_capture, sizeof(long), size, fp) != size) {
227 PyErr_SetString(SvError, "writing failed");
228 return NULL;
229 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000230
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000231 Py_INCREF(Py_None);
232 return Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000233}
234
Barry Warsawf630f6b1996-12-13 01:24:29 +0000235static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000236svc_FindVisibleRegion(captureobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000237{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000238 void *visible;
239 int width;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000240
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000241 if (!PyArg_Parse(args, ""))
242 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000243
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000244 if (svFindVisibleRegion(self->ob_svideo->ob_svideo,
245 self->ob_capture, &visible,
246 self->ob_info.width))
247 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000248
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000249 if (visible == NULL) {
250 PyErr_SetString(SvError, "data in wrong format");
251 return NULL;
252 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000253
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000254 return newcaptureobject(self->ob_svideo, visible, 0);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000255}
256
Barry Warsawf630f6b1996-12-13 01:24:29 +0000257static PyMethodDef capture_methods[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000258 {"YUVtoRGB", (PyCFunction)svc_YUVtoRGB, METH_OLDARGS},
259 {"RGB8toRGB32", (PyCFunction)svc_RGB8toRGB32, METH_OLDARGS},
260 {"InterleaveFields", (PyCFunction)svc_InterleaveFields, METH_OLDARGS},
261 {"UnlockCaptureData", (PyCFunction)svc_UnlockCaptureData, METH_OLDARGS},
262 {"FindVisibleRegion", (PyCFunction)svc_FindVisibleRegion, METH_OLDARGS},
263 {"GetFields", (PyCFunction)svc_GetFields, METH_OLDARGS},
264 {"YUVtoYUV422DC", (PyCFunction)svc_YUVtoYUV422DC, METH_OLDARGS},
265 {"YUVtoYUV422DC_quarter",(PyCFunction)svc_YUVtoYUV422DC_quarter, METH_OLDARGS},
266 {"YUVtoYUV422DC_sixteenth",(PyCFunction)svc_YUVtoYUV422DC_sixteenth, METH_OLDARGS},
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000267#ifdef USE_GL
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000268 {"lrectwrite", (PyCFunction)svc_lrectwrite, METH_OLDARGS},
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000269#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000270 {"writefile", (PyCFunction)svc_writefile, METH_OLDARGS},
271 {NULL, NULL} /* sentinel */
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000272};
273
274static void
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000275capture_dealloc(captureobject *self)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000276{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000277 if (self->ob_capture != NULL) {
278 if (self->ob_mustunlock)
279 (void)svUnlockCaptureData(self->ob_svideo->ob_svideo,
280 self->ob_capture);
281 self->ob_capture = NULL;
282 Py_DECREF(self->ob_svideo);
283 self->ob_svideo = NULL;
284 }
285 PyObject_Del(self);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000286}
287
Barry Warsawf630f6b1996-12-13 01:24:29 +0000288static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000289capture_getattr(svobject *self, char *name)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000290{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000291 return Py_FindMethod(capture_methods, (PyObject *)self, name);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000292}
293
Barry Warsawf630f6b1996-12-13 01:24:29 +0000294PyTypeObject Capturetype = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000295 PyObject_HEAD_INIT(&PyType_Type)
296 0, /*ob_size*/
297 "sv.capture", /*tp_name*/
298 sizeof(captureobject), /*tp_size*/
299 0, /*tp_itemsize*/
300 /* methods */
301 (destructor)capture_dealloc, /*tp_dealloc*/
302 0, /*tp_print*/
303 (getattrfunc)capture_getattr, /*tp_getattr*/
304 0, /*tp_setattr*/
305 0, /*tp_compare*/
306 0, /*tp_repr*/
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000307};
308
Barry Warsawf630f6b1996-12-13 01:24:29 +0000309static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000310newcaptureobject(svobject *self, void *ptr, int mustunlock)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000311{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000312 captureobject *p;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000313
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000314 p = PyObject_New(captureobject, &Capturetype);
315 if (p == NULL)
316 return NULL;
317 p->ob_svideo = self;
318 Py_INCREF(self);
319 p->ob_capture = ptr;
320 p->ob_mustunlock = mustunlock;
321 p->ob_info = self->ob_info;
322 return (PyObject *) p;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000323}
324
Barry Warsawf630f6b1996-12-13 01:24:29 +0000325static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000326sv_GetCaptureData(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000327{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000328 void *ptr;
329 long fieldID;
330 PyObject *res, *c;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000331
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000332 if (!PyArg_Parse(args, ""))
333 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000334
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000335 if (svGetCaptureData(self->ob_svideo, &ptr, &fieldID))
336 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000337
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000338 if (ptr == NULL) {
339 PyErr_SetString(SvError, "no data available");
340 return NULL;
341 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000342
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000343 c = newcaptureobject(self, ptr, 1);
344 if (c == NULL)
345 return NULL;
346 res = Py_BuildValue("(Oi)", c, fieldID);
347 Py_DECREF(c);
348 return res;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000349}
350
Barry Warsawf630f6b1996-12-13 01:24:29 +0000351static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000352sv_BindGLWindow(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000353{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000354 long wid;
355 int mode;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000356
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000357 if (!PyArg_Parse(args, "(ii)", &wid, &mode))
358 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000359
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000360 if (svBindGLWindow(self->ob_svideo, wid, mode))
361 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000362
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000363 Py_INCREF(Py_None);
364 return Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000365}
366
Barry Warsawf630f6b1996-12-13 01:24:29 +0000367static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000368sv_EndContinuousCapture(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000369{
370
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000371 if (!PyArg_Parse(args, ""))
372 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000373
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000374 if (svEndContinuousCapture(self->ob_svideo))
375 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000376
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000377 Py_INCREF(Py_None);
378 return Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000379}
380
Barry Warsawf630f6b1996-12-13 01:24:29 +0000381static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000382sv_IsVideoDisplayed(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000383{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000384 int v;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000385
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000386 if (!PyArg_Parse(args, ""))
387 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000388
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000389 v = svIsVideoDisplayed(self->ob_svideo);
390 if (v == -1)
391 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000392
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000393 return PyInt_FromLong((long) v);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000394}
395
Barry Warsawf630f6b1996-12-13 01:24:29 +0000396static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000397sv_OutputOffset(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000398{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000399 int x_offset;
400 int y_offset;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000401
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000402 if (!PyArg_Parse(args, "(ii)", &x_offset, &y_offset))
403 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000404
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000405 if (svOutputOffset(self->ob_svideo, x_offset, y_offset))
406 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000407
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000408 Py_INCREF(Py_None);
409 return Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000410}
411
Barry Warsawf630f6b1996-12-13 01:24:29 +0000412static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000413sv_PutFrame(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000414{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000415 char *buffer;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000416
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000417 if (!PyArg_Parse(args, "s", &buffer))
418 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000419
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000420 if (svPutFrame(self->ob_svideo, buffer))
421 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000422
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000423 Py_INCREF(Py_None);
424 return Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000425}
426
Barry Warsawf630f6b1996-12-13 01:24:29 +0000427static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000428sv_QuerySize(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000429{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000430 int w;
431 int h;
432 int rw;
433 int rh;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000434
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000435 if (!PyArg_Parse(args, "(ii)", &w, &h))
436 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000437
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000438 if (svQuerySize(self->ob_svideo, w, h, &rw, &rh))
439 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000440
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000441 return Py_BuildValue("(ii)", (long) rw, (long) rh);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000442}
443
Barry Warsawf630f6b1996-12-13 01:24:29 +0000444static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000445sv_SetSize(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000446{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000447 int w;
448 int h;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000449
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000450 if (!PyArg_Parse(args, "(ii)", &w, &h))
451 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000452
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000453 if (svSetSize(self->ob_svideo, w, h))
454 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000455
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000456 Py_INCREF(Py_None);
457 return Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000458}
459
Barry Warsawf630f6b1996-12-13 01:24:29 +0000460static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000461sv_SetStdDefaults(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000462{
463
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000464 if (!PyArg_Parse(args, ""))
465 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000466
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000467 if (svSetStdDefaults(self->ob_svideo))
468 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000469
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000470 Py_INCREF(Py_None);
471 return Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000472}
473
Barry Warsawf630f6b1996-12-13 01:24:29 +0000474static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000475sv_UseExclusive(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000476{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000477 boolean onoff;
478 int mode;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000479
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000480 if (!PyArg_Parse(args, "(ii)", &onoff, &mode))
481 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000482
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000483 if (svUseExclusive(self->ob_svideo, onoff, mode))
484 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000485
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000486 Py_INCREF(Py_None);
487 return Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000488}
489
Barry Warsawf630f6b1996-12-13 01:24:29 +0000490static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000491sv_WindowOffset(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000492{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000493 int x_offset;
494 int y_offset;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000495
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000496 if (!PyArg_Parse(args, "(ii)", &x_offset, &y_offset))
497 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000498
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000499 if (svWindowOffset(self->ob_svideo, x_offset, y_offset))
500 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000501
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000502 Py_INCREF(Py_None);
503 return Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000504}
505
Barry Warsawf630f6b1996-12-13 01:24:29 +0000506static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000507sv_CaptureBurst(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000508{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000509 int bytes, i;
510 svCaptureInfo info;
511 void *bitvector = NULL;
512 PyObject *videodata = NULL;
513 PyObject *bitvecobj = NULL;
514 PyObject *res = NULL;
515 static PyObject *evenitem, *odditem;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000516
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000517 if (!PyArg_Parse(args, "(iiiii)", &info.format,
518 &info.width, &info.height,
519 &info.size, &info.samplingrate))
520 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000521
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000522 switch (info.format) {
523 case SV_RGB8_FRAMES:
524 bitvector = malloc(SV_BITVEC_SIZE(info.size));
525 break;
526 case SV_YUV411_FRAMES_AND_BLANKING_BUFFER:
527 break;
528 default:
529 PyErr_SetString(SvError, "illegal format specified");
530 return NULL;
531 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000532
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000533 if (svQueryCaptureBufferSize(self->ob_svideo, &info, &bytes)) {
534 res = sv_error();
535 goto finally;
536 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000537
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000538 if (!(videodata = PyString_FromStringAndSize(NULL, bytes)))
539 goto finally;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000540
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000541 /* XXX -- need to do something about the bitvector */
542 {
543 char* str = PyString_AsString(videodata);
544 if (!str)
545 goto finally;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000546
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000547 if (svCaptureBurst(self->ob_svideo, &info, str, bitvector)) {
548 res = sv_error();
549 goto finally;
550 }
551 }
Barry Warsawf630f6b1996-12-13 01:24:29 +0000552
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000553 if (bitvector) {
554 if (evenitem == NULL) {
555 if (!(evenitem = PyInt_FromLong(0)))
556 goto finally;
557 }
558 if (odditem == NULL) {
559 if (!(odditem = PyInt_FromLong(1)))
560 goto finally;
561 }
562 if (!(bitvecobj = PyTuple_New(2 * info.size)))
563 goto finally;
Barry Warsawf630f6b1996-12-13 01:24:29 +0000564
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000565 for (i = 0; i < 2 * info.size; i++) {
566 int sts;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000567
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000568 if (SV_GET_FIELD(bitvector, i) == SV_EVEN_FIELD) {
569 Py_INCREF(evenitem);
570 sts = PyTuple_SetItem(bitvecobj, i, evenitem);
571 } else {
572 Py_INCREF(odditem);
573 sts = PyTuple_SetItem(bitvecobj, i, odditem);
574 }
575 if (sts < 0)
576 goto finally;
577 }
578 } else {
579 bitvecobj = Py_None;
580 Py_INCREF(Py_None);
581 }
582
583 res = Py_BuildValue("((iiiii)OO)", info.format,
584 info.width, info.height,
585 info.size, info.samplingrate,
586 videodata, bitvecobj);
Barry Warsawf630f6b1996-12-13 01:24:29 +0000587
588 finally:
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000589 if (bitvector)
590 free(bitvector);
Barry Warsawf630f6b1996-12-13 01:24:29 +0000591
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000592 Py_XDECREF(videodata);
593 Py_XDECREF(bitvecobj);
594 return res;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000595}
596
Barry Warsawf630f6b1996-12-13 01:24:29 +0000597static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000598sv_CaptureOneFrame(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000599{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000600 svCaptureInfo info;
601 int format, width, height;
602 int bytes;
603 PyObject *videodata = NULL;
604 PyObject *res = NULL;
605 char *str;
Barry Warsawf630f6b1996-12-13 01:24:29 +0000606
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000607 if (!PyArg_Parse(args, "(iii)", &format, &width, &height))
608 return NULL;
Barry Warsawf630f6b1996-12-13 01:24:29 +0000609
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000610 info.format = format;
611 info.width = width;
612 info.height = height;
613 info.size = 0;
614 info.samplingrate = 0;
615 if (svQueryCaptureBufferSize(self->ob_svideo, &info, &bytes))
616 return sv_error();
Barry Warsawf630f6b1996-12-13 01:24:29 +0000617
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000618 if (!(videodata = PyString_FromStringAndSize(NULL, bytes)))
619 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000620
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000621 str = PyString_AsString(videodata);
622 if (!str)
623 goto finally;
624
625 if (svCaptureOneFrame(self->ob_svideo, format, &width, &height, str)) {
626 res = sv_error();
627 goto finally;
628 }
629
630 res = Py_BuildValue("(iiO)", width, height, videodata);
Barry Warsawf630f6b1996-12-13 01:24:29 +0000631
632 finally:
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000633 Py_XDECREF(videodata);
634 return res;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000635}
636
Barry Warsawf630f6b1996-12-13 01:24:29 +0000637static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000638sv_InitContinuousCapture(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000639{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000640 svCaptureInfo info;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000641
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000642 if (!PyArg_Parse(args, "(iiiii)", &info.format,
643 &info.width, &info.height,
644 &info.size, &info.samplingrate))
645 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000646
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000647 if (svInitContinuousCapture(self->ob_svideo, &info))
648 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000649
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000650 self->ob_info = info;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000651
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000652 return Py_BuildValue("(iiiii)", info.format, info.width, info.height,
653 info.size, info.samplingrate);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000654}
655
Barry Warsawf630f6b1996-12-13 01:24:29 +0000656static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000657sv_LoadMap(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000658{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000659 PyObject *rgb;
660 PyObject *res = NULL;
661 rgb_tuple *mapp = NULL;
662 int maptype;
663 int i, j; /* indices */
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000664
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000665 if (!PyArg_Parse(args, "(iO)", &maptype, &rgb))
666 return NULL;
Barry Warsawf630f6b1996-12-13 01:24:29 +0000667
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000668 if (!PyList_Check(rgb) || PyList_Size(rgb) != 256) {
669 PyErr_BadArgument();
670 return NULL;
671 }
Barry Warsawf630f6b1996-12-13 01:24:29 +0000672
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000673 if (!(mapp = PyMem_NEW(rgb_tuple, 256)))
674 return PyErr_NoMemory();
Barry Warsawf630f6b1996-12-13 01:24:29 +0000675
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000676 for (i = 0; i < 256; i++) {
677 PyObject* v = PyList_GetItem(rgb, i);
678 if (!v)
679 goto finally;
Barry Warsawf630f6b1996-12-13 01:24:29 +0000680
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000681 if (!PyTuple_Check(v) || PyTuple_Size(v) != 3) {
682 PyErr_BadArgument();
683 goto finally;
684 }
685 for (j = 0; j < 3; j++) {
686 PyObject* cell = PyTuple_GetItem(v, j);
687 if (!cell)
688 goto finally;
Barry Warsawf630f6b1996-12-13 01:24:29 +0000689
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000690 if (!PyInt_Check(cell)) {
691 PyErr_BadArgument();
692 goto finally;
693 }
694 switch (j) {
695 case 0: mapp[i].red = PyInt_AsLong(cell); break;
696 case 1: mapp[i].blue = PyInt_AsLong(cell); break;
697 case 2: mapp[i].green = PyInt_AsLong(cell); break;
698 }
699 if (PyErr_Occurred())
700 goto finally;
701 }
702 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000703
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000704 if (svLoadMap(self->ob_svideo, maptype, mapp)) {
705 res = sv_error();
706 goto finally;
707 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000708
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000709 Py_INCREF(Py_None);
710 res = Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000711
Barry Warsawf630f6b1996-12-13 01:24:29 +0000712 finally:
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000713 PyMem_DEL(mapp);
714 return res;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000715}
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000716
Barry Warsawf630f6b1996-12-13 01:24:29 +0000717static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000718sv_CloseVideo(svobject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000719{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000720 if (!PyArg_Parse(args, ""))
721 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000722
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000723 if (svCloseVideo(self->ob_svideo))
724 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000725
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000726 self->ob_svideo = NULL;
727 Py_INCREF(Py_None);
728 return Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000729}
730
Barry Warsawf630f6b1996-12-13 01:24:29 +0000731static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000732doParams(svobject *self, PyObject *args,
733 int (*func)(SV_nodeP, long *, int), int modified)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000734{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000735 PyObject *list;
736 PyObject *res = NULL;
737 long *PVbuffer = NULL;
738 long length;
739 int i;
Barry Warsawf630f6b1996-12-13 01:24:29 +0000740
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000741 if (!PyArg_Parse(args, "O", &list))
742 return NULL;
Barry Warsawf630f6b1996-12-13 01:24:29 +0000743
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000744 if (!PyList_Check(list)) {
745 PyErr_BadArgument();
746 return NULL;
747 }
Barry Warsawf630f6b1996-12-13 01:24:29 +0000748
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000749 if ((length = PyList_Size(list)) < 0)
750 return NULL;
Barry Warsawf630f6b1996-12-13 01:24:29 +0000751
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000752 PVbuffer = PyMem_NEW(long, length);
753 if (PVbuffer == NULL)
754 return PyErr_NoMemory();
Barry Warsawf630f6b1996-12-13 01:24:29 +0000755
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000756 for (i = 0; i < length; i++) {
757 PyObject *v = PyList_GetItem(list, i);
758 if (!v)
759 goto finally;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000760
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000761 if (!PyInt_Check(v)) {
762 PyErr_BadArgument();
763 goto finally;
764 }
765 PVbuffer[i] = PyInt_AsLong(v);
766 /* can't just test the return value, because what if the
767 value was -1?!
768 */
769 if (PVbuffer[i] == -1 && PyErr_Occurred())
770 goto finally;
771 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000772
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000773 if ((*func)(self->ob_svideo, PVbuffer, length)) {
774 res = sv_error();
775 goto finally;
776 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000777
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000778 if (modified) {
779 for (i = 0; i < length; i++) {
780 PyObject* v = PyInt_FromLong(PVbuffer[i]);
781 if (!v || PyList_SetItem(list, i, v) < 0)
782 goto finally;
783 }
784 }
785
786 Py_INCREF(Py_None);
787 res = Py_None;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000788
Barry Warsawf630f6b1996-12-13 01:24:29 +0000789 finally:
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000790 PyMem_DEL(PVbuffer);
791 return res;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000792}
793
Barry Warsawf630f6b1996-12-13 01:24:29 +0000794static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000795sv_GetParam(PyObject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000796{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000797 return doParams(self, args, svGetParam, 1);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000798}
799
Barry Warsawf630f6b1996-12-13 01:24:29 +0000800static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000801sv_GetParamRange(PyObject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000802{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000803 return doParams(self, args, svGetParamRange, 1);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000804}
805
Barry Warsawf630f6b1996-12-13 01:24:29 +0000806static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000807sv_SetParam(PyObject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000808{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000809 return doParams(self, args, svSetParam, 0);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000810}
811
Barry Warsawf630f6b1996-12-13 01:24:29 +0000812static PyMethodDef svideo_methods[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000813 {"BindGLWindow", (PyCFunction)sv_BindGLWindow, METH_OLDARGS},
814 {"EndContinuousCapture",(PyCFunction)sv_EndContinuousCapture, METH_OLDARGS},
815 {"IsVideoDisplayed", (PyCFunction)sv_IsVideoDisplayed, METH_OLDARGS},
816 {"OutputOffset", (PyCFunction)sv_OutputOffset, METH_OLDARGS},
817 {"PutFrame", (PyCFunction)sv_PutFrame, METH_OLDARGS},
818 {"QuerySize", (PyCFunction)sv_QuerySize, METH_OLDARGS},
819 {"SetSize", (PyCFunction)sv_SetSize, METH_OLDARGS},
820 {"SetStdDefaults", (PyCFunction)sv_SetStdDefaults, METH_OLDARGS},
821 {"UseExclusive", (PyCFunction)sv_UseExclusive, METH_OLDARGS},
822 {"WindowOffset", (PyCFunction)sv_WindowOffset, METH_OLDARGS},
823 {"InitContinuousCapture",(PyCFunction)sv_InitContinuousCapture, METH_OLDARGS},
824 {"CaptureBurst", (PyCFunction)sv_CaptureBurst, METH_OLDARGS},
825 {"CaptureOneFrame", (PyCFunction)sv_CaptureOneFrame, METH_OLDARGS},
826 {"GetCaptureData", (PyCFunction)sv_GetCaptureData, METH_OLDARGS},
827 {"CloseVideo", (PyCFunction)sv_CloseVideo, METH_OLDARGS},
828 {"LoadMap", (PyCFunction)sv_LoadMap, METH_OLDARGS},
829 {"GetParam", (PyCFunction)sv_GetParam, METH_OLDARGS},
830 {"GetParamRange", (PyCFunction)sv_GetParamRange, METH_OLDARGS},
831 {"SetParam", (PyCFunction)sv_SetParam, METH_OLDARGS},
832 {NULL, NULL} /* sentinel */
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000833};
834
Barry Warsawf630f6b1996-12-13 01:24:29 +0000835static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000836sv_conversion(PyObject *self, PyObject *args, void (*function)(),
837 int inputfactor, float factor)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000838{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000839 int invert, width, height, inputlength;
840 char *input, *str;
841 PyObject *output;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000842
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000843 if (!PyArg_Parse(args, "(is#ii)", &invert,
844 &input, &inputlength, &width, &height))
845 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000846
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000847 if (width * height * inputfactor > inputlength) {
848 PyErr_SetString(SvError, "input buffer not long enough");
849 return NULL;
850 }
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000851
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000852 if (!(output = PyString_FromStringAndSize(NULL,
853 (int)(width * height * factor))))
854 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000855
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000856 str = PyString_AsString(output);
857 if (!str) {
858 Py_DECREF(output);
859 return NULL;
860 }
861 (*function)(invert, input, str, width, height);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000862
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000863 return output;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000864}
865
Barry Warsawf630f6b1996-12-13 01:24:29 +0000866static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000867sv_InterleaveFields(PyObject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000868{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000869 return sv_conversion(self, args, svInterleaveFields, 1, 1.0);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000870}
871
Barry Warsawf630f6b1996-12-13 01:24:29 +0000872static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000873sv_RGB8toRGB32(PyObject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000874{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000875 return sv_conversion(self, args, svRGB8toRGB32, 1, (float) sizeof(long));
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000876}
877
Barry Warsawf630f6b1996-12-13 01:24:29 +0000878static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000879sv_YUVtoRGB(PyObject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000880{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000881 return sv_conversion(self, args, svYUVtoRGB, 2, (float) sizeof(long));
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000882}
883
884static void
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000885svideo_dealloc(svobject *self)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000886{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000887 if (self->ob_svideo != NULL)
888 (void) svCloseVideo(self->ob_svideo);
889 PyObject_Del(self);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000890}
891
Barry Warsawf630f6b1996-12-13 01:24:29 +0000892static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000893svideo_getattr(svobject *self, char *name)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000894{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000895 return Py_FindMethod(svideo_methods, (PyObject *)self, name);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000896}
897
Barry Warsawf630f6b1996-12-13 01:24:29 +0000898PyTypeObject Svtype = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000899 PyObject_HEAD_INIT(&PyType_Type)
900 0, /*ob_size*/
901 "sv.sv", /*tp_name*/
902 sizeof(svobject), /*tp_size*/
903 0, /*tp_itemsize*/
904 /* methods */
905 (destructor)svideo_dealloc, /*tp_dealloc*/
906 0, /*tp_print*/
907 (getattrfunc)svideo_getattr, /*tp_getattr*/
908 0, /*tp_setattr*/
909 0, /*tp_compare*/
910 0, /*tp_repr*/
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000911};
912
Barry Warsawf630f6b1996-12-13 01:24:29 +0000913static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000914newsvobject(SV_nodeP svp)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000915{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000916 svobject *p;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000917
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000918 p = PyObject_New(svobject, &Svtype);
919 if (p == NULL)
920 return NULL;
921 p->ob_svideo = svp;
922 p->ob_info.format = 0;
923 p->ob_info.size = 0;
924 p->ob_info.width = 0;
925 p->ob_info.height = 0;
926 p->ob_info.samplingrate = 0;
927 return (PyObject *) p;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000928}
929
Barry Warsawf630f6b1996-12-13 01:24:29 +0000930static PyObject *
Peter Schneider-Kamp286da3b2000-07-10 12:43:58 +0000931sv_OpenVideo(PyObject *self, PyObject *args)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000932{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000933 SV_nodeP svp;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000934
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000935 if (!PyArg_Parse(args, ""))
936 return NULL;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000937
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000938 svp = svOpenVideo();
939 if (svp == NULL)
940 return sv_error();
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000941
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000942 return newsvobject(svp);
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000943}
944
Barry Warsawf630f6b1996-12-13 01:24:29 +0000945static PyMethodDef sv_methods[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000946 {"InterleaveFields", (PyCFunction)sv_InterleaveFields, METH_OLDARGS},
947 {"RGB8toRGB32", (PyCFunction)sv_RGB8toRGB32, METH_OLDARGS},
948 {"YUVtoRGB", (PyCFunction)sv_YUVtoRGB, METH_OLDARGS},
949 {"OpenVideo", (PyCFunction)sv_OpenVideo, METH_OLDARGS},
950 {NULL, NULL} /* Sentinel */
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000951};
952
953void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000954initsv(void)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000955{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000956 PyObject *m, *d;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000957
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000958 if (PyErr_WarnPy3k("the sv module has been removed in "
959 "Python 3.0", 2) < 0)
960 return;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000961
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000962 m = Py_InitModule("sv", sv_methods);
963 if (m == NULL)
964 return;
965 d = PyModule_GetDict(m);
966
967 SvError = PyErr_NewException("sv.error", NULL, NULL);
968 if (SvError == NULL || PyDict_SetItemString(d, "error", SvError) != 0)
969 return;
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000970}