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
|
//------------------------------------------------------------------------------
// GxB_IndexUnaryOp_new: create a new user-defined index_unary operator
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
// Create a new a index_unary operator: z = f (x,i,j,thunk). The
// index_unary function signature must be:
// void f (void *z, const void *x, int64_t i, int64_t j, const void *thunk)
// and then it must recast its inputs (x and thunk) and output (z) arguments
// internally as needed. When used with a GrB_Vector, j is zero.
#include "GB.h"
GrB_Info GxB_IndexUnaryOp_new // create a named user-created IndexUnaryOp
(
GrB_IndexUnaryOp *op, // handle for the new IndexUnary operator
GxB_index_unary_function function, // pointer to index_unary function
GrB_Type ztype, // type of output z
GrB_Type xtype, // type of input x (the A(i,j) entry)
GrB_Type ytype, // type of input y (the scalar)
const char *idxop_name, // name of the user function
const char *idxop_defn // definition of the user function
)
{
//--------------------------------------------------------------------------
// check inputs
//--------------------------------------------------------------------------
GB_WHERE1 ("GxB_IndexUnaryOp_new (op, function, ztype, xtype, ytype"
", name, defn)") ;
GB_RETURN_IF_NULL (op) ;
(*op) = NULL ;
GB_RETURN_IF_NULL (function) ;
GB_RETURN_IF_NULL_OR_FAULTY (ztype) ;
GB_RETURN_IF_NULL_OR_FAULTY (xtype) ;
GB_RETURN_IF_NULL_OR_FAULTY (ytype) ;
//--------------------------------------------------------------------------
// allocate the index_unary op
//--------------------------------------------------------------------------
size_t header_size ;
(*op) = GB_MALLOC (1, struct GB_IndexUnaryOp_opaque, &header_size) ;
if (*op == NULL)
{
// out of memory
return (GrB_OUT_OF_MEMORY) ;
}
(*op)->header_size = header_size ;
//--------------------------------------------------------------------------
// initialize the index_unary operator
//--------------------------------------------------------------------------
(*op)->magic = GB_MAGIC ;
(*op)->ztype = ztype ;
(*op)->xtype = xtype ;
(*op)->ytype = ytype ; // thunk type
(*op)->unop_function = NULL ;
(*op)->idxunop_function = function ;
(*op)->binop_function = NULL ;
(*op)->selop_function = NULL ;
(*op)->opcode = GB_USER_idxunop_code ;
//--------------------------------------------------------------------------
// get the index_unary op name and defn
//--------------------------------------------------------------------------
GrB_Info info = GB_op_name_and_defn ((*op)->name, &((*op)->defn),
&((*op)->defn_size), idxop_name, idxop_defn,
"GxB_index_unary_function", 24) ;
if (info != GrB_SUCCESS)
{
// out of memory
GB_FREE (op, header_size) ;
return (info) ;
}
//--------------------------------------------------------------------------
// return result
//--------------------------------------------------------------------------
ASSERT_INDEXUNARYOP_OK ((*op), "new user-defined index_unary op", GB0) ;
return (GrB_SUCCESS) ;
}
|
