diff options
Diffstat (limited to 'gr-uhd/python/uhd/__init__.py')
| -rw-r--r-- | gr-uhd/python/uhd/__init__.py | 95 |
1 files changed, 31 insertions, 64 deletions
diff --git a/gr-uhd/python/uhd/__init__.py b/gr-uhd/python/uhd/__init__.py index 0fa188ae23..85a8f2c142 100644 --- a/gr-uhd/python/uhd/__init__.py +++ b/gr-uhd/python/uhd/__init__.py @@ -16,51 +16,43 @@ line. from __future__ import absolute_import from __future__ import unicode_literals +import uhd # TODO: verify uhd python is installed as a dependency for gr-uhd with python + ######################################################################## # Prepare uhd swig module to make it more pythonic ######################################################################## -def _prepare_uhd_swig(): +def _prepare_uhd_python(): try: - from . import uhd_swig + from . import uhd_python except ImportError: import os dirname, filename = os.path.split(os.path.abspath(__file__)) - __path__.append(os.path.join(dirname, "..", "..", "swig")) - from . import uhd_swig + __path__.append(os.path.join(dirname, "bindings")) + from . import uhd_python #some useful typedefs for the user - setattr(uhd_swig, 'freq_range_t', uhd_swig.meta_range_t) - setattr(uhd_swig, 'gain_range_t', uhd_swig.meta_range_t) + setattr(uhd_python, 'freq_range_t', uhd_python.meta_range_t) + setattr(uhd_python, 'gain_range_t', uhd_python.meta_range_t) #Make the python tune request object inherit from float #so that it can be passed in GRC as a frequency parameter. #The type checking in GRC will accept the tune request. #Also use kwargs to construct individual struct elements. - class tune_request_t(uhd_swig.tune_request_t, float): - def __new__(self, *args, **kwargs): return float.__new__(self) + class tune_request_t(uhd_python.tune_request_t): + # def __new__(self, *args, **kwargs): return float.__new__(self) def __float__(self): return self.target_freq def __init__(self, *args, **kwargs): - super(tune_request_t, self).__init__(*args) + super().__init__(*args) for key, val in list(kwargs.items()): setattr(self, key, val) - setattr(uhd_swig, 'tune_request_t', tune_request_t) - - #Make the python tune request object inherit from string - #so that it can be passed in GRC as a string parameter. - #The type checking in GRC will accept the device address. - #Define the set/get item special methods for dict access. - class device_addr_t(uhd_swig.device_addr_t, str): - def __new__(self, *args): return str.__new__(self) - def __getitem__(self, key): return self.get(key) - def __setitem__(self, key, val): self.set(key, val) - def __init__(self, *args, **kwargs): - super(device_addr_t, self).__init__(*args) - if args and isinstance(args[0], device_addr_t): - for key in list(args[0].keys()): self[key] = args[0][key] - setattr(uhd_swig, 'device_addr_t', device_addr_t) + setattr(uhd_python, 'tune_request_t', tune_request_t) #make the streamer args take **kwargs on init - class stream_args_t(uhd_swig.stream_args_t): + class stream_args_t(uhd_python.stream_args_t): def __init__(self, *args, **kwargs): + # UHD Python API doesn't have default args for stream_args_t + # If empty args, then append empty str's + while len(args) < 2: + args += ("",) super(stream_args_t, self).__init__(*args) for key, val in list(kwargs.items()): #for some reason, I can't assign a list in the constructor @@ -68,60 +60,35 @@ def _prepare_uhd_swig(): if key == 'channels': for v in val: self.channels.append(v) elif key == 'args': - self.args = device_addr_t(val) + self.args = uhd_python.device_addr_t(val) else: setattr(self, key, val) - setattr(uhd_swig, 'stream_args_t', stream_args_t) - #handle general things on all uhd_swig attributes + # FIXME: stream_args_t.channels.append does not work due to copy operation of STL vectors + setattr(uhd_python, 'stream_args_t', stream_args_t) + + #handle general things on all uhd_python attributes #Install the __str__ and __repr__ handlers if applicable #Create aliases for uhd swig attributes to avoid the "_t" - for attr in dir(uhd_swig): - myobj = getattr(uhd_swig, attr) + for attr in dir(uhd_python): + myobj = getattr(uhd_python, attr) if hasattr(myobj, 'to_string'): myobj.__repr__ = lambda o: o.to_string().strip() if hasattr(myobj, 'to_pp_string'): myobj.__str__ = lambda o: o.to_pp_string().strip() if hasattr(myobj, 'to_bool'): myobj.__nonzero__ = lambda o: o.to_bool() if hasattr(myobj, 'to_int'): myobj.__int__ = lambda o: o.to_int() if hasattr(myobj, 'to_real'): myobj.__float__ = lambda o: o.to_real() - if attr.endswith('_t'): setattr(uhd_swig, attr[:-2], myobj) + if attr.endswith('_t'): setattr(uhd_python, attr[:-2], myobj) #make a new find devices that casts everything with the pythonized device_addr_t which has __str__ def find_devices(*args, **kwargs): def to_pythonized_dev_addr(dev_addr): - new_dev_addr = uhd_swig.device_addr_t() + new_dev_addr = uhd_python.device_addr_t() for key in list(dev_addr.keys()): new_dev_addr[key] = dev_addr.get(key) return new_dev_addr - return __builtins__['map'](to_pythonized_dev_addr, uhd_swig.find_devices_raw(*args, **kwargs)) - setattr(uhd_swig, 'find_devices', find_devices) - - #Cast constructor args (FIXME swig handle overloads?) - for attr in ('usrp_source', 'usrp_sink', 'amsg_source'): - def constructor_factory(old_constructor): - def constructor_interceptor(*args, **kwargs): - args = list(args) - kwargs = dict(kwargs) - for index, key, cast in ( - (0, 'device_addr', device_addr), - ): - if len(args) > index: - args[index] = cast(args[index]) - if key in kwargs: - kwargs[key] = cast(kwargs[key]) - #don't pass kwargs, it confuses swig, map into args list: - for key in ('device_addr', 'stream_args', - 'issue_stream_cmd_on_start', 'tsb_tag_name', 'msgq'): - if key in kwargs: args.append(kwargs[key]) - return old_constructor(*args) - return constructor_interceptor - setattr(uhd_swig, attr, constructor_factory(getattr(uhd_swig, attr))) - - #Aliases for deprecated constructors - setattr(uhd_swig, 'single_usrp_source', uhd_swig.usrp_source) - setattr(uhd_swig, 'single_usrp_sink', uhd_swig.usrp_sink) - setattr(uhd_swig, 'multi_usrp_source', uhd_swig.usrp_source) - setattr(uhd_swig, 'multi_usrp_sink', uhd_swig.usrp_sink) + return __builtins__['map'](to_pythonized_dev_addr, uhd_python.find_devices_raw(*args, **kwargs)) + setattr(uhd_python, 'find_devices', find_devices) ######################################################################## -# Initialize this module with the contents of uhd swig +# Initialize this module with the contents of uhd pybind ######################################################################## -_prepare_uhd_swig() -from .uhd_swig import * +_prepare_uhd_python() +from .uhd_python import * |