|
18 | 18 |
|
19 | 19 | #include "stdlib/math/base/special/cpolar.h"
|
20 | 20 | #include "stdlib/napi/argv.h"
|
| 21 | +#include "stdlib/napi/argv_complex128.h" |
| 22 | +#include "stdlib/napi/argv_float64array.h" |
| 23 | +#include "stdlib/napi/export.h" |
21 | 24 | #include <node_api.h>
|
22 | 25 |
|
23 | 26 | /**
|
|
28 | 31 | * @return Node-API value
|
29 | 32 | */
|
30 | 33 | static napi_value addon( napi_env env, napi_callback_info info ) {
|
31 |
| - napi_status status; |
32 |
| - |
33 |
| - // Get callback arguments: |
34 |
| - size_t argc = 2; |
35 |
| - napi_value argv[ 2 ]; |
36 |
| - status = napi_get_cb_info( env, info, &argc, argv, NULL, NULL ); |
37 |
| - assert( status == napi_ok ); |
38 |
| - |
39 |
| - // Check whether we were provided the correct number of arguments: |
40 |
| - if ( argc < 2 ) { |
41 |
| - status = napi_throw_error( env, NULL, "invalid invocation. Insufficient arguments." ); |
42 |
| - assert( status == napi_ok ); |
43 |
| - return NULL; |
44 |
| - } |
45 |
| - if ( argc > 2 ) { |
46 |
| - status = napi_throw_error( env, NULL, "invalid invocation. Too many arguments." ); |
47 |
| - assert( status == napi_ok ); |
48 |
| - return NULL; |
49 |
| - } |
50 |
| - |
51 |
| - bool res; |
52 |
| - status = napi_is_typedarray( env, argv[ 0 ], &res ); |
53 |
| - assert( status == napi_ok ); |
54 |
| - if ( res == false ) { |
55 |
| - status = napi_throw_type_error( env, NULL, "invalid argument. First argument must be a Float64Array." ); |
56 |
| - assert( status == napi_ok ); |
57 |
| - return NULL; |
58 |
| - } |
59 |
| - |
60 |
| - // Get the first element out |
61 |
| - napi_typedarray_type vtype0; |
62 |
| - size_t len; |
63 |
| - void *Out; |
64 |
| - status = napi_get_typedarray_info( env, argv[ 0 ], &vtype0, &len, &Out, NULL, NULL ); |
65 |
| - assert( status == napi_ok ); |
66 |
| - if ( vtype0 != napi_float64_array ) { |
67 |
| - status = napi_throw_type_error( env, NULL, "invalid argument. First argument must be a Float64Array." ); |
68 |
| - assert( status == napi_ok ); |
69 |
| - return NULL; |
70 |
| - } |
71 |
| - if ( len != 2 ) { |
72 |
| - status = napi_throw_range_error( env, NULL, "invalid argument. First argument must have 2 elements." ); |
73 |
| - assert( status == napi_ok ); |
74 |
| - return NULL; |
75 |
| - } |
76 |
| - |
77 |
| - // Get the real component |
78 |
| - napi_value xre; |
79 |
| - status = napi_get_named_property( env, argv[ 1 ], "re", &xre ); |
80 |
| - assert( status == napi_ok ); |
81 |
| - |
82 |
| - napi_valuetype xretype; |
83 |
| - status = napi_typeof( env, xre, &xretype ); |
84 |
| - assert( status == napi_ok ); |
85 |
| - if ( xretype != napi_number ) { |
86 |
| - status = napi_throw_type_error( env, NULL, "invalid argument. First argument must have a real component which is a number." ); |
87 |
| - assert( status == napi_ok ); |
88 |
| - return NULL; |
89 |
| - } |
90 |
| - |
91 |
| - // Get the imaginary component |
92 |
| - napi_value xim; |
93 |
| - status = napi_get_named_property( env, argv[ 1 ], "im", &xim ); |
94 |
| - assert( status == napi_ok ); |
95 |
| - |
96 |
| - napi_valuetype ximtype; |
97 |
| - status = napi_typeof( env, xim, &ximtype ); |
98 |
| - assert( status == napi_ok ); |
99 |
| - if ( ximtype != napi_number ) { |
100 |
| - status = napi_throw_type_error( env, NULL, "invalid argument. First argument must have an imaginary component which a number." ); |
101 |
| - assert( status == napi_ok ); |
102 |
| - return NULL; |
103 |
| - } |
104 |
| - |
105 |
| - double re; |
106 |
| - status = napi_get_value_double( env, xre, &re ); |
107 |
| - assert( status == napi_ok ); |
108 |
| - |
109 |
| - double im; |
110 |
| - status = napi_get_value_double( env, xim, &im ); |
111 |
| - assert( status == napi_ok ); |
112 |
| - |
113 |
| - double cabs; |
114 |
| - double cphase; |
115 |
| - stdlib_base_cpolar( stdlib_complex128( re, im ), &cabs, &cphase ); |
116 |
| - |
117 |
| - double *op = (double *)Out; |
118 |
| - op[ 0 ] = cabs; |
119 |
| - op[ 1 ] = cphase; |
120 |
| - |
| 34 | + STDLIB_NAPI_ARGV( env, info, argv, argc, 2 ); |
| 35 | + STDLIB_NAPI_ARGV_COMPLEX128( env, z, argv, 0 ); |
| 36 | + STDLIB_NAPI_ARGV_FLOAT64ARRAY( env, o, olen, argv, 1 ); |
| 37 | + stdlib_base_cpolar( z, &o[ 0 ], &o[ 1 ] ); |
121 | 38 | return NULL;
|
122 | 39 | }
|
123 | 40 |
|
124 |
| -/** |
125 |
| -* Initializes a Node-API module. |
126 |
| -* |
127 |
| -* @param env environment under which the function is invoked |
128 |
| -* @param exports exports object |
129 |
| -* @return main export |
130 |
| -*/ |
131 |
| -static napi_value init( napi_env env, napi_value exports ) { |
132 |
| - napi_value fcn; |
133 |
| - napi_status status = napi_create_function( env, "exports", NAPI_AUTO_LENGTH, addon, NULL, &fcn ); |
134 |
| - assert( status == napi_ok ); |
135 |
| - return fcn; |
136 |
| -} |
137 |
| - |
138 |
| -NAPI_MODULE( NODE_GYP_MODULE_NAME, init ) |
| 41 | +STDLIB_NAPI_MODULE_EXPORT_FCN( addon ) |
0 commit comments