[Python] Add new methods into matrix class (full mat C API support)

Author: EgorOrachyov

python/pycubool/bridge.py

@@ -99,6 +99,17 @@ def get_reduce_hints(time_check):
     return hints
 
 
+def get_reduce_vector_hints(transpose, time_check):
+    hints = _hint_no
+
+    if transpose:
+        hints |= _hint_transpose
+    if time_check:
+        hints |= _hint_time_check
+
+    return hints
+
+
 def get_kronecker_hints(time_check):
     hints = _hint_no
 
@@ -128,6 +139,15 @@ def get_vxm_hints(time_check):
     return hints
 
 
+def get_mxv_hints(time_check):
+    hints = _hint_no
+
+    if time_check:
+        hints |= _hint_time_check
+
+    return hints
+
+
 def get_ewiseadd_hints(time_check):
     hints = _hint_no
 

python/pycubool/matrix.py

@@ -7,6 +7,7 @@
 
 from . import wrapper
 from . import bridge
+from . import vector
 
 
 __all__ = [
@@ -490,6 +491,34 @@ def extract_matrix(self, i, j, shape, out=None, time_check=False):
         bridge.check(status)
         return out
 
+    def extract_row(self, i, out=None):
+        if out is None:
+            out = Vector.empty(self.ncols)
+
+        status = wrapper.loaded_dll.cuBool_Matrix_ExtractRow(
+            out.hnd,
+            self.hnd,
+            ctypes.c_uint(i),
+            ctypes.c_uint(0)
+        )
+
+        bridge.check(status)
+        return out
+
+    def extract_col(self, j, out=None):
+        if out is None:
+            out = Vector.empty(self.nrows)
+
+        status = wrapper.loaded_dll.cuBool_Matrix_ExtractCol(
+            out.hnd,
+            self.hnd,
+            ctypes.c_uint(j),
+            ctypes.c_uint(0)
+        )
+
+        bridge.check(status)
+        return out
+
     def mxm(self, other, out=None, accumulate=False, time_check=False):
         """
         Matrix-matrix multiplication in boolean semiring with "x = and" and "+ = or" operations.
@@ -532,6 +561,20 @@ def mxm(self, other, out=None, accumulate=False, time_check=False):
         bridge.check(status)
         return out
 
+    def mxv(self, other, out=None, time_check=False):
+        if out is None:
+            out = Vector.empty(self.nrows)
+
+        status = wrapper.loaded_dll.cuBool_MxV(
+            out.hnd,
+            self.hnd,
+            other.hnd,
+            ctypes.c_uint(bridge.get_mxv_hints(time_check=time_check))
+        )
+
+        bridge.check(status)
+        return out
+
     def kronecker(self, other, out=None, time_check=False):
         """
         Matrix-matrix kronecker product with boolean "x = and" operation.
@@ -643,6 +686,20 @@ def reduce(self, time_check=False):
         bridge.check(status)
         return out
 
+    def reduce_vector(self, out=None, transpose=False, time_check=False):
+        if out is None:
+            nrows = self.ncols if transpose else self.nrows
+            out = Vector.empty(nrows)
+
+        status = wrapper.loaded_dll.cuBool_Matrix_Reduce(
+            out.hnd,
+            self.hnd,
+            ctypes.c_uint(bridge.get_reduce_vector_hints(transpose=transpose, time_check=time_check))
+        )
+
+        bridge.check(status)
+        return out
+
     def equals(self, other) -> bool:
         """
         Compare two matrices. Returns true if they are equal.