/* 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 = PyTuple_Pack(2, 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");