extensions.c
55 lines
| 1.5 KiB
| text/x-c
|
CLexer
Juan C. Valdez
|
r878 | #include <Python.h> | ||
| #include <numpy/arrayobject.h> | ||||
| #include <math.h> | ||||
| static PyObject *hildebrand_sekhon(PyObject *self, PyObject *args); | ||||
| static PyMethodDef extensionsMethods[] = { | ||||
| { "hildebrand_sekhon", (PyCFunction)hildebrand_sekhon, METH_VARARGS, "get noise with" }, | ||||
| { NULL, NULL, 0, NULL } | ||||
| }; | ||||
| PyMODINIT_FUNC initcSchain() { | ||||
| Py_InitModule("cSchain", extensionsMethods); | ||||
| import_array(); | ||||
| } | ||||
| static PyObject *hildebrand_sekhon(PyObject *self, PyObject *args) { | ||||
| /* Do your stuff here. */ | ||||
| double navg; | ||||
| PyObject *data_obj, *data_array; | ||||
| if (!PyArg_ParseTuple(args, "Od", &data_obj, &navg)) return NULL; | ||||
| data_array = PyArray_FROM_OTF(data_obj, NPY_FLOAT64, NPY_IN_ARRAY); | ||||
| if (data_array == NULL) { | ||||
| Py_XDECREF(data_array); | ||||
| Py_XDECREF(data_obj); | ||||
| return NULL; | ||||
| } | ||||
| double *sortdata = (double*)PyArray_DATA(data_array); | ||||
| int lenOfData = (int)PyArray_SIZE(data_array) ; | ||||
| double nums_min = lenOfData*0.2; | ||||
| if (nums_min <= 5) nums_min = 5; | ||||
| double sump = 0; | ||||
| double sumq = 0; | ||||
| int j = 0; | ||||
| int cont = 1; | ||||
| double rtest = 0; | ||||
| while ((cont == 1) && (j < lenOfData)) { | ||||
| sump = sump + sortdata[j]; | ||||
| sumq = sumq + pow(sortdata[j], 2); | ||||
| if (j > nums_min) { | ||||
| rtest = (double)j/(j-1) + 1/navg; | ||||
| if ((sumq*j) > (rtest*pow(sump, 2))) { | ||||
| j = j - 1; | ||||
| sump = sump - sortdata[j]; | ||||
| sumq = sumq - pow(sortdata[j],2); | ||||
| cont = 0; | ||||
| } | ||||
| } | ||||
| j = j + 1; | ||||
| } | ||||
| double lnoise = sump / j; | ||||
| return Py_BuildValue("d", lnoise); | ||||
| } | ||||