graphblas_sparse_linear_algebra 0.62.0

Wrapper for SuiteSparse:GraphBLAS
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

use crate::collections::sparse_matrix::element::GetMatrixElementCoordinate;
use crate::collections::sparse_matrix::element::GetMatrixElementValue;
use crate::collections::sparse_matrix::ColumnIndex;
use crate::collections::sparse_matrix::RowIndex;
use crate::collections::sparse_matrix::{GetGraphblasSparseMatrix, SparseMatrix};
use crate::context::CallGraphBlasContext;
use crate::error::SparseLinearAlgebraError;
use crate::graphblas_bindings::{
    GrB_Matrix_setElement_BOOL, GrB_Matrix_setElement_FP32, GrB_Matrix_setElement_FP64,
    GrB_Matrix_setElement_INT16, GrB_Matrix_setElement_INT32, GrB_Matrix_setElement_INT64,
    GrB_Matrix_setElement_INT8, GrB_Matrix_setElement_UINT16, GrB_Matrix_setElement_UINT32,
    GrB_Matrix_setElement_UINT64, GrB_Matrix_setElement_UINT8,
};
use crate::index::IndexConversion;
use crate::value_type::utilities_to_implement_traits_for_all_value_types::implement_1_type_macro_for_all_value_types_and_typed_graphblas_function_with_implementation_type;
use crate::value_type::ConvertScalar;
use crate::value_type::ValueType;

pub trait SetSparseMatrixElement<T: ValueType> {
    fn set_value(
        &mut self,
        row_index: RowIndex,
        column_index: ColumnIndex,
        value: T,
    ) -> Result<(), SparseLinearAlgebraError>;
    fn set_element(
        &mut self,
        element: impl GetMatrixElementCoordinate + GetMatrixElementValue<T>,
    ) -> Result<(), SparseLinearAlgebraError>;
}

impl<T: ValueType + SetSparseMatrixElementTyped<T>> SetSparseMatrixElement<T> for SparseMatrix<T> {
    fn set_value(
        &mut self,
        row_index: RowIndex,
        column_index: ColumnIndex,
        value: T,
    ) -> Result<(), SparseLinearAlgebraError> {
        T::set_graphblas_matrix_value(self, row_index, column_index, value)
    }

    fn set_element(
        &mut self,
        element: impl GetMatrixElementCoordinate + GetMatrixElementValue<T>,
    ) -> Result<(), SparseLinearAlgebraError> {
        T::set_graphblas_matrix_element(self, element)
    }
}

pub trait SetSparseMatrixElementTyped<T: ValueType> {
    fn set_graphblas_matrix_value(
        matrix: &mut impl GetGraphblasSparseMatrix,
        row_index: RowIndex,
        column_index: ColumnIndex,
        value: T,
    ) -> Result<(), SparseLinearAlgebraError>;

    fn set_graphblas_matrix_element(
        matrix: &mut impl GetGraphblasSparseMatrix,
        element: impl GetMatrixElementCoordinate + GetMatrixElementValue<T>,
    ) -> Result<(), SparseLinearAlgebraError>;
}

macro_rules! implement_set_element_typed {
    ($value_type:ty, $conversion_target_type:ty, $add_element_function:ident) => {
        impl SetSparseMatrixElementTyped<$value_type> for $value_type {
            fn set_graphblas_matrix_value(
                matrix: &mut impl GetGraphblasSparseMatrix,
                row_index: RowIndex,
                column_index: ColumnIndex,
                value: $value_type,
            ) -> Result<(), SparseLinearAlgebraError> {
                let row_index_to_set = row_index.as_graphblas_index()?;
                let column_index_to_set = column_index.as_graphblas_index()?;
                let element_value = value.to_type()?;
                matrix.context_ref().call(
                    || unsafe {
                        $add_element_function(
                            matrix.graphblas_matrix_ptr(),
                            element_value,
                            row_index_to_set,
                            column_index_to_set,
                        )
                    },
                    unsafe { &matrix.graphblas_matrix_ptr() },
                )?;
                Ok(())
            }

            fn set_graphblas_matrix_element(
                matrix: &mut impl GetGraphblasSparseMatrix,
                element: impl GetMatrixElementCoordinate + GetMatrixElementValue<$value_type>,
            ) -> Result<(), SparseLinearAlgebraError> {
                <$value_type>::set_graphblas_matrix_value(
                    matrix,
                    element.row_index(),
                    element.column_index(),
                    element.value(),
                )
            }
        }
    };
}

implement_1_type_macro_for_all_value_types_and_typed_graphblas_function_with_implementation_type!(
    implement_set_element_typed,
    GrB_Matrix_setElement
);