Clone xlogy from scipy

Author: lixfz

ylearn/sklearn_ex/cloned/scipy/README.txt

@@ -0,0 +1,2 @@
+This directory is a fork from scipy 0.13.3
+See: https://github.com/scipy/scipy

ylearn/sklearn_ex/cloned/scipy/__init__.py

@@ -0,0 +1,9 @@
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#include <numpy/npy_math.h>
+
+#ifdef __cplusplus
+}
+#endif

ylearn/sklearn_ex/cloned/scipy/_complexstuff.h

@@ -0,0 +1,66 @@
+# -*-cython-*-
+#
+# Common functions required when doing complex arithmetic with Cython.
+#
+
+cimport numpy as np
+cimport libc.math
+
+cdef extern from "_complexstuff.h":
+    double npy_cabs(np.npy_cdouble z) nogil
+    np.npy_cdouble npy_clog(np.npy_cdouble z) nogil
+    np.npy_cdouble npy_cexp(np.npy_cdouble z) nogil
+    double npy_log1p(double x) nogil
+    int npy_isnan(double x) nogil
+    int npy_isinf(double x) nogil
+    int npy_isfinite(double x) nogil
+    double inf "NPY_INFINITY"
+    double pi "NPY_PI"
+    double nan "NPY_NAN"
+
+ctypedef double complex double_complex
+
+ctypedef fused number_t:
+    double
+    double_complex
+
+cdef inline bint zisnan(number_t x) nogil:
+    if number_t is double_complex:
+        return npy_isnan(x.real) or npy_isnan(x.imag)
+    else:
+        return npy_isnan(x)
+
+cdef inline bint zisfinite(number_t x) nogil:
+    if number_t is double_complex:
+        return npy_isfinite(x.real) and npy_isfinite(x.imag)
+    else:
+        return npy_isfinite(x)
+
+cdef inline bint zisinf(number_t x) nogil:
+    if number_t is double_complex:
+        return not zisnan(x) and not zisfinite(x)
+    else:
+        return npy_isinf(x)
+
+cdef inline double zabs(number_t x) nogil:
+    if number_t is double_complex:
+        return npy_cabs((<np.npy_cdouble*>&x)[0])
+    else:
+        return libc.math.fabs(x)
+
+cdef inline number_t zlog(number_t x) nogil:
+    cdef np.npy_cdouble r
+    if number_t is double_complex:
+        r = npy_clog((<np.npy_cdouble*>&x)[0])
+        return (<double_complex*>&r)[0]
+    else:
+        return libc.math.log(x)
+
+cdef inline number_t zexp(number_t x) nogil:
+    cdef np.npy_cdouble r
+    if number_t is double_complex:
+        r = npy_cexp((<np.npy_cdouble*>&x)[0])
+        return (<double_complex*>&r)[0]
+    else:
+        return libc.math.exp(x)
+

ylearn/sklearn_ex/cloned/scipy/_complexstuff.pxd

@@ -0,0 +1,10 @@
+# -*-cython-*-
+
+from ._complexstuff cimport zlog, zisnan, number_t
+
+cdef inline number_t xlogy(number_t x, number_t y) nogil:
+    if x == 0 and not zisnan(y):
+        return 0
+    else:
+        return x * zlog(y)
+

ylearn/sklearn_ex/cloned/scipy/_xlogy.pxd

@@ -25,7 +25,8 @@ cimport numpy as np
 np.import_array()
 
 from numpy.math cimport INFINITY
-from scipy.special.cython_special cimport xlogy
+## from scipy.special.cython_special cimport xlogy
+from ..scipy._xlogy cimport xlogy
 
 from ._utils cimport log
 from ._utils cimport WeightedMedianCalculator