1 //------------------------------------------------------------------------------
2 // GB_mex_ewise_alias5: C<M> = A+M
3 //------------------------------------------------------------------------------
4
5 // SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2021, All Rights Reserved.
6 // SPDX-License-Identifier: Apache-2.0
7
8 //------------------------------------------------------------------------------
9
10 #include "GB_mex.h"
11
12 #define USAGE "C = GB_mex_ewise_alias5 (C, M, op, A, desc)"
13
14 #define FREE_ALL \
15 { \
16 GrB_Matrix_free_(&A) ; \
17 GrB_Matrix_free_(&M) ; \
18 GrB_Matrix_free_(&C) ; \
19 GrB_Descriptor_free_(&desc) ; \
20 GB_mx_put_global (true) ; \
21 }
22
mexFunction(int nargout,mxArray * pargout[],int nargin,const mxArray * pargin[])23 void mexFunction
24 (
25 int nargout,
26 mxArray *pargout [ ],
27 int nargin,
28 const mxArray *pargin [ ]
29 )
30 {
31
32 bool malloc_debug = GB_mx_get_global (true) ;
33 GrB_Matrix C = NULL, M = NULL, A = NULL ;
34 GrB_Descriptor desc = NULL ;
35
36 // check inputs
37 if (nargout > 1 || nargin < 4 || nargin > 5)
38 {
39 mexErrMsgTxt ("Usage: " USAGE) ;
40 }
41
42 // get C (make a deep copy)
43 #define GET_DEEP_COPY \
44 C = GB_mx_mxArray_to_Matrix (pargin [0], "C input", true, true) ;
45 #define FREE_DEEP_COPY GrB_Matrix_free_(&C) ;
46 GET_DEEP_COPY ;
47 if (C == NULL)
48 {
49 FREE_ALL ;
50 mexErrMsgTxt ("C failed") ;
51 }
52
53 // get M (shallow copy)
54 M = GB_mx_mxArray_to_Matrix (pargin [1], "M input", false, true) ;
55 if (M == NULL)
56 {
57 FREE_ALL ;
58 mexErrMsgTxt ("M failed") ;
59 }
60
61 // get op
62 bool user_complex = (Complex != GxB_FC64) && (C->type == Complex) ;
63 GrB_BinaryOp op ;
64 if (!GB_mx_mxArray_to_BinaryOp (&op, pargin [2], "op",
65 C->type, user_complex) || op == NULL)
66 {
67 FREE_ALL ;
68 mexErrMsgTxt ("op failed") ;
69 }
70
71 // get A (shallow copy)
72 A = GB_mx_mxArray_to_Matrix (pargin [3], "A input", false, true) ;
73 if (A == NULL)
74 {
75 FREE_ALL ;
76 mexErrMsgTxt ("A failed") ;
77 }
78
79 // get desc
80 if (!GB_mx_mxArray_to_Descriptor (&desc, PARGIN (4), "desc"))
81 {
82 FREE_ALL ;
83 mexErrMsgTxt ("desc failed") ;
84 }
85
86 // C<M> = A+M
87 METHOD (GrB_Matrix_eWiseAdd_BinaryOp_(C, M, NULL, op, A, M, desc)) ;
88
89 // return C to MATLAB as a struct and free the GraphBLAS C
90 pargout [0] = GB_mx_Matrix_to_mxArray (&C, "C output", true) ;
91
92 FREE_ALL ;
93 }
94
95