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
103
104
105
106
|
//------------------------------------------------------------------------------
// GxB_UnaryOp_new: create a new named unary operator
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
// a unary operator: z = f (x). The unary function signature must be
// void f (void *z, const void *x), and then it must recast its input and
// output arguments internally as needed.
#include "GB.h"
GrB_Info GxB_UnaryOp_new // create a new user-defined unary operator
(
GrB_UnaryOp *unaryop, // handle for the new unary operator
GxB_unary_function function, // pointer to the unary function
GrB_Type ztype, // type of output z
GrB_Type xtype, // type of input x
const char *unop_name, // name of the user function
const char *unop_defn // definition of the user function
)
{
//--------------------------------------------------------------------------
// check inputs
//--------------------------------------------------------------------------
GB_WHERE1 ("GxB_UnaryOp_new (unaryop, function, ztype, xtype, name, defn)");
GB_RETURN_IF_NULL (unaryop) ;
(*unaryop) = NULL ;
GB_RETURN_IF_NULL (function) ;
GB_RETURN_IF_NULL_OR_FAULTY (ztype) ;
GB_RETURN_IF_NULL_OR_FAULTY (xtype) ;
//--------------------------------------------------------------------------
// create the unary op
//--------------------------------------------------------------------------
// allocate the unary operator
size_t header_size ;
(*unaryop) = GB_MALLOC (1, struct GB_UnaryOp_opaque, &header_size) ;
if (*unaryop == NULL)
{
// out of memory
return (GrB_OUT_OF_MEMORY) ;
}
// initialize the unary operator
GrB_UnaryOp op = *unaryop ;
op->magic = GB_MAGIC ;
op->header_size = header_size ;
op->xtype = xtype ;
op->ztype = ztype ;
op->ytype = NULL ;
op->unop_function = function ;
op->idxunop_function = NULL ;
op->binop_function = NULL ;
op->selop_function = NULL ;
op->opcode = GB_USER_unop_code ;
//--------------------------------------------------------------------------
// get the unary op name and defn
//--------------------------------------------------------------------------
GrB_Info info = GB_op_name_and_defn (op->name, &(op->defn),
&(op->defn_size), unop_name, unop_defn, "GxB_unary_function", 18) ;
if (info != GrB_SUCCESS)
{
// out of memory
GB_FREE (unaryop, header_size) ;
return (info) ;
}
//--------------------------------------------------------------------------
// return result
//--------------------------------------------------------------------------
ASSERT_UNARYOP_OK (op, "new user-defined unary op", GB0) ;
return (GrB_SUCCESS) ;
}
//------------------------------------------------------------------------------
// GB_UnaryOp_new: create a new user-defined unary operator (historical)
//------------------------------------------------------------------------------
// This method was only accessible via the GrB_UnaryOp_new macro in v5.1.x
// and earlier. The GrB_UnaryOp_new macro in v5.2.x and later calls
// GxB_UnaryOp_new instead.
GrB_Info GB_UnaryOp_new // create a new user-defined unary operator
(
GrB_UnaryOp *unaryop, // handle for the new unary operator
GxB_unary_function function, // pointer to the unary function
GrB_Type ztype, // type of output z
GrB_Type xtype, // type of input x
const char *name // name of the user function
)
{
return (GxB_UnaryOp_new (unaryop, function, ztype, xtype, name, NULL)) ;
}
|
