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
|
//------------------------------------------------------------------------------
// GrB_Matrix_extractTuples: extract all tuples from a matrix
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
// Extracts all tuples from a matrix, like [I,J,X] = find (A) in MATLAB. If
// any parameter I, J and/or X is NULL, then that component is not extracted.
// The size of the I, J, and X arrays (those that are not NULL) is given by
// nvals, which must be at least as large as GrB_nvals (&nvals, A). The values
// in the matrix are typecasted to the type of X, as needed.
// If any parameter I, J, and/or X is NULL, that component is not extracted.
// For example, to extract just the row and col indices, pass I and J as
// non-NULL, and X as NULL. This is like [I,J,~] = find (A) in MATLAB.
// If A is iso and X is not NULL, the iso scalar Ax [0] is expanded into X.
#include "GB.h"
#include "extractTuples/GB_extractTuples.h"
#define GB_EXTRACT_TUPLES(function_name,ctype,xtype) \
GrB_Info function_name /* extract tuples from a matrix */ \
( \
uint64_t *I, /* array for returning row indices of tuples */ \
uint64_t *J, /* array for returning col indices of tuples */ \
ctype *X, /* array for returning values of tuples */ \
uint64_t *p_nvals, /* I,J,X size on input; # tuples on output */ \
const GrB_Matrix A /* matrix to extract tuples from */ \
) \
{ \
GB_WHERE_1 (A, GB_STR(function_name) " (I, J, X, nvals, A)") ; \
GB_RETURN_IF_NULL (A) ; \
GB_RETURN_IF_NULL (p_nvals) ; \
GB_BURBLE_START (GB_STR(function_name)) ; \
info = GB_extractTuples (I, false, J, false, X, p_nvals, xtype, A, Werk);\
GB_BURBLE_END ; \
GB_PRAGMA (omp flush) \
return (info) ; \
}
// with 64-bit I and J arrays
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_BOOL , bool , GrB_BOOL )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_INT8 , int8_t , GrB_INT8 )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_INT16 , int16_t , GrB_INT16 )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_INT32 , int32_t , GrB_INT32 )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_INT64 , int64_t , GrB_INT64 )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_UINT8 , uint8_t , GrB_UINT8 )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_UINT16, uint16_t , GrB_UINT16)
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_UINT32, uint32_t , GrB_UINT32)
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_UINT64, uint64_t , GrB_UINT64)
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_FP32 , float , GrB_FP32 )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_FP64 , double , GrB_FP64 )
GB_EXTRACT_TUPLES (GxB_Matrix_extractTuples_FC32 , GxB_FC32_t , GxB_FC32 )
GB_EXTRACT_TUPLES (GxB_Matrix_extractTuples_FC64 , GxB_FC64_t , GxB_FC64 )
//------------------------------------------------------------------------------
// GrB_Matrix_extractTuples_UDT: extract from a matrix with user-defined type
//------------------------------------------------------------------------------
GrB_Info GrB_Matrix_extractTuples_UDT
(
uint64_t *I, // array for returning row indices of tuples
uint64_t *J, // array for returning col indices of tuples
void *X, // array for returning values of tuples
uint64_t *p_nvals, // I,J,X size on input; # tuples on output
const GrB_Matrix A // matrix to extract tuples from
)
{
GB_WHERE_1 (A, "GrB_Matrix_extractTuples_UDT (I, J, X, nvals, A)") ;
GB_RETURN_IF_NULL (A) ;
GB_RETURN_IF_NULL (p_nvals) ;
GB_BURBLE_START ("GrB_Matrix_extractTuples_UDT") ;
if (A->type->code != GB_UDT_code)
{
// A must have a user-defined type
return (GrB_DOMAIN_MISMATCH) ;
}
info = GB_extractTuples (I, false, J, false, X, p_nvals, A->type, A, Werk) ;
GB_BURBLE_END ;
GB_PRAGMA (omp flush)
return (info) ;
}
|
