diff options
Diffstat (limited to 'gr-uhd/python/uhd/__init__.py')
| -rw-r--r-- | gr-uhd/python/uhd/__init__.py | 47 |
1 files changed, 30 insertions, 17 deletions
diff --git a/gr-uhd/python/uhd/__init__.py b/gr-uhd/python/uhd/__init__.py index 7c0f23dde1..1ffa94e821 100644 --- a/gr-uhd/python/uhd/__init__.py +++ b/gr-uhd/python/uhd/__init__.py @@ -16,6 +16,8 @@ line. ######################################################################## # Prepare uhd swig module to make it more pythonic ######################################################################## + + def _prepare_uhd_python(): try: from . import uhd_python @@ -25,45 +27,56 @@ def _prepare_uhd_python(): __path__.append(os.path.join(dirname, "bindings")) from . import uhd_python - #some useful typedefs for the user + # some useful typedefs for the user 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. + # 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_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().__init__(*args) - for key, val in list(kwargs.items()): setattr(self, key, val) + for key, val in list(kwargs.items()): + setattr(self, key, val) setattr(uhd_python, 'tune_request_t', tune_request_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" + # 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_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_python, attr[:-2], myobj) + 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_python, attr[:-2], myobj) - #make a new find devices that casts everything with the pythonized device_addr_t which has __str__ + # 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_python.device_addr_t() - for key in list(dev_addr.keys()): new_dev_addr[key] = dev_addr.get(key) + 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_python.find_devices_raw(*args, **kwargs)) setattr(uhd_python, 'find_devices', find_devices) + ######################################################################## # Initialize this module with the contents of uhd pybind ######################################################################## _prepare_uhd_python() + from .uhd_python import * |