mirror of
https://github.com/EttusResearch/uhd.git
synced 2026-10-25 17:36:10 +00:00
22f7036ffb
GNU Radio-based tests are not used anymore. The gr- prefix is removed from the USRP test folder name and tests from the apps folder that do not depend on GNU Radio are kept in the top level folder.
80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
"""
|
|
Source Generator Plugins for RF Test Scripts
|
|
"""
|
|
|
|
import importlib
|
|
from inspect import isclass
|
|
from builtins import input, object
|
|
from six import itervalues
|
|
|
|
|
|
###############################################################################
|
|
# Source Generator Plugins
|
|
###############################################################################
|
|
class SourceGenerator(object):
|
|
"""
|
|
Parent class for source generators.
|
|
"""
|
|
def __init__(self, log, **_):
|
|
self.log = log
|
|
|
|
def tune(self, freq, power_lvl_dbm):
|
|
"""
|
|
Set the TX frequency. This function can block until the Tx frequency is
|
|
stable, so if LOs need to settle or whatnot, it's OK for this function
|
|
to take a while to return. After it's returned, the assumption is that
|
|
we can start RXing.
|
|
"""
|
|
raise NotImplementedError()
|
|
|
|
def tear_down(self):
|
|
"""Do all necessary clean-up for the source generator"""
|
|
pass
|
|
|
|
|
|
class ManualSourceGenerator(SourceGenerator):
|
|
"""
|
|
Not really a source generator, but a command-line interaction with the user
|
|
to manually set the TX source.
|
|
"""
|
|
def tune(self, freq, power_lvl_dbm):
|
|
"""
|
|
Ask the user to set the TX frequency. Once the user continues, the
|
|
assumption is that we can start RXing.
|
|
"""
|
|
input("Please tune the signal generator to {:.3f} MHz and {:.1f} dBm, "
|
|
"then press Enter".format(freq / 1e6, power_lvl_dbm))
|
|
|
|
|
|
DEFAULT_SOURCE_GENERATORS = {
|
|
'default': ManualSourceGenerator,
|
|
}
|
|
DEFAULT_SOURCE_GENERATOR = 'default'
|
|
|
|
|
|
def get_source_generator(log=None, src_gen_id=None, **kwargs):
|
|
"""
|
|
Factory function to get a source generator
|
|
"""
|
|
src_gen_id = src_gen_id or DEFAULT_SOURCE_GENERATOR
|
|
if src_gen_id in DEFAULT_SOURCE_GENERATORS:
|
|
return DEFAULT_SOURCE_GENERATORS.get(src_gen_id)(log, **kwargs)
|
|
try:
|
|
module = importlib.import_module(src_gen_id)
|
|
except ImportError:
|
|
raise RuntimeError("Could not find source generator plugin `{}'!".format(
|
|
src_gen_id))
|
|
src_gens = [
|
|
x for x in itervalues(module.__dict__)
|
|
if isclass(x) and issubclass(x, SourceGenerator) and x != SourceGenerator
|
|
]
|
|
if not src_gens:
|
|
raise RuntimeError(
|
|
"Could not find any source generator classes in module `{}'!"
|
|
.format(src_gen_id))
|
|
if len(src_gens) > 1:
|
|
raise RuntimeError(
|
|
"Ambiguous source generator plugin `{}'! Too many generator classes: {}"
|
|
.format(src_gen_id, src_gens))
|
|
return src_gens[0](log, **kwargs)
|