diff options
| author | Josh Morman <jmorman@gnuradio.org> | 2021-11-24 12:53:58 -0500 |
|---|---|---|
| committer | mormj <34754695+mormj@users.noreply.github.com> | 2021-11-24 14:41:53 -0500 |
| commit | 51ec89501552c35b2bd0721c0501302f224fe44d (patch) | |
| tree | 442036c424c286ece0976dcd3c36891cddefcd25 /gnuradio-runtime/python/gnuradio/gr/gateway.py | |
| parent | e426b9ad20dbbf41326c103b31a7418a80362b06 (diff) | |
runtime: pep8 formatting
Signed-off-by: Josh Morman <jmorman@gnuradio.org>
Diffstat (limited to 'gnuradio-runtime/python/gnuradio/gr/gateway.py')
| -rw-r--r-- | gnuradio-runtime/python/gnuradio/gr/gateway.py | 69 |
1 files changed, 36 insertions, 33 deletions
diff --git a/gnuradio-runtime/python/gnuradio/gr/gateway.py b/gnuradio-runtime/python/gnuradio/gr/gateway.py index 3be5e64eeb..7091a0586c 100644 --- a/gnuradio-runtime/python/gnuradio/gr/gateway.py +++ b/gnuradio-runtime/python/gnuradio/gr/gateway.py @@ -8,7 +8,6 @@ # - import numpy import ctypes @@ -80,7 +79,7 @@ class py_io_signature(object): return () if nports <= ntypes: return self.__types[:nports] - return self.__types + [self.__types[-1]]*(nports-ntypes) + return self.__types + [self.__types[-1]] * (nports - ntypes) def __iter__(self): """ @@ -106,7 +105,7 @@ class gateway_block(object): # Normalize the many Python ways of saying 'nothing' to '()' in_sig = in_sig or () out_sig = out_sig or () - + # Backward compatibility: array of type strings -> py_io_signature if type(in_sig) is py_io_signature: self.__in_sig = in_sig @@ -138,17 +137,17 @@ class gateway_block(object): Makes this block connectable by hier/top block python """ return self.gateway.to_basic_block() - + def fixed_rate_noutput_to_ninput(self, noutput_items): return int((noutput_items * self._decim / self._interp) + self.gateway.history() - 1) def handle_forecast(self, noutput_items, ninputs): """ - This is the handler function for forecast calls from + This is the handler function for forecast calls from block_gateway in c++ across pybind11 wrappers """ return self.forecast(noutput_items, ninputs) - + # return ninput_items_required def forecast(self, noutput_items, ninputs): @@ -156,50 +155,51 @@ class gateway_block(object): forecast is only called from a general block this is the default implementation """ - ninput_items_required = [0]*ninputs + ninput_items_required = [0] * ninputs for i in range(ninputs): - ninput_items_required[i] = noutput_items + self.gateway.history() - 1 + ninput_items_required[i] = noutput_items + \ + self.gateway.history() - 1 return ninput_items_required - def handle_general_work(self, noutput_items, - ninput_items, - input_items, - output_items): + def handle_general_work(self, noutput_items, + ninput_items, + input_items, + output_items): ninputs = len(input_items) noutputs = len(output_items) in_types = self.in_sig().port_types(ninputs) out_types = self.out_sig().port_types(noutputs) - ctypes.pythonapi.PyCapsule_GetPointer.restype = ctypes.c_void_p - ctypes.pythonapi.PyCapsule_GetPointer.argtypes = [ctypes.py_object, ctypes.c_char_p] + ctypes.pythonapi.PyCapsule_GetPointer.argtypes = [ + ctypes.py_object, ctypes.c_char_p] if self._block_type != gr.GW_BLOCK_GENERAL: - ii=[pointer_to_ndarray( - ctypes.pythonapi.PyCapsule_GetPointer(input_items[i],None), + ii = [pointer_to_ndarray( + ctypes.pythonapi.PyCapsule_GetPointer(input_items[i], None), in_types[i], self.fixed_rate_noutput_to_ninput(noutput_items) ) for i in range(ninputs)] else: - ii=[pointer_to_ndarray( - ctypes.pythonapi.PyCapsule_GetPointer(input_items[i],None), + ii = [pointer_to_ndarray( + ctypes.pythonapi.PyCapsule_GetPointer(input_items[i], None), in_types[i], ninput_items[i] ) for i in range(ninputs)] - oo=[pointer_to_ndarray( - ctypes.pythonapi.PyCapsule_GetPointer(output_items[i],None), + oo = [pointer_to_ndarray( + ctypes.pythonapi.PyCapsule_GetPointer(output_items[i], None), out_types[i], noutput_items - ) for i in range(noutputs)] + ) for i in range(noutputs)] if self._block_type != gr.GW_BLOCK_GENERAL: - r = self.work(ii,oo) + r = self.work(ii, oo) self.consume_items(r) else: - r = self.general_work(ii,oo) + r = self.general_work(ii, oo) return r @@ -224,7 +224,8 @@ class gateway_block(object): return self.__out_sig def set_msg_handler(self, which_port, handler_function): - self.gateway.set_msg_handler_pybind(which_port, handler_function.__name__) + self.gateway.set_msg_handler_pybind( + which_port, handler_function.__name__) # Save handler object in class so it's not garbage collected self.msg_handlers[which_port] = handler_function @@ -247,11 +248,10 @@ class gateway_block(object): # return self.gateway.produce(which_output, how_many_items) - ######################################################################## # Wrappers for the user to inherit from ######################################################################## -class basic_block(gateway_block): +class basic_block(gateway_block): """ Args: name (str): block name @@ -263,6 +263,7 @@ class basic_block(gateway_block): For backward compatibility, a sequence of numpy type names is also accepted as an io signature. """ + def __init__(self, name, in_sig, out_sig): gateway_block.__init__(self, name=name, @@ -271,10 +272,11 @@ class basic_block(gateway_block): block_type=gr.GW_BLOCK_GENERAL ) - def consume_items(self, nitems ): + def consume_items(self, nitems): pass -class sync_block(gateway_block): + +class sync_block(gateway_block): """ Args: name (str): block name @@ -286,6 +288,7 @@ class sync_block(gateway_block): For backward compatibility, a sequence of numpy type names is also accepted as an io signature. """ + def __init__(self, name, in_sig, out_sig): gateway_block.__init__(self, name=name, @@ -296,7 +299,7 @@ class sync_block(gateway_block): self._decim = 1 self._interp = 1 - def consume_items(self, nitems ): + def consume_items(self, nitems): if (nitems > 0): self.gateway.consume_each(nitems) @@ -313,6 +316,7 @@ class decim_block(gateway_block): For backward compatibility, a sequence of numpy type names is also accepted as an io signature. """ + def __init__(self, name, in_sig, out_sig, decim): gateway_block.__init__(self, name=name, @@ -325,7 +329,7 @@ class decim_block(gateway_block): self.gateway.set_relative_rate(self._interp, self._decim) self.gateway.set_output_multiple(self._interp) - def consume_items(self, nitems ): + def consume_items(self, nitems): if (nitems > 0): self.gateway.consume_each(int(nitems * self._decim)) @@ -333,7 +337,6 @@ class decim_block(gateway_block): return [self.fixed_rate_noutput_to_ninput(noutput_items) for ii in range(ninputs)] - class interp_block(gateway_block): """ Args: @@ -346,6 +349,7 @@ class interp_block(gateway_block): For backward compatibility, a sequence of numpy type names is also accepted as an io signature. """ + def __init__(self, name, in_sig, out_sig, interp): gateway_block.__init__(self, name=name, @@ -358,8 +362,7 @@ class interp_block(gateway_block): self.gateway.set_relative_rate(self._interp, self._decim) self.gateway.set_output_multiple(self._interp) - - def consume_items(self, nitems ): + def consume_items(self, nitems): if (nitems > 0): self.gateway.consume_each(int(nitems / self._interp)) |