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
|
function result = gb_entries (A, varargin)
%GB_ENTRIES count or query the entries of a matrix.
% Implements GrB.entries (A, ...) and GrB.nonz (A, ...).
% SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
% SPDX-License-Identifier: Apache-2.0
% get the string arguments
dim = 'all' ; % 'all', 'row', or 'col'
kind = 'count' ; % 'count', 'list', or 'degree'
for k = 1:nargin-1
arg = varargin {k} ;
switch arg
case { 'all', 'row', 'col' }
dim = arg ;
case { 'count', 'list', 'degree' }
kind = arg ;
otherwise
error ('GrB:error', 'unknown option') ;
end
end
if (isequal (dim, 'all'))
switch kind
case 'count'
% number of entries in A
% e = GrB.entries (A)
result = gbnvals (A) ;
case 'list'
% list of values of unique entries
% X = GrB.entries (A, 'list')
result = unique (gbextractvalues (A)) ;
otherwise
error ('GrB:error', '''all'' and ''degree'' cannot be combined') ;
end
else
% get the row or column degree
result = gbdegree (A, dim) ; % dim is 'row' or 'col'
switch kind
case 'count'
% number of non-empty rows/cols
% e = GrB.entries (A, 'row')
% e = GrB.entries (A, 'col')
result = gbnvals (gbselect (result, 'nonzero')) ;
case 'list'
% list of non-empty rows/cols
% I = GrB.entries (A, 'row', 'list')
% J = GrB.entries (A, 'col', 'list')
desc.base = 'one-based int' ;
result = gbextracttuples (gbselect (result, 'nonzero'), desc) ;
% case 'degree'
% degree of all rows/cols
% d = GrB.entries (A, 'row', 'degree')
% d = GrB.entries (A, 'col', 'degree')
% result is returned as a GraphBLAS struct, already computed above
end
end
|
