cinap_lenrekadd hg and python4bb9e2215 years agoHistory
465 lines · 10.4 KB · C
RawDownload
 /* Sad objects */ #include "Python.h"#include "structmember.h" #ifdef HAVE_SYS_AUDIOIO_H#define SOLARIS#endif #ifdef HAVE_FCNTL_H#include <fcntl.h>#endif #include <stropts.h>#include <sys/ioctl.h>#ifdef SOLARIS#include <sys/audioio.h>#else#include <sun/audioio.h>#endif /* #define offsetof(str,mem) ((int)(((str *)0)->mem)) */ typedef struct {	PyObject_HEAD	int	x_fd;		/* The open file */	int	x_icount;	/* # samples read */	int	x_ocount;	/* # samples written */	int	x_isctl;	/* True if control device */	} sadobject; typedef struct {	PyObject_HEAD	audio_info_t ai;} sadstatusobject; static PyTypeObject Sadtype;static PyTypeObject Sadstatustype;static sadstatusobject *sads_alloc(void);	/* Forward */ static PyObject *SunAudioError; #define is_sadobject(v)		((v)->ob_type == &Sadtype)#define is_sadstatusobject(v)	((v)->ob_type == &Sadstatustype)  static sadobject *newsadobject(PyObject *args){	sadobject *xp;	int fd;	char *mode;	int imode;	char* basedev;	char* ctldev;	char* opendev; 	/* Check arg for r/w/rw */	if (!PyArg_ParseTuple(args, "s", &mode))		return NULL;	if (strcmp(mode, "r") == 0)		imode = 0;	else if (strcmp(mode, "w") == 0)		imode = 1;	else if (strcmp(mode, "rw") == 0)		imode = 2;	else if (strcmp(mode, "control") == 0)		imode = -1;	else {		PyErr_SetString(SunAudioError,			  "Mode should be one of 'r', 'w', 'rw' or 'control'");		return NULL;	}		/* Open the correct device.  The base device name comes from the	 * AUDIODEV environment variable first, then /dev/audio.  The	 * control device tacks "ctl" onto the base device name.	 */	basedev = getenv("AUDIODEV");	if (!basedev)		basedev = "/dev/audio";	ctldev = PyMem_NEW(char, strlen(basedev) + 4);	if (!ctldev) {		PyErr_NoMemory();		return NULL;	}	strcpy(ctldev, basedev);	strcat(ctldev, "ctl"); 	if (imode < 0) {		opendev = ctldev;		fd = open(ctldev, 2);	}	else {		opendev = basedev;		fd = open(basedev, imode);	}	if (fd < 0) {		PyErr_SetFromErrnoWithFilename(SunAudioError, opendev);		PyMem_DEL(ctldev);		return NULL;	}	PyMem_DEL(ctldev); 	/* Create and initialize the object */	xp = PyObject_New(sadobject, &Sadtype);	if (xp == NULL) {		close(fd);		return NULL;	}	xp->x_fd = fd;	xp->x_icount = xp->x_ocount = 0;	xp->x_isctl = (imode < 0);		return xp;} /* Sad methods */ static voidsad_dealloc(sadobject *xp){        close(xp->x_fd);	PyObject_Del(xp);} static PyObject *sad_read(sadobject *self, PyObject *args){        int size, count;	char *cp;	PyObject *rv;	        if (!PyArg_ParseTuple(args, "i:read", &size))		return NULL;	rv = PyString_FromStringAndSize(NULL, size);	if (rv == NULL)		return NULL; 	if (!(cp = PyString_AsString(rv)))		goto finally; 	count = read(self->x_fd, cp, size);	if (count < 0) {		PyErr_SetFromErrno(SunAudioError);		goto finally;	}#if 0	/* TBD: why print this message if you can handle the condition?	 * assume it's debugging info which we can just as well get rid	 * of.  in any case this message should *not* be using printf!	 */	if (count != size)		printf("sunaudio: funny read rv %d wtd %d\n", count, size);#endif	self->x_icount += count;	return rv;   finally:	Py_DECREF(rv);	return NULL;} static PyObject *sad_write(sadobject *self, PyObject *args){        char *cp;	int count, size;	        if (!PyArg_ParseTuple(args, "s#:write", &cp, &size))		return NULL; 	count = write(self->x_fd, cp, size);	if (count < 0) {		PyErr_SetFromErrno(SunAudioError);		return NULL;	}#if 0	if (count != size)		printf("sunaudio: funny write rv %d wanted %d\n", count, size);#endif	self->x_ocount += count;		Py_INCREF(Py_None);	return Py_None;} static PyObject *sad_getinfo(sadobject *self){	sadstatusobject *rv; 	if (!(rv = sads_alloc()))		return NULL; 	if (ioctl(self->x_fd, AUDIO_GETINFO, &rv->ai) < 0) {		PyErr_SetFromErrno(SunAudioError);		Py_DECREF(rv);		return NULL;	}	return (PyObject *)rv;} static PyObject *sad_setinfo(sadobject *self, sadstatusobject *arg){	if (!is_sadstatusobject(arg)) {		PyErr_SetString(PyExc_TypeError,				"Must be sun audio status object");		return NULL;	}	if (ioctl(self->x_fd, AUDIO_SETINFO, &arg->ai) < 0) {		PyErr_SetFromErrno(SunAudioError);		return NULL;	}	Py_INCREF(Py_None);	return Py_None;} static PyObject *sad_ibufcount(sadobject *self){	audio_info_t ai;    	if (ioctl(self->x_fd, AUDIO_GETINFO, &ai) < 0) {		PyErr_SetFromErrno(SunAudioError);		return NULL;	}	return PyInt_FromLong(ai.record.samples - self->x_icount);} static PyObject *sad_obufcount(sadobject *self){	audio_info_t ai;    	if (ioctl(self->x_fd, AUDIO_GETINFO, &ai) < 0) {		PyErr_SetFromErrno(SunAudioError);		return NULL;	}	/* x_ocount is in bytes, whereas play.samples is in frames */	/* we want frames */	return PyInt_FromLong(self->x_ocount / (ai.play.channels *						ai.play.precision / 8) -			      ai.play.samples);} static PyObject *sad_drain(sadobject *self){	if (ioctl(self->x_fd, AUDIO_DRAIN, 0) < 0) {		PyErr_SetFromErrno(SunAudioError);		return NULL;	}	Py_INCREF(Py_None);	return Py_None;} #ifdef SOLARISstatic PyObject *sad_getdev(sadobject *self){	struct audio_device ad; 	if (ioctl(self->x_fd, AUDIO_GETDEV, &ad) < 0) {		PyErr_SetFromErrno(SunAudioError);		return NULL;	}	return Py_BuildValue("(sss)", ad.name, ad.version, ad.config);}#endif static PyObject *sad_flush(sadobject *self){	if (ioctl(self->x_fd, I_FLUSH, FLUSHW) < 0) {		PyErr_SetFromErrno(SunAudioError);		return NULL;	}	Py_INCREF(Py_None);	return Py_None;} static PyObject *sad_close(sadobject *self){    	if (self->x_fd >= 0) {		close(self->x_fd);		self->x_fd = -1;	}	Py_INCREF(Py_None);	return Py_None;} static PyObject *sad_fileno(sadobject *self){	return PyInt_FromLong(self->x_fd);}  static PyMethodDef sad_methods[] = {        { "read",	(PyCFunction)sad_read, METH_VARARGS },        { "write",	(PyCFunction)sad_write, METH_VARARGS },        { "ibufcount",	(PyCFunction)sad_ibufcount, METH_NOARGS },        { "obufcount",	(PyCFunction)sad_obufcount, METH_NOARGS },#define CTL_METHODS 4        { "getinfo",	(PyCFunction)sad_getinfo, METH_NOARGS },        { "setinfo",	(PyCFunction)sad_setinfo, METH_O},        { "drain",	(PyCFunction)sad_drain, METH_NOARGS },        { "flush",	(PyCFunction)sad_flush, METH_NOARGS },#ifdef SOLARIS	{ "getdev",	(PyCFunction)sad_getdev, METH_NOARGS },#endif        { "close",	(PyCFunction)sad_close, METH_NOARGS },	{ "fileno",     (PyCFunction)sad_fileno, METH_NOARGS },	{NULL,		NULL}		/* sentinel */}; static PyObject *sad_getattr(sadobject *xp, char *name){	if (xp->x_isctl)		return Py_FindMethod(sad_methods+CTL_METHODS,				     (PyObject *)xp, name);	else		return Py_FindMethod(sad_methods, (PyObject *)xp, name);} /* ----------------------------------------------------------------- */ static sadstatusobject *sads_alloc(void) {	return PyObject_New(sadstatusobject, &Sadstatustype);} static voidsads_dealloc(sadstatusobject *xp){	PyMem_DEL(xp);} #define OFF(x) offsetof(audio_info_t,x)static struct memberlist sads_ml[] = {	{ "i_sample_rate",	T_UINT,		OFF(record.sample_rate) },	{ "i_channels",		T_UINT,		OFF(record.channels) },	{ "i_precision",	T_UINT,		OFF(record.precision) },	{ "i_encoding",		T_UINT,		OFF(record.encoding) },	{ "i_gain",		T_UINT,		OFF(record.gain) },	{ "i_port",		T_UINT,		OFF(record.port) },	{ "i_samples",		T_UINT,		OFF(record.samples) },	{ "i_eof",		T_UINT,		OFF(record.eof) },	{ "i_pause",		T_UBYTE,	OFF(record.pause) },	{ "i_error",		T_UBYTE,	OFF(record.error) },	{ "i_waiting",		T_UBYTE,	OFF(record.waiting) },	{ "i_open",		T_UBYTE,	OFF(record.open) ,	 RO},	{ "i_active",		T_UBYTE,	OFF(record.active) ,	 RO},#ifdef SOLARIS	{ "i_buffer_size",	T_UINT,		OFF(record.buffer_size) },	{ "i_balance",		T_UBYTE,	OFF(record.balance) },	{ "i_avail_ports",	T_UINT,		OFF(record.avail_ports) },#endif 	{ "o_sample_rate",	T_UINT,		OFF(play.sample_rate) },	{ "o_channels",		T_UINT,		OFF(play.channels) },	{ "o_precision",	T_UINT,		OFF(play.precision) },	{ "o_encoding",		T_UINT,		OFF(play.encoding) },	{ "o_gain",		T_UINT,		OFF(play.gain) },	{ "o_port",		T_UINT,		OFF(play.port) },	{ "o_samples",		T_UINT,		OFF(play.samples) },	{ "o_eof",		T_UINT,		OFF(play.eof) },	{ "o_pause",		T_UBYTE,	OFF(play.pause) },	{ "o_error",		T_UBYTE,	OFF(play.error) },	{ "o_waiting",		T_UBYTE,	OFF(play.waiting) },	{ "o_open",		T_UBYTE,	OFF(play.open) ,	 RO},	{ "o_active",		T_UBYTE,	OFF(play.active) ,	 RO},#ifdef SOLARIS	{ "o_buffer_size",	T_UINT,		OFF(play.buffer_size) },	{ "o_balance",		T_UBYTE,	OFF(play.balance) },	{ "o_avail_ports",	T_UINT,		OFF(play.avail_ports) },#endif 	{ "monitor_gain",	T_UINT,		OFF(monitor_gain) },        { NULL,                 0,              0},}; static PyObject *sads_getattr(sadstatusobject *xp, char *name){	return PyMember_Get((char *)&xp->ai, sads_ml, name);} static intsads_setattr(sadstatusobject *xp, char *name, PyObject *v){ 	if (v == NULL) {		PyErr_SetString(PyExc_TypeError,				"can't delete sun audio status attributes");		return -1;	}	return PyMember_Set((char *)&xp->ai, sads_ml, name, v);} /* ------------------------------------------------------------------- */  static PyTypeObject Sadtype = {	PyObject_HEAD_INIT(&PyType_Type)	0,				/*ob_size*/	"sunaudiodev.sun_audio_device",	/*tp_name*/	sizeof(sadobject),		/*tp_size*/	0,				/*tp_itemsize*/	/* methods */	(destructor)sad_dealloc,	/*tp_dealloc*/	0,				/*tp_print*/	(getattrfunc)sad_getattr,	/*tp_getattr*/	0,				/*tp_setattr*/	0,				/*tp_compare*/	0,				/*tp_repr*/}; static PyTypeObject Sadstatustype = {	PyObject_HEAD_INIT(&PyType_Type)	0,				/*ob_size*/	"sunaudiodev.sun_audio_device_status", /*tp_name*/	sizeof(sadstatusobject),	/*tp_size*/	0,				/*tp_itemsize*/	/* methods */	(destructor)sads_dealloc,	/*tp_dealloc*/	0,				/*tp_print*/	(getattrfunc)sads_getattr,	/*tp_getattr*/	(setattrfunc)sads_setattr,	/*tp_setattr*/	0,				/*tp_compare*/	0,				/*tp_repr*/};/* ------------------------------------------------------------------- */ static PyObject *sadopen(PyObject *self, PyObject *args){	return (PyObject *)newsadobject(args);}    static PyMethodDef sunaudiodev_methods[] = {    { "open", sadopen, METH_VARARGS },    { 0, 0 },}; voidinitsunaudiodev(void){	PyObject *m, *d; 	m = Py_InitModule("sunaudiodev", sunaudiodev_methods);	if (m == NULL)		return;	d = PyModule_GetDict(m);	SunAudioError = PyErr_NewException("sunaudiodev.error", NULL, NULL);	if (SunAudioError)		PyDict_SetItemString(d, "error", SunAudioError);}