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;