[Python] Expose new C API into python bridge

Author: EgorOrachyov

python/pycubool/bridge.py

@@ -36,6 +36,7 @@
 _hint_log_all = 128
 _hint_no_duplicates = 256
 _hint_time_check = 512
+_hint_transpose = 1024
 
 
 def get_log_hints(default=True, error=False, warning=False):
@@ -136,7 +137,10 @@ def load_and_configure(cubool_lib_path: str):
     index_t = ctypes.c_uint
     hints_t = ctypes.c_uint
     matrix_p = ctypes.c_void_p
+    vector_p = ctypes.c_void_p
+
     p_to_matrix_p = ctypes.POINTER(matrix_p)
+    p_to_vector_p = ctypes.POINTER(vector_p)
 
     lib.cuBool_SetupLogging.restype = status_t
     lib.cuBool_SetupLogging.argtypes = [
@@ -155,8 +159,8 @@ def load_and_configure(cubool_lib_path: str):
     lib.cuBool_Matrix_New.restype = status_t
     lib.cuBool_Matrix_New.argtypes = [
         p_to_matrix_p,
-        ctypes.c_uint,
-        ctypes.c_uint
+        index_t,
+        index_t
     ]
 
     lib.cuBool_Matrix_Free.restype = status_t
@@ -167,17 +171,17 @@ def load_and_configure(cubool_lib_path: str):
     lib.cuBool_Matrix_Build.restype = status_t
     lib.cuBool_Matrix_Build.argtypes = [
         matrix_p,
-        ctypes.POINTER(ctypes.c_uint),
-        ctypes.POINTER(ctypes.c_uint),
-        ctypes.c_uint,
+        ctypes.POINTER(index_t),
+        ctypes.POINTER(index_t),
+        index_t,
         hints_t
     ]
 
     lib.cuBool_Matrix_SetElement.restype = status_t
     lib.cuBool_Matrix_SetElement.argtypes = [
         matrix_p,
-        ctypes.c_uint,
-        ctypes.c_uint
+        index_t,
+        index_t
     ]
 
     lib.cuBool_Matrix_SetMarker.restype = status_t
@@ -190,15 +194,15 @@ def load_and_configure(cubool_lib_path: str):
     lib.cuBool_Matrix_Marker.argtypes = [
         matrix_p,
         ctypes.POINTER(ctypes.c_char),
-        ctypes.POINTER(ctypes.c_uint)
+        ctypes.POINTER(index_t)
     ]
 
     lib.cuBool_Matrix_ExtractPairs.restype = status_t
     lib.cuBool_Matrix_ExtractPairs.argtypes = [
         matrix_p,
-        ctypes.POINTER(ctypes.c_uint),
-        ctypes.POINTER(ctypes.c_uint),
-        ctypes.POINTER(ctypes.c_uint)
+        ctypes.POINTER(index_t),
+        ctypes.POINTER(index_t),
+        ctypes.POINTER(index_t)
     ]
 
     lib.cuBool_Matrix_ExtractSubMatrix.restype = status_t
@@ -212,6 +216,22 @@ def load_and_configure(cubool_lib_path: str):
         hints_t
     ]
 
+    lib.cuBool_Matrix_ExtractRow.restype = status_t
+    lib.cuBool_Matrix_ExtractRow.argtypes = [
+        vector_p,
+        matrix_p,
+        index_t,
+        hints_t
+    ]
+
+    lib.cuBool_Matrix_ExtractCol.restype = status_t
+    lib.cuBool_Matrix_ExtractCol.argtypes = [
+        vector_p,
+        matrix_p,
+        index_t,
+        hints_t
+    ]
+
     lib.cuBool_Matrix_Duplicate.restype = status_t
     lib.cuBool_Matrix_Duplicate.argtypes = [
         matrix_p,
@@ -228,13 +248,13 @@ def load_and_configure(cubool_lib_path: str):
     lib.cuBool_Matrix_Nrows.restype = status_t
     lib.cuBool_Matrix_Nrows.argtype = [
         matrix_p,
-        ctypes.POINTER(ctypes.c_uint)
+        ctypes.POINTER(index_t)
     ]
 
     lib.cuBool_Matrix_Ncols.restype = status_t
     lib.cuBool_Matrix_Ncols.argtype = [
         matrix_p,
-        ctypes.POINTER(ctypes.c_uint)
+        ctypes.POINTER(index_t)
     ]
 
     lib.cuBool_Matrix_Nvals.restype = status_t
@@ -243,6 +263,13 @@ def load_and_configure(cubool_lib_path: str):
         ctypes.POINTER(ctypes.c_size_t)
     ]
 
+    lib.cuBool_Matrix_Reduce.restype = status_t
+    lib.cuBool_Matrix_Reduce.argtype = [
+        vector_p,
+        matrix_p,
+        hints_t
+    ]
+
     lib.cuBool_Matrix_Reduce2.restype = status_t
     lib.cuBool_Matrix_Reduce2.argtype = [
         matrix_p,
@@ -258,6 +285,93 @@ def load_and_configure(cubool_lib_path: str):
         hints_t
     ]
 
+    lib.cuBool_Vector_New.restype = status_t
+    lib.cuBool_Vector_New.argtypes = [
+        p_to_vector_p,
+        index_t
+    ]
+
+    lib.cuBool_Vector_Build.restype = status_t
+    lib.cuBool_Vector_Build.argtypes = [
+        vector_p,
+        ctypes.POINTER(index_t),
+        index_t,
+        hints_t
+    ]
+
+    lib.cuBool_Vector_SetElement.restype = status_t
+    lib.cuBool_Vector_SetElement.argtypes = [
+        vector_p,
+        index_t
+    ]
+
+    lib.cuBool_Vector_SetMarker.restype = status_t
+    lib.cuBool_Vector_SetMarker.argtypes = [
+        vector_p,
+        ctypes.POINTER(ctypes.c_char)
+    ]
+
+    lib.cuBool_Vector_Marker.restype = status_t
+    lib.cuBool_Vector_Marker.argtypes = [
+        vector_p,
+        ctypes.POINTER(ctypes.c_char),
+        ctypes.POINTER(index_t)
+    ]
+
+    lib.cuBool_Vector_ExtractValues.restype = status_t
+    lib.cuBool_Vector_ExtractValues.argtypes = [
+        vector_p,
+        ctypes.POINTER(index_t),
+        ctypes.POINTER(index_t)
+    ]
+
+    lib.cuBool_Vector_ExtractSubVector.restype = status_t
+    lib.cuBool_Vector_ExtractSubVector.argtypes = [
+        vector_p,
+        vector_p,
+        index_t,
+        index_t,
+        hints_t
+    ]
+
+    lib.cuBool_Vector_Duplicate.restype = status_t
+    lib.cuBool_Vector_Duplicate.argtypes = [
+        vector_p,
+        p_to_vector_p
+    ]
+
+    lib.cuBool_Vector_Nvals.restype = status_t
+    lib.cuBool_Vector_Nvals.argtypes = [
+        vector_p,
+        ctypes.POINTER(index_t)
+    ]
+
+    lib.cuBool_Vector_Nrows.restype = status_t
+    lib.cuBool_Vector_Nrows.argtypes = [
+        vector_p,
+        ctypes.POINTER(index_t)
+    ]
+
+    lib.cuBool_Vector_Free.restype = status_t
+    lib.cuBool_Vector_Free.argtypes = [
+        vector_p
+    ]
+
+    lib.cuBool_Vector_Reduce.restype = status_t
+    lib.cuBool_Vector_Reduce.argtypes = [
+        ctypes.POINTER(index_t),
+        vector_p,
+        hints_t
+    ]
+
+    lib.cuBool_Vector_EWiseAdd.restype = status_t
+    lib.cuBool_Vector_EWiseAdd.argtypes = [
+        vector_p,
+        vector_p,
+        vector_p,
+        hints_t
+    ]
+
     lib.cuBool_MxM.restype = status_t
     lib.cuBool_MxM.argtypes = [
         matrix_p,
@@ -266,6 +380,22 @@ def load_and_configure(cubool_lib_path: str):
         hints_t
     ]
 
+    lib.cuBool_MxV.restype = status_t
+    lib.cuBool_MxV.argtypes = [
+        vector_p,
+        matrix_p,
+        vector_p,
+        hints_t
+    ]
+
+    lib.cuBool_VxM.restype = status_t
+    lib.cuBool_VxM.argtypes = [
+        vector_p,
+        vector_p,
+        matrix_p,
+        hints_t
+    ]
+
     lib.cuBool_Kronecker.restype = status_t
     lib.cuBool_Kronecker.argtypes = [
         matrix_p,

python/pycubool/matrix.py

@@ -40,7 +40,7 @@ class Matrix:
     - equality check
 
     Debug features:
-    - String markers
+    - string markers
     """
 
     __slots__ = ["hnd"]

python/pycubool/vector.py

@@ -0,0 +1,47 @@
+"""
+Vector primitive.
+"""
+
+import ctypes
+import random
+
+from . import wrapper
+from . import bridge
+
+
+__all__ = [
+    "Vector"
+]
+
+
+class Vector:
+    """
+    Wrapper for cuBool sparse boolean vector type.
+
+    Vector class supports all cuBool C API Vector functions.
+    Also Vector class provides additional fancy functions/operations for better user experience.
+
+    Vector creation:
+    - empty
+    - from list data
+    - random generated
+
+    Vector operations:
+    - vxm
+    - ewiseadd
+    - reduce
+    - vector extraction
+
+    Vector functions:
+    - to string
+    - values iterating
+    - equality check
+
+    Debug features:
+    - string markers
+    """
+
+    __slots__ = ["hnd"]
+
+    def __init__(self, hnd):
+        self.hnd = hnd