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

use std::{marker::PhantomData, sync::Arc};

use suitesparse_graphblas_sys::GrB_Matrix;

use crate::collections::sparse_matrix::{GetGraphblasSparseMatrix, SparseMatrix};
use crate::context::{Context, GetContext};
use crate::value_type::ValueType;

pub struct GraphblasMatrixHandleUntyped<'a> {
    pointer_to_graphblas_matrix: GrB_Matrix,
    context: Arc<Context>,
    _lifetime_marker: PhantomData<&'a GrB_Matrix>,
}

pub struct GraphblasMatrixHandle<'a, T: ValueType> {
    pointer_to_graphblas_matrix: GrB_Matrix,
    context: Arc<Context>,
    _lifetime_and_value_type_marker: PhantomData<&'a T>,
}

impl<'a> GraphblasMatrixHandleUntyped<'a> {
    pub fn from_sparse_matrix(
        sparse_matrix: &'a (impl GetGraphblasSparseMatrix + GetContext),
    ) -> Self {
        Self {
            pointer_to_graphblas_matrix: unsafe { sparse_matrix.graphblas_matrix_ptr() },
            context: sparse_matrix.context(),
            _lifetime_marker: PhantomData,
        }
    }
}

impl<'a, T: ValueType> GraphblasMatrixHandle<'a, T> {
    pub fn from_sparse_matrix(sparse_matrix: &SparseMatrix<T>) -> Self {
        Self {
            pointer_to_graphblas_matrix: unsafe { sparse_matrix.graphblas_matrix_ptr() },
            context: sparse_matrix.context(),
            _lifetime_and_value_type_marker: PhantomData,
        }
    }
}

impl<'a> GetGraphblasSparseMatrix for GraphblasMatrixHandleUntyped<'a> {
    unsafe fn graphblas_matrix_ptr(&self) -> GrB_Matrix {
        self.pointer_to_graphblas_matrix
    }

    unsafe fn graphblas_matrix_ptr_ref(&self) -> &GrB_Matrix {
        &self.pointer_to_graphblas_matrix
    }

    unsafe fn graphblas_matrix_ptr_mut(&mut self) -> &mut GrB_Matrix {
        &mut self.pointer_to_graphblas_matrix
    }
}

impl<'a> GetContext for GraphblasMatrixHandleUntyped<'a> {
    fn context(&self) -> Arc<Context> {
        self.context.to_owned()
    }

    fn context_ref(&self) -> &Arc<Context> {
        &self.context
    }
}

impl<'a, T: ValueType> GetGraphblasSparseMatrix for GraphblasMatrixHandle<'a, T> {
    unsafe fn graphblas_matrix_ptr(&self) -> GrB_Matrix {
        self.pointer_to_graphblas_matrix
    }

    unsafe fn graphblas_matrix_ptr_ref(&self) -> &GrB_Matrix {
        &self.pointer_to_graphblas_matrix
    }

    unsafe fn graphblas_matrix_ptr_mut(&mut self) -> &mut GrB_Matrix {
        &mut self.pointer_to_graphblas_matrix
    }
}

impl<'a, T: ValueType> GetContext for GraphblasMatrixHandle<'a, T> {
    fn context(&self) -> Arc<Context> {
        self.context.to_owned()
    }

    fn context_ref(&self) -> &Arc<Context> {
        &self.context
    }
}