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
|
//------------------------------------------------------------------------------
// GxB_Matrix_unpack_CSR: unpack a matrix in CSR format
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
#include "GB_export.h"
#define GB_FREE_ALL ;
GrB_Info GxB_Matrix_unpack_CSR // unpack a CSR matrix
(
GrB_Matrix A, // matrix to unpack (type, nrows, ncols unchanged)
GrB_Index **Ap, // row "pointers"
GrB_Index **Aj, // column indices
void **Ax, // values
GrB_Index *Ap_size, // size of Ap in bytes
GrB_Index *Aj_size, // size of Aj in bytes
GrB_Index *Ax_size, // size of Ax in bytes
bool *iso, // if true, A is iso
bool *jumbled, // if true, indices in each row may be unsorted
const GrB_Descriptor desc
)
{
//--------------------------------------------------------------------------
// check inputs and get the descriptor
//--------------------------------------------------------------------------
GB_WHERE1 ("GxB_Matrix_unpack_CSR (A, "
"&Ap, &Aj, &Ax, &Ap_size, &Aj_size, &Ax_size, &iso, "
"&jumbled, desc)") ;
GB_BURBLE_START ("GxB_Matrix_unpack_CSR") ;
GB_RETURN_IF_NULL_OR_FAULTY (A) ;
GB_GET_DESCRIPTOR (info, desc, xx1, xx2, xx3, xx4, xx5, xx6, xx7) ;
//--------------------------------------------------------------------------
// ensure the matrix is sparse by-row
//--------------------------------------------------------------------------
if (A->is_csc)
{
// A = A', done in-place, to put A in by-row format
GBURBLE ("(transpose) ") ;
GB_OK (GB_transpose_in_place (A, false, Context)) ;
}
//--------------------------------------------------------------------------
// finish any pending work
//--------------------------------------------------------------------------
if (jumbled == NULL)
{
// the unpacked matrix cannot be jumbled
GB_MATRIX_WAIT (A) ;
}
else
{
// the unpacked matrix is allowed to be jumbled
GB_MATRIX_WAIT_IF_PENDING_OR_ZOMBIES (A) ;
}
//--------------------------------------------------------------------------
// ensure the matrix is sparse
//--------------------------------------------------------------------------
GB_OK (GB_convert_any_to_sparse (A, Context)) ;
//--------------------------------------------------------------------------
// unpack the matrix
//--------------------------------------------------------------------------
ASSERT (GB_IS_SPARSE (A)) ;
ASSERT (!(A->is_csc)) ;
ASSERT (!GB_ZOMBIES (A)) ;
ASSERT (GB_IMPLIES (jumbled == NULL, !GB_JUMBLED (A))) ;
ASSERT (!GB_PENDING (A)) ;
int sparsity ;
bool is_csc ;
GrB_Type type ;
GrB_Index vlen, vdim ;
info = GB_export (true, &A, &type, &vlen, &vdim, false,
Ap, Ap_size, // Ap
NULL, NULL, // Ah
NULL, NULL, // Ab
Aj, Aj_size, // Ai
Ax, Ax_size, // Ax
NULL, jumbled, NULL, // jumbled or not
&sparsity, &is_csc, // sparse by row
iso, Context) ;
if (info == GrB_SUCCESS)
{
ASSERT (sparsity == GxB_SPARSE) ;
ASSERT (!is_csc) ;
}
GB_BURBLE_END ;
return (info) ;
}
|
