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
|
//------------------------------------------------------------------------------
// GB_positional_offset: return the offset of a positional operator
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
#include "GB.h"
int64_t GB_positional_offset // return the positional scalar is int64_t
(
GB_Opcode opcode, // opcode of positional operator
GrB_Scalar scalar, // scalar for idxunops, or NULL
bool *depends_on_j // if true, the op depends on j
)
{
int64_t iscalar = 0 ;
if (scalar != NULL)
{
// get the value of the scalar, for positional idxunops
GB_cast_scalar (&iscalar, GB_INT64_code, scalar->x, scalar->type->code,
scalar->type->size) ;
}
bool j_dep = false ;
switch (opcode)
{
// offset 0, does not depend on j
case GB_POSITIONI_unop_code : // z = position_i(A(i,j)) == i
case GB_FIRSTI_binop_code : // z = first_i(A(i,j),y) == i
case GB_SECONDI_binop_code : // z = second_i(x,A(i,j)) == i
j_dep = false ;
iscalar = 0 ;
break ;
// offset 1, does not depend on j
case GB_POSITIONI1_unop_code : // z = position_i1(A(i,j)) == i+1
case GB_FIRSTI1_binop_code : // z = first_i1(A(i,j),y) == i+1
case GB_SECONDI1_binop_code : // z = second_i1(x,A(i,j)) == i+1
j_dep = false ;
iscalar = 1 ;
break ;
// offset 0, depends on j
case GB_POSITIONJ_unop_code : // z = position_j(A(i,j)) == j
case GB_FIRSTJ_binop_code : // z = first_j(A(i,j),y) == j
case GB_SECONDJ_binop_code : // z = second_j(x,A(i,j)) == j
j_dep = true ;
iscalar = 0 ;
break ;
// offset 1, depends on j
case GB_POSITIONJ1_unop_code : // z = position_j1(A(i,j)) == j+1
case GB_FIRSTJ1_binop_code : // z = first_j1(A(i,j),y) == j+1
case GB_SECONDJ1_binop_code : // z = second_j1(x,A(i,j)) == j+1
j_dep = true ;
iscalar = 1 ;
break ;
// idxunops that do not depend on j
case GB_ROWINDEX_idxunop_code : // (i+y): row index - y
case GB_ROWLE_idxunop_code : // (i <= y): A (0:y,:)
case GB_ROWGT_idxunop_code : // (i > y): A (y+1:nrows-1,:)
j_dep = false ;
break ;
// idxunops that depend on j
case GB_COLINDEX_idxunop_code : // (j+y): col index - y
case GB_DIAGINDEX_idxunop_code : // (j-(i+y)): diag index + y
case GB_FLIPDIAGINDEX_idxunop_code :// (i-(j+y)), internal use
case GB_TRIL_idxunop_code : // (j <= (i+y)): tril (A,y)
case GB_TRIU_idxunop_code : // (j >= (i+y)): triu (A,y)
case GB_DIAG_idxunop_code : // (j == (i+y)): diag(A,y)
case GB_OFFDIAG_idxunop_code : // (j != (i+y)): offdiag(A,y)
case GB_COLLE_idxunop_code : // (j <= y): A (:,0:y)
case GB_COLGT_idxunop_code : // (j > y): A (:,y+1:ncols-1)
j_dep = true ;
break ;
// all other operators
default: ;
}
if (depends_on_j != NULL)
{
(*depends_on_j) = j_dep ;
}
return (iscalar) ;
}
|
