aboutsummaryrefslogtreecommitdiffstats
path: root/mpm/python/usrp_mpm/simulator/sim_dboard.py
blob: a264eae9cd9f7ffe3855414cf1c5adeb005aa32c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#
# Copyright 2020 Ettus Research, a National Instruments Brand
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
from usrp_mpm.dboard_manager import DboardManagerBase
from usrp_mpm.mpmlog import get_logger
from usrp_mpm.mpmutils import to_native_str

class SimulatedDboardBase(DboardManagerBase):
    """
    A class to simulate daughterboards in a simulated device.
    """

    # Extra Overridables:
    # This is a list of extra methods to add to the class
    # Each element should be either a string or a tuple of
    # (string, function). Each element is added to the class
    # with the name being the string and the function being
    # the function, or the default of taking any amount of
    # arguments and returning None
    extra_methods = []

    def __init__(self, slot_idx, **kwargs):
        super().__init__(slot_idx, **kwargs)
        self.log = get_logger("sim_db-{}".format(slot_idx))
        self.device_info = {
            'pid': to_native_str(self.__class__.pids[0]),
            'serial': to_native_str("todo:serial-here"),
            'rev': to_native_str("1"),
            'eeprom_version': to_native_str('0')
        }
        self.rev = int(self.device_info['rev'])

        self.log.trace("This is a rev: {}".format(chr(65 + self.rev)))
        self._make_extra_methods()

    def init(self, args):
        self.log.trace("sim_db#init called")
        return True

    def tear_down(self):
        self.log.trace("sim_db#tear_down called")

    def _make_extra_methods(self):
        for entry in self.__class__.extra_methods:
            func = None
            prop_name = None
            if type(entry) is tuple:
                func = entry[1]
                prop_name = entry[0]
            else:
                func = lambda *args: None
                prop_name = entry
            # default values are needed because loop iterations don't create a new scope in python
            def wrapped_func(*args, prop_name=prop_name, func=func):
                self.log.debug("Called {} with args: {}".format(prop_name, args))
                return func(*args)
            setattr(self, prop_name, wrapped_func)