Skip to content

C
cpython
Kaydet (Commit) da940d8f authored tarafından Fred Drake's avatar Fred Drake
Dosyalara gözat
Seçenekler

Add method names to PyArg_ParseTuple() calls for better error messages. 

Convert to four-space indents.
üst 5443c49f
Hide whitespace changes
Inline Side-by-side
Showing with 298 additions and 313 deletions
Modules/linuxaudiodev.c
Dosyayı görüntüle @ da940d8f
......@@ -32,24 +32,24 @@
typedef unsigned long uint32_t;
typedef struct {
PyObject_HEAD;
int x_fd; /* The open file */
int x_icount; /* Input count */
int x_ocount; /* Output count */
uint32_t x_afmts; /* Supported audio formats */
PyObject_HEAD;
int x_fd; /* The open file */
int x_icount; /* Input count */
int x_ocount; /* Output count */
uint32_t x_afmts; /* Supported audio formats */
} lad_t;
static struct {
int a_bps;
uint32_t a_fmt;
int a_bps;
uint32_t a_fmt;
} audio_types[] = {
{ 8, AFMT_MU_LAW },
{ 8, AFMT_U8 },
{ 8, AFMT_S8 },
{ 16, AFMT_U16_BE },
{ 16, AFMT_U16_LE },
{ 16, AFMT_S16_BE },
{ 16, AFMT_S16_LE },
{ 8, AFMT_MU_LAW },
{ 8, AFMT_U8 },
{ 8, AFMT_S8 },
{ 16, AFMT_U16_BE },
{ 16, AFMT_U16_LE },
{ 16, AFMT_S16_BE },
{ 16, AFMT_S16_LE },
};
......@@ -60,238 +60,225 @@ static PyObject *LinuxAudioError;
static lad_t *
newladobject(PyObject *arg)
{
lad_t *xp;
int fd, afmts, imode;
char *mode;
char *basedev;
char *ctldev;
char *opendev;
/* Check arg for r/w/rw */
if (!PyArg_ParseTuple(arg, "s", &mode)) return NULL;
if (strcmp(mode, "r") == 0)
imode = 0;
else if (strcmp(mode, "w") == 0)
imode = 1;
else {
PyErr_SetString(LinuxAudioError, "Mode should be one of 'r', or 'w'");
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/dsp";
if ((fd = open(basedev, imode)) < 0) {
PyErr_SetFromErrnoWithFilename(LinuxAudioError, basedev);
return NULL;
}
if (imode) {
if (ioctl(fd, SNDCTL_DSP_NONBLOCK, NULL) < 0) {
PyErr_SetFromErrnoWithFilename(LinuxAudioError, basedev);
return NULL;
lad_t *xp;
int fd, afmts, imode;
char *mode;
char *basedev;
char *ctldev;
char *opendev;
/* Check arg for r/w/rw */
if (!PyArg_ParseTuple(arg, "s:open", &mode)) return NULL;
if (strcmp(mode, "r") == 0)
imode = 0;
else if (strcmp(mode, "w") == 0)
imode = 1;
else {
PyErr_SetString(LinuxAudioError, "Mode should be one of 'r', or 'w'");
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/dsp";
if ((fd = open(basedev, imode)) < 0) {
PyErr_SetFromErrnoWithFilename(LinuxAudioError, basedev);
return NULL;
}
if (imode && ioctl(fd, SNDCTL_DSP_NONBLOCK, NULL) < 0) {
PyErr_SetFromErrnoWithFilename(LinuxAudioError, basedev);
return NULL;
}
if (ioctl(fd, SNDCTL_DSP_GETFMTS, &afmts) < 0) {
PyErr_SetFromErrnoWithFilename(LinuxAudioError, basedev);
return NULL;
}
}
if (ioctl(fd, SNDCTL_DSP_GETFMTS, &afmts) < 0) {
PyErr_SetFromErrnoWithFilename(LinuxAudioError, basedev);
return NULL;
}
/* Create and initialize the object */
if ((xp = PyObject_New(lad_t, &Ladtype)) == NULL) {
close(fd);
return NULL;
}
xp->x_fd = fd;
xp->x_icount = xp->x_ocount = 0;
xp->x_afmts = afmts;
return xp;
/* Create and initialize the object */
if ((xp = PyObject_New(lad_t, &Ladtype)) == NULL) {
close(fd);
return NULL;
}
xp->x_fd = fd;
xp->x_icount = xp->x_ocount = 0;
xp->x_afmts = afmts;
return xp;
}
static void
lad_dealloc(lad_t *xp)
{
close(xp->x_fd);
PyObject_Del(xp);
close(xp->x_fd);
PyObject_Del(xp);
}
static PyObject *
lad_read(lad_t *self, PyObject *args)
{
int size, count;
char *cp;
PyObject *rv;
int size, count;
char *cp;
PyObject *rv;
if (!PyArg_ParseTuple(args, "i", &size)) return NULL;
rv = PyString_FromStringAndSize(NULL, size);
if (rv == NULL) return NULL;
if (!(cp = PyString_AsString(rv))) {
Py_DECREF(rv);
return NULL;
}
if ((count = read(self->x_fd, cp, size)) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
Py_DECREF(rv);
return NULL;
}
self->x_icount += count;
return 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))) {
Py_DECREF(rv);
return NULL;
}
if ((count = read(self->x_fd, cp, size)) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
Py_DECREF(rv);
return NULL;
}
self->x_icount += count;
return rv;
}
static PyObject *
lad_write(lad_t *self, PyObject *args)
{
char *cp;
int rv, size;
char *cp;
int rv, size;
if (!PyArg_ParseTuple(args, "s#", &cp, &size)) return NULL;
while (size > 0) {
if ((rv = write(self->x_fd, cp, size)) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
if (!PyArg_ParseTuple(args, "s#:write", &cp, &size)) return NULL;
while (size > 0) {
if ((rv = write(self->x_fd, cp, size)) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
self->x_ocount += rv;
size -= rv;
cp += rv;
}
self->x_ocount += rv;
size -= rv;
cp += rv;
}
Py_INCREF(Py_None);
return Py_None;
Py_INCREF(Py_None);
return Py_None;
}
static PyObject *
lad_close(lad_t *self, PyObject *args)
{
if (!PyArg_ParseTuple(args, "")) return NULL;
if (self->x_fd >= 0) {
close(self->x_fd);
self->x_fd = -1;
}
Py_INCREF(Py_None);
return Py_None;
if (!PyArg_ParseTuple(args, ":close")) return NULL;
if (self->x_fd >= 0) {
close(self->x_fd);
self->x_fd = -1;
}
Py_INCREF(Py_None);
return Py_None;
}
static PyObject *
lad_fileno(lad_t *self, PyObject *args)
{
if (!PyArg_ParseTuple(args, "")) return NULL;
return PyInt_FromLong(self->x_fd);
if (!PyArg_ParseTuple(args, ":fileno")) return NULL;
return PyInt_FromLong(self->x_fd);
}
static PyObject *
lad_setparameters(lad_t *self, PyObject *args)
{
int rate, ssize, nchannels, stereo, n, fmt;
int rate, ssize, nchannels, stereo, n, fmt;
if (!PyArg_ParseTuple(args, "iiii", &rate, &ssize, &nchannels, &fmt))
return NULL;
if (rate < 0 || ssize < 0 || (nchannels != 1 && nchannels != 2)) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (ioctl(self->x_fd, SOUND_PCM_WRITE_RATE, &rate) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (ioctl(self->x_fd, SNDCTL_DSP_SAMPLESIZE, &ssize) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
stereo = (nchannels == 1)? 0: (nchannels == 2)? 1: -1;
if (ioctl(self->x_fd, SNDCTL_DSP_STEREO, &stereo) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
for (n = 0; n != sizeof(audio_types) / sizeof(audio_types[0]); n++)
if (fmt == audio_types[n].a_fmt)
break;
if (n == sizeof(audio_types) / sizeof(audio_types[0]) ||
audio_types[n].a_bps != ssize ||
(self->x_afmts & audio_types[n].a_fmt) == 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (ioctl(self->x_fd, SNDCTL_DSP_SETFMT, &audio_types[n].a_fmt) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (!PyArg_ParseTuple(args, "iiii:setparameters",
&rate, &ssize, &nchannels, &fmt))
return NULL;
Py_INCREF(Py_None);
return Py_None;
if (rate < 0 || ssize < 0 || (nchannels != 1 && nchannels != 2)) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (ioctl(self->x_fd, SOUND_PCM_WRITE_RATE, &rate) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (ioctl(self->x_fd, SNDCTL_DSP_SAMPLESIZE, &ssize) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
stereo = (nchannels == 1)? 0: (nchannels == 2)? 1: -1;
if (ioctl(self->x_fd, SNDCTL_DSP_STEREO, &stereo) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
for (n = 0; n != sizeof(audio_types) / sizeof(audio_types[0]); n++)
if (fmt == audio_types[n].a_fmt)
break;
if (n == sizeof(audio_types) / sizeof(audio_types[0]) ||
audio_types[n].a_bps != ssize ||
(self->x_afmts & audio_types[n].a_fmt) == 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (ioctl(self->x_fd, SNDCTL_DSP_SETFMT, &audio_types[n].a_fmt) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
Py_INCREF(Py_None);
return Py_None;
}
static int
_ssize(lad_t *self, int *nchannels, int *ssize)
{
int fmt;
fmt = 0;
if (ioctl(self->x_fd, SNDCTL_DSP_SETFMT, &fmt) < 0)
return -errno;
switch (fmt) {
case AFMT_MU_LAW:
case AFMT_A_LAW:
case AFMT_U8:
case AFMT_S8:
*ssize = sizeof(char);
break;
case AFMT_S16_LE:
case AFMT_S16_BE:
case AFMT_U16_LE:
case AFMT_U16_BE:
*ssize = sizeof(short);
break;
case AFMT_MPEG:
case AFMT_IMA_ADPCM:
default:
return -EOPNOTSUPP;
}
*nchannels = 0;
if (ioctl(self->x_fd, SNDCTL_DSP_CHANNELS, nchannels) < 0)
return -errno;
return 0;
int fmt;
fmt = 0;
if (ioctl(self->x_fd, SNDCTL_DSP_SETFMT, &fmt) < 0)
return -errno;
switch (fmt) {
case AFMT_MU_LAW:
case AFMT_A_LAW:
case AFMT_U8:
case AFMT_S8:
*ssize = sizeof(char);
break;
case AFMT_S16_LE:
case AFMT_S16_BE:
case AFMT_U16_LE:
case AFMT_U16_BE:
*ssize = sizeof(short);
break;
case AFMT_MPEG:
case AFMT_IMA_ADPCM:
default:
return -EOPNOTSUPP;
}
*nchannels = 0;
if (ioctl(self->x_fd, SNDCTL_DSP_CHANNELS, nchannels) < 0)
return -errno;
return 0;
}
/* bufsize returns the size of the hardware audio buffer in number
of samples */
of samples */
static PyObject *
lad_bufsize(lad_t *self, PyObject *args)
{
audio_buf_info ai;
int nchannels, ssize;
if (!PyArg_ParseTuple(args, "")) return NULL;
if (_ssize(self, &nchannels, &ssize) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
audio_buf_info ai;
int nchannels, ssize;
if (ioctl(self->x_fd, SNDCTL_DSP_GETOSPACE, &ai) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (!PyArg_ParseTuple(args, ":bufsize")) return NULL;
return PyInt_FromLong((ai.fragstotal * ai.fragsize) / (nchannels * ssize));
if (_ssize(self, &nchannels, &ssize) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (ioctl(self->x_fd, SNDCTL_DSP_GETOSPACE, &ai) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
return PyInt_FromLong((ai.fragstotal * ai.fragsize) / (nchannels * ssize));
}
/* obufcount returns the number of samples that are available in the
......@@ -299,168 +286,166 @@ lad_bufsize(lad_t *self, PyObject *args)
static PyObject *
lad_obufcount(lad_t *self, PyObject *args)
{
audio_buf_info ai;
int nchannels, ssize;
audio_buf_info ai;
int nchannels, ssize;
if (!PyArg_ParseTuple(args, "")) return NULL;
if (!PyArg_ParseTuple(args, ":obufcount"))
return NULL;
if (_ssize(self, &nchannels, &ssize) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (ioctl(self->x_fd, SNDCTL_DSP_GETOSPACE, &ai) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
return PyInt_FromLong((ai.fragstotal * ai.fragsize - ai.bytes) /
(ssize * nchannels));
if (_ssize(self, &nchannels, &ssize) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (ioctl(self->x_fd, SNDCTL_DSP_GETOSPACE, &ai) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
return PyInt_FromLong((ai.fragstotal * ai.fragsize - ai.bytes) /
(ssize * nchannels));
}
/* obufcount returns the number of samples that can be played without
blocking */
blocking */
static PyObject *
lad_obuffree(lad_t *self, PyObject *args)
{
audio_buf_info ai;
int nchannels, ssize;
audio_buf_info ai;
int nchannels, ssize;
if (!PyArg_ParseTuple(args, "")) return NULL;
if (!PyArg_ParseTuple(args, ":obuffree"))
return NULL;
if (_ssize(self, &nchannels, &ssize) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (ioctl(self->x_fd, SNDCTL_DSP_GETOSPACE, &ai) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
return PyInt_FromLong(ai.bytes / (ssize * nchannels));
if (_ssize(self, &nchannels, &ssize) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
if (ioctl(self->x_fd, SNDCTL_DSP_GETOSPACE, &ai) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
return PyInt_FromLong(ai.bytes / (ssize * nchannels));
}
/* Flush the device */
static PyObject *
lad_flush(lad_t *self, PyObject *args)
{
if (!PyArg_ParseTuple(args, "")) return NULL;
if (!PyArg_ParseTuple(args, ":flush")) return NULL;
if (ioctl(self->x_fd, SNDCTL_DSP_SYNC, NULL) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
Py_INCREF(Py_None);
return Py_None;
if (ioctl(self->x_fd, SNDCTL_DSP_SYNC, NULL) < 0) {
PyErr_SetFromErrno(LinuxAudioError);
return NULL;
}
Py_INCREF(Py_None);
return Py_None;
}
static PyMethodDef lad_methods[] = {
{ "read", (PyCFunction)lad_read, 1 },
{ "write", (PyCFunction)lad_write, 1 },
{ "setparameters", (PyCFunction)lad_setparameters, 1 },
{ "bufsize", (PyCFunction)lad_bufsize, 1 },
{ "obufcount", (PyCFunction)lad_obufcount, 1 },
{ "obuffree", (PyCFunction)lad_obuffree, 1 },
{ "flush", (PyCFunction)lad_flush, 1 },
{ "close", (PyCFunction)lad_close, 1 },
{ "fileno", (PyCFunction)lad_fileno, 1 },
{ NULL, NULL} /* sentinel */
{ "read", (PyCFunction)lad_read, METH_VARARGS },
{ "write", (PyCFunction)lad_write, METH_VARARGS },
{ "setparameters", (PyCFunction)lad_setparameters, METH_VARARGS },
{ "bufsize", (PyCFunction)lad_bufsize, METH_VARARGS },
{ "obufcount", (PyCFunction)lad_obufcount, METH_VARARGS },
{ "obuffree", (PyCFunction)lad_obuffree, METH_VARARGS },
{ "flush", (PyCFunction)lad_flush, METH_VARARGS },
{ "close", (PyCFunction)lad_close, METH_VARARGS },
{ "fileno", (PyCFunction)lad_fileno, METH_VARARGS },
{ NULL, NULL} /* sentinel */
};
static PyObject *
lad_getattr(lad_t *xp, char *name)
{
return Py_FindMethod(lad_methods, (PyObject *)xp, name);
return Py_FindMethod(lad_methods, (PyObject *)xp, name);
}
static PyTypeObject Ladtype = {
PyObject_HEAD_INIT(&PyType_Type)
0, /*ob_size*/
"linux_audio_device", /*tp_name*/
sizeof(lad_t), /*tp_size*/
0, /*tp_itemsize*/
/* methods */
(destructor)lad_dealloc, /*tp_dealloc*/
0, /*tp_print*/
(getattrfunc)lad_getattr, /*tp_getattr*/
0, /*tp_setattr*/
0, /*tp_compare*/
0, /*tp_repr*/
PyObject_HEAD_INIT(&PyType_Type)
0, /*ob_size*/
"linux_audio_device", /*tp_name*/
sizeof(lad_t), /*tp_size*/
0, /*tp_itemsize*/
/* methods */
(destructor)lad_dealloc, /*tp_dealloc*/
0, /*tp_print*/
(getattrfunc)lad_getattr, /*tp_getattr*/
0, /*tp_setattr*/
0, /*tp_compare*/
0, /*tp_repr*/
};
static PyObject *
ladopen(PyObject *self, PyObject *args)
{
return (PyObject *)newladobject(args);
return (PyObject *)newladobject(args);
}
static PyMethodDef linuxaudiodev_methods[] = {
{ "open", ladopen, 1 },
{ 0, 0 },
{ "open", ladopen, METH_VARARGS },
{ 0, 0 },
};
static int
ins(PyObject *d, char *symbol, long value)
{
PyObject* v = PyInt_FromLong(value);
if (!v || PyDict_SetItemString(d, symbol, v) < 0)
return -1; /* triggers fatal error */
PyObject* v = PyInt_FromLong(value);
if (!v || PyDict_SetItemString(d, symbol, v) < 0)
return -1; /* triggers fatal error */
Py_DECREF(v);
return 0;
Py_DECREF(v);
return 0;
}
void
initlinuxaudiodev()
{
PyObject *m, *d, *x;
PyObject *m, *d, *x;
m = Py_InitModule("linuxaudiodev", linuxaudiodev_methods);
d = PyModule_GetDict(m);
LinuxAudioError = PyErr_NewException("linuxaudiodev.error", NULL, NULL);
if (LinuxAudioError)
PyDict_SetItemString(d, "error", LinuxAudioError);
x = PyInt_FromLong((long) AFMT_MU_LAW);
if (x == NULL || PyDict_SetItemString(d, "AFMT_MU_LAW", x) < 0)
goto error;
Py_DECREF(x);
x = PyInt_FromLong((long) AFMT_U8);
if (x == NULL || PyDict_SetItemString(d, "AFMT_U8", x) < 0)
goto error;
Py_DECREF(x);
x = PyInt_FromLong((long) AFMT_S8);
if (x == NULL || PyDict_SetItemString(d, "AFMT_S8", x) < 0)
goto error;
Py_DECREF(x);
x = PyInt_FromLong((long) AFMT_U16_BE);
if (x == NULL || PyDict_SetItemString(d, "AFMT_U16_BE", x) < 0)
goto error;
Py_DECREF(x);
x = PyInt_FromLong((long) AFMT_U16_LE);
if (x == NULL || PyDict_SetItemString(d, "AFMT_U16_LE", x) < 0)
goto error;
Py_DECREF(x);
x = PyInt_FromLong((long) AFMT_S16_BE);
if (x == NULL || PyDict_SetItemString(d, "AFMT_S16_BE", x) < 0)
goto error;
Py_DECREF(x);
x = PyInt_FromLong((long) AFMT_S16_LE);
if (x == NULL || PyDict_SetItemString(d, "AFMT_S16_LE", x) < 0)
goto error;
Py_DECREF(x);
/* Check for errors */
if (PyErr_Occurred()) {
error:
Py_FatalError("can't initialize module linuxaudiodev");
}
m = Py_InitModule("linuxaudiodev", linuxaudiodev_methods);
d = PyModule_GetDict(m);
LinuxAudioError = PyErr_NewException("linuxaudiodev.error", NULL, NULL);
if (LinuxAudioError)
PyDict_SetItemString(d, "error", LinuxAudioError);
x = PyInt_FromLong((long) AFMT_MU_LAW);
if (x == NULL || PyDict_SetItemString(d, "AFMT_MU_LAW", x) < 0)
goto error;
Py_DECREF(x);
x = PyInt_FromLong((long) AFMT_U8);
if (x == NULL || PyDict_SetItemString(d, "AFMT_U8", x) < 0)
goto error;
Py_DECREF(x);
x = PyInt_FromLong((long) AFMT_S8);
if (x == NULL || PyDict_SetItemString(d, "AFMT_S8", x) < 0)
goto error;
Py_DECREF(x);
x = PyInt_FromLong((long) AFMT_U16_BE);
if (x == NULL || PyDict_SetItemString(d, "AFMT_U16_BE", x) < 0)
goto error;
Py_DECREF(x);
x = PyInt_FromLong((long) AFMT_U16_LE);
if (x == NULL || PyDict_SetItemString(d, "AFMT_U16_LE", x) < 0)
goto error;
Py_DECREF(x);
x = PyInt_FromLong((long) AFMT_S16_BE);
if (x == NULL || PyDict_SetItemString(d, "AFMT_S16_BE", x) < 0)
goto error;
Py_DECREF(x);
x = PyInt_FromLong((long) AFMT_S16_LE);
if (x == NULL || PyDict_SetItemString(d, "AFMT_S16_LE", x) < 0)
goto error;
Py_DECREF(x);
/* Check for errors */
if (PyErr_Occurred()) {
error:
Py_FatalError("can't initialize module linuxaudiodev");
}
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment