Generated by Cython 0.29.32
Yellow lines hint at Python interaction.
Click on a line that starts with a "+
" to see the C code that Cython generated for it.
Raw output: _ba.c
+01: cimport libc.stdlib
__pyx_t_2 = __Pyx_PyDict_NewPresized(0); if (unlikely(!__pyx_t_2)) __PYX_ERR(1, 1, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_2); if (PyDict_SetItem(__pyx_d, __pyx_n_s_test, __pyx_t_2) < 0) __PYX_ERR(1, 1, __pyx_L1_error) __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
02: from cpython cimport Py_INCREF, Py_DECREF, bool
03: cimport numpy as np
+04: np.import_array()
__pyx_t_1 = __pyx_f_5numpy_import_array(); if (unlikely(__pyx_t_1 == ((int)-1))) __PYX_ERR(1, 4, __pyx_L1_error)
05:
06: cdef class bitarray:
07: cdef:
08: BIT_ARRAY* bitarr
09: bit_index_t nbits ### REDUNDANT as the BIT_ARRAY struct has this
10:
+11: def __cinit__(self, bit_index_t n):
/* Python wrapper */ static int __pyx_pw_2bp_3_ba_8bitarray_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/ static int __pyx_pw_2bp_3_ba_8bitarray_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) { bit_index_t __pyx_v_n; int __pyx_r; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("__cinit__ (wrapper)", 0); { static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_n,0}; PyObject* values[1] = {0}; if (unlikely(__pyx_kwds)) { Py_ssize_t kw_args; const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args); switch (pos_args) { case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0); CYTHON_FALLTHROUGH; case 0: break; default: goto __pyx_L5_argtuple_error; } kw_args = PyDict_Size(__pyx_kwds); switch (pos_args) { case 0: if (likely((values[0] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_n)) != 0)) kw_args--; else goto __pyx_L5_argtuple_error; } if (unlikely(kw_args > 0)) { if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "__cinit__") < 0)) __PYX_ERR(1, 11, __pyx_L3_error) } } else if (PyTuple_GET_SIZE(__pyx_args) != 1) { goto __pyx_L5_argtuple_error; } else { values[0] = PyTuple_GET_ITEM(__pyx_args, 0); } __pyx_v_n = __Pyx_PyInt_As_uint64_t(values[0]); if (unlikely((__pyx_v_n == ((bit_index_t)-1)) && PyErr_Occurred())) __PYX_ERR(1, 11, __pyx_L3_error) } goto __pyx_L4_argument_unpacking_done; __pyx_L5_argtuple_error:; __Pyx_RaiseArgtupleInvalid("__cinit__", 1, 1, 1, PyTuple_GET_SIZE(__pyx_args)); __PYX_ERR(1, 11, __pyx_L3_error) __pyx_L3_error:; __Pyx_AddTraceback("bp._ba.bitarray.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename); __Pyx_RefNannyFinishContext(); return -1; __pyx_L4_argument_unpacking_done:; __pyx_r = __pyx_pf_2bp_3_ba_8bitarray___cinit__(((struct __pyx_obj_2bp_3_ba_bitarray *)__pyx_v_self), __pyx_v_n); int __pyx_lineno = 0; const char *__pyx_filename = NULL; int __pyx_clineno = 0; /* function exit code */ __Pyx_RefNannyFinishContext(); return __pyx_r; } static int __pyx_pf_2bp_3_ba_8bitarray___cinit__(struct __pyx_obj_2bp_3_ba_bitarray *__pyx_v_self, bit_index_t __pyx_v_n) { int __pyx_r; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("__cinit__", 0); /* … */ /* function exit code */ __pyx_r = 0; __Pyx_RefNannyFinishContext(); return __pyx_r; }
+12: self.nbits = n
__pyx_v_self->nbits = __pyx_v_n;
+13: self.bitarr = bit_array_create(n)
__pyx_v_self->bitarr = bit_array_create(__pyx_v_n);
14:
+15: def __dealloc__(self):
/* Python wrapper */ static void __pyx_pw_2bp_3_ba_8bitarray_3__dealloc__(PyObject *__pyx_v_self); /*proto*/ static void __pyx_pw_2bp_3_ba_8bitarray_3__dealloc__(PyObject *__pyx_v_self) { __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("__dealloc__ (wrapper)", 0); __pyx_pf_2bp_3_ba_8bitarray_2__dealloc__(((struct __pyx_obj_2bp_3_ba_bitarray *)__pyx_v_self)); /* function exit code */ __Pyx_RefNannyFinishContext(); } static void __pyx_pf_2bp_3_ba_8bitarray_2__dealloc__(struct __pyx_obj_2bp_3_ba_bitarray *__pyx_v_self) { __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("__dealloc__", 0); /* … */ /* function exit code */ __Pyx_RefNannyFinishContext(); }
+16: bit_array_free(self.bitarr)
bit_array_free(__pyx_v_self->bitarr);
17:
+18: def __str__(self):
/* Python wrapper */ static PyObject *__pyx_pw_2bp_3_ba_8bitarray_5__str__(PyObject *__pyx_v_self); /*proto*/ static PyObject *__pyx_pw_2bp_3_ba_8bitarray_5__str__(PyObject *__pyx_v_self) { PyObject *__pyx_r = 0; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("__str__ (wrapper)", 0); __pyx_r = __pyx_pf_2bp_3_ba_8bitarray_4__str__(((struct __pyx_obj_2bp_3_ba_bitarray *)__pyx_v_self)); /* function exit code */ __Pyx_RefNannyFinishContext(); return __pyx_r; } static PyObject *__pyx_pf_2bp_3_ba_8bitarray_4__str__(struct __pyx_obj_2bp_3_ba_bitarray *__pyx_v_self) { char *__pyx_v_str_; PyObject *__pyx_v_result = 0; PyObject *__pyx_r = NULL; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("__str__", 0); /* … */ /* function exit code */ __pyx_L1_error:; __Pyx_XDECREF(__pyx_t_1); __Pyx_AddTraceback("bp._ba.bitarray.__str__", __pyx_clineno, __pyx_lineno, __pyx_filename); __pyx_r = NULL; __pyx_L0:; __Pyx_XDECREF(__pyx_v_result); __Pyx_XGIVEREF(__pyx_r); __Pyx_RefNannyFinishContext(); return __pyx_r; }
19: cdef char* str_
20: cdef object result
21:
+22: str_ = <char*>libc.stdlib.malloc(self.nbits + 1)
__pyx_v_str_ = ((char *)malloc((__pyx_v_self->nbits + 1)));
+23: result = tounicode(bit_array_to_str(self.bitarr, str_))
__pyx_t_1 = __pyx_f_2bp_3_ba_tounicode(bit_array_to_str(__pyx_v_self->bitarr, __pyx_v_str_)); if (unlikely(!__pyx_t_1)) __PYX_ERR(1, 23, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __pyx_v_result = __pyx_t_1; __pyx_t_1 = 0;
+24: libc.stdlib.free(str_)
free(__pyx_v_str_);
25:
+26: return result
__Pyx_XDECREF(__pyx_r); __Pyx_INCREF(__pyx_v_result); __pyx_r = __pyx_v_result; goto __pyx_L0;
27:
+28: def __getitem__(self, bit_index_t i):
/* Python wrapper */ static PyObject *__pyx_pw_2bp_3_ba_8bitarray_7__getitem__(PyObject *__pyx_v_self, PyObject *__pyx_arg_i); /*proto*/ static PyObject *__pyx_pw_2bp_3_ba_8bitarray_7__getitem__(PyObject *__pyx_v_self, PyObject *__pyx_arg_i) { bit_index_t __pyx_v_i; PyObject *__pyx_r = 0; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("__getitem__ (wrapper)", 0); assert(__pyx_arg_i); { __pyx_v_i = __Pyx_PyInt_As_uint64_t(__pyx_arg_i); if (unlikely((__pyx_v_i == ((bit_index_t)-1)) && PyErr_Occurred())) __PYX_ERR(1, 28, __pyx_L3_error) } goto __pyx_L4_argument_unpacking_done; __pyx_L3_error:; __Pyx_AddTraceback("bp._ba.bitarray.__getitem__", __pyx_clineno, __pyx_lineno, __pyx_filename); __Pyx_RefNannyFinishContext(); return NULL; __pyx_L4_argument_unpacking_done:; __pyx_r = __pyx_pf_2bp_3_ba_8bitarray_6__getitem__(((struct __pyx_obj_2bp_3_ba_bitarray *)__pyx_v_self), ((bit_index_t)__pyx_v_i)); int __pyx_lineno = 0; const char *__pyx_filename = NULL; int __pyx_clineno = 0; /* function exit code */ __Pyx_RefNannyFinishContext(); return __pyx_r; } static PyObject *__pyx_pf_2bp_3_ba_8bitarray_6__getitem__(struct __pyx_obj_2bp_3_ba_bitarray *__pyx_v_self, bit_index_t __pyx_v_i) { PyObject *__pyx_r = NULL; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("__getitem__", 0); /* … */ /* function exit code */ __pyx_L1_error:; __Pyx_XDECREF(__pyx_t_1); __Pyx_AddTraceback("bp._ba.bitarray.__getitem__", __pyx_clineno, __pyx_lineno, __pyx_filename); __pyx_r = NULL; __pyx_L0:; __Pyx_XGIVEREF(__pyx_r); __Pyx_RefNannyFinishContext(); return __pyx_r; }
29: # could inline the macro version as well, less safe but should be
30: # thinner code
+31: return bit_array_get_bit(self.bitarr, i)
__Pyx_XDECREF(__pyx_r); __pyx_t_1 = __Pyx_PyInt_From_char(bit_array_get_bit(__pyx_v_self->bitarr, __pyx_v_i)); if (unlikely(!__pyx_t_1)) __PYX_ERR(1, 31, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __pyx_r = __pyx_t_1; __pyx_t_1 = 0; goto __pyx_L0;
32:
+33: def __setitem__(self, bit_index_t i, bool v):
/* Python wrapper */ static int __pyx_pw_2bp_3_ba_8bitarray_9__setitem__(PyObject *__pyx_v_self, PyObject *__pyx_arg_i, PyObject *__pyx_v_v); /*proto*/ static int __pyx_pw_2bp_3_ba_8bitarray_9__setitem__(PyObject *__pyx_v_self, PyObject *__pyx_arg_i, PyObject *__pyx_v_v) { bit_index_t __pyx_v_i; int __pyx_r; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("__setitem__ (wrapper)", 0); assert(__pyx_arg_i); { __pyx_v_i = __Pyx_PyInt_As_uint64_t(__pyx_arg_i); if (unlikely((__pyx_v_i == ((bit_index_t)-1)) && PyErr_Occurred())) __PYX_ERR(1, 33, __pyx_L3_error) } goto __pyx_L4_argument_unpacking_done; __pyx_L3_error:; __Pyx_AddTraceback("bp._ba.bitarray.__setitem__", __pyx_clineno, __pyx_lineno, __pyx_filename); __Pyx_RefNannyFinishContext(); return -1; __pyx_L4_argument_unpacking_done:; if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_v), __pyx_ptype_7cpython_4bool_bool, 1, "v", 0))) __PYX_ERR(1, 33, __pyx_L1_error) __pyx_r = __pyx_pf_2bp_3_ba_8bitarray_8__setitem__(((struct __pyx_obj_2bp_3_ba_bitarray *)__pyx_v_self), ((bit_index_t)__pyx_v_i), ((PyBoolObject *)__pyx_v_v)); int __pyx_lineno = 0; const char *__pyx_filename = NULL; int __pyx_clineno = 0; /* function exit code */ goto __pyx_L0; __pyx_L1_error:; __pyx_r = -1; __pyx_L0:; __Pyx_RefNannyFinishContext(); return __pyx_r; } static int __pyx_pf_2bp_3_ba_8bitarray_8__setitem__(struct __pyx_obj_2bp_3_ba_bitarray *__pyx_v_self, bit_index_t __pyx_v_i, PyBoolObject *__pyx_v_v) { int __pyx_r; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("__setitem__", 0); /* … */ /* function exit code */ __pyx_r = 0; goto __pyx_L0; __pyx_L1_error:; __Pyx_AddTraceback("bp._ba.bitarray.__setitem__", __pyx_clineno, __pyx_lineno, __pyx_filename); __pyx_r = -1; __pyx_L0:; __Pyx_RefNannyFinishContext(); return __pyx_r; }
+34: if v:
__pyx_t_1 = __Pyx_PyObject_IsTrue(((PyObject *)__pyx_v_v)); if (unlikely(__pyx_t_1 < 0)) __PYX_ERR(1, 34, __pyx_L1_error) if (__pyx_t_1) { /* … */ goto __pyx_L3; }
+35: bit_array_set_bit(self.bitarr, i)
bit_array_set_bit(__pyx_v_self->bitarr, __pyx_v_i);
36: else:
+37: bit_array_clear_bit(self.bitarr, i)
/*else*/ { bit_array_clear_bit(__pyx_v_self->bitarr, __pyx_v_i); } __pyx_L3:;
38:
39:
40:
+41: cdef unicode tounicode(char* s):
static PyObject *__pyx_f_2bp_3_ba_tounicode(char *__pyx_v_s) { PyObject *__pyx_r = NULL; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("tounicode", 0); /* … */ /* function exit code */ __pyx_L1_error:; __Pyx_XDECREF(__pyx_t_1); __Pyx_AddTraceback("bp._ba.tounicode", __pyx_clineno, __pyx_lineno, __pyx_filename); __pyx_r = 0; __pyx_L0:; __Pyx_XGIVEREF(__pyx_r); __Pyx_RefNannyFinishContext(); return __pyx_r; }
42: # from http://docs.cython.org/en/latest/src/tutorial/strings.html
+43: return s.decode('UTF-8', 'strict')
__Pyx_XDECREF(__pyx_r); __pyx_t_1 = __Pyx_decode_c_string(__pyx_v_s, 0, strlen(__pyx_v_s), NULL, NULL, PyUnicode_DecodeUTF8); if (unlikely(!__pyx_t_1)) __PYX_ERR(1, 43, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __pyx_r = ((PyObject*)__pyx_t_1); __pyx_t_1 = 0; goto __pyx_L0;
44:
45:
+46: cpdef bitarray bitarray_factory(object vec):
static PyObject *__pyx_pw_2bp_3_ba_1bitarray_factory(PyObject *__pyx_self, PyObject *__pyx_v_vec); /*proto*/ static struct __pyx_obj_2bp_3_ba_bitarray *__pyx_f_2bp_3_ba_bitarray_factory(PyObject *__pyx_v_vec, CYTHON_UNUSED int __pyx_skip_dispatch) { int __pyx_v_i; int __pyx_v_n; struct __pyx_obj_2bp_3_ba_bitarray *__pyx_v_result = 0; struct __pyx_obj_2bp_3_ba_bitarray *__pyx_r = NULL; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("bitarray_factory", 0); /* … */ /* function exit code */ __pyx_L1_error:; __Pyx_XDECREF(__pyx_t_2); __Pyx_XDECREF(__pyx_t_3); __Pyx_AddTraceback("bp._ba.bitarray_factory", __pyx_clineno, __pyx_lineno, __pyx_filename); __pyx_r = 0; __pyx_L0:; __Pyx_XDECREF((PyObject *)__pyx_v_result); __Pyx_XGIVEREF((PyObject *)__pyx_r); __Pyx_RefNannyFinishContext(); return __pyx_r; } /* Python wrapper */ static PyObject *__pyx_pw_2bp_3_ba_1bitarray_factory(PyObject *__pyx_self, PyObject *__pyx_v_vec); /*proto*/ static char __pyx_doc_2bp_3_ba_bitarray_factory[] = "Construct a bitarray from a vector\n\n A value at a position in the vector is considered set if the value \n evaluates True.\n\n Parameters\n ----------\n vec : object\n An iterable\n\n Returns\n -------\n bitarray\n The bitarray based off of topo\n "; static PyObject *__pyx_pw_2bp_3_ba_1bitarray_factory(PyObject *__pyx_self, PyObject *__pyx_v_vec) { PyObject *__pyx_r = 0; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("bitarray_factory (wrapper)", 0); __pyx_r = __pyx_pf_2bp_3_ba_bitarray_factory(__pyx_self, ((PyObject *)__pyx_v_vec)); /* function exit code */ __Pyx_RefNannyFinishContext(); return __pyx_r; } static PyObject *__pyx_pf_2bp_3_ba_bitarray_factory(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_vec) { PyObject *__pyx_r = NULL; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("bitarray_factory", 0); __Pyx_XDECREF(__pyx_r); __pyx_t_1 = ((PyObject *)__pyx_f_2bp_3_ba_bitarray_factory(__pyx_v_vec, 0)); if (unlikely(!__pyx_t_1)) __PYX_ERR(1, 46, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __pyx_r = __pyx_t_1; __pyx_t_1 = 0; goto __pyx_L0; /* function exit code */ __pyx_L1_error:; __Pyx_XDECREF(__pyx_t_1); __Pyx_AddTraceback("bp._ba.bitarray_factory", __pyx_clineno, __pyx_lineno, __pyx_filename); __pyx_r = NULL; __pyx_L0:; __Pyx_XGIVEREF(__pyx_r); __Pyx_RefNannyFinishContext(); return __pyx_r; }
47: """Construct a bitarray from a vector
48:
49: A value at a position in the vector is considered set if the value
50: evaluates True.
51:
52: Parameters
53: ----------
54: vec : object
55: An iterable
56:
57: Returns
58: -------
59: bitarray
60: The bitarray based off of topo
61: """
62: cdef int i
+63: cdef int n = len(vec)
__pyx_t_1 = PyObject_Length(__pyx_v_vec); if (unlikely(__pyx_t_1 == ((Py_ssize_t)-1))) __PYX_ERR(1, 63, __pyx_L1_error) __pyx_v_n = __pyx_t_1;
64: cdef bitarray result
65:
+66: result = bitarray(n)
__pyx_t_2 = __Pyx_PyInt_From_int(__pyx_v_n); if (unlikely(!__pyx_t_2)) __PYX_ERR(1, 66, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_2); __pyx_t_3 = __Pyx_PyObject_CallOneArg(((PyObject *)__pyx_ptype_2bp_3_ba_bitarray), __pyx_t_2); if (unlikely(!__pyx_t_3)) __PYX_ERR(1, 66, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_3); __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0; __pyx_v_result = ((struct __pyx_obj_2bp_3_ba_bitarray *)__pyx_t_3); __pyx_t_3 = 0;
+67: for i in range(n):
__pyx_t_4 = __pyx_v_n; __pyx_t_5 = __pyx_t_4; for (__pyx_t_6 = 0; __pyx_t_6 < __pyx_t_5; __pyx_t_6+=1) { __pyx_v_i = __pyx_t_6;
+68: if vec[i]:
__pyx_t_3 = __Pyx_GetItemInt(__pyx_v_vec, __pyx_v_i, int, 1, __Pyx_PyInt_From_int, 0, 1, 1); if (unlikely(!__pyx_t_3)) __PYX_ERR(1, 68, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_3); __pyx_t_7 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_7 < 0)) __PYX_ERR(1, 68, __pyx_L1_error) __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0; if (__pyx_t_7) { /* … */ } }
+69: result[i] = True
if (unlikely(__Pyx_SetItemInt(((PyObject *)__pyx_v_result), __pyx_v_i, Py_True, int, 1, __Pyx_PyInt_From_int, 0, 1, 1) < 0)) __PYX_ERR(1, 69, __pyx_L1_error)
70:
+71: return result
__Pyx_XDECREF(((PyObject *)__pyx_r)); __Pyx_INCREF(((PyObject *)__pyx_v_result)); __pyx_r = __pyx_v_result; goto __pyx_L0;