feat: [wip] support cardinality in rpqmatrix algo

Author: suvorovrain

experimental/algorithm/LAGraph_RPQMatrix.c

@@ -129,6 +129,44 @@ char msg[LAGRAPH_MSG_LEN] ;
 #include <stdbool.h>
 #include <stdio.h>
 
+void cardinality_lor(GrB_Index *Crr, GrB_Index *Ccr, GrB_Index Arr, GrB_Index Acr, GrB_Index Brr, GrB_Index Bcr, GrB_Index dim, GrB_Index Annz, GrB_Index Bnnz) {
+    GrB_Index Crr_result, Ccr_result ;
+
+    *Crr = Crr_result ;
+    *Ccr = Ccr_result ;
+};
+
+void cardinality_concat(GrB_Index *Crr, GrB_Index *Ccr, GrB_Index Arr, GrB_Index Acr, GrB_Index Brr, GrB_Index Bcr, GrB_Index dim, GrB_Index Annz, GrB_Index Bnnz) {
+     GrB_Index Crr_result, Ccr_result ;
+
+    *Crr = Crr_result ;
+    *Ccr = Ccr_result ;
+};
+
+void cardinality_kleene(GrB_Index *Crr, GrB_Index *Ccr, GrB_Index Arr, GrB_Index Acr, GrB_Index Brr, GrB_Index Bcr, GrB_Index dim, GrB_Index Annz, GrB_Index Bnnz) {
+     GrB_Index Crr_result, Ccr_result ;
+
+    *Crr = Crr_result ;
+    *Ccr = Ccr_result ;
+};
+
+GrB_Info LAGraph_RPQMatrix_reduce(GrB_Index *res, GrB_Matrix mat, uint8_t reduce_type){
+    GrB_Index nvals ;
+    GrB_Vector reduce ;
+    if (reduce_type == 0)
+    {
+        GrB_Matrix_reduce_Monoid(reduce,NULL,NULL,GxB_ANY_BOOL_MONOID,mat,NULL) ;
+    }
+    else if (reduce_type == 1)
+    {
+        GrB_Matrix_reduce_Monoid(reduce, NULL,NULL,GxB_ANY_BOOL_MONOID,mat,GrB_DESC_T0) ;
+    }
+    GrB_Vector_nvals(&nvals,reduce) ;
+    *res = nvals ;
+    return (GrB_SUCCESS) ;
+}
+
+
 GrB_Info LAGraph_RPQMatrix_check(RPQMatrixPlan *plan, GrB_Index *dimension, char *msg)
 {
     if (plan == NULL)
@@ -137,7 +175,7 @@ GrB_Info LAGraph_RPQMatrix_check(RPQMatrixPlan *plan, GrB_Index *dimension, char
     }
     if (plan->op == RPQ_MATRIX_OP_LABEL)
     {
-        LG_ASSERT(plan->mat != NULL, GrB_NULL_POINTER);
+        LG_ASSERT(plan->mat != NULL, GrB_NULL_POINTER) ;
         GrB_Index nrows, ncols ;
         OK(GrB_Matrix_nrows(&nrows, plan->mat)) ;
         OK(GrB_Matrix_ncols(&ncols, plan->mat)) ;
@@ -224,7 +262,12 @@ static GrB_Info LAGraph_RPQMatrixLor(RPQMatrixPlan *plan, char *msg)
     GRB_TRY(GrB_eWiseAdd(res, GrB_NULL, GrB_NULL,
                          op, lhs_mat, rhs_mat, GrB_DESC_R)) ;
     plan->res_mat = res ;
-
+    // GrB_Index n ;
+    // GrB_Matrix_ncols(&n, lhs_mat) ;
+    // GrB_Index Annz, Bnnz;
+    // GrB_Matrix_nvals(&Annz, lhs_mat) ;
+    // GrB_Matrix_nvals(&Bnnz, rhs_mat) ;
+    // cardinality_lor(plan->row_reduce, plan->col_reduce, lhs->row_reduce, lhs->col_reduce, rhs->row_reduce, rhs->col_reduce, n, Annz, Bnnz) ;
     return (GrB_SUCCESS) ;
 }
 
@@ -254,6 +297,12 @@ static GrB_Info LAGraph_RPQMatrixConcat(RPQMatrixPlan *plan, char *msg)
     GRB_TRY(GrB_mxm(res, GrB_NULL, GrB_NULL,
                     sr, lhs_mat, rhs_mat, GrB_DESC_R)) ;
     plan->res_mat = res ;
+    // GrB_Index n ;
+    // GrB_Matrix_ncols(&n, lhs_mat) ;
+    // GrB_Index Annz, Bnnz;
+    // GrB_Matrix_nvals(&Annz, lhs_mat) ;
+    // GrB_Matrix_nvals(&Bnnz, rhs_mat) ;
+    // cardinality_concat(plan->row_reduce, plan->col_reduce, lhs->row_reduce, lhs->col_reduce, rhs->row_reduce, rhs->col_reduce, n, Annz, Bnnz) ;
 
     return (GrB_SUCCESS) ;
 }

experimental/test/test_RPQMatrix.c

@@ -88,15 +88,28 @@ void test_RPQMatrix_CONCAT(void)
     LAGraph_Init(msg) ;
 
     GrB_Matrix A, B ;
+    GrB_Index nA, nB ;
+    GrB_Matrix_nrows(&nA, A) ;
+    GrB_Matrix_nrows(&nB, B) ;
     load_matrix(&A, "rpq_data/a.mtx") ;
     load_matrix(&B, "rpq_data/b.mtx") ;
-
-    RPQMatrixPlan planA = {.op = RPQ_MATRIX_OP_LABEL, .mat = A} ;
-    RPQMatrixPlan planB = {.op = RPQ_MATRIX_OP_LABEL, .mat = B} ;
+    GrB_Vector A_row_reduce, A_col_reduce, B_row_reduce, B_col_reduce ;
+    GrB_Vector_new(A_row_reduce, GrB_BOOL, nA) ; 
+    GrB_Vector_new(A_col_reduce, GrB_BOOL, nA) ; 
+    GrB_Vector_new(B_row_reduce, GrB_BOOL, nB) ; 
+    GrB_Vector_new(B_col_reduce, GrB_BOOL, nB) ; 
+    GrB_Matrix_reduce_Monoid(A_row_reduce,NULL,NULL,GxB_ANY_BOOL_MONOID,A,NULL) ;
+    GrB_Matrix_reduce_Monoid(A_col_reduce,NULL,NULL,GxB_ANY_BOOL_MONOID,A,GrB_DESC_T0) ;
+    GrB_Matrix_reduce_Monoid(B_row_reduce,NULL,NULL,GxB_ANY_BOOL_MONOID,B,NULL) ;
+    GrB_Matrix_reduce_Monoid(B_col_reduce,NULL,NULL,GxB_ANY_BOOL_MONOID,B,GrB_DESC_T0) ;
+
+    RPQMatrixPlan planA = {.op = RPQ_MATRIX_OP_LABEL, .mat = A, .col_reduced = A_col_reduce, .row_reduced = A_row_reduce} ;
+    RPQMatrixPlan planB = {.op = RPQ_MATRIX_OP_LABEL, .mat = B, .col_reduced = B_col_reduce, .row_reduced = B_row_reduce} ;
     RPQMatrixPlan concat = {
         .op = RPQ_MATRIX_OP_CONCAT,
         .lhs = &planA,
-        .rhs = &planB} ;
+        .rhs = &planB
+    } ;
     GrB_Index result_nnz ;
     OK(LAGraph_RPQMatrix(&result_nnz, &concat, msg)) ;
 

include/LAGraphX.h

@@ -946,6 +946,12 @@ typedef struct RPQMatrixPlan
     GrB_Matrix mat ;            // adjacency matrix of label.
                                 // Only for leafes. Should be NULL
                                 // in other cases
+    GrB_Index row_reduced ;     // number of nonzero elements in 
+                                // reduced  by rows `mat` matrix.
+                                // Only for leafes
+    GrB_Index col_reduced ;     // number of nonzero elements in 
+                                // reduced by columns `mat` matrix.
+                                // Only for leafes
     GrB_Matrix res_mat ;        // result of subtree execution.
                                 // Should be NULL
 } RPQMatrixPlan ;