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

use suitesparse_graphblas_sys::GrB_Matrix_removeElement;

use crate::{
    collections::sparse_matrix::{
        ColumnIndex, GetCoordinateIndices, GetGraphblasSparseMatrix, RowIndex, SparseMatrix,
    },
    context::CallGraphBlasContext,
    error::SparseLinearAlgebraError,
    index::IndexConversion,
    value_type::ValueType,
};

pub trait DropSparseMatrixElement {
    fn drop_element_with_coordinate(
        &mut self,
        coordinate: impl GetCoordinateIndices,
    ) -> Result<(), SparseLinearAlgebraError>;
    fn drop_element(
        &mut self,
        row_index: RowIndex,
        column_index: ColumnIndex,
    ) -> Result<(), SparseLinearAlgebraError>;
}

impl<T: ValueType> DropSparseMatrixElement for SparseMatrix<T> {
    fn drop_element_with_coordinate(
        &mut self,
        coordinate: impl GetCoordinateIndices,
    ) -> Result<(), SparseLinearAlgebraError> {
        drop_sparse_matrix_element_with_coordinate(self, coordinate)
    }
    fn drop_element(
        &mut self,
        row_index: RowIndex,
        column_index: ColumnIndex,
    ) -> Result<(), SparseLinearAlgebraError> {
        drop_sparse_matrix_element(self, row_index, column_index)
    }
}

pub fn drop_sparse_matrix_element(
    matrix: &mut impl GetGraphblasSparseMatrix,
    row_index: RowIndex,
    column_index: ColumnIndex,
) -> Result<(), SparseLinearAlgebraError> {
    let row_index_to_delete = row_index.as_graphblas_index()?;
    let column_index_to_delete = column_index.as_graphblas_index()?;

    matrix.context_ref().call(
        || unsafe {
            GrB_Matrix_removeElement(
                matrix.graphblas_matrix_ptr(),
                row_index_to_delete,
                column_index_to_delete,
            )
        },
        unsafe { &matrix.graphblas_matrix_ptr() },
    )?;
    Ok(())
}

pub fn drop_sparse_matrix_element_with_coordinate(
    matrix: &mut impl GetGraphblasSparseMatrix,
    coordinate: impl GetCoordinateIndices,
) -> Result<(), SparseLinearAlgebraError> {
    let row_index_to_delete = coordinate.row_index().to_graphblas_index()?;
    let column_index_to_delete = coordinate.column_index().to_graphblas_index()?;

    matrix.context_ref().call(
        || unsafe {
            GrB_Matrix_removeElement(
                matrix.graphblas_matrix_ptr(),
                row_index_to_delete,
                column_index_to_delete,
            )
        },
        unsafe { &matrix.graphblas_matrix_ptr() },
    )?;
    Ok(())
}