12206 lines
556 KiB
C
12206 lines
556 KiB
C
/* Generated by Cython 0.16 on Thu Jun 6 13:59:16 2013 */
|
|
|
|
#define PY_SSIZE_T_CLEAN
|
|
#include "Python.h"
|
|
#ifndef Py_PYTHON_H
|
|
#error Python headers needed to compile C extensions, please install development version of Python.
|
|
#elif PY_VERSION_HEX < 0x02040000
|
|
#error Cython requires Python 2.4+.
|
|
#else
|
|
#include <stddef.h> /* For offsetof */
|
|
#ifndef offsetof
|
|
#define offsetof(type, member) ( (size_t) & ((type*)0) -> member )
|
|
#endif
|
|
|
|
#if !defined(WIN32) && !defined(MS_WINDOWS)
|
|
#ifndef __stdcall
|
|
#define __stdcall
|
|
#endif
|
|
#ifndef __cdecl
|
|
#define __cdecl
|
|
#endif
|
|
#ifndef __fastcall
|
|
#define __fastcall
|
|
#endif
|
|
#endif
|
|
|
|
#ifndef DL_IMPORT
|
|
#define DL_IMPORT(t) t
|
|
#endif
|
|
#ifndef DL_EXPORT
|
|
#define DL_EXPORT(t) t
|
|
#endif
|
|
|
|
#ifndef PY_LONG_LONG
|
|
#define PY_LONG_LONG LONG_LONG
|
|
#endif
|
|
|
|
#ifndef Py_HUGE_VAL
|
|
#define Py_HUGE_VAL HUGE_VAL
|
|
#endif
|
|
|
|
#ifdef PYPY_VERSION
|
|
#define CYTHON_COMPILING_IN_PYPY 1
|
|
#define CYTHON_COMPILING_IN_CPYTHON 0
|
|
#else
|
|
#define CYTHON_COMPILING_IN_PYPY 0
|
|
#define CYTHON_COMPILING_IN_CPYTHON 1
|
|
#endif
|
|
|
|
#if CYTHON_COMPILING_IN_PYPY
|
|
#define __Pyx_PyCFunction_Call PyObject_Call
|
|
#else
|
|
#define __Pyx_PyCFunction_Call PyCFunction_Call
|
|
#endif
|
|
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
typedef int Py_ssize_t;
|
|
#define PY_SSIZE_T_MAX INT_MAX
|
|
#define PY_SSIZE_T_MIN INT_MIN
|
|
#define PY_FORMAT_SIZE_T ""
|
|
#define PyInt_FromSsize_t(z) PyInt_FromLong(z)
|
|
#define PyInt_AsSsize_t(o) __Pyx_PyInt_AsInt(o)
|
|
#define PyNumber_Index(o) PyNumber_Int(o)
|
|
#define PyIndex_Check(o) PyNumber_Check(o)
|
|
#define PyErr_WarnEx(category, message, stacklevel) PyErr_Warn(category, message)
|
|
#define __PYX_BUILD_PY_SSIZE_T "i"
|
|
#else
|
|
#define __PYX_BUILD_PY_SSIZE_T "n"
|
|
#endif
|
|
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
#define Py_REFCNT(ob) (((PyObject*)(ob))->ob_refcnt)
|
|
#define Py_TYPE(ob) (((PyObject*)(ob))->ob_type)
|
|
#define Py_SIZE(ob) (((PyVarObject*)(ob))->ob_size)
|
|
#define PyVarObject_HEAD_INIT(type, size) \
|
|
PyObject_HEAD_INIT(type) size,
|
|
#define PyType_Modified(t)
|
|
|
|
typedef struct {
|
|
void *buf;
|
|
PyObject *obj;
|
|
Py_ssize_t len;
|
|
Py_ssize_t itemsize;
|
|
int readonly;
|
|
int ndim;
|
|
char *format;
|
|
Py_ssize_t *shape;
|
|
Py_ssize_t *strides;
|
|
Py_ssize_t *suboffsets;
|
|
void *internal;
|
|
} Py_buffer;
|
|
|
|
#define PyBUF_SIMPLE 0
|
|
#define PyBUF_WRITABLE 0x0001
|
|
#define PyBUF_FORMAT 0x0004
|
|
#define PyBUF_ND 0x0008
|
|
#define PyBUF_STRIDES (0x0010 | PyBUF_ND)
|
|
#define PyBUF_C_CONTIGUOUS (0x0020 | PyBUF_STRIDES)
|
|
#define PyBUF_F_CONTIGUOUS (0x0040 | PyBUF_STRIDES)
|
|
#define PyBUF_ANY_CONTIGUOUS (0x0080 | PyBUF_STRIDES)
|
|
#define PyBUF_INDIRECT (0x0100 | PyBUF_STRIDES)
|
|
#define PyBUF_RECORDS (PyBUF_STRIDES | PyBUF_FORMAT | PyBUF_WRITABLE)
|
|
#define PyBUF_FULL (PyBUF_INDIRECT | PyBUF_FORMAT | PyBUF_WRITABLE)
|
|
|
|
typedef int (*getbufferproc)(PyObject *, Py_buffer *, int);
|
|
typedef void (*releasebufferproc)(PyObject *, Py_buffer *);
|
|
#endif
|
|
|
|
#if PY_MAJOR_VERSION < 3
|
|
#define __Pyx_BUILTIN_MODULE_NAME "__builtin__"
|
|
#define __Pyx_PyCode_New(a, k, l, s, f, code, c, n, v, fv, cell, fn, name, fline, lnos) \
|
|
PyCode_New(a, l, s, f, code, c, n, v, fv, cell, fn, name, fline, lnos)
|
|
#else
|
|
#define __Pyx_BUILTIN_MODULE_NAME "builtins"
|
|
#define __Pyx_PyCode_New(a, k, l, s, f, code, c, n, v, fv, cell, fn, name, fline, lnos) \
|
|
PyCode_New(a, k, l, s, f, code, c, n, v, fv, cell, fn, name, fline, lnos)
|
|
#endif
|
|
|
|
#if PY_MAJOR_VERSION < 3 && PY_MINOR_VERSION < 6
|
|
#define PyUnicode_FromString(s) PyUnicode_Decode(s, strlen(s), "UTF-8", "strict")
|
|
#endif
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
#define Py_TPFLAGS_CHECKTYPES 0
|
|
#define Py_TPFLAGS_HAVE_INDEX 0
|
|
#endif
|
|
|
|
#if (PY_VERSION_HEX < 0x02060000) || (PY_MAJOR_VERSION >= 3)
|
|
#define Py_TPFLAGS_HAVE_NEWBUFFER 0
|
|
#endif
|
|
|
|
|
|
#if PY_VERSION_HEX > 0x03030000 && defined(PyUnicode_GET_LENGTH)
|
|
#define CYTHON_PEP393_ENABLED 1
|
|
#define __Pyx_PyUnicode_GET_LENGTH(u) PyUnicode_GET_LENGTH(u)
|
|
#define __Pyx_PyUnicode_READ_CHAR(u, i) PyUnicode_READ_CHAR(u, i)
|
|
#else
|
|
#define CYTHON_PEP393_ENABLED 0
|
|
#define __Pyx_PyUnicode_GET_LENGTH(u) PyUnicode_GET_SIZE(u)
|
|
#define __Pyx_PyUnicode_READ_CHAR(u, i) ((Py_UCS4)(PyUnicode_AS_UNICODE(u)[i]))
|
|
#endif
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
#define PyBaseString_Type PyUnicode_Type
|
|
#define PyStringObject PyUnicodeObject
|
|
#define PyString_Type PyUnicode_Type
|
|
#define PyString_Check PyUnicode_Check
|
|
#define PyString_CheckExact PyUnicode_CheckExact
|
|
#endif
|
|
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
#define PyBytesObject PyStringObject
|
|
#define PyBytes_Type PyString_Type
|
|
#define PyBytes_Check PyString_Check
|
|
#define PyBytes_CheckExact PyString_CheckExact
|
|
#define PyBytes_FromString PyString_FromString
|
|
#define PyBytes_FromStringAndSize PyString_FromStringAndSize
|
|
#define PyBytes_FromFormat PyString_FromFormat
|
|
#define PyBytes_DecodeEscape PyString_DecodeEscape
|
|
#define PyBytes_AsString PyString_AsString
|
|
#define PyBytes_AsStringAndSize PyString_AsStringAndSize
|
|
#define PyBytes_Size PyString_Size
|
|
#define PyBytes_AS_STRING PyString_AS_STRING
|
|
#define PyBytes_GET_SIZE PyString_GET_SIZE
|
|
#define PyBytes_Repr PyString_Repr
|
|
#define PyBytes_Concat PyString_Concat
|
|
#define PyBytes_ConcatAndDel PyString_ConcatAndDel
|
|
#endif
|
|
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
#define PySet_Check(obj) PyObject_TypeCheck(obj, &PySet_Type)
|
|
#define PyFrozenSet_Check(obj) PyObject_TypeCheck(obj, &PyFrozenSet_Type)
|
|
#endif
|
|
#ifndef PySet_CheckExact
|
|
#define PySet_CheckExact(obj) (Py_TYPE(obj) == &PySet_Type)
|
|
#endif
|
|
|
|
#define __Pyx_TypeCheck(obj, type) PyObject_TypeCheck(obj, (PyTypeObject *)type)
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
#define PyIntObject PyLongObject
|
|
#define PyInt_Type PyLong_Type
|
|
#define PyInt_Check(op) PyLong_Check(op)
|
|
#define PyInt_CheckExact(op) PyLong_CheckExact(op)
|
|
#define PyInt_FromString PyLong_FromString
|
|
#define PyInt_FromUnicode PyLong_FromUnicode
|
|
#define PyInt_FromLong PyLong_FromLong
|
|
#define PyInt_FromSize_t PyLong_FromSize_t
|
|
#define PyInt_FromSsize_t PyLong_FromSsize_t
|
|
#define PyInt_AsLong PyLong_AsLong
|
|
#define PyInt_AS_LONG PyLong_AS_LONG
|
|
#define PyInt_AsSsize_t PyLong_AsSsize_t
|
|
#define PyInt_AsUnsignedLongMask PyLong_AsUnsignedLongMask
|
|
#define PyInt_AsUnsignedLongLongMask PyLong_AsUnsignedLongLongMask
|
|
#endif
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
#define PyBoolObject PyLongObject
|
|
#endif
|
|
|
|
#if PY_VERSION_HEX < 0x03020000
|
|
typedef long Py_hash_t;
|
|
#define __Pyx_PyInt_FromHash_t PyInt_FromLong
|
|
#define __Pyx_PyInt_AsHash_t PyInt_AsLong
|
|
#else
|
|
#define __Pyx_PyInt_FromHash_t PyInt_FromSsize_t
|
|
#define __Pyx_PyInt_AsHash_t PyInt_AsSsize_t
|
|
#endif
|
|
|
|
#if (PY_MAJOR_VERSION < 3) || (PY_VERSION_HEX >= 0x03010300)
|
|
#define __Pyx_PySequence_GetSlice(obj, a, b) PySequence_GetSlice(obj, a, b)
|
|
#define __Pyx_PySequence_SetSlice(obj, a, b, value) PySequence_SetSlice(obj, a, b, value)
|
|
#define __Pyx_PySequence_DelSlice(obj, a, b) PySequence_DelSlice(obj, a, b)
|
|
#else
|
|
#define __Pyx_PySequence_GetSlice(obj, a, b) (unlikely(!(obj)) ? \
|
|
(PyErr_SetString(PyExc_SystemError, "null argument to internal routine"), (PyObject*)0) : \
|
|
(likely((obj)->ob_type->tp_as_mapping) ? (PySequence_GetSlice(obj, a, b)) : \
|
|
(PyErr_Format(PyExc_TypeError, "'%.200s' object is unsliceable", (obj)->ob_type->tp_name), (PyObject*)0)))
|
|
#define __Pyx_PySequence_SetSlice(obj, a, b, value) (unlikely(!(obj)) ? \
|
|
(PyErr_SetString(PyExc_SystemError, "null argument to internal routine"), -1) : \
|
|
(likely((obj)->ob_type->tp_as_mapping) ? (PySequence_SetSlice(obj, a, b, value)) : \
|
|
(PyErr_Format(PyExc_TypeError, "'%.200s' object doesn't support slice assignment", (obj)->ob_type->tp_name), -1)))
|
|
#define __Pyx_PySequence_DelSlice(obj, a, b) (unlikely(!(obj)) ? \
|
|
(PyErr_SetString(PyExc_SystemError, "null argument to internal routine"), -1) : \
|
|
(likely((obj)->ob_type->tp_as_mapping) ? (PySequence_DelSlice(obj, a, b)) : \
|
|
(PyErr_Format(PyExc_TypeError, "'%.200s' object doesn't support slice deletion", (obj)->ob_type->tp_name), -1)))
|
|
#endif
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
#define PyMethod_New(func, self, klass) ((self) ? PyMethod_New(func, self) : PyInstanceMethod_New(func))
|
|
#endif
|
|
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
#define __Pyx_GetAttrString(o,n) PyObject_GetAttrString((o),((char *)(n)))
|
|
#define __Pyx_SetAttrString(o,n,a) PyObject_SetAttrString((o),((char *)(n)),(a))
|
|
#define __Pyx_DelAttrString(o,n) PyObject_DelAttrString((o),((char *)(n)))
|
|
#else
|
|
#define __Pyx_GetAttrString(o,n) PyObject_GetAttrString((o),(n))
|
|
#define __Pyx_SetAttrString(o,n,a) PyObject_SetAttrString((o),(n),(a))
|
|
#define __Pyx_DelAttrString(o,n) PyObject_DelAttrString((o),(n))
|
|
#endif
|
|
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
#define __Pyx_NAMESTR(n) ((char *)(n))
|
|
#define __Pyx_DOCSTR(n) ((char *)(n))
|
|
#else
|
|
#define __Pyx_NAMESTR(n) (n)
|
|
#define __Pyx_DOCSTR(n) (n)
|
|
#endif
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
#define __Pyx_PyNumber_Divide(x,y) PyNumber_TrueDivide(x,y)
|
|
#define __Pyx_PyNumber_InPlaceDivide(x,y) PyNumber_InPlaceTrueDivide(x,y)
|
|
#else
|
|
#define __Pyx_PyNumber_Divide(x,y) PyNumber_Divide(x,y)
|
|
#define __Pyx_PyNumber_InPlaceDivide(x,y) PyNumber_InPlaceDivide(x,y)
|
|
#endif
|
|
|
|
#ifndef __PYX_EXTERN_C
|
|
#ifdef __cplusplus
|
|
#define __PYX_EXTERN_C extern "C"
|
|
#else
|
|
#define __PYX_EXTERN_C extern
|
|
#endif
|
|
#endif
|
|
|
|
#if defined(WIN32) || defined(MS_WINDOWS)
|
|
#define _USE_MATH_DEFINES
|
|
#endif
|
|
#include <math.h>
|
|
#define __PYX_HAVE__sklearn__utils__mst___traversal
|
|
#define __PYX_HAVE_API__sklearn__utils__mst___traversal
|
|
#include "stdio.h"
|
|
#include "stdlib.h"
|
|
#include "numpy/arrayobject.h"
|
|
#include "numpy/ufuncobject.h"
|
|
#ifdef _OPENMP
|
|
#include <omp.h>
|
|
#endif /* _OPENMP */
|
|
|
|
#ifdef PYREX_WITHOUT_ASSERTIONS
|
|
#define CYTHON_WITHOUT_ASSERTIONS
|
|
#endif
|
|
|
|
|
|
/* inline attribute */
|
|
#ifndef CYTHON_INLINE
|
|
#if defined(__GNUC__)
|
|
#define CYTHON_INLINE __inline__
|
|
#elif defined(_MSC_VER)
|
|
#define CYTHON_INLINE __inline
|
|
#elif defined (__STDC_VERSION__) && __STDC_VERSION__ >= 199901L
|
|
#define CYTHON_INLINE inline
|
|
#else
|
|
#define CYTHON_INLINE
|
|
#endif
|
|
#endif
|
|
|
|
/* unused attribute */
|
|
#ifndef CYTHON_UNUSED
|
|
# if defined(__GNUC__)
|
|
# if !(defined(__cplusplus)) || (__GNUC__ > 3 || (__GNUC__ == 3 && __GNUC_MINOR__ >= 4))
|
|
# define CYTHON_UNUSED __attribute__ ((__unused__))
|
|
# else
|
|
# define CYTHON_UNUSED
|
|
# endif
|
|
# elif defined(__ICC) || (defined(__INTEL_COMPILER) && !defined(_MSC_VER))
|
|
# define CYTHON_UNUSED __attribute__ ((__unused__))
|
|
# else
|
|
# define CYTHON_UNUSED
|
|
# endif
|
|
#endif
|
|
|
|
typedef struct {PyObject **p; char *s; const long n; const char* encoding; const char is_unicode; const char is_str; const char intern; } __Pyx_StringTabEntry; /*proto*/
|
|
|
|
|
|
/* Type Conversion Predeclarations */
|
|
|
|
#define __Pyx_PyBytes_FromUString(s) PyBytes_FromString((char*)s)
|
|
#define __Pyx_PyBytes_AsUString(s) ((unsigned char*) PyBytes_AsString(s))
|
|
|
|
#define __Pyx_Owned_Py_None(b) (Py_INCREF(Py_None), Py_None)
|
|
#define __Pyx_PyBool_FromLong(b) ((b) ? (Py_INCREF(Py_True), Py_True) : (Py_INCREF(Py_False), Py_False))
|
|
static CYTHON_INLINE int __Pyx_PyObject_IsTrue(PyObject*);
|
|
static CYTHON_INLINE PyObject* __Pyx_PyNumber_Int(PyObject* x);
|
|
|
|
static CYTHON_INLINE Py_ssize_t __Pyx_PyIndex_AsSsize_t(PyObject*);
|
|
static CYTHON_INLINE PyObject * __Pyx_PyInt_FromSize_t(size_t);
|
|
static CYTHON_INLINE size_t __Pyx_PyInt_AsSize_t(PyObject*);
|
|
|
|
#define __pyx_PyFloat_AsDouble(x) (PyFloat_CheckExact(x) ? PyFloat_AS_DOUBLE(x) : PyFloat_AsDouble(x))
|
|
#define __pyx_PyFloat_AsFloat(x) ((float) __pyx_PyFloat_AsDouble(x))
|
|
|
|
#ifdef __GNUC__
|
|
/* Test for GCC > 2.95 */
|
|
#if __GNUC__ > 2 || (__GNUC__ == 2 && (__GNUC_MINOR__ > 95))
|
|
#define likely(x) __builtin_expect(!!(x), 1)
|
|
#define unlikely(x) __builtin_expect(!!(x), 0)
|
|
#else /* __GNUC__ > 2 ... */
|
|
#define likely(x) (x)
|
|
#define unlikely(x) (x)
|
|
#endif /* __GNUC__ > 2 ... */
|
|
#else /* __GNUC__ */
|
|
#define likely(x) (x)
|
|
#define unlikely(x) (x)
|
|
#endif /* __GNUC__ */
|
|
|
|
static PyObject *__pyx_m;
|
|
static PyObject *__pyx_b;
|
|
static PyObject *__pyx_empty_tuple;
|
|
static PyObject *__pyx_empty_bytes;
|
|
static int __pyx_lineno;
|
|
static int __pyx_clineno = 0;
|
|
static const char * __pyx_cfilenm= __FILE__;
|
|
static const char *__pyx_filename;
|
|
|
|
#if !defined(CYTHON_CCOMPLEX)
|
|
#if defined(__cplusplus)
|
|
#define CYTHON_CCOMPLEX 1
|
|
#elif defined(_Complex_I)
|
|
#define CYTHON_CCOMPLEX 1
|
|
#else
|
|
#define CYTHON_CCOMPLEX 0
|
|
#endif
|
|
#endif
|
|
#if CYTHON_CCOMPLEX
|
|
#ifdef __cplusplus
|
|
#include <complex>
|
|
#else
|
|
#include <complex.h>
|
|
#endif
|
|
#endif
|
|
#if CYTHON_CCOMPLEX && !defined(__cplusplus) && defined(__sun__) && defined(__GNUC__)
|
|
#undef _Complex_I
|
|
#define _Complex_I 1.0fj
|
|
#endif
|
|
|
|
|
|
static const char *__pyx_f[] = {
|
|
"_traversal.pyx",
|
|
"numpy.pxd",
|
|
};
|
|
#define IS_UNSIGNED(type) (((type) -1) > 0)
|
|
struct __Pyx_StructField_;
|
|
#define __PYX_BUF_FLAGS_PACKED_STRUCT (1 << 0)
|
|
typedef struct {
|
|
const char* name; /* for error messages only */
|
|
struct __Pyx_StructField_* fields;
|
|
size_t size; /* sizeof(type) */
|
|
size_t arraysize[8]; /* length of array in each dimension */
|
|
int ndim;
|
|
char typegroup; /* _R_eal, _C_omplex, Signed _I_nt, _U_nsigned int, _S_truct, _P_ointer, _O_bject */
|
|
char is_unsigned;
|
|
int flags;
|
|
} __Pyx_TypeInfo;
|
|
typedef struct __Pyx_StructField_ {
|
|
__Pyx_TypeInfo* type;
|
|
const char* name;
|
|
size_t offset;
|
|
} __Pyx_StructField;
|
|
typedef struct {
|
|
__Pyx_StructField* field;
|
|
size_t parent_offset;
|
|
} __Pyx_BufFmt_StackElem;
|
|
typedef struct {
|
|
__Pyx_StructField root;
|
|
__Pyx_BufFmt_StackElem* head;
|
|
size_t fmt_offset;
|
|
size_t new_count, enc_count;
|
|
size_t struct_alignment;
|
|
int is_complex;
|
|
char enc_type;
|
|
char new_packmode;
|
|
char enc_packmode;
|
|
char is_valid_array;
|
|
} __Pyx_BufFmt_Context;
|
|
|
|
|
|
/* "numpy.pxd":722
|
|
* # in Cython to enable them only on the right systems.
|
|
*
|
|
* ctypedef npy_int8 int8_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_int16 int16_t
|
|
* ctypedef npy_int32 int32_t
|
|
*/
|
|
typedef npy_int8 __pyx_t_5numpy_int8_t;
|
|
|
|
/* "numpy.pxd":723
|
|
*
|
|
* ctypedef npy_int8 int8_t
|
|
* ctypedef npy_int16 int16_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_int32 int32_t
|
|
* ctypedef npy_int64 int64_t
|
|
*/
|
|
typedef npy_int16 __pyx_t_5numpy_int16_t;
|
|
|
|
/* "numpy.pxd":724
|
|
* ctypedef npy_int8 int8_t
|
|
* ctypedef npy_int16 int16_t
|
|
* ctypedef npy_int32 int32_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_int64 int64_t
|
|
* #ctypedef npy_int96 int96_t
|
|
*/
|
|
typedef npy_int32 __pyx_t_5numpy_int32_t;
|
|
|
|
/* "numpy.pxd":725
|
|
* ctypedef npy_int16 int16_t
|
|
* ctypedef npy_int32 int32_t
|
|
* ctypedef npy_int64 int64_t # <<<<<<<<<<<<<<
|
|
* #ctypedef npy_int96 int96_t
|
|
* #ctypedef npy_int128 int128_t
|
|
*/
|
|
typedef npy_int64 __pyx_t_5numpy_int64_t;
|
|
|
|
/* "numpy.pxd":729
|
|
* #ctypedef npy_int128 int128_t
|
|
*
|
|
* ctypedef npy_uint8 uint8_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_uint16 uint16_t
|
|
* ctypedef npy_uint32 uint32_t
|
|
*/
|
|
typedef npy_uint8 __pyx_t_5numpy_uint8_t;
|
|
|
|
/* "numpy.pxd":730
|
|
*
|
|
* ctypedef npy_uint8 uint8_t
|
|
* ctypedef npy_uint16 uint16_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_uint32 uint32_t
|
|
* ctypedef npy_uint64 uint64_t
|
|
*/
|
|
typedef npy_uint16 __pyx_t_5numpy_uint16_t;
|
|
|
|
/* "numpy.pxd":731
|
|
* ctypedef npy_uint8 uint8_t
|
|
* ctypedef npy_uint16 uint16_t
|
|
* ctypedef npy_uint32 uint32_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_uint64 uint64_t
|
|
* #ctypedef npy_uint96 uint96_t
|
|
*/
|
|
typedef npy_uint32 __pyx_t_5numpy_uint32_t;
|
|
|
|
/* "numpy.pxd":732
|
|
* ctypedef npy_uint16 uint16_t
|
|
* ctypedef npy_uint32 uint32_t
|
|
* ctypedef npy_uint64 uint64_t # <<<<<<<<<<<<<<
|
|
* #ctypedef npy_uint96 uint96_t
|
|
* #ctypedef npy_uint128 uint128_t
|
|
*/
|
|
typedef npy_uint64 __pyx_t_5numpy_uint64_t;
|
|
|
|
/* "numpy.pxd":736
|
|
* #ctypedef npy_uint128 uint128_t
|
|
*
|
|
* ctypedef npy_float32 float32_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_float64 float64_t
|
|
* #ctypedef npy_float80 float80_t
|
|
*/
|
|
typedef npy_float32 __pyx_t_5numpy_float32_t;
|
|
|
|
/* "numpy.pxd":737
|
|
*
|
|
* ctypedef npy_float32 float32_t
|
|
* ctypedef npy_float64 float64_t # <<<<<<<<<<<<<<
|
|
* #ctypedef npy_float80 float80_t
|
|
* #ctypedef npy_float128 float128_t
|
|
*/
|
|
typedef npy_float64 __pyx_t_5numpy_float64_t;
|
|
|
|
/* "numpy.pxd":746
|
|
* # The int types are mapped a bit surprising --
|
|
* # numpy.int corresponds to 'l' and numpy.long to 'q'
|
|
* ctypedef npy_long int_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_longlong long_t
|
|
* ctypedef npy_longlong longlong_t
|
|
*/
|
|
typedef npy_long __pyx_t_5numpy_int_t;
|
|
|
|
/* "numpy.pxd":747
|
|
* # numpy.int corresponds to 'l' and numpy.long to 'q'
|
|
* ctypedef npy_long int_t
|
|
* ctypedef npy_longlong long_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_longlong longlong_t
|
|
*
|
|
*/
|
|
typedef npy_longlong __pyx_t_5numpy_long_t;
|
|
|
|
/* "numpy.pxd":748
|
|
* ctypedef npy_long int_t
|
|
* ctypedef npy_longlong long_t
|
|
* ctypedef npy_longlong longlong_t # <<<<<<<<<<<<<<
|
|
*
|
|
* ctypedef npy_ulong uint_t
|
|
*/
|
|
typedef npy_longlong __pyx_t_5numpy_longlong_t;
|
|
|
|
/* "numpy.pxd":750
|
|
* ctypedef npy_longlong longlong_t
|
|
*
|
|
* ctypedef npy_ulong uint_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_ulonglong ulong_t
|
|
* ctypedef npy_ulonglong ulonglong_t
|
|
*/
|
|
typedef npy_ulong __pyx_t_5numpy_uint_t;
|
|
|
|
/* "numpy.pxd":751
|
|
*
|
|
* ctypedef npy_ulong uint_t
|
|
* ctypedef npy_ulonglong ulong_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_ulonglong ulonglong_t
|
|
*
|
|
*/
|
|
typedef npy_ulonglong __pyx_t_5numpy_ulong_t;
|
|
|
|
/* "numpy.pxd":752
|
|
* ctypedef npy_ulong uint_t
|
|
* ctypedef npy_ulonglong ulong_t
|
|
* ctypedef npy_ulonglong ulonglong_t # <<<<<<<<<<<<<<
|
|
*
|
|
* ctypedef npy_intp intp_t
|
|
*/
|
|
typedef npy_ulonglong __pyx_t_5numpy_ulonglong_t;
|
|
|
|
/* "numpy.pxd":754
|
|
* ctypedef npy_ulonglong ulonglong_t
|
|
*
|
|
* ctypedef npy_intp intp_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_uintp uintp_t
|
|
*
|
|
*/
|
|
typedef npy_intp __pyx_t_5numpy_intp_t;
|
|
|
|
/* "numpy.pxd":755
|
|
*
|
|
* ctypedef npy_intp intp_t
|
|
* ctypedef npy_uintp uintp_t # <<<<<<<<<<<<<<
|
|
*
|
|
* ctypedef npy_double float_t
|
|
*/
|
|
typedef npy_uintp __pyx_t_5numpy_uintp_t;
|
|
|
|
/* "numpy.pxd":757
|
|
* ctypedef npy_uintp uintp_t
|
|
*
|
|
* ctypedef npy_double float_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_double double_t
|
|
* ctypedef npy_longdouble longdouble_t
|
|
*/
|
|
typedef npy_double __pyx_t_5numpy_float_t;
|
|
|
|
/* "numpy.pxd":758
|
|
*
|
|
* ctypedef npy_double float_t
|
|
* ctypedef npy_double double_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_longdouble longdouble_t
|
|
*
|
|
*/
|
|
typedef npy_double __pyx_t_5numpy_double_t;
|
|
|
|
/* "numpy.pxd":759
|
|
* ctypedef npy_double float_t
|
|
* ctypedef npy_double double_t
|
|
* ctypedef npy_longdouble longdouble_t # <<<<<<<<<<<<<<
|
|
*
|
|
* ctypedef npy_cfloat cfloat_t
|
|
*/
|
|
typedef npy_longdouble __pyx_t_5numpy_longdouble_t;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":19
|
|
*
|
|
* DTYPE = np.float64
|
|
* ctypedef np.float64_t DTYPE_t # <<<<<<<<<<<<<<
|
|
*
|
|
* ITYPE = np.int32
|
|
*/
|
|
typedef __pyx_t_5numpy_float64_t __pyx_t_7sklearn_5utils_3mst_10_traversal_DTYPE_t;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":22
|
|
*
|
|
* ITYPE = np.int32
|
|
* ctypedef np.int32_t ITYPE_t # <<<<<<<<<<<<<<
|
|
*
|
|
* # EPS is the precision of DTYPE
|
|
*/
|
|
typedef __pyx_t_5numpy_int32_t __pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t;
|
|
#if CYTHON_CCOMPLEX
|
|
#ifdef __cplusplus
|
|
typedef ::std::complex< float > __pyx_t_float_complex;
|
|
#else
|
|
typedef float _Complex __pyx_t_float_complex;
|
|
#endif
|
|
#else
|
|
typedef struct { float real, imag; } __pyx_t_float_complex;
|
|
#endif
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#ifdef __cplusplus
|
|
typedef ::std::complex< double > __pyx_t_double_complex;
|
|
#else
|
|
typedef double _Complex __pyx_t_double_complex;
|
|
#endif
|
|
#else
|
|
typedef struct { double real, imag; } __pyx_t_double_complex;
|
|
#endif
|
|
|
|
|
|
/*--- Type declarations ---*/
|
|
|
|
/* "numpy.pxd":761
|
|
* ctypedef npy_longdouble longdouble_t
|
|
*
|
|
* ctypedef npy_cfloat cfloat_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_cdouble cdouble_t
|
|
* ctypedef npy_clongdouble clongdouble_t
|
|
*/
|
|
typedef npy_cfloat __pyx_t_5numpy_cfloat_t;
|
|
|
|
/* "numpy.pxd":762
|
|
*
|
|
* ctypedef npy_cfloat cfloat_t
|
|
* ctypedef npy_cdouble cdouble_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_clongdouble clongdouble_t
|
|
*
|
|
*/
|
|
typedef npy_cdouble __pyx_t_5numpy_cdouble_t;
|
|
|
|
/* "numpy.pxd":763
|
|
* ctypedef npy_cfloat cfloat_t
|
|
* ctypedef npy_cdouble cdouble_t
|
|
* ctypedef npy_clongdouble clongdouble_t # <<<<<<<<<<<<<<
|
|
*
|
|
* ctypedef npy_cdouble complex_t
|
|
*/
|
|
typedef npy_clongdouble __pyx_t_5numpy_clongdouble_t;
|
|
|
|
/* "numpy.pxd":765
|
|
* ctypedef npy_clongdouble clongdouble_t
|
|
*
|
|
* ctypedef npy_cdouble complex_t # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew1(a):
|
|
*/
|
|
typedef npy_cdouble __pyx_t_5numpy_complex_t;
|
|
#ifndef CYTHON_REFNANNY
|
|
#define CYTHON_REFNANNY 0
|
|
#endif
|
|
#if CYTHON_REFNANNY
|
|
typedef struct {
|
|
void (*INCREF)(void*, PyObject*, int);
|
|
void (*DECREF)(void*, PyObject*, int);
|
|
void (*GOTREF)(void*, PyObject*, int);
|
|
void (*GIVEREF)(void*, PyObject*, int);
|
|
void* (*SetupContext)(const char*, int, const char*);
|
|
void (*FinishContext)(void**);
|
|
} __Pyx_RefNannyAPIStruct;
|
|
static __Pyx_RefNannyAPIStruct *__Pyx_RefNanny = NULL;
|
|
static __Pyx_RefNannyAPIStruct *__Pyx_RefNannyImportAPI(const char *modname); /*proto*/
|
|
#define __Pyx_RefNannyDeclarations void *__pyx_refnanny = NULL;
|
|
#ifdef WITH_THREAD
|
|
#define __Pyx_RefNannySetupContext(name, acquire_gil) \
|
|
if (acquire_gil) { \
|
|
PyGILState_STATE __pyx_gilstate_save = PyGILState_Ensure(); \
|
|
__pyx_refnanny = __Pyx_RefNanny->SetupContext((name), __LINE__, __FILE__); \
|
|
PyGILState_Release(__pyx_gilstate_save); \
|
|
} else { \
|
|
__pyx_refnanny = __Pyx_RefNanny->SetupContext((name), __LINE__, __FILE__); \
|
|
}
|
|
#else
|
|
#define __Pyx_RefNannySetupContext(name, acquire_gil) \
|
|
__pyx_refnanny = __Pyx_RefNanny->SetupContext((name), __LINE__, __FILE__)
|
|
#endif
|
|
#define __Pyx_RefNannyFinishContext() \
|
|
__Pyx_RefNanny->FinishContext(&__pyx_refnanny)
|
|
#define __Pyx_INCREF(r) __Pyx_RefNanny->INCREF(__pyx_refnanny, (PyObject *)(r), __LINE__)
|
|
#define __Pyx_DECREF(r) __Pyx_RefNanny->DECREF(__pyx_refnanny, (PyObject *)(r), __LINE__)
|
|
#define __Pyx_GOTREF(r) __Pyx_RefNanny->GOTREF(__pyx_refnanny, (PyObject *)(r), __LINE__)
|
|
#define __Pyx_GIVEREF(r) __Pyx_RefNanny->GIVEREF(__pyx_refnanny, (PyObject *)(r), __LINE__)
|
|
#define __Pyx_XINCREF(r) do { if((r) != NULL) {__Pyx_INCREF(r); }} while(0)
|
|
#define __Pyx_XDECREF(r) do { if((r) != NULL) {__Pyx_DECREF(r); }} while(0)
|
|
#define __Pyx_XGOTREF(r) do { if((r) != NULL) {__Pyx_GOTREF(r); }} while(0)
|
|
#define __Pyx_XGIVEREF(r) do { if((r) != NULL) {__Pyx_GIVEREF(r);}} while(0)
|
|
#else
|
|
#define __Pyx_RefNannyDeclarations
|
|
#define __Pyx_RefNannySetupContext(name, acquire_gil)
|
|
#define __Pyx_RefNannyFinishContext()
|
|
#define __Pyx_INCREF(r) Py_INCREF(r)
|
|
#define __Pyx_DECREF(r) Py_DECREF(r)
|
|
#define __Pyx_GOTREF(r)
|
|
#define __Pyx_GIVEREF(r)
|
|
#define __Pyx_XINCREF(r) Py_XINCREF(r)
|
|
#define __Pyx_XDECREF(r) Py_XDECREF(r)
|
|
#define __Pyx_XGOTREF(r)
|
|
#define __Pyx_XGIVEREF(r)
|
|
#endif /* CYTHON_REFNANNY */
|
|
#define __Pyx_CLEAR(r) do { PyObject* tmp = ((PyObject*)(r)); r = NULL; __Pyx_DECREF(tmp);} while(0)
|
|
#define __Pyx_XCLEAR(r) do { if((r) != NULL) {PyObject* tmp = ((PyObject*)(r)); r = NULL; __Pyx_DECREF(tmp);}} while(0)
|
|
|
|
static PyObject *__Pyx_GetName(PyObject *dict, PyObject *name); /*proto*/
|
|
|
|
static void __Pyx_RaiseDoubleKeywordsError(const char* func_name, PyObject* kw_name); /*proto*/
|
|
|
|
static int __Pyx_ParseOptionalKeywords(PyObject *kwds, PyObject **argnames[], \
|
|
PyObject *kwds2, PyObject *values[], Py_ssize_t num_pos_args, \
|
|
const char* function_name); /*proto*/
|
|
|
|
static void __Pyx_RaiseArgtupleInvalid(const char* func_name, int exact,
|
|
Py_ssize_t num_min, Py_ssize_t num_max, Py_ssize_t num_found); /*proto*/
|
|
|
|
static CYTHON_INLINE void __Pyx_ErrRestore(PyObject *type, PyObject *value, PyObject *tb); /*proto*/
|
|
static CYTHON_INLINE void __Pyx_ErrFetch(PyObject **type, PyObject **value, PyObject **tb); /*proto*/
|
|
|
|
static void __Pyx_Raise(PyObject *type, PyObject *value, PyObject *tb, PyObject *cause); /*proto*/
|
|
|
|
static CYTHON_INLINE PyObject *__Pyx_GetItemInt_Generic(PyObject *o, PyObject* j) {
|
|
PyObject *r;
|
|
if (!j) return NULL;
|
|
r = PyObject_GetItem(o, j);
|
|
Py_DECREF(j);
|
|
return r;
|
|
}
|
|
#define __Pyx_GetItemInt_List(o, i, size, to_py_func) (((size) <= sizeof(Py_ssize_t)) ? \
|
|
__Pyx_GetItemInt_List_Fast(o, i) : \
|
|
__Pyx_GetItemInt_Generic(o, to_py_func(i)))
|
|
static CYTHON_INLINE PyObject *__Pyx_GetItemInt_List_Fast(PyObject *o, Py_ssize_t i) {
|
|
if (likely(o != Py_None)) {
|
|
if (likely((0 <= i) & (i < PyList_GET_SIZE(o)))) {
|
|
PyObject *r = PyList_GET_ITEM(o, i);
|
|
Py_INCREF(r);
|
|
return r;
|
|
}
|
|
else if ((-PyList_GET_SIZE(o) <= i) & (i < 0)) {
|
|
PyObject *r = PyList_GET_ITEM(o, PyList_GET_SIZE(o) + i);
|
|
Py_INCREF(r);
|
|
return r;
|
|
}
|
|
}
|
|
return __Pyx_GetItemInt_Generic(o, PyInt_FromSsize_t(i));
|
|
}
|
|
#define __Pyx_GetItemInt_Tuple(o, i, size, to_py_func) (((size) <= sizeof(Py_ssize_t)) ? \
|
|
__Pyx_GetItemInt_Tuple_Fast(o, i) : \
|
|
__Pyx_GetItemInt_Generic(o, to_py_func(i)))
|
|
static CYTHON_INLINE PyObject *__Pyx_GetItemInt_Tuple_Fast(PyObject *o, Py_ssize_t i) {
|
|
if (likely(o != Py_None)) {
|
|
if (likely((0 <= i) & (i < PyTuple_GET_SIZE(o)))) {
|
|
PyObject *r = PyTuple_GET_ITEM(o, i);
|
|
Py_INCREF(r);
|
|
return r;
|
|
}
|
|
else if ((-PyTuple_GET_SIZE(o) <= i) & (i < 0)) {
|
|
PyObject *r = PyTuple_GET_ITEM(o, PyTuple_GET_SIZE(o) + i);
|
|
Py_INCREF(r);
|
|
return r;
|
|
}
|
|
}
|
|
return __Pyx_GetItemInt_Generic(o, PyInt_FromSsize_t(i));
|
|
}
|
|
#define __Pyx_GetItemInt(o, i, size, to_py_func) (((size) <= sizeof(Py_ssize_t)) ? \
|
|
__Pyx_GetItemInt_Fast(o, i) : \
|
|
__Pyx_GetItemInt_Generic(o, to_py_func(i)))
|
|
static CYTHON_INLINE PyObject *__Pyx_GetItemInt_Fast(PyObject *o, Py_ssize_t i) {
|
|
if (PyList_CheckExact(o)) {
|
|
Py_ssize_t n = (likely(i >= 0)) ? i : i + PyList_GET_SIZE(o);
|
|
if (likely((n >= 0) & (n < PyList_GET_SIZE(o)))) {
|
|
PyObject *r = PyList_GET_ITEM(o, n);
|
|
Py_INCREF(r);
|
|
return r;
|
|
}
|
|
}
|
|
else if (PyTuple_CheckExact(o)) {
|
|
Py_ssize_t n = (likely(i >= 0)) ? i : i + PyTuple_GET_SIZE(o);
|
|
if (likely((n >= 0) & (n < PyTuple_GET_SIZE(o)))) {
|
|
PyObject *r = PyTuple_GET_ITEM(o, n);
|
|
Py_INCREF(r);
|
|
return r;
|
|
}
|
|
}
|
|
else if (likely(i >= 0)) {
|
|
PySequenceMethods *m = Py_TYPE(o)->tp_as_sequence;
|
|
if (likely(m && m->sq_item)) {
|
|
return m->sq_item(o, i);
|
|
}
|
|
}
|
|
return __Pyx_GetItemInt_Generic(o, PyInt_FromSsize_t(i));
|
|
}
|
|
|
|
static CYTHON_INLINE int __Pyx_TypeTest(PyObject *obj, PyTypeObject *type); /*proto*/
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseNeedMoreValuesError(Py_ssize_t index);
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseTooManyValuesError(Py_ssize_t expected);
|
|
|
|
static int __Pyx_IternextUnpackEndCheck(PyObject *retval, Py_ssize_t expected); /*proto*/
|
|
|
|
static CYTHON_INLINE int __Pyx_GetBufferAndValidate(Py_buffer* buf, PyObject* obj,
|
|
__Pyx_TypeInfo* dtype, int flags, int nd, int cast, __Pyx_BufFmt_StackElem* stack);
|
|
static CYTHON_INLINE void __Pyx_SafeReleaseBuffer(Py_buffer* info);
|
|
|
|
static void __Pyx_RaiseBufferIndexError(int axis); /*proto*/
|
|
|
|
#define __Pyx_BufPtrCContig1d(type, buf, i0, s0) ((type)buf + i0)
|
|
static void __Pyx_RaiseBufferFallbackError(void); /*proto*/
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseNoneNotIterableError(void);
|
|
|
|
static void __Pyx_UnpackTupleError(PyObject *, Py_ssize_t index); /*proto*/
|
|
|
|
typedef struct {
|
|
Py_ssize_t shape, strides, suboffsets;
|
|
} __Pyx_Buf_DimInfo;
|
|
typedef struct {
|
|
size_t refcount;
|
|
Py_buffer pybuffer;
|
|
} __Pyx_Buffer;
|
|
typedef struct {
|
|
__Pyx_Buffer *rcbuffer;
|
|
char *data;
|
|
__Pyx_Buf_DimInfo diminfo[8];
|
|
} __Pyx_LocalBuf_ND;
|
|
|
|
#if PY_MAJOR_VERSION < 3
|
|
static int __Pyx_GetBuffer(PyObject *obj, Py_buffer *view, int flags);
|
|
static void __Pyx_ReleaseBuffer(Py_buffer *view);
|
|
#else
|
|
#define __Pyx_GetBuffer PyObject_GetBuffer
|
|
#define __Pyx_ReleaseBuffer PyBuffer_Release
|
|
#endif
|
|
|
|
static Py_ssize_t __Pyx_zeros[] = {0, 0, 0, 0, 0, 0, 0, 0};
|
|
static Py_ssize_t __Pyx_minusones[] = {-1, -1, -1, -1, -1, -1, -1, -1};
|
|
|
|
static PyObject *__Pyx_Import(PyObject *name, PyObject *from_list, long level); /*proto*/
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseImportError(PyObject *name);
|
|
|
|
#include <string.h>
|
|
|
|
static CYTHON_INLINE int __Pyx_PyBytes_Equals(PyObject* s1, PyObject* s2, int equals); /*proto*/
|
|
|
|
static CYTHON_INLINE int __Pyx_PyUnicode_Equals(PyObject* s1, PyObject* s2, int equals); /*proto*/
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
#define __Pyx_PyString_Equals __Pyx_PyUnicode_Equals
|
|
#else
|
|
#define __Pyx_PyString_Equals __Pyx_PyBytes_Equals
|
|
#endif
|
|
|
|
static CYTHON_INLINE PyObject *__Pyx_PyInt_to_py_npy_int32(npy_int32);
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#ifdef __cplusplus
|
|
#define __Pyx_CREAL(z) ((z).real())
|
|
#define __Pyx_CIMAG(z) ((z).imag())
|
|
#else
|
|
#define __Pyx_CREAL(z) (__real__(z))
|
|
#define __Pyx_CIMAG(z) (__imag__(z))
|
|
#endif
|
|
#else
|
|
#define __Pyx_CREAL(z) ((z).real)
|
|
#define __Pyx_CIMAG(z) ((z).imag)
|
|
#endif
|
|
#if defined(_WIN32) && defined(__cplusplus) && CYTHON_CCOMPLEX
|
|
#define __Pyx_SET_CREAL(z,x) ((z).real(x))
|
|
#define __Pyx_SET_CIMAG(z,y) ((z).imag(y))
|
|
#else
|
|
#define __Pyx_SET_CREAL(z,x) __Pyx_CREAL(z) = (x)
|
|
#define __Pyx_SET_CIMAG(z,y) __Pyx_CIMAG(z) = (y)
|
|
#endif
|
|
|
|
static CYTHON_INLINE __pyx_t_float_complex __pyx_t_float_complex_from_parts(float, float);
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#define __Pyx_c_eqf(a, b) ((a)==(b))
|
|
#define __Pyx_c_sumf(a, b) ((a)+(b))
|
|
#define __Pyx_c_difff(a, b) ((a)-(b))
|
|
#define __Pyx_c_prodf(a, b) ((a)*(b))
|
|
#define __Pyx_c_quotf(a, b) ((a)/(b))
|
|
#define __Pyx_c_negf(a) (-(a))
|
|
#ifdef __cplusplus
|
|
#define __Pyx_c_is_zerof(z) ((z)==(float)0)
|
|
#define __Pyx_c_conjf(z) (::std::conj(z))
|
|
#if 1
|
|
#define __Pyx_c_absf(z) (::std::abs(z))
|
|
#define __Pyx_c_powf(a, b) (::std::pow(a, b))
|
|
#endif
|
|
#else
|
|
#define __Pyx_c_is_zerof(z) ((z)==0)
|
|
#define __Pyx_c_conjf(z) (conjf(z))
|
|
#if 1
|
|
#define __Pyx_c_absf(z) (cabsf(z))
|
|
#define __Pyx_c_powf(a, b) (cpowf(a, b))
|
|
#endif
|
|
#endif
|
|
#else
|
|
static CYTHON_INLINE int __Pyx_c_eqf(__pyx_t_float_complex, __pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_sumf(__pyx_t_float_complex, __pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_difff(__pyx_t_float_complex, __pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_prodf(__pyx_t_float_complex, __pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_quotf(__pyx_t_float_complex, __pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_negf(__pyx_t_float_complex);
|
|
static CYTHON_INLINE int __Pyx_c_is_zerof(__pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_conjf(__pyx_t_float_complex);
|
|
#if 1
|
|
static CYTHON_INLINE float __Pyx_c_absf(__pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_powf(__pyx_t_float_complex, __pyx_t_float_complex);
|
|
#endif
|
|
#endif
|
|
|
|
static CYTHON_INLINE __pyx_t_double_complex __pyx_t_double_complex_from_parts(double, double);
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#define __Pyx_c_eq(a, b) ((a)==(b))
|
|
#define __Pyx_c_sum(a, b) ((a)+(b))
|
|
#define __Pyx_c_diff(a, b) ((a)-(b))
|
|
#define __Pyx_c_prod(a, b) ((a)*(b))
|
|
#define __Pyx_c_quot(a, b) ((a)/(b))
|
|
#define __Pyx_c_neg(a) (-(a))
|
|
#ifdef __cplusplus
|
|
#define __Pyx_c_is_zero(z) ((z)==(double)0)
|
|
#define __Pyx_c_conj(z) (::std::conj(z))
|
|
#if 1
|
|
#define __Pyx_c_abs(z) (::std::abs(z))
|
|
#define __Pyx_c_pow(a, b) (::std::pow(a, b))
|
|
#endif
|
|
#else
|
|
#define __Pyx_c_is_zero(z) ((z)==0)
|
|
#define __Pyx_c_conj(z) (conj(z))
|
|
#if 1
|
|
#define __Pyx_c_abs(z) (cabs(z))
|
|
#define __Pyx_c_pow(a, b) (cpow(a, b))
|
|
#endif
|
|
#endif
|
|
#else
|
|
static CYTHON_INLINE int __Pyx_c_eq(__pyx_t_double_complex, __pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_sum(__pyx_t_double_complex, __pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_diff(__pyx_t_double_complex, __pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_prod(__pyx_t_double_complex, __pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_quot(__pyx_t_double_complex, __pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_neg(__pyx_t_double_complex);
|
|
static CYTHON_INLINE int __Pyx_c_is_zero(__pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_conj(__pyx_t_double_complex);
|
|
#if 1
|
|
static CYTHON_INLINE double __Pyx_c_abs(__pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_pow(__pyx_t_double_complex, __pyx_t_double_complex);
|
|
#endif
|
|
#endif
|
|
|
|
static CYTHON_INLINE unsigned char __Pyx_PyInt_AsUnsignedChar(PyObject *);
|
|
|
|
static CYTHON_INLINE unsigned short __Pyx_PyInt_AsUnsignedShort(PyObject *);
|
|
|
|
static CYTHON_INLINE unsigned int __Pyx_PyInt_AsUnsignedInt(PyObject *);
|
|
|
|
static CYTHON_INLINE char __Pyx_PyInt_AsChar(PyObject *);
|
|
|
|
static CYTHON_INLINE short __Pyx_PyInt_AsShort(PyObject *);
|
|
|
|
static CYTHON_INLINE int __Pyx_PyInt_AsInt(PyObject *);
|
|
|
|
static CYTHON_INLINE signed char __Pyx_PyInt_AsSignedChar(PyObject *);
|
|
|
|
static CYTHON_INLINE signed short __Pyx_PyInt_AsSignedShort(PyObject *);
|
|
|
|
static CYTHON_INLINE signed int __Pyx_PyInt_AsSignedInt(PyObject *);
|
|
|
|
static CYTHON_INLINE int __Pyx_PyInt_AsLongDouble(PyObject *);
|
|
|
|
static CYTHON_INLINE unsigned long __Pyx_PyInt_AsUnsignedLong(PyObject *);
|
|
|
|
static CYTHON_INLINE unsigned PY_LONG_LONG __Pyx_PyInt_AsUnsignedLongLong(PyObject *);
|
|
|
|
static CYTHON_INLINE long __Pyx_PyInt_AsLong(PyObject *);
|
|
|
|
static CYTHON_INLINE PY_LONG_LONG __Pyx_PyInt_AsLongLong(PyObject *);
|
|
|
|
static CYTHON_INLINE signed long __Pyx_PyInt_AsSignedLong(PyObject *);
|
|
|
|
static CYTHON_INLINE signed PY_LONG_LONG __Pyx_PyInt_AsSignedLongLong(PyObject *);
|
|
|
|
static void __Pyx_WriteUnraisable(const char *name, int clineno,
|
|
int lineno, const char *filename); /*proto*/
|
|
|
|
static int __Pyx_check_binary_version(void);
|
|
|
|
#if !defined(__Pyx_PyIdentifier_FromString)
|
|
#if PY_MAJOR_VERSION < 3
|
|
#define __Pyx_PyIdentifier_FromString(s) PyString_FromString(s)
|
|
#else
|
|
#define __Pyx_PyIdentifier_FromString(s) PyUnicode_FromString(s)
|
|
#endif
|
|
#endif
|
|
|
|
static PyTypeObject *__Pyx_ImportType(const char *module_name, const char *class_name, size_t size, int strict); /*proto*/
|
|
|
|
static PyObject *__Pyx_ImportModule(const char *name); /*proto*/
|
|
|
|
typedef struct {
|
|
int code_line;
|
|
PyCodeObject* code_object;
|
|
} __Pyx_CodeObjectCacheEntry;
|
|
struct __Pyx_CodeObjectCache {
|
|
int count;
|
|
int max_count;
|
|
__Pyx_CodeObjectCacheEntry* entries;
|
|
};
|
|
static struct __Pyx_CodeObjectCache __pyx_code_cache = {0,0,NULL};
|
|
static int __pyx_bisect_code_objects(__Pyx_CodeObjectCacheEntry* entries, int count, int code_line);
|
|
static PyCodeObject *__pyx_find_code_object(int code_line);
|
|
static void __pyx_insert_code_object(int code_line, PyCodeObject* code_object);
|
|
|
|
static void __Pyx_AddTraceback(const char *funcname, int c_line,
|
|
int py_line, const char *filename); /*proto*/
|
|
|
|
static int __Pyx_InitStrings(__Pyx_StringTabEntry *t); /*proto*/
|
|
|
|
|
|
/* Module declarations from 'cpython.buffer' */
|
|
|
|
/* Module declarations from 'cpython.ref' */
|
|
|
|
/* Module declarations from 'libc.stdio' */
|
|
|
|
/* Module declarations from 'cpython.object' */
|
|
|
|
/* Module declarations from 'libc.stdlib' */
|
|
|
|
/* Module declarations from 'numpy' */
|
|
|
|
/* Module declarations from 'numpy' */
|
|
static PyTypeObject *__pyx_ptype_5numpy_dtype = 0;
|
|
static PyTypeObject *__pyx_ptype_5numpy_flatiter = 0;
|
|
static PyTypeObject *__pyx_ptype_5numpy_broadcast = 0;
|
|
static PyTypeObject *__pyx_ptype_5numpy_ndarray = 0;
|
|
static PyTypeObject *__pyx_ptype_5numpy_ufunc = 0;
|
|
static CYTHON_INLINE char *__pyx_f_5numpy__util_dtypestring(PyArray_Descr *, char *, char *, int *); /*proto*/
|
|
|
|
/* Module declarations from 'cython' */
|
|
|
|
/* Module declarations from 'libc' */
|
|
|
|
/* Module declarations from 'sklearn.utils.mst._traversal' */
|
|
static __pyx_t_7sklearn_5utils_3mst_10_traversal_DTYPE_t __pyx_v_7sklearn_5utils_3mst_10_traversal_DTYPE_EPS;
|
|
static __pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t __pyx_v_7sklearn_5utils_3mst_10_traversal_NULL_IDX;
|
|
static unsigned int __pyx_f_7sklearn_5utils_3mst_10_traversal__breadth_first_directed(unsigned int, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *); /*proto*/
|
|
static unsigned int __pyx_f_7sklearn_5utils_3mst_10_traversal__breadth_first_undirected(unsigned int, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *); /*proto*/
|
|
static unsigned int __pyx_f_7sklearn_5utils_3mst_10_traversal__depth_first_directed(unsigned int, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *); /*proto*/
|
|
static unsigned int __pyx_f_7sklearn_5utils_3mst_10_traversal__depth_first_undirected(unsigned int, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *); /*proto*/
|
|
static int __pyx_f_7sklearn_5utils_3mst_10_traversal__connected_components_directed(PyArrayObject *, PyArrayObject *, PyArrayObject *); /*proto*/
|
|
static int __pyx_f_7sklearn_5utils_3mst_10_traversal__connected_components_undirected(PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *, PyArrayObject *); /*proto*/
|
|
static __Pyx_TypeInfo __Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t = { "ITYPE_t", NULL, sizeof(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t), { 0 }, 0, 'I', IS_UNSIGNED(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t), 0 };
|
|
#define __Pyx_MODULE_NAME "sklearn.utils.mst._traversal"
|
|
int __pyx_module_is_main_sklearn__utils__mst___traversal = 0;
|
|
|
|
/* Implementation of 'sklearn.utils.mst._traversal' */
|
|
static PyObject *__pyx_builtin_ValueError;
|
|
static PyObject *__pyx_builtin_range;
|
|
static PyObject *__pyx_builtin_RuntimeError;
|
|
static PyObject *__pyx_pf_7sklearn_5utils_3mst_10_traversal_connected_components(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csgraph, PyObject *__pyx_v_directed, PyObject *__pyx_v_connection, PyObject *__pyx_v_return_labels); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_5utils_3mst_10_traversal_2breadth_first_tree(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csgraph, PyObject *__pyx_v_i_start, PyObject *__pyx_v_directed); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_5utils_3mst_10_traversal_4depth_first_tree(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csgraph, PyObject *__pyx_v_i_start, PyObject *__pyx_v_directed); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_5utils_3mst_10_traversal_6breadth_first_order(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csgraph, PyObject *__pyx_v_i_start, PyObject *__pyx_v_directed, PyObject *__pyx_v_return_predecessors); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_5utils_3mst_10_traversal_8depth_first_order(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csgraph, PyObject *__pyx_v_i_start, PyObject *__pyx_v_directed, PyObject *__pyx_v_return_predecessors); /* proto */
|
|
static int __pyx_pf_5numpy_7ndarray___getbuffer__(PyArrayObject *__pyx_v_self, Py_buffer *__pyx_v_info, int __pyx_v_flags); /* proto */
|
|
static void __pyx_pf_5numpy_7ndarray_2__releasebuffer__(PyArrayObject *__pyx_v_self, Py_buffer *__pyx_v_info); /* proto */
|
|
static char __pyx_k_3[] = "connection must be 'weak' or 'strong'";
|
|
static char __pyx_k_11[] = "ndarray is not C contiguous";
|
|
static char __pyx_k_13[] = "ndarray is not Fortran contiguous";
|
|
static char __pyx_k_15[] = "Non-native byte order not supported";
|
|
static char __pyx_k_17[] = "unknown dtype code in numpy.pxd (%d)";
|
|
static char __pyx_k_18[] = "Format string allocated too short, see comment in numpy.pxd";
|
|
static char __pyx_k_21[] = "Format string allocated too short.";
|
|
static char __pyx_k_23[] = "\nRoutines for traversing graphs in compressed sparse format\n";
|
|
static char __pyx_k_24[] = "scipy.sparse";
|
|
static char __pyx_k_27[] = "connected_components";
|
|
static char __pyx_k_28[] = "/home/x006/Code/scikit-learn/sklearn/utils/mst/_traversal.pyx";
|
|
static char __pyx_k_29[] = "sklearn.utils.mst._traversal";
|
|
static char __pyx_k_38[] = "breadth_first_tree (line 99)";
|
|
static char __pyx_k_39[] = "\n breadth_first_tree(csgraph, i_start, directed=True)\n\n Return the tree generated by a breadth-first search\n\n Note that a breadth-first tree from a specified node is unique.\n\n .. versionadded:: 0.11.0\n\n Parameters\n ----------\n csgraph : array_like or sparse matrix\n The N x N matrix representing the compressed sparse graph. The input\n csgraph will be converted to csr format for the calculation.\n i_start : int\n The index of starting node.\n directed : bool, optional\n If True (default), then operate on a directed graph: only\n move from point i to point j along paths csgraph[i, j].\n If False, then find the shortest path on an undirected graph: the\n algorithm can progress from point i to j along csgraph[i, j] or\n csgraph[j, i].\n\n Returns\n -------\n cstree : csr matrix\n The N x N directed compressed-sparse representation of the breadth-\n first tree drawn from csgraph, starting at the specified node.\n\n Examples\n --------\n The following example shows the computation of a depth-first tree\n over a simple four-component graph, starting at node 0::\n\n input graph breadth first tree from (0)\n\n (0) (0)\n / \\ / \\\n 3 8 3 8\n / \\ / \\\n (3)---5---(1) (3) (1)\n \\ / /\n 6 2 2\n \\ / /\n (2) (2)\n\n In compressed sparse representation, the solution looks like this:\n\n >>> from scipy.sparse import csr_matrix\n >>> from scipy.sparse.csgraph import breadth_first_tree\n >>> X = csr_matrix([[0, 8, 0, 3],\n ... [0, 0, 2, 5],\n ... [0, 0, 0, 6],""\n ... [0, 0, 0, 0]])\n >>> Tcsr = breadth_first_tree(X, 0, directed=False)\n >>> Tcsr.toarray().astype(int)\n array([[0, 8, 0, 3],\n [0, 0, 2, 0],\n [0, 0, 0, 0],\n [0, 0, 0, 0]])\n\n Note that the resulting graph is a Directed Acyclic Graph which spans\n the graph. A breadth-first tree from a given node is unique.\n ";
|
|
static char __pyx_k_40[] = "depth_first_tree (line 169)";
|
|
static char __pyx_k_41[] = "\n depth_first_tree(csgraph, i_start, directed=True)\n\n Return a tree generated by a depth-first search.\n\n Note that a tree generated by a depth-first search is not unique:\n it depends on the order that the children of each node are searched.\n\n .. versionadded:: 0.11.0\n\n Parameters\n ----------\n csgraph : array_like or sparse matrix\n The N x N matrix representing the compressed sparse graph. The input\n csgraph will be converted to csr format for the calculation.\n i_start : int\n The index of starting node.\n directed : bool, optional\n If True (default), then operate on a directed graph: only\n move from point i to point j along paths csgraph[i, j].\n If False, then find the shortest path on an undirected graph: the\n algorithm can progress from point i to j along csgraph[i, j] or\n csgraph[j, i].\n\n Returns\n -------\n cstree : csr matrix\n The N x N directed compressed-sparse representation of the depth-\n first tree drawn from csgraph, starting at the specified node.\n\n Examples\n --------\n The following example shows the computation of a depth-first tree\n over a simple four-component graph, starting at node 0::\n\n input graph depth first tree from (0)\n\n (0) (0)\n / \\ \\\n 3 8 8\n / \\ \\\n (3)---5---(1) (3) (1)\n \\ / \\ /\n 6 2 6 2\n \\ / \\ /\n (2) (2)\n\n In compressed sparse representation, the solution looks like this:\n\n >>> from scipy.sparse import csr_matrix\n >>> from scipy.sparse.csgraph import depth_first_tree\n >>> X = csr_matrix([[0, 8, 0, 3],\n ... "" [0, 0, 2, 5],\n ... [0, 0, 0, 6],\n ... [0, 0, 0, 0]])\n >>> Tcsr = depth_first_tree(X, 0, directed=False)\n >>> Tcsr.toarray().astype(int)\n array([[0, 8, 0, 0],\n [0, 0, 2, 0],\n [0, 0, 0, 6],\n [0, 0, 0, 0]])\n\n Note that the resulting graph is a Directed Acyclic Graph which spans\n the graph. Unlike a breadth-first tree, a depth-first tree of a given\n graph is not unique if the graph contains cycles. If the above solution\n had begun with the edge connecting nodes 0 and 3, the result would have\n been different.\n ";
|
|
static char __pyx_k__B[] = "B";
|
|
static char __pyx_k__H[] = "H";
|
|
static char __pyx_k__I[] = "I";
|
|
static char __pyx_k__L[] = "L";
|
|
static char __pyx_k__N[] = "N";
|
|
static char __pyx_k__O[] = "O";
|
|
static char __pyx_k__Q[] = "Q";
|
|
static char __pyx_k__T[] = "T";
|
|
static char __pyx_k__b[] = "b";
|
|
static char __pyx_k__d[] = "d";
|
|
static char __pyx_k__f[] = "f";
|
|
static char __pyx_k__g[] = "g";
|
|
static char __pyx_k__h[] = "h";
|
|
static char __pyx_k__i[] = "i";
|
|
static char __pyx_k__l[] = "l";
|
|
static char __pyx_k__q[] = "q";
|
|
static char __pyx_k__Zd[] = "Zd";
|
|
static char __pyx_k__Zf[] = "Zf";
|
|
static char __pyx_k__Zg[] = "Zg";
|
|
static char __pyx_k__np[] = "np";
|
|
static char __pyx_k__fill[] = "fill";
|
|
static char __pyx_k__flag[] = "flag";
|
|
static char __pyx_k__weak[] = "weak";
|
|
static char __pyx_k__DTYPE[] = "DTYPE";
|
|
static char __pyx_k__ITYPE[] = "ITYPE";
|
|
static char __pyx_k__dtype[] = "dtype";
|
|
static char __pyx_k__empty[] = "empty";
|
|
static char __pyx_k__int32[] = "int32";
|
|
static char __pyx_k__lower[] = "lower";
|
|
static char __pyx_k__numpy[] = "numpy";
|
|
static char __pyx_k__range[] = "range";
|
|
static char __pyx_k__shape[] = "shape";
|
|
static char __pyx_k__tocsr[] = "tocsr";
|
|
static char __pyx_k__zeros[] = "zeros";
|
|
static char __pyx_k__indptr[] = "indptr";
|
|
static char __pyx_k__labels[] = "labels";
|
|
static char __pyx_k__length[] = "length";
|
|
static char __pyx_k__strong[] = "strong";
|
|
static char __pyx_k__csgraph[] = "csgraph";
|
|
static char __pyx_k__float64[] = "float64";
|
|
static char __pyx_k__i_start[] = "i_start";
|
|
static char __pyx_k__indices[] = "indices";
|
|
static char __pyx_k____main__[] = "__main__";
|
|
static char __pyx_k____test__[] = "__test__";
|
|
static char __pyx_k__directed[] = "directed";
|
|
static char __pyx_k__csgraph_T[] = "csgraph_T";
|
|
static char __pyx_k__node_list[] = "node_list";
|
|
static char __pyx_k__root_list[] = "root_list";
|
|
static char __pyx_k__ValueError[] = "ValueError";
|
|
static char __pyx_k__connection[] = "connection";
|
|
static char __pyx_k__csr_matrix[] = "csr_matrix";
|
|
static char __pyx_k__isspmatrix[] = "isspmatrix";
|
|
static char __pyx_k__RuntimeError[] = "RuntimeError";
|
|
static char __pyx_k___graph_tools[] = "_graph_tools";
|
|
static char __pyx_k__dense_output[] = "dense_output";
|
|
static char __pyx_k__n_components[] = "n_components";
|
|
static char __pyx_k__predecessors[] = "predecessors";
|
|
static char __pyx_k__return_labels[] = "return_labels";
|
|
static char __pyx_k__isspmatrix_csc[] = "isspmatrix_csc";
|
|
static char __pyx_k__isspmatrix_csr[] = "isspmatrix_csr";
|
|
static char __pyx_k__validate_graph[] = "validate_graph";
|
|
static char __pyx_k__depth_first_tree[] = "depth_first_tree";
|
|
static char __pyx_k__reconstruct_path[] = "reconstruct_path";
|
|
static char __pyx_k___graph_validation[] = "_graph_validation";
|
|
static char __pyx_k__depth_first_order[] = "depth_first_order";
|
|
static char __pyx_k__breadth_first_tree[] = "breadth_first_tree";
|
|
static char __pyx_k__breadth_first_order[] = "breadth_first_order";
|
|
static char __pyx_k__return_predecessors[] = "return_predecessors";
|
|
static PyObject *__pyx_kp_u_11;
|
|
static PyObject *__pyx_kp_u_13;
|
|
static PyObject *__pyx_kp_u_15;
|
|
static PyObject *__pyx_kp_u_17;
|
|
static PyObject *__pyx_kp_u_18;
|
|
static PyObject *__pyx_kp_u_21;
|
|
static PyObject *__pyx_n_s_24;
|
|
static PyObject *__pyx_n_s_27;
|
|
static PyObject *__pyx_kp_s_28;
|
|
static PyObject *__pyx_n_s_29;
|
|
static PyObject *__pyx_kp_s_3;
|
|
static PyObject *__pyx_kp_u_38;
|
|
static PyObject *__pyx_kp_u_39;
|
|
static PyObject *__pyx_kp_u_40;
|
|
static PyObject *__pyx_kp_u_41;
|
|
static PyObject *__pyx_n_s__DTYPE;
|
|
static PyObject *__pyx_n_s__ITYPE;
|
|
static PyObject *__pyx_n_s__N;
|
|
static PyObject *__pyx_n_s__RuntimeError;
|
|
static PyObject *__pyx_n_s__T;
|
|
static PyObject *__pyx_n_s__ValueError;
|
|
static PyObject *__pyx_n_s____main__;
|
|
static PyObject *__pyx_n_s____test__;
|
|
static PyObject *__pyx_n_s___graph_tools;
|
|
static PyObject *__pyx_n_s___graph_validation;
|
|
static PyObject *__pyx_n_s__breadth_first_order;
|
|
static PyObject *__pyx_n_s__breadth_first_tree;
|
|
static PyObject *__pyx_n_s__connection;
|
|
static PyObject *__pyx_n_s__csgraph;
|
|
static PyObject *__pyx_n_s__csgraph_T;
|
|
static PyObject *__pyx_n_s__csr_matrix;
|
|
static PyObject *__pyx_n_s__dense_output;
|
|
static PyObject *__pyx_n_s__depth_first_order;
|
|
static PyObject *__pyx_n_s__depth_first_tree;
|
|
static PyObject *__pyx_n_s__directed;
|
|
static PyObject *__pyx_n_s__dtype;
|
|
static PyObject *__pyx_n_s__empty;
|
|
static PyObject *__pyx_n_s__fill;
|
|
static PyObject *__pyx_n_s__flag;
|
|
static PyObject *__pyx_n_s__float64;
|
|
static PyObject *__pyx_n_s__i_start;
|
|
static PyObject *__pyx_n_s__indices;
|
|
static PyObject *__pyx_n_s__indptr;
|
|
static PyObject *__pyx_n_s__int32;
|
|
static PyObject *__pyx_n_s__isspmatrix;
|
|
static PyObject *__pyx_n_s__isspmatrix_csc;
|
|
static PyObject *__pyx_n_s__isspmatrix_csr;
|
|
static PyObject *__pyx_n_s__labels;
|
|
static PyObject *__pyx_n_s__length;
|
|
static PyObject *__pyx_n_s__lower;
|
|
static PyObject *__pyx_n_s__n_components;
|
|
static PyObject *__pyx_n_s__node_list;
|
|
static PyObject *__pyx_n_s__np;
|
|
static PyObject *__pyx_n_s__numpy;
|
|
static PyObject *__pyx_n_s__predecessors;
|
|
static PyObject *__pyx_n_s__range;
|
|
static PyObject *__pyx_n_s__reconstruct_path;
|
|
static PyObject *__pyx_n_s__return_labels;
|
|
static PyObject *__pyx_n_s__return_predecessors;
|
|
static PyObject *__pyx_n_s__root_list;
|
|
static PyObject *__pyx_n_s__shape;
|
|
static PyObject *__pyx_n_s__strong;
|
|
static PyObject *__pyx_n_s__tocsr;
|
|
static PyObject *__pyx_n_s__validate_graph;
|
|
static PyObject *__pyx_n_s__weak;
|
|
static PyObject *__pyx_n_s__zeros;
|
|
static PyObject *__pyx_int_neg_1;
|
|
static PyObject *__pyx_int_15;
|
|
static PyObject *__pyx_k_1;
|
|
static PyObject *__pyx_k_2;
|
|
static PyObject *__pyx_k_5;
|
|
static PyObject *__pyx_k_6;
|
|
static PyObject *__pyx_k_7;
|
|
static PyObject *__pyx_k_8;
|
|
static PyObject *__pyx_k_9;
|
|
static PyObject *__pyx_k_10;
|
|
static PyObject *__pyx_k_tuple_4;
|
|
static PyObject *__pyx_k_tuple_12;
|
|
static PyObject *__pyx_k_tuple_14;
|
|
static PyObject *__pyx_k_tuple_16;
|
|
static PyObject *__pyx_k_tuple_19;
|
|
static PyObject *__pyx_k_tuple_20;
|
|
static PyObject *__pyx_k_tuple_22;
|
|
static PyObject *__pyx_k_tuple_25;
|
|
static PyObject *__pyx_k_tuple_30;
|
|
static PyObject *__pyx_k_tuple_32;
|
|
static PyObject *__pyx_k_tuple_34;
|
|
static PyObject *__pyx_k_tuple_36;
|
|
static PyObject *__pyx_k_codeobj_26;
|
|
static PyObject *__pyx_k_codeobj_31;
|
|
static PyObject *__pyx_k_codeobj_33;
|
|
static PyObject *__pyx_k_codeobj_35;
|
|
static PyObject *__pyx_k_codeobj_37;
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_5utils_3mst_10_traversal_1connected_components(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static char __pyx_doc_7sklearn_5utils_3mst_10_traversal_connected_components[] = "\n connected_components(csgraph, directed=True, connection='weak',\n return_labels=True)\n\n Analyze the connected components of a sparse graph\n\n .. versionadded:: 0.11.0\n\n Parameters\n ----------\n csgraph : array_like or sparse matrix\n The N x N matrix representing the compressed sparse graph. The input\n csgraph will be converted to csr format for the calculation.\n directed : bool, optional\n If True (default), then operate on a directed graph: only\n move from point i to point j along paths csgraph[i, j].\n If False, then find the shortest path on an undirected graph: the\n algorithm can progress from point i to j along csgraph[i, j] or\n csgraph[j, i].\n connection : str, optional\n ['weak'|'strong']. For directed graphs, the type of connection to\n use. Nodes i and j are strongly connected if a path exists both\n from i to j and from j to i. Nodes i and j are weakly connected if\n only one of these paths exists. If directed == False, this keyword\n is not referenced.\n return_labels : str, optional\n If True (default), then return the labels for each of the connected\n components.\n\n Returns\n -------\n n_components: int\n The number of connected components.\n labels: ndarray\n The length-N array of labels of the connected components.\n ";
|
|
static PyMethodDef __pyx_mdef_7sklearn_5utils_3mst_10_traversal_1connected_components = {__Pyx_NAMESTR("connected_components"), (PyCFunction)__pyx_pw_7sklearn_5utils_3mst_10_traversal_1connected_components, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_5utils_3mst_10_traversal_connected_components)};
|
|
static PyObject *__pyx_pw_7sklearn_5utils_3mst_10_traversal_1connected_components(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
PyObject *__pyx_v_csgraph = 0;
|
|
PyObject *__pyx_v_directed = 0;
|
|
PyObject *__pyx_v_connection = 0;
|
|
PyObject *__pyx_v_return_labels = 0;
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__csgraph,&__pyx_n_s__directed,&__pyx_n_s__connection,&__pyx_n_s__return_labels,0};
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("connected_components (wrapper)", 0);
|
|
__pyx_self = __pyx_self;
|
|
{
|
|
PyObject* values[4] = {0,0,0,0};
|
|
values[1] = __pyx_k_1;
|
|
values[2] = ((PyObject *)__pyx_n_s__weak);
|
|
values[3] = __pyx_k_2;
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__csgraph);
|
|
if (likely(values[0])) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s__directed);
|
|
if (value) { values[1] = value; kw_args--; }
|
|
}
|
|
case 2:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s__connection);
|
|
if (value) { values[2] = value; kw_args--; }
|
|
}
|
|
case 3:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s__return_labels);
|
|
if (value) { values[3] = value; kw_args--; }
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "connected_components") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 30; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else {
|
|
switch (PyTuple_GET_SIZE(__pyx_args)) {
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
}
|
|
__pyx_v_csgraph = values[0];
|
|
__pyx_v_directed = values[1];
|
|
__pyx_v_connection = values[2];
|
|
__pyx_v_return_labels = values[3];
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("connected_components", 0, 1, 4, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 30; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.utils.mst._traversal.connected_components", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return NULL;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
__pyx_r = __pyx_pf_7sklearn_5utils_3mst_10_traversal_connected_components(__pyx_self, __pyx_v_csgraph, __pyx_v_directed, __pyx_v_connection, __pyx_v_return_labels);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":30
|
|
* cdef ITYPE_t NULL_IDX = -9999
|
|
*
|
|
* def connected_components(csgraph, directed=True, connection='weak', # <<<<<<<<<<<<<<
|
|
* return_labels=True):
|
|
* """
|
|
*/
|
|
|
|
static PyObject *__pyx_pf_7sklearn_5utils_3mst_10_traversal_connected_components(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csgraph, PyObject *__pyx_v_directed, PyObject *__pyx_v_connection, PyObject *__pyx_v_return_labels) {
|
|
PyObject *__pyx_v_labels = NULL;
|
|
int __pyx_v_n_components;
|
|
PyObject *__pyx_v_csgraph_T = NULL;
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
int __pyx_t_3;
|
|
int __pyx_t_4;
|
|
int __pyx_t_5;
|
|
PyObject *__pyx_t_6 = NULL;
|
|
PyObject *__pyx_t_7 = NULL;
|
|
PyObject *__pyx_t_8 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("connected_components", 0);
|
|
__Pyx_INCREF(__pyx_v_csgraph);
|
|
__Pyx_INCREF(__pyx_v_directed);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":68
|
|
* The length-N array of labels of the connected components.
|
|
* """
|
|
* if connection.lower() not in ['weak', 'strong']: # <<<<<<<<<<<<<<
|
|
* raise ValueError("connection must be 'weak' or 'strong'")
|
|
*
|
|
*/
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_v_connection, __pyx_n_s__lower); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 68; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_empty_tuple), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 68; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_3 = __Pyx_PyString_Equals(__pyx_t_2, ((PyObject *)__pyx_n_s__weak), Py_NE); if (unlikely(__pyx_t_3 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 68; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (((int)__pyx_t_3)) {
|
|
__pyx_t_4 = __Pyx_PyString_Equals(__pyx_t_2, ((PyObject *)__pyx_n_s__strong), Py_NE); if (unlikely(__pyx_t_4 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 68; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_5 = ((int)__pyx_t_4);
|
|
} else {
|
|
__pyx_t_5 = ((int)__pyx_t_3);
|
|
}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_3 = __pyx_t_5;
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":69
|
|
* """
|
|
* if connection.lower() not in ['weak', 'strong']:
|
|
* raise ValueError("connection must be 'weak' or 'strong'") # <<<<<<<<<<<<<<
|
|
*
|
|
* # weak connections <=> components of undirected graph
|
|
*/
|
|
__pyx_t_2 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_k_tuple_4), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 69; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_Raise(__pyx_t_2, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 69; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
goto __pyx_L3;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":72
|
|
*
|
|
* # weak connections <=> components of undirected graph
|
|
* if connection.lower() == 'weak': # <<<<<<<<<<<<<<
|
|
* directed = False
|
|
*
|
|
*/
|
|
__pyx_t_2 = PyObject_GetAttr(__pyx_v_connection, __pyx_n_s__lower); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 72; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_1 = PyObject_Call(__pyx_t_2, ((PyObject *)__pyx_empty_tuple), NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 72; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_3 = __Pyx_PyString_Equals(__pyx_t_1, ((PyObject *)__pyx_n_s__weak), Py_EQ); if (unlikely(__pyx_t_3 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 72; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":73
|
|
* # weak connections <=> components of undirected graph
|
|
* if connection.lower() == 'weak':
|
|
* directed = False # <<<<<<<<<<<<<<
|
|
*
|
|
* csgraph = validate_graph(csgraph, directed,
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyBool_FromLong(0); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 73; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_v_directed);
|
|
__pyx_v_directed = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L4;
|
|
}
|
|
__pyx_L4:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":75
|
|
* directed = False
|
|
*
|
|
* csgraph = validate_graph(csgraph, directed, # <<<<<<<<<<<<<<
|
|
* dense_output=False)
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__validate_graph); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 75; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyTuple_New(2); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 75; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_v_csgraph);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_v_csgraph);
|
|
__Pyx_GIVEREF(__pyx_v_csgraph);
|
|
__Pyx_INCREF(__pyx_v_directed);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 1, __pyx_v_directed);
|
|
__Pyx_GIVEREF(__pyx_v_directed);
|
|
__pyx_t_6 = PyDict_New(); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 75; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_6));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":76
|
|
*
|
|
* csgraph = validate_graph(csgraph, directed,
|
|
* dense_output=False) # <<<<<<<<<<<<<<
|
|
*
|
|
* labels = np.empty(csgraph.shape[0], dtype=ITYPE)
|
|
*/
|
|
__pyx_t_7 = __Pyx_PyBool_FromLong(0); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 76; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
if (PyDict_SetItem(__pyx_t_6, ((PyObject *)__pyx_n_s__dense_output), __pyx_t_7) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 75; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__pyx_t_7 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_t_2), ((PyObject *)__pyx_t_6)); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 75; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_6)); __pyx_t_6 = 0;
|
|
__Pyx_DECREF(__pyx_v_csgraph);
|
|
__pyx_v_csgraph = __pyx_t_7;
|
|
__pyx_t_7 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":78
|
|
* dense_output=False)
|
|
*
|
|
* labels = np.empty(csgraph.shape[0], dtype=ITYPE) # <<<<<<<<<<<<<<
|
|
* labels.fill(NULL_IDX)
|
|
*
|
|
*/
|
|
__pyx_t_7 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 78; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__pyx_t_6 = PyObject_GetAttr(__pyx_t_7, __pyx_n_s__empty); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 78; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__pyx_t_7 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__shape); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 78; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__pyx_t_2 = __Pyx_GetItemInt(__pyx_t_7, 0, sizeof(long), PyInt_FromLong); if (!__pyx_t_2) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 78; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__pyx_t_7 = PyTuple_New(1); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 78; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
PyTuple_SET_ITEM(__pyx_t_7, 0, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = PyDict_New(); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 78; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_2));
|
|
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__ITYPE); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 78; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_t_2, ((PyObject *)__pyx_n_s__dtype), __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 78; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_Call(__pyx_t_6, ((PyObject *)__pyx_t_7), ((PyObject *)__pyx_t_2)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 78; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_7)); __pyx_t_7 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__pyx_v_labels = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":79
|
|
*
|
|
* labels = np.empty(csgraph.shape[0], dtype=ITYPE)
|
|
* labels.fill(NULL_IDX) # <<<<<<<<<<<<<<
|
|
*
|
|
* if directed:
|
|
*/
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_v_labels, __pyx_n_s__fill); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = __Pyx_PyInt_to_py_npy_int32(__pyx_v_7sklearn_5utils_3mst_10_traversal_NULL_IDX); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_7 = PyTuple_New(1); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
PyTuple_SET_ITEM(__pyx_t_7, 0, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_t_7), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_7)); __pyx_t_7 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":81
|
|
* labels.fill(NULL_IDX)
|
|
*
|
|
* if directed: # <<<<<<<<<<<<<<
|
|
* n_components = _connected_components_directed(csgraph.indices,
|
|
* csgraph.indptr,
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_IsTrue(__pyx_v_directed); if (unlikely(__pyx_t_3 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 81; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":82
|
|
*
|
|
* if directed:
|
|
* n_components = _connected_components_directed(csgraph.indices, # <<<<<<<<<<<<<<
|
|
* csgraph.indptr,
|
|
* labels)
|
|
*/
|
|
__pyx_t_2 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__indices); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 82; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 82; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":83
|
|
* if directed:
|
|
* n_components = _connected_components_directed(csgraph.indices,
|
|
* csgraph.indptr, # <<<<<<<<<<<<<<
|
|
* labels)
|
|
* else:
|
|
*/
|
|
__pyx_t_7 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__indptr); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 83; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
if (!(likely(((__pyx_t_7) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_7, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 83; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":84
|
|
* n_components = _connected_components_directed(csgraph.indices,
|
|
* csgraph.indptr,
|
|
* labels) # <<<<<<<<<<<<<<
|
|
* else:
|
|
* csgraph_T = csgraph.T.tocsr()
|
|
*/
|
|
if (!(likely(((__pyx_v_labels) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_labels, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 84; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_1 = __pyx_v_labels;
|
|
__Pyx_INCREF(__pyx_t_1);
|
|
__pyx_v_n_components = __pyx_f_7sklearn_5utils_3mst_10_traversal__connected_components_directed(((PyArrayObject *)__pyx_t_2), ((PyArrayObject *)__pyx_t_7), ((PyArrayObject *)__pyx_t_1));
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
goto __pyx_L5;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":86
|
|
* labels)
|
|
* else:
|
|
* csgraph_T = csgraph.T.tocsr() # <<<<<<<<<<<<<<
|
|
* n_components = _connected_components_undirected(csgraph.indices,
|
|
* csgraph.indptr,
|
|
*/
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__T); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 86; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_7 = PyObject_GetAttr(__pyx_t_1, __pyx_n_s__tocsr); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 86; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_Call(__pyx_t_7, ((PyObject *)__pyx_empty_tuple), NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 86; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__pyx_v_csgraph_T = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":87
|
|
* else:
|
|
* csgraph_T = csgraph.T.tocsr()
|
|
* n_components = _connected_components_undirected(csgraph.indices, # <<<<<<<<<<<<<<
|
|
* csgraph.indptr,
|
|
* csgraph_T.indices,
|
|
*/
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__indices); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 87; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 87; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":88
|
|
* csgraph_T = csgraph.T.tocsr()
|
|
* n_components = _connected_components_undirected(csgraph.indices,
|
|
* csgraph.indptr, # <<<<<<<<<<<<<<
|
|
* csgraph_T.indices,
|
|
* csgraph_T.indptr,
|
|
*/
|
|
__pyx_t_7 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__indptr); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 88; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
if (!(likely(((__pyx_t_7) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_7, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 88; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":89
|
|
* n_components = _connected_components_undirected(csgraph.indices,
|
|
* csgraph.indptr,
|
|
* csgraph_T.indices, # <<<<<<<<<<<<<<
|
|
* csgraph_T.indptr,
|
|
* labels)
|
|
*/
|
|
__pyx_t_2 = PyObject_GetAttr(__pyx_v_csgraph_T, __pyx_n_s__indices); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 89; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 89; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":90
|
|
* csgraph.indptr,
|
|
* csgraph_T.indices,
|
|
* csgraph_T.indptr, # <<<<<<<<<<<<<<
|
|
* labels)
|
|
*
|
|
*/
|
|
__pyx_t_6 = PyObject_GetAttr(__pyx_v_csgraph_T, __pyx_n_s__indptr); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 90; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
if (!(likely(((__pyx_t_6) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_6, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 90; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":91
|
|
* csgraph_T.indices,
|
|
* csgraph_T.indptr,
|
|
* labels) # <<<<<<<<<<<<<<
|
|
*
|
|
* if return_labels:
|
|
*/
|
|
if (!(likely(((__pyx_v_labels) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_labels, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 91; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_8 = __pyx_v_labels;
|
|
__Pyx_INCREF(__pyx_t_8);
|
|
__pyx_v_n_components = __pyx_f_7sklearn_5utils_3mst_10_traversal__connected_components_undirected(((PyArrayObject *)__pyx_t_1), ((PyArrayObject *)__pyx_t_7), ((PyArrayObject *)__pyx_t_2), ((PyArrayObject *)__pyx_t_6), ((PyArrayObject *)__pyx_t_8));
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
__Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
|
|
}
|
|
__pyx_L5:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":93
|
|
* labels)
|
|
*
|
|
* if return_labels: # <<<<<<<<<<<<<<
|
|
* return n_components, labels
|
|
* else:
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_IsTrue(__pyx_v_return_labels); if (unlikely(__pyx_t_3 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 93; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":94
|
|
*
|
|
* if return_labels:
|
|
* return n_components, labels # <<<<<<<<<<<<<<
|
|
* else:
|
|
* return n_components
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_8 = PyInt_FromLong(__pyx_v_n_components); if (unlikely(!__pyx_t_8)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 94; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_8);
|
|
__pyx_t_6 = PyTuple_New(2); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 94; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
PyTuple_SET_ITEM(__pyx_t_6, 0, __pyx_t_8);
|
|
__Pyx_GIVEREF(__pyx_t_8);
|
|
__Pyx_INCREF(__pyx_v_labels);
|
|
PyTuple_SET_ITEM(__pyx_t_6, 1, __pyx_v_labels);
|
|
__Pyx_GIVEREF(__pyx_v_labels);
|
|
__pyx_t_8 = 0;
|
|
__pyx_r = ((PyObject *)__pyx_t_6);
|
|
__pyx_t_6 = 0;
|
|
goto __pyx_L0;
|
|
goto __pyx_L6;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":96
|
|
* return n_components, labels
|
|
* else:
|
|
* return n_components # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_6 = PyInt_FromLong(__pyx_v_n_components); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 96; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__pyx_r = __pyx_t_6;
|
|
__pyx_t_6 = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
__pyx_L6:;
|
|
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_6);
|
|
__Pyx_XDECREF(__pyx_t_7);
|
|
__Pyx_XDECREF(__pyx_t_8);
|
|
__Pyx_AddTraceback("sklearn.utils.mst._traversal.connected_components", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF(__pyx_v_labels);
|
|
__Pyx_XDECREF(__pyx_v_csgraph_T);
|
|
__Pyx_XDECREF(__pyx_v_csgraph);
|
|
__Pyx_XDECREF(__pyx_v_directed);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_5utils_3mst_10_traversal_3breadth_first_tree(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static char __pyx_doc_7sklearn_5utils_3mst_10_traversal_2breadth_first_tree[] = "\n breadth_first_tree(csgraph, i_start, directed=True)\n\n Return the tree generated by a breadth-first search\n\n Note that a breadth-first tree from a specified node is unique.\n\n .. versionadded:: 0.11.0\n\n Parameters\n ----------\n csgraph : array_like or sparse matrix\n The N x N matrix representing the compressed sparse graph. The input\n csgraph will be converted to csr format for the calculation.\n i_start : int\n The index of starting node.\n directed : bool, optional\n If True (default), then operate on a directed graph: only\n move from point i to point j along paths csgraph[i, j].\n If False, then find the shortest path on an undirected graph: the\n algorithm can progress from point i to j along csgraph[i, j] or\n csgraph[j, i].\n\n Returns\n -------\n cstree : csr matrix\n The N x N directed compressed-sparse representation of the breadth-\n first tree drawn from csgraph, starting at the specified node.\n\n Examples\n --------\n The following example shows the computation of a depth-first tree\n over a simple four-component graph, starting at node 0::\n\n input graph breadth first tree from (0)\n\n (0) (0)\n / \\ / \\\n 3 8 3 8\n / \\ / \\\n (3)---5---(1) (3) (1)\n \\ / /\n 6 2 2\n \\ / /\n (2) (2)\n\n In compressed sparse representation, the solution looks like this:\n\n >>> from scipy.sparse import csr_matrix\n >>> from scipy.sparse.csgraph import breadth_first_tree\n >>> X = csr_matrix([[0, 8, 0, 3],\n ... [0, 0, 2, 5],\n ... [0, 0, 0, 6],""\n ... [0, 0, 0, 0]])\n >>> Tcsr = breadth_first_tree(X, 0, directed=False)\n >>> Tcsr.toarray().astype(int)\n array([[0, 8, 0, 3],\n [0, 0, 2, 0],\n [0, 0, 0, 0],\n [0, 0, 0, 0]])\n\n Note that the resulting graph is a Directed Acyclic Graph which spans\n the graph. A breadth-first tree from a given node is unique.\n ";
|
|
static PyMethodDef __pyx_mdef_7sklearn_5utils_3mst_10_traversal_3breadth_first_tree = {__Pyx_NAMESTR("breadth_first_tree"), (PyCFunction)__pyx_pw_7sklearn_5utils_3mst_10_traversal_3breadth_first_tree, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_5utils_3mst_10_traversal_2breadth_first_tree)};
|
|
static PyObject *__pyx_pw_7sklearn_5utils_3mst_10_traversal_3breadth_first_tree(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
PyObject *__pyx_v_csgraph = 0;
|
|
PyObject *__pyx_v_i_start = 0;
|
|
PyObject *__pyx_v_directed = 0;
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__csgraph,&__pyx_n_s__i_start,&__pyx_n_s__directed,0};
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("breadth_first_tree (wrapper)", 0);
|
|
__pyx_self = __pyx_self;
|
|
{
|
|
PyObject* values[3] = {0,0,0};
|
|
values[2] = __pyx_k_5;
|
|
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);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__csgraph);
|
|
if (likely(values[0])) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__i_start);
|
|
if (likely(values[1])) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("breadth_first_tree", 0, 2, 3, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 99; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s__directed);
|
|
if (value) { values[2] = value; kw_args--; }
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "breadth_first_tree") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 99; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else {
|
|
switch (PyTuple_GET_SIZE(__pyx_args)) {
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
}
|
|
__pyx_v_csgraph = values[0];
|
|
__pyx_v_i_start = values[1];
|
|
__pyx_v_directed = values[2];
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("breadth_first_tree", 0, 2, 3, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 99; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.utils.mst._traversal.breadth_first_tree", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return NULL;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
__pyx_r = __pyx_pf_7sklearn_5utils_3mst_10_traversal_2breadth_first_tree(__pyx_self, __pyx_v_csgraph, __pyx_v_i_start, __pyx_v_directed);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":99
|
|
*
|
|
*
|
|
* def breadth_first_tree(csgraph, i_start, directed=True): # <<<<<<<<<<<<<<
|
|
* r"""
|
|
* breadth_first_tree(csgraph, i_start, directed=True)
|
|
*/
|
|
|
|
static PyObject *__pyx_pf_7sklearn_5utils_3mst_10_traversal_2breadth_first_tree(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csgraph, PyObject *__pyx_v_i_start, PyObject *__pyx_v_directed) {
|
|
CYTHON_UNUSED PyObject *__pyx_v_node_list = NULL;
|
|
PyObject *__pyx_v_predecessors = NULL;
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
PyObject *(*__pyx_t_5)(PyObject *);
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("breadth_first_tree", 0);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":164
|
|
* the graph. A breadth-first tree from a given node is unique.
|
|
* """
|
|
* node_list, predecessors = breadth_first_order(csgraph, i_start, # <<<<<<<<<<<<<<
|
|
* directed, True)
|
|
* return reconstruct_path(csgraph, predecessors, directed)
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__breadth_first_order); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 164; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":165
|
|
* """
|
|
* node_list, predecessors = breadth_first_order(csgraph, i_start,
|
|
* directed, True) # <<<<<<<<<<<<<<
|
|
* return reconstruct_path(csgraph, predecessors, directed)
|
|
*
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyBool_FromLong(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 165; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_3 = PyTuple_New(4); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 164; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_v_csgraph);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_v_csgraph);
|
|
__Pyx_GIVEREF(__pyx_v_csgraph);
|
|
__Pyx_INCREF(__pyx_v_i_start);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 1, __pyx_v_i_start);
|
|
__Pyx_GIVEREF(__pyx_v_i_start);
|
|
__Pyx_INCREF(__pyx_v_directed);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 2, __pyx_v_directed);
|
|
__Pyx_GIVEREF(__pyx_v_directed);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 3, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_t_3), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 164; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 0;
|
|
if ((likely(PyTuple_CheckExact(__pyx_t_2))) || (PyList_CheckExact(__pyx_t_2))) {
|
|
PyObject* sequence = __pyx_t_2;
|
|
if (likely(PyTuple_CheckExact(sequence))) {
|
|
if (unlikely(PyTuple_GET_SIZE(sequence) != 2)) {
|
|
if (PyTuple_GET_SIZE(sequence) > 2) __Pyx_RaiseTooManyValuesError(2);
|
|
else __Pyx_RaiseNeedMoreValuesError(PyTuple_GET_SIZE(sequence));
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 164; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_3 = PyTuple_GET_ITEM(sequence, 0);
|
|
__pyx_t_1 = PyTuple_GET_ITEM(sequence, 1);
|
|
} else {
|
|
if (unlikely(PyList_GET_SIZE(sequence) != 2)) {
|
|
if (PyList_GET_SIZE(sequence) > 2) __Pyx_RaiseTooManyValuesError(2);
|
|
else __Pyx_RaiseNeedMoreValuesError(PyList_GET_SIZE(sequence));
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 164; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_3 = PyList_GET_ITEM(sequence, 0);
|
|
__pyx_t_1 = PyList_GET_ITEM(sequence, 1);
|
|
}
|
|
__Pyx_INCREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
} else {
|
|
Py_ssize_t index = -1;
|
|
__pyx_t_4 = PyObject_GetIter(__pyx_t_2); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 164; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_5 = Py_TYPE(__pyx_t_4)->tp_iternext;
|
|
index = 0; __pyx_t_3 = __pyx_t_5(__pyx_t_4); if (unlikely(!__pyx_t_3)) goto __pyx_L3_unpacking_failed;
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
index = 1; __pyx_t_1 = __pyx_t_5(__pyx_t_4); if (unlikely(!__pyx_t_1)) goto __pyx_L3_unpacking_failed;
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (__Pyx_IternextUnpackEndCheck(__pyx_t_5(__pyx_t_4), 2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 164; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
goto __pyx_L4_unpacking_done;
|
|
__pyx_L3_unpacking_failed:;
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
if (PyErr_Occurred() && PyErr_ExceptionMatches(PyExc_StopIteration)) PyErr_Clear();
|
|
if (!PyErr_Occurred()) __Pyx_RaiseNeedMoreValuesError(index);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 164; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_L4_unpacking_done:;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":164
|
|
* the graph. A breadth-first tree from a given node is unique.
|
|
* """
|
|
* node_list, predecessors = breadth_first_order(csgraph, i_start, # <<<<<<<<<<<<<<
|
|
* directed, True)
|
|
* return reconstruct_path(csgraph, predecessors, directed)
|
|
*/
|
|
__pyx_v_node_list = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
__pyx_v_predecessors = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":166
|
|
* node_list, predecessors = breadth_first_order(csgraph, i_start,
|
|
* directed, True)
|
|
* return reconstruct_path(csgraph, predecessors, directed) # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_2 = __Pyx_GetName(__pyx_m, __pyx_n_s__reconstruct_path); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 166; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_1 = PyTuple_New(3); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 166; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_INCREF(__pyx_v_csgraph);
|
|
PyTuple_SET_ITEM(__pyx_t_1, 0, __pyx_v_csgraph);
|
|
__Pyx_GIVEREF(__pyx_v_csgraph);
|
|
__Pyx_INCREF(__pyx_v_predecessors);
|
|
PyTuple_SET_ITEM(__pyx_t_1, 1, __pyx_v_predecessors);
|
|
__Pyx_GIVEREF(__pyx_v_predecessors);
|
|
__Pyx_INCREF(__pyx_v_directed);
|
|
PyTuple_SET_ITEM(__pyx_t_1, 2, __pyx_v_directed);
|
|
__Pyx_GIVEREF(__pyx_v_directed);
|
|
__pyx_t_3 = PyObject_Call(__pyx_t_2, ((PyObject *)__pyx_t_1), NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 166; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_1)); __pyx_t_1 = 0;
|
|
__pyx_r = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.utils.mst._traversal.breadth_first_tree", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF(__pyx_v_node_list);
|
|
__Pyx_XDECREF(__pyx_v_predecessors);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_5utils_3mst_10_traversal_5depth_first_tree(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static char __pyx_doc_7sklearn_5utils_3mst_10_traversal_4depth_first_tree[] = "\n depth_first_tree(csgraph, i_start, directed=True)\n\n Return a tree generated by a depth-first search.\n\n Note that a tree generated by a depth-first search is not unique:\n it depends on the order that the children of each node are searched.\n\n .. versionadded:: 0.11.0\n\n Parameters\n ----------\n csgraph : array_like or sparse matrix\n The N x N matrix representing the compressed sparse graph. The input\n csgraph will be converted to csr format for the calculation.\n i_start : int\n The index of starting node.\n directed : bool, optional\n If True (default), then operate on a directed graph: only\n move from point i to point j along paths csgraph[i, j].\n If False, then find the shortest path on an undirected graph: the\n algorithm can progress from point i to j along csgraph[i, j] or\n csgraph[j, i].\n\n Returns\n -------\n cstree : csr matrix\n The N x N directed compressed-sparse representation of the depth-\n first tree drawn from csgraph, starting at the specified node.\n\n Examples\n --------\n The following example shows the computation of a depth-first tree\n over a simple four-component graph, starting at node 0::\n\n input graph depth first tree from (0)\n\n (0) (0)\n / \\ \\\n 3 8 8\n / \\ \\\n (3)---5---(1) (3) (1)\n \\ / \\ /\n 6 2 6 2\n \\ / \\ /\n (2) (2)\n\n In compressed sparse representation, the solution looks like this:\n\n >>> from scipy.sparse import csr_matrix\n >>> from scipy.sparse.csgraph import depth_first_tree\n >>> X = csr_matrix([[0, 8, 0, 3],\n ... "" [0, 0, 2, 5],\n ... [0, 0, 0, 6],\n ... [0, 0, 0, 0]])\n >>> Tcsr = depth_first_tree(X, 0, directed=False)\n >>> Tcsr.toarray().astype(int)\n array([[0, 8, 0, 0],\n [0, 0, 2, 0],\n [0, 0, 0, 6],\n [0, 0, 0, 0]])\n\n Note that the resulting graph is a Directed Acyclic Graph which spans\n the graph. Unlike a breadth-first tree, a depth-first tree of a given\n graph is not unique if the graph contains cycles. If the above solution\n had begun with the edge connecting nodes 0 and 3, the result would have\n been different.\n ";
|
|
static PyMethodDef __pyx_mdef_7sklearn_5utils_3mst_10_traversal_5depth_first_tree = {__Pyx_NAMESTR("depth_first_tree"), (PyCFunction)__pyx_pw_7sklearn_5utils_3mst_10_traversal_5depth_first_tree, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_5utils_3mst_10_traversal_4depth_first_tree)};
|
|
static PyObject *__pyx_pw_7sklearn_5utils_3mst_10_traversal_5depth_first_tree(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
PyObject *__pyx_v_csgraph = 0;
|
|
PyObject *__pyx_v_i_start = 0;
|
|
PyObject *__pyx_v_directed = 0;
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__csgraph,&__pyx_n_s__i_start,&__pyx_n_s__directed,0};
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("depth_first_tree (wrapper)", 0);
|
|
__pyx_self = __pyx_self;
|
|
{
|
|
PyObject* values[3] = {0,0,0};
|
|
values[2] = __pyx_k_6;
|
|
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);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__csgraph);
|
|
if (likely(values[0])) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__i_start);
|
|
if (likely(values[1])) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("depth_first_tree", 0, 2, 3, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 169; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s__directed);
|
|
if (value) { values[2] = value; kw_args--; }
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "depth_first_tree") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 169; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else {
|
|
switch (PyTuple_GET_SIZE(__pyx_args)) {
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
}
|
|
__pyx_v_csgraph = values[0];
|
|
__pyx_v_i_start = values[1];
|
|
__pyx_v_directed = values[2];
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("depth_first_tree", 0, 2, 3, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 169; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.utils.mst._traversal.depth_first_tree", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return NULL;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
__pyx_r = __pyx_pf_7sklearn_5utils_3mst_10_traversal_4depth_first_tree(__pyx_self, __pyx_v_csgraph, __pyx_v_i_start, __pyx_v_directed);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":169
|
|
*
|
|
*
|
|
* def depth_first_tree(csgraph, i_start, directed=True): # <<<<<<<<<<<<<<
|
|
* r"""
|
|
* depth_first_tree(csgraph, i_start, directed=True)
|
|
*/
|
|
|
|
static PyObject *__pyx_pf_7sklearn_5utils_3mst_10_traversal_4depth_first_tree(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csgraph, PyObject *__pyx_v_i_start, PyObject *__pyx_v_directed) {
|
|
CYTHON_UNUSED PyObject *__pyx_v_node_list = NULL;
|
|
PyObject *__pyx_v_predecessors = NULL;
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
PyObject *(*__pyx_t_5)(PyObject *);
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("depth_first_tree", 0);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":238
|
|
* been different.
|
|
* """
|
|
* node_list, predecessors = depth_first_order(csgraph, i_start, # <<<<<<<<<<<<<<
|
|
* directed, True)
|
|
* return reconstruct_path(csgraph, predecessors, directed)
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__depth_first_order); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 238; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":239
|
|
* """
|
|
* node_list, predecessors = depth_first_order(csgraph, i_start,
|
|
* directed, True) # <<<<<<<<<<<<<<
|
|
* return reconstruct_path(csgraph, predecessors, directed)
|
|
*
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyBool_FromLong(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 239; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_3 = PyTuple_New(4); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 238; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_v_csgraph);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_v_csgraph);
|
|
__Pyx_GIVEREF(__pyx_v_csgraph);
|
|
__Pyx_INCREF(__pyx_v_i_start);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 1, __pyx_v_i_start);
|
|
__Pyx_GIVEREF(__pyx_v_i_start);
|
|
__Pyx_INCREF(__pyx_v_directed);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 2, __pyx_v_directed);
|
|
__Pyx_GIVEREF(__pyx_v_directed);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 3, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_t_3), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 238; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 0;
|
|
if ((likely(PyTuple_CheckExact(__pyx_t_2))) || (PyList_CheckExact(__pyx_t_2))) {
|
|
PyObject* sequence = __pyx_t_2;
|
|
if (likely(PyTuple_CheckExact(sequence))) {
|
|
if (unlikely(PyTuple_GET_SIZE(sequence) != 2)) {
|
|
if (PyTuple_GET_SIZE(sequence) > 2) __Pyx_RaiseTooManyValuesError(2);
|
|
else __Pyx_RaiseNeedMoreValuesError(PyTuple_GET_SIZE(sequence));
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 238; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_3 = PyTuple_GET_ITEM(sequence, 0);
|
|
__pyx_t_1 = PyTuple_GET_ITEM(sequence, 1);
|
|
} else {
|
|
if (unlikely(PyList_GET_SIZE(sequence) != 2)) {
|
|
if (PyList_GET_SIZE(sequence) > 2) __Pyx_RaiseTooManyValuesError(2);
|
|
else __Pyx_RaiseNeedMoreValuesError(PyList_GET_SIZE(sequence));
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 238; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_3 = PyList_GET_ITEM(sequence, 0);
|
|
__pyx_t_1 = PyList_GET_ITEM(sequence, 1);
|
|
}
|
|
__Pyx_INCREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
} else {
|
|
Py_ssize_t index = -1;
|
|
__pyx_t_4 = PyObject_GetIter(__pyx_t_2); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 238; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_5 = Py_TYPE(__pyx_t_4)->tp_iternext;
|
|
index = 0; __pyx_t_3 = __pyx_t_5(__pyx_t_4); if (unlikely(!__pyx_t_3)) goto __pyx_L3_unpacking_failed;
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
index = 1; __pyx_t_1 = __pyx_t_5(__pyx_t_4); if (unlikely(!__pyx_t_1)) goto __pyx_L3_unpacking_failed;
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (__Pyx_IternextUnpackEndCheck(__pyx_t_5(__pyx_t_4), 2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 238; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
goto __pyx_L4_unpacking_done;
|
|
__pyx_L3_unpacking_failed:;
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
if (PyErr_Occurred() && PyErr_ExceptionMatches(PyExc_StopIteration)) PyErr_Clear();
|
|
if (!PyErr_Occurred()) __Pyx_RaiseNeedMoreValuesError(index);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 238; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_L4_unpacking_done:;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":238
|
|
* been different.
|
|
* """
|
|
* node_list, predecessors = depth_first_order(csgraph, i_start, # <<<<<<<<<<<<<<
|
|
* directed, True)
|
|
* return reconstruct_path(csgraph, predecessors, directed)
|
|
*/
|
|
__pyx_v_node_list = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
__pyx_v_predecessors = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":240
|
|
* node_list, predecessors = depth_first_order(csgraph, i_start,
|
|
* directed, True)
|
|
* return reconstruct_path(csgraph, predecessors, directed) # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_2 = __Pyx_GetName(__pyx_m, __pyx_n_s__reconstruct_path); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 240; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_1 = PyTuple_New(3); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 240; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_INCREF(__pyx_v_csgraph);
|
|
PyTuple_SET_ITEM(__pyx_t_1, 0, __pyx_v_csgraph);
|
|
__Pyx_GIVEREF(__pyx_v_csgraph);
|
|
__Pyx_INCREF(__pyx_v_predecessors);
|
|
PyTuple_SET_ITEM(__pyx_t_1, 1, __pyx_v_predecessors);
|
|
__Pyx_GIVEREF(__pyx_v_predecessors);
|
|
__Pyx_INCREF(__pyx_v_directed);
|
|
PyTuple_SET_ITEM(__pyx_t_1, 2, __pyx_v_directed);
|
|
__Pyx_GIVEREF(__pyx_v_directed);
|
|
__pyx_t_3 = PyObject_Call(__pyx_t_2, ((PyObject *)__pyx_t_1), NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 240; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_1)); __pyx_t_1 = 0;
|
|
__pyx_r = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.utils.mst._traversal.depth_first_tree", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF(__pyx_v_node_list);
|
|
__Pyx_XDECREF(__pyx_v_predecessors);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_5utils_3mst_10_traversal_7breadth_first_order(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static char __pyx_doc_7sklearn_5utils_3mst_10_traversal_6breadth_first_order[] = "\n breadth_first_order(csgraph, i_start, directed=True, return_predecessors=True)\n\n Return a breadth-first ordering starting with specified node.\n\n Note that a breadth-first order is not unique, but the tree which it\n generates is unique.\n\n .. versionadded:: 0.11.0\n\n Parameters\n ----------\n csgraph : array_like or sparse matrix\n The N x N compressed sparse graph. The input csgraph will be\n converted to csr format for the calculation.\n i_start : int\n The index of starting node.\n directed : bool, optional\n If True (default), then operate on a directed graph: only\n move from point i to point j along paths csgraph[i, j].\n If False, then find the shortest path on an undirected graph: the\n algorithm can progress from point i to j along csgraph[i, j] or\n csgraph[j, i].\n return_predecessors : bool, optional\n If True (default), then return the predecesor array (see below).\n\n Returns\n -------\n node_array : ndarray, one dimension\n The breadth-first list of nodes, starting with specified node. The\n length of node_array is the number of nodes reachable from the\n specified node.\n predecessors : ndarray, one dimension\n Returned only if return_predecessors is True.\n The length-N list of predecessors of each node in a breadth-first\n tree. If node i is in the tree, then its parent is given by\n predecessors[i]. If node i is not in the tree (and for the parent\n node) then predecessors[i] = -9999.\n ";
|
|
static PyMethodDef __pyx_mdef_7sklearn_5utils_3mst_10_traversal_7breadth_first_order = {__Pyx_NAMESTR("breadth_first_order"), (PyCFunction)__pyx_pw_7sklearn_5utils_3mst_10_traversal_7breadth_first_order, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_5utils_3mst_10_traversal_6breadth_first_order)};
|
|
static PyObject *__pyx_pw_7sklearn_5utils_3mst_10_traversal_7breadth_first_order(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
PyObject *__pyx_v_csgraph = 0;
|
|
PyObject *__pyx_v_i_start = 0;
|
|
PyObject *__pyx_v_directed = 0;
|
|
PyObject *__pyx_v_return_predecessors = 0;
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__csgraph,&__pyx_n_s__i_start,&__pyx_n_s__directed,&__pyx_n_s__return_predecessors,0};
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("breadth_first_order (wrapper)", 0);
|
|
__pyx_self = __pyx_self;
|
|
{
|
|
PyObject* values[4] = {0,0,0,0};
|
|
values[2] = __pyx_k_7;
|
|
values[3] = __pyx_k_8;
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__csgraph);
|
|
if (likely(values[0])) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__i_start);
|
|
if (likely(values[1])) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("breadth_first_order", 0, 2, 4, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 243; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s__directed);
|
|
if (value) { values[2] = value; kw_args--; }
|
|
}
|
|
case 3:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s__return_predecessors);
|
|
if (value) { values[3] = value; kw_args--; }
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "breadth_first_order") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 243; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else {
|
|
switch (PyTuple_GET_SIZE(__pyx_args)) {
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
}
|
|
__pyx_v_csgraph = values[0];
|
|
__pyx_v_i_start = values[1];
|
|
__pyx_v_directed = values[2];
|
|
__pyx_v_return_predecessors = values[3];
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("breadth_first_order", 0, 2, 4, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 243; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.utils.mst._traversal.breadth_first_order", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return NULL;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
__pyx_r = __pyx_pf_7sklearn_5utils_3mst_10_traversal_6breadth_first_order(__pyx_self, __pyx_v_csgraph, __pyx_v_i_start, __pyx_v_directed, __pyx_v_return_predecessors);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":243
|
|
*
|
|
*
|
|
* def breadth_first_order(csgraph, i_start, # <<<<<<<<<<<<<<
|
|
* directed=True, return_predecessors=True):
|
|
* """
|
|
*/
|
|
|
|
static PyObject *__pyx_pf_7sklearn_5utils_3mst_10_traversal_6breadth_first_order(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csgraph, PyObject *__pyx_v_i_start, PyObject *__pyx_v_directed, PyObject *__pyx_v_return_predecessors) {
|
|
int __pyx_v_N;
|
|
PyArrayObject *__pyx_v_node_list = 0;
|
|
PyArrayObject *__pyx_v_predecessors = 0;
|
|
unsigned int __pyx_v_length;
|
|
PyObject *__pyx_v_csgraph_T = NULL;
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_t_5;
|
|
int __pyx_t_6;
|
|
unsigned int __pyx_t_7;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("breadth_first_order", 0);
|
|
__Pyx_INCREF(__pyx_v_csgraph);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":285
|
|
* """
|
|
* global NULL_IDX
|
|
* csgraph = validate_graph(csgraph, directed, dense_output=False) # <<<<<<<<<<<<<<
|
|
* cdef int N = csgraph.shape[0]
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__validate_graph); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 285; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyTuple_New(2); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 285; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_v_csgraph);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_v_csgraph);
|
|
__Pyx_GIVEREF(__pyx_v_csgraph);
|
|
__Pyx_INCREF(__pyx_v_directed);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 1, __pyx_v_directed);
|
|
__Pyx_GIVEREF(__pyx_v_directed);
|
|
__pyx_t_3 = PyDict_New(); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 285; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_3));
|
|
__pyx_t_4 = __Pyx_PyBool_FromLong(0); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 285; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
if (PyDict_SetItem(__pyx_t_3, ((PyObject *)__pyx_n_s__dense_output), __pyx_t_4) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 285; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_4 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_t_2), ((PyObject *)__pyx_t_3)); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 285; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_v_csgraph);
|
|
__pyx_v_csgraph = __pyx_t_4;
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":286
|
|
* global NULL_IDX
|
|
* csgraph = validate_graph(csgraph, directed, dense_output=False)
|
|
* cdef int N = csgraph.shape[0] # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef np.ndarray node_list = np.empty(N, dtype=ITYPE)
|
|
*/
|
|
__pyx_t_4 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__shape); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 286; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = __Pyx_GetItemInt(__pyx_t_4, 0, sizeof(long), PyInt_FromLong); if (!__pyx_t_3) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 286; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_5 = __Pyx_PyInt_AsInt(__pyx_t_3); if (unlikely((__pyx_t_5 == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 286; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_v_N = __pyx_t_5;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":288
|
|
* cdef int N = csgraph.shape[0]
|
|
*
|
|
* cdef np.ndarray node_list = np.empty(N, dtype=ITYPE) # <<<<<<<<<<<<<<
|
|
* cdef np.ndarray predecessors = np.empty(N, dtype=ITYPE)
|
|
* node_list.fill(NULL_IDX)
|
|
*/
|
|
__pyx_t_3 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 288; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyObject_GetAttr(__pyx_t_3, __pyx_n_s__empty); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 288; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_3 = PyInt_FromLong(__pyx_v_N); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 288; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 288; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = PyDict_New(); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 288; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_3));
|
|
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__ITYPE); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 288; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_t_3, ((PyObject *)__pyx_n_s__dtype), __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 288; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_Call(__pyx_t_4, ((PyObject *)__pyx_t_2), ((PyObject *)__pyx_t_3)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 288; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 0;
|
|
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 288; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_node_list = ((PyArrayObject *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":289
|
|
*
|
|
* cdef np.ndarray node_list = np.empty(N, dtype=ITYPE)
|
|
* cdef np.ndarray predecessors = np.empty(N, dtype=ITYPE) # <<<<<<<<<<<<<<
|
|
* node_list.fill(NULL_IDX)
|
|
* predecessors.fill(NULL_IDX)
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 289; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_3 = PyObject_GetAttr(__pyx_t_1, __pyx_n_s__empty); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 289; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyInt_FromLong(__pyx_v_N); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 289; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 289; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 289; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_1));
|
|
__pyx_t_4 = __Pyx_GetName(__pyx_m, __pyx_n_s__ITYPE); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 289; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
if (PyDict_SetItem(__pyx_t_1, ((PyObject *)__pyx_n_s__dtype), __pyx_t_4) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 289; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_4 = PyObject_Call(__pyx_t_3, ((PyObject *)__pyx_t_2), ((PyObject *)__pyx_t_1)); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 289; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_1)); __pyx_t_1 = 0;
|
|
if (!(likely(((__pyx_t_4) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_4, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 289; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_predecessors = ((PyArrayObject *)__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":290
|
|
* cdef np.ndarray node_list = np.empty(N, dtype=ITYPE)
|
|
* cdef np.ndarray predecessors = np.empty(N, dtype=ITYPE)
|
|
* node_list.fill(NULL_IDX) # <<<<<<<<<<<<<<
|
|
* predecessors.fill(NULL_IDX)
|
|
*
|
|
*/
|
|
__pyx_t_4 = PyObject_GetAttr(((PyObject *)__pyx_v_node_list), __pyx_n_s__fill); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_1 = __Pyx_PyInt_to_py_npy_int32(__pyx_v_7sklearn_5utils_3mst_10_traversal_NULL_IDX); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_Call(__pyx_t_4, ((PyObject *)__pyx_t_2), NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":291
|
|
* cdef np.ndarray predecessors = np.empty(N, dtype=ITYPE)
|
|
* node_list.fill(NULL_IDX)
|
|
* predecessors.fill(NULL_IDX) # <<<<<<<<<<<<<<
|
|
*
|
|
* if directed:
|
|
*/
|
|
__pyx_t_1 = PyObject_GetAttr(((PyObject *)__pyx_v_predecessors), __pyx_n_s__fill); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 291; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = __Pyx_PyInt_to_py_npy_int32(__pyx_v_7sklearn_5utils_3mst_10_traversal_NULL_IDX); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 291; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 291; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_t_4), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 291; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_4)); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":293
|
|
* predecessors.fill(NULL_IDX)
|
|
*
|
|
* if directed: # <<<<<<<<<<<<<<
|
|
* length = _breadth_first_directed(i_start,
|
|
* csgraph.indices, csgraph.indptr,
|
|
*/
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_v_directed); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 293; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":294
|
|
*
|
|
* if directed:
|
|
* length = _breadth_first_directed(i_start, # <<<<<<<<<<<<<<
|
|
* csgraph.indices, csgraph.indptr,
|
|
* node_list, predecessors)
|
|
*/
|
|
__pyx_t_7 = __Pyx_PyInt_AsUnsignedInt(__pyx_v_i_start); if (unlikely((__pyx_t_7 == (unsigned int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 294; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":295
|
|
* if directed:
|
|
* length = _breadth_first_directed(i_start,
|
|
* csgraph.indices, csgraph.indptr, # <<<<<<<<<<<<<<
|
|
* node_list, predecessors)
|
|
* else:
|
|
*/
|
|
__pyx_t_2 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__indices); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 295; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 295; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_4 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__indptr); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 295; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
if (!(likely(((__pyx_t_4) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_4, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 295; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":296
|
|
* length = _breadth_first_directed(i_start,
|
|
* csgraph.indices, csgraph.indptr,
|
|
* node_list, predecessors) # <<<<<<<<<<<<<<
|
|
* else:
|
|
* csgraph_T = csgraph.T.tocsr()
|
|
*/
|
|
__pyx_v_length = __pyx_f_7sklearn_5utils_3mst_10_traversal__breadth_first_directed(__pyx_t_7, ((PyArrayObject *)__pyx_t_2), ((PyArrayObject *)__pyx_t_4), ((PyArrayObject *)__pyx_v_node_list), ((PyArrayObject *)__pyx_v_predecessors));
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
goto __pyx_L3;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":298
|
|
* node_list, predecessors)
|
|
* else:
|
|
* csgraph_T = csgraph.T.tocsr() # <<<<<<<<<<<<<<
|
|
* length = _breadth_first_undirected(i_start,
|
|
* csgraph.indices, csgraph.indptr,
|
|
*/
|
|
__pyx_t_4 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__T); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 298; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_2 = PyObject_GetAttr(__pyx_t_4, __pyx_n_s__tocsr); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 298; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_4 = PyObject_Call(__pyx_t_2, ((PyObject *)__pyx_empty_tuple), NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 298; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_v_csgraph_T = __pyx_t_4;
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":299
|
|
* else:
|
|
* csgraph_T = csgraph.T.tocsr()
|
|
* length = _breadth_first_undirected(i_start, # <<<<<<<<<<<<<<
|
|
* csgraph.indices, csgraph.indptr,
|
|
* csgraph_T.indices, csgraph_T.indptr,
|
|
*/
|
|
__pyx_t_7 = __Pyx_PyInt_AsUnsignedInt(__pyx_v_i_start); if (unlikely((__pyx_t_7 == (unsigned int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 299; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":300
|
|
* csgraph_T = csgraph.T.tocsr()
|
|
* length = _breadth_first_undirected(i_start,
|
|
* csgraph.indices, csgraph.indptr, # <<<<<<<<<<<<<<
|
|
* csgraph_T.indices, csgraph_T.indptr,
|
|
* node_list, predecessors)
|
|
*/
|
|
__pyx_t_4 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__indices); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 300; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
if (!(likely(((__pyx_t_4) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_4, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 300; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_2 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__indptr); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 300; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 300; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":301
|
|
* length = _breadth_first_undirected(i_start,
|
|
* csgraph.indices, csgraph.indptr,
|
|
* csgraph_T.indices, csgraph_T.indptr, # <<<<<<<<<<<<<<
|
|
* node_list, predecessors)
|
|
*
|
|
*/
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_v_csgraph_T, __pyx_n_s__indices); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 301; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 301; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_3 = PyObject_GetAttr(__pyx_v_csgraph_T, __pyx_n_s__indptr); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 301; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
if (!(likely(((__pyx_t_3) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_3, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 301; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":302
|
|
* csgraph.indices, csgraph.indptr,
|
|
* csgraph_T.indices, csgraph_T.indptr,
|
|
* node_list, predecessors) # <<<<<<<<<<<<<<
|
|
*
|
|
* if return_predecessors:
|
|
*/
|
|
__pyx_v_length = __pyx_f_7sklearn_5utils_3mst_10_traversal__breadth_first_undirected(__pyx_t_7, ((PyArrayObject *)__pyx_t_4), ((PyArrayObject *)__pyx_t_2), ((PyArrayObject *)__pyx_t_1), ((PyArrayObject *)__pyx_t_3), ((PyArrayObject *)__pyx_v_node_list), ((PyArrayObject *)__pyx_v_predecessors));
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":304
|
|
* node_list, predecessors)
|
|
*
|
|
* if return_predecessors: # <<<<<<<<<<<<<<
|
|
* return node_list[:length], predecessors
|
|
* else:
|
|
*/
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_v_return_predecessors); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 304; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":305
|
|
*
|
|
* if return_predecessors:
|
|
* return node_list[:length], predecessors # <<<<<<<<<<<<<<
|
|
* else:
|
|
* return node_list[:length]
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_3 = __Pyx_PySequence_GetSlice(((PyObject *)__pyx_v_node_list), 0, __pyx_v_length); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 305; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_1 = PyTuple_New(2); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 305; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
PyTuple_SET_ITEM(__pyx_t_1, 0, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_predecessors));
|
|
PyTuple_SET_ITEM(__pyx_t_1, 1, ((PyObject *)__pyx_v_predecessors));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_predecessors));
|
|
__pyx_t_3 = 0;
|
|
__pyx_r = ((PyObject *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
goto __pyx_L4;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":307
|
|
* return node_list[:length], predecessors
|
|
* else:
|
|
* return node_list[:length] # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = __Pyx_PySequence_GetSlice(((PyObject *)__pyx_v_node_list), 0, __pyx_v_length); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 307; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
__pyx_L4:;
|
|
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.utils.mst._traversal.breadth_first_order", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_node_list);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_predecessors);
|
|
__Pyx_XDECREF(__pyx_v_csgraph_T);
|
|
__Pyx_XDECREF(__pyx_v_csgraph);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":310
|
|
*
|
|
*
|
|
* cdef unsigned int _breadth_first_directed( # <<<<<<<<<<<<<<
|
|
* unsigned int head_node,
|
|
* np.ndarray[ITYPE_t, ndim=1, mode='c'] indices,
|
|
*/
|
|
|
|
static unsigned int __pyx_f_7sklearn_5utils_3mst_10_traversal__breadth_first_directed(unsigned int __pyx_v_head_node, PyArrayObject *__pyx_v_indices, PyArrayObject *__pyx_v_indptr, PyArrayObject *__pyx_v_node_list, PyArrayObject *__pyx_v_predecessors) {
|
|
unsigned int __pyx_v_i;
|
|
unsigned int __pyx_v_pnode;
|
|
unsigned int __pyx_v_cnode;
|
|
unsigned int __pyx_v_i_nl;
|
|
unsigned int __pyx_v_i_nl_end;
|
|
CYTHON_UNUSED unsigned int __pyx_v_N;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indices;
|
|
__Pyx_Buffer __pyx_pybuffer_indices;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indptr;
|
|
__Pyx_Buffer __pyx_pybuffer_indptr;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_node_list;
|
|
__Pyx_Buffer __pyx_pybuffer_node_list;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_predecessors;
|
|
__Pyx_Buffer __pyx_pybuffer_predecessors;
|
|
unsigned int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
long __pyx_t_1;
|
|
int __pyx_t_2;
|
|
int __pyx_t_3;
|
|
unsigned int __pyx_t_4;
|
|
unsigned int __pyx_t_5;
|
|
long __pyx_t_6;
|
|
__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t __pyx_t_7;
|
|
unsigned int __pyx_t_8;
|
|
unsigned int __pyx_t_9;
|
|
unsigned int __pyx_t_10;
|
|
unsigned int __pyx_t_11;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_breadth_first_directed", 0);
|
|
__pyx_pybuffer_indices.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indices.refcount = 0;
|
|
__pyx_pybuffernd_indices.data = NULL;
|
|
__pyx_pybuffernd_indices.rcbuffer = &__pyx_pybuffer_indices;
|
|
__pyx_pybuffer_indptr.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indptr.refcount = 0;
|
|
__pyx_pybuffernd_indptr.data = NULL;
|
|
__pyx_pybuffernd_indptr.rcbuffer = &__pyx_pybuffer_indptr;
|
|
__pyx_pybuffer_node_list.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_node_list.refcount = 0;
|
|
__pyx_pybuffernd_node_list.data = NULL;
|
|
__pyx_pybuffernd_node_list.rcbuffer = &__pyx_pybuffer_node_list;
|
|
__pyx_pybuffer_predecessors.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_predecessors.refcount = 0;
|
|
__pyx_pybuffernd_predecessors.data = NULL;
|
|
__pyx_pybuffernd_predecessors.rcbuffer = &__pyx_pybuffer_predecessors;
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indices.rcbuffer->pybuffer, (PyObject*)__pyx_v_indices, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 310; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indices.diminfo[0].strides = __pyx_pybuffernd_indices.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indices.diminfo[0].shape = __pyx_pybuffernd_indices.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indptr.rcbuffer->pybuffer, (PyObject*)__pyx_v_indptr, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 310; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indptr.diminfo[0].strides = __pyx_pybuffernd_indptr.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indptr.diminfo[0].shape = __pyx_pybuffernd_indptr.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_node_list.rcbuffer->pybuffer, (PyObject*)__pyx_v_node_list, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 310; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_node_list.diminfo[0].strides = __pyx_pybuffernd_node_list.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_node_list.diminfo[0].shape = __pyx_pybuffernd_node_list.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_predecessors.rcbuffer->pybuffer, (PyObject*)__pyx_v_predecessors, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 310; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_predecessors.diminfo[0].strides = __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_predecessors.diminfo[0].shape = __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.shape[0];
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":329
|
|
* cdef unsigned int i, pnode, cnode
|
|
* cdef unsigned int i_nl, i_nl_end
|
|
* cdef unsigned int N = node_list.shape[0] # <<<<<<<<<<<<<<
|
|
*
|
|
* node_list[0] = head_node
|
|
*/
|
|
__pyx_v_N = (__pyx_v_node_list->dimensions[0]);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":331
|
|
* cdef unsigned int N = node_list.shape[0]
|
|
*
|
|
* node_list[0] = head_node # <<<<<<<<<<<<<<
|
|
* i_nl = 0
|
|
* i_nl_end = 1
|
|
*/
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_2 = -1;
|
|
if (__pyx_t_1 < 0) {
|
|
__pyx_t_1 += __pyx_pybuffernd_node_list.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_1 < 0)) __pyx_t_2 = 0;
|
|
} else if (unlikely(__pyx_t_1 >= __pyx_pybuffernd_node_list.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 331; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_node_list.rcbuffer->pybuffer.buf, __pyx_t_1, __pyx_pybuffernd_node_list.diminfo[0].strides) = __pyx_v_head_node;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":332
|
|
*
|
|
* node_list[0] = head_node
|
|
* i_nl = 0 # <<<<<<<<<<<<<<
|
|
* i_nl_end = 1
|
|
*
|
|
*/
|
|
__pyx_v_i_nl = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":333
|
|
* node_list[0] = head_node
|
|
* i_nl = 0
|
|
* i_nl_end = 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* while i_nl < i_nl_end:
|
|
*/
|
|
__pyx_v_i_nl_end = 1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":335
|
|
* i_nl_end = 1
|
|
*
|
|
* while i_nl < i_nl_end: # <<<<<<<<<<<<<<
|
|
* pnode = node_list[i_nl]
|
|
*
|
|
*/
|
|
while (1) {
|
|
__pyx_t_3 = (__pyx_v_i_nl < __pyx_v_i_nl_end);
|
|
if (!__pyx_t_3) break;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":336
|
|
*
|
|
* while i_nl < i_nl_end:
|
|
* pnode = node_list[i_nl] # <<<<<<<<<<<<<<
|
|
*
|
|
* for i from indptr[pnode] <= i < indptr[pnode + 1]:
|
|
*/
|
|
__pyx_t_4 = __pyx_v_i_nl;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_4 >= (size_t)__pyx_pybuffernd_node_list.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 336; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_pnode = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_node_list.rcbuffer->pybuffer.buf, __pyx_t_4, __pyx_pybuffernd_node_list.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":338
|
|
* pnode = node_list[i_nl]
|
|
*
|
|
* for i from indptr[pnode] <= i < indptr[pnode + 1]: # <<<<<<<<<<<<<<
|
|
* cnode = indices[i]
|
|
* if (cnode == head_node):
|
|
*/
|
|
__pyx_t_5 = __pyx_v_pnode;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_5 >= (size_t)__pyx_pybuffernd_indptr.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 338; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_6 = (__pyx_v_pnode + 1);
|
|
__pyx_t_2 = -1;
|
|
if (__pyx_t_6 < 0) {
|
|
__pyx_t_6 += __pyx_pybuffernd_indptr.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_6 < 0)) __pyx_t_2 = 0;
|
|
} else if (unlikely(__pyx_t_6 >= __pyx_pybuffernd_indptr.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 338; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_7 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr.rcbuffer->pybuffer.buf, __pyx_t_6, __pyx_pybuffernd_indptr.diminfo[0].strides));
|
|
for (__pyx_v_i = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr.rcbuffer->pybuffer.buf, __pyx_t_5, __pyx_pybuffernd_indptr.diminfo[0].strides)); __pyx_v_i < __pyx_t_7; __pyx_v_i++) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":339
|
|
*
|
|
* for i from indptr[pnode] <= i < indptr[pnode + 1]:
|
|
* cnode = indices[i] # <<<<<<<<<<<<<<
|
|
* if (cnode == head_node):
|
|
* continue
|
|
*/
|
|
__pyx_t_8 = __pyx_v_i;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_8 >= (size_t)__pyx_pybuffernd_indices.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 339; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_cnode = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indices.rcbuffer->pybuffer.buf, __pyx_t_8, __pyx_pybuffernd_indices.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":340
|
|
* for i from indptr[pnode] <= i < indptr[pnode + 1]:
|
|
* cnode = indices[i]
|
|
* if (cnode == head_node): # <<<<<<<<<<<<<<
|
|
* continue
|
|
* elif (predecessors[cnode] == NULL_IDX):
|
|
*/
|
|
__pyx_t_3 = (__pyx_v_cnode == __pyx_v_head_node);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":341
|
|
* cnode = indices[i]
|
|
* if (cnode == head_node):
|
|
* continue # <<<<<<<<<<<<<<
|
|
* elif (predecessors[cnode] == NULL_IDX):
|
|
* node_list[i_nl_end] = cnode
|
|
*/
|
|
goto __pyx_L5_continue;
|
|
goto __pyx_L7;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":342
|
|
* if (cnode == head_node):
|
|
* continue
|
|
* elif (predecessors[cnode] == NULL_IDX): # <<<<<<<<<<<<<<
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode
|
|
*/
|
|
__pyx_t_9 = __pyx_v_cnode;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_9 >= (size_t)__pyx_pybuffernd_predecessors.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 342; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_3 = ((*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.buf, __pyx_t_9, __pyx_pybuffernd_predecessors.diminfo[0].strides)) == __pyx_v_7sklearn_5utils_3mst_10_traversal_NULL_IDX);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":343
|
|
* continue
|
|
* elif (predecessors[cnode] == NULL_IDX):
|
|
* node_list[i_nl_end] = cnode # <<<<<<<<<<<<<<
|
|
* predecessors[cnode] = pnode
|
|
* i_nl_end += 1
|
|
*/
|
|
__pyx_t_10 = __pyx_v_i_nl_end;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_10 >= (size_t)__pyx_pybuffernd_node_list.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 343; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_node_list.rcbuffer->pybuffer.buf, __pyx_t_10, __pyx_pybuffernd_node_list.diminfo[0].strides) = __pyx_v_cnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":344
|
|
* elif (predecessors[cnode] == NULL_IDX):
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode # <<<<<<<<<<<<<<
|
|
* i_nl_end += 1
|
|
*
|
|
*/
|
|
__pyx_t_11 = __pyx_v_cnode;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_11 >= (size_t)__pyx_pybuffernd_predecessors.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 344; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.buf, __pyx_t_11, __pyx_pybuffernd_predecessors.diminfo[0].strides) = __pyx_v_pnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":345
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode
|
|
* i_nl_end += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* i_nl += 1
|
|
*/
|
|
__pyx_v_i_nl_end = (__pyx_v_i_nl_end + 1);
|
|
goto __pyx_L7;
|
|
}
|
|
__pyx_L7:;
|
|
__pyx_L5_continue:;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":347
|
|
* i_nl_end += 1
|
|
*
|
|
* i_nl += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* return i_nl
|
|
*/
|
|
__pyx_v_i_nl = (__pyx_v_i_nl + 1);
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":349
|
|
* i_nl += 1
|
|
*
|
|
* return i_nl # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = __pyx_v_i_nl;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_node_list.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_predecessors.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_WriteUnraisable("sklearn.utils.mst._traversal._breadth_first_directed", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_node_list.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_predecessors.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":352
|
|
*
|
|
*
|
|
* cdef unsigned int _breadth_first_undirected( # <<<<<<<<<<<<<<
|
|
* unsigned int head_node,
|
|
* np.ndarray[ITYPE_t, ndim=1, mode='c'] indices1,
|
|
*/
|
|
|
|
static unsigned int __pyx_f_7sklearn_5utils_3mst_10_traversal__breadth_first_undirected(unsigned int __pyx_v_head_node, PyArrayObject *__pyx_v_indices1, PyArrayObject *__pyx_v_indptr1, PyArrayObject *__pyx_v_indices2, PyArrayObject *__pyx_v_indptr2, PyArrayObject *__pyx_v_node_list, PyArrayObject *__pyx_v_predecessors) {
|
|
unsigned int __pyx_v_i;
|
|
unsigned int __pyx_v_pnode;
|
|
unsigned int __pyx_v_cnode;
|
|
unsigned int __pyx_v_i_nl;
|
|
unsigned int __pyx_v_i_nl_end;
|
|
CYTHON_UNUSED unsigned int __pyx_v_N;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indices1;
|
|
__Pyx_Buffer __pyx_pybuffer_indices1;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indices2;
|
|
__Pyx_Buffer __pyx_pybuffer_indices2;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indptr1;
|
|
__Pyx_Buffer __pyx_pybuffer_indptr1;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indptr2;
|
|
__Pyx_Buffer __pyx_pybuffer_indptr2;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_node_list;
|
|
__Pyx_Buffer __pyx_pybuffer_node_list;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_predecessors;
|
|
__Pyx_Buffer __pyx_pybuffer_predecessors;
|
|
unsigned int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
long __pyx_t_1;
|
|
int __pyx_t_2;
|
|
int __pyx_t_3;
|
|
unsigned int __pyx_t_4;
|
|
unsigned int __pyx_t_5;
|
|
long __pyx_t_6;
|
|
__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t __pyx_t_7;
|
|
unsigned int __pyx_t_8;
|
|
unsigned int __pyx_t_9;
|
|
unsigned int __pyx_t_10;
|
|
unsigned int __pyx_t_11;
|
|
unsigned int __pyx_t_12;
|
|
long __pyx_t_13;
|
|
unsigned int __pyx_t_14;
|
|
unsigned int __pyx_t_15;
|
|
unsigned int __pyx_t_16;
|
|
unsigned int __pyx_t_17;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_breadth_first_undirected", 0);
|
|
__pyx_pybuffer_indices1.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indices1.refcount = 0;
|
|
__pyx_pybuffernd_indices1.data = NULL;
|
|
__pyx_pybuffernd_indices1.rcbuffer = &__pyx_pybuffer_indices1;
|
|
__pyx_pybuffer_indptr1.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indptr1.refcount = 0;
|
|
__pyx_pybuffernd_indptr1.data = NULL;
|
|
__pyx_pybuffernd_indptr1.rcbuffer = &__pyx_pybuffer_indptr1;
|
|
__pyx_pybuffer_indices2.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indices2.refcount = 0;
|
|
__pyx_pybuffernd_indices2.data = NULL;
|
|
__pyx_pybuffernd_indices2.rcbuffer = &__pyx_pybuffer_indices2;
|
|
__pyx_pybuffer_indptr2.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indptr2.refcount = 0;
|
|
__pyx_pybuffernd_indptr2.data = NULL;
|
|
__pyx_pybuffernd_indptr2.rcbuffer = &__pyx_pybuffer_indptr2;
|
|
__pyx_pybuffer_node_list.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_node_list.refcount = 0;
|
|
__pyx_pybuffernd_node_list.data = NULL;
|
|
__pyx_pybuffernd_node_list.rcbuffer = &__pyx_pybuffer_node_list;
|
|
__pyx_pybuffer_predecessors.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_predecessors.refcount = 0;
|
|
__pyx_pybuffernd_predecessors.data = NULL;
|
|
__pyx_pybuffernd_predecessors.rcbuffer = &__pyx_pybuffer_predecessors;
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indices1.rcbuffer->pybuffer, (PyObject*)__pyx_v_indices1, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indices1.diminfo[0].strides = __pyx_pybuffernd_indices1.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indices1.diminfo[0].shape = __pyx_pybuffernd_indices1.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indptr1.rcbuffer->pybuffer, (PyObject*)__pyx_v_indptr1, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indptr1.diminfo[0].strides = __pyx_pybuffernd_indptr1.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indptr1.diminfo[0].shape = __pyx_pybuffernd_indptr1.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indices2.rcbuffer->pybuffer, (PyObject*)__pyx_v_indices2, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indices2.diminfo[0].strides = __pyx_pybuffernd_indices2.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indices2.diminfo[0].shape = __pyx_pybuffernd_indices2.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indptr2.rcbuffer->pybuffer, (PyObject*)__pyx_v_indptr2, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indptr2.diminfo[0].strides = __pyx_pybuffernd_indptr2.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indptr2.diminfo[0].shape = __pyx_pybuffernd_indptr2.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_node_list.rcbuffer->pybuffer, (PyObject*)__pyx_v_node_list, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_node_list.diminfo[0].strides = __pyx_pybuffernd_node_list.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_node_list.diminfo[0].shape = __pyx_pybuffernd_node_list.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_predecessors.rcbuffer->pybuffer, (PyObject*)__pyx_v_predecessors, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_predecessors.diminfo[0].strides = __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_predecessors.diminfo[0].shape = __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.shape[0];
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":375
|
|
* cdef unsigned int i, pnode, cnode
|
|
* cdef unsigned int i_nl, i_nl_end
|
|
* cdef unsigned int N = node_list.shape[0] # <<<<<<<<<<<<<<
|
|
*
|
|
* node_list[0] = head_node
|
|
*/
|
|
__pyx_v_N = (__pyx_v_node_list->dimensions[0]);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":377
|
|
* cdef unsigned int N = node_list.shape[0]
|
|
*
|
|
* node_list[0] = head_node # <<<<<<<<<<<<<<
|
|
* i_nl = 0
|
|
* i_nl_end = 1
|
|
*/
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_2 = -1;
|
|
if (__pyx_t_1 < 0) {
|
|
__pyx_t_1 += __pyx_pybuffernd_node_list.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_1 < 0)) __pyx_t_2 = 0;
|
|
} else if (unlikely(__pyx_t_1 >= __pyx_pybuffernd_node_list.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 377; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_node_list.rcbuffer->pybuffer.buf, __pyx_t_1, __pyx_pybuffernd_node_list.diminfo[0].strides) = __pyx_v_head_node;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":378
|
|
*
|
|
* node_list[0] = head_node
|
|
* i_nl = 0 # <<<<<<<<<<<<<<
|
|
* i_nl_end = 1
|
|
*
|
|
*/
|
|
__pyx_v_i_nl = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":379
|
|
* node_list[0] = head_node
|
|
* i_nl = 0
|
|
* i_nl_end = 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* while i_nl < i_nl_end:
|
|
*/
|
|
__pyx_v_i_nl_end = 1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":381
|
|
* i_nl_end = 1
|
|
*
|
|
* while i_nl < i_nl_end: # <<<<<<<<<<<<<<
|
|
* pnode = node_list[i_nl]
|
|
*
|
|
*/
|
|
while (1) {
|
|
__pyx_t_3 = (__pyx_v_i_nl < __pyx_v_i_nl_end);
|
|
if (!__pyx_t_3) break;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":382
|
|
*
|
|
* while i_nl < i_nl_end:
|
|
* pnode = node_list[i_nl] # <<<<<<<<<<<<<<
|
|
*
|
|
* for i from indptr1[pnode] <= i < indptr1[pnode + 1]:
|
|
*/
|
|
__pyx_t_4 = __pyx_v_i_nl;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_4 >= (size_t)__pyx_pybuffernd_node_list.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 382; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_pnode = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_node_list.rcbuffer->pybuffer.buf, __pyx_t_4, __pyx_pybuffernd_node_list.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":384
|
|
* pnode = node_list[i_nl]
|
|
*
|
|
* for i from indptr1[pnode] <= i < indptr1[pnode + 1]: # <<<<<<<<<<<<<<
|
|
* cnode = indices1[i]
|
|
* if (cnode == head_node):
|
|
*/
|
|
__pyx_t_5 = __pyx_v_pnode;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_5 >= (size_t)__pyx_pybuffernd_indptr1.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 384; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_6 = (__pyx_v_pnode + 1);
|
|
__pyx_t_2 = -1;
|
|
if (__pyx_t_6 < 0) {
|
|
__pyx_t_6 += __pyx_pybuffernd_indptr1.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_6 < 0)) __pyx_t_2 = 0;
|
|
} else if (unlikely(__pyx_t_6 >= __pyx_pybuffernd_indptr1.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 384; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_7 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr1.rcbuffer->pybuffer.buf, __pyx_t_6, __pyx_pybuffernd_indptr1.diminfo[0].strides));
|
|
for (__pyx_v_i = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr1.rcbuffer->pybuffer.buf, __pyx_t_5, __pyx_pybuffernd_indptr1.diminfo[0].strides)); __pyx_v_i < __pyx_t_7; __pyx_v_i++) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":385
|
|
*
|
|
* for i from indptr1[pnode] <= i < indptr1[pnode + 1]:
|
|
* cnode = indices1[i] # <<<<<<<<<<<<<<
|
|
* if (cnode == head_node):
|
|
* continue
|
|
*/
|
|
__pyx_t_8 = __pyx_v_i;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_8 >= (size_t)__pyx_pybuffernd_indices1.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 385; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_cnode = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indices1.rcbuffer->pybuffer.buf, __pyx_t_8, __pyx_pybuffernd_indices1.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":386
|
|
* for i from indptr1[pnode] <= i < indptr1[pnode + 1]:
|
|
* cnode = indices1[i]
|
|
* if (cnode == head_node): # <<<<<<<<<<<<<<
|
|
* continue
|
|
* elif (predecessors[cnode] == NULL_IDX):
|
|
*/
|
|
__pyx_t_3 = (__pyx_v_cnode == __pyx_v_head_node);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":387
|
|
* cnode = indices1[i]
|
|
* if (cnode == head_node):
|
|
* continue # <<<<<<<<<<<<<<
|
|
* elif (predecessors[cnode] == NULL_IDX):
|
|
* node_list[i_nl_end] = cnode
|
|
*/
|
|
goto __pyx_L5_continue;
|
|
goto __pyx_L7;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":388
|
|
* if (cnode == head_node):
|
|
* continue
|
|
* elif (predecessors[cnode] == NULL_IDX): # <<<<<<<<<<<<<<
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode
|
|
*/
|
|
__pyx_t_9 = __pyx_v_cnode;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_9 >= (size_t)__pyx_pybuffernd_predecessors.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 388; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_3 = ((*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.buf, __pyx_t_9, __pyx_pybuffernd_predecessors.diminfo[0].strides)) == __pyx_v_7sklearn_5utils_3mst_10_traversal_NULL_IDX);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":389
|
|
* continue
|
|
* elif (predecessors[cnode] == NULL_IDX):
|
|
* node_list[i_nl_end] = cnode # <<<<<<<<<<<<<<
|
|
* predecessors[cnode] = pnode
|
|
* i_nl_end += 1
|
|
*/
|
|
__pyx_t_10 = __pyx_v_i_nl_end;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_10 >= (size_t)__pyx_pybuffernd_node_list.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 389; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_node_list.rcbuffer->pybuffer.buf, __pyx_t_10, __pyx_pybuffernd_node_list.diminfo[0].strides) = __pyx_v_cnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":390
|
|
* elif (predecessors[cnode] == NULL_IDX):
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode # <<<<<<<<<<<<<<
|
|
* i_nl_end += 1
|
|
*
|
|
*/
|
|
__pyx_t_11 = __pyx_v_cnode;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_11 >= (size_t)__pyx_pybuffernd_predecessors.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 390; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.buf, __pyx_t_11, __pyx_pybuffernd_predecessors.diminfo[0].strides) = __pyx_v_pnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":391
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode
|
|
* i_nl_end += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* for i from indptr2[pnode] <= i < indptr2[pnode + 1]:
|
|
*/
|
|
__pyx_v_i_nl_end = (__pyx_v_i_nl_end + 1);
|
|
goto __pyx_L7;
|
|
}
|
|
__pyx_L7:;
|
|
__pyx_L5_continue:;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":393
|
|
* i_nl_end += 1
|
|
*
|
|
* for i from indptr2[pnode] <= i < indptr2[pnode + 1]: # <<<<<<<<<<<<<<
|
|
* cnode = indices2[i]
|
|
* if (cnode == head_node):
|
|
*/
|
|
__pyx_t_12 = __pyx_v_pnode;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_12 >= (size_t)__pyx_pybuffernd_indptr2.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 393; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_13 = (__pyx_v_pnode + 1);
|
|
__pyx_t_2 = -1;
|
|
if (__pyx_t_13 < 0) {
|
|
__pyx_t_13 += __pyx_pybuffernd_indptr2.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_13 < 0)) __pyx_t_2 = 0;
|
|
} else if (unlikely(__pyx_t_13 >= __pyx_pybuffernd_indptr2.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 393; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_7 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr2.rcbuffer->pybuffer.buf, __pyx_t_13, __pyx_pybuffernd_indptr2.diminfo[0].strides));
|
|
for (__pyx_v_i = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr2.rcbuffer->pybuffer.buf, __pyx_t_12, __pyx_pybuffernd_indptr2.diminfo[0].strides)); __pyx_v_i < __pyx_t_7; __pyx_v_i++) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":394
|
|
*
|
|
* for i from indptr2[pnode] <= i < indptr2[pnode + 1]:
|
|
* cnode = indices2[i] # <<<<<<<<<<<<<<
|
|
* if (cnode == head_node):
|
|
* continue
|
|
*/
|
|
__pyx_t_14 = __pyx_v_i;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_14 >= (size_t)__pyx_pybuffernd_indices2.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 394; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_cnode = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indices2.rcbuffer->pybuffer.buf, __pyx_t_14, __pyx_pybuffernd_indices2.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":395
|
|
* for i from indptr2[pnode] <= i < indptr2[pnode + 1]:
|
|
* cnode = indices2[i]
|
|
* if (cnode == head_node): # <<<<<<<<<<<<<<
|
|
* continue
|
|
* elif (predecessors[cnode] == NULL_IDX):
|
|
*/
|
|
__pyx_t_3 = (__pyx_v_cnode == __pyx_v_head_node);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":396
|
|
* cnode = indices2[i]
|
|
* if (cnode == head_node):
|
|
* continue # <<<<<<<<<<<<<<
|
|
* elif (predecessors[cnode] == NULL_IDX):
|
|
* node_list[i_nl_end] = cnode
|
|
*/
|
|
goto __pyx_L8_continue;
|
|
goto __pyx_L10;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":397
|
|
* if (cnode == head_node):
|
|
* continue
|
|
* elif (predecessors[cnode] == NULL_IDX): # <<<<<<<<<<<<<<
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode
|
|
*/
|
|
__pyx_t_15 = __pyx_v_cnode;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_15 >= (size_t)__pyx_pybuffernd_predecessors.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 397; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_3 = ((*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.buf, __pyx_t_15, __pyx_pybuffernd_predecessors.diminfo[0].strides)) == __pyx_v_7sklearn_5utils_3mst_10_traversal_NULL_IDX);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":398
|
|
* continue
|
|
* elif (predecessors[cnode] == NULL_IDX):
|
|
* node_list[i_nl_end] = cnode # <<<<<<<<<<<<<<
|
|
* predecessors[cnode] = pnode
|
|
* i_nl_end += 1
|
|
*/
|
|
__pyx_t_16 = __pyx_v_i_nl_end;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_16 >= (size_t)__pyx_pybuffernd_node_list.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 398; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_node_list.rcbuffer->pybuffer.buf, __pyx_t_16, __pyx_pybuffernd_node_list.diminfo[0].strides) = __pyx_v_cnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":399
|
|
* elif (predecessors[cnode] == NULL_IDX):
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode # <<<<<<<<<<<<<<
|
|
* i_nl_end += 1
|
|
*
|
|
*/
|
|
__pyx_t_17 = __pyx_v_cnode;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_17 >= (size_t)__pyx_pybuffernd_predecessors.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 399; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.buf, __pyx_t_17, __pyx_pybuffernd_predecessors.diminfo[0].strides) = __pyx_v_pnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":400
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode
|
|
* i_nl_end += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* i_nl += 1
|
|
*/
|
|
__pyx_v_i_nl_end = (__pyx_v_i_nl_end + 1);
|
|
goto __pyx_L10;
|
|
}
|
|
__pyx_L10:;
|
|
__pyx_L8_continue:;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":402
|
|
* i_nl_end += 1
|
|
*
|
|
* i_nl += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* return i_nl
|
|
*/
|
|
__pyx_v_i_nl = (__pyx_v_i_nl + 1);
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":404
|
|
* i_nl += 1
|
|
*
|
|
* return i_nl # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = __pyx_v_i_nl;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices1.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices2.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr1.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr2.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_node_list.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_predecessors.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_WriteUnraisable("sklearn.utils.mst._traversal._breadth_first_undirected", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices1.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices2.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr1.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr2.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_node_list.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_predecessors.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_5utils_3mst_10_traversal_9depth_first_order(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static char __pyx_doc_7sklearn_5utils_3mst_10_traversal_8depth_first_order[] = "\n depth_first_order(csgraph, i_start, directed=True, return_predecessors=True)\n\n Return a depth-first ordering starting with specified node.\n\n Note that a depth-first order is not unique. Furthermore, for graphs\n with cycles, the tree generated by a depth-first search is not\n unique either.\n\n .. versionadded:: 0.11.0\n\n Parameters\n ----------\n csgraph : array_like or sparse matrix\n The N x N compressed sparse graph. The input csgraph will be\n converted to csr format for the calculation.\n i_start : int\n The index of starting node.\n directed : bool, optional\n If True (default), then operate on a directed graph: only\n move from point i to point j along paths csgraph[i, j].\n If False, then find the shortest path on an undirected graph: the\n algorithm can progress from point i to j along csgraph[i, j] or\n csgraph[j, i].\n return_predecessors : bool, optional\n If True (default), then return the predecesor array (see below).\n\n Returns\n -------\n node_array : ndarray, one dimension\n The breadth-first list of nodes, starting with specified node. The\n length of node_array is the number of nodes reachable from the\n specified node.\n predecessors : ndarray, one dimension\n Returned only if return_predecessors is True.\n The length-N list of predecessors of each node in a breadth-first\n tree. If node i is in the tree, then its parent is given by\n predecessors[i]. If node i is not in the tree (and for the parent\n node) then predecessors[i] = -9999.\n ";
|
|
static PyMethodDef __pyx_mdef_7sklearn_5utils_3mst_10_traversal_9depth_first_order = {__Pyx_NAMESTR("depth_first_order"), (PyCFunction)__pyx_pw_7sklearn_5utils_3mst_10_traversal_9depth_first_order, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_5utils_3mst_10_traversal_8depth_first_order)};
|
|
static PyObject *__pyx_pw_7sklearn_5utils_3mst_10_traversal_9depth_first_order(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
PyObject *__pyx_v_csgraph = 0;
|
|
PyObject *__pyx_v_i_start = 0;
|
|
PyObject *__pyx_v_directed = 0;
|
|
PyObject *__pyx_v_return_predecessors = 0;
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__csgraph,&__pyx_n_s__i_start,&__pyx_n_s__directed,&__pyx_n_s__return_predecessors,0};
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("depth_first_order (wrapper)", 0);
|
|
__pyx_self = __pyx_self;
|
|
{
|
|
PyObject* values[4] = {0,0,0,0};
|
|
values[2] = __pyx_k_9;
|
|
values[3] = __pyx_k_10;
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__csgraph);
|
|
if (likely(values[0])) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__i_start);
|
|
if (likely(values[1])) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("depth_first_order", 0, 2, 4, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 407; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s__directed);
|
|
if (value) { values[2] = value; kw_args--; }
|
|
}
|
|
case 3:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s__return_predecessors);
|
|
if (value) { values[3] = value; kw_args--; }
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "depth_first_order") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 407; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else {
|
|
switch (PyTuple_GET_SIZE(__pyx_args)) {
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
}
|
|
__pyx_v_csgraph = values[0];
|
|
__pyx_v_i_start = values[1];
|
|
__pyx_v_directed = values[2];
|
|
__pyx_v_return_predecessors = values[3];
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("depth_first_order", 0, 2, 4, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 407; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.utils.mst._traversal.depth_first_order", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return NULL;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
__pyx_r = __pyx_pf_7sklearn_5utils_3mst_10_traversal_8depth_first_order(__pyx_self, __pyx_v_csgraph, __pyx_v_i_start, __pyx_v_directed, __pyx_v_return_predecessors);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":407
|
|
*
|
|
*
|
|
* def depth_first_order(csgraph, i_start, # <<<<<<<<<<<<<<
|
|
* directed=True, return_predecessors=True):
|
|
* """
|
|
*/
|
|
|
|
static PyObject *__pyx_pf_7sklearn_5utils_3mst_10_traversal_8depth_first_order(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csgraph, PyObject *__pyx_v_i_start, PyObject *__pyx_v_directed, PyObject *__pyx_v_return_predecessors) {
|
|
int __pyx_v_N;
|
|
PyObject *__pyx_v_node_list = NULL;
|
|
PyObject *__pyx_v_predecessors = NULL;
|
|
PyObject *__pyx_v_root_list = NULL;
|
|
PyObject *__pyx_v_flag = NULL;
|
|
unsigned int __pyx_v_length;
|
|
PyObject *__pyx_v_csgraph_T = NULL;
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_t_5;
|
|
int __pyx_t_6;
|
|
unsigned int __pyx_t_7;
|
|
PyObject *__pyx_t_8 = NULL;
|
|
PyObject *__pyx_t_9 = NULL;
|
|
PyObject *__pyx_t_10 = NULL;
|
|
PyObject *__pyx_t_11 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("depth_first_order", 0);
|
|
__Pyx_INCREF(__pyx_v_csgraph);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":450
|
|
* """
|
|
* global NULL_IDX
|
|
* csgraph = validate_graph(csgraph, directed, dense_output=False) # <<<<<<<<<<<<<<
|
|
* cdef int N = csgraph.shape[0]
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__validate_graph); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 450; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyTuple_New(2); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 450; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_v_csgraph);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_v_csgraph);
|
|
__Pyx_GIVEREF(__pyx_v_csgraph);
|
|
__Pyx_INCREF(__pyx_v_directed);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 1, __pyx_v_directed);
|
|
__Pyx_GIVEREF(__pyx_v_directed);
|
|
__pyx_t_3 = PyDict_New(); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 450; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_3));
|
|
__pyx_t_4 = __Pyx_PyBool_FromLong(0); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 450; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
if (PyDict_SetItem(__pyx_t_3, ((PyObject *)__pyx_n_s__dense_output), __pyx_t_4) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 450; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_4 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_t_2), ((PyObject *)__pyx_t_3)); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 450; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_v_csgraph);
|
|
__pyx_v_csgraph = __pyx_t_4;
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":451
|
|
* global NULL_IDX
|
|
* csgraph = validate_graph(csgraph, directed, dense_output=False)
|
|
* cdef int N = csgraph.shape[0] # <<<<<<<<<<<<<<
|
|
*
|
|
* node_list = np.empty(N, dtype=ITYPE)
|
|
*/
|
|
__pyx_t_4 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__shape); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 451; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = __Pyx_GetItemInt(__pyx_t_4, 0, sizeof(long), PyInt_FromLong); if (!__pyx_t_3) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 451; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_5 = __Pyx_PyInt_AsInt(__pyx_t_3); if (unlikely((__pyx_t_5 == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 451; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_v_N = __pyx_t_5;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":453
|
|
* cdef int N = csgraph.shape[0]
|
|
*
|
|
* node_list = np.empty(N, dtype=ITYPE) # <<<<<<<<<<<<<<
|
|
* predecessors = np.empty(N, dtype=ITYPE)
|
|
* root_list = np.empty(N, dtype=ITYPE)
|
|
*/
|
|
__pyx_t_3 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 453; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyObject_GetAttr(__pyx_t_3, __pyx_n_s__empty); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 453; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_3 = PyInt_FromLong(__pyx_v_N); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 453; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 453; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = PyDict_New(); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 453; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_3));
|
|
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__ITYPE); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 453; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_t_3, ((PyObject *)__pyx_n_s__dtype), __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 453; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_Call(__pyx_t_4, ((PyObject *)__pyx_t_2), ((PyObject *)__pyx_t_3)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 453; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 0;
|
|
__pyx_v_node_list = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":454
|
|
*
|
|
* node_list = np.empty(N, dtype=ITYPE)
|
|
* predecessors = np.empty(N, dtype=ITYPE) # <<<<<<<<<<<<<<
|
|
* root_list = np.empty(N, dtype=ITYPE)
|
|
* flag = np.zeros(N, dtype=ITYPE)
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 454; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_3 = PyObject_GetAttr(__pyx_t_1, __pyx_n_s__empty); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 454; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyInt_FromLong(__pyx_v_N); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 454; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 454; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 454; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_1));
|
|
__pyx_t_4 = __Pyx_GetName(__pyx_m, __pyx_n_s__ITYPE); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 454; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
if (PyDict_SetItem(__pyx_t_1, ((PyObject *)__pyx_n_s__dtype), __pyx_t_4) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 454; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_4 = PyObject_Call(__pyx_t_3, ((PyObject *)__pyx_t_2), ((PyObject *)__pyx_t_1)); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 454; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_1)); __pyx_t_1 = 0;
|
|
__pyx_v_predecessors = __pyx_t_4;
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":455
|
|
* node_list = np.empty(N, dtype=ITYPE)
|
|
* predecessors = np.empty(N, dtype=ITYPE)
|
|
* root_list = np.empty(N, dtype=ITYPE) # <<<<<<<<<<<<<<
|
|
* flag = np.zeros(N, dtype=ITYPE)
|
|
* node_list.fill(NULL_IDX)
|
|
*/
|
|
__pyx_t_4 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 455; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_t_4, __pyx_n_s__empty); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 455; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_4 = PyInt_FromLong(__pyx_v_N); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 455; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 455; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_4 = PyDict_New(); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 455; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_4));
|
|
__pyx_t_3 = __Pyx_GetName(__pyx_m, __pyx_n_s__ITYPE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 455; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
if (PyDict_SetItem(__pyx_t_4, ((PyObject *)__pyx_n_s__dtype), __pyx_t_3) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 455; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_3 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_t_2), ((PyObject *)__pyx_t_4)); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 455; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_4)); __pyx_t_4 = 0;
|
|
__pyx_v_root_list = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":456
|
|
* predecessors = np.empty(N, dtype=ITYPE)
|
|
* root_list = np.empty(N, dtype=ITYPE)
|
|
* flag = np.zeros(N, dtype=ITYPE) # <<<<<<<<<<<<<<
|
|
* node_list.fill(NULL_IDX)
|
|
* predecessors.fill(NULL_IDX)
|
|
*/
|
|
__pyx_t_3 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 456; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyObject_GetAttr(__pyx_t_3, __pyx_n_s__zeros); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 456; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_3 = PyInt_FromLong(__pyx_v_N); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 456; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 456; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = PyDict_New(); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 456; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_3));
|
|
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__ITYPE); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 456; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_t_3, ((PyObject *)__pyx_n_s__dtype), __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 456; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_Call(__pyx_t_4, ((PyObject *)__pyx_t_2), ((PyObject *)__pyx_t_3)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 456; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 0;
|
|
__pyx_v_flag = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":457
|
|
* root_list = np.empty(N, dtype=ITYPE)
|
|
* flag = np.zeros(N, dtype=ITYPE)
|
|
* node_list.fill(NULL_IDX) # <<<<<<<<<<<<<<
|
|
* predecessors.fill(NULL_IDX)
|
|
* root_list.fill(NULL_IDX)
|
|
*/
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_v_node_list, __pyx_n_s__fill); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 457; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_3 = __Pyx_PyInt_to_py_npy_int32(__pyx_v_7sklearn_5utils_3mst_10_traversal_NULL_IDX); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 457; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 457; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_t_2), NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 457; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":458
|
|
* flag = np.zeros(N, dtype=ITYPE)
|
|
* node_list.fill(NULL_IDX)
|
|
* predecessors.fill(NULL_IDX) # <<<<<<<<<<<<<<
|
|
* root_list.fill(NULL_IDX)
|
|
*
|
|
*/
|
|
__pyx_t_3 = PyObject_GetAttr(__pyx_v_predecessors, __pyx_n_s__fill); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 458; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = __Pyx_PyInt_to_py_npy_int32(__pyx_v_7sklearn_5utils_3mst_10_traversal_NULL_IDX); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 458; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_1 = PyTuple_New(1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 458; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
PyTuple_SET_ITEM(__pyx_t_1, 0, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = PyObject_Call(__pyx_t_3, ((PyObject *)__pyx_t_1), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 458; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_1)); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":459
|
|
* node_list.fill(NULL_IDX)
|
|
* predecessors.fill(NULL_IDX)
|
|
* root_list.fill(NULL_IDX) # <<<<<<<<<<<<<<
|
|
*
|
|
* if directed:
|
|
*/
|
|
__pyx_t_2 = PyObject_GetAttr(__pyx_v_root_list, __pyx_n_s__fill); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 459; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_1 = __Pyx_PyInt_to_py_npy_int32(__pyx_v_7sklearn_5utils_3mst_10_traversal_NULL_IDX); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 459; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 459; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_Call(__pyx_t_2, ((PyObject *)__pyx_t_3), NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 459; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":461
|
|
* root_list.fill(NULL_IDX)
|
|
*
|
|
* if directed: # <<<<<<<<<<<<<<
|
|
* length = _depth_first_directed(i_start,
|
|
* csgraph.indices, csgraph.indptr,
|
|
*/
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_v_directed); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 461; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":462
|
|
*
|
|
* if directed:
|
|
* length = _depth_first_directed(i_start, # <<<<<<<<<<<<<<
|
|
* csgraph.indices, csgraph.indptr,
|
|
* node_list, predecessors,
|
|
*/
|
|
__pyx_t_7 = __Pyx_PyInt_AsUnsignedInt(__pyx_v_i_start); if (unlikely((__pyx_t_7 == (unsigned int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 462; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":463
|
|
* if directed:
|
|
* length = _depth_first_directed(i_start,
|
|
* csgraph.indices, csgraph.indptr, # <<<<<<<<<<<<<<
|
|
* node_list, predecessors,
|
|
* root_list, flag)
|
|
*/
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__indices); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_3 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__indptr); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
if (!(likely(((__pyx_t_3) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_3, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":464
|
|
* length = _depth_first_directed(i_start,
|
|
* csgraph.indices, csgraph.indptr,
|
|
* node_list, predecessors, # <<<<<<<<<<<<<<
|
|
* root_list, flag)
|
|
* else:
|
|
*/
|
|
if (!(likely(((__pyx_v_node_list) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_node_list, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 464; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_2 = __pyx_v_node_list;
|
|
__Pyx_INCREF(__pyx_t_2);
|
|
if (!(likely(((__pyx_v_predecessors) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_predecessors, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 464; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_4 = __pyx_v_predecessors;
|
|
__Pyx_INCREF(__pyx_t_4);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":465
|
|
* csgraph.indices, csgraph.indptr,
|
|
* node_list, predecessors,
|
|
* root_list, flag) # <<<<<<<<<<<<<<
|
|
* else:
|
|
* csgraph_T = csgraph.T.tocsr()
|
|
*/
|
|
if (!(likely(((__pyx_v_root_list) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_root_list, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 465; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_8 = __pyx_v_root_list;
|
|
__Pyx_INCREF(__pyx_t_8);
|
|
if (!(likely(((__pyx_v_flag) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_flag, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 465; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_9 = __pyx_v_flag;
|
|
__Pyx_INCREF(__pyx_t_9);
|
|
__pyx_v_length = __pyx_f_7sklearn_5utils_3mst_10_traversal__depth_first_directed(__pyx_t_7, ((PyArrayObject *)__pyx_t_1), ((PyArrayObject *)__pyx_t_3), ((PyArrayObject *)__pyx_t_2), ((PyArrayObject *)__pyx_t_4), ((PyArrayObject *)__pyx_t_8), ((PyArrayObject *)__pyx_t_9));
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
|
|
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
|
|
goto __pyx_L3;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":467
|
|
* root_list, flag)
|
|
* else:
|
|
* csgraph_T = csgraph.T.tocsr() # <<<<<<<<<<<<<<
|
|
* length = _depth_first_undirected(i_start,
|
|
* csgraph.indices, csgraph.indptr,
|
|
*/
|
|
__pyx_t_9 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__T); if (unlikely(!__pyx_t_9)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 467; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_9);
|
|
__pyx_t_8 = PyObject_GetAttr(__pyx_t_9, __pyx_n_s__tocsr); if (unlikely(!__pyx_t_8)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 467; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_8);
|
|
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
|
|
__pyx_t_9 = PyObject_Call(__pyx_t_8, ((PyObject *)__pyx_empty_tuple), NULL); if (unlikely(!__pyx_t_9)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 467; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_9);
|
|
__Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
|
|
__pyx_v_csgraph_T = __pyx_t_9;
|
|
__pyx_t_9 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":468
|
|
* else:
|
|
* csgraph_T = csgraph.T.tocsr()
|
|
* length = _depth_first_undirected(i_start, # <<<<<<<<<<<<<<
|
|
* csgraph.indices, csgraph.indptr,
|
|
* csgraph_T.indices, csgraph_T.indptr,
|
|
*/
|
|
__pyx_t_7 = __Pyx_PyInt_AsUnsignedInt(__pyx_v_i_start); if (unlikely((__pyx_t_7 == (unsigned int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 468; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":469
|
|
* csgraph_T = csgraph.T.tocsr()
|
|
* length = _depth_first_undirected(i_start,
|
|
* csgraph.indices, csgraph.indptr, # <<<<<<<<<<<<<<
|
|
* csgraph_T.indices, csgraph_T.indptr,
|
|
* node_list, predecessors,
|
|
*/
|
|
__pyx_t_9 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__indices); if (unlikely(!__pyx_t_9)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 469; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_9);
|
|
if (!(likely(((__pyx_t_9) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_9, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 469; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_8 = PyObject_GetAttr(__pyx_v_csgraph, __pyx_n_s__indptr); if (unlikely(!__pyx_t_8)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 469; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_8);
|
|
if (!(likely(((__pyx_t_8) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_8, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 469; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":470
|
|
* length = _depth_first_undirected(i_start,
|
|
* csgraph.indices, csgraph.indptr,
|
|
* csgraph_T.indices, csgraph_T.indptr, # <<<<<<<<<<<<<<
|
|
* node_list, predecessors,
|
|
* root_list, flag)
|
|
*/
|
|
__pyx_t_4 = PyObject_GetAttr(__pyx_v_csgraph_T, __pyx_n_s__indices); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 470; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
if (!(likely(((__pyx_t_4) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_4, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 470; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_2 = PyObject_GetAttr(__pyx_v_csgraph_T, __pyx_n_s__indptr); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 470; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 470; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":471
|
|
* csgraph.indices, csgraph.indptr,
|
|
* csgraph_T.indices, csgraph_T.indptr,
|
|
* node_list, predecessors, # <<<<<<<<<<<<<<
|
|
* root_list, flag)
|
|
*
|
|
*/
|
|
if (!(likely(((__pyx_v_node_list) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_node_list, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 471; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_3 = __pyx_v_node_list;
|
|
__Pyx_INCREF(__pyx_t_3);
|
|
if (!(likely(((__pyx_v_predecessors) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_predecessors, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 471; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_1 = __pyx_v_predecessors;
|
|
__Pyx_INCREF(__pyx_t_1);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":472
|
|
* csgraph_T.indices, csgraph_T.indptr,
|
|
* node_list, predecessors,
|
|
* root_list, flag) # <<<<<<<<<<<<<<
|
|
*
|
|
* if return_predecessors:
|
|
*/
|
|
if (!(likely(((__pyx_v_root_list) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_root_list, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 472; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_10 = __pyx_v_root_list;
|
|
__Pyx_INCREF(__pyx_t_10);
|
|
if (!(likely(((__pyx_v_flag) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_flag, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 472; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_11 = __pyx_v_flag;
|
|
__Pyx_INCREF(__pyx_t_11);
|
|
__pyx_v_length = __pyx_f_7sklearn_5utils_3mst_10_traversal__depth_first_undirected(__pyx_t_7, ((PyArrayObject *)__pyx_t_9), ((PyArrayObject *)__pyx_t_8), ((PyArrayObject *)__pyx_t_4), ((PyArrayObject *)__pyx_t_2), ((PyArrayObject *)__pyx_t_3), ((PyArrayObject *)__pyx_t_1), ((PyArrayObject *)__pyx_t_10), ((PyArrayObject *)__pyx_t_11));
|
|
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
|
|
__Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
|
|
__Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":474
|
|
* root_list, flag)
|
|
*
|
|
* if return_predecessors: # <<<<<<<<<<<<<<
|
|
* return node_list[:length], predecessors
|
|
* else:
|
|
*/
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_v_return_predecessors); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 474; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":475
|
|
*
|
|
* if return_predecessors:
|
|
* return node_list[:length], predecessors # <<<<<<<<<<<<<<
|
|
* else:
|
|
* return node_list[:length]
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_11 = __Pyx_PySequence_GetSlice(__pyx_v_node_list, 0, __pyx_v_length); if (unlikely(!__pyx_t_11)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 475; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_11);
|
|
__pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 475; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_10);
|
|
PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_t_11);
|
|
__Pyx_GIVEREF(__pyx_t_11);
|
|
__Pyx_INCREF(__pyx_v_predecessors);
|
|
PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_predecessors);
|
|
__Pyx_GIVEREF(__pyx_v_predecessors);
|
|
__pyx_t_11 = 0;
|
|
__pyx_r = ((PyObject *)__pyx_t_10);
|
|
__pyx_t_10 = 0;
|
|
goto __pyx_L0;
|
|
goto __pyx_L4;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":477
|
|
* return node_list[:length], predecessors
|
|
* else:
|
|
* return node_list[:length] # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_10 = __Pyx_PySequence_GetSlice(__pyx_v_node_list, 0, __pyx_v_length); if (unlikely(!__pyx_t_10)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 477; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_10);
|
|
__pyx_r = __pyx_t_10;
|
|
__pyx_t_10 = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
__pyx_L4:;
|
|
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_XDECREF(__pyx_t_8);
|
|
__Pyx_XDECREF(__pyx_t_9);
|
|
__Pyx_XDECREF(__pyx_t_10);
|
|
__Pyx_XDECREF(__pyx_t_11);
|
|
__Pyx_AddTraceback("sklearn.utils.mst._traversal.depth_first_order", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF(__pyx_v_node_list);
|
|
__Pyx_XDECREF(__pyx_v_predecessors);
|
|
__Pyx_XDECREF(__pyx_v_root_list);
|
|
__Pyx_XDECREF(__pyx_v_flag);
|
|
__Pyx_XDECREF(__pyx_v_csgraph_T);
|
|
__Pyx_XDECREF(__pyx_v_csgraph);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":480
|
|
*
|
|
*
|
|
* cdef unsigned int _depth_first_directed( # <<<<<<<<<<<<<<
|
|
* unsigned int head_node,
|
|
* np.ndarray[ITYPE_t, ndim=1, mode='c'] indices,
|
|
*/
|
|
|
|
static unsigned int __pyx_f_7sklearn_5utils_3mst_10_traversal__depth_first_directed(unsigned int __pyx_v_head_node, PyArrayObject *__pyx_v_indices, PyArrayObject *__pyx_v_indptr, PyArrayObject *__pyx_v_node_list, PyArrayObject *__pyx_v_predecessors, PyArrayObject *__pyx_v_root_list, PyArrayObject *__pyx_v_flag) {
|
|
unsigned int __pyx_v_i;
|
|
unsigned int __pyx_v_i_nl_end;
|
|
unsigned int __pyx_v_cnode;
|
|
unsigned int __pyx_v_pnode;
|
|
unsigned int __pyx_v_N;
|
|
int __pyx_v_no_children;
|
|
int __pyx_v_i_root;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_flag;
|
|
__Pyx_Buffer __pyx_pybuffer_flag;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indices;
|
|
__Pyx_Buffer __pyx_pybuffer_indices;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indptr;
|
|
__Pyx_Buffer __pyx_pybuffer_indptr;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_node_list;
|
|
__Pyx_Buffer __pyx_pybuffer_node_list;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_predecessors;
|
|
__Pyx_Buffer __pyx_pybuffer_predecessors;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_root_list;
|
|
__Pyx_Buffer __pyx_pybuffer_root_list;
|
|
unsigned int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
long __pyx_t_1;
|
|
int __pyx_t_2;
|
|
long __pyx_t_3;
|
|
unsigned int __pyx_t_4;
|
|
int __pyx_t_5;
|
|
int __pyx_t_6;
|
|
unsigned int __pyx_t_7;
|
|
long __pyx_t_8;
|
|
__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t __pyx_t_9;
|
|
unsigned int __pyx_t_10;
|
|
unsigned int __pyx_t_11;
|
|
__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t __pyx_t_12;
|
|
int __pyx_t_13;
|
|
unsigned int __pyx_t_14;
|
|
unsigned int __pyx_t_15;
|
|
unsigned int __pyx_t_16;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_depth_first_directed", 0);
|
|
__pyx_pybuffer_indices.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indices.refcount = 0;
|
|
__pyx_pybuffernd_indices.data = NULL;
|
|
__pyx_pybuffernd_indices.rcbuffer = &__pyx_pybuffer_indices;
|
|
__pyx_pybuffer_indptr.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indptr.refcount = 0;
|
|
__pyx_pybuffernd_indptr.data = NULL;
|
|
__pyx_pybuffernd_indptr.rcbuffer = &__pyx_pybuffer_indptr;
|
|
__pyx_pybuffer_node_list.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_node_list.refcount = 0;
|
|
__pyx_pybuffernd_node_list.data = NULL;
|
|
__pyx_pybuffernd_node_list.rcbuffer = &__pyx_pybuffer_node_list;
|
|
__pyx_pybuffer_predecessors.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_predecessors.refcount = 0;
|
|
__pyx_pybuffernd_predecessors.data = NULL;
|
|
__pyx_pybuffernd_predecessors.rcbuffer = &__pyx_pybuffer_predecessors;
|
|
__pyx_pybuffer_root_list.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_root_list.refcount = 0;
|
|
__pyx_pybuffernd_root_list.data = NULL;
|
|
__pyx_pybuffernd_root_list.rcbuffer = &__pyx_pybuffer_root_list;
|
|
__pyx_pybuffer_flag.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_flag.refcount = 0;
|
|
__pyx_pybuffernd_flag.data = NULL;
|
|
__pyx_pybuffernd_flag.rcbuffer = &__pyx_pybuffer_flag;
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indices.rcbuffer->pybuffer, (PyObject*)__pyx_v_indices, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 480; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indices.diminfo[0].strides = __pyx_pybuffernd_indices.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indices.diminfo[0].shape = __pyx_pybuffernd_indices.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indptr.rcbuffer->pybuffer, (PyObject*)__pyx_v_indptr, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 480; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indptr.diminfo[0].strides = __pyx_pybuffernd_indptr.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indptr.diminfo[0].shape = __pyx_pybuffernd_indptr.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_node_list.rcbuffer->pybuffer, (PyObject*)__pyx_v_node_list, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 480; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_node_list.diminfo[0].strides = __pyx_pybuffernd_node_list.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_node_list.diminfo[0].shape = __pyx_pybuffernd_node_list.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_predecessors.rcbuffer->pybuffer, (PyObject*)__pyx_v_predecessors, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 480; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_predecessors.diminfo[0].strides = __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_predecessors.diminfo[0].shape = __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_root_list.rcbuffer->pybuffer, (PyObject*)__pyx_v_root_list, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 480; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_root_list.diminfo[0].strides = __pyx_pybuffernd_root_list.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_root_list.diminfo[0].shape = __pyx_pybuffernd_root_list.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_flag.rcbuffer->pybuffer, (PyObject*)__pyx_v_flag, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 480; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_flag.diminfo[0].strides = __pyx_pybuffernd_flag.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_flag.diminfo[0].shape = __pyx_pybuffernd_flag.rcbuffer->pybuffer.shape[0];
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":489
|
|
* np.ndarray[ITYPE_t, ndim=1, mode='c'] flag):
|
|
* cdef unsigned int i, j, i_nl_end, cnode, pnode
|
|
* cdef unsigned int N = node_list.shape[0] # <<<<<<<<<<<<<<
|
|
* cdef int no_children, i_root
|
|
*
|
|
*/
|
|
__pyx_v_N = (__pyx_v_node_list->dimensions[0]);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":492
|
|
* cdef int no_children, i_root
|
|
*
|
|
* node_list[0] = head_node # <<<<<<<<<<<<<<
|
|
* root_list[0] = head_node
|
|
* i_root = 0
|
|
*/
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_2 = -1;
|
|
if (__pyx_t_1 < 0) {
|
|
__pyx_t_1 += __pyx_pybuffernd_node_list.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_1 < 0)) __pyx_t_2 = 0;
|
|
} else if (unlikely(__pyx_t_1 >= __pyx_pybuffernd_node_list.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 492; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_node_list.rcbuffer->pybuffer.buf, __pyx_t_1, __pyx_pybuffernd_node_list.diminfo[0].strides) = __pyx_v_head_node;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":493
|
|
*
|
|
* node_list[0] = head_node
|
|
* root_list[0] = head_node # <<<<<<<<<<<<<<
|
|
* i_root = 0
|
|
* i_nl_end = 1
|
|
*/
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_2 = -1;
|
|
if (__pyx_t_3 < 0) {
|
|
__pyx_t_3 += __pyx_pybuffernd_root_list.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_3 < 0)) __pyx_t_2 = 0;
|
|
} else if (unlikely(__pyx_t_3 >= __pyx_pybuffernd_root_list.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 493; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_root_list.rcbuffer->pybuffer.buf, __pyx_t_3, __pyx_pybuffernd_root_list.diminfo[0].strides) = __pyx_v_head_node;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":494
|
|
* node_list[0] = head_node
|
|
* root_list[0] = head_node
|
|
* i_root = 0 # <<<<<<<<<<<<<<
|
|
* i_nl_end = 1
|
|
* flag[head_node] = 1
|
|
*/
|
|
__pyx_v_i_root = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":495
|
|
* root_list[0] = head_node
|
|
* i_root = 0
|
|
* i_nl_end = 1 # <<<<<<<<<<<<<<
|
|
* flag[head_node] = 1
|
|
*
|
|
*/
|
|
__pyx_v_i_nl_end = 1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":496
|
|
* i_root = 0
|
|
* i_nl_end = 1
|
|
* flag[head_node] = 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* while i_root >= 0:
|
|
*/
|
|
__pyx_t_4 = __pyx_v_head_node;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_4 >= (size_t)__pyx_pybuffernd_flag.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 496; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_flag.rcbuffer->pybuffer.buf, __pyx_t_4, __pyx_pybuffernd_flag.diminfo[0].strides) = 1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":498
|
|
* flag[head_node] = 1
|
|
*
|
|
* while i_root >= 0: # <<<<<<<<<<<<<<
|
|
* pnode = root_list[i_root]
|
|
* no_children = True
|
|
*/
|
|
while (1) {
|
|
__pyx_t_5 = (__pyx_v_i_root >= 0);
|
|
if (!__pyx_t_5) break;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":499
|
|
*
|
|
* while i_root >= 0:
|
|
* pnode = root_list[i_root] # <<<<<<<<<<<<<<
|
|
* no_children = True
|
|
* for i from indptr[pnode] <= i < indptr[pnode + 1]:
|
|
*/
|
|
__pyx_t_2 = __pyx_v_i_root;
|
|
__pyx_t_6 = -1;
|
|
if (__pyx_t_2 < 0) {
|
|
__pyx_t_2 += __pyx_pybuffernd_root_list.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_2 < 0)) __pyx_t_6 = 0;
|
|
} else if (unlikely(__pyx_t_2 >= __pyx_pybuffernd_root_list.diminfo[0].shape)) __pyx_t_6 = 0;
|
|
if (unlikely(__pyx_t_6 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_6);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 499; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_pnode = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_root_list.rcbuffer->pybuffer.buf, __pyx_t_2, __pyx_pybuffernd_root_list.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":500
|
|
* while i_root >= 0:
|
|
* pnode = root_list[i_root]
|
|
* no_children = True # <<<<<<<<<<<<<<
|
|
* for i from indptr[pnode] <= i < indptr[pnode + 1]:
|
|
* cnode = indices[i]
|
|
*/
|
|
__pyx_v_no_children = 1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":501
|
|
* pnode = root_list[i_root]
|
|
* no_children = True
|
|
* for i from indptr[pnode] <= i < indptr[pnode + 1]: # <<<<<<<<<<<<<<
|
|
* cnode = indices[i]
|
|
* if flag[cnode]:
|
|
*/
|
|
__pyx_t_7 = __pyx_v_pnode;
|
|
__pyx_t_6 = -1;
|
|
if (unlikely(__pyx_t_7 >= (size_t)__pyx_pybuffernd_indptr.diminfo[0].shape)) __pyx_t_6 = 0;
|
|
if (unlikely(__pyx_t_6 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_6);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 501; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_8 = (__pyx_v_pnode + 1);
|
|
__pyx_t_6 = -1;
|
|
if (__pyx_t_8 < 0) {
|
|
__pyx_t_8 += __pyx_pybuffernd_indptr.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_8 < 0)) __pyx_t_6 = 0;
|
|
} else if (unlikely(__pyx_t_8 >= __pyx_pybuffernd_indptr.diminfo[0].shape)) __pyx_t_6 = 0;
|
|
if (unlikely(__pyx_t_6 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_6);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 501; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_9 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr.rcbuffer->pybuffer.buf, __pyx_t_8, __pyx_pybuffernd_indptr.diminfo[0].strides));
|
|
for (__pyx_v_i = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr.rcbuffer->pybuffer.buf, __pyx_t_7, __pyx_pybuffernd_indptr.diminfo[0].strides)); __pyx_v_i < __pyx_t_9; __pyx_v_i++) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":502
|
|
* no_children = True
|
|
* for i from indptr[pnode] <= i < indptr[pnode + 1]:
|
|
* cnode = indices[i] # <<<<<<<<<<<<<<
|
|
* if flag[cnode]:
|
|
* continue
|
|
*/
|
|
__pyx_t_10 = __pyx_v_i;
|
|
__pyx_t_6 = -1;
|
|
if (unlikely(__pyx_t_10 >= (size_t)__pyx_pybuffernd_indices.diminfo[0].shape)) __pyx_t_6 = 0;
|
|
if (unlikely(__pyx_t_6 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_6);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 502; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_cnode = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indices.rcbuffer->pybuffer.buf, __pyx_t_10, __pyx_pybuffernd_indices.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":503
|
|
* for i from indptr[pnode] <= i < indptr[pnode + 1]:
|
|
* cnode = indices[i]
|
|
* if flag[cnode]: # <<<<<<<<<<<<<<
|
|
* continue
|
|
* else:
|
|
*/
|
|
__pyx_t_11 = __pyx_v_cnode;
|
|
__pyx_t_6 = -1;
|
|
if (unlikely(__pyx_t_11 >= (size_t)__pyx_pybuffernd_flag.diminfo[0].shape)) __pyx_t_6 = 0;
|
|
if (unlikely(__pyx_t_6 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_6);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 503; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_12 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_flag.rcbuffer->pybuffer.buf, __pyx_t_11, __pyx_pybuffernd_flag.diminfo[0].strides));
|
|
if (__pyx_t_12) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":504
|
|
* cnode = indices[i]
|
|
* if flag[cnode]:
|
|
* continue # <<<<<<<<<<<<<<
|
|
* else:
|
|
* i_root += 1
|
|
*/
|
|
goto __pyx_L5_continue;
|
|
goto __pyx_L7;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":506
|
|
* continue
|
|
* else:
|
|
* i_root += 1 # <<<<<<<<<<<<<<
|
|
* root_list[i_root] = cnode
|
|
* node_list[i_nl_end] = cnode
|
|
*/
|
|
__pyx_v_i_root = (__pyx_v_i_root + 1);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":507
|
|
* else:
|
|
* i_root += 1
|
|
* root_list[i_root] = cnode # <<<<<<<<<<<<<<
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode
|
|
*/
|
|
__pyx_t_6 = __pyx_v_i_root;
|
|
__pyx_t_13 = -1;
|
|
if (__pyx_t_6 < 0) {
|
|
__pyx_t_6 += __pyx_pybuffernd_root_list.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_6 < 0)) __pyx_t_13 = 0;
|
|
} else if (unlikely(__pyx_t_6 >= __pyx_pybuffernd_root_list.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 507; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_root_list.rcbuffer->pybuffer.buf, __pyx_t_6, __pyx_pybuffernd_root_list.diminfo[0].strides) = __pyx_v_cnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":508
|
|
* i_root += 1
|
|
* root_list[i_root] = cnode
|
|
* node_list[i_nl_end] = cnode # <<<<<<<<<<<<<<
|
|
* predecessors[cnode] = pnode
|
|
* flag[cnode] = 1
|
|
*/
|
|
__pyx_t_14 = __pyx_v_i_nl_end;
|
|
__pyx_t_13 = -1;
|
|
if (unlikely(__pyx_t_14 >= (size_t)__pyx_pybuffernd_node_list.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 508; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_node_list.rcbuffer->pybuffer.buf, __pyx_t_14, __pyx_pybuffernd_node_list.diminfo[0].strides) = __pyx_v_cnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":509
|
|
* root_list[i_root] = cnode
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode # <<<<<<<<<<<<<<
|
|
* flag[cnode] = 1
|
|
* i_nl_end += 1
|
|
*/
|
|
__pyx_t_15 = __pyx_v_cnode;
|
|
__pyx_t_13 = -1;
|
|
if (unlikely(__pyx_t_15 >= (size_t)__pyx_pybuffernd_predecessors.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 509; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.buf, __pyx_t_15, __pyx_pybuffernd_predecessors.diminfo[0].strides) = __pyx_v_pnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":510
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode
|
|
* flag[cnode] = 1 # <<<<<<<<<<<<<<
|
|
* i_nl_end += 1
|
|
* no_children = False
|
|
*/
|
|
__pyx_t_16 = __pyx_v_cnode;
|
|
__pyx_t_13 = -1;
|
|
if (unlikely(__pyx_t_16 >= (size_t)__pyx_pybuffernd_flag.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 510; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_flag.rcbuffer->pybuffer.buf, __pyx_t_16, __pyx_pybuffernd_flag.diminfo[0].strides) = 1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":511
|
|
* predecessors[cnode] = pnode
|
|
* flag[cnode] = 1
|
|
* i_nl_end += 1 # <<<<<<<<<<<<<<
|
|
* no_children = False
|
|
* break
|
|
*/
|
|
__pyx_v_i_nl_end = (__pyx_v_i_nl_end + 1);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":512
|
|
* flag[cnode] = 1
|
|
* i_nl_end += 1
|
|
* no_children = False # <<<<<<<<<<<<<<
|
|
* break
|
|
*
|
|
*/
|
|
__pyx_v_no_children = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":513
|
|
* i_nl_end += 1
|
|
* no_children = False
|
|
* break # <<<<<<<<<<<<<<
|
|
*
|
|
* if i_nl_end == N:
|
|
*/
|
|
goto __pyx_L6_break;
|
|
}
|
|
__pyx_L7:;
|
|
__pyx_L5_continue:;
|
|
}
|
|
__pyx_L6_break:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":515
|
|
* break
|
|
*
|
|
* if i_nl_end == N: # <<<<<<<<<<<<<<
|
|
* break
|
|
*
|
|
*/
|
|
__pyx_t_5 = (__pyx_v_i_nl_end == __pyx_v_N);
|
|
if (__pyx_t_5) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":516
|
|
*
|
|
* if i_nl_end == N:
|
|
* break # <<<<<<<<<<<<<<
|
|
*
|
|
* if no_children:
|
|
*/
|
|
goto __pyx_L4_break;
|
|
goto __pyx_L8;
|
|
}
|
|
__pyx_L8:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":518
|
|
* break
|
|
*
|
|
* if no_children: # <<<<<<<<<<<<<<
|
|
* i_root -= 1
|
|
*
|
|
*/
|
|
if (__pyx_v_no_children) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":519
|
|
*
|
|
* if no_children:
|
|
* i_root -= 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* return i_nl_end
|
|
*/
|
|
__pyx_v_i_root = (__pyx_v_i_root - 1);
|
|
goto __pyx_L9;
|
|
}
|
|
__pyx_L9:;
|
|
}
|
|
__pyx_L4_break:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":521
|
|
* i_root -= 1
|
|
*
|
|
* return i_nl_end # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = __pyx_v_i_nl_end;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_flag.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_node_list.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_predecessors.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_root_list.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_WriteUnraisable("sklearn.utils.mst._traversal._depth_first_directed", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_flag.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_node_list.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_predecessors.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_root_list.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":524
|
|
*
|
|
*
|
|
* cdef unsigned int _depth_first_undirected( # <<<<<<<<<<<<<<
|
|
* unsigned int head_node,
|
|
* np.ndarray[ITYPE_t, ndim=1, mode='c'] indices1,
|
|
*/
|
|
|
|
static unsigned int __pyx_f_7sklearn_5utils_3mst_10_traversal__depth_first_undirected(unsigned int __pyx_v_head_node, PyArrayObject *__pyx_v_indices1, PyArrayObject *__pyx_v_indptr1, PyArrayObject *__pyx_v_indices2, PyArrayObject *__pyx_v_indptr2, PyArrayObject *__pyx_v_node_list, PyArrayObject *__pyx_v_predecessors, PyArrayObject *__pyx_v_root_list, PyArrayObject *__pyx_v_flag) {
|
|
unsigned int __pyx_v_i;
|
|
unsigned int __pyx_v_i_nl_end;
|
|
unsigned int __pyx_v_cnode;
|
|
unsigned int __pyx_v_pnode;
|
|
unsigned int __pyx_v_N;
|
|
int __pyx_v_no_children;
|
|
int __pyx_v_i_root;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_flag;
|
|
__Pyx_Buffer __pyx_pybuffer_flag;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indices1;
|
|
__Pyx_Buffer __pyx_pybuffer_indices1;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indices2;
|
|
__Pyx_Buffer __pyx_pybuffer_indices2;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indptr1;
|
|
__Pyx_Buffer __pyx_pybuffer_indptr1;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indptr2;
|
|
__Pyx_Buffer __pyx_pybuffer_indptr2;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_node_list;
|
|
__Pyx_Buffer __pyx_pybuffer_node_list;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_predecessors;
|
|
__Pyx_Buffer __pyx_pybuffer_predecessors;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_root_list;
|
|
__Pyx_Buffer __pyx_pybuffer_root_list;
|
|
unsigned int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
long __pyx_t_1;
|
|
int __pyx_t_2;
|
|
long __pyx_t_3;
|
|
unsigned int __pyx_t_4;
|
|
int __pyx_t_5;
|
|
int __pyx_t_6;
|
|
unsigned int __pyx_t_7;
|
|
long __pyx_t_8;
|
|
__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t __pyx_t_9;
|
|
unsigned int __pyx_t_10;
|
|
unsigned int __pyx_t_11;
|
|
__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t __pyx_t_12;
|
|
int __pyx_t_13;
|
|
unsigned int __pyx_t_14;
|
|
unsigned int __pyx_t_15;
|
|
unsigned int __pyx_t_16;
|
|
unsigned int __pyx_t_17;
|
|
long __pyx_t_18;
|
|
unsigned int __pyx_t_19;
|
|
unsigned int __pyx_t_20;
|
|
int __pyx_t_21;
|
|
unsigned int __pyx_t_22;
|
|
unsigned int __pyx_t_23;
|
|
unsigned int __pyx_t_24;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_depth_first_undirected", 0);
|
|
__pyx_pybuffer_indices1.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indices1.refcount = 0;
|
|
__pyx_pybuffernd_indices1.data = NULL;
|
|
__pyx_pybuffernd_indices1.rcbuffer = &__pyx_pybuffer_indices1;
|
|
__pyx_pybuffer_indptr1.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indptr1.refcount = 0;
|
|
__pyx_pybuffernd_indptr1.data = NULL;
|
|
__pyx_pybuffernd_indptr1.rcbuffer = &__pyx_pybuffer_indptr1;
|
|
__pyx_pybuffer_indices2.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indices2.refcount = 0;
|
|
__pyx_pybuffernd_indices2.data = NULL;
|
|
__pyx_pybuffernd_indices2.rcbuffer = &__pyx_pybuffer_indices2;
|
|
__pyx_pybuffer_indptr2.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indptr2.refcount = 0;
|
|
__pyx_pybuffernd_indptr2.data = NULL;
|
|
__pyx_pybuffernd_indptr2.rcbuffer = &__pyx_pybuffer_indptr2;
|
|
__pyx_pybuffer_node_list.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_node_list.refcount = 0;
|
|
__pyx_pybuffernd_node_list.data = NULL;
|
|
__pyx_pybuffernd_node_list.rcbuffer = &__pyx_pybuffer_node_list;
|
|
__pyx_pybuffer_predecessors.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_predecessors.refcount = 0;
|
|
__pyx_pybuffernd_predecessors.data = NULL;
|
|
__pyx_pybuffernd_predecessors.rcbuffer = &__pyx_pybuffer_predecessors;
|
|
__pyx_pybuffer_root_list.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_root_list.refcount = 0;
|
|
__pyx_pybuffernd_root_list.data = NULL;
|
|
__pyx_pybuffernd_root_list.rcbuffer = &__pyx_pybuffer_root_list;
|
|
__pyx_pybuffer_flag.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_flag.refcount = 0;
|
|
__pyx_pybuffernd_flag.data = NULL;
|
|
__pyx_pybuffernd_flag.rcbuffer = &__pyx_pybuffer_flag;
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indices1.rcbuffer->pybuffer, (PyObject*)__pyx_v_indices1, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 524; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indices1.diminfo[0].strides = __pyx_pybuffernd_indices1.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indices1.diminfo[0].shape = __pyx_pybuffernd_indices1.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indptr1.rcbuffer->pybuffer, (PyObject*)__pyx_v_indptr1, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 524; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indptr1.diminfo[0].strides = __pyx_pybuffernd_indptr1.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indptr1.diminfo[0].shape = __pyx_pybuffernd_indptr1.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indices2.rcbuffer->pybuffer, (PyObject*)__pyx_v_indices2, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 524; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indices2.diminfo[0].strides = __pyx_pybuffernd_indices2.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indices2.diminfo[0].shape = __pyx_pybuffernd_indices2.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indptr2.rcbuffer->pybuffer, (PyObject*)__pyx_v_indptr2, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 524; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indptr2.diminfo[0].strides = __pyx_pybuffernd_indptr2.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indptr2.diminfo[0].shape = __pyx_pybuffernd_indptr2.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_node_list.rcbuffer->pybuffer, (PyObject*)__pyx_v_node_list, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 524; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_node_list.diminfo[0].strides = __pyx_pybuffernd_node_list.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_node_list.diminfo[0].shape = __pyx_pybuffernd_node_list.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_predecessors.rcbuffer->pybuffer, (PyObject*)__pyx_v_predecessors, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 524; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_predecessors.diminfo[0].strides = __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_predecessors.diminfo[0].shape = __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_root_list.rcbuffer->pybuffer, (PyObject*)__pyx_v_root_list, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 524; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_root_list.diminfo[0].strides = __pyx_pybuffernd_root_list.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_root_list.diminfo[0].shape = __pyx_pybuffernd_root_list.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_flag.rcbuffer->pybuffer, (PyObject*)__pyx_v_flag, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 524; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_flag.diminfo[0].strides = __pyx_pybuffernd_flag.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_flag.diminfo[0].shape = __pyx_pybuffernd_flag.rcbuffer->pybuffer.shape[0];
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":535
|
|
* np.ndarray[ITYPE_t, ndim=1, mode='c'] flag):
|
|
* cdef unsigned int i, j, i_nl_end, cnode, pnode
|
|
* cdef unsigned int N = node_list.shape[0] # <<<<<<<<<<<<<<
|
|
* cdef int no_children, i_root
|
|
*
|
|
*/
|
|
__pyx_v_N = (__pyx_v_node_list->dimensions[0]);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":538
|
|
* cdef int no_children, i_root
|
|
*
|
|
* node_list[0] = head_node # <<<<<<<<<<<<<<
|
|
* root_list[0] = head_node
|
|
* i_root = 0
|
|
*/
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_2 = -1;
|
|
if (__pyx_t_1 < 0) {
|
|
__pyx_t_1 += __pyx_pybuffernd_node_list.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_1 < 0)) __pyx_t_2 = 0;
|
|
} else if (unlikely(__pyx_t_1 >= __pyx_pybuffernd_node_list.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 538; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_node_list.rcbuffer->pybuffer.buf, __pyx_t_1, __pyx_pybuffernd_node_list.diminfo[0].strides) = __pyx_v_head_node;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":539
|
|
*
|
|
* node_list[0] = head_node
|
|
* root_list[0] = head_node # <<<<<<<<<<<<<<
|
|
* i_root = 0
|
|
* i_nl_end = 1
|
|
*/
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_2 = -1;
|
|
if (__pyx_t_3 < 0) {
|
|
__pyx_t_3 += __pyx_pybuffernd_root_list.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_3 < 0)) __pyx_t_2 = 0;
|
|
} else if (unlikely(__pyx_t_3 >= __pyx_pybuffernd_root_list.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 539; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_root_list.rcbuffer->pybuffer.buf, __pyx_t_3, __pyx_pybuffernd_root_list.diminfo[0].strides) = __pyx_v_head_node;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":540
|
|
* node_list[0] = head_node
|
|
* root_list[0] = head_node
|
|
* i_root = 0 # <<<<<<<<<<<<<<
|
|
* i_nl_end = 1
|
|
* flag[head_node] = 1
|
|
*/
|
|
__pyx_v_i_root = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":541
|
|
* root_list[0] = head_node
|
|
* i_root = 0
|
|
* i_nl_end = 1 # <<<<<<<<<<<<<<
|
|
* flag[head_node] = 1
|
|
*
|
|
*/
|
|
__pyx_v_i_nl_end = 1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":542
|
|
* i_root = 0
|
|
* i_nl_end = 1
|
|
* flag[head_node] = 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* while i_root >= 0:
|
|
*/
|
|
__pyx_t_4 = __pyx_v_head_node;
|
|
__pyx_t_2 = -1;
|
|
if (unlikely(__pyx_t_4 >= (size_t)__pyx_pybuffernd_flag.diminfo[0].shape)) __pyx_t_2 = 0;
|
|
if (unlikely(__pyx_t_2 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_2);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 542; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_flag.rcbuffer->pybuffer.buf, __pyx_t_4, __pyx_pybuffernd_flag.diminfo[0].strides) = 1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":544
|
|
* flag[head_node] = 1
|
|
*
|
|
* while i_root >= 0: # <<<<<<<<<<<<<<
|
|
* pnode = root_list[i_root]
|
|
* no_children = True
|
|
*/
|
|
while (1) {
|
|
__pyx_t_5 = (__pyx_v_i_root >= 0);
|
|
if (!__pyx_t_5) break;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":545
|
|
*
|
|
* while i_root >= 0:
|
|
* pnode = root_list[i_root] # <<<<<<<<<<<<<<
|
|
* no_children = True
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_i_root;
|
|
__pyx_t_6 = -1;
|
|
if (__pyx_t_2 < 0) {
|
|
__pyx_t_2 += __pyx_pybuffernd_root_list.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_2 < 0)) __pyx_t_6 = 0;
|
|
} else if (unlikely(__pyx_t_2 >= __pyx_pybuffernd_root_list.diminfo[0].shape)) __pyx_t_6 = 0;
|
|
if (unlikely(__pyx_t_6 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_6);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 545; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_pnode = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_root_list.rcbuffer->pybuffer.buf, __pyx_t_2, __pyx_pybuffernd_root_list.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":546
|
|
* while i_root >= 0:
|
|
* pnode = root_list[i_root]
|
|
* no_children = True # <<<<<<<<<<<<<<
|
|
*
|
|
* for i from indptr1[pnode] <= i < indptr1[pnode + 1]:
|
|
*/
|
|
__pyx_v_no_children = 1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":548
|
|
* no_children = True
|
|
*
|
|
* for i from indptr1[pnode] <= i < indptr1[pnode + 1]: # <<<<<<<<<<<<<<
|
|
* cnode = indices1[i]
|
|
* if flag[cnode]:
|
|
*/
|
|
__pyx_t_7 = __pyx_v_pnode;
|
|
__pyx_t_6 = -1;
|
|
if (unlikely(__pyx_t_7 >= (size_t)__pyx_pybuffernd_indptr1.diminfo[0].shape)) __pyx_t_6 = 0;
|
|
if (unlikely(__pyx_t_6 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_6);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 548; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_8 = (__pyx_v_pnode + 1);
|
|
__pyx_t_6 = -1;
|
|
if (__pyx_t_8 < 0) {
|
|
__pyx_t_8 += __pyx_pybuffernd_indptr1.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_8 < 0)) __pyx_t_6 = 0;
|
|
} else if (unlikely(__pyx_t_8 >= __pyx_pybuffernd_indptr1.diminfo[0].shape)) __pyx_t_6 = 0;
|
|
if (unlikely(__pyx_t_6 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_6);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 548; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_9 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr1.rcbuffer->pybuffer.buf, __pyx_t_8, __pyx_pybuffernd_indptr1.diminfo[0].strides));
|
|
for (__pyx_v_i = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr1.rcbuffer->pybuffer.buf, __pyx_t_7, __pyx_pybuffernd_indptr1.diminfo[0].strides)); __pyx_v_i < __pyx_t_9; __pyx_v_i++) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":549
|
|
*
|
|
* for i from indptr1[pnode] <= i < indptr1[pnode + 1]:
|
|
* cnode = indices1[i] # <<<<<<<<<<<<<<
|
|
* if flag[cnode]:
|
|
* continue
|
|
*/
|
|
__pyx_t_10 = __pyx_v_i;
|
|
__pyx_t_6 = -1;
|
|
if (unlikely(__pyx_t_10 >= (size_t)__pyx_pybuffernd_indices1.diminfo[0].shape)) __pyx_t_6 = 0;
|
|
if (unlikely(__pyx_t_6 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_6);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 549; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_cnode = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indices1.rcbuffer->pybuffer.buf, __pyx_t_10, __pyx_pybuffernd_indices1.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":550
|
|
* for i from indptr1[pnode] <= i < indptr1[pnode + 1]:
|
|
* cnode = indices1[i]
|
|
* if flag[cnode]: # <<<<<<<<<<<<<<
|
|
* continue
|
|
* else:
|
|
*/
|
|
__pyx_t_11 = __pyx_v_cnode;
|
|
__pyx_t_6 = -1;
|
|
if (unlikely(__pyx_t_11 >= (size_t)__pyx_pybuffernd_flag.diminfo[0].shape)) __pyx_t_6 = 0;
|
|
if (unlikely(__pyx_t_6 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_6);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 550; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_12 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_flag.rcbuffer->pybuffer.buf, __pyx_t_11, __pyx_pybuffernd_flag.diminfo[0].strides));
|
|
if (__pyx_t_12) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":551
|
|
* cnode = indices1[i]
|
|
* if flag[cnode]:
|
|
* continue # <<<<<<<<<<<<<<
|
|
* else:
|
|
* i_root += 1
|
|
*/
|
|
goto __pyx_L5_continue;
|
|
goto __pyx_L7;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":553
|
|
* continue
|
|
* else:
|
|
* i_root += 1 # <<<<<<<<<<<<<<
|
|
* root_list[i_root] = cnode
|
|
* node_list[i_nl_end] = cnode
|
|
*/
|
|
__pyx_v_i_root = (__pyx_v_i_root + 1);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":554
|
|
* else:
|
|
* i_root += 1
|
|
* root_list[i_root] = cnode # <<<<<<<<<<<<<<
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode
|
|
*/
|
|
__pyx_t_6 = __pyx_v_i_root;
|
|
__pyx_t_13 = -1;
|
|
if (__pyx_t_6 < 0) {
|
|
__pyx_t_6 += __pyx_pybuffernd_root_list.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_6 < 0)) __pyx_t_13 = 0;
|
|
} else if (unlikely(__pyx_t_6 >= __pyx_pybuffernd_root_list.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 554; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_root_list.rcbuffer->pybuffer.buf, __pyx_t_6, __pyx_pybuffernd_root_list.diminfo[0].strides) = __pyx_v_cnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":555
|
|
* i_root += 1
|
|
* root_list[i_root] = cnode
|
|
* node_list[i_nl_end] = cnode # <<<<<<<<<<<<<<
|
|
* predecessors[cnode] = pnode
|
|
* flag[cnode] = 1
|
|
*/
|
|
__pyx_t_14 = __pyx_v_i_nl_end;
|
|
__pyx_t_13 = -1;
|
|
if (unlikely(__pyx_t_14 >= (size_t)__pyx_pybuffernd_node_list.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 555; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_node_list.rcbuffer->pybuffer.buf, __pyx_t_14, __pyx_pybuffernd_node_list.diminfo[0].strides) = __pyx_v_cnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":556
|
|
* root_list[i_root] = cnode
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode # <<<<<<<<<<<<<<
|
|
* flag[cnode] = 1
|
|
* i_nl_end += 1
|
|
*/
|
|
__pyx_t_15 = __pyx_v_cnode;
|
|
__pyx_t_13 = -1;
|
|
if (unlikely(__pyx_t_15 >= (size_t)__pyx_pybuffernd_predecessors.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 556; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.buf, __pyx_t_15, __pyx_pybuffernd_predecessors.diminfo[0].strides) = __pyx_v_pnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":557
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode
|
|
* flag[cnode] = 1 # <<<<<<<<<<<<<<
|
|
* i_nl_end += 1
|
|
* no_children = False
|
|
*/
|
|
__pyx_t_16 = __pyx_v_cnode;
|
|
__pyx_t_13 = -1;
|
|
if (unlikely(__pyx_t_16 >= (size_t)__pyx_pybuffernd_flag.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_flag.rcbuffer->pybuffer.buf, __pyx_t_16, __pyx_pybuffernd_flag.diminfo[0].strides) = 1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":558
|
|
* predecessors[cnode] = pnode
|
|
* flag[cnode] = 1
|
|
* i_nl_end += 1 # <<<<<<<<<<<<<<
|
|
* no_children = False
|
|
* break
|
|
*/
|
|
__pyx_v_i_nl_end = (__pyx_v_i_nl_end + 1);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":559
|
|
* flag[cnode] = 1
|
|
* i_nl_end += 1
|
|
* no_children = False # <<<<<<<<<<<<<<
|
|
* break
|
|
*
|
|
*/
|
|
__pyx_v_no_children = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":560
|
|
* i_nl_end += 1
|
|
* no_children = False
|
|
* break # <<<<<<<<<<<<<<
|
|
*
|
|
* if no_children:
|
|
*/
|
|
goto __pyx_L6_break;
|
|
}
|
|
__pyx_L7:;
|
|
__pyx_L5_continue:;
|
|
}
|
|
__pyx_L6_break:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":562
|
|
* break
|
|
*
|
|
* if no_children: # <<<<<<<<<<<<<<
|
|
* for i from indptr2[pnode] <= i < indptr2[pnode + 1]:
|
|
* cnode = indices2[i]
|
|
*/
|
|
if (__pyx_v_no_children) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":563
|
|
*
|
|
* if no_children:
|
|
* for i from indptr2[pnode] <= i < indptr2[pnode + 1]: # <<<<<<<<<<<<<<
|
|
* cnode = indices2[i]
|
|
* if flag[cnode]:
|
|
*/
|
|
__pyx_t_17 = __pyx_v_pnode;
|
|
__pyx_t_13 = -1;
|
|
if (unlikely(__pyx_t_17 >= (size_t)__pyx_pybuffernd_indptr2.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 563; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_18 = (__pyx_v_pnode + 1);
|
|
__pyx_t_13 = -1;
|
|
if (__pyx_t_18 < 0) {
|
|
__pyx_t_18 += __pyx_pybuffernd_indptr2.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_18 < 0)) __pyx_t_13 = 0;
|
|
} else if (unlikely(__pyx_t_18 >= __pyx_pybuffernd_indptr2.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 563; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_9 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr2.rcbuffer->pybuffer.buf, __pyx_t_18, __pyx_pybuffernd_indptr2.diminfo[0].strides));
|
|
for (__pyx_v_i = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr2.rcbuffer->pybuffer.buf, __pyx_t_17, __pyx_pybuffernd_indptr2.diminfo[0].strides)); __pyx_v_i < __pyx_t_9; __pyx_v_i++) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":564
|
|
* if no_children:
|
|
* for i from indptr2[pnode] <= i < indptr2[pnode + 1]:
|
|
* cnode = indices2[i] # <<<<<<<<<<<<<<
|
|
* if flag[cnode]:
|
|
* continue
|
|
*/
|
|
__pyx_t_19 = __pyx_v_i;
|
|
__pyx_t_13 = -1;
|
|
if (unlikely(__pyx_t_19 >= (size_t)__pyx_pybuffernd_indices2.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 564; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_cnode = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indices2.rcbuffer->pybuffer.buf, __pyx_t_19, __pyx_pybuffernd_indices2.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":565
|
|
* for i from indptr2[pnode] <= i < indptr2[pnode + 1]:
|
|
* cnode = indices2[i]
|
|
* if flag[cnode]: # <<<<<<<<<<<<<<
|
|
* continue
|
|
* else:
|
|
*/
|
|
__pyx_t_20 = __pyx_v_cnode;
|
|
__pyx_t_13 = -1;
|
|
if (unlikely(__pyx_t_20 >= (size_t)__pyx_pybuffernd_flag.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 565; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_12 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_flag.rcbuffer->pybuffer.buf, __pyx_t_20, __pyx_pybuffernd_flag.diminfo[0].strides));
|
|
if (__pyx_t_12) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":566
|
|
* cnode = indices2[i]
|
|
* if flag[cnode]:
|
|
* continue # <<<<<<<<<<<<<<
|
|
* else:
|
|
* i_root += 1
|
|
*/
|
|
goto __pyx_L9_continue;
|
|
goto __pyx_L11;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":568
|
|
* continue
|
|
* else:
|
|
* i_root += 1 # <<<<<<<<<<<<<<
|
|
* root_list[i_root] = cnode
|
|
* node_list[i_nl_end] = cnode
|
|
*/
|
|
__pyx_v_i_root = (__pyx_v_i_root + 1);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":569
|
|
* else:
|
|
* i_root += 1
|
|
* root_list[i_root] = cnode # <<<<<<<<<<<<<<
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode
|
|
*/
|
|
__pyx_t_13 = __pyx_v_i_root;
|
|
__pyx_t_21 = -1;
|
|
if (__pyx_t_13 < 0) {
|
|
__pyx_t_13 += __pyx_pybuffernd_root_list.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_13 < 0)) __pyx_t_21 = 0;
|
|
} else if (unlikely(__pyx_t_13 >= __pyx_pybuffernd_root_list.diminfo[0].shape)) __pyx_t_21 = 0;
|
|
if (unlikely(__pyx_t_21 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_21);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 569; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_root_list.rcbuffer->pybuffer.buf, __pyx_t_13, __pyx_pybuffernd_root_list.diminfo[0].strides) = __pyx_v_cnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":570
|
|
* i_root += 1
|
|
* root_list[i_root] = cnode
|
|
* node_list[i_nl_end] = cnode # <<<<<<<<<<<<<<
|
|
* predecessors[cnode] = pnode
|
|
* flag[cnode] = 1
|
|
*/
|
|
__pyx_t_22 = __pyx_v_i_nl_end;
|
|
__pyx_t_21 = -1;
|
|
if (unlikely(__pyx_t_22 >= (size_t)__pyx_pybuffernd_node_list.diminfo[0].shape)) __pyx_t_21 = 0;
|
|
if (unlikely(__pyx_t_21 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_21);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 570; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_node_list.rcbuffer->pybuffer.buf, __pyx_t_22, __pyx_pybuffernd_node_list.diminfo[0].strides) = __pyx_v_cnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":571
|
|
* root_list[i_root] = cnode
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode # <<<<<<<<<<<<<<
|
|
* flag[cnode] = 1
|
|
* i_nl_end += 1
|
|
*/
|
|
__pyx_t_23 = __pyx_v_cnode;
|
|
__pyx_t_21 = -1;
|
|
if (unlikely(__pyx_t_23 >= (size_t)__pyx_pybuffernd_predecessors.diminfo[0].shape)) __pyx_t_21 = 0;
|
|
if (unlikely(__pyx_t_21 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_21);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 571; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_predecessors.rcbuffer->pybuffer.buf, __pyx_t_23, __pyx_pybuffernd_predecessors.diminfo[0].strides) = __pyx_v_pnode;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":572
|
|
* node_list[i_nl_end] = cnode
|
|
* predecessors[cnode] = pnode
|
|
* flag[cnode] = 1 # <<<<<<<<<<<<<<
|
|
* i_nl_end += 1
|
|
* no_children = False
|
|
*/
|
|
__pyx_t_24 = __pyx_v_cnode;
|
|
__pyx_t_21 = -1;
|
|
if (unlikely(__pyx_t_24 >= (size_t)__pyx_pybuffernd_flag.diminfo[0].shape)) __pyx_t_21 = 0;
|
|
if (unlikely(__pyx_t_21 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_21);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 572; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_flag.rcbuffer->pybuffer.buf, __pyx_t_24, __pyx_pybuffernd_flag.diminfo[0].strides) = 1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":573
|
|
* predecessors[cnode] = pnode
|
|
* flag[cnode] = 1
|
|
* i_nl_end += 1 # <<<<<<<<<<<<<<
|
|
* no_children = False
|
|
* break
|
|
*/
|
|
__pyx_v_i_nl_end = (__pyx_v_i_nl_end + 1);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":574
|
|
* flag[cnode] = 1
|
|
* i_nl_end += 1
|
|
* no_children = False # <<<<<<<<<<<<<<
|
|
* break
|
|
*
|
|
*/
|
|
__pyx_v_no_children = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":575
|
|
* i_nl_end += 1
|
|
* no_children = False
|
|
* break # <<<<<<<<<<<<<<
|
|
*
|
|
* if i_nl_end == N:
|
|
*/
|
|
goto __pyx_L10_break;
|
|
}
|
|
__pyx_L11:;
|
|
__pyx_L9_continue:;
|
|
}
|
|
__pyx_L10_break:;
|
|
goto __pyx_L8;
|
|
}
|
|
__pyx_L8:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":577
|
|
* break
|
|
*
|
|
* if i_nl_end == N: # <<<<<<<<<<<<<<
|
|
* break
|
|
*
|
|
*/
|
|
__pyx_t_5 = (__pyx_v_i_nl_end == __pyx_v_N);
|
|
if (__pyx_t_5) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":578
|
|
*
|
|
* if i_nl_end == N:
|
|
* break # <<<<<<<<<<<<<<
|
|
*
|
|
* if no_children:
|
|
*/
|
|
goto __pyx_L4_break;
|
|
goto __pyx_L12;
|
|
}
|
|
__pyx_L12:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":580
|
|
* break
|
|
*
|
|
* if no_children: # <<<<<<<<<<<<<<
|
|
* i_root -= 1
|
|
*
|
|
*/
|
|
if (__pyx_v_no_children) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":581
|
|
*
|
|
* if no_children:
|
|
* i_root -= 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* return i_nl_end
|
|
*/
|
|
__pyx_v_i_root = (__pyx_v_i_root - 1);
|
|
goto __pyx_L13;
|
|
}
|
|
__pyx_L13:;
|
|
}
|
|
__pyx_L4_break:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":583
|
|
* i_root -= 1
|
|
*
|
|
* return i_nl_end # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = __pyx_v_i_nl_end;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_flag.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices1.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices2.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr1.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr2.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_node_list.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_predecessors.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_root_list.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_WriteUnraisable("sklearn.utils.mst._traversal._depth_first_undirected", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_flag.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices1.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices2.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr1.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr2.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_node_list.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_predecessors.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_root_list.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":586
|
|
*
|
|
*
|
|
* cdef int _connected_components_directed( # <<<<<<<<<<<<<<
|
|
* np.ndarray[ITYPE_t, ndim=1, mode='c'] indices,
|
|
* np.ndarray[ITYPE_t, ndim=1, mode='c'] indptr,
|
|
*/
|
|
|
|
static int __pyx_f_7sklearn_5utils_3mst_10_traversal__connected_components_directed(PyArrayObject *__pyx_v_indices, PyArrayObject *__pyx_v_indptr, PyArrayObject *__pyx_v_labels) {
|
|
int __pyx_v_v;
|
|
int __pyx_v_w;
|
|
int __pyx_v_index;
|
|
int __pyx_v_low_v;
|
|
int __pyx_v_low_w;
|
|
int __pyx_v_label;
|
|
int __pyx_v_j;
|
|
int __pyx_v_SS_head;
|
|
int __pyx_v_root;
|
|
int __pyx_v_stack_head;
|
|
int __pyx_v_f;
|
|
int __pyx_v_b;
|
|
int __pyx_v_VOID;
|
|
int __pyx_v_END;
|
|
int __pyx_v_N;
|
|
PyArrayObject *__pyx_v_SS = 0;
|
|
PyArrayObject *__pyx_v_lowlinks = 0;
|
|
PyArrayObject *__pyx_v_stack_f = 0;
|
|
PyArrayObject *__pyx_v_stack_b = 0;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_SS;
|
|
__Pyx_Buffer __pyx_pybuffer_SS;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indices;
|
|
__Pyx_Buffer __pyx_pybuffer_indices;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indptr;
|
|
__Pyx_Buffer __pyx_pybuffer_indptr;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_labels;
|
|
__Pyx_Buffer __pyx_pybuffer_labels;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_lowlinks;
|
|
__Pyx_Buffer __pyx_pybuffer_lowlinks;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_stack_b;
|
|
__Pyx_Buffer __pyx_pybuffer_stack_b;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_stack_f;
|
|
__Pyx_Buffer __pyx_pybuffer_stack_f;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
PyObject *__pyx_t_5 = NULL;
|
|
PyObject *__pyx_t_6 = NULL;
|
|
PyObject *__pyx_t_7 = NULL;
|
|
int __pyx_t_8;
|
|
int __pyx_t_9;
|
|
int __pyx_t_10;
|
|
int __pyx_t_11;
|
|
int __pyx_t_12;
|
|
int __pyx_t_13;
|
|
int __pyx_t_14;
|
|
int __pyx_t_15;
|
|
int __pyx_t_16;
|
|
long __pyx_t_17;
|
|
__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t __pyx_t_18;
|
|
int __pyx_t_19;
|
|
int __pyx_t_20;
|
|
int __pyx_t_21;
|
|
int __pyx_t_22;
|
|
int __pyx_t_23;
|
|
int __pyx_t_24;
|
|
int __pyx_t_25;
|
|
int __pyx_t_26;
|
|
int __pyx_t_27;
|
|
int __pyx_t_28;
|
|
int __pyx_t_29;
|
|
int __pyx_t_30;
|
|
int __pyx_t_31;
|
|
int __pyx_t_32;
|
|
int __pyx_t_33;
|
|
int __pyx_t_34;
|
|
long __pyx_t_35;
|
|
int __pyx_t_36;
|
|
__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t __pyx_t_37;
|
|
int __pyx_t_38;
|
|
int __pyx_t_39;
|
|
int __pyx_t_40;
|
|
int __pyx_t_41;
|
|
int __pyx_t_42;
|
|
int __pyx_t_43;
|
|
int __pyx_t_44;
|
|
int __pyx_t_45;
|
|
int __pyx_t_46;
|
|
int __pyx_t_47;
|
|
PyArrayObject *__pyx_t_48 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_connected_components_directed", 0);
|
|
__Pyx_INCREF((PyObject *)__pyx_v_labels);
|
|
__pyx_pybuffer_SS.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_SS.refcount = 0;
|
|
__pyx_pybuffernd_SS.data = NULL;
|
|
__pyx_pybuffernd_SS.rcbuffer = &__pyx_pybuffer_SS;
|
|
__pyx_pybuffer_lowlinks.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_lowlinks.refcount = 0;
|
|
__pyx_pybuffernd_lowlinks.data = NULL;
|
|
__pyx_pybuffernd_lowlinks.rcbuffer = &__pyx_pybuffer_lowlinks;
|
|
__pyx_pybuffer_stack_f.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_stack_f.refcount = 0;
|
|
__pyx_pybuffernd_stack_f.data = NULL;
|
|
__pyx_pybuffernd_stack_f.rcbuffer = &__pyx_pybuffer_stack_f;
|
|
__pyx_pybuffer_stack_b.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_stack_b.refcount = 0;
|
|
__pyx_pybuffernd_stack_b.data = NULL;
|
|
__pyx_pybuffernd_stack_b.rcbuffer = &__pyx_pybuffer_stack_b;
|
|
__pyx_pybuffer_indices.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indices.refcount = 0;
|
|
__pyx_pybuffernd_indices.data = NULL;
|
|
__pyx_pybuffernd_indices.rcbuffer = &__pyx_pybuffer_indices;
|
|
__pyx_pybuffer_indptr.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indptr.refcount = 0;
|
|
__pyx_pybuffernd_indptr.data = NULL;
|
|
__pyx_pybuffernd_indptr.rcbuffer = &__pyx_pybuffer_indptr;
|
|
__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_indices.rcbuffer->pybuffer, (PyObject*)__pyx_v_indices, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 586; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indices.diminfo[0].strides = __pyx_pybuffernd_indices.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indices.diminfo[0].shape = __pyx_pybuffernd_indices.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indptr.rcbuffer->pybuffer, (PyObject*)__pyx_v_indptr, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 586; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indptr.diminfo[0].strides = __pyx_pybuffernd_indptr.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indptr.diminfo[0].shape = __pyx_pybuffernd_indptr.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___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 586; __pyx_clineno = __LINE__; goto __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];
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":604
|
|
* cdef int v, w, index, low_v, low_w, label, j
|
|
* cdef int SS_head, root, stack_head, f, b
|
|
* cdef int VOID = -1 # <<<<<<<<<<<<<<
|
|
* cdef int END = -2
|
|
* cdef int N = labels.shape[0]
|
|
*/
|
|
__pyx_v_VOID = -1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":605
|
|
* cdef int SS_head, root, stack_head, f, b
|
|
* cdef int VOID = -1
|
|
* cdef int END = -2 # <<<<<<<<<<<<<<
|
|
* cdef int N = labels.shape[0]
|
|
* cdef np.ndarray[ITYPE_t, ndim=1, mode="c"] SS, lowlinks, stack_f, stack_b
|
|
*/
|
|
__pyx_v_END = -2;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":606
|
|
* cdef int VOID = -1
|
|
* cdef int END = -2
|
|
* cdef int N = labels.shape[0] # <<<<<<<<<<<<<<
|
|
* cdef np.ndarray[ITYPE_t, ndim=1, mode="c"] SS, lowlinks, stack_f, stack_b
|
|
*
|
|
*/
|
|
__pyx_v_N = (__pyx_v_labels->dimensions[0]);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":609
|
|
* cdef np.ndarray[ITYPE_t, ndim=1, mode="c"] SS, lowlinks, stack_f, stack_b
|
|
*
|
|
* lowlinks = labels # <<<<<<<<<<<<<<
|
|
* SS = np.ndarray((N,), dtype=ITYPE)
|
|
* stack_b = np.ndarray((N,), dtype=ITYPE)
|
|
*/
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_lowlinks.rcbuffer->pybuffer);
|
|
__pyx_t_1 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_lowlinks.rcbuffer->pybuffer, (PyObject*)((PyArrayObject *)__pyx_v_labels), &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack);
|
|
if (unlikely(__pyx_t_1 < 0)) {
|
|
PyErr_Fetch(&__pyx_t_2, &__pyx_t_3, &__pyx_t_4);
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_lowlinks.rcbuffer->pybuffer, (PyObject*)__pyx_v_lowlinks, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {
|
|
Py_XDECREF(__pyx_t_2); Py_XDECREF(__pyx_t_3); Py_XDECREF(__pyx_t_4);
|
|
__Pyx_RaiseBufferFallbackError();
|
|
} else {
|
|
PyErr_Restore(__pyx_t_2, __pyx_t_3, __pyx_t_4);
|
|
}
|
|
}
|
|
__pyx_pybuffernd_lowlinks.diminfo[0].strides = __pyx_pybuffernd_lowlinks.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_lowlinks.diminfo[0].shape = __pyx_pybuffernd_lowlinks.rcbuffer->pybuffer.shape[0];
|
|
if (unlikely(__pyx_t_1 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 609; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_labels));
|
|
__pyx_v_lowlinks = ((PyArrayObject *)__pyx_v_labels);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":610
|
|
*
|
|
* lowlinks = labels
|
|
* SS = np.ndarray((N,), dtype=ITYPE) # <<<<<<<<<<<<<<
|
|
* stack_b = np.ndarray((N,), dtype=ITYPE)
|
|
* stack_f = SS
|
|
*/
|
|
__pyx_t_5 = PyInt_FromLong(__pyx_v_N); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 610; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_6 = PyTuple_New(1); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 610; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
PyTuple_SET_ITEM(__pyx_t_6, 0, __pyx_t_5);
|
|
__Pyx_GIVEREF(__pyx_t_5);
|
|
__pyx_t_5 = 0;
|
|
__pyx_t_5 = PyTuple_New(1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 610; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
PyTuple_SET_ITEM(__pyx_t_5, 0, ((PyObject *)__pyx_t_6));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_t_6));
|
|
__pyx_t_6 = 0;
|
|
__pyx_t_6 = PyDict_New(); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 610; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_6));
|
|
__pyx_t_7 = __Pyx_GetName(__pyx_m, __pyx_n_s__ITYPE); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 610; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
if (PyDict_SetItem(__pyx_t_6, ((PyObject *)__pyx_n_s__dtype), __pyx_t_7) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 610; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__pyx_t_7 = PyObject_Call(((PyObject *)((PyObject*)__pyx_ptype_5numpy_ndarray)), ((PyObject *)__pyx_t_5), ((PyObject *)__pyx_t_6)); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 610; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_5)); __pyx_t_5 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_6)); __pyx_t_6 = 0;
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_SS.rcbuffer->pybuffer);
|
|
__pyx_t_1 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_SS.rcbuffer->pybuffer, (PyObject*)((PyArrayObject *)__pyx_t_7), &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack);
|
|
if (unlikely(__pyx_t_1 < 0)) {
|
|
PyErr_Fetch(&__pyx_t_4, &__pyx_t_3, &__pyx_t_2);
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_SS.rcbuffer->pybuffer, (PyObject*)__pyx_v_SS, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {
|
|
Py_XDECREF(__pyx_t_4); Py_XDECREF(__pyx_t_3); Py_XDECREF(__pyx_t_2);
|
|
__Pyx_RaiseBufferFallbackError();
|
|
} else {
|
|
PyErr_Restore(__pyx_t_4, __pyx_t_3, __pyx_t_2);
|
|
}
|
|
}
|
|
__pyx_pybuffernd_SS.diminfo[0].strides = __pyx_pybuffernd_SS.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_SS.diminfo[0].shape = __pyx_pybuffernd_SS.rcbuffer->pybuffer.shape[0];
|
|
if (unlikely(__pyx_t_1 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 610; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_SS = ((PyArrayObject *)__pyx_t_7);
|
|
__pyx_t_7 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":611
|
|
* lowlinks = labels
|
|
* SS = np.ndarray((N,), dtype=ITYPE)
|
|
* stack_b = np.ndarray((N,), dtype=ITYPE) # <<<<<<<<<<<<<<
|
|
* stack_f = SS
|
|
*
|
|
*/
|
|
__pyx_t_7 = PyInt_FromLong(__pyx_v_N); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 611; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__pyx_t_6 = PyTuple_New(1); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 611; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
PyTuple_SET_ITEM(__pyx_t_6, 0, __pyx_t_7);
|
|
__Pyx_GIVEREF(__pyx_t_7);
|
|
__pyx_t_7 = 0;
|
|
__pyx_t_7 = PyTuple_New(1); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 611; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
PyTuple_SET_ITEM(__pyx_t_7, 0, ((PyObject *)__pyx_t_6));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_t_6));
|
|
__pyx_t_6 = 0;
|
|
__pyx_t_6 = PyDict_New(); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 611; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_6));
|
|
__pyx_t_5 = __Pyx_GetName(__pyx_m, __pyx_n_s__ITYPE); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 611; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
if (PyDict_SetItem(__pyx_t_6, ((PyObject *)__pyx_n_s__dtype), __pyx_t_5) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 611; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_5 = PyObject_Call(((PyObject *)((PyObject*)__pyx_ptype_5numpy_ndarray)), ((PyObject *)__pyx_t_7), ((PyObject *)__pyx_t_6)); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 611; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_7)); __pyx_t_7 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_6)); __pyx_t_6 = 0;
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_stack_b.rcbuffer->pybuffer);
|
|
__pyx_t_1 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_stack_b.rcbuffer->pybuffer, (PyObject*)((PyArrayObject *)__pyx_t_5), &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack);
|
|
if (unlikely(__pyx_t_1 < 0)) {
|
|
PyErr_Fetch(&__pyx_t_2, &__pyx_t_3, &__pyx_t_4);
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_stack_b.rcbuffer->pybuffer, (PyObject*)__pyx_v_stack_b, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {
|
|
Py_XDECREF(__pyx_t_2); Py_XDECREF(__pyx_t_3); Py_XDECREF(__pyx_t_4);
|
|
__Pyx_RaiseBufferFallbackError();
|
|
} else {
|
|
PyErr_Restore(__pyx_t_2, __pyx_t_3, __pyx_t_4);
|
|
}
|
|
}
|
|
__pyx_pybuffernd_stack_b.diminfo[0].strides = __pyx_pybuffernd_stack_b.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_stack_b.diminfo[0].shape = __pyx_pybuffernd_stack_b.rcbuffer->pybuffer.shape[0];
|
|
if (unlikely(__pyx_t_1 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 611; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_stack_b = ((PyArrayObject *)__pyx_t_5);
|
|
__pyx_t_5 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":612
|
|
* SS = np.ndarray((N,), dtype=ITYPE)
|
|
* stack_b = np.ndarray((N,), dtype=ITYPE)
|
|
* stack_f = SS # <<<<<<<<<<<<<<
|
|
*
|
|
* # The stack of nodes which have been backtracked and are in the current SCC
|
|
*/
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_stack_f.rcbuffer->pybuffer);
|
|
__pyx_t_1 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_stack_f.rcbuffer->pybuffer, (PyObject*)__pyx_v_SS, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack);
|
|
if (unlikely(__pyx_t_1 < 0)) {
|
|
PyErr_Fetch(&__pyx_t_4, &__pyx_t_3, &__pyx_t_2);
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_stack_f.rcbuffer->pybuffer, (PyObject*)__pyx_v_stack_f, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {
|
|
Py_XDECREF(__pyx_t_4); Py_XDECREF(__pyx_t_3); Py_XDECREF(__pyx_t_2);
|
|
__Pyx_RaiseBufferFallbackError();
|
|
} else {
|
|
PyErr_Restore(__pyx_t_4, __pyx_t_3, __pyx_t_2);
|
|
}
|
|
}
|
|
__pyx_pybuffernd_stack_f.diminfo[0].strides = __pyx_pybuffernd_stack_f.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_stack_f.diminfo[0].shape = __pyx_pybuffernd_stack_f.rcbuffer->pybuffer.shape[0];
|
|
if (unlikely(__pyx_t_1 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 612; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_SS));
|
|
__pyx_v_stack_f = __pyx_v_SS;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":615
|
|
*
|
|
* # The stack of nodes which have been backtracked and are in the current SCC
|
|
* SS.fill(VOID) # <<<<<<<<<<<<<<
|
|
* SS_head = END
|
|
*
|
|
*/
|
|
__pyx_t_5 = PyObject_GetAttr(((PyObject *)__pyx_v_SS), __pyx_n_s__fill); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 615; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_6 = PyInt_FromLong(__pyx_v_VOID); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 615; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__pyx_t_7 = PyTuple_New(1); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 615; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
PyTuple_SET_ITEM(__pyx_t_7, 0, __pyx_t_6);
|
|
__Pyx_GIVEREF(__pyx_t_6);
|
|
__pyx_t_6 = 0;
|
|
__pyx_t_6 = PyObject_Call(__pyx_t_5, ((PyObject *)__pyx_t_7), NULL); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 615; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_7)); __pyx_t_7 = 0;
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":616
|
|
* # The stack of nodes which have been backtracked and are in the current SCC
|
|
* SS.fill(VOID)
|
|
* SS_head = END # <<<<<<<<<<<<<<
|
|
*
|
|
* # The array containing the lowlinks of nodes not yet assigned an SCC. Shares
|
|
*/
|
|
__pyx_v_SS_head = __pyx_v_END;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":620
|
|
* # The array containing the lowlinks of nodes not yet assigned an SCC. Shares
|
|
* # memory with the labels array, since they are not used at the same time.
|
|
* lowlinks.fill(VOID) # <<<<<<<<<<<<<<
|
|
*
|
|
* # The DFS stack. Stored with both forwards and backwards pointers to allow
|
|
*/
|
|
__pyx_t_6 = PyObject_GetAttr(((PyObject *)__pyx_v_lowlinks), __pyx_n_s__fill); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 620; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__pyx_t_7 = PyInt_FromLong(__pyx_v_VOID); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 620; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__pyx_t_5 = PyTuple_New(1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 620; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
PyTuple_SET_ITEM(__pyx_t_5, 0, __pyx_t_7);
|
|
__Pyx_GIVEREF(__pyx_t_7);
|
|
__pyx_t_7 = 0;
|
|
__pyx_t_7 = PyObject_Call(__pyx_t_6, ((PyObject *)__pyx_t_5), NULL); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 620; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_5)); __pyx_t_5 = 0;
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":626
|
|
* # each node once. stack_f shares memory with SS, as nodes aren't put on the
|
|
* # SS stack until after they've been popped from the DFS stack.
|
|
* stack_head = END # <<<<<<<<<<<<<<
|
|
* stack_f.fill(VOID)
|
|
* stack_b.fill(VOID)
|
|
*/
|
|
__pyx_v_stack_head = __pyx_v_END;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":627
|
|
* # SS stack until after they've been popped from the DFS stack.
|
|
* stack_head = END
|
|
* stack_f.fill(VOID) # <<<<<<<<<<<<<<
|
|
* stack_b.fill(VOID)
|
|
*
|
|
*/
|
|
__pyx_t_7 = PyObject_GetAttr(((PyObject *)__pyx_v_stack_f), __pyx_n_s__fill); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 627; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__pyx_t_5 = PyInt_FromLong(__pyx_v_VOID); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 627; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_6 = PyTuple_New(1); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 627; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
PyTuple_SET_ITEM(__pyx_t_6, 0, __pyx_t_5);
|
|
__Pyx_GIVEREF(__pyx_t_5);
|
|
__pyx_t_5 = 0;
|
|
__pyx_t_5 = PyObject_Call(__pyx_t_7, ((PyObject *)__pyx_t_6), NULL); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 627; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_6)); __pyx_t_6 = 0;
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":628
|
|
* stack_head = END
|
|
* stack_f.fill(VOID)
|
|
* stack_b.fill(VOID) # <<<<<<<<<<<<<<
|
|
*
|
|
* index = 0
|
|
*/
|
|
__pyx_t_5 = PyObject_GetAttr(((PyObject *)__pyx_v_stack_b), __pyx_n_s__fill); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 628; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_6 = PyInt_FromLong(__pyx_v_VOID); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 628; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__pyx_t_7 = PyTuple_New(1); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 628; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
PyTuple_SET_ITEM(__pyx_t_7, 0, __pyx_t_6);
|
|
__Pyx_GIVEREF(__pyx_t_6);
|
|
__pyx_t_6 = 0;
|
|
__pyx_t_6 = PyObject_Call(__pyx_t_5, ((PyObject *)__pyx_t_7), NULL); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 628; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_7)); __pyx_t_7 = 0;
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":630
|
|
* stack_b.fill(VOID)
|
|
*
|
|
* index = 0 # <<<<<<<<<<<<<<
|
|
* # Count SCC labels backwards so as not to class with lowlinks values.
|
|
* label = N - 1
|
|
*/
|
|
__pyx_v_index = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":632
|
|
* index = 0
|
|
* # Count SCC labels backwards so as not to class with lowlinks values.
|
|
* label = N - 1 # <<<<<<<<<<<<<<
|
|
* for v in range(N):
|
|
* if lowlinks[v] == VOID:
|
|
*/
|
|
__pyx_v_label = (__pyx_v_N - 1);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":633
|
|
* # Count SCC labels backwards so as not to class with lowlinks values.
|
|
* label = N - 1
|
|
* for v in range(N): # <<<<<<<<<<<<<<
|
|
* if lowlinks[v] == VOID:
|
|
* # DFS-stack push
|
|
*/
|
|
__pyx_t_1 = __pyx_v_N;
|
|
for (__pyx_t_8 = 0; __pyx_t_8 < __pyx_t_1; __pyx_t_8+=1) {
|
|
__pyx_v_v = __pyx_t_8;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":634
|
|
* label = N - 1
|
|
* for v in range(N):
|
|
* if lowlinks[v] == VOID: # <<<<<<<<<<<<<<
|
|
* # DFS-stack push
|
|
* stack_head = v
|
|
*/
|
|
__pyx_t_9 = __pyx_v_v;
|
|
__pyx_t_10 = -1;
|
|
if (__pyx_t_9 < 0) {
|
|
__pyx_t_9 += __pyx_pybuffernd_lowlinks.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_9 < 0)) __pyx_t_10 = 0;
|
|
} else if (unlikely(__pyx_t_9 >= __pyx_pybuffernd_lowlinks.diminfo[0].shape)) __pyx_t_10 = 0;
|
|
if (unlikely(__pyx_t_10 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_10);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 634; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_11 = ((*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_lowlinks.rcbuffer->pybuffer.buf, __pyx_t_9, __pyx_pybuffernd_lowlinks.diminfo[0].strides)) == __pyx_v_VOID);
|
|
if (__pyx_t_11) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":636
|
|
* if lowlinks[v] == VOID:
|
|
* # DFS-stack push
|
|
* stack_head = v # <<<<<<<<<<<<<<
|
|
* stack_f[v] = END
|
|
* stack_b[v] = END
|
|
*/
|
|
__pyx_v_stack_head = __pyx_v_v;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":637
|
|
* # DFS-stack push
|
|
* stack_head = v
|
|
* stack_f[v] = END # <<<<<<<<<<<<<<
|
|
* stack_b[v] = END
|
|
* while stack_head != END:
|
|
*/
|
|
__pyx_t_10 = __pyx_v_v;
|
|
__pyx_t_12 = -1;
|
|
if (__pyx_t_10 < 0) {
|
|
__pyx_t_10 += __pyx_pybuffernd_stack_f.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_10 < 0)) __pyx_t_12 = 0;
|
|
} else if (unlikely(__pyx_t_10 >= __pyx_pybuffernd_stack_f.diminfo[0].shape)) __pyx_t_12 = 0;
|
|
if (unlikely(__pyx_t_12 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_12);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 637; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_f.rcbuffer->pybuffer.buf, __pyx_t_10, __pyx_pybuffernd_stack_f.diminfo[0].strides) = __pyx_v_END;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":638
|
|
* stack_head = v
|
|
* stack_f[v] = END
|
|
* stack_b[v] = END # <<<<<<<<<<<<<<
|
|
* while stack_head != END:
|
|
* v = stack_head
|
|
*/
|
|
__pyx_t_12 = __pyx_v_v;
|
|
__pyx_t_13 = -1;
|
|
if (__pyx_t_12 < 0) {
|
|
__pyx_t_12 += __pyx_pybuffernd_stack_b.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_12 < 0)) __pyx_t_13 = 0;
|
|
} else if (unlikely(__pyx_t_12 >= __pyx_pybuffernd_stack_b.diminfo[0].shape)) __pyx_t_13 = 0;
|
|
if (unlikely(__pyx_t_13 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_13);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 638; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_b.rcbuffer->pybuffer.buf, __pyx_t_12, __pyx_pybuffernd_stack_b.diminfo[0].strides) = __pyx_v_END;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":639
|
|
* stack_f[v] = END
|
|
* stack_b[v] = END
|
|
* while stack_head != END: # <<<<<<<<<<<<<<
|
|
* v = stack_head
|
|
* if lowlinks[v] == VOID:
|
|
*/
|
|
while (1) {
|
|
__pyx_t_11 = (__pyx_v_stack_head != __pyx_v_END);
|
|
if (!__pyx_t_11) break;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":640
|
|
* stack_b[v] = END
|
|
* while stack_head != END:
|
|
* v = stack_head # <<<<<<<<<<<<<<
|
|
* if lowlinks[v] == VOID:
|
|
* lowlinks[v] = index
|
|
*/
|
|
__pyx_v_v = __pyx_v_stack_head;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":641
|
|
* while stack_head != END:
|
|
* v = stack_head
|
|
* if lowlinks[v] == VOID: # <<<<<<<<<<<<<<
|
|
* lowlinks[v] = index
|
|
* index += 1
|
|
*/
|
|
__pyx_t_13 = __pyx_v_v;
|
|
__pyx_t_14 = -1;
|
|
if (__pyx_t_13 < 0) {
|
|
__pyx_t_13 += __pyx_pybuffernd_lowlinks.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_13 < 0)) __pyx_t_14 = 0;
|
|
} else if (unlikely(__pyx_t_13 >= __pyx_pybuffernd_lowlinks.diminfo[0].shape)) __pyx_t_14 = 0;
|
|
if (unlikely(__pyx_t_14 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_14);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 641; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_11 = ((*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_lowlinks.rcbuffer->pybuffer.buf, __pyx_t_13, __pyx_pybuffernd_lowlinks.diminfo[0].strides)) == __pyx_v_VOID);
|
|
if (__pyx_t_11) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":642
|
|
* v = stack_head
|
|
* if lowlinks[v] == VOID:
|
|
* lowlinks[v] = index # <<<<<<<<<<<<<<
|
|
* index += 1
|
|
*
|
|
*/
|
|
__pyx_t_14 = __pyx_v_v;
|
|
__pyx_t_15 = -1;
|
|
if (__pyx_t_14 < 0) {
|
|
__pyx_t_14 += __pyx_pybuffernd_lowlinks.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_14 < 0)) __pyx_t_15 = 0;
|
|
} else if (unlikely(__pyx_t_14 >= __pyx_pybuffernd_lowlinks.diminfo[0].shape)) __pyx_t_15 = 0;
|
|
if (unlikely(__pyx_t_15 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_15);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 642; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_lowlinks.rcbuffer->pybuffer.buf, __pyx_t_14, __pyx_pybuffernd_lowlinks.diminfo[0].strides) = __pyx_v_index;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":643
|
|
* if lowlinks[v] == VOID:
|
|
* lowlinks[v] = index
|
|
* index += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Add successor nodes
|
|
*/
|
|
__pyx_v_index = (__pyx_v_index + 1);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":646
|
|
*
|
|
* # Add successor nodes
|
|
* for j from indptr[v] <= j < indptr[v+1]: # <<<<<<<<<<<<<<
|
|
* w = indices[j]
|
|
* if lowlinks[w] == VOID:
|
|
*/
|
|
__pyx_t_15 = __pyx_v_v;
|
|
__pyx_t_16 = -1;
|
|
if (__pyx_t_15 < 0) {
|
|
__pyx_t_15 += __pyx_pybuffernd_indptr.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_15 < 0)) __pyx_t_16 = 0;
|
|
} else if (unlikely(__pyx_t_15 >= __pyx_pybuffernd_indptr.diminfo[0].shape)) __pyx_t_16 = 0;
|
|
if (unlikely(__pyx_t_16 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_16);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 646; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_17 = (__pyx_v_v + 1);
|
|
__pyx_t_16 = -1;
|
|
if (__pyx_t_17 < 0) {
|
|
__pyx_t_17 += __pyx_pybuffernd_indptr.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_17 < 0)) __pyx_t_16 = 0;
|
|
} else if (unlikely(__pyx_t_17 >= __pyx_pybuffernd_indptr.diminfo[0].shape)) __pyx_t_16 = 0;
|
|
if (unlikely(__pyx_t_16 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_16);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 646; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_18 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr.rcbuffer->pybuffer.buf, __pyx_t_17, __pyx_pybuffernd_indptr.diminfo[0].strides));
|
|
for (__pyx_v_j = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr.rcbuffer->pybuffer.buf, __pyx_t_15, __pyx_pybuffernd_indptr.diminfo[0].strides)); __pyx_v_j < __pyx_t_18; __pyx_v_j++) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":647
|
|
* # Add successor nodes
|
|
* for j from indptr[v] <= j < indptr[v+1]:
|
|
* w = indices[j] # <<<<<<<<<<<<<<
|
|
* if lowlinks[w] == VOID:
|
|
* # DFS-stack push
|
|
*/
|
|
__pyx_t_16 = __pyx_v_j;
|
|
__pyx_t_19 = -1;
|
|
if (__pyx_t_16 < 0) {
|
|
__pyx_t_16 += __pyx_pybuffernd_indices.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_16 < 0)) __pyx_t_19 = 0;
|
|
} else if (unlikely(__pyx_t_16 >= __pyx_pybuffernd_indices.diminfo[0].shape)) __pyx_t_19 = 0;
|
|
if (unlikely(__pyx_t_19 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_19);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 647; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_w = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indices.rcbuffer->pybuffer.buf, __pyx_t_16, __pyx_pybuffernd_indices.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":648
|
|
* for j from indptr[v] <= j < indptr[v+1]:
|
|
* w = indices[j]
|
|
* if lowlinks[w] == VOID: # <<<<<<<<<<<<<<
|
|
* # DFS-stack push
|
|
* if stack_f[w] != VOID:
|
|
*/
|
|
__pyx_t_19 = __pyx_v_w;
|
|
__pyx_t_20 = -1;
|
|
if (__pyx_t_19 < 0) {
|
|
__pyx_t_19 += __pyx_pybuffernd_lowlinks.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_19 < 0)) __pyx_t_20 = 0;
|
|
} else if (unlikely(__pyx_t_19 >= __pyx_pybuffernd_lowlinks.diminfo[0].shape)) __pyx_t_20 = 0;
|
|
if (unlikely(__pyx_t_20 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_20);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 648; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_11 = ((*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_lowlinks.rcbuffer->pybuffer.buf, __pyx_t_19, __pyx_pybuffernd_lowlinks.diminfo[0].strides)) == __pyx_v_VOID);
|
|
if (__pyx_t_11) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":650
|
|
* if lowlinks[w] == VOID:
|
|
* # DFS-stack push
|
|
* if stack_f[w] != VOID: # <<<<<<<<<<<<<<
|
|
* # w is already inside the stack, so excise it.
|
|
* f = stack_f[w]
|
|
*/
|
|
__pyx_t_20 = __pyx_v_w;
|
|
__pyx_t_21 = -1;
|
|
if (__pyx_t_20 < 0) {
|
|
__pyx_t_20 += __pyx_pybuffernd_stack_f.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_20 < 0)) __pyx_t_21 = 0;
|
|
} else if (unlikely(__pyx_t_20 >= __pyx_pybuffernd_stack_f.diminfo[0].shape)) __pyx_t_21 = 0;
|
|
if (unlikely(__pyx_t_21 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_21);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 650; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_11 = ((*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_f.rcbuffer->pybuffer.buf, __pyx_t_20, __pyx_pybuffernd_stack_f.diminfo[0].strides)) != __pyx_v_VOID);
|
|
if (__pyx_t_11) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":652
|
|
* if stack_f[w] != VOID:
|
|
* # w is already inside the stack, so excise it.
|
|
* f = stack_f[w] # <<<<<<<<<<<<<<
|
|
* b = stack_b[w]
|
|
* if b != END:
|
|
*/
|
|
__pyx_t_21 = __pyx_v_w;
|
|
__pyx_t_22 = -1;
|
|
if (__pyx_t_21 < 0) {
|
|
__pyx_t_21 += __pyx_pybuffernd_stack_f.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_21 < 0)) __pyx_t_22 = 0;
|
|
} else if (unlikely(__pyx_t_21 >= __pyx_pybuffernd_stack_f.diminfo[0].shape)) __pyx_t_22 = 0;
|
|
if (unlikely(__pyx_t_22 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_22);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 652; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_f = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_f.rcbuffer->pybuffer.buf, __pyx_t_21, __pyx_pybuffernd_stack_f.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":653
|
|
* # w is already inside the stack, so excise it.
|
|
* f = stack_f[w]
|
|
* b = stack_b[w] # <<<<<<<<<<<<<<
|
|
* if b != END:
|
|
* stack_f[b] = f
|
|
*/
|
|
__pyx_t_22 = __pyx_v_w;
|
|
__pyx_t_23 = -1;
|
|
if (__pyx_t_22 < 0) {
|
|
__pyx_t_22 += __pyx_pybuffernd_stack_b.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_22 < 0)) __pyx_t_23 = 0;
|
|
} else if (unlikely(__pyx_t_22 >= __pyx_pybuffernd_stack_b.diminfo[0].shape)) __pyx_t_23 = 0;
|
|
if (unlikely(__pyx_t_23 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_23);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 653; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_b = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_b.rcbuffer->pybuffer.buf, __pyx_t_22, __pyx_pybuffernd_stack_b.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":654
|
|
* f = stack_f[w]
|
|
* b = stack_b[w]
|
|
* if b != END: # <<<<<<<<<<<<<<
|
|
* stack_f[b] = f
|
|
* if f != END:
|
|
*/
|
|
__pyx_t_11 = (__pyx_v_b != __pyx_v_END);
|
|
if (__pyx_t_11) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":655
|
|
* b = stack_b[w]
|
|
* if b != END:
|
|
* stack_f[b] = f # <<<<<<<<<<<<<<
|
|
* if f != END:
|
|
* stack_b[f] = b
|
|
*/
|
|
__pyx_t_23 = __pyx_v_b;
|
|
__pyx_t_24 = -1;
|
|
if (__pyx_t_23 < 0) {
|
|
__pyx_t_23 += __pyx_pybuffernd_stack_f.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_23 < 0)) __pyx_t_24 = 0;
|
|
} else if (unlikely(__pyx_t_23 >= __pyx_pybuffernd_stack_f.diminfo[0].shape)) __pyx_t_24 = 0;
|
|
if (unlikely(__pyx_t_24 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_24);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 655; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_f.rcbuffer->pybuffer.buf, __pyx_t_23, __pyx_pybuffernd_stack_f.diminfo[0].strides) = __pyx_v_f;
|
|
goto __pyx_L13;
|
|
}
|
|
__pyx_L13:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":656
|
|
* if b != END:
|
|
* stack_f[b] = f
|
|
* if f != END: # <<<<<<<<<<<<<<
|
|
* stack_b[f] = b
|
|
*
|
|
*/
|
|
__pyx_t_11 = (__pyx_v_f != __pyx_v_END);
|
|
if (__pyx_t_11) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":657
|
|
* stack_f[b] = f
|
|
* if f != END:
|
|
* stack_b[f] = b # <<<<<<<<<<<<<<
|
|
*
|
|
* stack_f[w] = stack_head
|
|
*/
|
|
__pyx_t_24 = __pyx_v_f;
|
|
__pyx_t_25 = -1;
|
|
if (__pyx_t_24 < 0) {
|
|
__pyx_t_24 += __pyx_pybuffernd_stack_b.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_24 < 0)) __pyx_t_25 = 0;
|
|
} else if (unlikely(__pyx_t_24 >= __pyx_pybuffernd_stack_b.diminfo[0].shape)) __pyx_t_25 = 0;
|
|
if (unlikely(__pyx_t_25 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_25);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 657; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_b.rcbuffer->pybuffer.buf, __pyx_t_24, __pyx_pybuffernd_stack_b.diminfo[0].strides) = __pyx_v_b;
|
|
goto __pyx_L14;
|
|
}
|
|
__pyx_L14:;
|
|
goto __pyx_L12;
|
|
}
|
|
__pyx_L12:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":659
|
|
* stack_b[f] = b
|
|
*
|
|
* stack_f[w] = stack_head # <<<<<<<<<<<<<<
|
|
* stack_b[w] = END
|
|
* stack_b[stack_head] = w
|
|
*/
|
|
__pyx_t_25 = __pyx_v_w;
|
|
__pyx_t_26 = -1;
|
|
if (__pyx_t_25 < 0) {
|
|
__pyx_t_25 += __pyx_pybuffernd_stack_f.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_25 < 0)) __pyx_t_26 = 0;
|
|
} else if (unlikely(__pyx_t_25 >= __pyx_pybuffernd_stack_f.diminfo[0].shape)) __pyx_t_26 = 0;
|
|
if (unlikely(__pyx_t_26 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_26);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 659; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_f.rcbuffer->pybuffer.buf, __pyx_t_25, __pyx_pybuffernd_stack_f.diminfo[0].strides) = __pyx_v_stack_head;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":660
|
|
*
|
|
* stack_f[w] = stack_head
|
|
* stack_b[w] = END # <<<<<<<<<<<<<<
|
|
* stack_b[stack_head] = w
|
|
* stack_head = w
|
|
*/
|
|
__pyx_t_26 = __pyx_v_w;
|
|
__pyx_t_27 = -1;
|
|
if (__pyx_t_26 < 0) {
|
|
__pyx_t_26 += __pyx_pybuffernd_stack_b.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_26 < 0)) __pyx_t_27 = 0;
|
|
} else if (unlikely(__pyx_t_26 >= __pyx_pybuffernd_stack_b.diminfo[0].shape)) __pyx_t_27 = 0;
|
|
if (unlikely(__pyx_t_27 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_27);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 660; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_b.rcbuffer->pybuffer.buf, __pyx_t_26, __pyx_pybuffernd_stack_b.diminfo[0].strides) = __pyx_v_END;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":661
|
|
* stack_f[w] = stack_head
|
|
* stack_b[w] = END
|
|
* stack_b[stack_head] = w # <<<<<<<<<<<<<<
|
|
* stack_head = w
|
|
*
|
|
*/
|
|
__pyx_t_27 = __pyx_v_stack_head;
|
|
__pyx_t_28 = -1;
|
|
if (__pyx_t_27 < 0) {
|
|
__pyx_t_27 += __pyx_pybuffernd_stack_b.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_27 < 0)) __pyx_t_28 = 0;
|
|
} else if (unlikely(__pyx_t_27 >= __pyx_pybuffernd_stack_b.diminfo[0].shape)) __pyx_t_28 = 0;
|
|
if (unlikely(__pyx_t_28 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_28);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 661; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_b.rcbuffer->pybuffer.buf, __pyx_t_27, __pyx_pybuffernd_stack_b.diminfo[0].strides) = __pyx_v_w;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":662
|
|
* stack_b[w] = END
|
|
* stack_b[stack_head] = w
|
|
* stack_head = w # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_stack_head = __pyx_v_w;
|
|
goto __pyx_L11;
|
|
}
|
|
__pyx_L11:;
|
|
}
|
|
goto __pyx_L8;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":666
|
|
* else:
|
|
* # DFS-stack pop
|
|
* stack_head = stack_f[v] # <<<<<<<<<<<<<<
|
|
* if stack_head >= 0:
|
|
* stack_b[stack_head] = END
|
|
*/
|
|
__pyx_t_28 = __pyx_v_v;
|
|
__pyx_t_29 = -1;
|
|
if (__pyx_t_28 < 0) {
|
|
__pyx_t_28 += __pyx_pybuffernd_stack_f.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_28 < 0)) __pyx_t_29 = 0;
|
|
} else if (unlikely(__pyx_t_28 >= __pyx_pybuffernd_stack_f.diminfo[0].shape)) __pyx_t_29 = 0;
|
|
if (unlikely(__pyx_t_29 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_29);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 666; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_stack_head = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_f.rcbuffer->pybuffer.buf, __pyx_t_28, __pyx_pybuffernd_stack_f.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":667
|
|
* # DFS-stack pop
|
|
* stack_head = stack_f[v]
|
|
* if stack_head >= 0: # <<<<<<<<<<<<<<
|
|
* stack_b[stack_head] = END
|
|
* stack_f[v] = VOID
|
|
*/
|
|
__pyx_t_11 = (__pyx_v_stack_head >= 0);
|
|
if (__pyx_t_11) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":668
|
|
* stack_head = stack_f[v]
|
|
* if stack_head >= 0:
|
|
* stack_b[stack_head] = END # <<<<<<<<<<<<<<
|
|
* stack_f[v] = VOID
|
|
* stack_b[v] = VOID
|
|
*/
|
|
__pyx_t_29 = __pyx_v_stack_head;
|
|
__pyx_t_30 = -1;
|
|
if (__pyx_t_29 < 0) {
|
|
__pyx_t_29 += __pyx_pybuffernd_stack_b.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_29 < 0)) __pyx_t_30 = 0;
|
|
} else if (unlikely(__pyx_t_29 >= __pyx_pybuffernd_stack_b.diminfo[0].shape)) __pyx_t_30 = 0;
|
|
if (unlikely(__pyx_t_30 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_30);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 668; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_b.rcbuffer->pybuffer.buf, __pyx_t_29, __pyx_pybuffernd_stack_b.diminfo[0].strides) = __pyx_v_END;
|
|
goto __pyx_L15;
|
|
}
|
|
__pyx_L15:;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":669
|
|
* if stack_head >= 0:
|
|
* stack_b[stack_head] = END
|
|
* stack_f[v] = VOID # <<<<<<<<<<<<<<
|
|
* stack_b[v] = VOID
|
|
*
|
|
*/
|
|
__pyx_t_30 = __pyx_v_v;
|
|
__pyx_t_31 = -1;
|
|
if (__pyx_t_30 < 0) {
|
|
__pyx_t_30 += __pyx_pybuffernd_stack_f.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_30 < 0)) __pyx_t_31 = 0;
|
|
} else if (unlikely(__pyx_t_30 >= __pyx_pybuffernd_stack_f.diminfo[0].shape)) __pyx_t_31 = 0;
|
|
if (unlikely(__pyx_t_31 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_31);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 669; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_f.rcbuffer->pybuffer.buf, __pyx_t_30, __pyx_pybuffernd_stack_f.diminfo[0].strides) = __pyx_v_VOID;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":670
|
|
* stack_b[stack_head] = END
|
|
* stack_f[v] = VOID
|
|
* stack_b[v] = VOID # <<<<<<<<<<<<<<
|
|
*
|
|
* root = 1 # True
|
|
*/
|
|
__pyx_t_31 = __pyx_v_v;
|
|
__pyx_t_32 = -1;
|
|
if (__pyx_t_31 < 0) {
|
|
__pyx_t_31 += __pyx_pybuffernd_stack_b.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_31 < 0)) __pyx_t_32 = 0;
|
|
} else if (unlikely(__pyx_t_31 >= __pyx_pybuffernd_stack_b.diminfo[0].shape)) __pyx_t_32 = 0;
|
|
if (unlikely(__pyx_t_32 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_32);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 670; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_stack_b.rcbuffer->pybuffer.buf, __pyx_t_31, __pyx_pybuffernd_stack_b.diminfo[0].strides) = __pyx_v_VOID;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":672
|
|
* stack_b[v] = VOID
|
|
*
|
|
* root = 1 # True # <<<<<<<<<<<<<<
|
|
* low_v = lowlinks[v]
|
|
* for j from indptr[v] <= j < indptr[v+1]:
|
|
*/
|
|
__pyx_v_root = 1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":673
|
|
*
|
|
* root = 1 # True
|
|
* low_v = lowlinks[v] # <<<<<<<<<<<<<<
|
|
* for j from indptr[v] <= j < indptr[v+1]:
|
|
* low_w = lowlinks[indices[j]]
|
|
*/
|
|
__pyx_t_32 = __pyx_v_v;
|
|
__pyx_t_33 = -1;
|
|
if (__pyx_t_32 < 0) {
|
|
__pyx_t_32 += __pyx_pybuffernd_lowlinks.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_32 < 0)) __pyx_t_33 = 0;
|
|
} else if (unlikely(__pyx_t_32 >= __pyx_pybuffernd_lowlinks.diminfo[0].shape)) __pyx_t_33 = 0;
|
|
if (unlikely(__pyx_t_33 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_33);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 673; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_low_v = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_lowlinks.rcbuffer->pybuffer.buf, __pyx_t_32, __pyx_pybuffernd_lowlinks.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":674
|
|
* root = 1 # True
|
|
* low_v = lowlinks[v]
|
|
* for j from indptr[v] <= j < indptr[v+1]: # <<<<<<<<<<<<<<
|
|
* low_w = lowlinks[indices[j]]
|
|
* if low_w < low_v:
|
|
*/
|
|
__pyx_t_33 = __pyx_v_v;
|
|
__pyx_t_34 = -1;
|
|
if (__pyx_t_33 < 0) {
|
|
__pyx_t_33 += __pyx_pybuffernd_indptr.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_33 < 0)) __pyx_t_34 = 0;
|
|
} else if (unlikely(__pyx_t_33 >= __pyx_pybuffernd_indptr.diminfo[0].shape)) __pyx_t_34 = 0;
|
|
if (unlikely(__pyx_t_34 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_34);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 674; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_35 = (__pyx_v_v + 1);
|
|
__pyx_t_34 = -1;
|
|
if (__pyx_t_35 < 0) {
|
|
__pyx_t_35 += __pyx_pybuffernd_indptr.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_35 < 0)) __pyx_t_34 = 0;
|
|
} else if (unlikely(__pyx_t_35 >= __pyx_pybuffernd_indptr.diminfo[0].shape)) __pyx_t_34 = 0;
|
|
if (unlikely(__pyx_t_34 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_34);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 674; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_18 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr.rcbuffer->pybuffer.buf, __pyx_t_35, __pyx_pybuffernd_indptr.diminfo[0].strides));
|
|
for (__pyx_v_j = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr.rcbuffer->pybuffer.buf, __pyx_t_33, __pyx_pybuffernd_indptr.diminfo[0].strides)); __pyx_v_j < __pyx_t_18; __pyx_v_j++) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":675
|
|
* low_v = lowlinks[v]
|
|
* for j from indptr[v] <= j < indptr[v+1]:
|
|
* low_w = lowlinks[indices[j]] # <<<<<<<<<<<<<<
|
|
* if low_w < low_v:
|
|
* low_v = low_w
|
|
*/
|
|
__pyx_t_34 = __pyx_v_j;
|
|
__pyx_t_36 = -1;
|
|
if (__pyx_t_34 < 0) {
|
|
__pyx_t_34 += __pyx_pybuffernd_indices.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_34 < 0)) __pyx_t_36 = 0;
|
|
} else if (unlikely(__pyx_t_34 >= __pyx_pybuffernd_indices.diminfo[0].shape)) __pyx_t_36 = 0;
|
|
if (unlikely(__pyx_t_36 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_36);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 675; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_37 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indices.rcbuffer->pybuffer.buf, __pyx_t_34, __pyx_pybuffernd_indices.diminfo[0].strides));
|
|
__pyx_t_36 = -1;
|
|
if (__pyx_t_37 < 0) {
|
|
__pyx_t_37 += __pyx_pybuffernd_lowlinks.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_37 < 0)) __pyx_t_36 = 0;
|
|
} else if (unlikely(__pyx_t_37 >= __pyx_pybuffernd_lowlinks.diminfo[0].shape)) __pyx_t_36 = 0;
|
|
if (unlikely(__pyx_t_36 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_36);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 675; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_low_w = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_lowlinks.rcbuffer->pybuffer.buf, __pyx_t_37, __pyx_pybuffernd_lowlinks.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":676
|
|
* for j from indptr[v] <= j < indptr[v+1]:
|
|
* low_w = lowlinks[indices[j]]
|
|
* if low_w < low_v: # <<<<<<<<<<<<<<
|
|
* low_v = low_w
|
|
* root = 0 # False
|
|
*/
|
|
__pyx_t_11 = (__pyx_v_low_w < __pyx_v_low_v);
|
|
if (__pyx_t_11) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":677
|
|
* low_w = lowlinks[indices[j]]
|
|
* if low_w < low_v:
|
|
* low_v = low_w # <<<<<<<<<<<<<<
|
|
* root = 0 # False
|
|
* lowlinks[v] = low_v
|
|
*/
|
|
__pyx_v_low_v = __pyx_v_low_w;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":678
|
|
* if low_w < low_v:
|
|
* low_v = low_w
|
|
* root = 0 # False # <<<<<<<<<<<<<<
|
|
* lowlinks[v] = low_v
|
|
*
|
|
*/
|
|
__pyx_v_root = 0;
|
|
goto __pyx_L18;
|
|
}
|
|
__pyx_L18:;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":679
|
|
* low_v = low_w
|
|
* root = 0 # False
|
|
* lowlinks[v] = low_v # <<<<<<<<<<<<<<
|
|
*
|
|
* if root: # Found a root node
|
|
*/
|
|
__pyx_t_36 = __pyx_v_v;
|
|
__pyx_t_38 = -1;
|
|
if (__pyx_t_36 < 0) {
|
|
__pyx_t_36 += __pyx_pybuffernd_lowlinks.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_36 < 0)) __pyx_t_38 = 0;
|
|
} else if (unlikely(__pyx_t_36 >= __pyx_pybuffernd_lowlinks.diminfo[0].shape)) __pyx_t_38 = 0;
|
|
if (unlikely(__pyx_t_38 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_38);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 679; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_lowlinks.rcbuffer->pybuffer.buf, __pyx_t_36, __pyx_pybuffernd_lowlinks.diminfo[0].strides) = __pyx_v_low_v;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":681
|
|
* lowlinks[v] = low_v
|
|
*
|
|
* if root: # Found a root node # <<<<<<<<<<<<<<
|
|
* index -= 1
|
|
* # while S not empty and rindex[v] <= rindex[top[S]
|
|
*/
|
|
if (__pyx_v_root) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":682
|
|
*
|
|
* if root: # Found a root node
|
|
* index -= 1 # <<<<<<<<<<<<<<
|
|
* # while S not empty and rindex[v] <= rindex[top[S]
|
|
* while SS_head != END and lowlinks[v] <= lowlinks[SS_head]:
|
|
*/
|
|
__pyx_v_index = (__pyx_v_index - 1);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":684
|
|
* index -= 1
|
|
* # while S not empty and rindex[v] <= rindex[top[S]
|
|
* while SS_head != END and lowlinks[v] <= lowlinks[SS_head]: # <<<<<<<<<<<<<<
|
|
* w = SS_head # w = pop(S)
|
|
* SS_head = SS[w]
|
|
*/
|
|
while (1) {
|
|
__pyx_t_11 = (__pyx_v_SS_head != __pyx_v_END);
|
|
if (__pyx_t_11) {
|
|
__pyx_t_38 = __pyx_v_v;
|
|
__pyx_t_39 = -1;
|
|
if (__pyx_t_38 < 0) {
|
|
__pyx_t_38 += __pyx_pybuffernd_lowlinks.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_38 < 0)) __pyx_t_39 = 0;
|
|
} else if (unlikely(__pyx_t_38 >= __pyx_pybuffernd_lowlinks.diminfo[0].shape)) __pyx_t_39 = 0;
|
|
if (unlikely(__pyx_t_39 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_39);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 684; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_39 = __pyx_v_SS_head;
|
|
__pyx_t_40 = -1;
|
|
if (__pyx_t_39 < 0) {
|
|
__pyx_t_39 += __pyx_pybuffernd_lowlinks.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_39 < 0)) __pyx_t_40 = 0;
|
|
} else if (unlikely(__pyx_t_39 >= __pyx_pybuffernd_lowlinks.diminfo[0].shape)) __pyx_t_40 = 0;
|
|
if (unlikely(__pyx_t_40 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_40);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 684; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_41 = ((*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_lowlinks.rcbuffer->pybuffer.buf, __pyx_t_38, __pyx_pybuffernd_lowlinks.diminfo[0].strides)) <= (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_lowlinks.rcbuffer->pybuffer.buf, __pyx_t_39, __pyx_pybuffernd_lowlinks.diminfo[0].strides)));
|
|
__pyx_t_42 = __pyx_t_41;
|
|
} else {
|
|
__pyx_t_42 = __pyx_t_11;
|
|
}
|
|
if (!__pyx_t_42) break;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":685
|
|
* # while S not empty and rindex[v] <= rindex[top[S]
|
|
* while SS_head != END and lowlinks[v] <= lowlinks[SS_head]:
|
|
* w = SS_head # w = pop(S) # <<<<<<<<<<<<<<
|
|
* SS_head = SS[w]
|
|
* SS[w] = VOID
|
|
*/
|
|
__pyx_v_w = __pyx_v_SS_head;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":686
|
|
* while SS_head != END and lowlinks[v] <= lowlinks[SS_head]:
|
|
* w = SS_head # w = pop(S)
|
|
* SS_head = SS[w] # <<<<<<<<<<<<<<
|
|
* SS[w] = VOID
|
|
*
|
|
*/
|
|
__pyx_t_40 = __pyx_v_w;
|
|
__pyx_t_43 = -1;
|
|
if (__pyx_t_40 < 0) {
|
|
__pyx_t_40 += __pyx_pybuffernd_SS.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_40 < 0)) __pyx_t_43 = 0;
|
|
} else if (unlikely(__pyx_t_40 >= __pyx_pybuffernd_SS.diminfo[0].shape)) __pyx_t_43 = 0;
|
|
if (unlikely(__pyx_t_43 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_43);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 686; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_SS_head = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_SS.rcbuffer->pybuffer.buf, __pyx_t_40, __pyx_pybuffernd_SS.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":687
|
|
* w = SS_head # w = pop(S)
|
|
* SS_head = SS[w]
|
|
* SS[w] = VOID # <<<<<<<<<<<<<<
|
|
*
|
|
* labels[w] = label # rindex[w] = c
|
|
*/
|
|
__pyx_t_43 = __pyx_v_w;
|
|
__pyx_t_44 = -1;
|
|
if (__pyx_t_43 < 0) {
|
|
__pyx_t_43 += __pyx_pybuffernd_SS.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_43 < 0)) __pyx_t_44 = 0;
|
|
} else if (unlikely(__pyx_t_43 >= __pyx_pybuffernd_SS.diminfo[0].shape)) __pyx_t_44 = 0;
|
|
if (unlikely(__pyx_t_44 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_44);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 687; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_SS.rcbuffer->pybuffer.buf, __pyx_t_43, __pyx_pybuffernd_SS.diminfo[0].strides) = __pyx_v_VOID;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":689
|
|
* SS[w] = VOID
|
|
*
|
|
* labels[w] = label # rindex[w] = c # <<<<<<<<<<<<<<
|
|
* index -= 1 # index = index - 1
|
|
* labels[v] = label # rindex[v] = c
|
|
*/
|
|
__pyx_t_44 = __pyx_v_w;
|
|
__pyx_t_45 = -1;
|
|
if (__pyx_t_44 < 0) {
|
|
__pyx_t_44 += __pyx_pybuffernd_labels.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_44 < 0)) __pyx_t_45 = 0;
|
|
} else if (unlikely(__pyx_t_44 >= __pyx_pybuffernd_labels.diminfo[0].shape)) __pyx_t_45 = 0;
|
|
if (unlikely(__pyx_t_45 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_45);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 689; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_labels.rcbuffer->pybuffer.buf, __pyx_t_44, __pyx_pybuffernd_labels.diminfo[0].strides) = __pyx_v_label;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":690
|
|
*
|
|
* labels[w] = label # rindex[w] = c
|
|
* index -= 1 # index = index - 1 # <<<<<<<<<<<<<<
|
|
* labels[v] = label # rindex[v] = c
|
|
* label -= 1 # c = c - 1
|
|
*/
|
|
__pyx_v_index = (__pyx_v_index - 1);
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":691
|
|
* labels[w] = label # rindex[w] = c
|
|
* index -= 1 # index = index - 1
|
|
* labels[v] = label # rindex[v] = c # <<<<<<<<<<<<<<
|
|
* label -= 1 # c = c - 1
|
|
* else:
|
|
*/
|
|
__pyx_t_45 = __pyx_v_v;
|
|
__pyx_t_46 = -1;
|
|
if (__pyx_t_45 < 0) {
|
|
__pyx_t_45 += __pyx_pybuffernd_labels.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_45 < 0)) __pyx_t_46 = 0;
|
|
} else if (unlikely(__pyx_t_45 >= __pyx_pybuffernd_labels.diminfo[0].shape)) __pyx_t_46 = 0;
|
|
if (unlikely(__pyx_t_46 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_46);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 691; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_labels.rcbuffer->pybuffer.buf, __pyx_t_45, __pyx_pybuffernd_labels.diminfo[0].strides) = __pyx_v_label;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":692
|
|
* index -= 1 # index = index - 1
|
|
* labels[v] = label # rindex[v] = c
|
|
* label -= 1 # c = c - 1 # <<<<<<<<<<<<<<
|
|
* else:
|
|
* SS[v] = SS_head # push(S, v)
|
|
*/
|
|
__pyx_v_label = (__pyx_v_label - 1);
|
|
goto __pyx_L19;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":694
|
|
* label -= 1 # c = c - 1
|
|
* else:
|
|
* SS[v] = SS_head # push(S, v) # <<<<<<<<<<<<<<
|
|
* SS_head = v
|
|
*
|
|
*/
|
|
__pyx_t_46 = __pyx_v_v;
|
|
__pyx_t_47 = -1;
|
|
if (__pyx_t_46 < 0) {
|
|
__pyx_t_46 += __pyx_pybuffernd_SS.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_46 < 0)) __pyx_t_47 = 0;
|
|
} else if (unlikely(__pyx_t_46 >= __pyx_pybuffernd_SS.diminfo[0].shape)) __pyx_t_47 = 0;
|
|
if (unlikely(__pyx_t_47 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_47);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 694; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_SS.rcbuffer->pybuffer.buf, __pyx_t_46, __pyx_pybuffernd_SS.diminfo[0].strides) = __pyx_v_SS_head;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":695
|
|
* else:
|
|
* SS[v] = SS_head # push(S, v)
|
|
* SS_head = v # <<<<<<<<<<<<<<
|
|
*
|
|
* # labels count down from N-1 to zero. Modify them so they
|
|
*/
|
|
__pyx_v_SS_head = __pyx_v_v;
|
|
}
|
|
__pyx_L19:;
|
|
}
|
|
__pyx_L8:;
|
|
}
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":699
|
|
* # labels count down from N-1 to zero. Modify them so they
|
|
* # count upward from 0
|
|
* labels *= -1 # <<<<<<<<<<<<<<
|
|
* labels += (N - 1)
|
|
* return (N - 1) - label
|
|
*/
|
|
__pyx_t_6 = PyNumber_InPlaceMultiply(((PyObject *)__pyx_v_labels), __pyx_int_neg_1); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 699; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
if (!(likely(((__pyx_t_6) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_6, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 699; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_48 = ((PyArrayObject *)__pyx_t_6);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
|
|
__pyx_t_1 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_labels.rcbuffer->pybuffer, (PyObject*)__pyx_t_48, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack);
|
|
if (unlikely(__pyx_t_1 < 0)) {
|
|
PyErr_Fetch(&__pyx_t_2, &__pyx_t_3, &__pyx_t_4);
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_labels.rcbuffer->pybuffer, (PyObject*)__pyx_v_labels, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {
|
|
Py_XDECREF(__pyx_t_2); Py_XDECREF(__pyx_t_3); Py_XDECREF(__pyx_t_4);
|
|
__Pyx_RaiseBufferFallbackError();
|
|
} else {
|
|
PyErr_Restore(__pyx_t_2, __pyx_t_3, __pyx_t_4);
|
|
}
|
|
}
|
|
__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];
|
|
if (unlikely(__pyx_t_1 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 699; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_48 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_v_labels));
|
|
__pyx_v_labels = ((PyArrayObject *)__pyx_t_6);
|
|
__pyx_t_6 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":700
|
|
* # count upward from 0
|
|
* labels *= -1
|
|
* labels += (N - 1) # <<<<<<<<<<<<<<
|
|
* return (N - 1) - label
|
|
*
|
|
*/
|
|
__pyx_t_6 = PyInt_FromLong((__pyx_v_N - 1)); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 700; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__pyx_t_7 = PyNumber_InPlaceAdd(((PyObject *)__pyx_v_labels), __pyx_t_6); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 700; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
if (!(likely(((__pyx_t_7) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_7, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 700; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_48 = ((PyArrayObject *)__pyx_t_7);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
|
|
__pyx_t_1 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_labels.rcbuffer->pybuffer, (PyObject*)__pyx_t_48, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack);
|
|
if (unlikely(__pyx_t_1 < 0)) {
|
|
PyErr_Fetch(&__pyx_t_4, &__pyx_t_3, &__pyx_t_2);
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_labels.rcbuffer->pybuffer, (PyObject*)__pyx_v_labels, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {
|
|
Py_XDECREF(__pyx_t_4); Py_XDECREF(__pyx_t_3); Py_XDECREF(__pyx_t_2);
|
|
__Pyx_RaiseBufferFallbackError();
|
|
} else {
|
|
PyErr_Restore(__pyx_t_4, __pyx_t_3, __pyx_t_2);
|
|
}
|
|
}
|
|
__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];
|
|
if (unlikely(__pyx_t_1 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 700; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_48 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_v_labels));
|
|
__pyx_v_labels = ((PyArrayObject *)__pyx_t_7);
|
|
__pyx_t_7 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":701
|
|
* labels *= -1
|
|
* labels += (N - 1)
|
|
* return (N - 1) - label # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef int _connected_components_undirected(
|
|
*/
|
|
__pyx_r = ((__pyx_v_N - 1) - __pyx_v_label);
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_5);
|
|
__Pyx_XDECREF(__pyx_t_6);
|
|
__Pyx_XDECREF(__pyx_t_7);
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_SS.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_lowlinks.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_stack_b.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_stack_f.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_WriteUnraisable("sklearn.utils.mst._traversal._connected_components_directed", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_SS.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_lowlinks.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_stack_b.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_stack_f.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_SS);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_lowlinks);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_stack_f);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_stack_b);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_labels);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":703
|
|
* return (N - 1) - label
|
|
*
|
|
* cdef int _connected_components_undirected( # <<<<<<<<<<<<<<
|
|
* np.ndarray[ITYPE_t, ndim=1, mode='c'] indices1,
|
|
* np.ndarray[ITYPE_t, ndim=1, mode='c'] indptr1,
|
|
*/
|
|
|
|
static int __pyx_f_7sklearn_5utils_3mst_10_traversal__connected_components_undirected(PyArrayObject *__pyx_v_indices1, PyArrayObject *__pyx_v_indptr1, PyArrayObject *__pyx_v_indices2, PyArrayObject *__pyx_v_indptr2, PyArrayObject *__pyx_v_labels) {
|
|
int __pyx_v_v;
|
|
int __pyx_v_w;
|
|
int __pyx_v_j;
|
|
int __pyx_v_label;
|
|
int __pyx_v_SS_head;
|
|
int __pyx_v_N;
|
|
int __pyx_v_VOID;
|
|
int __pyx_v_END;
|
|
PyArrayObject *__pyx_v_SS = 0;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_SS;
|
|
__Pyx_Buffer __pyx_pybuffer_SS;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indices1;
|
|
__Pyx_Buffer __pyx_pybuffer_indices1;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indices2;
|
|
__Pyx_Buffer __pyx_pybuffer_indices2;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indptr1;
|
|
__Pyx_Buffer __pyx_pybuffer_indptr1;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indptr2;
|
|
__Pyx_Buffer __pyx_pybuffer_indptr2;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_labels;
|
|
__Pyx_Buffer __pyx_pybuffer_labels;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
int __pyx_t_4;
|
|
int __pyx_t_5;
|
|
int __pyx_t_6;
|
|
int __pyx_t_7;
|
|
int __pyx_t_8;
|
|
int __pyx_t_9;
|
|
int __pyx_t_10;
|
|
int __pyx_t_11;
|
|
int __pyx_t_12;
|
|
long __pyx_t_13;
|
|
__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t __pyx_t_14;
|
|
int __pyx_t_15;
|
|
int __pyx_t_16;
|
|
int __pyx_t_17;
|
|
int __pyx_t_18;
|
|
long __pyx_t_19;
|
|
int __pyx_t_20;
|
|
int __pyx_t_21;
|
|
int __pyx_t_22;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_connected_components_undirected", 0);
|
|
__pyx_pybuffer_SS.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_SS.refcount = 0;
|
|
__pyx_pybuffernd_SS.data = NULL;
|
|
__pyx_pybuffernd_SS.rcbuffer = &__pyx_pybuffer_SS;
|
|
__pyx_pybuffer_indices1.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indices1.refcount = 0;
|
|
__pyx_pybuffernd_indices1.data = NULL;
|
|
__pyx_pybuffernd_indices1.rcbuffer = &__pyx_pybuffer_indices1;
|
|
__pyx_pybuffer_indptr1.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indptr1.refcount = 0;
|
|
__pyx_pybuffernd_indptr1.data = NULL;
|
|
__pyx_pybuffernd_indptr1.rcbuffer = &__pyx_pybuffer_indptr1;
|
|
__pyx_pybuffer_indices2.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indices2.refcount = 0;
|
|
__pyx_pybuffernd_indices2.data = NULL;
|
|
__pyx_pybuffernd_indices2.rcbuffer = &__pyx_pybuffer_indices2;
|
|
__pyx_pybuffer_indptr2.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indptr2.refcount = 0;
|
|
__pyx_pybuffernd_indptr2.data = NULL;
|
|
__pyx_pybuffernd_indptr2.rcbuffer = &__pyx_pybuffer_indptr2;
|
|
__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_indices1.rcbuffer->pybuffer, (PyObject*)__pyx_v_indices1, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 703; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indices1.diminfo[0].strides = __pyx_pybuffernd_indices1.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indices1.diminfo[0].shape = __pyx_pybuffernd_indices1.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indptr1.rcbuffer->pybuffer, (PyObject*)__pyx_v_indptr1, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 703; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indptr1.diminfo[0].strides = __pyx_pybuffernd_indptr1.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indptr1.diminfo[0].shape = __pyx_pybuffernd_indptr1.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indices2.rcbuffer->pybuffer, (PyObject*)__pyx_v_indices2, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 703; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indices2.diminfo[0].strides = __pyx_pybuffernd_indices2.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indices2.diminfo[0].shape = __pyx_pybuffernd_indices2.rcbuffer->pybuffer.shape[0];
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indptr2.rcbuffer->pybuffer, (PyObject*)__pyx_v_indptr2, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 703; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_indptr2.diminfo[0].strides = __pyx_pybuffernd_indptr2.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indptr2.diminfo[0].shape = __pyx_pybuffernd_indptr2.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___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 703; __pyx_clineno = __LINE__; goto __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];
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":711
|
|
*
|
|
* cdef int v, w, j, label, SS_head
|
|
* cdef int N = labels.shape[0] # <<<<<<<<<<<<<<
|
|
* cdef int VOID = -1
|
|
* cdef int END = -2
|
|
*/
|
|
__pyx_v_N = (__pyx_v_labels->dimensions[0]);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":712
|
|
* cdef int v, w, j, label, SS_head
|
|
* cdef int N = labels.shape[0]
|
|
* cdef int VOID = -1 # <<<<<<<<<<<<<<
|
|
* cdef int END = -2
|
|
* labels.fill(VOID)
|
|
*/
|
|
__pyx_v_VOID = -1;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":713
|
|
* cdef int N = labels.shape[0]
|
|
* cdef int VOID = -1
|
|
* cdef int END = -2 # <<<<<<<<<<<<<<
|
|
* labels.fill(VOID)
|
|
* label = 0
|
|
*/
|
|
__pyx_v_END = -2;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":714
|
|
* cdef int VOID = -1
|
|
* cdef int END = -2
|
|
* labels.fill(VOID) # <<<<<<<<<<<<<<
|
|
* label = 0
|
|
*
|
|
*/
|
|
__pyx_t_1 = PyObject_GetAttr(((PyObject *)__pyx_v_labels), __pyx_n_s__fill); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 714; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyInt_FromLong(__pyx_v_VOID); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 714; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 714; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_t_3), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 714; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":715
|
|
* cdef int END = -2
|
|
* labels.fill(VOID)
|
|
* label = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Share memory for the stack and labels, since labels are only
|
|
*/
|
|
__pyx_v_label = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":719
|
|
* # Share memory for the stack and labels, since labels are only
|
|
* # applied once a node has been popped from the stack.
|
|
* cdef np.ndarray[ITYPE_t, ndim=1, mode="c"] SS = labels # <<<<<<<<<<<<<<
|
|
* SS_head = END
|
|
* for v in range(N):
|
|
*/
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_SS.rcbuffer->pybuffer, (PyObject*)((PyArrayObject *)__pyx_v_labels), &__Pyx_TypeInfo_nn___pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {
|
|
__pyx_v_SS = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_pybuffernd_SS.rcbuffer->pybuffer.buf = NULL;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 719; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
} else {__pyx_pybuffernd_SS.diminfo[0].strides = __pyx_pybuffernd_SS.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_SS.diminfo[0].shape = __pyx_pybuffernd_SS.rcbuffer->pybuffer.shape[0];
|
|
}
|
|
}
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_labels));
|
|
__pyx_v_SS = ((PyArrayObject *)__pyx_v_labels);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":720
|
|
* # applied once a node has been popped from the stack.
|
|
* cdef np.ndarray[ITYPE_t, ndim=1, mode="c"] SS = labels
|
|
* SS_head = END # <<<<<<<<<<<<<<
|
|
* for v in range(N):
|
|
* if labels[v] == VOID:
|
|
*/
|
|
__pyx_v_SS_head = __pyx_v_END;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":721
|
|
* cdef np.ndarray[ITYPE_t, ndim=1, mode="c"] SS = labels
|
|
* SS_head = END
|
|
* for v in range(N): # <<<<<<<<<<<<<<
|
|
* if labels[v] == VOID:
|
|
* # SS.push(v)
|
|
*/
|
|
__pyx_t_4 = __pyx_v_N;
|
|
for (__pyx_t_5 = 0; __pyx_t_5 < __pyx_t_4; __pyx_t_5+=1) {
|
|
__pyx_v_v = __pyx_t_5;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":722
|
|
* SS_head = END
|
|
* for v in range(N):
|
|
* if labels[v] == VOID: # <<<<<<<<<<<<<<
|
|
* # SS.push(v)
|
|
* SS_head = v
|
|
*/
|
|
__pyx_t_6 = __pyx_v_v;
|
|
__pyx_t_7 = -1;
|
|
if (__pyx_t_6 < 0) {
|
|
__pyx_t_6 += __pyx_pybuffernd_labels.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_6 < 0)) __pyx_t_7 = 0;
|
|
} else if (unlikely(__pyx_t_6 >= __pyx_pybuffernd_labels.diminfo[0].shape)) __pyx_t_7 = 0;
|
|
if (unlikely(__pyx_t_7 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_7);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 722; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_8 = ((*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_labels.rcbuffer->pybuffer.buf, __pyx_t_6, __pyx_pybuffernd_labels.diminfo[0].strides)) == __pyx_v_VOID);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":724
|
|
* if labels[v] == VOID:
|
|
* # SS.push(v)
|
|
* SS_head = v # <<<<<<<<<<<<<<
|
|
* SS[v] = END
|
|
*
|
|
*/
|
|
__pyx_v_SS_head = __pyx_v_v;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":725
|
|
* # SS.push(v)
|
|
* SS_head = v
|
|
* SS[v] = END # <<<<<<<<<<<<<<
|
|
*
|
|
* while SS_head != END:
|
|
*/
|
|
__pyx_t_7 = __pyx_v_v;
|
|
__pyx_t_9 = -1;
|
|
if (__pyx_t_7 < 0) {
|
|
__pyx_t_7 += __pyx_pybuffernd_SS.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_7 < 0)) __pyx_t_9 = 0;
|
|
} else if (unlikely(__pyx_t_7 >= __pyx_pybuffernd_SS.diminfo[0].shape)) __pyx_t_9 = 0;
|
|
if (unlikely(__pyx_t_9 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_9);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 725; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_SS.rcbuffer->pybuffer.buf, __pyx_t_7, __pyx_pybuffernd_SS.diminfo[0].strides) = __pyx_v_END;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":727
|
|
* SS[v] = END
|
|
*
|
|
* while SS_head != END: # <<<<<<<<<<<<<<
|
|
* # v = SS.pop()
|
|
* v = SS_head
|
|
*/
|
|
while (1) {
|
|
__pyx_t_8 = (__pyx_v_SS_head != __pyx_v_END);
|
|
if (!__pyx_t_8) break;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":729
|
|
* while SS_head != END:
|
|
* # v = SS.pop()
|
|
* v = SS_head # <<<<<<<<<<<<<<
|
|
* SS_head = SS[v]
|
|
*
|
|
*/
|
|
__pyx_v_v = __pyx_v_SS_head;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":730
|
|
* # v = SS.pop()
|
|
* v = SS_head
|
|
* SS_head = SS[v] # <<<<<<<<<<<<<<
|
|
*
|
|
* labels[v] = label
|
|
*/
|
|
__pyx_t_9 = __pyx_v_v;
|
|
__pyx_t_10 = -1;
|
|
if (__pyx_t_9 < 0) {
|
|
__pyx_t_9 += __pyx_pybuffernd_SS.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_9 < 0)) __pyx_t_10 = 0;
|
|
} else if (unlikely(__pyx_t_9 >= __pyx_pybuffernd_SS.diminfo[0].shape)) __pyx_t_10 = 0;
|
|
if (unlikely(__pyx_t_10 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_10);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 730; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_SS_head = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_SS.rcbuffer->pybuffer.buf, __pyx_t_9, __pyx_pybuffernd_SS.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":732
|
|
* SS_head = SS[v]
|
|
*
|
|
* labels[v] = label # <<<<<<<<<<<<<<
|
|
*
|
|
* # Push children onto the stack if they havn't been
|
|
*/
|
|
__pyx_t_10 = __pyx_v_v;
|
|
__pyx_t_11 = -1;
|
|
if (__pyx_t_10 < 0) {
|
|
__pyx_t_10 += __pyx_pybuffernd_labels.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_10 < 0)) __pyx_t_11 = 0;
|
|
} else if (unlikely(__pyx_t_10 >= __pyx_pybuffernd_labels.diminfo[0].shape)) __pyx_t_11 = 0;
|
|
if (unlikely(__pyx_t_11 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_11);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 732; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_labels.rcbuffer->pybuffer.buf, __pyx_t_10, __pyx_pybuffernd_labels.diminfo[0].strides) = __pyx_v_label;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":736
|
|
* # Push children onto the stack if they havn't been
|
|
* # seen at all yet.
|
|
* for j from indptr1[v] <= j < indptr1[v+1]: # <<<<<<<<<<<<<<
|
|
* w = indices1[j]
|
|
* if SS[w] == VOID:
|
|
*/
|
|
__pyx_t_11 = __pyx_v_v;
|
|
__pyx_t_12 = -1;
|
|
if (__pyx_t_11 < 0) {
|
|
__pyx_t_11 += __pyx_pybuffernd_indptr1.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_11 < 0)) __pyx_t_12 = 0;
|
|
} else if (unlikely(__pyx_t_11 >= __pyx_pybuffernd_indptr1.diminfo[0].shape)) __pyx_t_12 = 0;
|
|
if (unlikely(__pyx_t_12 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_12);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 736; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_13 = (__pyx_v_v + 1);
|
|
__pyx_t_12 = -1;
|
|
if (__pyx_t_13 < 0) {
|
|
__pyx_t_13 += __pyx_pybuffernd_indptr1.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_13 < 0)) __pyx_t_12 = 0;
|
|
} else if (unlikely(__pyx_t_13 >= __pyx_pybuffernd_indptr1.diminfo[0].shape)) __pyx_t_12 = 0;
|
|
if (unlikely(__pyx_t_12 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_12);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 736; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_14 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr1.rcbuffer->pybuffer.buf, __pyx_t_13, __pyx_pybuffernd_indptr1.diminfo[0].strides));
|
|
for (__pyx_v_j = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr1.rcbuffer->pybuffer.buf, __pyx_t_11, __pyx_pybuffernd_indptr1.diminfo[0].strides)); __pyx_v_j < __pyx_t_14; __pyx_v_j++) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":737
|
|
* # seen at all yet.
|
|
* for j from indptr1[v] <= j < indptr1[v+1]:
|
|
* w = indices1[j] # <<<<<<<<<<<<<<
|
|
* if SS[w] == VOID:
|
|
* SS[w] = SS_head
|
|
*/
|
|
__pyx_t_12 = __pyx_v_j;
|
|
__pyx_t_15 = -1;
|
|
if (__pyx_t_12 < 0) {
|
|
__pyx_t_12 += __pyx_pybuffernd_indices1.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_12 < 0)) __pyx_t_15 = 0;
|
|
} else if (unlikely(__pyx_t_12 >= __pyx_pybuffernd_indices1.diminfo[0].shape)) __pyx_t_15 = 0;
|
|
if (unlikely(__pyx_t_15 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_15);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 737; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_w = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indices1.rcbuffer->pybuffer.buf, __pyx_t_12, __pyx_pybuffernd_indices1.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":738
|
|
* for j from indptr1[v] <= j < indptr1[v+1]:
|
|
* w = indices1[j]
|
|
* if SS[w] == VOID: # <<<<<<<<<<<<<<
|
|
* SS[w] = SS_head
|
|
* SS_head = w
|
|
*/
|
|
__pyx_t_15 = __pyx_v_w;
|
|
__pyx_t_16 = -1;
|
|
if (__pyx_t_15 < 0) {
|
|
__pyx_t_15 += __pyx_pybuffernd_SS.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_15 < 0)) __pyx_t_16 = 0;
|
|
} else if (unlikely(__pyx_t_15 >= __pyx_pybuffernd_SS.diminfo[0].shape)) __pyx_t_16 = 0;
|
|
if (unlikely(__pyx_t_16 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_16);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 738; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_8 = ((*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_SS.rcbuffer->pybuffer.buf, __pyx_t_15, __pyx_pybuffernd_SS.diminfo[0].strides)) == __pyx_v_VOID);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":739
|
|
* w = indices1[j]
|
|
* if SS[w] == VOID:
|
|
* SS[w] = SS_head # <<<<<<<<<<<<<<
|
|
* SS_head = w
|
|
* for j from indptr2[v] <= j < indptr2[v+1]:
|
|
*/
|
|
__pyx_t_16 = __pyx_v_w;
|
|
__pyx_t_17 = -1;
|
|
if (__pyx_t_16 < 0) {
|
|
__pyx_t_16 += __pyx_pybuffernd_SS.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_16 < 0)) __pyx_t_17 = 0;
|
|
} else if (unlikely(__pyx_t_16 >= __pyx_pybuffernd_SS.diminfo[0].shape)) __pyx_t_17 = 0;
|
|
if (unlikely(__pyx_t_17 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_17);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 739; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_SS.rcbuffer->pybuffer.buf, __pyx_t_16, __pyx_pybuffernd_SS.diminfo[0].strides) = __pyx_v_SS_head;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":740
|
|
* if SS[w] == VOID:
|
|
* SS[w] = SS_head
|
|
* SS_head = w # <<<<<<<<<<<<<<
|
|
* for j from indptr2[v] <= j < indptr2[v+1]:
|
|
* w = indices2[j]
|
|
*/
|
|
__pyx_v_SS_head = __pyx_v_w;
|
|
goto __pyx_L10;
|
|
}
|
|
__pyx_L10:;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":741
|
|
* SS[w] = SS_head
|
|
* SS_head = w
|
|
* for j from indptr2[v] <= j < indptr2[v+1]: # <<<<<<<<<<<<<<
|
|
* w = indices2[j]
|
|
* if SS[w] == VOID:
|
|
*/
|
|
__pyx_t_17 = __pyx_v_v;
|
|
__pyx_t_18 = -1;
|
|
if (__pyx_t_17 < 0) {
|
|
__pyx_t_17 += __pyx_pybuffernd_indptr2.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_17 < 0)) __pyx_t_18 = 0;
|
|
} else if (unlikely(__pyx_t_17 >= __pyx_pybuffernd_indptr2.diminfo[0].shape)) __pyx_t_18 = 0;
|
|
if (unlikely(__pyx_t_18 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_18);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 741; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_19 = (__pyx_v_v + 1);
|
|
__pyx_t_18 = -1;
|
|
if (__pyx_t_19 < 0) {
|
|
__pyx_t_19 += __pyx_pybuffernd_indptr2.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_19 < 0)) __pyx_t_18 = 0;
|
|
} else if (unlikely(__pyx_t_19 >= __pyx_pybuffernd_indptr2.diminfo[0].shape)) __pyx_t_18 = 0;
|
|
if (unlikely(__pyx_t_18 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_18);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 741; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_14 = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr2.rcbuffer->pybuffer.buf, __pyx_t_19, __pyx_pybuffernd_indptr2.diminfo[0].strides));
|
|
for (__pyx_v_j = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indptr2.rcbuffer->pybuffer.buf, __pyx_t_17, __pyx_pybuffernd_indptr2.diminfo[0].strides)); __pyx_v_j < __pyx_t_14; __pyx_v_j++) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":742
|
|
* SS_head = w
|
|
* for j from indptr2[v] <= j < indptr2[v+1]:
|
|
* w = indices2[j] # <<<<<<<<<<<<<<
|
|
* if SS[w] == VOID:
|
|
* SS[w] = SS_head
|
|
*/
|
|
__pyx_t_18 = __pyx_v_j;
|
|
__pyx_t_20 = -1;
|
|
if (__pyx_t_18 < 0) {
|
|
__pyx_t_18 += __pyx_pybuffernd_indices2.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_18 < 0)) __pyx_t_20 = 0;
|
|
} else if (unlikely(__pyx_t_18 >= __pyx_pybuffernd_indices2.diminfo[0].shape)) __pyx_t_20 = 0;
|
|
if (unlikely(__pyx_t_20 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_20);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 742; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_v_w = (*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_indices2.rcbuffer->pybuffer.buf, __pyx_t_18, __pyx_pybuffernd_indices2.diminfo[0].strides));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":743
|
|
* for j from indptr2[v] <= j < indptr2[v+1]:
|
|
* w = indices2[j]
|
|
* if SS[w] == VOID: # <<<<<<<<<<<<<<
|
|
* SS[w] = SS_head
|
|
* SS_head = w
|
|
*/
|
|
__pyx_t_20 = __pyx_v_w;
|
|
__pyx_t_21 = -1;
|
|
if (__pyx_t_20 < 0) {
|
|
__pyx_t_20 += __pyx_pybuffernd_SS.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_20 < 0)) __pyx_t_21 = 0;
|
|
} else if (unlikely(__pyx_t_20 >= __pyx_pybuffernd_SS.diminfo[0].shape)) __pyx_t_21 = 0;
|
|
if (unlikely(__pyx_t_21 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_21);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 743; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_8 = ((*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_SS.rcbuffer->pybuffer.buf, __pyx_t_20, __pyx_pybuffernd_SS.diminfo[0].strides)) == __pyx_v_VOID);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":744
|
|
* w = indices2[j]
|
|
* if SS[w] == VOID:
|
|
* SS[w] = SS_head # <<<<<<<<<<<<<<
|
|
* SS_head = w
|
|
* label += 1
|
|
*/
|
|
__pyx_t_21 = __pyx_v_w;
|
|
__pyx_t_22 = -1;
|
|
if (__pyx_t_21 < 0) {
|
|
__pyx_t_21 += __pyx_pybuffernd_SS.diminfo[0].shape;
|
|
if (unlikely(__pyx_t_21 < 0)) __pyx_t_22 = 0;
|
|
} else if (unlikely(__pyx_t_21 >= __pyx_pybuffernd_SS.diminfo[0].shape)) __pyx_t_22 = 0;
|
|
if (unlikely(__pyx_t_22 != -1)) {
|
|
__Pyx_RaiseBufferIndexError(__pyx_t_22);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 744; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_5utils_3mst_10_traversal_ITYPE_t *, __pyx_pybuffernd_SS.rcbuffer->pybuffer.buf, __pyx_t_21, __pyx_pybuffernd_SS.diminfo[0].strides) = __pyx_v_SS_head;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":745
|
|
* if SS[w] == VOID:
|
|
* SS[w] = SS_head
|
|
* SS_head = w # <<<<<<<<<<<<<<
|
|
* label += 1
|
|
*
|
|
*/
|
|
__pyx_v_SS_head = __pyx_v_w;
|
|
goto __pyx_L13;
|
|
}
|
|
__pyx_L13:;
|
|
}
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":746
|
|
* SS[w] = SS_head
|
|
* SS_head = w
|
|
* label += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* return label
|
|
*/
|
|
__pyx_v_label = (__pyx_v_label + 1);
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":748
|
|
* label += 1
|
|
*
|
|
* return label # <<<<<<<<<<<<<<
|
|
*/
|
|
__pyx_r = __pyx_v_label;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_SS.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices1.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices2.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr1.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr2.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_WriteUnraisable("sklearn.utils.mst._traversal._connected_components_undirected", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_SS.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices1.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices2.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr1.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr2.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_labels.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_SS);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_5numpy_7ndarray_1__getbuffer__(PyObject *__pyx_v_self, Py_buffer *__pyx_v_info, int __pyx_v_flags); /*proto*/
|
|
static int __pyx_pw_5numpy_7ndarray_1__getbuffer__(PyObject *__pyx_v_self, Py_buffer *__pyx_v_info, int __pyx_v_flags) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__getbuffer__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_5numpy_7ndarray___getbuffer__(((PyArrayObject *)__pyx_v_self), ((Py_buffer *)__pyx_v_info), ((int)__pyx_v_flags));
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "numpy.pxd":193
|
|
* # experimental exception made for __getbuffer__ and __releasebuffer__
|
|
* # -- the details of this may change.
|
|
* def __getbuffer__(ndarray self, Py_buffer* info, int flags): # <<<<<<<<<<<<<<
|
|
* # This implementation of getbuffer is geared towards Cython
|
|
* # requirements, and does not yet fullfill the PEP.
|
|
*/
|
|
|
|
static int __pyx_pf_5numpy_7ndarray___getbuffer__(PyArrayObject *__pyx_v_self, Py_buffer *__pyx_v_info, int __pyx_v_flags) {
|
|
int __pyx_v_copy_shape;
|
|
int __pyx_v_i;
|
|
int __pyx_v_ndim;
|
|
int __pyx_v_endian_detector;
|
|
int __pyx_v_little_endian;
|
|
int __pyx_v_t;
|
|
char *__pyx_v_f;
|
|
PyArray_Descr *__pyx_v_descr = 0;
|
|
int __pyx_v_offset;
|
|
int __pyx_v_hasfields;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
int __pyx_t_2;
|
|
int __pyx_t_3;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_t_5;
|
|
int __pyx_t_6;
|
|
int __pyx_t_7;
|
|
PyObject *__pyx_t_8 = NULL;
|
|
char *__pyx_t_9;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__getbuffer__", 0);
|
|
if (__pyx_v_info != NULL) {
|
|
__pyx_v_info->obj = Py_None; __Pyx_INCREF(Py_None);
|
|
__Pyx_GIVEREF(__pyx_v_info->obj);
|
|
}
|
|
|
|
/* "numpy.pxd":199
|
|
* # of flags
|
|
*
|
|
* if info == NULL: return # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef int copy_shape, i, ndim
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_info == NULL);
|
|
if (__pyx_t_1) {
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
goto __pyx_L3;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "numpy.pxd":202
|
|
*
|
|
* cdef int copy_shape, i, ndim
|
|
* cdef int endian_detector = 1 # <<<<<<<<<<<<<<
|
|
* cdef bint little_endian = ((<char*>&endian_detector)[0] != 0)
|
|
*
|
|
*/
|
|
__pyx_v_endian_detector = 1;
|
|
|
|
/* "numpy.pxd":203
|
|
* cdef int copy_shape, i, ndim
|
|
* cdef int endian_detector = 1
|
|
* cdef bint little_endian = ((<char*>&endian_detector)[0] != 0) # <<<<<<<<<<<<<<
|
|
*
|
|
* ndim = PyArray_NDIM(self)
|
|
*/
|
|
__pyx_v_little_endian = ((((char *)(&__pyx_v_endian_detector))[0]) != 0);
|
|
|
|
/* "numpy.pxd":205
|
|
* cdef bint little_endian = ((<char*>&endian_detector)[0] != 0)
|
|
*
|
|
* ndim = PyArray_NDIM(self) # <<<<<<<<<<<<<<
|
|
*
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t):
|
|
*/
|
|
__pyx_v_ndim = PyArray_NDIM(__pyx_v_self);
|
|
|
|
/* "numpy.pxd":207
|
|
* ndim = PyArray_NDIM(self)
|
|
*
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t): # <<<<<<<<<<<<<<
|
|
* copy_shape = 1
|
|
* else:
|
|
*/
|
|
__pyx_t_1 = ((sizeof(npy_intp)) != (sizeof(Py_ssize_t)));
|
|
if (__pyx_t_1) {
|
|
|
|
/* "numpy.pxd":208
|
|
*
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t):
|
|
* copy_shape = 1 # <<<<<<<<<<<<<<
|
|
* else:
|
|
* copy_shape = 0
|
|
*/
|
|
__pyx_v_copy_shape = 1;
|
|
goto __pyx_L4;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "numpy.pxd":210
|
|
* copy_shape = 1
|
|
* else:
|
|
* copy_shape = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* if ((flags & pybuf.PyBUF_C_CONTIGUOUS == pybuf.PyBUF_C_CONTIGUOUS)
|
|
*/
|
|
__pyx_v_copy_shape = 0;
|
|
}
|
|
__pyx_L4:;
|
|
|
|
/* "numpy.pxd":212
|
|
* copy_shape = 0
|
|
*
|
|
* if ((flags & pybuf.PyBUF_C_CONTIGUOUS == pybuf.PyBUF_C_CONTIGUOUS) # <<<<<<<<<<<<<<
|
|
* and not PyArray_CHKFLAGS(self, NPY_C_CONTIGUOUS)):
|
|
* raise ValueError(u"ndarray is not C contiguous")
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_flags & PyBUF_C_CONTIGUOUS) == PyBUF_C_CONTIGUOUS);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "numpy.pxd":213
|
|
*
|
|
* if ((flags & pybuf.PyBUF_C_CONTIGUOUS == pybuf.PyBUF_C_CONTIGUOUS)
|
|
* and not PyArray_CHKFLAGS(self, NPY_C_CONTIGUOUS)): # <<<<<<<<<<<<<<
|
|
* raise ValueError(u"ndarray is not C contiguous")
|
|
*
|
|
*/
|
|
__pyx_t_2 = (!PyArray_CHKFLAGS(__pyx_v_self, NPY_C_CONTIGUOUS));
|
|
__pyx_t_3 = __pyx_t_2;
|
|
} else {
|
|
__pyx_t_3 = __pyx_t_1;
|
|
}
|
|
if (__pyx_t_3) {
|
|
|
|
/* "numpy.pxd":214
|
|
* if ((flags & pybuf.PyBUF_C_CONTIGUOUS == pybuf.PyBUF_C_CONTIGUOUS)
|
|
* and not PyArray_CHKFLAGS(self, NPY_C_CONTIGUOUS)):
|
|
* raise ValueError(u"ndarray is not C contiguous") # <<<<<<<<<<<<<<
|
|
*
|
|
* if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS)
|
|
*/
|
|
__pyx_t_4 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_k_tuple_12), NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 214; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_Raise(__pyx_t_4, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 214; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
|
|
/* "numpy.pxd":216
|
|
* raise ValueError(u"ndarray is not C contiguous")
|
|
*
|
|
* if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS) # <<<<<<<<<<<<<<
|
|
* and not PyArray_CHKFLAGS(self, NPY_F_CONTIGUOUS)):
|
|
* raise ValueError(u"ndarray is not Fortran contiguous")
|
|
*/
|
|
__pyx_t_3 = ((__pyx_v_flags & PyBUF_F_CONTIGUOUS) == PyBUF_F_CONTIGUOUS);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "numpy.pxd":217
|
|
*
|
|
* if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS)
|
|
* and not PyArray_CHKFLAGS(self, NPY_F_CONTIGUOUS)): # <<<<<<<<<<<<<<
|
|
* raise ValueError(u"ndarray is not Fortran contiguous")
|
|
*
|
|
*/
|
|
__pyx_t_1 = (!PyArray_CHKFLAGS(__pyx_v_self, NPY_F_CONTIGUOUS));
|
|
__pyx_t_2 = __pyx_t_1;
|
|
} else {
|
|
__pyx_t_2 = __pyx_t_3;
|
|
}
|
|
if (__pyx_t_2) {
|
|
|
|
/* "numpy.pxd":218
|
|
* if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS)
|
|
* and not PyArray_CHKFLAGS(self, NPY_F_CONTIGUOUS)):
|
|
* raise ValueError(u"ndarray is not Fortran contiguous") # <<<<<<<<<<<<<<
|
|
*
|
|
* info.buf = PyArray_DATA(self)
|
|
*/
|
|
__pyx_t_4 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_k_tuple_14), NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 218; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_Raise(__pyx_t_4, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 218; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
goto __pyx_L6;
|
|
}
|
|
__pyx_L6:;
|
|
|
|
/* "numpy.pxd":220
|
|
* raise ValueError(u"ndarray is not Fortran contiguous")
|
|
*
|
|
* info.buf = PyArray_DATA(self) # <<<<<<<<<<<<<<
|
|
* info.ndim = ndim
|
|
* if copy_shape:
|
|
*/
|
|
__pyx_v_info->buf = PyArray_DATA(__pyx_v_self);
|
|
|
|
/* "numpy.pxd":221
|
|
*
|
|
* info.buf = PyArray_DATA(self)
|
|
* info.ndim = ndim # <<<<<<<<<<<<<<
|
|
* if copy_shape:
|
|
* # Allocate new buffer for strides and shape info.
|
|
*/
|
|
__pyx_v_info->ndim = __pyx_v_ndim;
|
|
|
|
/* "numpy.pxd":222
|
|
* info.buf = PyArray_DATA(self)
|
|
* info.ndim = ndim
|
|
* if copy_shape: # <<<<<<<<<<<<<<
|
|
* # Allocate new buffer for strides and shape info.
|
|
* # This is allocated as one block, strides first.
|
|
*/
|
|
if (__pyx_v_copy_shape) {
|
|
|
|
/* "numpy.pxd":225
|
|
* # Allocate new buffer for strides and shape info.
|
|
* # This is allocated as one block, strides first.
|
|
* info.strides = <Py_ssize_t*>stdlib.malloc(sizeof(Py_ssize_t) * <size_t>ndim * 2) # <<<<<<<<<<<<<<
|
|
* info.shape = info.strides + ndim
|
|
* for i in range(ndim):
|
|
*/
|
|
__pyx_v_info->strides = ((Py_ssize_t *)malloc((((sizeof(Py_ssize_t)) * ((size_t)__pyx_v_ndim)) * 2)));
|
|
|
|
/* "numpy.pxd":226
|
|
* # This is allocated as one block, strides first.
|
|
* info.strides = <Py_ssize_t*>stdlib.malloc(sizeof(Py_ssize_t) * <size_t>ndim * 2)
|
|
* info.shape = info.strides + ndim # <<<<<<<<<<<<<<
|
|
* for i in range(ndim):
|
|
* info.strides[i] = PyArray_STRIDES(self)[i]
|
|
*/
|
|
__pyx_v_info->shape = (__pyx_v_info->strides + __pyx_v_ndim);
|
|
|
|
/* "numpy.pxd":227
|
|
* info.strides = <Py_ssize_t*>stdlib.malloc(sizeof(Py_ssize_t) * <size_t>ndim * 2)
|
|
* info.shape = info.strides + ndim
|
|
* for i in range(ndim): # <<<<<<<<<<<<<<
|
|
* info.strides[i] = PyArray_STRIDES(self)[i]
|
|
* info.shape[i] = PyArray_DIMS(self)[i]
|
|
*/
|
|
__pyx_t_5 = __pyx_v_ndim;
|
|
for (__pyx_t_6 = 0; __pyx_t_6 < __pyx_t_5; __pyx_t_6+=1) {
|
|
__pyx_v_i = __pyx_t_6;
|
|
|
|
/* "numpy.pxd":228
|
|
* info.shape = info.strides + ndim
|
|
* for i in range(ndim):
|
|
* info.strides[i] = PyArray_STRIDES(self)[i] # <<<<<<<<<<<<<<
|
|
* info.shape[i] = PyArray_DIMS(self)[i]
|
|
* else:
|
|
*/
|
|
(__pyx_v_info->strides[__pyx_v_i]) = (PyArray_STRIDES(__pyx_v_self)[__pyx_v_i]);
|
|
|
|
/* "numpy.pxd":229
|
|
* for i in range(ndim):
|
|
* info.strides[i] = PyArray_STRIDES(self)[i]
|
|
* info.shape[i] = PyArray_DIMS(self)[i] # <<<<<<<<<<<<<<
|
|
* else:
|
|
* info.strides = <Py_ssize_t*>PyArray_STRIDES(self)
|
|
*/
|
|
(__pyx_v_info->shape[__pyx_v_i]) = (PyArray_DIMS(__pyx_v_self)[__pyx_v_i]);
|
|
}
|
|
goto __pyx_L7;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "numpy.pxd":231
|
|
* info.shape[i] = PyArray_DIMS(self)[i]
|
|
* else:
|
|
* info.strides = <Py_ssize_t*>PyArray_STRIDES(self) # <<<<<<<<<<<<<<
|
|
* info.shape = <Py_ssize_t*>PyArray_DIMS(self)
|
|
* info.suboffsets = NULL
|
|
*/
|
|
__pyx_v_info->strides = ((Py_ssize_t *)PyArray_STRIDES(__pyx_v_self));
|
|
|
|
/* "numpy.pxd":232
|
|
* else:
|
|
* info.strides = <Py_ssize_t*>PyArray_STRIDES(self)
|
|
* info.shape = <Py_ssize_t*>PyArray_DIMS(self) # <<<<<<<<<<<<<<
|
|
* info.suboffsets = NULL
|
|
* info.itemsize = PyArray_ITEMSIZE(self)
|
|
*/
|
|
__pyx_v_info->shape = ((Py_ssize_t *)PyArray_DIMS(__pyx_v_self));
|
|
}
|
|
__pyx_L7:;
|
|
|
|
/* "numpy.pxd":233
|
|
* info.strides = <Py_ssize_t*>PyArray_STRIDES(self)
|
|
* info.shape = <Py_ssize_t*>PyArray_DIMS(self)
|
|
* info.suboffsets = NULL # <<<<<<<<<<<<<<
|
|
* info.itemsize = PyArray_ITEMSIZE(self)
|
|
* info.readonly = not PyArray_ISWRITEABLE(self)
|
|
*/
|
|
__pyx_v_info->suboffsets = NULL;
|
|
|
|
/* "numpy.pxd":234
|
|
* info.shape = <Py_ssize_t*>PyArray_DIMS(self)
|
|
* info.suboffsets = NULL
|
|
* info.itemsize = PyArray_ITEMSIZE(self) # <<<<<<<<<<<<<<
|
|
* info.readonly = not PyArray_ISWRITEABLE(self)
|
|
*
|
|
*/
|
|
__pyx_v_info->itemsize = PyArray_ITEMSIZE(__pyx_v_self);
|
|
|
|
/* "numpy.pxd":235
|
|
* info.suboffsets = NULL
|
|
* info.itemsize = PyArray_ITEMSIZE(self)
|
|
* info.readonly = not PyArray_ISWRITEABLE(self) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef int t
|
|
*/
|
|
__pyx_v_info->readonly = (!PyArray_ISWRITEABLE(__pyx_v_self));
|
|
|
|
/* "numpy.pxd":238
|
|
*
|
|
* cdef int t
|
|
* cdef char* f = NULL # <<<<<<<<<<<<<<
|
|
* cdef dtype descr = self.descr
|
|
* cdef list stack
|
|
*/
|
|
__pyx_v_f = NULL;
|
|
|
|
/* "numpy.pxd":239
|
|
* cdef int t
|
|
* cdef char* f = NULL
|
|
* cdef dtype descr = self.descr # <<<<<<<<<<<<<<
|
|
* cdef list stack
|
|
* cdef int offset
|
|
*/
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_self->descr));
|
|
__pyx_v_descr = __pyx_v_self->descr;
|
|
|
|
/* "numpy.pxd":243
|
|
* cdef int offset
|
|
*
|
|
* cdef bint hasfields = PyDataType_HASFIELDS(descr) # <<<<<<<<<<<<<<
|
|
*
|
|
* if not hasfields and not copy_shape:
|
|
*/
|
|
__pyx_v_hasfields = PyDataType_HASFIELDS(__pyx_v_descr);
|
|
|
|
/* "numpy.pxd":245
|
|
* cdef bint hasfields = PyDataType_HASFIELDS(descr)
|
|
*
|
|
* if not hasfields and not copy_shape: # <<<<<<<<<<<<<<
|
|
* # do not call releasebuffer
|
|
* info.obj = None
|
|
*/
|
|
__pyx_t_2 = (!__pyx_v_hasfields);
|
|
if (__pyx_t_2) {
|
|
__pyx_t_3 = (!__pyx_v_copy_shape);
|
|
__pyx_t_1 = __pyx_t_3;
|
|
} else {
|
|
__pyx_t_1 = __pyx_t_2;
|
|
}
|
|
if (__pyx_t_1) {
|
|
|
|
/* "numpy.pxd":247
|
|
* if not hasfields and not copy_shape:
|
|
* # do not call releasebuffer
|
|
* info.obj = None # <<<<<<<<<<<<<<
|
|
* else:
|
|
* # need to call releasebuffer
|
|
*/
|
|
__Pyx_INCREF(Py_None);
|
|
__Pyx_GIVEREF(Py_None);
|
|
__Pyx_GOTREF(__pyx_v_info->obj);
|
|
__Pyx_DECREF(__pyx_v_info->obj);
|
|
__pyx_v_info->obj = Py_None;
|
|
goto __pyx_L10;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "numpy.pxd":250
|
|
* else:
|
|
* # need to call releasebuffer
|
|
* info.obj = self # <<<<<<<<<<<<<<
|
|
*
|
|
* if not hasfields:
|
|
*/
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_self));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_self));
|
|
__Pyx_GOTREF(__pyx_v_info->obj);
|
|
__Pyx_DECREF(__pyx_v_info->obj);
|
|
__pyx_v_info->obj = ((PyObject *)__pyx_v_self);
|
|
}
|
|
__pyx_L10:;
|
|
|
|
/* "numpy.pxd":252
|
|
* info.obj = self
|
|
*
|
|
* if not hasfields: # <<<<<<<<<<<<<<
|
|
* t = descr.type_num
|
|
* if ((descr.byteorder == '>' and little_endian) or
|
|
*/
|
|
__pyx_t_1 = (!__pyx_v_hasfields);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "numpy.pxd":253
|
|
*
|
|
* if not hasfields:
|
|
* t = descr.type_num # <<<<<<<<<<<<<<
|
|
* if ((descr.byteorder == '>' and little_endian) or
|
|
* (descr.byteorder == '<' and not little_endian)):
|
|
*/
|
|
__pyx_v_t = __pyx_v_descr->type_num;
|
|
|
|
/* "numpy.pxd":254
|
|
* if not hasfields:
|
|
* t = descr.type_num
|
|
* if ((descr.byteorder == '>' and little_endian) or # <<<<<<<<<<<<<<
|
|
* (descr.byteorder == '<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported")
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_descr->byteorder == '>');
|
|
if (__pyx_t_1) {
|
|
__pyx_t_2 = __pyx_v_little_endian;
|
|
} else {
|
|
__pyx_t_2 = __pyx_t_1;
|
|
}
|
|
if (!__pyx_t_2) {
|
|
|
|
/* "numpy.pxd":255
|
|
* t = descr.type_num
|
|
* if ((descr.byteorder == '>' and little_endian) or
|
|
* (descr.byteorder == '<' and not little_endian)): # <<<<<<<<<<<<<<
|
|
* raise ValueError(u"Non-native byte order not supported")
|
|
* if t == NPY_BYTE: f = "b"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_descr->byteorder == '<');
|
|
if (__pyx_t_1) {
|
|
__pyx_t_3 = (!__pyx_v_little_endian);
|
|
__pyx_t_7 = __pyx_t_3;
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_1;
|
|
}
|
|
__pyx_t_1 = __pyx_t_7;
|
|
} else {
|
|
__pyx_t_1 = __pyx_t_2;
|
|
}
|
|
if (__pyx_t_1) {
|
|
|
|
/* "numpy.pxd":256
|
|
* if ((descr.byteorder == '>' and little_endian) or
|
|
* (descr.byteorder == '<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported") # <<<<<<<<<<<<<<
|
|
* if t == NPY_BYTE: f = "b"
|
|
* elif t == NPY_UBYTE: f = "B"
|
|
*/
|
|
__pyx_t_4 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_k_tuple_16), NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 256; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_Raise(__pyx_t_4, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 256; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
goto __pyx_L12;
|
|
}
|
|
__pyx_L12:;
|
|
|
|
/* "numpy.pxd":257
|
|
* (descr.byteorder == '<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported")
|
|
* if t == NPY_BYTE: f = "b" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_UBYTE: f = "B"
|
|
* elif t == NPY_SHORT: f = "h"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_BYTE);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__b;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":258
|
|
* raise ValueError(u"Non-native byte order not supported")
|
|
* if t == NPY_BYTE: f = "b"
|
|
* elif t == NPY_UBYTE: f = "B" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_SHORT: f = "h"
|
|
* elif t == NPY_USHORT: f = "H"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_UBYTE);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__B;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":259
|
|
* if t == NPY_BYTE: f = "b"
|
|
* elif t == NPY_UBYTE: f = "B"
|
|
* elif t == NPY_SHORT: f = "h" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_USHORT: f = "H"
|
|
* elif t == NPY_INT: f = "i"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_SHORT);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__h;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":260
|
|
* elif t == NPY_UBYTE: f = "B"
|
|
* elif t == NPY_SHORT: f = "h"
|
|
* elif t == NPY_USHORT: f = "H" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_INT: f = "i"
|
|
* elif t == NPY_UINT: f = "I"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_USHORT);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__H;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":261
|
|
* elif t == NPY_SHORT: f = "h"
|
|
* elif t == NPY_USHORT: f = "H"
|
|
* elif t == NPY_INT: f = "i" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_UINT: f = "I"
|
|
* elif t == NPY_LONG: f = "l"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_INT);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__i;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":262
|
|
* elif t == NPY_USHORT: f = "H"
|
|
* elif t == NPY_INT: f = "i"
|
|
* elif t == NPY_UINT: f = "I" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_LONG: f = "l"
|
|
* elif t == NPY_ULONG: f = "L"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_UINT);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__I;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":263
|
|
* elif t == NPY_INT: f = "i"
|
|
* elif t == NPY_UINT: f = "I"
|
|
* elif t == NPY_LONG: f = "l" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_ULONG: f = "L"
|
|
* elif t == NPY_LONGLONG: f = "q"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_LONG);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__l;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":264
|
|
* elif t == NPY_UINT: f = "I"
|
|
* elif t == NPY_LONG: f = "l"
|
|
* elif t == NPY_ULONG: f = "L" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_LONGLONG: f = "q"
|
|
* elif t == NPY_ULONGLONG: f = "Q"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_ULONG);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__L;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":265
|
|
* elif t == NPY_LONG: f = "l"
|
|
* elif t == NPY_ULONG: f = "L"
|
|
* elif t == NPY_LONGLONG: f = "q" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_ULONGLONG: f = "Q"
|
|
* elif t == NPY_FLOAT: f = "f"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_LONGLONG);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__q;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":266
|
|
* elif t == NPY_ULONG: f = "L"
|
|
* elif t == NPY_LONGLONG: f = "q"
|
|
* elif t == NPY_ULONGLONG: f = "Q" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_FLOAT: f = "f"
|
|
* elif t == NPY_DOUBLE: f = "d"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_ULONGLONG);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__Q;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":267
|
|
* elif t == NPY_LONGLONG: f = "q"
|
|
* elif t == NPY_ULONGLONG: f = "Q"
|
|
* elif t == NPY_FLOAT: f = "f" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_DOUBLE: f = "d"
|
|
* elif t == NPY_LONGDOUBLE: f = "g"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_FLOAT);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__f;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":268
|
|
* elif t == NPY_ULONGLONG: f = "Q"
|
|
* elif t == NPY_FLOAT: f = "f"
|
|
* elif t == NPY_DOUBLE: f = "d" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_LONGDOUBLE: f = "g"
|
|
* elif t == NPY_CFLOAT: f = "Zf"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_DOUBLE);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__d;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":269
|
|
* elif t == NPY_FLOAT: f = "f"
|
|
* elif t == NPY_DOUBLE: f = "d"
|
|
* elif t == NPY_LONGDOUBLE: f = "g" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_CFLOAT: f = "Zf"
|
|
* elif t == NPY_CDOUBLE: f = "Zd"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_LONGDOUBLE);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__g;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":270
|
|
* elif t == NPY_DOUBLE: f = "d"
|
|
* elif t == NPY_LONGDOUBLE: f = "g"
|
|
* elif t == NPY_CFLOAT: f = "Zf" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_CDOUBLE: f = "Zd"
|
|
* elif t == NPY_CLONGDOUBLE: f = "Zg"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_CFLOAT);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__Zf;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":271
|
|
* elif t == NPY_LONGDOUBLE: f = "g"
|
|
* elif t == NPY_CFLOAT: f = "Zf"
|
|
* elif t == NPY_CDOUBLE: f = "Zd" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_CLONGDOUBLE: f = "Zg"
|
|
* elif t == NPY_OBJECT: f = "O"
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_CDOUBLE);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__Zd;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":272
|
|
* elif t == NPY_CFLOAT: f = "Zf"
|
|
* elif t == NPY_CDOUBLE: f = "Zd"
|
|
* elif t == NPY_CLONGDOUBLE: f = "Zg" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_OBJECT: f = "O"
|
|
* else:
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_CLONGDOUBLE);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__Zg;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "numpy.pxd":273
|
|
* elif t == NPY_CDOUBLE: f = "Zd"
|
|
* elif t == NPY_CLONGDOUBLE: f = "Zg"
|
|
* elif t == NPY_OBJECT: f = "O" # <<<<<<<<<<<<<<
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_t == NPY_OBJECT);
|
|
if (__pyx_t_1) {
|
|
__pyx_v_f = __pyx_k__O;
|
|
goto __pyx_L13;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "numpy.pxd":275
|
|
* elif t == NPY_OBJECT: f = "O"
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t) # <<<<<<<<<<<<<<
|
|
* info.format = f
|
|
* return
|
|
*/
|
|
__pyx_t_4 = PyInt_FromLong(__pyx_v_t); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 275; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_8 = PyNumber_Remainder(((PyObject *)__pyx_kp_u_17), __pyx_t_4); if (unlikely(!__pyx_t_8)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 275; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_8));
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 275; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, ((PyObject *)__pyx_t_8));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_t_8));
|
|
__pyx_t_8 = 0;
|
|
__pyx_t_8 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_t_4), NULL); if (unlikely(!__pyx_t_8)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 275; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_8);
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_4)); __pyx_t_4 = 0;
|
|
__Pyx_Raise(__pyx_t_8, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
|
|
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 275; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_L13:;
|
|
|
|
/* "numpy.pxd":276
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
|
|
* info.format = f # <<<<<<<<<<<<<<
|
|
* return
|
|
* else:
|
|
*/
|
|
__pyx_v_info->format = __pyx_v_f;
|
|
|
|
/* "numpy.pxd":277
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
|
|
* info.format = f
|
|
* return # <<<<<<<<<<<<<<
|
|
* else:
|
|
* info.format = <char*>stdlib.malloc(_buffer_format_string_len)
|
|
*/
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
goto __pyx_L11;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "numpy.pxd":279
|
|
* return
|
|
* else:
|
|
* info.format = <char*>stdlib.malloc(_buffer_format_string_len) # <<<<<<<<<<<<<<
|
|
* info.format[0] = '^' # Native data types, manual alignment
|
|
* offset = 0
|
|
*/
|
|
__pyx_v_info->format = ((char *)malloc(255));
|
|
|
|
/* "numpy.pxd":280
|
|
* else:
|
|
* info.format = <char*>stdlib.malloc(_buffer_format_string_len)
|
|
* info.format[0] = '^' # Native data types, manual alignment # <<<<<<<<<<<<<<
|
|
* offset = 0
|
|
* f = _util_dtypestring(descr, info.format + 1,
|
|
*/
|
|
(__pyx_v_info->format[0]) = '^';
|
|
|
|
/* "numpy.pxd":281
|
|
* info.format = <char*>stdlib.malloc(_buffer_format_string_len)
|
|
* info.format[0] = '^' # Native data types, manual alignment
|
|
* offset = 0 # <<<<<<<<<<<<<<
|
|
* f = _util_dtypestring(descr, info.format + 1,
|
|
* info.format + _buffer_format_string_len,
|
|
*/
|
|
__pyx_v_offset = 0;
|
|
|
|
/* "numpy.pxd":284
|
|
* f = _util_dtypestring(descr, info.format + 1,
|
|
* info.format + _buffer_format_string_len,
|
|
* &offset) # <<<<<<<<<<<<<<
|
|
* f[0] = 0 # Terminate format string
|
|
*
|
|
*/
|
|
__pyx_t_9 = __pyx_f_5numpy__util_dtypestring(__pyx_v_descr, (__pyx_v_info->format + 1), (__pyx_v_info->format + 255), (&__pyx_v_offset)); if (unlikely(__pyx_t_9 == NULL)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 282; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_f = __pyx_t_9;
|
|
|
|
/* "numpy.pxd":285
|
|
* info.format + _buffer_format_string_len,
|
|
* &offset)
|
|
* f[0] = 0 # Terminate format string # <<<<<<<<<<<<<<
|
|
*
|
|
* def __releasebuffer__(ndarray self, Py_buffer* info):
|
|
*/
|
|
(__pyx_v_f[0]) = 0;
|
|
}
|
|
__pyx_L11:;
|
|
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_XDECREF(__pyx_t_8);
|
|
__Pyx_AddTraceback("numpy.ndarray.__getbuffer__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
if (__pyx_v_info != NULL && __pyx_v_info->obj != NULL) {
|
|
__Pyx_GOTREF(__pyx_v_info->obj);
|
|
__Pyx_DECREF(__pyx_v_info->obj); __pyx_v_info->obj = NULL;
|
|
}
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
if (__pyx_v_info != NULL && __pyx_v_info->obj == Py_None) {
|
|
__Pyx_GOTREF(Py_None);
|
|
__Pyx_DECREF(Py_None); __pyx_v_info->obj = NULL;
|
|
}
|
|
__pyx_L2:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_descr);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static void __pyx_pw_5numpy_7ndarray_3__releasebuffer__(PyObject *__pyx_v_self, Py_buffer *__pyx_v_info); /*proto*/
|
|
static void __pyx_pw_5numpy_7ndarray_3__releasebuffer__(PyObject *__pyx_v_self, Py_buffer *__pyx_v_info) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__releasebuffer__ (wrapper)", 0);
|
|
__pyx_pf_5numpy_7ndarray_2__releasebuffer__(((PyArrayObject *)__pyx_v_self), ((Py_buffer *)__pyx_v_info));
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "numpy.pxd":287
|
|
* f[0] = 0 # Terminate format string
|
|
*
|
|
* def __releasebuffer__(ndarray self, Py_buffer* info): # <<<<<<<<<<<<<<
|
|
* if PyArray_HASFIELDS(self):
|
|
* stdlib.free(info.format)
|
|
*/
|
|
|
|
static void __pyx_pf_5numpy_7ndarray_2__releasebuffer__(PyArrayObject *__pyx_v_self, Py_buffer *__pyx_v_info) {
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
__Pyx_RefNannySetupContext("__releasebuffer__", 0);
|
|
|
|
/* "numpy.pxd":288
|
|
*
|
|
* def __releasebuffer__(ndarray self, Py_buffer* info):
|
|
* if PyArray_HASFIELDS(self): # <<<<<<<<<<<<<<
|
|
* stdlib.free(info.format)
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t):
|
|
*/
|
|
__pyx_t_1 = PyArray_HASFIELDS(__pyx_v_self);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "numpy.pxd":289
|
|
* def __releasebuffer__(ndarray self, Py_buffer* info):
|
|
* if PyArray_HASFIELDS(self):
|
|
* stdlib.free(info.format) # <<<<<<<<<<<<<<
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t):
|
|
* stdlib.free(info.strides)
|
|
*/
|
|
free(__pyx_v_info->format);
|
|
goto __pyx_L3;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "numpy.pxd":290
|
|
* if PyArray_HASFIELDS(self):
|
|
* stdlib.free(info.format)
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t): # <<<<<<<<<<<<<<
|
|
* stdlib.free(info.strides)
|
|
* # info.shape was stored after info.strides in the same block
|
|
*/
|
|
__pyx_t_1 = ((sizeof(npy_intp)) != (sizeof(Py_ssize_t)));
|
|
if (__pyx_t_1) {
|
|
|
|
/* "numpy.pxd":291
|
|
* stdlib.free(info.format)
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t):
|
|
* stdlib.free(info.strides) # <<<<<<<<<<<<<<
|
|
* # info.shape was stored after info.strides in the same block
|
|
*
|
|
*/
|
|
free(__pyx_v_info->strides);
|
|
goto __pyx_L4;
|
|
}
|
|
__pyx_L4:;
|
|
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "numpy.pxd":767
|
|
* ctypedef npy_cdouble complex_t
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew1(a): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(1, <void*>a)
|
|
*
|
|
*/
|
|
|
|
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew1(PyObject *__pyx_v_a) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("PyArray_MultiIterNew1", 0);
|
|
|
|
/* "numpy.pxd":768
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew1(a):
|
|
* return PyArray_MultiIterNew(1, <void*>a) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew2(a, b):
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = PyArray_MultiIterNew(1, ((void *)__pyx_v_a)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 768; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("numpy.PyArray_MultiIterNew1", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "numpy.pxd":770
|
|
* return PyArray_MultiIterNew(1, <void*>a)
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew2(a, b): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(2, <void*>a, <void*>b)
|
|
*
|
|
*/
|
|
|
|
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew2(PyObject *__pyx_v_a, PyObject *__pyx_v_b) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("PyArray_MultiIterNew2", 0);
|
|
|
|
/* "numpy.pxd":771
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew2(a, b):
|
|
* return PyArray_MultiIterNew(2, <void*>a, <void*>b) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew3(a, b, c):
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = PyArray_MultiIterNew(2, ((void *)__pyx_v_a), ((void *)__pyx_v_b)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 771; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("numpy.PyArray_MultiIterNew2", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "numpy.pxd":773
|
|
* return PyArray_MultiIterNew(2, <void*>a, <void*>b)
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew3(a, b, c): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(3, <void*>a, <void*>b, <void*> c)
|
|
*
|
|
*/
|
|
|
|
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew3(PyObject *__pyx_v_a, PyObject *__pyx_v_b, PyObject *__pyx_v_c) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("PyArray_MultiIterNew3", 0);
|
|
|
|
/* "numpy.pxd":774
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew3(a, b, c):
|
|
* return PyArray_MultiIterNew(3, <void*>a, <void*>b, <void*> c) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew4(a, b, c, d):
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = PyArray_MultiIterNew(3, ((void *)__pyx_v_a), ((void *)__pyx_v_b), ((void *)__pyx_v_c)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 774; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("numpy.PyArray_MultiIterNew3", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "numpy.pxd":776
|
|
* return PyArray_MultiIterNew(3, <void*>a, <void*>b, <void*> c)
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew4(a, b, c, d): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(4, <void*>a, <void*>b, <void*>c, <void*> d)
|
|
*
|
|
*/
|
|
|
|
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew4(PyObject *__pyx_v_a, PyObject *__pyx_v_b, PyObject *__pyx_v_c, PyObject *__pyx_v_d) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("PyArray_MultiIterNew4", 0);
|
|
|
|
/* "numpy.pxd":777
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew4(a, b, c, d):
|
|
* return PyArray_MultiIterNew(4, <void*>a, <void*>b, <void*>c, <void*> d) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew5(a, b, c, d, e):
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = PyArray_MultiIterNew(4, ((void *)__pyx_v_a), ((void *)__pyx_v_b), ((void *)__pyx_v_c), ((void *)__pyx_v_d)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 777; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("numpy.PyArray_MultiIterNew4", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "numpy.pxd":779
|
|
* return PyArray_MultiIterNew(4, <void*>a, <void*>b, <void*>c, <void*> d)
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew5(a, b, c, d, e): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(5, <void*>a, <void*>b, <void*>c, <void*> d, <void*> e)
|
|
*
|
|
*/
|
|
|
|
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew5(PyObject *__pyx_v_a, PyObject *__pyx_v_b, PyObject *__pyx_v_c, PyObject *__pyx_v_d, PyObject *__pyx_v_e) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("PyArray_MultiIterNew5", 0);
|
|
|
|
/* "numpy.pxd":780
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew5(a, b, c, d, e):
|
|
* return PyArray_MultiIterNew(5, <void*>a, <void*>b, <void*>c, <void*> d, <void*> e) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline char* _util_dtypestring(dtype descr, char* f, char* end, int* offset) except NULL:
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = PyArray_MultiIterNew(5, ((void *)__pyx_v_a), ((void *)__pyx_v_b), ((void *)__pyx_v_c), ((void *)__pyx_v_d), ((void *)__pyx_v_e)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 780; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("numpy.PyArray_MultiIterNew5", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "numpy.pxd":782
|
|
* return PyArray_MultiIterNew(5, <void*>a, <void*>b, <void*>c, <void*> d, <void*> e)
|
|
*
|
|
* cdef inline char* _util_dtypestring(dtype descr, char* f, char* end, int* offset) except NULL: # <<<<<<<<<<<<<<
|
|
* # Recursive utility function used in __getbuffer__ to get format
|
|
* # string. The new location in the format string is returned.
|
|
*/
|
|
|
|
static CYTHON_INLINE char *__pyx_f_5numpy__util_dtypestring(PyArray_Descr *__pyx_v_descr, char *__pyx_v_f, char *__pyx_v_end, int *__pyx_v_offset) {
|
|
PyArray_Descr *__pyx_v_child = 0;
|
|
int __pyx_v_endian_detector;
|
|
int __pyx_v_little_endian;
|
|
PyObject *__pyx_v_fields = 0;
|
|
PyObject *__pyx_v_childname = NULL;
|
|
PyObject *__pyx_v_new_offset = NULL;
|
|
PyObject *__pyx_v_t = NULL;
|
|
char *__pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
Py_ssize_t __pyx_t_2;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
PyObject *__pyx_t_5 = NULL;
|
|
int __pyx_t_6;
|
|
int __pyx_t_7;
|
|
int __pyx_t_8;
|
|
int __pyx_t_9;
|
|
long __pyx_t_10;
|
|
char *__pyx_t_11;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_util_dtypestring", 0);
|
|
|
|
/* "numpy.pxd":789
|
|
* cdef int delta_offset
|
|
* cdef tuple i
|
|
* cdef int endian_detector = 1 # <<<<<<<<<<<<<<
|
|
* cdef bint little_endian = ((<char*>&endian_detector)[0] != 0)
|
|
* cdef tuple fields
|
|
*/
|
|
__pyx_v_endian_detector = 1;
|
|
|
|
/* "numpy.pxd":790
|
|
* cdef tuple i
|
|
* cdef int endian_detector = 1
|
|
* cdef bint little_endian = ((<char*>&endian_detector)[0] != 0) # <<<<<<<<<<<<<<
|
|
* cdef tuple fields
|
|
*
|
|
*/
|
|
__pyx_v_little_endian = ((((char *)(&__pyx_v_endian_detector))[0]) != 0);
|
|
|
|
/* "numpy.pxd":793
|
|
* cdef tuple fields
|
|
*
|
|
* for childname in descr.names: # <<<<<<<<<<<<<<
|
|
* fields = descr.fields[childname]
|
|
* child, new_offset = fields
|
|
*/
|
|
if (unlikely(((PyObject *)__pyx_v_descr->names) == Py_None)) {
|
|
PyErr_SetString(PyExc_TypeError, "'NoneType' object is not iterable"); {__pyx_filename = __pyx_f[1]; __pyx_lineno = 793; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_1 = ((PyObject *)__pyx_v_descr->names); __Pyx_INCREF(__pyx_t_1); __pyx_t_2 = 0;
|
|
for (;;) {
|
|
if (__pyx_t_2 >= PyTuple_GET_SIZE(__pyx_t_1)) break;
|
|
__pyx_t_3 = PyTuple_GET_ITEM(__pyx_t_1, __pyx_t_2); __Pyx_INCREF(__pyx_t_3); __pyx_t_2++;
|
|
__Pyx_XDECREF(__pyx_v_childname);
|
|
__pyx_v_childname = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
|
|
/* "numpy.pxd":794
|
|
*
|
|
* for childname in descr.names:
|
|
* fields = descr.fields[childname] # <<<<<<<<<<<<<<
|
|
* child, new_offset = fields
|
|
*
|
|
*/
|
|
__pyx_t_3 = PyObject_GetItem(__pyx_v_descr->fields, __pyx_v_childname); if (!__pyx_t_3) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 794; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
if (!(likely(PyTuple_CheckExact(__pyx_t_3))||((__pyx_t_3) == Py_None)||(PyErr_Format(PyExc_TypeError, "Expected tuple, got %.200s", Py_TYPE(__pyx_t_3)->tp_name), 0))) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 794; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_XDECREF(((PyObject *)__pyx_v_fields));
|
|
__pyx_v_fields = ((PyObject*)__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
|
|
/* "numpy.pxd":795
|
|
* for childname in descr.names:
|
|
* fields = descr.fields[childname]
|
|
* child, new_offset = fields # <<<<<<<<<<<<<<
|
|
*
|
|
* if (end - f) - (new_offset - offset[0]) < 15:
|
|
*/
|
|
if (likely(PyTuple_CheckExact(((PyObject *)__pyx_v_fields)))) {
|
|
PyObject* sequence = ((PyObject *)__pyx_v_fields);
|
|
if (unlikely(PyTuple_GET_SIZE(sequence) != 2)) {
|
|
if (PyTuple_GET_SIZE(sequence) > 2) __Pyx_RaiseTooManyValuesError(2);
|
|
else __Pyx_RaiseNeedMoreValuesError(PyTuple_GET_SIZE(sequence));
|
|
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 795; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_3 = PyTuple_GET_ITEM(sequence, 0);
|
|
__pyx_t_4 = PyTuple_GET_ITEM(sequence, 1);
|
|
__Pyx_INCREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_t_4);
|
|
} else {
|
|
__Pyx_UnpackTupleError(((PyObject *)__pyx_v_fields), 2);
|
|
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 795; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
if (!(likely(((__pyx_t_3) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_3, __pyx_ptype_5numpy_dtype))))) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 795; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_XDECREF(((PyObject *)__pyx_v_child));
|
|
__pyx_v_child = ((PyArray_Descr *)__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__Pyx_XDECREF(__pyx_v_new_offset);
|
|
__pyx_v_new_offset = __pyx_t_4;
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "numpy.pxd":797
|
|
* child, new_offset = fields
|
|
*
|
|
* if (end - f) - (new_offset - offset[0]) < 15: # <<<<<<<<<<<<<<
|
|
* raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd")
|
|
*
|
|
*/
|
|
__pyx_t_4 = PyInt_FromLong((__pyx_v_end - __pyx_v_f)); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 797; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = PyInt_FromLong((__pyx_v_offset[0])); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 797; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = PyNumber_Subtract(__pyx_v_new_offset, __pyx_t_3); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 797; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_3 = PyNumber_Subtract(__pyx_t_4, __pyx_t_5); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 797; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_t_3, __pyx_int_15, Py_LT); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 797; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 797; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (__pyx_t_6) {
|
|
|
|
/* "numpy.pxd":798
|
|
*
|
|
* if (end - f) - (new_offset - offset[0]) < 15:
|
|
* raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd") # <<<<<<<<<<<<<<
|
|
*
|
|
* if ((child.byteorder == '>' and little_endian) or
|
|
*/
|
|
__pyx_t_5 = PyObject_Call(__pyx_builtin_RuntimeError, ((PyObject *)__pyx_k_tuple_19), NULL); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 798; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_Raise(__pyx_t_5, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 798; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
|
|
/* "numpy.pxd":800
|
|
* raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd")
|
|
*
|
|
* if ((child.byteorder == '>' and little_endian) or # <<<<<<<<<<<<<<
|
|
* (child.byteorder == '<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported")
|
|
*/
|
|
__pyx_t_6 = (__pyx_v_child->byteorder == '>');
|
|
if (__pyx_t_6) {
|
|
__pyx_t_7 = __pyx_v_little_endian;
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_6;
|
|
}
|
|
if (!__pyx_t_7) {
|
|
|
|
/* "numpy.pxd":801
|
|
*
|
|
* if ((child.byteorder == '>' and little_endian) or
|
|
* (child.byteorder == '<' and not little_endian)): # <<<<<<<<<<<<<<
|
|
* raise ValueError(u"Non-native byte order not supported")
|
|
* # One could encode it in the format string and have Cython
|
|
*/
|
|
__pyx_t_6 = (__pyx_v_child->byteorder == '<');
|
|
if (__pyx_t_6) {
|
|
__pyx_t_8 = (!__pyx_v_little_endian);
|
|
__pyx_t_9 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_9 = __pyx_t_6;
|
|
}
|
|
__pyx_t_6 = __pyx_t_9;
|
|
} else {
|
|
__pyx_t_6 = __pyx_t_7;
|
|
}
|
|
if (__pyx_t_6) {
|
|
|
|
/* "numpy.pxd":802
|
|
* if ((child.byteorder == '>' and little_endian) or
|
|
* (child.byteorder == '<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported") # <<<<<<<<<<<<<<
|
|
* # One could encode it in the format string and have Cython
|
|
* # complain instead, BUT: < and > in format strings also imply
|
|
*/
|
|
__pyx_t_5 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_k_tuple_20), NULL); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 802; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_Raise(__pyx_t_5, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 802; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
goto __pyx_L6;
|
|
}
|
|
__pyx_L6:;
|
|
|
|
/* "numpy.pxd":812
|
|
*
|
|
* # Output padding bytes
|
|
* while offset[0] < new_offset: # <<<<<<<<<<<<<<
|
|
* f[0] = 120 # "x"; pad byte
|
|
* f += 1
|
|
*/
|
|
while (1) {
|
|
__pyx_t_5 = PyInt_FromLong((__pyx_v_offset[0])); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 812; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_t_5, __pyx_v_new_offset, Py_LT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 812; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 812; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (!__pyx_t_6) break;
|
|
|
|
/* "numpy.pxd":813
|
|
* # Output padding bytes
|
|
* while offset[0] < new_offset:
|
|
* f[0] = 120 # "x"; pad byte # <<<<<<<<<<<<<<
|
|
* f += 1
|
|
* offset[0] += 1
|
|
*/
|
|
(__pyx_v_f[0]) = 120;
|
|
|
|
/* "numpy.pxd":814
|
|
* while offset[0] < new_offset:
|
|
* f[0] = 120 # "x"; pad byte
|
|
* f += 1 # <<<<<<<<<<<<<<
|
|
* offset[0] += 1
|
|
*
|
|
*/
|
|
__pyx_v_f = (__pyx_v_f + 1);
|
|
|
|
/* "numpy.pxd":815
|
|
* f[0] = 120 # "x"; pad byte
|
|
* f += 1
|
|
* offset[0] += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* offset[0] += child.itemsize
|
|
*/
|
|
__pyx_t_10 = 0;
|
|
(__pyx_v_offset[__pyx_t_10]) = ((__pyx_v_offset[__pyx_t_10]) + 1);
|
|
}
|
|
|
|
/* "numpy.pxd":817
|
|
* offset[0] += 1
|
|
*
|
|
* offset[0] += child.itemsize # <<<<<<<<<<<<<<
|
|
*
|
|
* if not PyDataType_HASFIELDS(child):
|
|
*/
|
|
__pyx_t_10 = 0;
|
|
(__pyx_v_offset[__pyx_t_10]) = ((__pyx_v_offset[__pyx_t_10]) + __pyx_v_child->elsize);
|
|
|
|
/* "numpy.pxd":819
|
|
* offset[0] += child.itemsize
|
|
*
|
|
* if not PyDataType_HASFIELDS(child): # <<<<<<<<<<<<<<
|
|
* t = child.type_num
|
|
* if end - f < 5:
|
|
*/
|
|
__pyx_t_6 = (!PyDataType_HASFIELDS(__pyx_v_child));
|
|
if (__pyx_t_6) {
|
|
|
|
/* "numpy.pxd":820
|
|
*
|
|
* if not PyDataType_HASFIELDS(child):
|
|
* t = child.type_num # <<<<<<<<<<<<<<
|
|
* if end - f < 5:
|
|
* raise RuntimeError(u"Format string allocated too short.")
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(__pyx_v_child->type_num); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 820; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_v_t);
|
|
__pyx_v_t = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
|
|
/* "numpy.pxd":821
|
|
* if not PyDataType_HASFIELDS(child):
|
|
* t = child.type_num
|
|
* if end - f < 5: # <<<<<<<<<<<<<<
|
|
* raise RuntimeError(u"Format string allocated too short.")
|
|
*
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_end - __pyx_v_f) < 5);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "numpy.pxd":822
|
|
* t = child.type_num
|
|
* if end - f < 5:
|
|
* raise RuntimeError(u"Format string allocated too short.") # <<<<<<<<<<<<<<
|
|
*
|
|
* # Until ticket #99 is fixed, use integers to avoid warnings
|
|
*/
|
|
__pyx_t_3 = PyObject_Call(__pyx_builtin_RuntimeError, ((PyObject *)__pyx_k_tuple_22), NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 822; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_Raise(__pyx_t_3, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 822; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
goto __pyx_L10;
|
|
}
|
|
__pyx_L10:;
|
|
|
|
/* "numpy.pxd":825
|
|
*
|
|
* # Until ticket #99 is fixed, use integers to avoid warnings
|
|
* if t == NPY_BYTE: f[0] = 98 #"b" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_UBYTE: f[0] = 66 #"B"
|
|
* elif t == NPY_SHORT: f[0] = 104 #"h"
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_BYTE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 825; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 825; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 825; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 98;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":826
|
|
* # Until ticket #99 is fixed, use integers to avoid warnings
|
|
* if t == NPY_BYTE: f[0] = 98 #"b"
|
|
* elif t == NPY_UBYTE: f[0] = 66 #"B" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_SHORT: f[0] = 104 #"h"
|
|
* elif t == NPY_USHORT: f[0] = 72 #"H"
|
|
*/
|
|
__pyx_t_5 = PyInt_FromLong(NPY_UBYTE); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 826; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 826; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 826; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 66;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":827
|
|
* if t == NPY_BYTE: f[0] = 98 #"b"
|
|
* elif t == NPY_UBYTE: f[0] = 66 #"B"
|
|
* elif t == NPY_SHORT: f[0] = 104 #"h" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_USHORT: f[0] = 72 #"H"
|
|
* elif t == NPY_INT: f[0] = 105 #"i"
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_SHORT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 827; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 827; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 827; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 104;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":828
|
|
* elif t == NPY_UBYTE: f[0] = 66 #"B"
|
|
* elif t == NPY_SHORT: f[0] = 104 #"h"
|
|
* elif t == NPY_USHORT: f[0] = 72 #"H" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_INT: f[0] = 105 #"i"
|
|
* elif t == NPY_UINT: f[0] = 73 #"I"
|
|
*/
|
|
__pyx_t_5 = PyInt_FromLong(NPY_USHORT); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 828; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 828; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 828; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 72;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":829
|
|
* elif t == NPY_SHORT: f[0] = 104 #"h"
|
|
* elif t == NPY_USHORT: f[0] = 72 #"H"
|
|
* elif t == NPY_INT: f[0] = 105 #"i" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_UINT: f[0] = 73 #"I"
|
|
* elif t == NPY_LONG: f[0] = 108 #"l"
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_INT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 829; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 829; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 829; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 105;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":830
|
|
* elif t == NPY_USHORT: f[0] = 72 #"H"
|
|
* elif t == NPY_INT: f[0] = 105 #"i"
|
|
* elif t == NPY_UINT: f[0] = 73 #"I" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_LONG: f[0] = 108 #"l"
|
|
* elif t == NPY_ULONG: f[0] = 76 #"L"
|
|
*/
|
|
__pyx_t_5 = PyInt_FromLong(NPY_UINT); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 830; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 830; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 830; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 73;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":831
|
|
* elif t == NPY_INT: f[0] = 105 #"i"
|
|
* elif t == NPY_UINT: f[0] = 73 #"I"
|
|
* elif t == NPY_LONG: f[0] = 108 #"l" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_ULONG: f[0] = 76 #"L"
|
|
* elif t == NPY_LONGLONG: f[0] = 113 #"q"
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_LONG); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 831; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 831; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 831; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 108;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":832
|
|
* elif t == NPY_UINT: f[0] = 73 #"I"
|
|
* elif t == NPY_LONG: f[0] = 108 #"l"
|
|
* elif t == NPY_ULONG: f[0] = 76 #"L" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_LONGLONG: f[0] = 113 #"q"
|
|
* elif t == NPY_ULONGLONG: f[0] = 81 #"Q"
|
|
*/
|
|
__pyx_t_5 = PyInt_FromLong(NPY_ULONG); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 76;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":833
|
|
* elif t == NPY_LONG: f[0] = 108 #"l"
|
|
* elif t == NPY_ULONG: f[0] = 76 #"L"
|
|
* elif t == NPY_LONGLONG: f[0] = 113 #"q" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_ULONGLONG: f[0] = 81 #"Q"
|
|
* elif t == NPY_FLOAT: f[0] = 102 #"f"
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_LONGLONG); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 833; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 833; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 833; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 113;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":834
|
|
* elif t == NPY_ULONG: f[0] = 76 #"L"
|
|
* elif t == NPY_LONGLONG: f[0] = 113 #"q"
|
|
* elif t == NPY_ULONGLONG: f[0] = 81 #"Q" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_FLOAT: f[0] = 102 #"f"
|
|
* elif t == NPY_DOUBLE: f[0] = 100 #"d"
|
|
*/
|
|
__pyx_t_5 = PyInt_FromLong(NPY_ULONGLONG); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 834; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 834; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 834; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 81;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":835
|
|
* elif t == NPY_LONGLONG: f[0] = 113 #"q"
|
|
* elif t == NPY_ULONGLONG: f[0] = 81 #"Q"
|
|
* elif t == NPY_FLOAT: f[0] = 102 #"f" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_DOUBLE: f[0] = 100 #"d"
|
|
* elif t == NPY_LONGDOUBLE: f[0] = 103 #"g"
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_FLOAT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 835; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 835; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 835; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 102;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":836
|
|
* elif t == NPY_ULONGLONG: f[0] = 81 #"Q"
|
|
* elif t == NPY_FLOAT: f[0] = 102 #"f"
|
|
* elif t == NPY_DOUBLE: f[0] = 100 #"d" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_LONGDOUBLE: f[0] = 103 #"g"
|
|
* elif t == NPY_CFLOAT: f[0] = 90; f[1] = 102; f += 1 # Zf
|
|
*/
|
|
__pyx_t_5 = PyInt_FromLong(NPY_DOUBLE); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 836; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 836; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 836; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 100;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":837
|
|
* elif t == NPY_FLOAT: f[0] = 102 #"f"
|
|
* elif t == NPY_DOUBLE: f[0] = 100 #"d"
|
|
* elif t == NPY_LONGDOUBLE: f[0] = 103 #"g" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_CFLOAT: f[0] = 90; f[1] = 102; f += 1 # Zf
|
|
* elif t == NPY_CDOUBLE: f[0] = 90; f[1] = 100; f += 1 # Zd
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_LONGDOUBLE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 837; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 837; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 837; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 103;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":838
|
|
* elif t == NPY_DOUBLE: f[0] = 100 #"d"
|
|
* elif t == NPY_LONGDOUBLE: f[0] = 103 #"g"
|
|
* elif t == NPY_CFLOAT: f[0] = 90; f[1] = 102; f += 1 # Zf # <<<<<<<<<<<<<<
|
|
* elif t == NPY_CDOUBLE: f[0] = 90; f[1] = 100; f += 1 # Zd
|
|
* elif t == NPY_CLONGDOUBLE: f[0] = 90; f[1] = 103; f += 1 # Zg
|
|
*/
|
|
__pyx_t_5 = PyInt_FromLong(NPY_CFLOAT); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 838; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 838; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 838; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 90;
|
|
(__pyx_v_f[1]) = 102;
|
|
__pyx_v_f = (__pyx_v_f + 1);
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":839
|
|
* elif t == NPY_LONGDOUBLE: f[0] = 103 #"g"
|
|
* elif t == NPY_CFLOAT: f[0] = 90; f[1] = 102; f += 1 # Zf
|
|
* elif t == NPY_CDOUBLE: f[0] = 90; f[1] = 100; f += 1 # Zd # <<<<<<<<<<<<<<
|
|
* elif t == NPY_CLONGDOUBLE: f[0] = 90; f[1] = 103; f += 1 # Zg
|
|
* elif t == NPY_OBJECT: f[0] = 79 #"O"
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_CDOUBLE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 839; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 839; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 839; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 90;
|
|
(__pyx_v_f[1]) = 100;
|
|
__pyx_v_f = (__pyx_v_f + 1);
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":840
|
|
* elif t == NPY_CFLOAT: f[0] = 90; f[1] = 102; f += 1 # Zf
|
|
* elif t == NPY_CDOUBLE: f[0] = 90; f[1] = 100; f += 1 # Zd
|
|
* elif t == NPY_CLONGDOUBLE: f[0] = 90; f[1] = 103; f += 1 # Zg # <<<<<<<<<<<<<<
|
|
* elif t == NPY_OBJECT: f[0] = 79 #"O"
|
|
* else:
|
|
*/
|
|
__pyx_t_5 = PyInt_FromLong(NPY_CLONGDOUBLE); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 90;
|
|
(__pyx_v_f[1]) = 103;
|
|
__pyx_v_f = (__pyx_v_f + 1);
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "numpy.pxd":841
|
|
* elif t == NPY_CDOUBLE: f[0] = 90; f[1] = 100; f += 1 # Zd
|
|
* elif t == NPY_CLONGDOUBLE: f[0] = 90; f[1] = 103; f += 1 # Zg
|
|
* elif t == NPY_OBJECT: f[0] = 79 #"O" # <<<<<<<<<<<<<<
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_OBJECT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 841; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 841; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 841; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 79;
|
|
goto __pyx_L11;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "numpy.pxd":843
|
|
* elif t == NPY_OBJECT: f[0] = 79 #"O"
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t) # <<<<<<<<<<<<<<
|
|
* f += 1
|
|
* else:
|
|
*/
|
|
__pyx_t_5 = PyNumber_Remainder(((PyObject *)__pyx_kp_u_17), __pyx_v_t); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 843; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_5));
|
|
__pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 843; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, ((PyObject *)__pyx_t_5));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_t_5));
|
|
__pyx_t_5 = 0;
|
|
__pyx_t_5 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_t_3), NULL); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 843; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 0;
|
|
__Pyx_Raise(__pyx_t_5, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 843; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_L11:;
|
|
|
|
/* "numpy.pxd":844
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
|
|
* f += 1 # <<<<<<<<<<<<<<
|
|
* else:
|
|
* # Cython ignores struct boundary information ("T{...}"),
|
|
*/
|
|
__pyx_v_f = (__pyx_v_f + 1);
|
|
goto __pyx_L9;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "numpy.pxd":848
|
|
* # Cython ignores struct boundary information ("T{...}"),
|
|
* # so don't output it
|
|
* f = _util_dtypestring(child, f, end, offset) # <<<<<<<<<<<<<<
|
|
* return f
|
|
*
|
|
*/
|
|
__pyx_t_11 = __pyx_f_5numpy__util_dtypestring(__pyx_v_child, __pyx_v_f, __pyx_v_end, __pyx_v_offset); if (unlikely(__pyx_t_11 == NULL)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 848; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_f = __pyx_t_11;
|
|
}
|
|
__pyx_L9:;
|
|
}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "numpy.pxd":849
|
|
* # so don't output it
|
|
* f = _util_dtypestring(child, f, end, offset)
|
|
* return f # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = __pyx_v_f;
|
|
goto __pyx_L0;
|
|
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_XDECREF(__pyx_t_5);
|
|
__Pyx_AddTraceback("numpy._util_dtypestring", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_child);
|
|
__Pyx_XDECREF(__pyx_v_fields);
|
|
__Pyx_XDECREF(__pyx_v_childname);
|
|
__Pyx_XDECREF(__pyx_v_new_offset);
|
|
__Pyx_XDECREF(__pyx_v_t);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "numpy.pxd":964
|
|
*
|
|
*
|
|
* cdef inline void set_array_base(ndarray arr, object base): # <<<<<<<<<<<<<<
|
|
* cdef PyObject* baseptr
|
|
* if base is None:
|
|
*/
|
|
|
|
static CYTHON_INLINE void __pyx_f_5numpy_set_array_base(PyArrayObject *__pyx_v_arr, PyObject *__pyx_v_base) {
|
|
PyObject *__pyx_v_baseptr;
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
__Pyx_RefNannySetupContext("set_array_base", 0);
|
|
|
|
/* "numpy.pxd":966
|
|
* cdef inline void set_array_base(ndarray arr, object base):
|
|
* cdef PyObject* baseptr
|
|
* if base is None: # <<<<<<<<<<<<<<
|
|
* baseptr = NULL
|
|
* else:
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_base == Py_None);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "numpy.pxd":967
|
|
* cdef PyObject* baseptr
|
|
* if base is None:
|
|
* baseptr = NULL # <<<<<<<<<<<<<<
|
|
* else:
|
|
* Py_INCREF(base) # important to do this before decref below!
|
|
*/
|
|
__pyx_v_baseptr = NULL;
|
|
goto __pyx_L3;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "numpy.pxd":969
|
|
* baseptr = NULL
|
|
* else:
|
|
* Py_INCREF(base) # important to do this before decref below! # <<<<<<<<<<<<<<
|
|
* baseptr = <PyObject*>base
|
|
* Py_XDECREF(arr.base)
|
|
*/
|
|
Py_INCREF(__pyx_v_base);
|
|
|
|
/* "numpy.pxd":970
|
|
* else:
|
|
* Py_INCREF(base) # important to do this before decref below!
|
|
* baseptr = <PyObject*>base # <<<<<<<<<<<<<<
|
|
* Py_XDECREF(arr.base)
|
|
* arr.base = baseptr
|
|
*/
|
|
__pyx_v_baseptr = ((PyObject *)__pyx_v_base);
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "numpy.pxd":971
|
|
* Py_INCREF(base) # important to do this before decref below!
|
|
* baseptr = <PyObject*>base
|
|
* Py_XDECREF(arr.base) # <<<<<<<<<<<<<<
|
|
* arr.base = baseptr
|
|
*
|
|
*/
|
|
Py_XDECREF(__pyx_v_arr->base);
|
|
|
|
/* "numpy.pxd":972
|
|
* baseptr = <PyObject*>base
|
|
* Py_XDECREF(arr.base)
|
|
* arr.base = baseptr # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline object get_array_base(ndarray arr):
|
|
*/
|
|
__pyx_v_arr->base = __pyx_v_baseptr;
|
|
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "numpy.pxd":974
|
|
* arr.base = baseptr
|
|
*
|
|
* cdef inline object get_array_base(ndarray arr): # <<<<<<<<<<<<<<
|
|
* if arr.base is NULL:
|
|
* return None
|
|
*/
|
|
|
|
static CYTHON_INLINE PyObject *__pyx_f_5numpy_get_array_base(PyArrayObject *__pyx_v_arr) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
__Pyx_RefNannySetupContext("get_array_base", 0);
|
|
|
|
/* "numpy.pxd":975
|
|
*
|
|
* cdef inline object get_array_base(ndarray arr):
|
|
* if arr.base is NULL: # <<<<<<<<<<<<<<
|
|
* return None
|
|
* else:
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_arr->base == NULL);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "numpy.pxd":976
|
|
* cdef inline object get_array_base(ndarray arr):
|
|
* if arr.base is NULL:
|
|
* return None # <<<<<<<<<<<<<<
|
|
* else:
|
|
* return <object>arr.base
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__Pyx_INCREF(Py_None);
|
|
__pyx_r = Py_None;
|
|
goto __pyx_L0;
|
|
goto __pyx_L3;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "numpy.pxd":978
|
|
* return None
|
|
* else:
|
|
* return <object>arr.base # <<<<<<<<<<<<<<
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_arr->base));
|
|
__pyx_r = ((PyObject *)__pyx_v_arr->base);
|
|
goto __pyx_L0;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods[] = {
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
static struct PyModuleDef __pyx_moduledef = {
|
|
PyModuleDef_HEAD_INIT,
|
|
__Pyx_NAMESTR("_traversal"),
|
|
__Pyx_DOCSTR(__pyx_k_23), /* m_doc */
|
|
-1, /* m_size */
|
|
__pyx_methods /* m_methods */,
|
|
NULL, /* m_reload */
|
|
NULL, /* m_traverse */
|
|
NULL, /* m_clear */
|
|
NULL /* m_free */
|
|
};
|
|
#endif
|
|
|
|
static __Pyx_StringTabEntry __pyx_string_tab[] = {
|
|
{&__pyx_kp_u_11, __pyx_k_11, sizeof(__pyx_k_11), 0, 1, 0, 0},
|
|
{&__pyx_kp_u_13, __pyx_k_13, sizeof(__pyx_k_13), 0, 1, 0, 0},
|
|
{&__pyx_kp_u_15, __pyx_k_15, sizeof(__pyx_k_15), 0, 1, 0, 0},
|
|
{&__pyx_kp_u_17, __pyx_k_17, sizeof(__pyx_k_17), 0, 1, 0, 0},
|
|
{&__pyx_kp_u_18, __pyx_k_18, sizeof(__pyx_k_18), 0, 1, 0, 0},
|
|
{&__pyx_kp_u_21, __pyx_k_21, sizeof(__pyx_k_21), 0, 1, 0, 0},
|
|
{&__pyx_n_s_24, __pyx_k_24, sizeof(__pyx_k_24), 0, 0, 1, 1},
|
|
{&__pyx_n_s_27, __pyx_k_27, sizeof(__pyx_k_27), 0, 0, 1, 1},
|
|
{&__pyx_kp_s_28, __pyx_k_28, sizeof(__pyx_k_28), 0, 0, 1, 0},
|
|
{&__pyx_n_s_29, __pyx_k_29, sizeof(__pyx_k_29), 0, 0, 1, 1},
|
|
{&__pyx_kp_s_3, __pyx_k_3, sizeof(__pyx_k_3), 0, 0, 1, 0},
|
|
{&__pyx_kp_u_38, __pyx_k_38, sizeof(__pyx_k_38), 0, 1, 0, 0},
|
|
{&__pyx_kp_u_39, __pyx_k_39, sizeof(__pyx_k_39), 0, 1, 0, 0},
|
|
{&__pyx_kp_u_40, __pyx_k_40, sizeof(__pyx_k_40), 0, 1, 0, 0},
|
|
{&__pyx_kp_u_41, __pyx_k_41, sizeof(__pyx_k_41), 0, 1, 0, 0},
|
|
{&__pyx_n_s__DTYPE, __pyx_k__DTYPE, sizeof(__pyx_k__DTYPE), 0, 0, 1, 1},
|
|
{&__pyx_n_s__ITYPE, __pyx_k__ITYPE, sizeof(__pyx_k__ITYPE), 0, 0, 1, 1},
|
|
{&__pyx_n_s__N, __pyx_k__N, sizeof(__pyx_k__N), 0, 0, 1, 1},
|
|
{&__pyx_n_s__RuntimeError, __pyx_k__RuntimeError, sizeof(__pyx_k__RuntimeError), 0, 0, 1, 1},
|
|
{&__pyx_n_s__T, __pyx_k__T, sizeof(__pyx_k__T), 0, 0, 1, 1},
|
|
{&__pyx_n_s__ValueError, __pyx_k__ValueError, sizeof(__pyx_k__ValueError), 0, 0, 1, 1},
|
|
{&__pyx_n_s____main__, __pyx_k____main__, sizeof(__pyx_k____main__), 0, 0, 1, 1},
|
|
{&__pyx_n_s____test__, __pyx_k____test__, sizeof(__pyx_k____test__), 0, 0, 1, 1},
|
|
{&__pyx_n_s___graph_tools, __pyx_k___graph_tools, sizeof(__pyx_k___graph_tools), 0, 0, 1, 1},
|
|
{&__pyx_n_s___graph_validation, __pyx_k___graph_validation, sizeof(__pyx_k___graph_validation), 0, 0, 1, 1},
|
|
{&__pyx_n_s__breadth_first_order, __pyx_k__breadth_first_order, sizeof(__pyx_k__breadth_first_order), 0, 0, 1, 1},
|
|
{&__pyx_n_s__breadth_first_tree, __pyx_k__breadth_first_tree, sizeof(__pyx_k__breadth_first_tree), 0, 0, 1, 1},
|
|
{&__pyx_n_s__connection, __pyx_k__connection, sizeof(__pyx_k__connection), 0, 0, 1, 1},
|
|
{&__pyx_n_s__csgraph, __pyx_k__csgraph, sizeof(__pyx_k__csgraph), 0, 0, 1, 1},
|
|
{&__pyx_n_s__csgraph_T, __pyx_k__csgraph_T, sizeof(__pyx_k__csgraph_T), 0, 0, 1, 1},
|
|
{&__pyx_n_s__csr_matrix, __pyx_k__csr_matrix, sizeof(__pyx_k__csr_matrix), 0, 0, 1, 1},
|
|
{&__pyx_n_s__dense_output, __pyx_k__dense_output, sizeof(__pyx_k__dense_output), 0, 0, 1, 1},
|
|
{&__pyx_n_s__depth_first_order, __pyx_k__depth_first_order, sizeof(__pyx_k__depth_first_order), 0, 0, 1, 1},
|
|
{&__pyx_n_s__depth_first_tree, __pyx_k__depth_first_tree, sizeof(__pyx_k__depth_first_tree), 0, 0, 1, 1},
|
|
{&__pyx_n_s__directed, __pyx_k__directed, sizeof(__pyx_k__directed), 0, 0, 1, 1},
|
|
{&__pyx_n_s__dtype, __pyx_k__dtype, sizeof(__pyx_k__dtype), 0, 0, 1, 1},
|
|
{&__pyx_n_s__empty, __pyx_k__empty, sizeof(__pyx_k__empty), 0, 0, 1, 1},
|
|
{&__pyx_n_s__fill, __pyx_k__fill, sizeof(__pyx_k__fill), 0, 0, 1, 1},
|
|
{&__pyx_n_s__flag, __pyx_k__flag, sizeof(__pyx_k__flag), 0, 0, 1, 1},
|
|
{&__pyx_n_s__float64, __pyx_k__float64, sizeof(__pyx_k__float64), 0, 0, 1, 1},
|
|
{&__pyx_n_s__i_start, __pyx_k__i_start, sizeof(__pyx_k__i_start), 0, 0, 1, 1},
|
|
{&__pyx_n_s__indices, __pyx_k__indices, sizeof(__pyx_k__indices), 0, 0, 1, 1},
|
|
{&__pyx_n_s__indptr, __pyx_k__indptr, sizeof(__pyx_k__indptr), 0, 0, 1, 1},
|
|
{&__pyx_n_s__int32, __pyx_k__int32, sizeof(__pyx_k__int32), 0, 0, 1, 1},
|
|
{&__pyx_n_s__isspmatrix, __pyx_k__isspmatrix, sizeof(__pyx_k__isspmatrix), 0, 0, 1, 1},
|
|
{&__pyx_n_s__isspmatrix_csc, __pyx_k__isspmatrix_csc, sizeof(__pyx_k__isspmatrix_csc), 0, 0, 1, 1},
|
|
{&__pyx_n_s__isspmatrix_csr, __pyx_k__isspmatrix_csr, sizeof(__pyx_k__isspmatrix_csr), 0, 0, 1, 1},
|
|
{&__pyx_n_s__labels, __pyx_k__labels, sizeof(__pyx_k__labels), 0, 0, 1, 1},
|
|
{&__pyx_n_s__length, __pyx_k__length, sizeof(__pyx_k__length), 0, 0, 1, 1},
|
|
{&__pyx_n_s__lower, __pyx_k__lower, sizeof(__pyx_k__lower), 0, 0, 1, 1},
|
|
{&__pyx_n_s__n_components, __pyx_k__n_components, sizeof(__pyx_k__n_components), 0, 0, 1, 1},
|
|
{&__pyx_n_s__node_list, __pyx_k__node_list, sizeof(__pyx_k__node_list), 0, 0, 1, 1},
|
|
{&__pyx_n_s__np, __pyx_k__np, sizeof(__pyx_k__np), 0, 0, 1, 1},
|
|
{&__pyx_n_s__numpy, __pyx_k__numpy, sizeof(__pyx_k__numpy), 0, 0, 1, 1},
|
|
{&__pyx_n_s__predecessors, __pyx_k__predecessors, sizeof(__pyx_k__predecessors), 0, 0, 1, 1},
|
|
{&__pyx_n_s__range, __pyx_k__range, sizeof(__pyx_k__range), 0, 0, 1, 1},
|
|
{&__pyx_n_s__reconstruct_path, __pyx_k__reconstruct_path, sizeof(__pyx_k__reconstruct_path), 0, 0, 1, 1},
|
|
{&__pyx_n_s__return_labels, __pyx_k__return_labels, sizeof(__pyx_k__return_labels), 0, 0, 1, 1},
|
|
{&__pyx_n_s__return_predecessors, __pyx_k__return_predecessors, sizeof(__pyx_k__return_predecessors), 0, 0, 1, 1},
|
|
{&__pyx_n_s__root_list, __pyx_k__root_list, sizeof(__pyx_k__root_list), 0, 0, 1, 1},
|
|
{&__pyx_n_s__shape, __pyx_k__shape, sizeof(__pyx_k__shape), 0, 0, 1, 1},
|
|
{&__pyx_n_s__strong, __pyx_k__strong, sizeof(__pyx_k__strong), 0, 0, 1, 1},
|
|
{&__pyx_n_s__tocsr, __pyx_k__tocsr, sizeof(__pyx_k__tocsr), 0, 0, 1, 1},
|
|
{&__pyx_n_s__validate_graph, __pyx_k__validate_graph, sizeof(__pyx_k__validate_graph), 0, 0, 1, 1},
|
|
{&__pyx_n_s__weak, __pyx_k__weak, sizeof(__pyx_k__weak), 0, 0, 1, 1},
|
|
{&__pyx_n_s__zeros, __pyx_k__zeros, sizeof(__pyx_k__zeros), 0, 0, 1, 1},
|
|
{0, 0, 0, 0, 0, 0, 0}
|
|
};
|
|
static int __Pyx_InitCachedBuiltins(void) {
|
|
__pyx_builtin_ValueError = __Pyx_GetName(__pyx_b, __pyx_n_s__ValueError); if (!__pyx_builtin_ValueError) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 69; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_builtin_range = __Pyx_GetName(__pyx_b, __pyx_n_s__range); if (!__pyx_builtin_range) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 633; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_builtin_RuntimeError = __Pyx_GetName(__pyx_b, __pyx_n_s__RuntimeError); if (!__pyx_builtin_RuntimeError) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 798; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
return 0;
|
|
__pyx_L1_error:;
|
|
return -1;
|
|
}
|
|
|
|
static int __Pyx_InitCachedConstants(void) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__Pyx_InitCachedConstants", 0);
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":69
|
|
* """
|
|
* if connection.lower() not in ['weak', 'strong']:
|
|
* raise ValueError("connection must be 'weak' or 'strong'") # <<<<<<<<<<<<<<
|
|
*
|
|
* # weak connections <=> components of undirected graph
|
|
*/
|
|
__pyx_k_tuple_4 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 69; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_k_tuple_4);
|
|
__Pyx_INCREF(((PyObject *)__pyx_kp_s_3));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_4, 0, ((PyObject *)__pyx_kp_s_3));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_kp_s_3));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_4));
|
|
|
|
/* "numpy.pxd":214
|
|
* if ((flags & pybuf.PyBUF_C_CONTIGUOUS == pybuf.PyBUF_C_CONTIGUOUS)
|
|
* and not PyArray_CHKFLAGS(self, NPY_C_CONTIGUOUS)):
|
|
* raise ValueError(u"ndarray is not C contiguous") # <<<<<<<<<<<<<<
|
|
*
|
|
* if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS)
|
|
*/
|
|
__pyx_k_tuple_12 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_12)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 214; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_k_tuple_12);
|
|
__Pyx_INCREF(((PyObject *)__pyx_kp_u_11));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_12, 0, ((PyObject *)__pyx_kp_u_11));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_kp_u_11));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_12));
|
|
|
|
/* "numpy.pxd":218
|
|
* if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS)
|
|
* and not PyArray_CHKFLAGS(self, NPY_F_CONTIGUOUS)):
|
|
* raise ValueError(u"ndarray is not Fortran contiguous") # <<<<<<<<<<<<<<
|
|
*
|
|
* info.buf = PyArray_DATA(self)
|
|
*/
|
|
__pyx_k_tuple_14 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_14)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 218; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_k_tuple_14);
|
|
__Pyx_INCREF(((PyObject *)__pyx_kp_u_13));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_14, 0, ((PyObject *)__pyx_kp_u_13));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_kp_u_13));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_14));
|
|
|
|
/* "numpy.pxd":256
|
|
* if ((descr.byteorder == '>' and little_endian) or
|
|
* (descr.byteorder == '<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported") # <<<<<<<<<<<<<<
|
|
* if t == NPY_BYTE: f = "b"
|
|
* elif t == NPY_UBYTE: f = "B"
|
|
*/
|
|
__pyx_k_tuple_16 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_16)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 256; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_k_tuple_16);
|
|
__Pyx_INCREF(((PyObject *)__pyx_kp_u_15));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_16, 0, ((PyObject *)__pyx_kp_u_15));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_kp_u_15));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_16));
|
|
|
|
/* "numpy.pxd":798
|
|
*
|
|
* if (end - f) - (new_offset - offset[0]) < 15:
|
|
* raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd") # <<<<<<<<<<<<<<
|
|
*
|
|
* if ((child.byteorder == '>' and little_endian) or
|
|
*/
|
|
__pyx_k_tuple_19 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_19)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 798; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_k_tuple_19);
|
|
__Pyx_INCREF(((PyObject *)__pyx_kp_u_18));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_19, 0, ((PyObject *)__pyx_kp_u_18));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_kp_u_18));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_19));
|
|
|
|
/* "numpy.pxd":802
|
|
* if ((child.byteorder == '>' and little_endian) or
|
|
* (child.byteorder == '<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported") # <<<<<<<<<<<<<<
|
|
* # One could encode it in the format string and have Cython
|
|
* # complain instead, BUT: < and > in format strings also imply
|
|
*/
|
|
__pyx_k_tuple_20 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_20)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 802; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_k_tuple_20);
|
|
__Pyx_INCREF(((PyObject *)__pyx_kp_u_15));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_20, 0, ((PyObject *)__pyx_kp_u_15));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_kp_u_15));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_20));
|
|
|
|
/* "numpy.pxd":822
|
|
* t = child.type_num
|
|
* if end - f < 5:
|
|
* raise RuntimeError(u"Format string allocated too short.") # <<<<<<<<<<<<<<
|
|
*
|
|
* # Until ticket #99 is fixed, use integers to avoid warnings
|
|
*/
|
|
__pyx_k_tuple_22 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_22)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 822; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_k_tuple_22);
|
|
__Pyx_INCREF(((PyObject *)__pyx_kp_u_21));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_22, 0, ((PyObject *)__pyx_kp_u_21));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_kp_u_21));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_22));
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":30
|
|
* cdef ITYPE_t NULL_IDX = -9999
|
|
*
|
|
* def connected_components(csgraph, directed=True, connection='weak', # <<<<<<<<<<<<<<
|
|
* return_labels=True):
|
|
* """
|
|
*/
|
|
__pyx_k_tuple_25 = PyTuple_New(7); if (unlikely(!__pyx_k_tuple_25)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 30; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_k_tuple_25);
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__csgraph));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_25, 0, ((PyObject *)__pyx_n_s__csgraph));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__csgraph));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__directed));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_25, 1, ((PyObject *)__pyx_n_s__directed));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__directed));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__connection));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_25, 2, ((PyObject *)__pyx_n_s__connection));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__connection));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__return_labels));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_25, 3, ((PyObject *)__pyx_n_s__return_labels));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__return_labels));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__labels));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_25, 4, ((PyObject *)__pyx_n_s__labels));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__labels));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__n_components));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_25, 5, ((PyObject *)__pyx_n_s__n_components));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__n_components));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__csgraph_T));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_25, 6, ((PyObject *)__pyx_n_s__csgraph_T));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__csgraph_T));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_25));
|
|
__pyx_k_codeobj_26 = (PyObject*)__Pyx_PyCode_New(4, 0, 7, 0, 0, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_k_tuple_25, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_28, __pyx_n_s_27, 30, __pyx_empty_bytes); if (unlikely(!__pyx_k_codeobj_26)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 30; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":99
|
|
*
|
|
*
|
|
* def breadth_first_tree(csgraph, i_start, directed=True): # <<<<<<<<<<<<<<
|
|
* r"""
|
|
* breadth_first_tree(csgraph, i_start, directed=True)
|
|
*/
|
|
__pyx_k_tuple_30 = PyTuple_New(5); if (unlikely(!__pyx_k_tuple_30)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 99; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_k_tuple_30);
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__csgraph));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_30, 0, ((PyObject *)__pyx_n_s__csgraph));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__csgraph));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__i_start));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_30, 1, ((PyObject *)__pyx_n_s__i_start));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__i_start));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__directed));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_30, 2, ((PyObject *)__pyx_n_s__directed));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__directed));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__node_list));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_30, 3, ((PyObject *)__pyx_n_s__node_list));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__node_list));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__predecessors));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_30, 4, ((PyObject *)__pyx_n_s__predecessors));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__predecessors));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_30));
|
|
__pyx_k_codeobj_31 = (PyObject*)__Pyx_PyCode_New(3, 0, 5, 0, 0, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_k_tuple_30, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_28, __pyx_n_s__breadth_first_tree, 99, __pyx_empty_bytes); if (unlikely(!__pyx_k_codeobj_31)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 99; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":169
|
|
*
|
|
*
|
|
* def depth_first_tree(csgraph, i_start, directed=True): # <<<<<<<<<<<<<<
|
|
* r"""
|
|
* depth_first_tree(csgraph, i_start, directed=True)
|
|
*/
|
|
__pyx_k_tuple_32 = PyTuple_New(5); if (unlikely(!__pyx_k_tuple_32)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 169; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_k_tuple_32);
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__csgraph));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_32, 0, ((PyObject *)__pyx_n_s__csgraph));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__csgraph));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__i_start));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_32, 1, ((PyObject *)__pyx_n_s__i_start));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__i_start));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__directed));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_32, 2, ((PyObject *)__pyx_n_s__directed));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__directed));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__node_list));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_32, 3, ((PyObject *)__pyx_n_s__node_list));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__node_list));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__predecessors));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_32, 4, ((PyObject *)__pyx_n_s__predecessors));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__predecessors));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_32));
|
|
__pyx_k_codeobj_33 = (PyObject*)__Pyx_PyCode_New(3, 0, 5, 0, 0, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_k_tuple_32, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_28, __pyx_n_s__depth_first_tree, 169, __pyx_empty_bytes); if (unlikely(!__pyx_k_codeobj_33)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 169; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":243
|
|
*
|
|
*
|
|
* def breadth_first_order(csgraph, i_start, # <<<<<<<<<<<<<<
|
|
* directed=True, return_predecessors=True):
|
|
* """
|
|
*/
|
|
__pyx_k_tuple_34 = PyTuple_New(9); if (unlikely(!__pyx_k_tuple_34)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 243; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_k_tuple_34);
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__csgraph));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_34, 0, ((PyObject *)__pyx_n_s__csgraph));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__csgraph));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__i_start));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_34, 1, ((PyObject *)__pyx_n_s__i_start));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__i_start));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__directed));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_34, 2, ((PyObject *)__pyx_n_s__directed));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__directed));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__return_predecessors));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_34, 3, ((PyObject *)__pyx_n_s__return_predecessors));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__return_predecessors));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__N));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_34, 4, ((PyObject *)__pyx_n_s__N));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__N));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__node_list));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_34, 5, ((PyObject *)__pyx_n_s__node_list));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__node_list));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__predecessors));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_34, 6, ((PyObject *)__pyx_n_s__predecessors));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__predecessors));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__length));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_34, 7, ((PyObject *)__pyx_n_s__length));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__length));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__csgraph_T));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_34, 8, ((PyObject *)__pyx_n_s__csgraph_T));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__csgraph_T));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_34));
|
|
__pyx_k_codeobj_35 = (PyObject*)__Pyx_PyCode_New(4, 0, 9, 0, 0, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_k_tuple_34, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_28, __pyx_n_s__breadth_first_order, 243, __pyx_empty_bytes); if (unlikely(!__pyx_k_codeobj_35)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 243; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":407
|
|
*
|
|
*
|
|
* def depth_first_order(csgraph, i_start, # <<<<<<<<<<<<<<
|
|
* directed=True, return_predecessors=True):
|
|
* """
|
|
*/
|
|
__pyx_k_tuple_36 = PyTuple_New(11); if (unlikely(!__pyx_k_tuple_36)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 407; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_k_tuple_36);
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__csgraph));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_36, 0, ((PyObject *)__pyx_n_s__csgraph));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__csgraph));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__i_start));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_36, 1, ((PyObject *)__pyx_n_s__i_start));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__i_start));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__directed));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_36, 2, ((PyObject *)__pyx_n_s__directed));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__directed));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__return_predecessors));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_36, 3, ((PyObject *)__pyx_n_s__return_predecessors));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__return_predecessors));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__N));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_36, 4, ((PyObject *)__pyx_n_s__N));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__N));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__node_list));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_36, 5, ((PyObject *)__pyx_n_s__node_list));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__node_list));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__predecessors));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_36, 6, ((PyObject *)__pyx_n_s__predecessors));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__predecessors));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__root_list));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_36, 7, ((PyObject *)__pyx_n_s__root_list));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__root_list));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__flag));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_36, 8, ((PyObject *)__pyx_n_s__flag));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__flag));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__length));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_36, 9, ((PyObject *)__pyx_n_s__length));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__length));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__csgraph_T));
|
|
PyTuple_SET_ITEM(__pyx_k_tuple_36, 10, ((PyObject *)__pyx_n_s__csgraph_T));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__csgraph_T));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_36));
|
|
__pyx_k_codeobj_37 = (PyObject*)__Pyx_PyCode_New(4, 0, 11, 0, 0, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_k_tuple_36, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_28, __pyx_n_s__depth_first_order, 407, __pyx_empty_bytes); if (unlikely(!__pyx_k_codeobj_37)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 407; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_RefNannyFinishContext();
|
|
return 0;
|
|
__pyx_L1_error:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return -1;
|
|
}
|
|
|
|
static int __Pyx_InitGlobals(void) {
|
|
if (__Pyx_InitStrings(__pyx_string_tab) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__pyx_int_neg_1 = PyInt_FromLong(-1); if (unlikely(!__pyx_int_neg_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__pyx_int_15 = PyInt_FromLong(15); if (unlikely(!__pyx_int_15)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
return 0;
|
|
__pyx_L1_error:;
|
|
return -1;
|
|
}
|
|
|
|
#if PY_MAJOR_VERSION < 3
|
|
PyMODINIT_FUNC init_traversal(void); /*proto*/
|
|
PyMODINIT_FUNC init_traversal(void)
|
|
#else
|
|
PyMODINIT_FUNC PyInit__traversal(void); /*proto*/
|
|
PyMODINIT_FUNC PyInit__traversal(void)
|
|
#endif
|
|
{
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
#if CYTHON_REFNANNY
|
|
__Pyx_RefNanny = __Pyx_RefNannyImportAPI("refnanny");
|
|
if (!__Pyx_RefNanny) {
|
|
PyErr_Clear();
|
|
__Pyx_RefNanny = __Pyx_RefNannyImportAPI("Cython.Runtime.refnanny");
|
|
if (!__Pyx_RefNanny)
|
|
Py_FatalError("failed to import 'refnanny' module");
|
|
}
|
|
#endif
|
|
__Pyx_RefNannySetupContext("PyMODINIT_FUNC PyInit__traversal(void)", 0);
|
|
if ( __Pyx_check_binary_version() < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_empty_tuple = PyTuple_New(0); if (unlikely(!__pyx_empty_tuple)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_empty_bytes = PyBytes_FromStringAndSize("", 0); if (unlikely(!__pyx_empty_bytes)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#ifdef __Pyx_CyFunction_USED
|
|
if (__Pyx_CyFunction_init() < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#endif
|
|
#ifdef __Pyx_FusedFunction_USED
|
|
if (__pyx_FusedFunction_init() < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#endif
|
|
#ifdef __Pyx_Generator_USED
|
|
if (__pyx_Generator_init() < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#endif
|
|
/*--- Library function declarations ---*/
|
|
/*--- Threads initialization code ---*/
|
|
#if defined(__PYX_FORCE_INIT_THREADS) && __PYX_FORCE_INIT_THREADS
|
|
#ifdef WITH_THREAD /* Python build with threading support? */
|
|
PyEval_InitThreads();
|
|
#endif
|
|
#endif
|
|
/*--- Module creation code ---*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
__pyx_m = Py_InitModule4(__Pyx_NAMESTR("_traversal"), __pyx_methods, __Pyx_DOCSTR(__pyx_k_23), 0, PYTHON_API_VERSION);
|
|
#else
|
|
__pyx_m = PyModule_Create(&__pyx_moduledef);
|
|
#endif
|
|
if (!__pyx_m) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
#if PY_MAJOR_VERSION < 3
|
|
Py_INCREF(__pyx_m);
|
|
#endif
|
|
__pyx_b = PyImport_AddModule(__Pyx_NAMESTR(__Pyx_BUILTIN_MODULE_NAME));
|
|
if (!__pyx_b) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
if (__Pyx_SetAttrString(__pyx_m, "__builtins__", __pyx_b) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
/*--- Initialize various global constants etc. ---*/
|
|
if (unlikely(__Pyx_InitGlobals() < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__pyx_module_is_main_sklearn__utils__mst___traversal) {
|
|
if (__Pyx_SetAttrString(__pyx_m, "__name__", __pyx_n_s____main__) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
}
|
|
/*--- Builtin init code ---*/
|
|
if (unlikely(__Pyx_InitCachedBuiltins() < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
/*--- Constants init code ---*/
|
|
if (unlikely(__Pyx_InitCachedConstants() < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
/*--- Global init code ---*/
|
|
/*--- Variable export code ---*/
|
|
/*--- Function export code ---*/
|
|
/*--- Type init code ---*/
|
|
/*--- Type import code ---*/
|
|
__pyx_ptype_5numpy_dtype = __Pyx_ImportType("numpy", "dtype", sizeof(PyArray_Descr), 0); if (unlikely(!__pyx_ptype_5numpy_dtype)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 154; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_5numpy_flatiter = __Pyx_ImportType("numpy", "flatiter", sizeof(PyArrayIterObject), 0); if (unlikely(!__pyx_ptype_5numpy_flatiter)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 164; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_5numpy_broadcast = __Pyx_ImportType("numpy", "broadcast", sizeof(PyArrayMultiIterObject), 0); if (unlikely(!__pyx_ptype_5numpy_broadcast)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 168; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_5numpy_ndarray = __Pyx_ImportType("numpy", "ndarray", sizeof(PyArrayObject), 0); if (unlikely(!__pyx_ptype_5numpy_ndarray)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 177; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_5numpy_ufunc = __Pyx_ImportType("numpy", "ufunc", sizeof(PyUFuncObject), 0); if (unlikely(!__pyx_ptype_5numpy_ufunc)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 860; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
/*--- Variable import code ---*/
|
|
/*--- Function import code ---*/
|
|
/*--- Execution code ---*/
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":8
|
|
* # License: BSD, (C) 2012
|
|
*
|
|
* import numpy as np # <<<<<<<<<<<<<<
|
|
* cimport numpy as np
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_Import(((PyObject *)__pyx_n_s__numpy), 0, -1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 8; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__np, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 8; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":11
|
|
* cimport numpy as np
|
|
*
|
|
* from scipy.sparse import csr_matrix, isspmatrix, isspmatrix_csr, isspmatrix_csc # <<<<<<<<<<<<<<
|
|
* from ._graph_validation import validate_graph
|
|
* from ._graph_tools import reconstruct_path
|
|
*/
|
|
__pyx_t_1 = PyList_New(4); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 11; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__csr_matrix));
|
|
PyList_SET_ITEM(__pyx_t_1, 0, ((PyObject *)__pyx_n_s__csr_matrix));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__csr_matrix));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__isspmatrix));
|
|
PyList_SET_ITEM(__pyx_t_1, 1, ((PyObject *)__pyx_n_s__isspmatrix));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__isspmatrix));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__isspmatrix_csr));
|
|
PyList_SET_ITEM(__pyx_t_1, 2, ((PyObject *)__pyx_n_s__isspmatrix_csr));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__isspmatrix_csr));
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__isspmatrix_csc));
|
|
PyList_SET_ITEM(__pyx_t_1, 3, ((PyObject *)__pyx_n_s__isspmatrix_csc));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__isspmatrix_csc));
|
|
__pyx_t_2 = __Pyx_Import(((PyObject *)__pyx_n_s_24), ((PyObject *)__pyx_t_1), -1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 11; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_1)); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_t_2, __pyx_n_s__csr_matrix);
|
|
if (__pyx_t_1 == NULL) {
|
|
if (PyErr_ExceptionMatches(PyExc_AttributeError)) __Pyx_RaiseImportError(__pyx_n_s__csr_matrix);
|
|
if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 11; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__csr_matrix, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 11; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_t_2, __pyx_n_s__isspmatrix);
|
|
if (__pyx_t_1 == NULL) {
|
|
if (PyErr_ExceptionMatches(PyExc_AttributeError)) __Pyx_RaiseImportError(__pyx_n_s__isspmatrix);
|
|
if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 11; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__isspmatrix, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 11; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_t_2, __pyx_n_s__isspmatrix_csr);
|
|
if (__pyx_t_1 == NULL) {
|
|
if (PyErr_ExceptionMatches(PyExc_AttributeError)) __Pyx_RaiseImportError(__pyx_n_s__isspmatrix_csr);
|
|
if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 11; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__isspmatrix_csr, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 11; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_t_2, __pyx_n_s__isspmatrix_csc);
|
|
if (__pyx_t_1 == NULL) {
|
|
if (PyErr_ExceptionMatches(PyExc_AttributeError)) __Pyx_RaiseImportError(__pyx_n_s__isspmatrix_csc);
|
|
if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 11; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__isspmatrix_csc, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 11; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":12
|
|
*
|
|
* from scipy.sparse import csr_matrix, isspmatrix, isspmatrix_csr, isspmatrix_csc
|
|
* from ._graph_validation import validate_graph # <<<<<<<<<<<<<<
|
|
* from ._graph_tools import reconstruct_path
|
|
*
|
|
*/
|
|
__pyx_t_2 = PyList_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 12; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__validate_graph));
|
|
PyList_SET_ITEM(__pyx_t_2, 0, ((PyObject *)__pyx_n_s__validate_graph));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__validate_graph));
|
|
__pyx_t_1 = __Pyx_Import(((PyObject *)__pyx_n_s___graph_validation), ((PyObject *)__pyx_t_2), 1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 12; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
__pyx_t_2 = PyObject_GetAttr(__pyx_t_1, __pyx_n_s__validate_graph);
|
|
if (__pyx_t_2 == NULL) {
|
|
if (PyErr_ExceptionMatches(PyExc_AttributeError)) __Pyx_RaiseImportError(__pyx_n_s__validate_graph);
|
|
if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 12; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__validate_graph, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 12; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":13
|
|
* from scipy.sparse import csr_matrix, isspmatrix, isspmatrix_csr, isspmatrix_csc
|
|
* from ._graph_validation import validate_graph
|
|
* from ._graph_tools import reconstruct_path # <<<<<<<<<<<<<<
|
|
*
|
|
* cimport cython
|
|
*/
|
|
__pyx_t_1 = PyList_New(1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 13; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_INCREF(((PyObject *)__pyx_n_s__reconstruct_path));
|
|
PyList_SET_ITEM(__pyx_t_1, 0, ((PyObject *)__pyx_n_s__reconstruct_path));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_n_s__reconstruct_path));
|
|
__pyx_t_2 = __Pyx_Import(((PyObject *)__pyx_n_s___graph_tools), ((PyObject *)__pyx_t_1), 1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 13; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_1)); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_t_2, __pyx_n_s__reconstruct_path);
|
|
if (__pyx_t_1 == NULL) {
|
|
if (PyErr_ExceptionMatches(PyExc_AttributeError)) __Pyx_RaiseImportError(__pyx_n_s__reconstruct_path);
|
|
if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 13; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__reconstruct_path, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 13; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":18
|
|
* from libc cimport stdlib
|
|
*
|
|
* DTYPE = np.float64 # <<<<<<<<<<<<<<
|
|
* ctypedef np.float64_t DTYPE_t
|
|
*
|
|
*/
|
|
__pyx_t_2 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 18; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_1 = PyObject_GetAttr(__pyx_t_2, __pyx_n_s__float64); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 18; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__DTYPE, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 18; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":21
|
|
* ctypedef np.float64_t DTYPE_t
|
|
*
|
|
* ITYPE = np.int32 # <<<<<<<<<<<<<<
|
|
* ctypedef np.int32_t ITYPE_t
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 21; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyObject_GetAttr(__pyx_t_1, __pyx_n_s__int32); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 21; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__ITYPE, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 21; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":25
|
|
*
|
|
* # EPS is the precision of DTYPE
|
|
* cdef DTYPE_t DTYPE_EPS = 1E-15 # <<<<<<<<<<<<<<
|
|
*
|
|
* # NULL_IDX is the index used in predecessor matrices to store a non-path
|
|
*/
|
|
__pyx_v_7sklearn_5utils_3mst_10_traversal_DTYPE_EPS = 1E-15;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":28
|
|
*
|
|
* # NULL_IDX is the index used in predecessor matrices to store a non-path
|
|
* cdef ITYPE_t NULL_IDX = -9999 # <<<<<<<<<<<<<<
|
|
*
|
|
* def connected_components(csgraph, directed=True, connection='weak',
|
|
*/
|
|
__pyx_v_7sklearn_5utils_3mst_10_traversal_NULL_IDX = -9999;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":30
|
|
* cdef ITYPE_t NULL_IDX = -9999
|
|
*
|
|
* def connected_components(csgraph, directed=True, connection='weak', # <<<<<<<<<<<<<<
|
|
* return_labels=True):
|
|
* """
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyBool_FromLong(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 30; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_k_1 = __pyx_t_2;
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":31
|
|
*
|
|
* def connected_components(csgraph, directed=True, connection='weak',
|
|
* return_labels=True): # <<<<<<<<<<<<<<
|
|
* """
|
|
* connected_components(csgraph, directed=True, connection='weak',
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyBool_FromLong(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 31; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_k_2 = __pyx_t_2;
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":30
|
|
* cdef ITYPE_t NULL_IDX = -9999
|
|
*
|
|
* def connected_components(csgraph, directed=True, connection='weak', # <<<<<<<<<<<<<<
|
|
* return_labels=True):
|
|
* """
|
|
*/
|
|
__pyx_t_2 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_5utils_3mst_10_traversal_1connected_components, NULL, __pyx_n_s_29); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 30; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s_27, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 30; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":99
|
|
*
|
|
*
|
|
* def breadth_first_tree(csgraph, i_start, directed=True): # <<<<<<<<<<<<<<
|
|
* r"""
|
|
* breadth_first_tree(csgraph, i_start, directed=True)
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyBool_FromLong(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 99; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_k_5 = __pyx_t_2;
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_5utils_3mst_10_traversal_3breadth_first_tree, NULL, __pyx_n_s_29); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 99; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__breadth_first_tree, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 99; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":169
|
|
*
|
|
*
|
|
* def depth_first_tree(csgraph, i_start, directed=True): # <<<<<<<<<<<<<<
|
|
* r"""
|
|
* depth_first_tree(csgraph, i_start, directed=True)
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyBool_FromLong(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 169; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_k_6 = __pyx_t_2;
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_5utils_3mst_10_traversal_5depth_first_tree, NULL, __pyx_n_s_29); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 169; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__depth_first_tree, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 169; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":244
|
|
*
|
|
* def breadth_first_order(csgraph, i_start,
|
|
* directed=True, return_predecessors=True): # <<<<<<<<<<<<<<
|
|
* """
|
|
* breadth_first_order(csgraph, i_start, directed=True, return_predecessors=True)
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyBool_FromLong(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 244; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_k_7 = __pyx_t_2;
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = __Pyx_PyBool_FromLong(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 244; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_k_8 = __pyx_t_2;
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":243
|
|
*
|
|
*
|
|
* def breadth_first_order(csgraph, i_start, # <<<<<<<<<<<<<<
|
|
* directed=True, return_predecessors=True):
|
|
* """
|
|
*/
|
|
__pyx_t_2 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_5utils_3mst_10_traversal_7breadth_first_order, NULL, __pyx_n_s_29); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 243; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__breadth_first_order, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 243; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":408
|
|
*
|
|
* def depth_first_order(csgraph, i_start,
|
|
* directed=True, return_predecessors=True): # <<<<<<<<<<<<<<
|
|
* """
|
|
* depth_first_order(csgraph, i_start, directed=True, return_predecessors=True)
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyBool_FromLong(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 408; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_k_9 = __pyx_t_2;
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = __Pyx_PyBool_FromLong(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 408; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_k_10 = __pyx_t_2;
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":407
|
|
*
|
|
*
|
|
* def depth_first_order(csgraph, i_start, # <<<<<<<<<<<<<<
|
|
* directed=True, return_predecessors=True):
|
|
* """
|
|
*/
|
|
__pyx_t_2 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_5utils_3mst_10_traversal_9depth_first_order, NULL, __pyx_n_s_29); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 407; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__depth_first_order, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 407; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/utils/mst/_traversal.pyx":1
|
|
* """ # <<<<<<<<<<<<<<
|
|
* Routines for traversing graphs in compressed sparse format
|
|
* """
|
|
*/
|
|
__pyx_t_2 = PyDict_New(); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(((PyObject *)__pyx_t_2));
|
|
if (PyDict_SetItem(__pyx_t_2, ((PyObject *)__pyx_kp_u_38), ((PyObject *)__pyx_kp_u_39)) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (PyDict_SetItem(__pyx_t_2, ((PyObject *)__pyx_kp_u_40), ((PyObject *)__pyx_kp_u_41)) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (PyObject_SetAttr(__pyx_m, __pyx_n_s____test__, ((PyObject *)__pyx_t_2)) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
|
|
|
|
/* "numpy.pxd":974
|
|
* arr.base = baseptr
|
|
*
|
|
* cdef inline object get_array_base(ndarray arr): # <<<<<<<<<<<<<<
|
|
* if arr.base is NULL:
|
|
* return None
|
|
*/
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
if (__pyx_m) {
|
|
__Pyx_AddTraceback("init sklearn.utils.mst._traversal", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
Py_DECREF(__pyx_m); __pyx_m = 0;
|
|
} else if (!PyErr_Occurred()) {
|
|
PyErr_SetString(PyExc_ImportError, "init sklearn.utils.mst._traversal");
|
|
}
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
#if PY_MAJOR_VERSION < 3
|
|
return;
|
|
#else
|
|
return __pyx_m;
|
|
#endif
|
|
}
|
|
|
|
/* Runtime support code */
|
|
#if CYTHON_REFNANNY
|
|
static __Pyx_RefNannyAPIStruct *__Pyx_RefNannyImportAPI(const char *modname) {
|
|
PyObject *m = NULL, *p = NULL;
|
|
void *r = NULL;
|
|
m = PyImport_ImportModule((char *)modname);
|
|
if (!m) goto end;
|
|
p = PyObject_GetAttrString(m, (char *)"RefNannyAPI");
|
|
if (!p) goto end;
|
|
r = PyLong_AsVoidPtr(p);
|
|
end:
|
|
Py_XDECREF(p);
|
|
Py_XDECREF(m);
|
|
return (__Pyx_RefNannyAPIStruct *)r;
|
|
}
|
|
#endif /* CYTHON_REFNANNY */
|
|
|
|
static PyObject *__Pyx_GetName(PyObject *dict, PyObject *name) {
|
|
PyObject *result;
|
|
result = PyObject_GetAttr(dict, name);
|
|
if (!result) {
|
|
if (dict != __pyx_b) {
|
|
PyErr_Clear();
|
|
result = PyObject_GetAttr(__pyx_b, name);
|
|
}
|
|
if (!result) {
|
|
PyErr_SetObject(PyExc_NameError, name);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
static void __Pyx_RaiseDoubleKeywordsError(
|
|
const char* func_name,
|
|
PyObject* kw_name)
|
|
{
|
|
PyErr_Format(PyExc_TypeError,
|
|
#if PY_MAJOR_VERSION >= 3
|
|
"%s() got multiple values for keyword argument '%U'", func_name, kw_name);
|
|
#else
|
|
"%s() got multiple values for keyword argument '%s'", func_name,
|
|
PyString_AS_STRING(kw_name));
|
|
#endif
|
|
}
|
|
|
|
static int __Pyx_ParseOptionalKeywords(
|
|
PyObject *kwds,
|
|
PyObject **argnames[],
|
|
PyObject *kwds2,
|
|
PyObject *values[],
|
|
Py_ssize_t num_pos_args,
|
|
const char* function_name)
|
|
{
|
|
PyObject *key = 0, *value = 0;
|
|
Py_ssize_t pos = 0;
|
|
PyObject*** name;
|
|
PyObject*** first_kw_arg = argnames + num_pos_args;
|
|
while (PyDict_Next(kwds, &pos, &key, &value)) {
|
|
name = first_kw_arg;
|
|
while (*name && (**name != key)) name++;
|
|
if (*name) {
|
|
values[name-argnames] = value;
|
|
} else {
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (unlikely(!PyString_CheckExact(key)) && unlikely(!PyString_Check(key))) {
|
|
#else
|
|
if (unlikely(!PyUnicode_Check(key))) {
|
|
#endif
|
|
goto invalid_keyword_type;
|
|
} else {
|
|
for (name = first_kw_arg; *name; name++) {
|
|
#if PY_MAJOR_VERSION >= 3
|
|
if (PyUnicode_GET_SIZE(**name) == PyUnicode_GET_SIZE(key) &&
|
|
PyUnicode_Compare(**name, key) == 0) break;
|
|
#else
|
|
if (PyString_GET_SIZE(**name) == PyString_GET_SIZE(key) &&
|
|
_PyString_Eq(**name, key)) break;
|
|
#endif
|
|
}
|
|
if (*name) {
|
|
values[name-argnames] = value;
|
|
} else {
|
|
for (name=argnames; name != first_kw_arg; name++) {
|
|
if (**name == key) goto arg_passed_twice;
|
|
#if PY_MAJOR_VERSION >= 3
|
|
if (PyUnicode_GET_SIZE(**name) == PyUnicode_GET_SIZE(key) &&
|
|
PyUnicode_Compare(**name, key) == 0) goto arg_passed_twice;
|
|
#else
|
|
if (PyString_GET_SIZE(**name) == PyString_GET_SIZE(key) &&
|
|
_PyString_Eq(**name, key)) goto arg_passed_twice;
|
|
#endif
|
|
}
|
|
if (kwds2) {
|
|
if (unlikely(PyDict_SetItem(kwds2, key, value))) goto bad;
|
|
} else {
|
|
goto invalid_keyword;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return 0;
|
|
arg_passed_twice:
|
|
__Pyx_RaiseDoubleKeywordsError(function_name, **name);
|
|
goto bad;
|
|
invalid_keyword_type:
|
|
PyErr_Format(PyExc_TypeError,
|
|
"%s() keywords must be strings", function_name);
|
|
goto bad;
|
|
invalid_keyword:
|
|
PyErr_Format(PyExc_TypeError,
|
|
#if PY_MAJOR_VERSION < 3
|
|
"%s() got an unexpected keyword argument '%s'",
|
|
function_name, PyString_AsString(key));
|
|
#else
|
|
"%s() got an unexpected keyword argument '%U'",
|
|
function_name, key);
|
|
#endif
|
|
bad:
|
|
return -1;
|
|
}
|
|
|
|
static void __Pyx_RaiseArgtupleInvalid(
|
|
const char* func_name,
|
|
int exact,
|
|
Py_ssize_t num_min,
|
|
Py_ssize_t num_max,
|
|
Py_ssize_t num_found)
|
|
{
|
|
Py_ssize_t num_expected;
|
|
const char *more_or_less;
|
|
if (num_found < num_min) {
|
|
num_expected = num_min;
|
|
more_or_less = "at least";
|
|
} else {
|
|
num_expected = num_max;
|
|
more_or_less = "at most";
|
|
}
|
|
if (exact) {
|
|
more_or_less = "exactly";
|
|
}
|
|
PyErr_Format(PyExc_TypeError,
|
|
"%s() takes %s %"PY_FORMAT_SIZE_T"d positional argument%s (%"PY_FORMAT_SIZE_T"d given)",
|
|
func_name, more_or_less, num_expected,
|
|
(num_expected == 1) ? "" : "s", num_found);
|
|
}
|
|
|
|
static CYTHON_INLINE void __Pyx_ErrRestore(PyObject *type, PyObject *value, PyObject *tb) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
PyObject *tmp_type, *tmp_value, *tmp_tb;
|
|
PyThreadState *tstate = PyThreadState_GET();
|
|
tmp_type = tstate->curexc_type;
|
|
tmp_value = tstate->curexc_value;
|
|
tmp_tb = tstate->curexc_traceback;
|
|
tstate->curexc_type = type;
|
|
tstate->curexc_value = value;
|
|
tstate->curexc_traceback = tb;
|
|
Py_XDECREF(tmp_type);
|
|
Py_XDECREF(tmp_value);
|
|
Py_XDECREF(tmp_tb);
|
|
#else
|
|
PyErr_Restore(type, value, tb);
|
|
#endif
|
|
}
|
|
static CYTHON_INLINE void __Pyx_ErrFetch(PyObject **type, PyObject **value, PyObject **tb) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
PyThreadState *tstate = PyThreadState_GET();
|
|
*type = tstate->curexc_type;
|
|
*value = tstate->curexc_value;
|
|
*tb = tstate->curexc_traceback;
|
|
tstate->curexc_type = 0;
|
|
tstate->curexc_value = 0;
|
|
tstate->curexc_traceback = 0;
|
|
#else
|
|
PyErr_Fetch(type, value, tb);
|
|
#endif
|
|
}
|
|
|
|
#if PY_MAJOR_VERSION < 3
|
|
static void __Pyx_Raise(PyObject *type, PyObject *value, PyObject *tb,
|
|
CYTHON_UNUSED PyObject *cause) {
|
|
Py_XINCREF(type);
|
|
Py_XINCREF(value);
|
|
Py_XINCREF(tb);
|
|
if (tb == Py_None) {
|
|
Py_DECREF(tb);
|
|
tb = 0;
|
|
}
|
|
else if (tb != NULL && !PyTraceBack_Check(tb)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"raise: arg 3 must be a traceback or None");
|
|
goto raise_error;
|
|
}
|
|
if (value == NULL) {
|
|
value = Py_None;
|
|
Py_INCREF(value);
|
|
}
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
if (!PyClass_Check(type))
|
|
#else
|
|
if (!PyType_Check(type))
|
|
#endif
|
|
{
|
|
if (value != Py_None) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"instance exception may not have a separate value");
|
|
goto raise_error;
|
|
}
|
|
Py_DECREF(value);
|
|
value = type;
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
if (PyInstance_Check(type)) {
|
|
type = (PyObject*) ((PyInstanceObject*)type)->in_class;
|
|
Py_INCREF(type);
|
|
}
|
|
else {
|
|
type = 0;
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"raise: exception must be an old-style class or instance");
|
|
goto raise_error;
|
|
}
|
|
#else
|
|
type = (PyObject*) Py_TYPE(type);
|
|
Py_INCREF(type);
|
|
if (!PyType_IsSubtype((PyTypeObject *)type, (PyTypeObject *)PyExc_BaseException)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"raise: exception class must be a subclass of BaseException");
|
|
goto raise_error;
|
|
}
|
|
#endif
|
|
}
|
|
__Pyx_ErrRestore(type, value, tb);
|
|
return;
|
|
raise_error:
|
|
Py_XDECREF(value);
|
|
Py_XDECREF(type);
|
|
Py_XDECREF(tb);
|
|
return;
|
|
}
|
|
#else /* Python 3+ */
|
|
static void __Pyx_Raise(PyObject *type, PyObject *value, PyObject *tb, PyObject *cause) {
|
|
if (tb == Py_None) {
|
|
tb = 0;
|
|
} else if (tb && !PyTraceBack_Check(tb)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"raise: arg 3 must be a traceback or None");
|
|
goto bad;
|
|
}
|
|
if (value == Py_None)
|
|
value = 0;
|
|
if (PyExceptionInstance_Check(type)) {
|
|
if (value) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"instance exception may not have a separate value");
|
|
goto bad;
|
|
}
|
|
value = type;
|
|
type = (PyObject*) Py_TYPE(value);
|
|
} else if (!PyExceptionClass_Check(type)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"raise: exception class must be a subclass of BaseException");
|
|
goto bad;
|
|
}
|
|
if (cause) {
|
|
PyObject *fixed_cause;
|
|
if (PyExceptionClass_Check(cause)) {
|
|
fixed_cause = PyObject_CallObject(cause, NULL);
|
|
if (fixed_cause == NULL)
|
|
goto bad;
|
|
}
|
|
else if (PyExceptionInstance_Check(cause)) {
|
|
fixed_cause = cause;
|
|
Py_INCREF(fixed_cause);
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"exception causes must derive from "
|
|
"BaseException");
|
|
goto bad;
|
|
}
|
|
if (!value) {
|
|
value = PyObject_CallObject(type, NULL);
|
|
}
|
|
PyException_SetCause(value, fixed_cause);
|
|
}
|
|
PyErr_SetObject(type, value);
|
|
if (tb) {
|
|
PyThreadState *tstate = PyThreadState_GET();
|
|
PyObject* tmp_tb = tstate->curexc_traceback;
|
|
if (tb != tmp_tb) {
|
|
Py_INCREF(tb);
|
|
tstate->curexc_traceback = tb;
|
|
Py_XDECREF(tmp_tb);
|
|
}
|
|
}
|
|
bad:
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
|
|
|
|
static CYTHON_INLINE int __Pyx_TypeTest(PyObject *obj, PyTypeObject *type) {
|
|
if (unlikely(!type)) {
|
|
PyErr_Format(PyExc_SystemError, "Missing type object");
|
|
return 0;
|
|
}
|
|
if (likely(PyObject_TypeCheck(obj, type)))
|
|
return 1;
|
|
PyErr_Format(PyExc_TypeError, "Cannot convert %.200s to %.200s",
|
|
Py_TYPE(obj)->tp_name, type->tp_name);
|
|
return 0;
|
|
}
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseNeedMoreValuesError(Py_ssize_t index) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"need more than %"PY_FORMAT_SIZE_T"d value%s to unpack",
|
|
index, (index == 1) ? "" : "s");
|
|
}
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseTooManyValuesError(Py_ssize_t expected) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"too many values to unpack (expected %"PY_FORMAT_SIZE_T"d)", expected);
|
|
}
|
|
|
|
static int __Pyx_IternextUnpackEndCheck(PyObject *retval, Py_ssize_t expected) {
|
|
if (unlikely(retval)) {
|
|
Py_DECREF(retval);
|
|
__Pyx_RaiseTooManyValuesError(expected);
|
|
return -1;
|
|
} else if (PyErr_Occurred()) {
|
|
if (likely(PyErr_ExceptionMatches(PyExc_StopIteration))) {
|
|
PyErr_Clear();
|
|
return 0;
|
|
} else {
|
|
return -1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static CYTHON_INLINE int __Pyx_IsLittleEndian(void) {
|
|
unsigned int n = 1;
|
|
return *(unsigned char*)(&n) != 0;
|
|
}
|
|
static void __Pyx_BufFmt_Init(__Pyx_BufFmt_Context* ctx,
|
|
__Pyx_BufFmt_StackElem* stack,
|
|
__Pyx_TypeInfo* type) {
|
|
stack[0].field = &ctx->root;
|
|
stack[0].parent_offset = 0;
|
|
ctx->root.type = type;
|
|
ctx->root.name = "buffer dtype";
|
|
ctx->root.offset = 0;
|
|
ctx->head = stack;
|
|
ctx->head->field = &ctx->root;
|
|
ctx->fmt_offset = 0;
|
|
ctx->head->parent_offset = 0;
|
|
ctx->new_packmode = '@';
|
|
ctx->enc_packmode = '@';
|
|
ctx->new_count = 1;
|
|
ctx->enc_count = 0;
|
|
ctx->enc_type = 0;
|
|
ctx->is_complex = 0;
|
|
ctx->is_valid_array = 0;
|
|
ctx->struct_alignment = 0;
|
|
while (type->typegroup == 'S') {
|
|
++ctx->head;
|
|
ctx->head->field = type->fields;
|
|
ctx->head->parent_offset = 0;
|
|
type = type->fields->type;
|
|
}
|
|
}
|
|
static int __Pyx_BufFmt_ParseNumber(const char** ts) {
|
|
int count;
|
|
const char* t = *ts;
|
|
if (*t < '0' || *t > '9') {
|
|
return -1;
|
|
} else {
|
|
count = *t++ - '0';
|
|
while (*t >= '0' && *t < '9') {
|
|
count *= 10;
|
|
count += *t++ - '0';
|
|
}
|
|
}
|
|
*ts = t;
|
|
return count;
|
|
}
|
|
static int __Pyx_BufFmt_ExpectNumber(const char **ts) {
|
|
int number = __Pyx_BufFmt_ParseNumber(ts);
|
|
if (number == -1) /* First char was not a digit */
|
|
PyErr_Format(PyExc_ValueError,\
|
|
"Does not understand character buffer dtype format string ('%c')", **ts);
|
|
return number;
|
|
}
|
|
static void __Pyx_BufFmt_RaiseUnexpectedChar(char ch) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Unexpected format string character: '%c'", ch);
|
|
}
|
|
static const char* __Pyx_BufFmt_DescribeTypeChar(char ch, int is_complex) {
|
|
switch (ch) {
|
|
case 'b': return "'char'";
|
|
case 'B': return "'unsigned char'";
|
|
case 'h': return "'short'";
|
|
case 'H': return "'unsigned short'";
|
|
case 'i': return "'int'";
|
|
case 'I': return "'unsigned int'";
|
|
case 'l': return "'long'";
|
|
case 'L': return "'unsigned long'";
|
|
case 'q': return "'long long'";
|
|
case 'Q': return "'unsigned long long'";
|
|
case 'f': return (is_complex ? "'complex float'" : "'float'");
|
|
case 'd': return (is_complex ? "'complex double'" : "'double'");
|
|
case 'g': return (is_complex ? "'complex long double'" : "'long double'");
|
|
case 'T': return "a struct";
|
|
case 'O': return "Python object";
|
|
case 'P': return "a pointer";
|
|
case 's': case 'p': return "a string";
|
|
case 0: return "end";
|
|
default: return "unparseable format string";
|
|
}
|
|
}
|
|
static size_t __Pyx_BufFmt_TypeCharToStandardSize(char ch, int is_complex) {
|
|
switch (ch) {
|
|
case '?': case 'c': case 'b': case 'B': case 's': case 'p': return 1;
|
|
case 'h': case 'H': return 2;
|
|
case 'i': case 'I': case 'l': case 'L': return 4;
|
|
case 'q': case 'Q': return 8;
|
|
case 'f': return (is_complex ? 8 : 4);
|
|
case 'd': return (is_complex ? 16 : 8);
|
|
case 'g': {
|
|
PyErr_SetString(PyExc_ValueError, "Python does not define a standard format string size for long double ('g')..");
|
|
return 0;
|
|
}
|
|
case 'O': case 'P': return sizeof(void*);
|
|
default:
|
|
__Pyx_BufFmt_RaiseUnexpectedChar(ch);
|
|
return 0;
|
|
}
|
|
}
|
|
static size_t __Pyx_BufFmt_TypeCharToNativeSize(char ch, int is_complex) {
|
|
switch (ch) {
|
|
case 'c': case 'b': case 'B': case 's': case 'p': return 1;
|
|
case 'h': case 'H': return sizeof(short);
|
|
case 'i': case 'I': return sizeof(int);
|
|
case 'l': case 'L': return sizeof(long);
|
|
#ifdef HAVE_LONG_LONG
|
|
case 'q': case 'Q': return sizeof(PY_LONG_LONG);
|
|
#endif
|
|
case 'f': return sizeof(float) * (is_complex ? 2 : 1);
|
|
case 'd': return sizeof(double) * (is_complex ? 2 : 1);
|
|
case 'g': return sizeof(long double) * (is_complex ? 2 : 1);
|
|
case 'O': case 'P': return sizeof(void*);
|
|
default: {
|
|
__Pyx_BufFmt_RaiseUnexpectedChar(ch);
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
typedef struct { char c; short x; } __Pyx_st_short;
|
|
typedef struct { char c; int x; } __Pyx_st_int;
|
|
typedef struct { char c; long x; } __Pyx_st_long;
|
|
typedef struct { char c; float x; } __Pyx_st_float;
|
|
typedef struct { char c; double x; } __Pyx_st_double;
|
|
typedef struct { char c; long double x; } __Pyx_st_longdouble;
|
|
typedef struct { char c; void *x; } __Pyx_st_void_p;
|
|
#ifdef HAVE_LONG_LONG
|
|
typedef struct { char c; PY_LONG_LONG x; } __Pyx_st_longlong;
|
|
#endif
|
|
static size_t __Pyx_BufFmt_TypeCharToAlignment(char ch, int is_complex) {
|
|
switch (ch) {
|
|
case '?': case 'c': case 'b': case 'B': case 's': case 'p': return 1;
|
|
case 'h': case 'H': return sizeof(__Pyx_st_short) - sizeof(short);
|
|
case 'i': case 'I': return sizeof(__Pyx_st_int) - sizeof(int);
|
|
case 'l': case 'L': return sizeof(__Pyx_st_long) - sizeof(long);
|
|
#ifdef HAVE_LONG_LONG
|
|
case 'q': case 'Q': return sizeof(__Pyx_st_longlong) - sizeof(PY_LONG_LONG);
|
|
#endif
|
|
case 'f': return sizeof(__Pyx_st_float) - sizeof(float);
|
|
case 'd': return sizeof(__Pyx_st_double) - sizeof(double);
|
|
case 'g': return sizeof(__Pyx_st_longdouble) - sizeof(long double);
|
|
case 'P': case 'O': return sizeof(__Pyx_st_void_p) - sizeof(void*);
|
|
default:
|
|
__Pyx_BufFmt_RaiseUnexpectedChar(ch);
|
|
return 0;
|
|
}
|
|
}
|
|
/* These are for computing the padding at the end of the struct to align
|
|
on the first member of the struct. This will probably the same as above,
|
|
but we don't have any guarantees.
|
|
*/
|
|
typedef struct { short x; char c; } __Pyx_pad_short;
|
|
typedef struct { int x; char c; } __Pyx_pad_int;
|
|
typedef struct { long x; char c; } __Pyx_pad_long;
|
|
typedef struct { float x; char c; } __Pyx_pad_float;
|
|
typedef struct { double x; char c; } __Pyx_pad_double;
|
|
typedef struct { long double x; char c; } __Pyx_pad_longdouble;
|
|
typedef struct { void *x; char c; } __Pyx_pad_void_p;
|
|
#ifdef HAVE_LONG_LONG
|
|
typedef struct { PY_LONG_LONG x; char c; } __Pyx_pad_longlong;
|
|
#endif
|
|
static size_t __Pyx_BufFmt_TypeCharToPadding(char ch, int is_complex) {
|
|
switch (ch) {
|
|
case '?': case 'c': case 'b': case 'B': case 's': case 'p': return 1;
|
|
case 'h': case 'H': return sizeof(__Pyx_pad_short) - sizeof(short);
|
|
case 'i': case 'I': return sizeof(__Pyx_pad_int) - sizeof(int);
|
|
case 'l': case 'L': return sizeof(__Pyx_pad_long) - sizeof(long);
|
|
#ifdef HAVE_LONG_LONG
|
|
case 'q': case 'Q': return sizeof(__Pyx_pad_longlong) - sizeof(PY_LONG_LONG);
|
|
#endif
|
|
case 'f': return sizeof(__Pyx_pad_float) - sizeof(float);
|
|
case 'd': return sizeof(__Pyx_pad_double) - sizeof(double);
|
|
case 'g': return sizeof(__Pyx_pad_longdouble) - sizeof(long double);
|
|
case 'P': case 'O': return sizeof(__Pyx_pad_void_p) - sizeof(void*);
|
|
default:
|
|
__Pyx_BufFmt_RaiseUnexpectedChar(ch);
|
|
return 0;
|
|
}
|
|
}
|
|
static char __Pyx_BufFmt_TypeCharToGroup(char ch, int is_complex) {
|
|
switch (ch) {
|
|
case 'c': case 'b': case 'h': case 'i':
|
|
case 'l': case 'q': case 's': case 'p':
|
|
return 'I';
|
|
case 'B': case 'H': case 'I': case 'L': case 'Q':
|
|
return 'U';
|
|
case 'f': case 'd': case 'g':
|
|
return (is_complex ? 'C' : 'R');
|
|
case 'O':
|
|
return 'O';
|
|
case 'P':
|
|
return 'P';
|
|
default: {
|
|
__Pyx_BufFmt_RaiseUnexpectedChar(ch);
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
static void __Pyx_BufFmt_RaiseExpected(__Pyx_BufFmt_Context* ctx) {
|
|
if (ctx->head == NULL || ctx->head->field == &ctx->root) {
|
|
const char* expected;
|
|
const char* quote;
|
|
if (ctx->head == NULL) {
|
|
expected = "end";
|
|
quote = "";
|
|
} else {
|
|
expected = ctx->head->field->type->name;
|
|
quote = "'";
|
|
}
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Buffer dtype mismatch, expected %s%s%s but got %s",
|
|
quote, expected, quote,
|
|
__Pyx_BufFmt_DescribeTypeChar(ctx->enc_type, ctx->is_complex));
|
|
} else {
|
|
__Pyx_StructField* field = ctx->head->field;
|
|
__Pyx_StructField* parent = (ctx->head - 1)->field;
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Buffer dtype mismatch, expected '%s' but got %s in '%s.%s'",
|
|
field->type->name, __Pyx_BufFmt_DescribeTypeChar(ctx->enc_type, ctx->is_complex),
|
|
parent->type->name, field->name);
|
|
}
|
|
}
|
|
static int __Pyx_BufFmt_ProcessTypeChunk(__Pyx_BufFmt_Context* ctx) {
|
|
char group;
|
|
size_t size, offset, arraysize = 1;
|
|
if (ctx->enc_type == 0) return 0;
|
|
if (ctx->head->field->type->arraysize[0]) {
|
|
int i, ndim = 0;
|
|
if (ctx->enc_type == 's' || ctx->enc_type == 'p') {
|
|
ctx->is_valid_array = ctx->head->field->type->ndim == 1;
|
|
ndim = 1;
|
|
if (ctx->enc_count != ctx->head->field->type->arraysize[0]) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Expected a dimension of size %zu, got %zu",
|
|
ctx->head->field->type->arraysize[0], ctx->enc_count);
|
|
return -1;
|
|
}
|
|
}
|
|
if (!ctx->is_valid_array) {
|
|
PyErr_Format(PyExc_ValueError, "Expected %d dimensions, got %d",
|
|
ctx->head->field->type->ndim, ndim);
|
|
return -1;
|
|
}
|
|
for (i = 0; i < ctx->head->field->type->ndim; i++) {
|
|
arraysize *= ctx->head->field->type->arraysize[i];
|
|
}
|
|
ctx->is_valid_array = 0;
|
|
ctx->enc_count = 1;
|
|
}
|
|
group = __Pyx_BufFmt_TypeCharToGroup(ctx->enc_type, ctx->is_complex);
|
|
do {
|
|
__Pyx_StructField* field = ctx->head->field;
|
|
__Pyx_TypeInfo* type = field->type;
|
|
if (ctx->enc_packmode == '@' || ctx->enc_packmode == '^') {
|
|
size = __Pyx_BufFmt_TypeCharToNativeSize(ctx->enc_type, ctx->is_complex);
|
|
} else {
|
|
size = __Pyx_BufFmt_TypeCharToStandardSize(ctx->enc_type, ctx->is_complex);
|
|
}
|
|
if (ctx->enc_packmode == '@') {
|
|
size_t align_at = __Pyx_BufFmt_TypeCharToAlignment(ctx->enc_type, ctx->is_complex);
|
|
size_t align_mod_offset;
|
|
if (align_at == 0) return -1;
|
|
align_mod_offset = ctx->fmt_offset % align_at;
|
|
if (align_mod_offset > 0) ctx->fmt_offset += align_at - align_mod_offset;
|
|
if (ctx->struct_alignment == 0)
|
|
ctx->struct_alignment = __Pyx_BufFmt_TypeCharToPadding(ctx->enc_type,
|
|
ctx->is_complex);
|
|
}
|
|
if (type->size != size || type->typegroup != group) {
|
|
if (type->typegroup == 'C' && type->fields != NULL) {
|
|
size_t parent_offset = ctx->head->parent_offset + field->offset;
|
|
++ctx->head;
|
|
ctx->head->field = type->fields;
|
|
ctx->head->parent_offset = parent_offset;
|
|
continue;
|
|
}
|
|
__Pyx_BufFmt_RaiseExpected(ctx);
|
|
return -1;
|
|
}
|
|
offset = ctx->head->parent_offset + field->offset;
|
|
if (ctx->fmt_offset != offset) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Buffer dtype mismatch; next field is at offset %"PY_FORMAT_SIZE_T"d but %"PY_FORMAT_SIZE_T"d expected",
|
|
(Py_ssize_t)ctx->fmt_offset, (Py_ssize_t)offset);
|
|
return -1;
|
|
}
|
|
ctx->fmt_offset += size;
|
|
if (arraysize)
|
|
ctx->fmt_offset += (arraysize - 1) * size;
|
|
--ctx->enc_count; /* Consume from buffer string */
|
|
while (1) {
|
|
if (field == &ctx->root) {
|
|
ctx->head = NULL;
|
|
if (ctx->enc_count != 0) {
|
|
__Pyx_BufFmt_RaiseExpected(ctx);
|
|
return -1;
|
|
}
|
|
break; /* breaks both loops as ctx->enc_count == 0 */
|
|
}
|
|
ctx->head->field = ++field;
|
|
if (field->type == NULL) {
|
|
--ctx->head;
|
|
field = ctx->head->field;
|
|
continue;
|
|
} else if (field->type->typegroup == 'S') {
|
|
size_t parent_offset = ctx->head->parent_offset + field->offset;
|
|
if (field->type->fields->type == NULL) continue; /* empty struct */
|
|
field = field->type->fields;
|
|
++ctx->head;
|
|
ctx->head->field = field;
|
|
ctx->head->parent_offset = parent_offset;
|
|
break;
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
} while (ctx->enc_count);
|
|
ctx->enc_type = 0;
|
|
ctx->is_complex = 0;
|
|
return 0;
|
|
}
|
|
static CYTHON_INLINE PyObject *
|
|
__pyx_buffmt_parse_array(__Pyx_BufFmt_Context* ctx, const char** tsp)
|
|
{
|
|
const char *ts = *tsp;
|
|
int i = 0, number;
|
|
int ndim = ctx->head->field->type->ndim;
|
|
;
|
|
++ts;
|
|
if (ctx->new_count != 1) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"Cannot handle repeated arrays in format string");
|
|
return NULL;
|
|
}
|
|
if (__Pyx_BufFmt_ProcessTypeChunk(ctx) == -1) return NULL;
|
|
while (*ts && *ts != ')') {
|
|
if (isspace(*ts))
|
|
continue;
|
|
number = __Pyx_BufFmt_ExpectNumber(&ts);
|
|
if (number == -1) return NULL;
|
|
if (i < ndim && (size_t) number != ctx->head->field->type->arraysize[i])
|
|
return PyErr_Format(PyExc_ValueError,
|
|
"Expected a dimension of size %zu, got %d",
|
|
ctx->head->field->type->arraysize[i], number);
|
|
if (*ts != ',' && *ts != ')')
|
|
return PyErr_Format(PyExc_ValueError,
|
|
"Expected a comma in format string, got '%c'", *ts);
|
|
if (*ts == ',') ts++;
|
|
i++;
|
|
}
|
|
if (i != ndim)
|
|
return PyErr_Format(PyExc_ValueError, "Expected %d dimension(s), got %d",
|
|
ctx->head->field->type->ndim, i);
|
|
if (!*ts) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"Unexpected end of format string, expected ')'");
|
|
return NULL;
|
|
}
|
|
ctx->is_valid_array = 1;
|
|
ctx->new_count = 1;
|
|
*tsp = ++ts;
|
|
return Py_None;
|
|
}
|
|
static const char* __Pyx_BufFmt_CheckString(__Pyx_BufFmt_Context* ctx, const char* ts) {
|
|
int got_Z = 0;
|
|
while (1) {
|
|
switch(*ts) {
|
|
case 0:
|
|
if (ctx->enc_type != 0 && ctx->head == NULL) {
|
|
__Pyx_BufFmt_RaiseExpected(ctx);
|
|
return NULL;
|
|
}
|
|
if (__Pyx_BufFmt_ProcessTypeChunk(ctx) == -1) return NULL;
|
|
if (ctx->head != NULL) {
|
|
__Pyx_BufFmt_RaiseExpected(ctx);
|
|
return NULL;
|
|
}
|
|
return ts;
|
|
case ' ':
|
|
case 10:
|
|
case 13:
|
|
++ts;
|
|
break;
|
|
case '<':
|
|
if (!__Pyx_IsLittleEndian()) {
|
|
PyErr_SetString(PyExc_ValueError, "Little-endian buffer not supported on big-endian compiler");
|
|
return NULL;
|
|
}
|
|
ctx->new_packmode = '=';
|
|
++ts;
|
|
break;
|
|
case '>':
|
|
case '!':
|
|
if (__Pyx_IsLittleEndian()) {
|
|
PyErr_SetString(PyExc_ValueError, "Big-endian buffer not supported on little-endian compiler");
|
|
return NULL;
|
|
}
|
|
ctx->new_packmode = '=';
|
|
++ts;
|
|
break;
|
|
case '=':
|
|
case '@':
|
|
case '^':
|
|
ctx->new_packmode = *ts++;
|
|
break;
|
|
case 'T': /* substruct */
|
|
{
|
|
const char* ts_after_sub;
|
|
size_t i, struct_count = ctx->new_count;
|
|
size_t struct_alignment = ctx->struct_alignment;
|
|
ctx->new_count = 1;
|
|
++ts;
|
|
if (*ts != '{') {
|
|
PyErr_SetString(PyExc_ValueError, "Buffer acquisition: Expected '{' after 'T'");
|
|
return NULL;
|
|
}
|
|
if (__Pyx_BufFmt_ProcessTypeChunk(ctx) == -1) return NULL;
|
|
ctx->enc_type = 0; /* Erase processed last struct element */
|
|
ctx->enc_count = 0;
|
|
ctx->struct_alignment = 0;
|
|
++ts;
|
|
ts_after_sub = ts;
|
|
for (i = 0; i != struct_count; ++i) {
|
|
ts_after_sub = __Pyx_BufFmt_CheckString(ctx, ts);
|
|
if (!ts_after_sub) return NULL;
|
|
}
|
|
ts = ts_after_sub;
|
|
if (struct_alignment) ctx->struct_alignment = struct_alignment;
|
|
}
|
|
break;
|
|
case '}': /* end of substruct; either repeat or move on */
|
|
{
|
|
size_t alignment = ctx->struct_alignment;
|
|
++ts;
|
|
if (__Pyx_BufFmt_ProcessTypeChunk(ctx) == -1) return NULL;
|
|
ctx->enc_type = 0; /* Erase processed last struct element */
|
|
if (alignment && ctx->fmt_offset % alignment) {
|
|
ctx->fmt_offset += alignment - (ctx->fmt_offset % alignment);
|
|
}
|
|
}
|
|
return ts;
|
|
case 'x':
|
|
if (__Pyx_BufFmt_ProcessTypeChunk(ctx) == -1) return NULL;
|
|
ctx->fmt_offset += ctx->new_count;
|
|
ctx->new_count = 1;
|
|
ctx->enc_count = 0;
|
|
ctx->enc_type = 0;
|
|
ctx->enc_packmode = ctx->new_packmode;
|
|
++ts;
|
|
break;
|
|
case 'Z':
|
|
got_Z = 1;
|
|
++ts;
|
|
if (*ts != 'f' && *ts != 'd' && *ts != 'g') {
|
|
__Pyx_BufFmt_RaiseUnexpectedChar('Z');
|
|
return NULL;
|
|
} /* fall through */
|
|
case 'c': case 'b': case 'B': case 'h': case 'H': case 'i': case 'I':
|
|
case 'l': case 'L': case 'q': case 'Q':
|
|
case 'f': case 'd': case 'g':
|
|
case 'O': case 's': case 'p':
|
|
if (ctx->enc_type == *ts && got_Z == ctx->is_complex &&
|
|
ctx->enc_packmode == ctx->new_packmode) {
|
|
ctx->enc_count += ctx->new_count;
|
|
} else {
|
|
if (__Pyx_BufFmt_ProcessTypeChunk(ctx) == -1) return NULL;
|
|
ctx->enc_count = ctx->new_count;
|
|
ctx->enc_packmode = ctx->new_packmode;
|
|
ctx->enc_type = *ts;
|
|
ctx->is_complex = got_Z;
|
|
}
|
|
++ts;
|
|
ctx->new_count = 1;
|
|
got_Z = 0;
|
|
break;
|
|
case ':':
|
|
++ts;
|
|
while(*ts != ':') ++ts;
|
|
++ts;
|
|
break;
|
|
case '(':
|
|
if (!__pyx_buffmt_parse_array(ctx, &ts)) return NULL;
|
|
break;
|
|
default:
|
|
{
|
|
int number = __Pyx_BufFmt_ExpectNumber(&ts);
|
|
if (number == -1) return NULL;
|
|
ctx->new_count = (size_t)number;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
static CYTHON_INLINE void __Pyx_ZeroBuffer(Py_buffer* buf) {
|
|
buf->buf = NULL;
|
|
buf->obj = NULL;
|
|
buf->strides = __Pyx_zeros;
|
|
buf->shape = __Pyx_zeros;
|
|
buf->suboffsets = __Pyx_minusones;
|
|
}
|
|
static CYTHON_INLINE int __Pyx_GetBufferAndValidate(
|
|
Py_buffer* buf, PyObject* obj, __Pyx_TypeInfo* dtype, int flags,
|
|
int nd, int cast, __Pyx_BufFmt_StackElem* stack)
|
|
{
|
|
if (obj == Py_None || obj == NULL) {
|
|
__Pyx_ZeroBuffer(buf);
|
|
return 0;
|
|
}
|
|
buf->buf = NULL;
|
|
if (__Pyx_GetBuffer(obj, buf, flags) == -1) goto fail;
|
|
if (buf->ndim != nd) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Buffer has wrong number of dimensions (expected %d, got %d)",
|
|
nd, buf->ndim);
|
|
goto fail;
|
|
}
|
|
if (!cast) {
|
|
__Pyx_BufFmt_Context ctx;
|
|
__Pyx_BufFmt_Init(&ctx, stack, dtype);
|
|
if (!__Pyx_BufFmt_CheckString(&ctx, buf->format)) goto fail;
|
|
}
|
|
if ((unsigned)buf->itemsize != dtype->size) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Item size of buffer (%"PY_FORMAT_SIZE_T"d byte%s) does not match size of '%s' (%"PY_FORMAT_SIZE_T"d byte%s)",
|
|
buf->itemsize, (buf->itemsize > 1) ? "s" : "",
|
|
dtype->name, (Py_ssize_t)dtype->size, (dtype->size > 1) ? "s" : "");
|
|
goto fail;
|
|
}
|
|
if (buf->suboffsets == NULL) buf->suboffsets = __Pyx_minusones;
|
|
return 0;
|
|
fail:;
|
|
__Pyx_ZeroBuffer(buf);
|
|
return -1;
|
|
}
|
|
static CYTHON_INLINE void __Pyx_SafeReleaseBuffer(Py_buffer* info) {
|
|
if (info->buf == NULL) return;
|
|
if (info->suboffsets == __Pyx_minusones) info->suboffsets = NULL;
|
|
__Pyx_ReleaseBuffer(info);
|
|
}
|
|
|
|
static void __Pyx_RaiseBufferIndexError(int axis) {
|
|
PyErr_Format(PyExc_IndexError,
|
|
"Out of bounds on buffer access (axis %d)", axis);
|
|
}
|
|
|
|
static void __Pyx_RaiseBufferFallbackError(void) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Buffer acquisition failed on assignment; and then reacquiring the old buffer failed too!");
|
|
}
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseNoneNotIterableError(void) {
|
|
PyErr_SetString(PyExc_TypeError, "'NoneType' object is not iterable");
|
|
}
|
|
|
|
static void __Pyx_UnpackTupleError(PyObject *t, Py_ssize_t index) {
|
|
if (t == Py_None) {
|
|
__Pyx_RaiseNoneNotIterableError();
|
|
} else if (PyTuple_GET_SIZE(t) < index) {
|
|
__Pyx_RaiseNeedMoreValuesError(PyTuple_GET_SIZE(t));
|
|
} else {
|
|
__Pyx_RaiseTooManyValuesError(index);
|
|
}
|
|
}
|
|
|
|
#if PY_MAJOR_VERSION < 3
|
|
static int __Pyx_GetBuffer(PyObject *obj, Py_buffer *view, int flags) {
|
|
PyObject *getbuffer_cobj;
|
|
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
if (PyObject_CheckBuffer(obj)) return PyObject_GetBuffer(obj, view, flags);
|
|
#endif
|
|
|
|
if (PyObject_TypeCheck(obj, __pyx_ptype_5numpy_ndarray)) return __pyx_pw_5numpy_7ndarray_1__getbuffer__(obj, view, flags);
|
|
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
if (obj->ob_type->tp_dict &&
|
|
(getbuffer_cobj = PyMapping_GetItemString(obj->ob_type->tp_dict,
|
|
"__pyx_getbuffer"))) {
|
|
getbufferproc func;
|
|
|
|
#if PY_VERSION_HEX >= 0x02070000 && !(PY_MAJOR_VERSION == 3 && PY_MINOR_VERSION == 0)
|
|
func = (getbufferproc) PyCapsule_GetPointer(getbuffer_cobj, "getbuffer(obj, view, flags)");
|
|
#else
|
|
func = (getbufferproc) PyCObject_AsVoidPtr(getbuffer_cobj);
|
|
#endif
|
|
Py_DECREF(getbuffer_cobj);
|
|
if (!func)
|
|
goto fail;
|
|
|
|
return func(obj, view, flags);
|
|
} else {
|
|
PyErr_Clear();
|
|
}
|
|
#endif
|
|
|
|
PyErr_Format(PyExc_TypeError, "'%100s' does not have the buffer interface", Py_TYPE(obj)->tp_name);
|
|
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
fail:
|
|
#endif
|
|
|
|
return -1;
|
|
}
|
|
|
|
static void __Pyx_ReleaseBuffer(Py_buffer *view) {
|
|
PyObject *obj = view->obj;
|
|
PyObject *releasebuffer_cobj;
|
|
|
|
if (!obj) return;
|
|
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
if (PyObject_CheckBuffer(obj)) {
|
|
PyBuffer_Release(view);
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
if (PyObject_TypeCheck(obj, __pyx_ptype_5numpy_ndarray)) { __pyx_pw_5numpy_7ndarray_3__releasebuffer__(obj, view); return; }
|
|
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
if (obj->ob_type->tp_dict &&
|
|
(releasebuffer_cobj = PyMapping_GetItemString(obj->ob_type->tp_dict,
|
|
"__pyx_releasebuffer"))) {
|
|
releasebufferproc func;
|
|
|
|
#if PY_VERSION_HEX >= 0x02070000 && !(PY_MAJOR_VERSION == 3 && PY_MINOR_VERSION == 0)
|
|
func = (releasebufferproc) PyCapsule_GetPointer(releasebuffer_cobj, "releasebuffer(obj, view)");
|
|
#else
|
|
func = (releasebufferproc) PyCObject_AsVoidPtr(releasebuffer_cobj);
|
|
#endif
|
|
|
|
Py_DECREF(releasebuffer_cobj);
|
|
|
|
if (!func)
|
|
goto fail;
|
|
|
|
func(obj, view);
|
|
return;
|
|
} else {
|
|
PyErr_Clear();
|
|
}
|
|
#endif
|
|
|
|
goto nofail;
|
|
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
fail:
|
|
#endif
|
|
PyErr_WriteUnraisable(obj);
|
|
|
|
nofail:
|
|
Py_DECREF(obj);
|
|
view->obj = NULL;
|
|
}
|
|
|
|
#endif /* PY_MAJOR_VERSION < 3 */
|
|
|
|
static PyObject *__Pyx_Import(PyObject *name, PyObject *from_list, long level) {
|
|
PyObject *py_import = 0;
|
|
PyObject *empty_list = 0;
|
|
PyObject *module = 0;
|
|
PyObject *global_dict = 0;
|
|
PyObject *empty_dict = 0;
|
|
PyObject *list;
|
|
py_import = __Pyx_GetAttrString(__pyx_b, "__import__");
|
|
if (!py_import)
|
|
goto bad;
|
|
if (from_list)
|
|
list = from_list;
|
|
else {
|
|
empty_list = PyList_New(0);
|
|
if (!empty_list)
|
|
goto bad;
|
|
list = empty_list;
|
|
}
|
|
global_dict = PyModule_GetDict(__pyx_m);
|
|
if (!global_dict)
|
|
goto bad;
|
|
empty_dict = PyDict_New();
|
|
if (!empty_dict)
|
|
goto bad;
|
|
#if PY_VERSION_HEX >= 0x02050000
|
|
{
|
|
#if PY_MAJOR_VERSION >= 3
|
|
if (level == -1) {
|
|
if (strchr(__Pyx_MODULE_NAME, '.')) {
|
|
/* try package relative import first */
|
|
PyObject *py_level = PyInt_FromLong(1);
|
|
if (!py_level)
|
|
goto bad;
|
|
module = PyObject_CallFunctionObjArgs(py_import,
|
|
name, global_dict, empty_dict, list, py_level, NULL);
|
|
Py_DECREF(py_level);
|
|
if (!module) {
|
|
if (!PyErr_ExceptionMatches(PyExc_ImportError))
|
|
goto bad;
|
|
PyErr_Clear();
|
|
}
|
|
}
|
|
level = 0; /* try absolute import on failure */
|
|
}
|
|
#endif
|
|
if (!module) {
|
|
PyObject *py_level = PyInt_FromLong(level);
|
|
if (!py_level)
|
|
goto bad;
|
|
module = PyObject_CallFunctionObjArgs(py_import,
|
|
name, global_dict, empty_dict, list, py_level, NULL);
|
|
Py_DECREF(py_level);
|
|
}
|
|
}
|
|
#else
|
|
if (level>0) {
|
|
PyErr_SetString(PyExc_RuntimeError, "Relative import is not supported for Python <=2.4.");
|
|
goto bad;
|
|
}
|
|
module = PyObject_CallFunctionObjArgs(py_import,
|
|
name, global_dict, empty_dict, list, NULL);
|
|
#endif
|
|
bad:
|
|
Py_XDECREF(empty_list);
|
|
Py_XDECREF(py_import);
|
|
Py_XDECREF(empty_dict);
|
|
return module;
|
|
}
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseImportError(PyObject *name) {
|
|
#if PY_MAJOR_VERSION < 3
|
|
PyErr_Format(PyExc_ImportError, "cannot import name %.230s",
|
|
PyString_AsString(name));
|
|
#else
|
|
PyErr_Format(PyExc_ImportError, "cannot import name %S", name);
|
|
#endif
|
|
}
|
|
|
|
static CYTHON_INLINE int __Pyx_PyBytes_Equals(PyObject* s1, PyObject* s2, int equals) {
|
|
if (s1 == s2) {
|
|
return (equals == Py_EQ);
|
|
} else if (PyBytes_CheckExact(s1) & PyBytes_CheckExact(s2)) {
|
|
if (PyBytes_GET_SIZE(s1) != PyBytes_GET_SIZE(s2)) {
|
|
return (equals == Py_NE);
|
|
} else if (PyBytes_GET_SIZE(s1) == 1) {
|
|
if (equals == Py_EQ)
|
|
return (PyBytes_AS_STRING(s1)[0] == PyBytes_AS_STRING(s2)[0]);
|
|
else
|
|
return (PyBytes_AS_STRING(s1)[0] != PyBytes_AS_STRING(s2)[0]);
|
|
} else {
|
|
int result = memcmp(PyBytes_AS_STRING(s1), PyBytes_AS_STRING(s2), (size_t)PyBytes_GET_SIZE(s1));
|
|
return (equals == Py_EQ) ? (result == 0) : (result != 0);
|
|
}
|
|
} else if ((s1 == Py_None) & PyBytes_CheckExact(s2)) {
|
|
return (equals == Py_NE);
|
|
} else if ((s2 == Py_None) & PyBytes_CheckExact(s1)) {
|
|
return (equals == Py_NE);
|
|
} else {
|
|
int result;
|
|
PyObject* py_result = PyObject_RichCompare(s1, s2, equals);
|
|
if (!py_result)
|
|
return -1;
|
|
result = __Pyx_PyObject_IsTrue(py_result);
|
|
Py_DECREF(py_result);
|
|
return result;
|
|
}
|
|
}
|
|
|
|
static CYTHON_INLINE int __Pyx_PyUnicode_Equals(PyObject* s1, PyObject* s2, int equals) {
|
|
if (s1 == s2) {
|
|
return (equals == Py_EQ);
|
|
} else if (PyUnicode_CheckExact(s1) & PyUnicode_CheckExact(s2)) {
|
|
#if CYTHON_PEP393_ENABLED
|
|
if ((PyUnicode_READY(s1) < 0) || (PyUnicode_READY(s2) < 0))
|
|
return -1;
|
|
if (PyUnicode_GET_LENGTH(s1) != PyUnicode_GET_LENGTH(s2)) {
|
|
return (equals == Py_NE);
|
|
} else if (PyUnicode_GET_LENGTH(s1) == 1) {
|
|
Py_UCS4 ch1 = PyUnicode_READ_CHAR(s1, 0);
|
|
Py_UCS4 ch2 = PyUnicode_READ_CHAR(s2, 0);
|
|
return (equals == Py_EQ) ? (ch1 == ch2) : (ch1 != ch2);
|
|
#else
|
|
if (PyUnicode_GET_SIZE(s1) != PyUnicode_GET_SIZE(s2)) {
|
|
return (equals == Py_NE);
|
|
} else if (PyUnicode_GET_SIZE(s1) == 1) {
|
|
Py_UNICODE ch1 = PyUnicode_AS_UNICODE(s1)[0];
|
|
Py_UNICODE ch2 = PyUnicode_AS_UNICODE(s2)[0];
|
|
return (equals == Py_EQ) ? (ch1 == ch2) : (ch1 != ch2);
|
|
#endif
|
|
} else {
|
|
int result = PyUnicode_Compare(s1, s2);
|
|
if ((result == -1) && unlikely(PyErr_Occurred()))
|
|
return -1;
|
|
return (equals == Py_EQ) ? (result == 0) : (result != 0);
|
|
}
|
|
} else if ((s1 == Py_None) & PyUnicode_CheckExact(s2)) {
|
|
return (equals == Py_NE);
|
|
} else if ((s2 == Py_None) & PyUnicode_CheckExact(s1)) {
|
|
return (equals == Py_NE);
|
|
} else {
|
|
int result;
|
|
PyObject* py_result = PyObject_RichCompare(s1, s2, equals);
|
|
if (!py_result)
|
|
return -1;
|
|
result = __Pyx_PyObject_IsTrue(py_result);
|
|
Py_DECREF(py_result);
|
|
return result;
|
|
}
|
|
}
|
|
|
|
static CYTHON_INLINE PyObject *__Pyx_PyInt_to_py_npy_int32(npy_int32 val) {
|
|
const npy_int32 neg_one = (npy_int32)-1, const_zero = (npy_int32)0;
|
|
const int is_unsigned = const_zero < neg_one;
|
|
if ((sizeof(npy_int32) == sizeof(char)) ||
|
|
(sizeof(npy_int32) == sizeof(short))) {
|
|
return PyInt_FromLong((long)val);
|
|
} else if ((sizeof(npy_int32) == sizeof(int)) ||
|
|
(sizeof(npy_int32) == sizeof(long))) {
|
|
if (is_unsigned)
|
|
return PyLong_FromUnsignedLong((unsigned long)val);
|
|
else
|
|
return PyInt_FromLong((long)val);
|
|
} else if (sizeof(npy_int32) == sizeof(PY_LONG_LONG)) {
|
|
if (is_unsigned)
|
|
return PyLong_FromUnsignedLongLong((unsigned PY_LONG_LONG)val);
|
|
else
|
|
return PyLong_FromLongLong((PY_LONG_LONG)val);
|
|
} else {
|
|
int one = 1; int little = (int)*(unsigned char *)&one;
|
|
unsigned char *bytes = (unsigned char *)&val;
|
|
return _PyLong_FromByteArray(bytes, sizeof(npy_int32),
|
|
little, !is_unsigned);
|
|
}
|
|
}
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#ifdef __cplusplus
|
|
static CYTHON_INLINE __pyx_t_float_complex __pyx_t_float_complex_from_parts(float x, float y) {
|
|
return ::std::complex< float >(x, y);
|
|
}
|
|
#else
|
|
static CYTHON_INLINE __pyx_t_float_complex __pyx_t_float_complex_from_parts(float x, float y) {
|
|
return x + y*(__pyx_t_float_complex)_Complex_I;
|
|
}
|
|
#endif
|
|
#else
|
|
static CYTHON_INLINE __pyx_t_float_complex __pyx_t_float_complex_from_parts(float x, float y) {
|
|
__pyx_t_float_complex z;
|
|
z.real = x;
|
|
z.imag = y;
|
|
return z;
|
|
}
|
|
#endif
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#else
|
|
static CYTHON_INLINE int __Pyx_c_eqf(__pyx_t_float_complex a, __pyx_t_float_complex b) {
|
|
return (a.real == b.real) && (a.imag == b.imag);
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_sumf(__pyx_t_float_complex a, __pyx_t_float_complex b) {
|
|
__pyx_t_float_complex z;
|
|
z.real = a.real + b.real;
|
|
z.imag = a.imag + b.imag;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_difff(__pyx_t_float_complex a, __pyx_t_float_complex b) {
|
|
__pyx_t_float_complex z;
|
|
z.real = a.real - b.real;
|
|
z.imag = a.imag - b.imag;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_prodf(__pyx_t_float_complex a, __pyx_t_float_complex b) {
|
|
__pyx_t_float_complex z;
|
|
z.real = a.real * b.real - a.imag * b.imag;
|
|
z.imag = a.real * b.imag + a.imag * b.real;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_quotf(__pyx_t_float_complex a, __pyx_t_float_complex b) {
|
|
__pyx_t_float_complex z;
|
|
float denom = b.real * b.real + b.imag * b.imag;
|
|
z.real = (a.real * b.real + a.imag * b.imag) / denom;
|
|
z.imag = (a.imag * b.real - a.real * b.imag) / denom;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_negf(__pyx_t_float_complex a) {
|
|
__pyx_t_float_complex z;
|
|
z.real = -a.real;
|
|
z.imag = -a.imag;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE int __Pyx_c_is_zerof(__pyx_t_float_complex a) {
|
|
return (a.real == 0) && (a.imag == 0);
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_conjf(__pyx_t_float_complex a) {
|
|
__pyx_t_float_complex z;
|
|
z.real = a.real;
|
|
z.imag = -a.imag;
|
|
return z;
|
|
}
|
|
#if 1
|
|
static CYTHON_INLINE float __Pyx_c_absf(__pyx_t_float_complex z) {
|
|
#if !defined(HAVE_HYPOT) || defined(_MSC_VER)
|
|
return sqrtf(z.real*z.real + z.imag*z.imag);
|
|
#else
|
|
return hypotf(z.real, z.imag);
|
|
#endif
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_powf(__pyx_t_float_complex a, __pyx_t_float_complex b) {
|
|
__pyx_t_float_complex z;
|
|
float r, lnr, theta, z_r, z_theta;
|
|
if (b.imag == 0 && b.real == (int)b.real) {
|
|
if (b.real < 0) {
|
|
float denom = a.real * a.real + a.imag * a.imag;
|
|
a.real = a.real / denom;
|
|
a.imag = -a.imag / denom;
|
|
b.real = -b.real;
|
|
}
|
|
switch ((int)b.real) {
|
|
case 0:
|
|
z.real = 1;
|
|
z.imag = 0;
|
|
return z;
|
|
case 1:
|
|
return a;
|
|
case 2:
|
|
z = __Pyx_c_prodf(a, a);
|
|
return __Pyx_c_prodf(a, a);
|
|
case 3:
|
|
z = __Pyx_c_prodf(a, a);
|
|
return __Pyx_c_prodf(z, a);
|
|
case 4:
|
|
z = __Pyx_c_prodf(a, a);
|
|
return __Pyx_c_prodf(z, z);
|
|
}
|
|
}
|
|
if (a.imag == 0) {
|
|
if (a.real == 0) {
|
|
return a;
|
|
}
|
|
r = a.real;
|
|
theta = 0;
|
|
} else {
|
|
r = __Pyx_c_absf(a);
|
|
theta = atan2f(a.imag, a.real);
|
|
}
|
|
lnr = logf(r);
|
|
z_r = expf(lnr * b.real - theta * b.imag);
|
|
z_theta = theta * b.real + lnr * b.imag;
|
|
z.real = z_r * cosf(z_theta);
|
|
z.imag = z_r * sinf(z_theta);
|
|
return z;
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#ifdef __cplusplus
|
|
static CYTHON_INLINE __pyx_t_double_complex __pyx_t_double_complex_from_parts(double x, double y) {
|
|
return ::std::complex< double >(x, y);
|
|
}
|
|
#else
|
|
static CYTHON_INLINE __pyx_t_double_complex __pyx_t_double_complex_from_parts(double x, double y) {
|
|
return x + y*(__pyx_t_double_complex)_Complex_I;
|
|
}
|
|
#endif
|
|
#else
|
|
static CYTHON_INLINE __pyx_t_double_complex __pyx_t_double_complex_from_parts(double x, double y) {
|
|
__pyx_t_double_complex z;
|
|
z.real = x;
|
|
z.imag = y;
|
|
return z;
|
|
}
|
|
#endif
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#else
|
|
static CYTHON_INLINE int __Pyx_c_eq(__pyx_t_double_complex a, __pyx_t_double_complex b) {
|
|
return (a.real == b.real) && (a.imag == b.imag);
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_sum(__pyx_t_double_complex a, __pyx_t_double_complex b) {
|
|
__pyx_t_double_complex z;
|
|
z.real = a.real + b.real;
|
|
z.imag = a.imag + b.imag;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_diff(__pyx_t_double_complex a, __pyx_t_double_complex b) {
|
|
__pyx_t_double_complex z;
|
|
z.real = a.real - b.real;
|
|
z.imag = a.imag - b.imag;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_prod(__pyx_t_double_complex a, __pyx_t_double_complex b) {
|
|
__pyx_t_double_complex z;
|
|
z.real = a.real * b.real - a.imag * b.imag;
|
|
z.imag = a.real * b.imag + a.imag * b.real;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_quot(__pyx_t_double_complex a, __pyx_t_double_complex b) {
|
|
__pyx_t_double_complex z;
|
|
double denom = b.real * b.real + b.imag * b.imag;
|
|
z.real = (a.real * b.real + a.imag * b.imag) / denom;
|
|
z.imag = (a.imag * b.real - a.real * b.imag) / denom;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_neg(__pyx_t_double_complex a) {
|
|
__pyx_t_double_complex z;
|
|
z.real = -a.real;
|
|
z.imag = -a.imag;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE int __Pyx_c_is_zero(__pyx_t_double_complex a) {
|
|
return (a.real == 0) && (a.imag == 0);
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_conj(__pyx_t_double_complex a) {
|
|
__pyx_t_double_complex z;
|
|
z.real = a.real;
|
|
z.imag = -a.imag;
|
|
return z;
|
|
}
|
|
#if 1
|
|
static CYTHON_INLINE double __Pyx_c_abs(__pyx_t_double_complex z) {
|
|
#if !defined(HAVE_HYPOT) || defined(_MSC_VER)
|
|
return sqrt(z.real*z.real + z.imag*z.imag);
|
|
#else
|
|
return hypot(z.real, z.imag);
|
|
#endif
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_pow(__pyx_t_double_complex a, __pyx_t_double_complex b) {
|
|
__pyx_t_double_complex z;
|
|
double r, lnr, theta, z_r, z_theta;
|
|
if (b.imag == 0 && b.real == (int)b.real) {
|
|
if (b.real < 0) {
|
|
double denom = a.real * a.real + a.imag * a.imag;
|
|
a.real = a.real / denom;
|
|
a.imag = -a.imag / denom;
|
|
b.real = -b.real;
|
|
}
|
|
switch ((int)b.real) {
|
|
case 0:
|
|
z.real = 1;
|
|
z.imag = 0;
|
|
return z;
|
|
case 1:
|
|
return a;
|
|
case 2:
|
|
z = __Pyx_c_prod(a, a);
|
|
return __Pyx_c_prod(a, a);
|
|
case 3:
|
|
z = __Pyx_c_prod(a, a);
|
|
return __Pyx_c_prod(z, a);
|
|
case 4:
|
|
z = __Pyx_c_prod(a, a);
|
|
return __Pyx_c_prod(z, z);
|
|
}
|
|
}
|
|
if (a.imag == 0) {
|
|
if (a.real == 0) {
|
|
return a;
|
|
}
|
|
r = a.real;
|
|
theta = 0;
|
|
} else {
|
|
r = __Pyx_c_abs(a);
|
|
theta = atan2(a.imag, a.real);
|
|
}
|
|
lnr = log(r);
|
|
z_r = exp(lnr * b.real - theta * b.imag);
|
|
z_theta = theta * b.real + lnr * b.imag;
|
|
z.real = z_r * cos(z_theta);
|
|
z.imag = z_r * sin(z_theta);
|
|
return z;
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
static CYTHON_INLINE unsigned char __Pyx_PyInt_AsUnsignedChar(PyObject* x) {
|
|
const unsigned char neg_one = (unsigned char)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (sizeof(unsigned char) < sizeof(long)) {
|
|
long val = __Pyx_PyInt_AsLong(x);
|
|
if (unlikely(val != (long)(unsigned char)val)) {
|
|
if (!unlikely(val == -1 && PyErr_Occurred())) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
(is_unsigned && unlikely(val < 0)) ?
|
|
"can't convert negative value to unsigned char" :
|
|
"value too large to convert to unsigned char");
|
|
}
|
|
return (unsigned char)-1;
|
|
}
|
|
return (unsigned char)val;
|
|
}
|
|
return (unsigned char)__Pyx_PyInt_AsUnsignedLong(x);
|
|
}
|
|
|
|
static CYTHON_INLINE unsigned short __Pyx_PyInt_AsUnsignedShort(PyObject* x) {
|
|
const unsigned short neg_one = (unsigned short)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (sizeof(unsigned short) < sizeof(long)) {
|
|
long val = __Pyx_PyInt_AsLong(x);
|
|
if (unlikely(val != (long)(unsigned short)val)) {
|
|
if (!unlikely(val == -1 && PyErr_Occurred())) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
(is_unsigned && unlikely(val < 0)) ?
|
|
"can't convert negative value to unsigned short" :
|
|
"value too large to convert to unsigned short");
|
|
}
|
|
return (unsigned short)-1;
|
|
}
|
|
return (unsigned short)val;
|
|
}
|
|
return (unsigned short)__Pyx_PyInt_AsUnsignedLong(x);
|
|
}
|
|
|
|
static CYTHON_INLINE unsigned int __Pyx_PyInt_AsUnsignedInt(PyObject* x) {
|
|
const unsigned int neg_one = (unsigned int)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (sizeof(unsigned int) < sizeof(long)) {
|
|
long val = __Pyx_PyInt_AsLong(x);
|
|
if (unlikely(val != (long)(unsigned int)val)) {
|
|
if (!unlikely(val == -1 && PyErr_Occurred())) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
(is_unsigned && unlikely(val < 0)) ?
|
|
"can't convert negative value to unsigned int" :
|
|
"value too large to convert to unsigned int");
|
|
}
|
|
return (unsigned int)-1;
|
|
}
|
|
return (unsigned int)val;
|
|
}
|
|
return (unsigned int)__Pyx_PyInt_AsUnsignedLong(x);
|
|
}
|
|
|
|
static CYTHON_INLINE char __Pyx_PyInt_AsChar(PyObject* x) {
|
|
const char neg_one = (char)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (sizeof(char) < sizeof(long)) {
|
|
long val = __Pyx_PyInt_AsLong(x);
|
|
if (unlikely(val != (long)(char)val)) {
|
|
if (!unlikely(val == -1 && PyErr_Occurred())) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
(is_unsigned && unlikely(val < 0)) ?
|
|
"can't convert negative value to char" :
|
|
"value too large to convert to char");
|
|
}
|
|
return (char)-1;
|
|
}
|
|
return (char)val;
|
|
}
|
|
return (char)__Pyx_PyInt_AsLong(x);
|
|
}
|
|
|
|
static CYTHON_INLINE short __Pyx_PyInt_AsShort(PyObject* x) {
|
|
const short neg_one = (short)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (sizeof(short) < sizeof(long)) {
|
|
long val = __Pyx_PyInt_AsLong(x);
|
|
if (unlikely(val != (long)(short)val)) {
|
|
if (!unlikely(val == -1 && PyErr_Occurred())) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
(is_unsigned && unlikely(val < 0)) ?
|
|
"can't convert negative value to short" :
|
|
"value too large to convert to short");
|
|
}
|
|
return (short)-1;
|
|
}
|
|
return (short)val;
|
|
}
|
|
return (short)__Pyx_PyInt_AsLong(x);
|
|
}
|
|
|
|
static CYTHON_INLINE int __Pyx_PyInt_AsInt(PyObject* x) {
|
|
const int neg_one = (int)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (sizeof(int) < sizeof(long)) {
|
|
long val = __Pyx_PyInt_AsLong(x);
|
|
if (unlikely(val != (long)(int)val)) {
|
|
if (!unlikely(val == -1 && PyErr_Occurred())) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
(is_unsigned && unlikely(val < 0)) ?
|
|
"can't convert negative value to int" :
|
|
"value too large to convert to int");
|
|
}
|
|
return (int)-1;
|
|
}
|
|
return (int)val;
|
|
}
|
|
return (int)__Pyx_PyInt_AsLong(x);
|
|
}
|
|
|
|
static CYTHON_INLINE signed char __Pyx_PyInt_AsSignedChar(PyObject* x) {
|
|
const signed char neg_one = (signed char)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (sizeof(signed char) < sizeof(long)) {
|
|
long val = __Pyx_PyInt_AsLong(x);
|
|
if (unlikely(val != (long)(signed char)val)) {
|
|
if (!unlikely(val == -1 && PyErr_Occurred())) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
(is_unsigned && unlikely(val < 0)) ?
|
|
"can't convert negative value to signed char" :
|
|
"value too large to convert to signed char");
|
|
}
|
|
return (signed char)-1;
|
|
}
|
|
return (signed char)val;
|
|
}
|
|
return (signed char)__Pyx_PyInt_AsSignedLong(x);
|
|
}
|
|
|
|
static CYTHON_INLINE signed short __Pyx_PyInt_AsSignedShort(PyObject* x) {
|
|
const signed short neg_one = (signed short)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (sizeof(signed short) < sizeof(long)) {
|
|
long val = __Pyx_PyInt_AsLong(x);
|
|
if (unlikely(val != (long)(signed short)val)) {
|
|
if (!unlikely(val == -1 && PyErr_Occurred())) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
(is_unsigned && unlikely(val < 0)) ?
|
|
"can't convert negative value to signed short" :
|
|
"value too large to convert to signed short");
|
|
}
|
|
return (signed short)-1;
|
|
}
|
|
return (signed short)val;
|
|
}
|
|
return (signed short)__Pyx_PyInt_AsSignedLong(x);
|
|
}
|
|
|
|
static CYTHON_INLINE signed int __Pyx_PyInt_AsSignedInt(PyObject* x) {
|
|
const signed int neg_one = (signed int)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (sizeof(signed int) < sizeof(long)) {
|
|
long val = __Pyx_PyInt_AsLong(x);
|
|
if (unlikely(val != (long)(signed int)val)) {
|
|
if (!unlikely(val == -1 && PyErr_Occurred())) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
(is_unsigned && unlikely(val < 0)) ?
|
|
"can't convert negative value to signed int" :
|
|
"value too large to convert to signed int");
|
|
}
|
|
return (signed int)-1;
|
|
}
|
|
return (signed int)val;
|
|
}
|
|
return (signed int)__Pyx_PyInt_AsSignedLong(x);
|
|
}
|
|
|
|
static CYTHON_INLINE int __Pyx_PyInt_AsLongDouble(PyObject* x) {
|
|
const int neg_one = (int)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (sizeof(int) < sizeof(long)) {
|
|
long val = __Pyx_PyInt_AsLong(x);
|
|
if (unlikely(val != (long)(int)val)) {
|
|
if (!unlikely(val == -1 && PyErr_Occurred())) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
(is_unsigned && unlikely(val < 0)) ?
|
|
"can't convert negative value to int" :
|
|
"value too large to convert to int");
|
|
}
|
|
return (int)-1;
|
|
}
|
|
return (int)val;
|
|
}
|
|
return (int)__Pyx_PyInt_AsLong(x);
|
|
}
|
|
|
|
static CYTHON_INLINE unsigned long __Pyx_PyInt_AsUnsignedLong(PyObject* x) {
|
|
const unsigned long neg_one = (unsigned long)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
#if PY_VERSION_HEX < 0x03000000
|
|
if (likely(PyInt_Check(x))) {
|
|
long val = PyInt_AS_LONG(x);
|
|
if (is_unsigned && unlikely(val < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to unsigned long");
|
|
return (unsigned long)-1;
|
|
}
|
|
return (unsigned long)val;
|
|
} else
|
|
#endif
|
|
if (likely(PyLong_Check(x))) {
|
|
if (is_unsigned) {
|
|
if (unlikely(Py_SIZE(x) < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to unsigned long");
|
|
return (unsigned long)-1;
|
|
}
|
|
return (unsigned long)PyLong_AsUnsignedLong(x);
|
|
} else {
|
|
return (unsigned long)PyLong_AsLong(x);
|
|
}
|
|
} else {
|
|
unsigned long val;
|
|
PyObject *tmp = __Pyx_PyNumber_Int(x);
|
|
if (!tmp) return (unsigned long)-1;
|
|
val = __Pyx_PyInt_AsUnsignedLong(tmp);
|
|
Py_DECREF(tmp);
|
|
return val;
|
|
}
|
|
}
|
|
|
|
static CYTHON_INLINE unsigned PY_LONG_LONG __Pyx_PyInt_AsUnsignedLongLong(PyObject* x) {
|
|
const unsigned PY_LONG_LONG neg_one = (unsigned PY_LONG_LONG)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
#if PY_VERSION_HEX < 0x03000000
|
|
if (likely(PyInt_Check(x))) {
|
|
long val = PyInt_AS_LONG(x);
|
|
if (is_unsigned && unlikely(val < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to unsigned PY_LONG_LONG");
|
|
return (unsigned PY_LONG_LONG)-1;
|
|
}
|
|
return (unsigned PY_LONG_LONG)val;
|
|
} else
|
|
#endif
|
|
if (likely(PyLong_Check(x))) {
|
|
if (is_unsigned) {
|
|
if (unlikely(Py_SIZE(x) < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to unsigned PY_LONG_LONG");
|
|
return (unsigned PY_LONG_LONG)-1;
|
|
}
|
|
return (unsigned PY_LONG_LONG)PyLong_AsUnsignedLongLong(x);
|
|
} else {
|
|
return (unsigned PY_LONG_LONG)PyLong_AsLongLong(x);
|
|
}
|
|
} else {
|
|
unsigned PY_LONG_LONG val;
|
|
PyObject *tmp = __Pyx_PyNumber_Int(x);
|
|
if (!tmp) return (unsigned PY_LONG_LONG)-1;
|
|
val = __Pyx_PyInt_AsUnsignedLongLong(tmp);
|
|
Py_DECREF(tmp);
|
|
return val;
|
|
}
|
|
}
|
|
|
|
static CYTHON_INLINE long __Pyx_PyInt_AsLong(PyObject* x) {
|
|
const long neg_one = (long)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
#if PY_VERSION_HEX < 0x03000000
|
|
if (likely(PyInt_Check(x))) {
|
|
long val = PyInt_AS_LONG(x);
|
|
if (is_unsigned && unlikely(val < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to long");
|
|
return (long)-1;
|
|
}
|
|
return (long)val;
|
|
} else
|
|
#endif
|
|
if (likely(PyLong_Check(x))) {
|
|
if (is_unsigned) {
|
|
if (unlikely(Py_SIZE(x) < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to long");
|
|
return (long)-1;
|
|
}
|
|
return (long)PyLong_AsUnsignedLong(x);
|
|
} else {
|
|
return (long)PyLong_AsLong(x);
|
|
}
|
|
} else {
|
|
long val;
|
|
PyObject *tmp = __Pyx_PyNumber_Int(x);
|
|
if (!tmp) return (long)-1;
|
|
val = __Pyx_PyInt_AsLong(tmp);
|
|
Py_DECREF(tmp);
|
|
return val;
|
|
}
|
|
}
|
|
|
|
static CYTHON_INLINE PY_LONG_LONG __Pyx_PyInt_AsLongLong(PyObject* x) {
|
|
const PY_LONG_LONG neg_one = (PY_LONG_LONG)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
#if PY_VERSION_HEX < 0x03000000
|
|
if (likely(PyInt_Check(x))) {
|
|
long val = PyInt_AS_LONG(x);
|
|
if (is_unsigned && unlikely(val < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to PY_LONG_LONG");
|
|
return (PY_LONG_LONG)-1;
|
|
}
|
|
return (PY_LONG_LONG)val;
|
|
} else
|
|
#endif
|
|
if (likely(PyLong_Check(x))) {
|
|
if (is_unsigned) {
|
|
if (unlikely(Py_SIZE(x) < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to PY_LONG_LONG");
|
|
return (PY_LONG_LONG)-1;
|
|
}
|
|
return (PY_LONG_LONG)PyLong_AsUnsignedLongLong(x);
|
|
} else {
|
|
return (PY_LONG_LONG)PyLong_AsLongLong(x);
|
|
}
|
|
} else {
|
|
PY_LONG_LONG val;
|
|
PyObject *tmp = __Pyx_PyNumber_Int(x);
|
|
if (!tmp) return (PY_LONG_LONG)-1;
|
|
val = __Pyx_PyInt_AsLongLong(tmp);
|
|
Py_DECREF(tmp);
|
|
return val;
|
|
}
|
|
}
|
|
|
|
static CYTHON_INLINE signed long __Pyx_PyInt_AsSignedLong(PyObject* x) {
|
|
const signed long neg_one = (signed long)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
#if PY_VERSION_HEX < 0x03000000
|
|
if (likely(PyInt_Check(x))) {
|
|
long val = PyInt_AS_LONG(x);
|
|
if (is_unsigned && unlikely(val < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to signed long");
|
|
return (signed long)-1;
|
|
}
|
|
return (signed long)val;
|
|
} else
|
|
#endif
|
|
if (likely(PyLong_Check(x))) {
|
|
if (is_unsigned) {
|
|
if (unlikely(Py_SIZE(x) < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to signed long");
|
|
return (signed long)-1;
|
|
}
|
|
return (signed long)PyLong_AsUnsignedLong(x);
|
|
} else {
|
|
return (signed long)PyLong_AsLong(x);
|
|
}
|
|
} else {
|
|
signed long val;
|
|
PyObject *tmp = __Pyx_PyNumber_Int(x);
|
|
if (!tmp) return (signed long)-1;
|
|
val = __Pyx_PyInt_AsSignedLong(tmp);
|
|
Py_DECREF(tmp);
|
|
return val;
|
|
}
|
|
}
|
|
|
|
static CYTHON_INLINE signed PY_LONG_LONG __Pyx_PyInt_AsSignedLongLong(PyObject* x) {
|
|
const signed PY_LONG_LONG neg_one = (signed PY_LONG_LONG)-1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
#if PY_VERSION_HEX < 0x03000000
|
|
if (likely(PyInt_Check(x))) {
|
|
long val = PyInt_AS_LONG(x);
|
|
if (is_unsigned && unlikely(val < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to signed PY_LONG_LONG");
|
|
return (signed PY_LONG_LONG)-1;
|
|
}
|
|
return (signed PY_LONG_LONG)val;
|
|
} else
|
|
#endif
|
|
if (likely(PyLong_Check(x))) {
|
|
if (is_unsigned) {
|
|
if (unlikely(Py_SIZE(x) < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to signed PY_LONG_LONG");
|
|
return (signed PY_LONG_LONG)-1;
|
|
}
|
|
return (signed PY_LONG_LONG)PyLong_AsUnsignedLongLong(x);
|
|
} else {
|
|
return (signed PY_LONG_LONG)PyLong_AsLongLong(x);
|
|
}
|
|
} else {
|
|
signed PY_LONG_LONG val;
|
|
PyObject *tmp = __Pyx_PyNumber_Int(x);
|
|
if (!tmp) return (signed PY_LONG_LONG)-1;
|
|
val = __Pyx_PyInt_AsSignedLongLong(tmp);
|
|
Py_DECREF(tmp);
|
|
return val;
|
|
}
|
|
}
|
|
|
|
static void __Pyx_WriteUnraisable(const char *name, int clineno,
|
|
int lineno, const char *filename) {
|
|
PyObject *old_exc, *old_val, *old_tb;
|
|
PyObject *ctx;
|
|
__Pyx_ErrFetch(&old_exc, &old_val, &old_tb);
|
|
#if PY_MAJOR_VERSION < 3
|
|
ctx = PyString_FromString(name);
|
|
#else
|
|
ctx = PyUnicode_FromString(name);
|
|
#endif
|
|
__Pyx_ErrRestore(old_exc, old_val, old_tb);
|
|
if (!ctx) {
|
|
PyErr_WriteUnraisable(Py_None);
|
|
} else {
|
|
PyErr_WriteUnraisable(ctx);
|
|
Py_DECREF(ctx);
|
|
}
|
|
}
|
|
|
|
static int __Pyx_check_binary_version(void) {
|
|
char ctversion[4], rtversion[4];
|
|
PyOS_snprintf(ctversion, 4, "%d.%d", PY_MAJOR_VERSION, PY_MINOR_VERSION);
|
|
PyOS_snprintf(rtversion, 4, "%s", Py_GetVersion());
|
|
if (ctversion[0] != rtversion[0] || ctversion[2] != rtversion[2]) {
|
|
char message[200];
|
|
PyOS_snprintf(message, sizeof(message),
|
|
"compiletime version %s of module '%.100s' "
|
|
"does not match runtime version %s",
|
|
ctversion, __Pyx_MODULE_NAME, rtversion);
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
return PyErr_Warn(NULL, message);
|
|
#else
|
|
return PyErr_WarnEx(NULL, message, 1);
|
|
#endif
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
#ifndef __PYX_HAVE_RT_ImportType
|
|
#define __PYX_HAVE_RT_ImportType
|
|
static PyTypeObject *__Pyx_ImportType(const char *module_name, const char *class_name,
|
|
size_t size, int strict)
|
|
{
|
|
PyObject *py_module = 0;
|
|
PyObject *result = 0;
|
|
PyObject *py_name = 0;
|
|
char warning[200];
|
|
py_module = __Pyx_ImportModule(module_name);
|
|
if (!py_module)
|
|
goto bad;
|
|
py_name = __Pyx_PyIdentifier_FromString(class_name);
|
|
if (!py_name)
|
|
goto bad;
|
|
result = PyObject_GetAttr(py_module, py_name);
|
|
Py_DECREF(py_name);
|
|
py_name = 0;
|
|
Py_DECREF(py_module);
|
|
py_module = 0;
|
|
if (!result)
|
|
goto bad;
|
|
if (!PyType_Check(result)) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"%s.%s is not a type object",
|
|
module_name, class_name);
|
|
goto bad;
|
|
}
|
|
if (!strict && (size_t)((PyTypeObject *)result)->tp_basicsize > size) {
|
|
PyOS_snprintf(warning, sizeof(warning),
|
|
"%s.%s size changed, may indicate binary incompatibility",
|
|
module_name, class_name);
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
if (PyErr_Warn(NULL, warning) < 0) goto bad;
|
|
#else
|
|
if (PyErr_WarnEx(NULL, warning, 0) < 0) goto bad;
|
|
#endif
|
|
}
|
|
else if ((size_t)((PyTypeObject *)result)->tp_basicsize != size) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"%s.%s has the wrong size, try recompiling",
|
|
module_name, class_name);
|
|
goto bad;
|
|
}
|
|
return (PyTypeObject *)result;
|
|
bad:
|
|
Py_XDECREF(py_module);
|
|
Py_XDECREF(result);
|
|
return NULL;
|
|
}
|
|
#endif
|
|
|
|
#ifndef __PYX_HAVE_RT_ImportModule
|
|
#define __PYX_HAVE_RT_ImportModule
|
|
static PyObject *__Pyx_ImportModule(const char *name) {
|
|
PyObject *py_name = 0;
|
|
PyObject *py_module = 0;
|
|
py_name = __Pyx_PyIdentifier_FromString(name);
|
|
if (!py_name)
|
|
goto bad;
|
|
py_module = PyImport_Import(py_name);
|
|
Py_DECREF(py_name);
|
|
return py_module;
|
|
bad:
|
|
Py_XDECREF(py_name);
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
static int __pyx_bisect_code_objects(__Pyx_CodeObjectCacheEntry* entries, int count, int code_line) {
|
|
int start = 0, mid = 0, end = count - 1;
|
|
if (end >= 0 && code_line > entries[end].code_line) {
|
|
return count;
|
|
}
|
|
while (start < end) {
|
|
mid = (start + end) / 2;
|
|
if (code_line < entries[mid].code_line) {
|
|
end = mid;
|
|
} else if (code_line > entries[mid].code_line) {
|
|
start = mid + 1;
|
|
} else {
|
|
return mid;
|
|
}
|
|
}
|
|
if (code_line <= entries[mid].code_line) {
|
|
return mid;
|
|
} else {
|
|
return mid + 1;
|
|
}
|
|
}
|
|
static PyCodeObject *__pyx_find_code_object(int code_line) {
|
|
PyCodeObject* code_object;
|
|
int pos;
|
|
if (unlikely(!code_line) || unlikely(!__pyx_code_cache.entries)) {
|
|
return NULL;
|
|
}
|
|
pos = __pyx_bisect_code_objects(__pyx_code_cache.entries, __pyx_code_cache.count, code_line);
|
|
if (unlikely(pos >= __pyx_code_cache.count) || unlikely(__pyx_code_cache.entries[pos].code_line != code_line)) {
|
|
return NULL;
|
|
}
|
|
code_object = __pyx_code_cache.entries[pos].code_object;
|
|
Py_INCREF(code_object);
|
|
return code_object;
|
|
}
|
|
static void __pyx_insert_code_object(int code_line, PyCodeObject* code_object) {
|
|
int pos, i;
|
|
__Pyx_CodeObjectCacheEntry* entries = __pyx_code_cache.entries;
|
|
if (unlikely(!code_line)) {
|
|
return;
|
|
}
|
|
if (unlikely(!entries)) {
|
|
entries = (__Pyx_CodeObjectCacheEntry*)PyMem_Malloc(64*sizeof(__Pyx_CodeObjectCacheEntry));
|
|
if (likely(entries)) {
|
|
__pyx_code_cache.entries = entries;
|
|
__pyx_code_cache.max_count = 64;
|
|
__pyx_code_cache.count = 1;
|
|
entries[0].code_line = code_line;
|
|
entries[0].code_object = code_object;
|
|
Py_INCREF(code_object);
|
|
}
|
|
return;
|
|
}
|
|
pos = __pyx_bisect_code_objects(__pyx_code_cache.entries, __pyx_code_cache.count, code_line);
|
|
if ((pos < __pyx_code_cache.count) && unlikely(__pyx_code_cache.entries[pos].code_line == code_line)) {
|
|
PyCodeObject* tmp = entries[pos].code_object;
|
|
entries[pos].code_object = code_object;
|
|
Py_DECREF(tmp);
|
|
return;
|
|
}
|
|
if (__pyx_code_cache.count == __pyx_code_cache.max_count) {
|
|
int new_max = __pyx_code_cache.max_count + 64;
|
|
entries = (__Pyx_CodeObjectCacheEntry*)PyMem_Realloc(
|
|
__pyx_code_cache.entries, new_max*sizeof(__Pyx_CodeObjectCacheEntry));
|
|
if (unlikely(!entries)) {
|
|
return;
|
|
}
|
|
__pyx_code_cache.entries = entries;
|
|
__pyx_code_cache.max_count = new_max;
|
|
}
|
|
for (i=__pyx_code_cache.count; i>pos; i--) {
|
|
entries[i] = entries[i-1];
|
|
}
|
|
entries[pos].code_line = code_line;
|
|
entries[pos].code_object = code_object;
|
|
__pyx_code_cache.count++;
|
|
Py_INCREF(code_object);
|
|
}
|
|
|
|
#include "compile.h"
|
|
#include "frameobject.h"
|
|
#include "traceback.h"
|
|
static PyCodeObject* __Pyx_CreateCodeObjectForTraceback(
|
|
const char *funcname, int c_line,
|
|
int py_line, const char *filename) {
|
|
PyCodeObject *py_code = 0;
|
|
PyObject *py_srcfile = 0;
|
|
PyObject *py_funcname = 0;
|
|
#if PY_MAJOR_VERSION < 3
|
|
py_srcfile = PyString_FromString(filename);
|
|
#else
|
|
py_srcfile = PyUnicode_FromString(filename);
|
|
#endif
|
|
if (!py_srcfile) goto bad;
|
|
if (c_line) {
|
|
#if PY_MAJOR_VERSION < 3
|
|
py_funcname = PyString_FromFormat( "%s (%s:%d)", funcname, __pyx_cfilenm, c_line);
|
|
#else
|
|
py_funcname = PyUnicode_FromFormat( "%s (%s:%d)", funcname, __pyx_cfilenm, c_line);
|
|
#endif
|
|
}
|
|
else {
|
|
#if PY_MAJOR_VERSION < 3
|
|
py_funcname = PyString_FromString(funcname);
|
|
#else
|
|
py_funcname = PyUnicode_FromString(funcname);
|
|
#endif
|
|
}
|
|
if (!py_funcname) goto bad;
|
|
py_code = __Pyx_PyCode_New(
|
|
0, /*int argcount,*/
|
|
0, /*int kwonlyargcount,*/
|
|
0, /*int nlocals,*/
|
|
0, /*int stacksize,*/
|
|
0, /*int flags,*/
|
|
__pyx_empty_bytes, /*PyObject *code,*/
|
|
__pyx_empty_tuple, /*PyObject *consts,*/
|
|
__pyx_empty_tuple, /*PyObject *names,*/
|
|
__pyx_empty_tuple, /*PyObject *varnames,*/
|
|
__pyx_empty_tuple, /*PyObject *freevars,*/
|
|
__pyx_empty_tuple, /*PyObject *cellvars,*/
|
|
py_srcfile, /*PyObject *filename,*/
|
|
py_funcname, /*PyObject *name,*/
|
|
py_line, /*int firstlineno,*/
|
|
__pyx_empty_bytes /*PyObject *lnotab*/
|
|
);
|
|
Py_DECREF(py_srcfile);
|
|
Py_DECREF(py_funcname);
|
|
return py_code;
|
|
bad:
|
|
Py_XDECREF(py_srcfile);
|
|
Py_XDECREF(py_funcname);
|
|
return NULL;
|
|
}
|
|
static void __Pyx_AddTraceback(const char *funcname, int c_line,
|
|
int py_line, const char *filename) {
|
|
PyCodeObject *py_code = 0;
|
|
PyObject *py_globals = 0;
|
|
PyFrameObject *py_frame = 0;
|
|
py_code = __pyx_find_code_object(c_line ? c_line : py_line);
|
|
if (!py_code) {
|
|
py_code = __Pyx_CreateCodeObjectForTraceback(
|
|
funcname, c_line, py_line, filename);
|
|
if (!py_code) goto bad;
|
|
__pyx_insert_code_object(c_line ? c_line : py_line, py_code);
|
|
}
|
|
py_globals = PyModule_GetDict(__pyx_m);
|
|
if (!py_globals) goto bad;
|
|
py_frame = PyFrame_New(
|
|
PyThreadState_GET(), /*PyThreadState *tstate,*/
|
|
py_code, /*PyCodeObject *code,*/
|
|
py_globals, /*PyObject *globals,*/
|
|
0 /*PyObject *locals*/
|
|
);
|
|
if (!py_frame) goto bad;
|
|
py_frame->f_lineno = py_line;
|
|
PyTraceBack_Here(py_frame);
|
|
bad:
|
|
Py_XDECREF(py_code);
|
|
Py_XDECREF(py_frame);
|
|
}
|
|
|
|
static int __Pyx_InitStrings(__Pyx_StringTabEntry *t) {
|
|
while (t->p) {
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (t->is_unicode) {
|
|
*t->p = PyUnicode_DecodeUTF8(t->s, t->n - 1, NULL);
|
|
} else if (t->intern) {
|
|
*t->p = PyString_InternFromString(t->s);
|
|
} else {
|
|
*t->p = PyString_FromStringAndSize(t->s, t->n - 1);
|
|
}
|
|
#else /* Python 3+ has unicode identifiers */
|
|
if (t->is_unicode | t->is_str) {
|
|
if (t->intern) {
|
|
*t->p = PyUnicode_InternFromString(t->s);
|
|
} else if (t->encoding) {
|
|
*t->p = PyUnicode_Decode(t->s, t->n - 1, t->encoding, NULL);
|
|
} else {
|
|
*t->p = PyUnicode_FromStringAndSize(t->s, t->n - 1);
|
|
}
|
|
} else {
|
|
*t->p = PyBytes_FromStringAndSize(t->s, t->n - 1);
|
|
}
|
|
#endif
|
|
if (!*t->p)
|
|
return -1;
|
|
++t;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
/* Type Conversion Functions */
|
|
|
|
static CYTHON_INLINE int __Pyx_PyObject_IsTrue(PyObject* x) {
|
|
int is_true = x == Py_True;
|
|
if (is_true | (x == Py_False) | (x == Py_None)) return is_true;
|
|
else return PyObject_IsTrue(x);
|
|
}
|
|
|
|
static CYTHON_INLINE PyObject* __Pyx_PyNumber_Int(PyObject* x) {
|
|
PyNumberMethods *m;
|
|
const char *name = NULL;
|
|
PyObject *res = NULL;
|
|
#if PY_VERSION_HEX < 0x03000000
|
|
if (PyInt_Check(x) || PyLong_Check(x))
|
|
#else
|
|
if (PyLong_Check(x))
|
|
#endif
|
|
return Py_INCREF(x), x;
|
|
m = Py_TYPE(x)->tp_as_number;
|
|
#if PY_VERSION_HEX < 0x03000000
|
|
if (m && m->nb_int) {
|
|
name = "int";
|
|
res = PyNumber_Int(x);
|
|
}
|
|
else if (m && m->nb_long) {
|
|
name = "long";
|
|
res = PyNumber_Long(x);
|
|
}
|
|
#else
|
|
if (m && m->nb_int) {
|
|
name = "int";
|
|
res = PyNumber_Long(x);
|
|
}
|
|
#endif
|
|
if (res) {
|
|
#if PY_VERSION_HEX < 0x03000000
|
|
if (!PyInt_Check(res) && !PyLong_Check(res)) {
|
|
#else
|
|
if (!PyLong_Check(res)) {
|
|
#endif
|
|
PyErr_Format(PyExc_TypeError,
|
|
"__%s__ returned non-%s (type %.200s)",
|
|
name, name, Py_TYPE(res)->tp_name);
|
|
Py_DECREF(res);
|
|
return NULL;
|
|
}
|
|
}
|
|
else if (!PyErr_Occurred()) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"an integer is required");
|
|
}
|
|
return res;
|
|
}
|
|
|
|
static CYTHON_INLINE Py_ssize_t __Pyx_PyIndex_AsSsize_t(PyObject* b) {
|
|
Py_ssize_t ival;
|
|
PyObject* x = PyNumber_Index(b);
|
|
if (!x) return -1;
|
|
ival = PyInt_AsSsize_t(x);
|
|
Py_DECREF(x);
|
|
return ival;
|
|
}
|
|
|
|
static CYTHON_INLINE PyObject * __Pyx_PyInt_FromSize_t(size_t ival) {
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
if (ival <= LONG_MAX)
|
|
return PyInt_FromLong((long)ival);
|
|
else {
|
|
unsigned char *bytes = (unsigned char *) &ival;
|
|
int one = 1; int little = (int)*(unsigned char*)&one;
|
|
return _PyLong_FromByteArray(bytes, sizeof(size_t), little, 0);
|
|
}
|
|
#else
|
|
return PyInt_FromSize_t(ival);
|
|
#endif
|
|
}
|
|
|
|
static CYTHON_INLINE size_t __Pyx_PyInt_AsSize_t(PyObject* x) {
|
|
unsigned PY_LONG_LONG val = __Pyx_PyInt_AsUnsignedLongLong(x);
|
|
if (unlikely(val == (unsigned PY_LONG_LONG)-1 && PyErr_Occurred())) {
|
|
return (size_t)-1;
|
|
} else if (unlikely(val != (unsigned PY_LONG_LONG)(size_t)val)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"value too large to convert to size_t");
|
|
return (size_t)-1;
|
|
}
|
|
return (size_t)val;
|
|
}
|
|
|
|
|
|
#endif /* Py_PYTHON_H */
|