| /* Hey Emacs, this is -*-C-*- |
| ****************************************************************************** |
| * linuxaudiodev.c -- Linux audio device for python. |
| * |
| * Author : Peter Bosch |
| * Created On : Thu Mar 2 21:10:33 2000 |
| * Status : Unknown, Use with caution! |
| * |
| * Unless other notices are present in any part of this file |
| * explicitly claiming copyrights for other people and/or |
| * organizations, the contents of this file is fully copyright |
| * (C) 2000 Peter Bosch, all rights reserved. |
| ****************************************************************************** |
| */ |
| |
| #include "Python.h" |
| #include "structmember.h" |
| |
| #ifdef HAVE_FCNTL_H |
| #include <fcntl.h> |
| #else |
| #define O_RDONLY 00 |
| #define O_WRONLY 01 |
| #endif |
| |
| |
| #include <sys/ioctl.h> |
| #if defined(linux) |
| #include <linux/soundcard.h> |
| |
| #ifndef HAVE_STDINT_H |
| typedef unsigned long uint32_t; |
| #endif |
| |
| #elif defined(__FreeBSD__) |
| #include <machine/soundcard.h> |
| |
| #ifndef SNDCTL_DSP_CHANNELS |
| #define SNDCTL_DSP_CHANNELS SOUND_PCM_WRITE_CHANNELS |
| #endif |
| |
| #endif |
| |
| typedef struct { |
| PyObject_HEAD |
| int x_fd; /* The open file */ |
| int x_mode; /* file mode */ |
| int x_icount; /* Input count */ |
| int x_ocount; /* Output count */ |
| uint32_t x_afmts; /* Audio formats supported by hardware*/ |
| } lad_t; |
| |
| /* XXX several format defined in soundcard.h are not supported, |
| including _NE (native endian) options and S32 options |
| */ |
| |
| static struct { |
| int a_bps; |
| uint32_t a_fmt; |
| char *a_name; |
| } audio_types[] = { |
| { 8, AFMT_MU_LAW, "logarithmic mu-law 8-bit audio" }, |
| { 8, AFMT_A_LAW, "logarithmic A-law 8-bit audio" }, |
| { 8, AFMT_U8, "linear unsigned 8-bit audio" }, |
| { 8, AFMT_S8, "linear signed 8-bit audio" }, |
| { 16, AFMT_U16_BE, "linear unsigned 16-bit big-endian audio" }, |
| { 16, AFMT_U16_LE, "linear unsigned 16-bit little-endian audio" }, |
| { 16, AFMT_S16_BE, "linear signed 16-bit big-endian audio" }, |
| { 16, AFMT_S16_LE, "linear signed 16-bit little-endian audio" }, |
| { 16, AFMT_S16_NE, "linear signed 16-bit native-endian audio" }, |
| }; |
| |
| static int n_audio_types = sizeof(audio_types) / sizeof(audio_types[0]); |
| |
| static PyTypeObject Ladtype; |
| |
| static PyObject *LinuxAudioError; |
| |
| static lad_t * |
| newladobject(PyObject *arg) |
| { |
| lad_t *xp; |
| int fd, afmts, imode; |
| char *basedev = NULL; |
| char *mode = NULL; |
| |
| /* Two ways to call linuxaudiodev.open(): |
| open(device, mode) (for consistency with builtin open()) |
| open(mode) (for backwards compatibility) |
| because the *first* argument is optional, parsing args is |
| a wee bit tricky. */ |
| if (!PyArg_ParseTuple(arg, "s|s:open", &basedev, &mode)) |
| return NULL; |
| if (mode == NULL) { /* only one arg supplied */ |
| mode = basedev; |
| basedev = NULL; |
| } |
| |
| if (strcmp(mode, "r") == 0) |
| imode = O_RDONLY; |
| else if (strcmp(mode, "w") == 0) |
| imode = O_WRONLY; |
| else { |
| PyErr_SetString(LinuxAudioError, "mode should be 'r' or 'w'"); |
| return NULL; |
| } |
| |
| /* Open the correct device. The base device name comes from the |
| * AUDIODEV environment variable first, then /dev/dsp. The |
| * control device tacks "ctl" onto the base device name. |
| * |
| * Note that the only difference between /dev/audio and /dev/dsp |
| * is that the former uses logarithmic mu-law encoding and the |
| * latter uses 8-bit unsigned encoding. |
| */ |
| |
| if (basedev == NULL) { /* called with one arg */ |
| basedev = getenv("AUDIODEV"); |
| if (basedev == NULL) /* $AUDIODEV not set */ |
| basedev = "/dev/dsp"; |
| } |
| |
| if ((fd = open(basedev, imode)) == -1) { |
| PyErr_SetFromErrnoWithFilename(LinuxAudioError, basedev); |
| return NULL; |
| } |
| if (imode == O_WRONLY && ioctl(fd, SNDCTL_DSP_NONBLOCK, NULL) == -1) { |
| PyErr_SetFromErrnoWithFilename(LinuxAudioError, basedev); |
| return NULL; |
| } |
| if (ioctl(fd, SNDCTL_DSP_GETFMTS, &afmts) == -1) { |
| 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_mode = imode; |
| xp->x_icount = xp->x_ocount = 0; |
| xp->x_afmts = afmts; |
| return xp; |
| } |
| |
| static void |
| lad_dealloc(lad_t *xp) |
| { |
| /* if already closed, don't reclose it */ |
| if (xp->x_fd != -1) |
| close(xp->x_fd); |
| PyObject_Del(xp); |
| } |
| |
| static PyObject * |
| lad_read(lad_t *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; |
| cp = PyString_AS_STRING(rv); |
| if ((count = read(self->x_fd, cp, size)) < 0) { |
| PyErr_SetFromErrno(LinuxAudioError); |
| Py_DECREF(rv); |
| return NULL; |
| } |
| self->x_icount += count; |
| _PyString_Resize(&rv, count); |
| return rv; |
| } |
| |
| static PyObject * |
| lad_write(lad_t *self, PyObject *args) |
| { |
| char *cp; |
| int rv, size; |
| fd_set write_set_fds; |
| struct timeval tv; |
| int select_retval; |
| |
| if (!PyArg_ParseTuple(args, "s#:write", &cp, &size)) |
| return NULL; |
| |
| /* use select to wait for audio device to be available */ |
| FD_ZERO(&write_set_fds); |
| FD_SET(self->x_fd, &write_set_fds); |
| tv.tv_sec = 4; /* timeout values */ |
| tv.tv_usec = 0; |
| |
| while (size > 0) { |
| select_retval = select(self->x_fd+1, NULL, &write_set_fds, NULL, &tv); |
| tv.tv_sec = 1; tv.tv_usec = 0; /* willing to wait this long next time*/ |
| if (select_retval) { |
| if ((rv = write(self->x_fd, cp, size)) == -1) { |
| if (errno != EAGAIN) { |
| PyErr_SetFromErrno(LinuxAudioError); |
| return NULL; |
| } else { |
| errno = 0; /* EAGAIN: buffer is full, try again */ |
| } |
| } else { |
| self->x_ocount += rv; |
| size -= rv; |
| cp += rv; |
| } |
| } else { |
| /* printf("Not able to write to linux audio device within %ld seconds\n", tv.tv_sec); */ |
| PyErr_SetFromErrno(LinuxAudioError); |
| return NULL; |
| } |
| } |
| Py_INCREF(Py_None); |
| return Py_None; |
| } |
| |
| static PyObject * |
| lad_close(lad_t *self, PyObject *unused) |
| { |
| if (self->x_fd >= 0) { |
| close(self->x_fd); |
| self->x_fd = -1; |
| } |
| Py_RETURN_NONE; |
| } |
| |
| static PyObject * |
| lad_fileno(lad_t *self, PyObject *unused) |
| { |
| return PyInt_FromLong(self->x_fd); |
| } |
| |
| static PyObject * |
| lad_setparameters(lad_t *self, PyObject *args) |
| { |
| int rate, ssize, nchannels, n, fmt, emulate=0; |
| |
| if (!PyArg_ParseTuple(args, "iiii|i:setparameters", |
| &rate, &ssize, &nchannels, &fmt, &emulate)) |
| return NULL; |
| |
| if (rate < 0) { |
| PyErr_Format(PyExc_ValueError, "expected rate >= 0, not %d", |
| rate); |
| return NULL; |
| } |
| if (ssize < 0) { |
| PyErr_Format(PyExc_ValueError, "expected sample size >= 0, not %d", |
| ssize); |
| return NULL; |
| } |
| if (nchannels != 1 && nchannels != 2) { |
| PyErr_Format(PyExc_ValueError, "nchannels must be 1 or 2, not %d", |
| nchannels); |
| return NULL; |
| } |
| |
| for (n = 0; n < n_audio_types; n++) |
| if (fmt == audio_types[n].a_fmt) |
| break; |
| if (n == n_audio_types) { |
| PyErr_Format(PyExc_ValueError, "unknown audio encoding: %d", fmt); |
| return NULL; |
| } |
| if (audio_types[n].a_bps != ssize) { |
| PyErr_Format(PyExc_ValueError, |
| "for %s, expected sample size %d, not %d", |
| audio_types[n].a_name, audio_types[n].a_bps, ssize); |
| return NULL; |
| } |
| |
| if (emulate == 0) { |
| if ((self->x_afmts & audio_types[n].a_fmt) == 0) { |
| PyErr_Format(PyExc_ValueError, |
| "%s format not supported by device", |
| audio_types[n].a_name); |
| return NULL; |
| } |
| } |
| if (ioctl(self->x_fd, SNDCTL_DSP_SETFMT, |
| &audio_types[n].a_fmt) == -1) { |
| PyErr_SetFromErrno(LinuxAudioError); |
| return NULL; |
| } |
| if (ioctl(self->x_fd, SNDCTL_DSP_CHANNELS, &nchannels) == -1) { |
| PyErr_SetFromErrno(LinuxAudioError); |
| return NULL; |
| } |
| if (ioctl(self->x_fd, SNDCTL_DSP_SPEED, &rate) == -1) { |
| 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; |
| } |
| 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 */ |
| static PyObject * |
| lad_bufsize(lad_t *self, PyObject *unused) |
| { |
| audio_buf_info ai; |
| int nchannels=0, ssize=0; |
| |
| if (_ssize(self, &nchannels, &ssize) < 0 || !ssize || !nchannels) { |
| 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 |
| hardware for playing */ |
| static PyObject * |
| lad_obufcount(lad_t *self, PyObject *unused) |
| { |
| audio_buf_info ai; |
| int nchannels=0, ssize=0; |
| |
| if (_ssize(self, &nchannels, &ssize) < 0 || !ssize || !nchannels) { |
| 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 */ |
| static PyObject * |
| lad_obuffree(lad_t *self, PyObject *unused) |
| { |
| audio_buf_info ai; |
| int nchannels=0, ssize=0; |
| |
| if (_ssize(self, &nchannels, &ssize) < 0 || !ssize || !nchannels) { |
| 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 *unused) |
| { |
| if (ioctl(self->x_fd, SNDCTL_DSP_SYNC, NULL) == -1) { |
| PyErr_SetFromErrno(LinuxAudioError); |
| return NULL; |
| } |
| Py_RETURN_NONE; |
| } |
| |
| static PyObject * |
| lad_getptr(lad_t *self, PyObject *unused) |
| { |
| count_info info; |
| int req; |
| |
| if (self->x_mode == O_RDONLY) |
| req = SNDCTL_DSP_GETIPTR; |
| else |
| req = SNDCTL_DSP_GETOPTR; |
| if (ioctl(self->x_fd, req, &info) == -1) { |
| PyErr_SetFromErrno(LinuxAudioError); |
| return NULL; |
| } |
| return Py_BuildValue("iii", info.bytes, info.blocks, info.ptr); |
| } |
| |
| static PyMethodDef lad_methods[] = { |
| { "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_NOARGS }, |
| { "obuffree", (PyCFunction)lad_obuffree, METH_NOARGS }, |
| { "flush", (PyCFunction)lad_flush, METH_NOARGS }, |
| { "close", (PyCFunction)lad_close, METH_NOARGS }, |
| { "fileno", (PyCFunction)lad_fileno, METH_NOARGS }, |
| { "getptr", (PyCFunction)lad_getptr, METH_NOARGS }, |
| { NULL, NULL} /* sentinel */ |
| }; |
| |
| static PyObject * |
| lad_getattr(lad_t *xp, char *name) |
| { |
| return Py_FindMethod(lad_methods, (PyObject *)xp, name); |
| } |
| |
| static PyTypeObject Ladtype = { |
| PyVarObject_HEAD_INIT(&PyType_Type, 0) |
| "linuxaudiodev.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); |
| } |
| |
| static PyMethodDef linuxaudiodev_methods[] = { |
| { "open", ladopen, METH_VARARGS }, |
| { 0, 0 }, |
| }; |
| |
| void |
| initlinuxaudiodev(void) |
| { |
| PyObject *m; |
| |
| if (PyErr_WarnPy3k("the linuxaudiodev module has been removed in " |
| "Python 3.0; use the ossaudiodev module instead", 2) < 0) |
| return; |
| |
| m = Py_InitModule("linuxaudiodev", linuxaudiodev_methods); |
| if (m == NULL) |
| return; |
| |
| LinuxAudioError = PyErr_NewException("linuxaudiodev.error", NULL, NULL); |
| if (LinuxAudioError) |
| PyModule_AddObject(m, "error", LinuxAudioError); |
| |
| if (PyModule_AddIntConstant(m, "AFMT_MU_LAW", (long)AFMT_MU_LAW) == -1) |
| return; |
| if (PyModule_AddIntConstant(m, "AFMT_A_LAW", (long)AFMT_A_LAW) == -1) |
| return; |
| if (PyModule_AddIntConstant(m, "AFMT_U8", (long)AFMT_U8) == -1) |
| return; |
| if (PyModule_AddIntConstant(m, "AFMT_S8", (long)AFMT_S8) == -1) |
| return; |
| if (PyModule_AddIntConstant(m, "AFMT_U16_BE", (long)AFMT_U16_BE) == -1) |
| return; |
| if (PyModule_AddIntConstant(m, "AFMT_U16_LE", (long)AFMT_U16_LE) == -1) |
| return; |
| if (PyModule_AddIntConstant(m, "AFMT_S16_BE", (long)AFMT_S16_BE) == -1) |
| return; |
| if (PyModule_AddIntConstant(m, "AFMT_S16_LE", (long)AFMT_S16_LE) == -1) |
| return; |
| if (PyModule_AddIntConstant(m, "AFMT_S16_NE", (long)AFMT_S16_NE) == -1) |
| return; |
| |
| return; |
| } |
