| @@ -0,0 +1,86 | |||||
|
|
1 | #include <Python.h> | |||
|
|
2 | #include <numpy/arrayobject.h> | |||
|
|
3 | #include <math.h> | |||
|
|
4 | ||||
|
|
5 | ||||
|
|
6 | static PyObject *HS_algorithm(PyObject *self, PyObject *args) { | |||
|
|
7 | double navg; | |||
|
|
8 | PyObject *data_obj, *data_array; | |||
|
|
9 | ||||
|
|
10 | if (!PyArg_ParseTuple(args, "Od", &data_obj, &navg)) { | |||
|
|
11 | return NULL; | |||
|
|
12 | } | |||
|
|
13 | ||||
|
|
14 | data_array = PyArray_FROM_OTF(data_obj, NPY_FLOAT64, NPY_IN_ARRAY); | |||
|
|
15 | ||||
|
|
16 | if (data_array == NULL) { | |||
|
|
17 | Py_XDECREF(data_array); | |||
|
|
18 | Py_XDECREF(data_obj); | |||
|
|
19 | return NULL; | |||
|
|
20 | } | |||
|
|
21 | double *sortdata = (double*)PyArray_DATA(data_array); | |||
|
|
22 | int lenOfData = (int)PyArray_SIZE(data_array) ; | |||
|
|
23 | double nums_min = lenOfData*0.75; | |||
|
|
24 | if (nums_min <= 5) nums_min = 5; | |||
|
|
25 | double sump = 0; | |||
|
|
26 | double sumq = 0; | |||
|
|
27 | long j = 0; | |||
|
|
28 | int cont = 1; | |||
|
|
29 | double rtest = 0; | |||
|
|
30 | while ((cont == 1) && (j < lenOfData)) { | |||
|
|
31 | sump = sump + sortdata[j]; | |||
|
|
32 | sumq = sumq + pow(sortdata[j], 2); | |||
|
|
33 | if (j > nums_min) { | |||
|
|
34 | rtest = (double)j/(j-1) + 1/navg; | |||
|
|
35 | //printf("%ld\n", j); | |||
|
|
36 | //printf("%f \n", rtest); | |||
|
|
37 | //printf("%f \n", sump); | |||
|
|
38 | if ((sumq*j) > (rtest*pow(sump, 2))) { | |||
|
|
39 | j = j - 1; | |||
|
|
40 | sump = sump - sortdata[j]; | |||
|
|
41 | sumq = sumq - pow(sortdata[j],2); | |||
|
|
42 | cont = 0; | |||
|
|
43 | } | |||
|
|
44 | } | |||
|
|
45 | //printf("%ld\n", j); | |||
|
|
46 | j = j + 1; | |||
|
|
47 | } | |||
|
|
48 | ||||
|
|
49 | double lnoise = sump / j; | |||
|
|
50 | ||||
|
|
51 | Py_DECREF(data_array); | |||
|
|
52 | ||||
|
|
53 | // return PyLong_FromLong(lnoise); | |||
|
|
54 | return PyFloat_FromDouble(j); | |||
|
|
55 | } | |||
|
|
56 | ||||
|
|
57 | ||||
|
|
58 | static PyMethodDef noiseMethods[] = { | |||
|
|
59 | { "HS_algorithm", HS_algorithm, METH_VARARGS, "Applies hildebrand_sekhon algorithm" }, | |||
|
|
60 | { NULL, NULL, 0, NULL } | |||
|
|
61 | }; | |||
|
|
62 | ||||
|
|
63 | #if PY_MAJOR_VERSION >= 3 | |||
|
|
64 | ||||
|
|
65 | static struct PyModuleDef noisemodule = { | |||
|
|
66 | PyModuleDef_HEAD_INIT, | |||
|
|
67 | "_HS_algorithm", | |||
|
|
68 | "Applies hildebrand_sekhon algorithm", | |||
|
|
69 | -1, | |||
|
|
70 | noiseMethods | |||
|
|
71 | }; | |||
|
|
72 | ||||
|
|
73 | #endif | |||
|
|
74 | ||||
|
|
75 | #if PY_MAJOR_VERSION >= 3 | |||
|
|
76 | PyMODINIT_FUNC PyInit__HS_algorithm(void) { | |||
|
|
77 | Py_Initialize(); | |||
|
|
78 | import_array(); | |||
|
|
79 | return PyModule_Create(&noisemodule); | |||
|
|
80 | } | |||
|
|
81 | #else | |||
|
|
82 | PyMODINIT_FUNC init_HS_algorithm() { | |||
|
|
83 | Py_InitModule("_HS_algorithm", noiseMethods); | |||
|
|
84 | import_array(); | |||
|
|
85 | } | |||
|
|
86 | #endif | |||
General Comments 0
You need to be logged in to leave comments.
Login now