Generated by Cython 0.29.24
Yellow lines hint at Python interaction.
Click on a line that starts with a "+" to see the C code that Cython generated for it.
Raw output: _inner.cpp
+001: # Fast inner loop for EDBSCAN
__pyx_t_1 = __Pyx_PyDict_NewPresized(0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 1, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (PyDict_SetItem(__pyx_d, __pyx_n_s_test, __pyx_t_1) < 0) __PYX_ERR(0, 1, __pyx_L1_error) __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
002: # Author: Ruben Broekx
003: #
004: # Code forked from Scikit-Learn's DBSCAN
005: # Author: Lars Buitinck
006: # License: 3-clause BSD
007: #
008: # cython: boundscheck=False, wraparound=False, infer_types=True
009:
010: cimport cython
011: from libcpp cimport bool
012: from libcpp.vector cimport vector
013: from libc.stdlib cimport malloc, free
014: cimport numpy as np
+015: import numpy as np
__pyx_t_1 = __Pyx_Import(__pyx_n_s_numpy, 0, -1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 15, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (PyDict_SetItem(__pyx_d, __pyx_n_s_np, __pyx_t_1) < 0) __PYX_ERR(0, 15, __pyx_L1_error) __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
016:
+017: np.import_array()
__pyx_t_2 = __pyx_f_5numpy_import_array(); if (unlikely(__pyx_t_2 == ((int)-1))) __PYX_ERR(0, 17, __pyx_L1_error)
018:
019:
+020: cdef int pop(
static int __pyx_f_7edbscan_6_inner_pop(PyObject *__pyx_v_stack, PyArrayObject *__pyx_v_neighborhoods) {
int __pyx_v_best_size;
int __pyx_v_best_idx;
int __pyx_v_idx;
PyArrayObject *__pyx_v_neighborhood = 0;
__Pyx_LocalBuf_ND __pyx_pybuffernd_neighborhood;
__Pyx_Buffer __pyx_pybuffer_neighborhood;
__Pyx_LocalBuf_ND __pyx_pybuffernd_neighborhoods;
__Pyx_Buffer __pyx_pybuffer_neighborhoods;
int __pyx_r;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("pop", 0);
__pyx_pybuffer_neighborhood.pybuffer.buf = NULL;
__pyx_pybuffer_neighborhood.refcount = 0;
__pyx_pybuffernd_neighborhood.data = NULL;
__pyx_pybuffernd_neighborhood.rcbuffer = &__pyx_pybuffer_neighborhood;
__pyx_pybuffer_neighborhoods.pybuffer.buf = NULL;
__pyx_pybuffer_neighborhoods.refcount = 0;
__pyx_pybuffernd_neighborhoods.data = NULL;
__pyx_pybuffernd_neighborhoods.rcbuffer = &__pyx_pybuffer_neighborhoods;
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer, (PyObject*)__pyx_v_neighborhoods, &__Pyx_TypeInfo_object, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 20, __pyx_L1_error)
}
__pyx_pybuffernd_neighborhoods.diminfo[0].strides = __pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_neighborhoods.diminfo[0].shape = __pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer.shape[0];
/* … */
/* function exit code */
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_XDECREF(__pyx_t_3);
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
__Pyx_PyThreadState_declare
__Pyx_PyThreadState_assign
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_WriteUnraisable("edbscan._inner.pop", __pyx_clineno, __pyx_lineno, __pyx_filename, 1, 0);
__pyx_r = 0;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer);
__pyx_L2:;
__Pyx_XDECREF((PyObject *)__pyx_v_neighborhood);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
021: list stack,
022: np.ndarray[object, ndim=1] neighborhoods,
023: ):
024: """Pop index from stack with the highest density, calculated as number of neighbors."""
+025: cdef int best_size = -1
__pyx_v_best_size = -1;
+026: cdef int best_idx = -1
__pyx_v_best_idx = -1;
027: cdef int idx
028: cdef np.ndarray[np.npy_intp, ndim=1, mode='c'] neighborhood
029:
+030: for idx in stack:
if (unlikely(__pyx_v_stack == Py_None)) {
PyErr_SetString(PyExc_TypeError, "'NoneType' object is not iterable");
__PYX_ERR(0, 30, __pyx_L1_error)
}
__pyx_t_1 = __pyx_v_stack; __Pyx_INCREF(__pyx_t_1); __pyx_t_2 = 0;
for (;;) {
if (__pyx_t_2 >= PyList_GET_SIZE(__pyx_t_1)) break;
#if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
__pyx_t_3 = PyList_GET_ITEM(__pyx_t_1, __pyx_t_2); __Pyx_INCREF(__pyx_t_3); __pyx_t_2++; if (unlikely(0 < 0)) __PYX_ERR(0, 30, __pyx_L1_error)
#else
__pyx_t_3 = PySequence_ITEM(__pyx_t_1, __pyx_t_2); __pyx_t_2++; if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 30, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_3);
#endif
__pyx_t_4 = __Pyx_PyInt_As_int(__pyx_t_3); if (unlikely((__pyx_t_4 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 30, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_v_idx = __pyx_t_4;
/* … */
}
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
+031: neighborhood = neighborhoods[idx]
__pyx_t_5 = __pyx_v_idx;
__pyx_t_3 = (PyObject *) *__Pyx_BufPtrStrided1d(PyObject **, __pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer.buf, __pyx_t_5, __pyx_pybuffernd_neighborhoods.diminfo[0].strides);
__Pyx_INCREF((PyObject*)__pyx_t_3);
if (!(likely(((__pyx_t_3) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_3, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 31, __pyx_L1_error)
__pyx_t_6 = ((PyArrayObject *)__pyx_t_3);
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer);
__pyx_t_4 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer, (PyObject*)__pyx_t_6, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack);
if (unlikely(__pyx_t_4 < 0)) {
PyErr_Fetch(&__pyx_t_7, &__pyx_t_8, &__pyx_t_9);
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer, (PyObject*)__pyx_v_neighborhood, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {
Py_XDECREF(__pyx_t_7); Py_XDECREF(__pyx_t_8); Py_XDECREF(__pyx_t_9);
__Pyx_RaiseBufferFallbackError();
} else {
PyErr_Restore(__pyx_t_7, __pyx_t_8, __pyx_t_9);
}
__pyx_t_7 = __pyx_t_8 = __pyx_t_9 = 0;
}
__pyx_pybuffernd_neighborhood.diminfo[0].strides = __pyx_pybuffernd_neighborhood.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_neighborhood.diminfo[0].shape = __pyx_pybuffernd_neighborhood.rcbuffer->pybuffer.shape[0];
if (unlikely(__pyx_t_4 < 0)) __PYX_ERR(0, 31, __pyx_L1_error)
}
__pyx_t_6 = 0;
__Pyx_XDECREF_SET(__pyx_v_neighborhood, ((PyArrayObject *)__pyx_t_3));
__pyx_t_3 = 0;
+032: if neighborhood.shape[0] > best_size:
__pyx_t_10 = (((__pyx_v_neighborhood->dimensions[0]) > __pyx_v_best_size) != 0);
if (__pyx_t_10) {
/* … */
}
+033: best_size = neighborhood.shape[0]
__pyx_v_best_size = (__pyx_v_neighborhood->dimensions[0]);
+034: best_idx = idx
__pyx_v_best_idx = __pyx_v_idx;
035:
036: # Pop the best index from the stack.
+037: stack.remove(best_idx)
__pyx_t_1 = __Pyx_PyInt_From_int(__pyx_v_best_idx); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 37, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __pyx_t_3 = __Pyx_CallUnboundCMethod1(&__pyx_umethod_PyList_Type_remove, __pyx_v_stack, __pyx_t_1); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 37, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_3); __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0; __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
+038: return best_idx
__pyx_r = __pyx_v_best_idx; goto __pyx_L0;
039:
040:
+041: cdef int get_unlabeled(
static int __pyx_f_7edbscan_6_inner_get_unlabeled(PyArrayObject *__pyx_v_is_core, PyArrayObject *__pyx_v_neighborhoods, PyArrayObject *__pyx_v_labels) {
int __pyx_v_best_size;
int __pyx_v_best_idx;
PyArrayObject *__pyx_v_neighborhood = 0;
npy_intp __pyx_v_idx;
__Pyx_LocalBuf_ND __pyx_pybuffernd_is_core;
__Pyx_Buffer __pyx_pybuffer_is_core;
__Pyx_LocalBuf_ND __pyx_pybuffernd_labels;
__Pyx_Buffer __pyx_pybuffer_labels;
__Pyx_LocalBuf_ND __pyx_pybuffernd_neighborhood;
__Pyx_Buffer __pyx_pybuffer_neighborhood;
__Pyx_LocalBuf_ND __pyx_pybuffernd_neighborhoods;
__Pyx_Buffer __pyx_pybuffer_neighborhoods;
int __pyx_r;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("get_unlabeled", 0);
__pyx_pybuffer_neighborhood.pybuffer.buf = NULL;
__pyx_pybuffer_neighborhood.refcount = 0;
__pyx_pybuffernd_neighborhood.data = NULL;
__pyx_pybuffernd_neighborhood.rcbuffer = &__pyx_pybuffer_neighborhood;
__pyx_pybuffer_is_core.pybuffer.buf = NULL;
__pyx_pybuffer_is_core.refcount = 0;
__pyx_pybuffernd_is_core.data = NULL;
__pyx_pybuffernd_is_core.rcbuffer = &__pyx_pybuffer_is_core;
__pyx_pybuffer_neighborhoods.pybuffer.buf = NULL;
__pyx_pybuffer_neighborhoods.refcount = 0;
__pyx_pybuffernd_neighborhoods.data = NULL;
__pyx_pybuffernd_neighborhoods.rcbuffer = &__pyx_pybuffer_neighborhoods;
__pyx_pybuffer_labels.pybuffer.buf = NULL;
__pyx_pybuffer_labels.refcount = 0;
__pyx_pybuffernd_labels.data = NULL;
__pyx_pybuffernd_labels.rcbuffer = &__pyx_pybuffer_labels;
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_is_core.rcbuffer->pybuffer, (PyObject*)__pyx_v_is_core, &__Pyx_TypeInfo_nn___pyx_t_5numpy_uint8_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 41, __pyx_L1_error)
}
__pyx_pybuffernd_is_core.diminfo[0].strides = __pyx_pybuffernd_is_core.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_is_core.diminfo[0].shape = __pyx_pybuffernd_is_core.rcbuffer->pybuffer.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer, (PyObject*)__pyx_v_neighborhoods, &__Pyx_TypeInfo_object, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 41, __pyx_L1_error)
}
__pyx_pybuffernd_neighborhoods.diminfo[0].strides = __pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_neighborhoods.diminfo[0].shape = __pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_labels.rcbuffer->pybuffer, (PyObject*)__pyx_v_labels, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 41, __pyx_L1_error)
}
__pyx_pybuffernd_labels.diminfo[0].strides = __pyx_pybuffernd_labels.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_labels.diminfo[0].shape = __pyx_pybuffernd_labels.rcbuffer->pybuffer.shape[0];
/* … */
/* function exit code */
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_6);
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
__Pyx_PyThreadState_declare
__Pyx_PyThreadState_assign
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_is_core.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_WriteUnraisable("edbscan._inner.get_unlabeled", __pyx_clineno, __pyx_lineno, __pyx_filename, 1, 0);
__pyx_r = 0;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_is_core.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer);
__pyx_L2:;
__Pyx_XDECREF((PyObject *)__pyx_v_neighborhood);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
042: np.ndarray[np.uint8_t, ndim=1, mode='c'] is_core,
043: np.ndarray[object, ndim=1] neighborhoods,
044: np.ndarray[np.npy_intp, ndim=1, mode='c'] labels,
045: ):
046: """Get the best unlabeled core index."""
+047: cdef int best_size = -1
__pyx_v_best_size = -1;
+048: cdef int best_idx = -1
__pyx_v_best_idx = -1;
049: cdef np.ndarray[np.npy_intp, ndim=1, mode='c'] neighborhood
050:
+051: for idx in xrange(neighborhoods.shape[0]):
__pyx_t_1 = (__pyx_v_neighborhoods->dimensions[0]);
__pyx_t_2 = __pyx_t_1;
for (__pyx_t_3 = 0; __pyx_t_3 < __pyx_t_2; __pyx_t_3+=1) {
__pyx_v_idx = __pyx_t_3;
+052: if not is_core[idx]:
__pyx_t_4 = __pyx_v_idx;
__pyx_t_5 = ((!((*__Pyx_BufPtrCContig1d(__pyx_t_5numpy_uint8_t *, __pyx_pybuffernd_is_core.rcbuffer->pybuffer.buf, __pyx_t_4, __pyx_pybuffernd_is_core.diminfo[0].strides)) != 0)) != 0);
if (__pyx_t_5) {
/* … */
}
+053: continue
goto __pyx_L3_continue;
+054: if labels[idx] != -2:
__pyx_t_4 = __pyx_v_idx;
__pyx_t_5 = (((*__Pyx_BufPtrCContig1d(npy_intp *, __pyx_pybuffernd_labels.rcbuffer->pybuffer.buf, __pyx_t_4, __pyx_pybuffernd_labels.diminfo[0].strides)) != -2L) != 0);
if (__pyx_t_5) {
/* … */
}
+055: continue
goto __pyx_L3_continue;
056:
+057: neighborhood = neighborhoods[idx]
__pyx_t_4 = __pyx_v_idx;
__pyx_t_6 = (PyObject *) *__Pyx_BufPtrStrided1d(PyObject **, __pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer.buf, __pyx_t_4, __pyx_pybuffernd_neighborhoods.diminfo[0].strides);
__Pyx_INCREF((PyObject*)__pyx_t_6);
if (!(likely(((__pyx_t_6) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_6, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 57, __pyx_L1_error)
__pyx_t_7 = ((PyArrayObject *)__pyx_t_6);
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer);
__pyx_t_8 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer, (PyObject*)__pyx_t_7, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack);
if (unlikely(__pyx_t_8 < 0)) {
PyErr_Fetch(&__pyx_t_9, &__pyx_t_10, &__pyx_t_11);
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer, (PyObject*)__pyx_v_neighborhood, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {
Py_XDECREF(__pyx_t_9); Py_XDECREF(__pyx_t_10); Py_XDECREF(__pyx_t_11);
__Pyx_RaiseBufferFallbackError();
} else {
PyErr_Restore(__pyx_t_9, __pyx_t_10, __pyx_t_11);
}
__pyx_t_9 = __pyx_t_10 = __pyx_t_11 = 0;
}
__pyx_pybuffernd_neighborhood.diminfo[0].strides = __pyx_pybuffernd_neighborhood.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_neighborhood.diminfo[0].shape = __pyx_pybuffernd_neighborhood.rcbuffer->pybuffer.shape[0];
if (unlikely(__pyx_t_8 < 0)) __PYX_ERR(0, 57, __pyx_L1_error)
}
__pyx_t_7 = 0;
__Pyx_XDECREF_SET(__pyx_v_neighborhood, ((PyArrayObject *)__pyx_t_6));
__pyx_t_6 = 0;
+058: if neighborhood.shape[0] > best_size:
__pyx_t_5 = (((__pyx_v_neighborhood->dimensions[0]) > __pyx_v_best_size) != 0);
if (__pyx_t_5) {
/* … */
}
__pyx_L3_continue:;
}
+059: best_size = neighborhood.shape[0]
__pyx_v_best_size = (__pyx_v_neighborhood->dimensions[0]);
+060: best_idx = idx
__pyx_v_best_idx = __pyx_v_idx;
061:
062: # Pop the best index from the stack.
+063: return best_idx
__pyx_r = __pyx_v_best_idx; goto __pyx_L0;
064:
065:
+066: cdef bool value_in_array(
static bool __pyx_f_7edbscan_6_inner_value_in_array(int __pyx_v_val, std::vector<npy_intp> __pyx_v_arr) {
int __pyx_v_i;
bool __pyx_r;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("value_in_array", 0);
/* … */
/* function exit code */
__pyx_L0:;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
067: int val,
068: vector[np.npy_intp] arr,
069: ):
070: """Check if the value exists in the array."""
071: cdef int i
072:
+073: for i in arr:
__pyx_t_1 = __pyx_v_arr.begin();
for (;;) {
if (!(__pyx_t_1 != __pyx_v_arr.end())) break;
__pyx_t_2 = *__pyx_t_1;
++__pyx_t_1;
__pyx_v_i = __pyx_t_2;
/* … */
}
+074: if i == val:
__pyx_t_3 = ((__pyx_v_i == __pyx_v_val) != 0);
if (__pyx_t_3) {
/* … */
}
+075: return True
__pyx_r = 1;
goto __pyx_L0;
+076: return False
__pyx_r = 0; goto __pyx_L0;
077:
078:
+079: cdef int get_max_0(
static int __pyx_f_7edbscan_6_inner_get_max_0(PyArrayObject *__pyx_v_arr) {
int __pyx_v_max_val;
int __pyx_v_i;
int __pyx_v_idx;
__Pyx_LocalBuf_ND __pyx_pybuffernd_arr;
__Pyx_Buffer __pyx_pybuffer_arr;
int __pyx_r;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("get_max_0", 0);
__pyx_pybuffer_arr.pybuffer.buf = NULL;
__pyx_pybuffer_arr.refcount = 0;
__pyx_pybuffernd_arr.data = NULL;
__pyx_pybuffernd_arr.rcbuffer = &__pyx_pybuffer_arr;
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_arr.rcbuffer->pybuffer, (PyObject*)__pyx_v_arr, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 79, __pyx_L1_error)
}
__pyx_pybuffernd_arr.diminfo[0].strides = __pyx_pybuffernd_arr.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_arr.diminfo[0].shape = __pyx_pybuffernd_arr.rcbuffer->pybuffer.shape[0];
/* … */
/* function exit code */
__pyx_L1_error:;
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
__Pyx_PyThreadState_declare
__Pyx_PyThreadState_assign
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_arr.rcbuffer->pybuffer);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_WriteUnraisable("edbscan._inner.get_max_0", __pyx_clineno, __pyx_lineno, __pyx_filename, 1, 0);
__pyx_r = 0;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_arr.rcbuffer->pybuffer);
__pyx_L2:;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
080: np.ndarray[np.npy_intp, ndim=1, mode='c'] arr,
081: ):
082: """Get the maximum value in the array, that is at least zero."""
+083: cdef int max_val = 0
__pyx_v_max_val = 0;
084: cdef int i, idx
085:
+086: for i in xrange(arr.shape[0]):
__pyx_t_1 = (__pyx_v_arr->dimensions[0]);
__pyx_t_2 = __pyx_t_1;
for (__pyx_t_3 = 0; __pyx_t_3 < __pyx_t_2; __pyx_t_3+=1) {
__pyx_v_i = __pyx_t_3;
+087: idx = arr[i]
__pyx_t_4 = __pyx_v_i;
__pyx_v_idx = (*__Pyx_BufPtrCContig1d(npy_intp *, __pyx_pybuffernd_arr.rcbuffer->pybuffer.buf, __pyx_t_4, __pyx_pybuffernd_arr.diminfo[0].strides));
+088: if (idx + 1) > max_val:
__pyx_t_5 = (((__pyx_v_idx + 1) > __pyx_v_max_val) != 0);
if (__pyx_t_5) {
/* … */
}
}
+089: max_val = idx + 1
__pyx_v_max_val = (__pyx_v_idx + 1);
+090: return max_val
__pyx_r = __pyx_v_max_val; goto __pyx_L0;
091:
+092: cdef bool conflict(
static bool __pyx_f_7edbscan_6_inner_conflict(PyArrayObject *__pyx_v_neighborhood, PyArrayObject *__pyx_v_labels) {
int __pyx_v_label;
std::vector<npy_intp> __pyx_v_seen;
__Pyx_LocalBuf_ND __pyx_pybuffernd_labels;
__Pyx_Buffer __pyx_pybuffer_labels;
__Pyx_LocalBuf_ND __pyx_pybuffernd_neighborhood;
__Pyx_Buffer __pyx_pybuffer_neighborhood;
bool __pyx_r;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("conflict", 0);
__pyx_pybuffer_neighborhood.pybuffer.buf = NULL;
__pyx_pybuffer_neighborhood.refcount = 0;
__pyx_pybuffernd_neighborhood.data = NULL;
__pyx_pybuffernd_neighborhood.rcbuffer = &__pyx_pybuffer_neighborhood;
__pyx_pybuffer_labels.pybuffer.buf = NULL;
__pyx_pybuffer_labels.refcount = 0;
__pyx_pybuffernd_labels.data = NULL;
__pyx_pybuffernd_labels.rcbuffer = &__pyx_pybuffer_labels;
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer, (PyObject*)__pyx_v_neighborhood, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 92, __pyx_L1_error)
}
__pyx_pybuffernd_neighborhood.diminfo[0].strides = __pyx_pybuffernd_neighborhood.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_neighborhood.diminfo[0].shape = __pyx_pybuffernd_neighborhood.rcbuffer->pybuffer.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_labels.rcbuffer->pybuffer, (PyObject*)__pyx_v_labels, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 92, __pyx_L1_error)
}
__pyx_pybuffernd_labels.diminfo[0].strides = __pyx_pybuffernd_labels.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_labels.diminfo[0].shape = __pyx_pybuffernd_labels.rcbuffer->pybuffer.shape[0];
/* … */
/* function exit code */
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_XDECREF(__pyx_t_2);
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
__Pyx_PyThreadState_declare
__Pyx_PyThreadState_assign
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_WriteUnraisable("edbscan._inner.conflict", __pyx_clineno, __pyx_lineno, __pyx_filename, 1, 0);
__pyx_r = 0;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer);
__pyx_L2:;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
093: np.ndarray[np.npy_intp, ndim=1] neighborhood,
094: np.ndarray[np.npy_intp, ndim=1, mode='c'] labels,
095: ):
096: """Check if there's a conflict in the given neighborhood."""
097: cdef int label
098: cdef vector[np.npy_intp] seen
099:
+100: for label in labels[neighborhood]:
__pyx_t_1 = __Pyx_PyObject_GetItem(((PyObject *)__pyx_v_labels), ((PyObject *)__pyx_v_neighborhood)); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 100, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (likely(PyList_CheckExact(__pyx_t_1)) || PyTuple_CheckExact(__pyx_t_1)) { __pyx_t_2 = __pyx_t_1; __Pyx_INCREF(__pyx_t_2); __pyx_t_3 = 0; __pyx_t_4 = NULL; } else { __pyx_t_3 = -1; __pyx_t_2 = PyObject_GetIter(__pyx_t_1); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 100, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_2); __pyx_t_4 = Py_TYPE(__pyx_t_2)->tp_iternext; if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 100, __pyx_L1_error) } __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0; for (;;) { if (likely(!__pyx_t_4)) { if (likely(PyList_CheckExact(__pyx_t_2))) { if (__pyx_t_3 >= PyList_GET_SIZE(__pyx_t_2)) break; #if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS __pyx_t_1 = PyList_GET_ITEM(__pyx_t_2, __pyx_t_3); __Pyx_INCREF(__pyx_t_1); __pyx_t_3++; if (unlikely(0 < 0)) __PYX_ERR(0, 100, __pyx_L1_error) #else __pyx_t_1 = PySequence_ITEM(__pyx_t_2, __pyx_t_3); __pyx_t_3++; if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 100, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); #endif } else { if (__pyx_t_3 >= PyTuple_GET_SIZE(__pyx_t_2)) break; #if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS __pyx_t_1 = PyTuple_GET_ITEM(__pyx_t_2, __pyx_t_3); __Pyx_INCREF(__pyx_t_1); __pyx_t_3++; if (unlikely(0 < 0)) __PYX_ERR(0, 100, __pyx_L1_error) #else __pyx_t_1 = PySequence_ITEM(__pyx_t_2, __pyx_t_3); __pyx_t_3++; if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 100, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); #endif } } else { __pyx_t_1 = __pyx_t_4(__pyx_t_2); if (unlikely(!__pyx_t_1)) { PyObject* exc_type = PyErr_Occurred(); if (exc_type) { if (likely(__Pyx_PyErr_GivenExceptionMatches(exc_type, PyExc_StopIteration))) PyErr_Clear(); else __PYX_ERR(0, 100, __pyx_L1_error) } break; } __Pyx_GOTREF(__pyx_t_1); } __pyx_t_5 = __Pyx_PyInt_As_int(__pyx_t_1); if (unlikely((__pyx_t_5 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 100, __pyx_L1_error) __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0; __pyx_v_label = __pyx_t_5; /* … */ __pyx_L3_continue:; } __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
+101: if label == -2:
__pyx_t_6 = ((__pyx_v_label == -2L) != 0);
if (__pyx_t_6) {
/* … */
}
+102: continue
goto __pyx_L3_continue;
+103: if value_in_array(label, seen):
__pyx_t_6 = (__pyx_f_7edbscan_6_inner_value_in_array(__pyx_v_label, __pyx_v_seen) != 0);
if (__pyx_t_6) {
/* … */
}
+104: continue
goto __pyx_L3_continue;
+105: if label == -1:
__pyx_t_6 = ((__pyx_v_label == -1L) != 0);
if (__pyx_t_6) {
/* … */
}
+106: return True
__pyx_r = 1;
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
goto __pyx_L0;
+107: seen.push_back(label)
try {
__pyx_v_seen.push_back(__pyx_v_label);
} catch(...) {
__Pyx_CppExn2PyErr();
__PYX_ERR(0, 107, __pyx_L1_error)
}
+108: return seen.size() > 1
__pyx_r = (__pyx_v_seen.size() > 1); goto __pyx_L0;
109:
110:
+111: cpdef void inner(
static PyObject *__pyx_pw_7edbscan_6_inner_1inner(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static void __pyx_f_7edbscan_6_inner_inner(PyArrayObject *__pyx_v_is_core, PyArrayObject *__pyx_v_neighborhoods, PyArrayObject *__pyx_v_labels, CYTHON_UNUSED int __pyx_skip_dispatch) {
int __pyx_v_i;
int __pyx_v_idx;
int __pyx_v_n_idx;
PyObject *__pyx_v_stack = 0;
PyArrayObject *__pyx_v_neighborhood = 0;
__Pyx_LocalBuf_ND __pyx_pybuffernd_is_core;
__Pyx_Buffer __pyx_pybuffer_is_core;
__Pyx_LocalBuf_ND __pyx_pybuffernd_labels;
__Pyx_Buffer __pyx_pybuffer_labels;
__Pyx_LocalBuf_ND __pyx_pybuffernd_neighborhood;
__Pyx_Buffer __pyx_pybuffer_neighborhood;
__Pyx_LocalBuf_ND __pyx_pybuffernd_neighborhoods;
__Pyx_Buffer __pyx_pybuffer_neighborhoods;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("inner", 0);
__pyx_pybuffer_neighborhood.pybuffer.buf = NULL;
__pyx_pybuffer_neighborhood.refcount = 0;
__pyx_pybuffernd_neighborhood.data = NULL;
__pyx_pybuffernd_neighborhood.rcbuffer = &__pyx_pybuffer_neighborhood;
__pyx_pybuffer_is_core.pybuffer.buf = NULL;
__pyx_pybuffer_is_core.refcount = 0;
__pyx_pybuffernd_is_core.data = NULL;
__pyx_pybuffernd_is_core.rcbuffer = &__pyx_pybuffer_is_core;
__pyx_pybuffer_neighborhoods.pybuffer.buf = NULL;
__pyx_pybuffer_neighborhoods.refcount = 0;
__pyx_pybuffernd_neighborhoods.data = NULL;
__pyx_pybuffernd_neighborhoods.rcbuffer = &__pyx_pybuffer_neighborhoods;
__pyx_pybuffer_labels.pybuffer.buf = NULL;
__pyx_pybuffer_labels.refcount = 0;
__pyx_pybuffernd_labels.data = NULL;
__pyx_pybuffernd_labels.rcbuffer = &__pyx_pybuffer_labels;
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_is_core.rcbuffer->pybuffer, (PyObject*)__pyx_v_is_core, &__Pyx_TypeInfo_nn___pyx_t_5numpy_uint8_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 111, __pyx_L1_error)
}
__pyx_pybuffernd_is_core.diminfo[0].strides = __pyx_pybuffernd_is_core.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_is_core.diminfo[0].shape = __pyx_pybuffernd_is_core.rcbuffer->pybuffer.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer, (PyObject*)__pyx_v_neighborhoods, &__Pyx_TypeInfo_object, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 111, __pyx_L1_error)
}
__pyx_pybuffernd_neighborhoods.diminfo[0].strides = __pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_neighborhoods.diminfo[0].shape = __pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_labels.rcbuffer->pybuffer, (PyObject*)__pyx_v_labels, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 111, __pyx_L1_error)
}
__pyx_pybuffernd_labels.diminfo[0].strides = __pyx_pybuffernd_labels.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_labels.diminfo[0].shape = __pyx_pybuffernd_labels.rcbuffer->pybuffer.shape[0];
/* … */
/* function exit code */
goto __pyx_L0;
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
__Pyx_PyThreadState_declare
__Pyx_PyThreadState_assign
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_is_core.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_WriteUnraisable("edbscan._inner.inner", __pyx_clineno, __pyx_lineno, __pyx_filename, 1, 0);
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_is_core.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer);
__pyx_L2:;
__Pyx_XDECREF(__pyx_v_stack);
__Pyx_XDECREF((PyObject *)__pyx_v_neighborhood);
__Pyx_RefNannyFinishContext();
}
/* Python wrapper */
static PyObject *__pyx_pw_7edbscan_6_inner_1inner(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static char __pyx_doc_7edbscan_6_inner_inner[] = "Inner-loop of the EDBSCAN algorithm.";
static PyObject *__pyx_pw_7edbscan_6_inner_1inner(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
PyArrayObject *__pyx_v_is_core = 0;
PyArrayObject *__pyx_v_neighborhoods = 0;
PyArrayObject *__pyx_v_labels = 0;
PyObject *__pyx_r = 0;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("inner (wrapper)", 0);
{
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_is_core,&__pyx_n_s_neighborhoods,&__pyx_n_s_labels,0};
PyObject* values[3] = {0,0,0};
if (unlikely(__pyx_kwds)) {
Py_ssize_t kw_args;
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
switch (pos_args) {
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
CYTHON_FALLTHROUGH;
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
CYTHON_FALLTHROUGH;
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
CYTHON_FALLTHROUGH;
case 0: break;
default: goto __pyx_L5_argtuple_error;
}
kw_args = PyDict_Size(__pyx_kwds);
switch (pos_args) {
case 0:
if (likely((values[0] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_is_core)) != 0)) kw_args--;
else goto __pyx_L5_argtuple_error;
CYTHON_FALLTHROUGH;
case 1:
if (likely((values[1] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_neighborhoods)) != 0)) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("inner", 1, 3, 3, 1); __PYX_ERR(0, 111, __pyx_L3_error)
}
CYTHON_FALLTHROUGH;
case 2:
if (likely((values[2] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_labels)) != 0)) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("inner", 1, 3, 3, 2); __PYX_ERR(0, 111, __pyx_L3_error)
}
}
if (unlikely(kw_args > 0)) {
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "inner") < 0)) __PYX_ERR(0, 111, __pyx_L3_error)
}
} else if (PyTuple_GET_SIZE(__pyx_args) != 3) {
goto __pyx_L5_argtuple_error;
} else {
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
}
__pyx_v_is_core = ((PyArrayObject *)values[0]);
__pyx_v_neighborhoods = ((PyArrayObject *)values[1]);
__pyx_v_labels = ((PyArrayObject *)values[2]);
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("inner", 1, 3, 3, PyTuple_GET_SIZE(__pyx_args)); __PYX_ERR(0, 111, __pyx_L3_error)
__pyx_L3_error:;
__Pyx_AddTraceback("edbscan._inner.inner", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_is_core), __pyx_ptype_5numpy_ndarray, 1, "is_core", 0))) __PYX_ERR(0, 112, __pyx_L1_error)
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_neighborhoods), __pyx_ptype_5numpy_ndarray, 1, "neighborhoods", 0))) __PYX_ERR(0, 113, __pyx_L1_error)
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_labels), __pyx_ptype_5numpy_ndarray, 1, "labels", 0))) __PYX_ERR(0, 114, __pyx_L1_error)
__pyx_r = __pyx_pf_7edbscan_6_inner_inner(__pyx_self, __pyx_v_is_core, __pyx_v_neighborhoods, __pyx_v_labels);
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
/* function exit code */
goto __pyx_L0;
__pyx_L1_error:;
__pyx_r = NULL;
__pyx_L0:;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
static PyObject *__pyx_pf_7edbscan_6_inner_inner(CYTHON_UNUSED PyObject *__pyx_self, PyArrayObject *__pyx_v_is_core, PyArrayObject *__pyx_v_neighborhoods, PyArrayObject *__pyx_v_labels) {
__Pyx_LocalBuf_ND __pyx_pybuffernd_is_core;
__Pyx_Buffer __pyx_pybuffer_is_core;
__Pyx_LocalBuf_ND __pyx_pybuffernd_labels;
__Pyx_Buffer __pyx_pybuffer_labels;
__Pyx_LocalBuf_ND __pyx_pybuffernd_neighborhoods;
__Pyx_Buffer __pyx_pybuffer_neighborhoods;
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("inner", 0);
__pyx_pybuffer_is_core.pybuffer.buf = NULL;
__pyx_pybuffer_is_core.refcount = 0;
__pyx_pybuffernd_is_core.data = NULL;
__pyx_pybuffernd_is_core.rcbuffer = &__pyx_pybuffer_is_core;
__pyx_pybuffer_neighborhoods.pybuffer.buf = NULL;
__pyx_pybuffer_neighborhoods.refcount = 0;
__pyx_pybuffernd_neighborhoods.data = NULL;
__pyx_pybuffernd_neighborhoods.rcbuffer = &__pyx_pybuffer_neighborhoods;
__pyx_pybuffer_labels.pybuffer.buf = NULL;
__pyx_pybuffer_labels.refcount = 0;
__pyx_pybuffernd_labels.data = NULL;
__pyx_pybuffernd_labels.rcbuffer = &__pyx_pybuffer_labels;
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_is_core.rcbuffer->pybuffer, (PyObject*)__pyx_v_is_core, &__Pyx_TypeInfo_nn___pyx_t_5numpy_uint8_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 111, __pyx_L1_error)
}
__pyx_pybuffernd_is_core.diminfo[0].strides = __pyx_pybuffernd_is_core.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_is_core.diminfo[0].shape = __pyx_pybuffernd_is_core.rcbuffer->pybuffer.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer, (PyObject*)__pyx_v_neighborhoods, &__Pyx_TypeInfo_object, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 111, __pyx_L1_error)
}
__pyx_pybuffernd_neighborhoods.diminfo[0].strides = __pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_neighborhoods.diminfo[0].shape = __pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_labels.rcbuffer->pybuffer, (PyObject*)__pyx_v_labels, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 111, __pyx_L1_error)
}
__pyx_pybuffernd_labels.diminfo[0].strides = __pyx_pybuffernd_labels.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_labels.diminfo[0].shape = __pyx_pybuffernd_labels.rcbuffer->pybuffer.shape[0];
__Pyx_XDECREF(__pyx_r);
__pyx_t_1 = __Pyx_void_to_None(__pyx_f_7edbscan_6_inner_inner(__pyx_v_is_core, __pyx_v_neighborhoods, __pyx_v_labels, 0)); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 111, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
/* function exit code */
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
__Pyx_PyThreadState_declare
__Pyx_PyThreadState_assign
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_is_core.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_AddTraceback("edbscan._inner.inner", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_is_core.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer);
__pyx_L2:;
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
112: np.ndarray[np.uint8_t, ndim=1, mode='c'] is_core,
113: np.ndarray[object, ndim=1] neighborhoods,
114: np.ndarray[np.npy_intp, ndim=1, mode='c'] labels,
115: ):
116: """Inner-loop of the EDBSCAN algorithm."""
117: # Initialise the stack with all known clusters.
118: cdef int i, idx, n_idx
+119: cdef list stack = []
__pyx_t_1 = PyList_New(0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 119, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __pyx_v_stack = ((PyObject*)__pyx_t_1); __pyx_t_1 = 0;
120: cdef np.ndarray[np.npy_intp, ndim=1, mode='c'] neighborhood
121:
+122: for idx in xrange(labels.shape[0]):
__pyx_t_2 = (__pyx_v_labels->dimensions[0]);
__pyx_t_3 = __pyx_t_2;
for (__pyx_t_4 = 0; __pyx_t_4 < __pyx_t_3; __pyx_t_4+=1) {
__pyx_v_idx = __pyx_t_4;
+123: if labels[idx] >= 0:
__pyx_t_5 = __pyx_v_idx;
__pyx_t_6 = (((*__Pyx_BufPtrCContig1d(npy_intp *, __pyx_pybuffernd_labels.rcbuffer->pybuffer.buf, __pyx_t_5, __pyx_pybuffernd_labels.diminfo[0].strides)) >= 0) != 0);
if (__pyx_t_6) {
/* … */
}
}
+124: stack.append(idx)
__pyx_t_1 = __Pyx_PyInt_From_int(__pyx_v_idx); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 124, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __pyx_t_7 = __Pyx_PyList_Append(__pyx_v_stack, __pyx_t_1); if (unlikely(__pyx_t_7 == ((int)-1))) __PYX_ERR(0, 124, __pyx_L1_error) __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
125:
126: # Density-first search, where the density is defined by the number of neighbors a component has.
127: # The algorithm expands the most dense core components first and ends at the non-core points.
128: # Non-core points are labeled as part of a component, but don't expand their neighborhoods.
+129: while True:
while (1) {
+130: if len(stack) == 0:
__pyx_t_8 = PyList_GET_SIZE(__pyx_v_stack); if (unlikely(__pyx_t_8 == ((Py_ssize_t)-1))) __PYX_ERR(0, 130, __pyx_L1_error) __pyx_t_6 = ((__pyx_t_8 == 0) != 0); if (__pyx_t_6) { /* … */ }
+131: idx = get_unlabeled(is_core=is_core, neighborhoods=neighborhoods, labels=labels)
__pyx_v_idx = __pyx_f_7edbscan_6_inner_get_unlabeled(((PyArrayObject *)__pyx_v_is_core), ((PyArrayObject *)__pyx_v_neighborhoods), ((PyArrayObject *)__pyx_v_labels));
132:
133: # Break if no unlabeled found
+134: if idx == -1:
__pyx_t_6 = ((__pyx_v_idx == -1L) != 0);
if (__pyx_t_6) {
/* … */
}
+135: break
goto __pyx_L7_break;
136:
137: # Break if conflict found
+138: neighborhood = neighborhoods[idx]
__pyx_t_5 = __pyx_v_idx;
__pyx_t_1 = (PyObject *) *__Pyx_BufPtrStrided1d(PyObject **, __pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer.buf, __pyx_t_5, __pyx_pybuffernd_neighborhoods.diminfo[0].strides);
__Pyx_INCREF((PyObject*)__pyx_t_1);
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 138, __pyx_L1_error)
__pyx_t_9 = ((PyArrayObject *)__pyx_t_1);
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer);
__pyx_t_4 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer, (PyObject*)__pyx_t_9, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack);
if (unlikely(__pyx_t_4 < 0)) {
PyErr_Fetch(&__pyx_t_10, &__pyx_t_11, &__pyx_t_12);
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer, (PyObject*)__pyx_v_neighborhood, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {
Py_XDECREF(__pyx_t_10); Py_XDECREF(__pyx_t_11); Py_XDECREF(__pyx_t_12);
__Pyx_RaiseBufferFallbackError();
} else {
PyErr_Restore(__pyx_t_10, __pyx_t_11, __pyx_t_12);
}
__pyx_t_10 = __pyx_t_11 = __pyx_t_12 = 0;
}
__pyx_pybuffernd_neighborhood.diminfo[0].strides = __pyx_pybuffernd_neighborhood.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_neighborhood.diminfo[0].shape = __pyx_pybuffernd_neighborhood.rcbuffer->pybuffer.shape[0];
if (unlikely(__pyx_t_4 < 0)) __PYX_ERR(0, 138, __pyx_L1_error)
}
__pyx_t_9 = 0;
__Pyx_XDECREF_SET(__pyx_v_neighborhood, ((PyArrayObject *)__pyx_t_1));
__pyx_t_1 = 0;
+139: if conflict(neighborhood=neighborhood, labels=labels):
__pyx_t_6 = (__pyx_f_7edbscan_6_inner_conflict(((PyArrayObject *)__pyx_v_neighborhood), ((PyArrayObject *)__pyx_v_labels)) != 0);
if (__pyx_t_6) {
/* … */
}
+140: break
goto __pyx_L7_break;
141:
142: # Add to stack (new cluster found that doesn't lead to a conflict)
+143: labels[idx] = get_max_0(labels)
__pyx_t_5 = __pyx_v_idx;
*__Pyx_BufPtrCContig1d(npy_intp *, __pyx_pybuffernd_labels.rcbuffer->pybuffer.buf, __pyx_t_5, __pyx_pybuffernd_labels.diminfo[0].strides) = __pyx_f_7edbscan_6_inner_get_max_0(((PyArrayObject *)__pyx_v_labels));
+144: stack.append(idx)
__pyx_t_1 = __Pyx_PyInt_From_int(__pyx_v_idx); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 144, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __pyx_t_7 = __Pyx_PyList_Append(__pyx_v_stack, __pyx_t_1); if (unlikely(__pyx_t_7 == ((int)-1))) __PYX_ERR(0, 144, __pyx_L1_error) __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
145:
146: # Pop the most dense index in stack and expand if there's no conflict.
147: # Only add unlabeled core indices to the stack.
+148: idx = pop(stack=stack, neighborhoods=neighborhoods)
__pyx_v_idx = __pyx_f_7edbscan_6_inner_pop(__pyx_v_stack, ((PyArrayObject *)__pyx_v_neighborhoods));
+149: neighborhood = neighborhoods[idx]
__pyx_t_5 = __pyx_v_idx;
__pyx_t_1 = (PyObject *) *__Pyx_BufPtrStrided1d(PyObject **, __pyx_pybuffernd_neighborhoods.rcbuffer->pybuffer.buf, __pyx_t_5, __pyx_pybuffernd_neighborhoods.diminfo[0].strides);
__Pyx_INCREF((PyObject*)__pyx_t_1);
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 149, __pyx_L1_error)
__pyx_t_9 = ((PyArrayObject *)__pyx_t_1);
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer);
__pyx_t_4 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer, (PyObject*)__pyx_t_9, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack);
if (unlikely(__pyx_t_4 < 0)) {
PyErr_Fetch(&__pyx_t_12, &__pyx_t_11, &__pyx_t_10);
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_neighborhood.rcbuffer->pybuffer, (PyObject*)__pyx_v_neighborhood, &__Pyx_TypeInfo_nn_npy_intp, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {
Py_XDECREF(__pyx_t_12); Py_XDECREF(__pyx_t_11); Py_XDECREF(__pyx_t_10);
__Pyx_RaiseBufferFallbackError();
} else {
PyErr_Restore(__pyx_t_12, __pyx_t_11, __pyx_t_10);
}
__pyx_t_12 = __pyx_t_11 = __pyx_t_10 = 0;
}
__pyx_pybuffernd_neighborhood.diminfo[0].strides = __pyx_pybuffernd_neighborhood.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_neighborhood.diminfo[0].shape = __pyx_pybuffernd_neighborhood.rcbuffer->pybuffer.shape[0];
if (unlikely(__pyx_t_4 < 0)) __PYX_ERR(0, 149, __pyx_L1_error)
}
__pyx_t_9 = 0;
__Pyx_XDECREF_SET(__pyx_v_neighborhood, ((PyArrayObject *)__pyx_t_1));
__pyx_t_1 = 0;
+150: if not conflict(neighborhood=neighborhood, labels=labels):
__pyx_t_6 = ((!(__pyx_f_7edbscan_6_inner_conflict(((PyArrayObject *)__pyx_v_neighborhood), ((PyArrayObject *)__pyx_v_labels)) != 0)) != 0);
if (__pyx_t_6) {
/* … */
}
}
__pyx_L7_break:;
+151: for i in xrange(neighborhood.shape[0]):
__pyx_t_2 = (__pyx_v_neighborhood->dimensions[0]);
__pyx_t_3 = __pyx_t_2;
for (__pyx_t_4 = 0; __pyx_t_4 < __pyx_t_3; __pyx_t_4+=1) {
__pyx_v_i = __pyx_t_4;
+152: n_idx = neighborhood[i]
__pyx_t_5 = __pyx_v_i;
__pyx_v_n_idx = (*__Pyx_BufPtrCContig1d(npy_intp *, __pyx_pybuffernd_neighborhood.rcbuffer->pybuffer.buf, __pyx_t_5, __pyx_pybuffernd_neighborhood.diminfo[0].strides));
153: # Label if it's an unlabeled sample
+154: if labels[n_idx] == -2:
__pyx_t_5 = __pyx_v_n_idx;
__pyx_t_6 = (((*__Pyx_BufPtrCContig1d(npy_intp *, __pyx_pybuffernd_labels.rcbuffer->pybuffer.buf, __pyx_t_5, __pyx_pybuffernd_labels.diminfo[0].strides)) == -2L) != 0);
if (__pyx_t_6) {
/* … */
}
}
+155: labels[n_idx] = labels[idx]
__pyx_t_5 = __pyx_v_idx;
__pyx_t_13 = __pyx_v_n_idx;
*__Pyx_BufPtrCContig1d(npy_intp *, __pyx_pybuffernd_labels.rcbuffer->pybuffer.buf, __pyx_t_13, __pyx_pybuffernd_labels.diminfo[0].strides) = (*__Pyx_BufPtrCContig1d(npy_intp *, __pyx_pybuffernd_labels.rcbuffer->pybuffer.buf, __pyx_t_5, __pyx_pybuffernd_labels.diminfo[0].strides));
156:
157: # Only possible to expand if it's a core point
+158: if is_core[n_idx]:
__pyx_t_5 = __pyx_v_n_idx;
__pyx_t_6 = ((*__Pyx_BufPtrCContig1d(__pyx_t_5numpy_uint8_t *, __pyx_pybuffernd_is_core.rcbuffer->pybuffer.buf, __pyx_t_5, __pyx_pybuffernd_is_core.diminfo[0].strides)) != 0);
if (__pyx_t_6) {
/* … */
}
+159: stack.append(n_idx)
__pyx_t_1 = __Pyx_PyInt_From_int(__pyx_v_n_idx); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 159, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __pyx_t_7 = __Pyx_PyList_Append(__pyx_v_stack, __pyx_t_1); if (unlikely(__pyx_t_7 == ((int)-1))) __PYX_ERR(0, 159, __pyx_L1_error) __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;