Added Write numpy to NIComplex over gRPC

Author: abhising-NI

build/templates/_grpc_stub_interpreter.py/numpy_read_method.py.mako

@@ -16,7 +16,9 @@
 %>\
 
     def ${full_func_name}(${param_names_method}):  # noqa: N802
+% if numpy_complex_params:
         import numpy
+% endif
 % if included_in_proto:
         ${capture_response}self._invoke(
             self._client.${grpc_name},

build/templates/_grpc_stub_interpreter.py/numpy_write_method.py.mako

@@ -1,3 +1,59 @@
 <%page args="f, config, method_template"/>\
-## numpy_read and numpy_write are identical for gRPC -- both return a NotImplementedError
-<%include file="/_grpc_stub_interpreter.py/numpy_read_method.py.mako" args="f=f, config=config, method_template=method_template" />\
+<%
+    '''Renders a GrpcStubInterpreter method corresponding to the passed-in function metadata.'''
+    import build.helper as helper
+    parameters = f['parameters']
+    full_func_name = f['interpreter_name'] + method_template['method_python_name_suffix']
+    method_decl_params = helper.get_params_snippet(f, helper.ParameterUsageOptions.INTERPRETER_METHOD_DECLARATION)
+    grpc_name = f.get('grpc_name', f['name'])
+    grpc_request_args = helper.get_params_snippet(f, helper.ParameterUsageOptions.GRPC_REQUEST_PARAMETERS)
+    return_statement = helper.get_grpc_interpreter_method_return_snippet(f['parameters'], config)
+    if return_statement == 'return':
+        return_statement = None
+    capture_response = 'response = ' if return_statement else ''
+    included_in_proto = f.get('included_in_proto', True)
+    numpy_complex_params = [
+        p for p in helper.filter_parameters(parameters, helper.ParameterUsageOptions.NUMPY_PARAMETERS)
+        if p['complex_type'] is not None and p.get('original_type') in ('NIComplexNumber[]', 'NIComplexNumberF32[]', 'NIComplexI16[]')
+    ]
+
+    # For numpy complex inputs, create NIComplex message lists and map them in the request args
+    for p in numpy_complex_params:
+        # Replace occurrences like "field=python_name" with "field=python_name_list"
+        grpc_request_args = grpc_request_args.replace(
+            p['grpc_name'] + '=' + p['python_name'],
+            p['grpc_name'] + '=' + p['python_name'] + '_list'
+        )
+
+%>\
+
+    def ${full_func_name}(${method_decl_params}):  # noqa: N802
+% if included_in_proto:
+% for p in numpy_complex_params:
+% if p['original_type'] == 'NIComplexNumber[]':
+        ${p['python_name']}_list = [
+            grpc_types.NIComplexNumber(real=val.real, imaginary=val.imag)
+            for val in ${p['python_name']}.ravel()
+        ]
+% elif p['original_type'] == 'NIComplexNumberF32[]':
+        ${p['python_name']}_list = [
+            grpc_types.NIComplexNumberF32(real=val.real, imaginary=val.imag)
+            for val in ${p['python_name']}.ravel()
+        ]
+% elif p['original_type'] == 'NIComplexI16[]':
+        ${p['python_name']}_list = [
+            grpc_types.NIComplexI16(real=int(val['real']), imaginary=int(val['imag']))
+            for val in ${p['python_name']}.ravel()
+        ]
+% endif
+% endfor
+        ${capture_response}self._invoke(
+            self._client.${grpc_name},
+            grpc_types.${grpc_name}Request(${grpc_request_args}),
+        )
+% if return_statement:
+        ${return_statement}
+% endif
+% else:
+        raise NotImplementedError('${full_func_name} is not supported over gRPC')
+% endif

src/nirfsg/metadata/nirfsg.proto

@@ -1274,13 +1274,11 @@ message CreateDeembeddingSparameterTableArrayRequest {
   string port = 2;
   string table_name = 3;
   repeated double frequencies = 4;
-  sint32 frequencies_size = 5;
-  repeated nidevice_grpc.NIComplexNumber sparameter_table = 6;
-  sint32 sparameter_table_size = 7;
-  sint32 number_of_ports = 8;
+  repeated nidevice_grpc.NIComplexNumber sparameter_table = 5;
+  sint32 number_of_ports = 6;
   oneof sparameter_orientation_enum {
-    SParameterOrientation sparameter_orientation = 9;
-    sint32 sparameter_orientation_raw = 10;
+    SParameterOrientation sparameter_orientation = 7;
+    sint32 sparameter_orientation_raw = 8;
   }
 }
 
@@ -1532,7 +1530,6 @@ message GetChannelNameResponse {
 
 message GetDeembeddingSparametersRequest {
   nidevice_grpc.Session vi = 1;
-  sint32 sparameters_array_size = 2;
 }
 
 message GetDeembeddingSparametersResponse {