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
|
//------------------------------------------------------------------------------
// gb_get_mxargs: get input arguments to a GraphBLAS mexFunction
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
// gb_get_mxargs collects all the input arguments for the 12 foundational
// GraphBLAS operations. The user-level view is described below. For
// the private mexFunctions, the descriptor optionally appears as the last
// argument. The matrix arguments are either built-in sparse or full matrices,
// GraphBLAS matrices. To call the mexFunction, the opaque content of the
// GraphBLAS matrices has been extracted, so they appear here as GraphBLAS
// structs (a built-in struct G whose first field is always G.GraphBLAS).
#include "gb_interface.h"
void gb_get_mxargs
(
// input:
int nargin, // # input arguments for mexFunction
const mxArray *pargin [ ], // input arguments for mexFunction
const char *usage, // usage to print, if too many args appear
// output:
mxArray *Matrix [6], // matrix arguments
int *nmatrices, // # of matrix arguments
mxArray *String [2], // string arguments
int *nstrings, // # of string arguments
mxArray *Cell [2], // cell array arguments
int *ncells, // # of cell array arguments
GrB_Descriptor *desc, // last argument is always the descriptor
base_enum_t *base, // desc.base
kind_enum_t *kind, // desc.kind
GxB_Format_Value *fmt, // desc.format : by row or by col
int *sparsity // desc.format : hypersparse/sparse/bitmap/full
// or 0 if not in the descriptor
)
{
//--------------------------------------------------------------------------
// find the descriptor
//--------------------------------------------------------------------------
(*desc) = NULL ;
(*kind) = KIND_GRB ;
(*fmt) = GxB_NO_FORMAT ;
(*sparsity) = 0 ;
(*base) = BASE_DEFAULT ;
if (nargin > 0)
{
(*desc) = gb_mxarray_to_descriptor (pargin [nargin-1], kind, fmt,
sparsity, base) ;
}
if ((*desc) != NULL)
{
// descriptor is present, remove it from further consideration
nargin-- ;
}
//--------------------------------------------------------------------------
// find the remaining arguments
//--------------------------------------------------------------------------
(*nmatrices) = 0 ;
(*nstrings) = 0 ;
(*ncells) = 0 ;
for (int k = 0 ; k < nargin ; k++)
{
if (mxIsCell (pargin [k]))
{
// I or J index arguments
if ((*ncells) >= 2)
{
ERROR ("only 2D indexing is supported") ;
}
Cell [(*ncells)++] = (mxArray *) pargin [k] ;
}
else if (mxIsChar (pargin [k]))
{
// accum operator, unary op, binary op, monoid, or semiring
if ((*nstrings) >= 2)
{
ERROR (usage) ;
}
String [(*nstrings)++] = (mxArray *) pargin [k] ;
}
else
{
// a matrix argument is C, M, A, or B
if ((*nmatrices) >= 6)
{
// at most 6 matrix inputs are allowed
ERROR (usage) ;
}
Matrix [(*nmatrices)++] = (mxArray *) pargin [k] ;
}
}
}
|
