mirror of https://github.com/python/cpython
487 lines
11 KiB
C
487 lines
11 KiB
C
|
|
/* Sad objects */
|
|
|
|
#include "Python.h"
|
|
#include "structmember.h"
|
|
|
|
#ifdef HAVE_SYS_AUDIOIO_H
|
|
#define SOLARIS
|
|
#endif
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#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;
|
|
|
|
staticforward PyTypeObject Sadtype;
|
|
staticforward 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_Parse(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 void
|
|
sad_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_Parse(args, "i", &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_Parse(args, "s#", &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, PyObject *args)
|
|
{
|
|
sadstatusobject *rv;
|
|
|
|
if (!PyArg_Parse(args, ""))
|
|
return NULL;
|
|
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, PyObject *args)
|
|
{
|
|
audio_info_t ai;
|
|
|
|
if (!PyArg_Parse(args, ""))
|
|
return NULL;
|
|
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, PyObject *args)
|
|
{
|
|
audio_info_t ai;
|
|
|
|
if (!PyArg_Parse(args, ""))
|
|
return NULL;
|
|
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, PyObject *args)
|
|
{
|
|
|
|
if (!PyArg_Parse(args, ""))
|
|
return NULL;
|
|
if (ioctl(self->x_fd, AUDIO_DRAIN, 0) < 0) {
|
|
PyErr_SetFromErrno(SunAudioError);
|
|
return NULL;
|
|
}
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
#ifdef SOLARIS
|
|
static PyObject *
|
|
sad_getdev(sadobject *self, PyObject *args)
|
|
{
|
|
struct audio_device ad;
|
|
|
|
if (!PyArg_Parse(args, ""))
|
|
return NULL;
|
|
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, PyObject *args)
|
|
{
|
|
|
|
if (!PyArg_Parse(args, ""))
|
|
return NULL;
|
|
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, PyObject *args)
|
|
{
|
|
|
|
if (!PyArg_Parse(args, ""))
|
|
return NULL;
|
|
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, PyObject *args)
|
|
{
|
|
if (!PyArg_Parse(args, ""))
|
|
return NULL;
|
|
|
|
return PyInt_FromLong(self->x_fd);
|
|
}
|
|
|
|
|
|
static PyMethodDef sad_methods[] = {
|
|
{ "read", (PyCFunction)sad_read },
|
|
{ "write", (PyCFunction)sad_write },
|
|
{ "ibufcount", (PyCFunction)sad_ibufcount },
|
|
{ "obufcount", (PyCFunction)sad_obufcount },
|
|
#define CTL_METHODS 4
|
|
{ "getinfo", (PyCFunction)sad_getinfo },
|
|
{ "setinfo", (PyCFunction)sad_setinfo },
|
|
{ "drain", (PyCFunction)sad_drain },
|
|
{ "flush", (PyCFunction)sad_flush },
|
|
#ifdef SOLARIS
|
|
{ "getdev", (PyCFunction)sad_getdev },
|
|
#endif
|
|
{ "close", (PyCFunction)sad_close },
|
|
{ "fileno", (PyCFunction)sad_fileno },
|
|
{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 void
|
|
sads_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 int
|
|
sads_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 },
|
|
{ 0, 0 },
|
|
};
|
|
|
|
void
|
|
initsunaudiodev(void)
|
|
{
|
|
PyObject *m, *d;
|
|
|
|
m = Py_InitModule("sunaudiodev", sunaudiodev_methods);
|
|
d = PyModule_GetDict(m);
|
|
SunAudioError = PyErr_NewException("sunaudiodev.error", NULL, NULL);
|
|
if (SunAudioError)
|
|
PyDict_SetItemString(d, "error", SunAudioError);
|
|
}
|