1 //------------------------------------------------------------------------------
2 // GB_mex_extractTuples: extract all tuples from a matrix or vector
3 //------------------------------------------------------------------------------
4
5 // SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2021, All Rights Reserved.
6 // SPDX-License-Identifier: Apache-2.0
7
8 //------------------------------------------------------------------------------
9
10 #include "GB_mex.h"
11
12 #define USAGE "[I,J,X] = GB_mex_extractTuples (A, xtype)"
13
14 #define FREE_ALL \
15 { \
16 GrB_Matrix_free_(&A) ; \
17 GB_mx_put_global (true) ; \
18 }
19
mexFunction(int nargout,mxArray * pargout[],int nargin,const mxArray * pargin[])20 void mexFunction
21 (
22 int nargout,
23 mxArray *pargout [ ],
24 int nargin,
25 const mxArray *pargin [ ]
26 )
27 {
28
29 bool malloc_debug = GB_mx_get_global (true) ;
30 GrB_Matrix A = NULL ;
31 GB_void *Y = NULL ;
32 GrB_Index nvals = 0 ;
33
34 // check inputs
35 if (nargout > 3 || nargin < 1 || nargin > 2)
36 {
37 mexErrMsgTxt ("Usage: " USAGE) ;
38 }
39
40 #define GET_DEEP_COPY ;
41 #define FREE_DEEP_COPY ;
42
43 // get A (shallow copy)
44 A = GB_mx_mxArray_to_Matrix (pargin [0], "A input", false, true) ;
45 if (A == NULL)
46 {
47 FREE_ALL ;
48 mexErrMsgTxt ("A failed") ;
49 }
50
51 // get the number of entries in A
52 GrB_Matrix_nvals (&nvals, A) ;
53
54 // create I
55 pargout [0] = GB_mx_create_full (nvals, 1, GrB_UINT64) ;
56 GrB_Index *I = (GrB_Index *) mxGetData (pargout [0]) ;
57
58 // create J
59 GrB_Index *J = NULL ;
60 if (nargout > 1)
61 {
62 pargout [1] = GB_mx_create_full (nvals, 1, GrB_UINT64) ;
63 J = (GrB_Index *) mxGetData (pargout [1]) ;
64 }
65
66 // create X
67 GB_void *X = NULL ;
68 GrB_Type xtype = GB_mx_string_to_Type (PARGIN (1), A->type) ;
69 if (nargout > 2)
70 {
71 pargout [2] = GB_mx_create_full (nvals, 1, xtype) ;
72 X = (GB_void *) mxGetData (pargout [2]) ;
73 }
74
75 // [I,J,X] = find (A)
76 if (GB_VECTOR_OK (A))
77 {
78 // test extract vector methods
79 GrB_Vector v = (GrB_Vector) A ;
80 switch (xtype->code)
81 {
82 case GB_BOOL_code : METHOD (GrB_Vector_extractTuples_BOOL_ (I, (bool *) X, &nvals, v)) ; break ;
83 case GB_INT8_code : METHOD (GrB_Vector_extractTuples_INT8_ (I, (int8_t *) X, &nvals, v)) ; break ;
84 case GB_UINT8_code : METHOD (GrB_Vector_extractTuples_UINT8_ (I, (uint8_t *) X, &nvals, v)) ; break ;
85 case GB_INT16_code : METHOD (GrB_Vector_extractTuples_INT16_ (I, (int16_t *) X, &nvals, v)) ; break ;
86 case GB_UINT16_code : METHOD (GrB_Vector_extractTuples_UINT16_(I, (uint16_t *) X, &nvals, v)) ; break ;
87 case GB_INT32_code : METHOD (GrB_Vector_extractTuples_INT32_ (I, (int32_t *) X, &nvals, v)) ; break ;
88 case GB_UINT32_code : METHOD (GrB_Vector_extractTuples_UINT32_(I, (uint32_t *) X, &nvals, v)) ; break ;
89 case GB_INT64_code : METHOD (GrB_Vector_extractTuples_INT64_ (I, (int64_t *) X, &nvals, v)) ; break ;
90 case GB_UINT64_code : METHOD (GrB_Vector_extractTuples_UINT64_(I, (uint64_t *) X, &nvals, v)) ; break ;
91 case GB_FP32_code : METHOD (GrB_Vector_extractTuples_FP32_ (I, (float *) X, &nvals, v)) ; break ;
92 case GB_FP64_code : METHOD (GrB_Vector_extractTuples_FP64_ (I, (double *) X, &nvals, v)) ; break ;
93 case GB_FC32_code : METHOD (GxB_Vector_extractTuples_FC32_ (I, (GxB_FC32_t *) X, &nvals, v)) ; break ;
94 case GB_FC64_code : METHOD (GxB_Vector_extractTuples_FC64_ (I, (GxB_FC64_t *) X, &nvals, v)) ; break ;
95 case GB_UDT_code : METHOD (GrB_Vector_extractTuples_UDT_ (I, (void *) X, &nvals, v)) ; break ;
96 default : FREE_ALL ; mexErrMsgTxt ("unsupported type") ;
97 }
98 if (J != NULL)
99 {
100 for (int64_t p = 0 ; p < nvals ; p++) J [p] = 0 ;
101 }
102 }
103 else
104 {
105 switch (xtype->code)
106 {
107 case GB_BOOL_code : METHOD (GrB_Matrix_extractTuples_BOOL_ (I, J, (bool *) X, &nvals, A)) ; break ;
108 case GB_INT8_code : METHOD (GrB_Matrix_extractTuples_INT8_ (I, J, (int8_t *) X, &nvals, A)) ; break ;
109 case GB_UINT8_code : METHOD (GrB_Matrix_extractTuples_UINT8_ (I, J, (uint8_t *) X, &nvals, A)) ; break ;
110 case GB_INT16_code : METHOD (GrB_Matrix_extractTuples_INT16_ (I, J, (int16_t *) X, &nvals, A)) ; break ;
111 case GB_UINT16_code : METHOD (GrB_Matrix_extractTuples_UINT16_(I, J, (uint16_t *) X, &nvals, A)) ; break ;
112 case GB_INT32_code : METHOD (GrB_Matrix_extractTuples_INT32_ (I, J, (int32_t *) X, &nvals, A)) ; break ;
113 case GB_UINT32_code : METHOD (GrB_Matrix_extractTuples_UINT32_(I, J, (uint32_t *) X, &nvals, A)) ; break ;
114 case GB_INT64_code : METHOD (GrB_Matrix_extractTuples_INT64_ (I, J, (int64_t *) X, &nvals, A)) ; break ;
115 case GB_UINT64_code : METHOD (GrB_Matrix_extractTuples_UINT64_(I, J, (uint64_t *) X, &nvals, A)) ; break ;
116 case GB_FP32_code : METHOD (GrB_Matrix_extractTuples_FP32_ (I, J, (float *) X, &nvals, A)) ; break ;
117 case GB_FP64_code : METHOD (GrB_Matrix_extractTuples_FP64_ (I, J, (double *) X, &nvals, A)) ; break;
118 case GB_FC32_code : METHOD (GxB_Matrix_extractTuples_FC32_ (I, J, (GxB_FC32_t *) X, &nvals, A)) ; break ;
119 case GB_FC64_code : METHOD (GxB_Matrix_extractTuples_FC64_ (I, J, (GxB_FC64_t *) X, &nvals, A)) ; break;
120 case GB_UDT_code : METHOD (GrB_Matrix_extractTuples_UDT_ (I, J, (void *) X, &nvals, A)) ; break;
121 default : FREE_ALL ; mexErrMsgTxt ("unsupported type") ;
122 }
123 }
124
125 FREE_ALL ;
126 }
127
128