mpm: Moved some SPI setup code to dboard base

1 files changed, 22 insertions, 0 deletions

diff --git a/mpm/python/usrp_mpm/dboard_manager/base.py b/mpm/python/usrp_mpm/dboard_manager/base.py
index 985c20484..5bf31784f 100644
--- a/mpm/python/usrp_mpm/dboard_manager/base.py
+++ b/mpm/python/usrp_mpm/dboard_manager/base.py
@@ -18,6 +18,7 @@
 dboard base implementation module
 """
 
+from six import iteritems
 from ..mpmlog import get_logger
 
 class DboardManagerBase(object):
@@ -32,10 +33,31 @@ class DboardManagerBase(object):
     # Very important: A list of PIDs that apply to the current device. Must be
     # list, even if there's only one entry.
     pids = []
+    # A dictionary that maps chips or components to chip selects for SPI.
+    # If this is given, a dictionary called self._spi_nodes is created which
+    # maps these keys to actual spidev paths. Also throws a warning/error if
+    # the SPI configuration is invalid.
+    spi_chipselect = {}
 
     def __init__(self, slot_idx, **kwargs):
         self.log = get_logger('dboardManager')
         self.slot_idx = slot_idx
+        self._init_spi_nodes(kwargs.get('spi_nodes', []))
+
+
+    def _init_spi_nodes(self, spi_devices):
+        """
+        docstring for _init_spi_nodes
+        """
+        if len(spi_devices) < len(self.spi_chipselect):
+            self.log.error("Expected {0} spi devices, found {1} spi devices".format(
+                len(self.spi_chipselect), len(spi_devices),
+            ))
+            raise RuntimeError("Not enough SPI devices found.")
+        self._spi_nodes = {}
+        for k, v in iteritems(self.spi_chipselect):
+            self._spi_nodes[k] = spi_devices[v]
+        self.log.debug("spidev device node map: {}".format(self._spi_nodes))
 
     def get_serial(self):
         return self._eeprom.get("serial", "")