summaryrefslogtreecommitdiff
path: root/numpy
diff options
context:
space:
mode:
authorCharles Harris <charlesr.harris@gmail.com>2008-07-13 22:15:13 +0000
committerCharles Harris <charlesr.harris@gmail.com>2008-07-13 22:15:13 +0000
commitc62bdcbb4bfbed32bc3df144ece9cce1dc11e775 (patch)
tree403620abd1354306d90beb3bf90ce4ce0ca1d126 /numpy
parent13b05eea92552d0d65f49033af5267b84049b2ee (diff)
downloadnumpy-c62bdcbb4bfbed32bc3df144ece9cce1dc11e775.tar.gz
Indentation and coding style cleanups.
Diffstat (limited to 'numpy')
-rw-r--r--numpy/fft/fftpack_litemodule.c417
1 files changed, 229 insertions, 188 deletions
diff --git a/numpy/fft/fftpack_litemodule.c b/numpy/fft/fftpack_litemodule.c
index 4abfe84d3..7df537eb8 100644
--- a/numpy/fft/fftpack_litemodule.c
+++ b/numpy/fft/fftpack_litemodule.c
@@ -6,82 +6,100 @@ static PyObject *ErrorObject;
/* ----------------------------------------------------- */
-static char fftpack_cfftf__doc__[] ="";
+static char fftpack_cfftf__doc__[] = "";
PyObject *
fftpack_cfftf(PyObject *self, PyObject *args)
{
- PyObject *op1, *op2;
- PyArrayObject *data;
- double *wsave, *dptr;
- int npts, nsave, nrepeats, i;
-
- if(!PyArg_ParseTuple(args, "OO", &op1, &op2)) return NULL;
- data = (PyArrayObject *)PyArray_CopyFromObject(op1, PyArray_CDOUBLE, 1, 0);
- if (data == NULL) return NULL;
- if (PyArray_As1D(&op2, (char **)&wsave, &nsave, PyArray_DOUBLE) == -1)
- goto fail;
- if (data == NULL) goto fail;
-
- npts = data->dimensions[data->nd-1];
- if (nsave != npts*4+15) {
- PyErr_SetString(ErrorObject, "invalid work array for fft size");
- goto fail;
- }
+ PyObject *op1, *op2;
+ PyArrayObject *data;
+ double *wsave, *dptr;
+ int npts, nsave, nrepeats, i;
+
+ if(!PyArg_ParseTuple(args, "OO", &op1, &op2)) {
+ return NULL;
+ }
+ data = (PyArrayObject *)PyArray_CopyFromObject(op1,
+ PyArray_CDOUBLE, 1, 0);
+ if (data == NULL) {
+ return NULL;
+ }
+ if (PyArray_As1D(&op2, (char **)&wsave, &nsave, PyArray_DOUBLE) == -1) {
+ goto fail;
+ }
+ if (data == NULL) {
+ goto fail;
+ }
+
+ npts = data->dimensions[data->nd - 1];
+ if (nsave != npts*4 + 15) {
+ PyErr_SetString(ErrorObject, "invalid work array for fft size");
+ goto fail;
+ }
+
+ nrepeats = PyArray_SIZE(data)/npts;
+ dptr = (double *)data->data;
+ NPY_SIGINT_ON;
+ for (i = 0; i < nrepeats; i++) {
+ cfftf(npts, dptr, wsave);
+ dptr += npts*2;
+ }
+ NPY_SIGINT_OFF;
+ PyArray_Free(op2, (char *)wsave);
+ return (PyObject *)data;
- nrepeats = PyArray_SIZE(data)/npts;
- dptr = (double *)data->data;
- NPY_SIGINT_ON
- for (i=0; i<nrepeats; i++) {
- cfftf(npts, dptr, wsave);
- dptr += npts*2;
- }
- NPY_SIGINT_OFF
- PyArray_Free(op2, (char *)wsave);
- return (PyObject *)data;
fail:
- PyArray_Free(op2, (char *)wsave);
- Py_DECREF(data);
- return NULL;
+ PyArray_Free(op2, (char *)wsave);
+ Py_DECREF(data);
+ return NULL;
}
-static char fftpack_cfftb__doc__[] ="";
+static char fftpack_cfftb__doc__[] = "";
PyObject *
fftpack_cfftb(PyObject *self, PyObject *args)
{
- PyObject *op1, *op2;
- PyArrayObject *data;
- double *wsave, *dptr;
- int npts, nsave, nrepeats, i;
-
- if(!PyArg_ParseTuple(args, "OO", &op1, &op2)) return NULL;
- data = (PyArrayObject *)PyArray_CopyFromObject(op1, PyArray_CDOUBLE, 1, 0);
- if (data == NULL) return NULL;
- if (PyArray_As1D(&op2, (char **)&wsave, &nsave, PyArray_DOUBLE) == -1)
- goto fail;
- if (data == NULL) goto fail;
-
- npts = data->dimensions[data->nd-1];
- if (nsave != npts*4+15) {
- PyErr_SetString(ErrorObject, "invalid work array for fft size");
- goto fail;
- }
+ PyObject *op1, *op2;
+ PyArrayObject *data;
+ double *wsave, *dptr;
+ int npts, nsave, nrepeats, i;
+
+ if(!PyArg_ParseTuple(args, "OO", &op1, &op2)) {
+ return NULL;
+ }
+ data = (PyArrayObject *)PyArray_CopyFromObject(op1,
+ PyArray_CDOUBLE, 1, 0);
+ if (data == NULL) {
+ return NULL;
+ }
+ if (PyArray_As1D(&op2, (char **)&wsave, &nsave, PyArray_DOUBLE) == -1) {
+ goto fail;
+ }
+ if (data == NULL) {
+ goto fail;
+ }
+
+ npts = data->dimensions[data->nd - 1];
+ if (nsave != npts*4 + 15) {
+ PyErr_SetString(ErrorObject, "invalid work array for fft size");
+ goto fail;
+ }
+
+ nrepeats = PyArray_SIZE(data)/npts;
+ dptr = (double *)data->data;
+ NPY_SIGINT_ON;
+ for (i = 0; i < nrepeats; i++) {
+ cfftb(npts, dptr, wsave);
+ dptr += npts*2;
+ }
+ NPY_SIGINT_OFF;
+ PyArray_Free(op2, (char *)wsave);
+ return (PyObject *)data;
- nrepeats = PyArray_SIZE(data)/npts;
- dptr = (double *)data->data;
- NPY_SIGINT_ON
- for (i=0; i<nrepeats; i++) {
- cfftb(npts, dptr, wsave);
- dptr += npts*2;
- }
- NPY_SIGINT_OFF
- PyArray_Free(op2, (char *)wsave);
- return (PyObject *)data;
fail:
- PyArray_Free(op2, (char *)wsave);
- Py_DECREF(data);
- return NULL;
+ PyArray_Free(op2, (char *)wsave);
+ Py_DECREF(data);
+ return NULL;
}
static char fftpack_cffti__doc__[] ="";
@@ -89,21 +107,25 @@ static char fftpack_cffti__doc__[] ="";
static PyObject *
fftpack_cffti(PyObject *self, PyObject *args)
{
- PyArrayObject *op;
- int dim, n;
-
- if (!PyArg_ParseTuple(args, "i", &n)) return NULL;
-
- dim = 4*n+15; /*Magic size needed by cffti*/
- /*Create a 1 dimensional array of dimensions of type double*/
- op = (PyArrayObject *)PyArray_FromDims(1, &dim, PyArray_DOUBLE);
- if (op == NULL) return NULL;
-
- NPY_SIGINT_ON
- cffti(n, (double *)((PyArrayObject*)op)->data);
- NPY_SIGINT_OFF
-
- return (PyObject *)op;
+ PyArrayObject *op;
+ int dim, n;
+
+ if (!PyArg_ParseTuple(args, "i", &n)) {
+ return NULL;
+ }
+ /*Magic size needed by cffti*/
+ dim = 4*n + 15;
+ /*Create a 1 dimensional array of dimensions of type double*/
+ op = (PyArrayObject *)PyArray_FromDims(1, &dim, PyArray_DOUBLE);
+ if (op == NULL) {
+ return NULL;
+ }
+
+ NPY_SIGINT_ON;
+ cffti(n, (double *)((PyArrayObject*)op)->data);
+ NPY_SIGINT_OFF;
+
+ return (PyObject *)op;
}
static char fftpack_rfftf__doc__[] ="";
@@ -111,53 +133,61 @@ static char fftpack_rfftf__doc__[] ="";
PyObject *
fftpack_rfftf(PyObject *self, PyObject *args)
{
- PyObject *op1, *op2;
- PyArrayObject *data, *ret;
- double *wsave, *dptr, *rptr;
- int npts, nsave, nrepeats, i, rstep;
-
- if(!PyArg_ParseTuple(args, "OO", &op1, &op2)) return NULL;
- data = (PyArrayObject *)PyArray_ContiguousFromObject(op1, PyArray_DOUBLE, 1, 0);
- if (data == NULL) return NULL;
- npts = data->dimensions[data->nd-1];
- data->dimensions[data->nd-1] = npts/2+1;
- ret = (PyArrayObject *)PyArray_Zeros(data->nd, data->dimensions,
- PyArray_DescrFromType(PyArray_CDOUBLE), 0);
- data->dimensions[data->nd-1] = npts;
- rstep = (ret->dimensions[ret->nd-1])*2;
-
- if (PyArray_As1D(&op2, (char **)&wsave, &nsave, PyArray_DOUBLE) == -1)
- goto fail;
- if (data == NULL || ret == NULL) goto fail;
-
- if (nsave != npts*2+15) {
- PyErr_SetString(ErrorObject, "invalid work array for fft size");
- goto fail;
- }
-
- nrepeats = PyArray_SIZE(data)/npts;
- rptr = (double *)ret->data;
- dptr = (double *)data->data;
+ PyObject *op1, *op2;
+ PyArrayObject *data, *ret;
+ double *wsave, *dptr, *rptr;
+ int npts, nsave, nrepeats, i, rstep;
+
+ if(!PyArg_ParseTuple(args, "OO", &op1, &op2)) {
+ return NULL;
+ }
+ data = (PyArrayObject *)PyArray_ContiguousFromObject(op1,
+ PyArray_DOUBLE, 1, 0);
+ if (data == NULL) {
+ return NULL;
+ }
+ npts = data->dimensions[data->nd-1];
+ data->dimensions[data->nd - 1] = npts/2 + 1;
+ ret = (PyArrayObject *)PyArray_Zeros(data->nd, data->dimensions,
+ PyArray_DescrFromType(PyArray_CDOUBLE), 0);
+ data->dimensions[data->nd - 1] = npts;
+ rstep = (ret->dimensions[ret->nd - 1])*2;
+
+ if (PyArray_As1D(&op2, (char **)&wsave, &nsave, PyArray_DOUBLE) == -1) {
+ goto fail;
+ }
+ if (data == NULL || ret == NULL) {
+ goto fail;
+ }
+ if (nsave != npts*2+15) {
+ PyErr_SetString(ErrorObject, "invalid work array for fft size");
+ goto fail;
+ }
+
+ nrepeats = PyArray_SIZE(data)/npts;
+ rptr = (double *)ret->data;
+ dptr = (double *)data->data;
+
+
+ NPY_SIGINT_ON;
+ for (i = 0; i < nrepeats; i++) {
+ memcpy((char *)(rptr+1), dptr, npts*sizeof(double));
+ rfftf(npts, rptr+1, wsave);
+ rptr[0] = rptr[1];
+ rptr[1] = 0.0;
+ rptr += rstep;
+ dptr += npts;
+ }
+ NPY_SIGINT_OFF;
+ PyArray_Free(op2, (char *)wsave);
+ Py_DECREF(data);
+ return (PyObject *)ret;
-
- NPY_SIGINT_ON
- for (i=0; i<nrepeats; i++) {
- memcpy((char *)(rptr+1), dptr, npts*sizeof(double));
- rfftf(npts, rptr+1, wsave);
- rptr[0] = rptr[1];
- rptr[1] = 0.0;
- rptr += rstep;
- dptr += npts;
- }
- NPY_SIGINT_OFF
- PyArray_Free(op2, (char *)wsave);
- Py_DECREF(data);
- return (PyObject *)ret;
fail:
- PyArray_Free(op2, (char *)wsave);
- Py_XDECREF(data);
- Py_XDECREF(ret);
- return NULL;
+ PyArray_Free(op2, (char *)wsave);
+ Py_XDECREF(data);
+ Py_XDECREF(ret);
+ return NULL;
}
static char fftpack_rfftb__doc__[] ="";
@@ -166,48 +196,56 @@ static char fftpack_rfftb__doc__[] ="";
PyObject *
fftpack_rfftb(PyObject *self, PyObject *args)
{
- PyObject *op1, *op2;
- PyArrayObject *data, *ret;
- double *wsave, *dptr, *rptr;
- int npts, nsave, nrepeats, i;
-
- if(!PyArg_ParseTuple(args, "OO", &op1, &op2)) return NULL;
- data = (PyArrayObject *)PyArray_ContiguousFromObject(op1, PyArray_CDOUBLE, 1, 0);
- if (data == NULL) return NULL;
- npts = data->dimensions[data->nd-1];
- ret = (PyArrayObject *)PyArray_Zeros(data->nd, data->dimensions,
- PyArray_DescrFromType(PyArray_DOUBLE), 0);
-
- if (PyArray_As1D(&op2, (char **)&wsave, &nsave, PyArray_DOUBLE) == -1)
- goto fail;
- if (data == NULL || ret == NULL) goto fail;
-
- if (nsave != npts*2+15) {
- PyErr_SetString(ErrorObject, "invalid work array for fft size");
- goto fail;
- }
+ PyObject *op1, *op2;
+ PyArrayObject *data, *ret;
+ double *wsave, *dptr, *rptr;
+ int npts, nsave, nrepeats, i;
+
+ if(!PyArg_ParseTuple(args, "OO", &op1, &op2)) {
+ return NULL;
+ }
+ data = (PyArrayObject *)PyArray_ContiguousFromObject(op1,
+ PyArray_CDOUBLE, 1, 0);
+ if (data == NULL) {
+ return NULL;
+ }
+ npts = data->dimensions[data->nd - 1];
+ ret = (PyArrayObject *)PyArray_Zeros(data->nd, data->dimensions,
+ PyArray_DescrFromType(PyArray_DOUBLE), 0);
+
+ if (PyArray_As1D(&op2, (char **)&wsave, &nsave, PyArray_DOUBLE) == -1) {
+ goto fail;
+ }
+ if (data == NULL || ret == NULL) {
+ goto fail;
+ }
+ if (nsave != npts*2 + 15) {
+ PyErr_SetString(ErrorObject, "invalid work array for fft size");
+ goto fail;
+ }
+
+ nrepeats = PyArray_SIZE(ret)/npts;
+ rptr = (double *)ret->data;
+ dptr = (double *)data->data;
+
+ NPY_SIGINT_ON;
+ for (i=0; i<nrepeats; i++) {
+ memcpy((char *)(rptr+1), (dptr+2), (npts-1)*sizeof(double));
+ rptr[0] = dptr[0];
+ rfftb(npts, rptr, wsave);
+ rptr += npts;
+ dptr += npts*2;
+ }
+ NPY_SIGINT_OFF;
+ PyArray_Free(op2, (char *)wsave);
+ Py_DECREF(data);
+ return (PyObject *)ret;
- nrepeats = PyArray_SIZE(ret)/npts;
- rptr = (double *)ret->data;
- dptr = (double *)data->data;
-
- NPY_SIGINT_ON
- for (i=0; i<nrepeats; i++) {
- memcpy((char *)(rptr+1), (dptr+2), (npts-1)*sizeof(double));
- rptr[0] = dptr[0];
- rfftb(npts, rptr, wsave);
- rptr += npts;
- dptr += npts*2;
- }
- NPY_SIGINT_OFF
- PyArray_Free(op2, (char *)wsave);
- Py_DECREF(data);
- return (PyObject *)ret;
fail:
- PyArray_Free(op2, (char *)wsave);
- Py_XDECREF(data);
- Py_XDECREF(ret);
- return NULL;
+ PyArray_Free(op2, (char *)wsave);
+ Py_XDECREF(data);
+ Py_XDECREF(ret);
+ return NULL;
}
@@ -219,16 +257,19 @@ fftpack_rffti(PyObject *self, PyObject *args)
PyArrayObject *op;
int dim, n;
- if (!PyArg_ParseTuple(args, "i", &n)) return NULL;
-
- dim = 2*n+15; /*Magic size needed by rffti*/
+ if (!PyArg_ParseTuple(args, "i", &n)) {
+ return NULL;
+ }
+ /*Magic size needed by rffti*/
+ dim = 2*n + 15;
/*Create a 1 dimensional array of dimensions of type double*/
op = (PyArrayObject *)PyArray_FromDims(1, &dim, PyArray_DOUBLE);
- if (op == NULL) return NULL;
-
- NPY_SIGINT_ON
+ if (op == NULL) {
+ return NULL;
+ }
+ NPY_SIGINT_ON;
rffti(n, (double *)((PyArrayObject*)op)->data);
- NPY_SIGINT_OFF
+ NPY_SIGINT_OFF;
return (PyObject *)op;
}
@@ -237,13 +278,13 @@ fftpack_rffti(PyObject *self, PyObject *args)
/* List of methods defined in the module */
static struct PyMethodDef fftpack_methods[] = {
- {"cfftf", fftpack_cfftf, 1, fftpack_cfftf__doc__},
- {"cfftb", fftpack_cfftb, 1, fftpack_cfftb__doc__},
- {"cffti", fftpack_cffti, 1, fftpack_cffti__doc__},
- {"rfftf", fftpack_rfftf, 1, fftpack_rfftf__doc__},
- {"rfftb", fftpack_rfftb, 1, fftpack_rfftb__doc__},
- {"rffti", fftpack_rffti, 1, fftpack_rffti__doc__},
- {NULL, NULL} /* sentinel */
+ {"cfftf", fftpack_cfftf, 1, fftpack_cfftf__doc__},
+ {"cfftb", fftpack_cfftb, 1, fftpack_cfftb__doc__},
+ {"cffti", fftpack_cffti, 1, fftpack_cffti__doc__},
+ {"rfftf", fftpack_rfftf, 1, fftpack_rfftf__doc__},
+ {"rfftb", fftpack_rfftb, 1, fftpack_rfftb__doc__},
+ {"rffti", fftpack_rffti, 1, fftpack_rffti__doc__},
+ {NULL, NULL} /* sentinel */
};
@@ -255,21 +296,21 @@ static char fftpack_module_documentation[] =
PyMODINIT_FUNC initfftpack_lite(void)
{
- PyObject *m, *d;
+ PyObject *m, *d;
- /* Create the module and add the functions */
- m = Py_InitModule4("fftpack_lite", fftpack_methods,
- fftpack_module_documentation,
- (PyObject*)NULL,PYTHON_API_VERSION);
+ /* Create the module and add the functions */
+ m = Py_InitModule4("fftpack_lite", fftpack_methods,
+ fftpack_module_documentation,
+ (PyObject*)NULL,PYTHON_API_VERSION);
- /* Import the array object */
- import_array();
+ /* Import the array object */
+ import_array();
- /* Add some symbolic constants to the module */
- d = PyModule_GetDict(m);
- ErrorObject = PyErr_NewException("fftpack.error", NULL, NULL);
- PyDict_SetItemString(d, "error", ErrorObject);
+ /* Add some symbolic constants to the module */
+ d = PyModule_GetDict(m);
+ ErrorObject = PyErr_NewException("fftpack.error", NULL, NULL);
+ PyDict_SetItemString(d, "error", ErrorObject);
- /* XXXX Add constants here */
+ /* XXXX Add constants here */
}