From 2a575bf9b5a4942f60e979161764b9e942699e1e Mon Sep 17 00:00:00 2001
From: Lars Amsel <lars.amsel@ni.com>
Date: Fri, 4 Jun 2021 08:27:50 +0200
Subject: uhd: Add support for the USRP X410
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: Lars Amsel <lars.amsel@ni.com>
Co-authored-by: Michael Auchter <michael.auchter@ni.com>
Co-authored-by: Martin Braun <martin.braun@ettus.com>
Co-authored-by: Paul Butler <paul.butler@ni.com>
Co-authored-by: Cristina Fuentes <cristina.fuentes-curiel@ni.com>
Co-authored-by: Humberto Jimenez <humberto.jimenez@ni.com>
Co-authored-by: Virendra Kakade <virendra.kakade@ni.com>
Co-authored-by: Lane Kolbly <lane.kolbly@ni.com>
Co-authored-by: Max Köhler <max.koehler@ni.com>
Co-authored-by: Andrew Lynch <andrew.lynch@ni.com>
Co-authored-by: Grant Meyerhoff <grant.meyerhoff@ni.com>
Co-authored-by: Ciro Nishiguchi <ciro.nishiguchi@ni.com>
Co-authored-by: Thomas Vogel <thomas.vogel@ni.com>
---
 mpm/python/usrp_mpm/sys_utils/sysfs_gpio.py | 31 ++++++++++++++++++++---------
 1 file changed, 22 insertions(+), 9 deletions(-)

(limited to 'mpm/python/usrp_mpm/sys_utils/sysfs_gpio.py')

diff --git a/mpm/python/usrp_mpm/sys_utils/sysfs_gpio.py b/mpm/python/usrp_mpm/sys_utils/sysfs_gpio.py
index b542123bc..056c28fd0 100644
--- a/mpm/python/usrp_mpm/sys_utils/sysfs_gpio.py
+++ b/mpm/python/usrp_mpm/sys_utils/sysfs_gpio.py
@@ -145,6 +145,7 @@ class SysFSGPIO(object):
         self._use_mask = use_mask
         self._ddr = ddr
         self._init_value = init_value
+        self._out_value = 0
         self.log.trace("Generating SysFSGPIO object for identifiers `{}'..."
                        .format(identifiers))
         self._gpio_dev, self._map_info = \
@@ -182,6 +183,8 @@ class SysFSGPIO(object):
                 open(os.path.join(GPIO_SYSFS_BASE_DIR, 'export'), 'w').write('{}'.format(gpio_num))
             ddr_str = 'out' if ddr_out else 'in'
             ddr_str = 'high' if ini_v else ddr_str
+            if ini_v and ddr_out:
+                self._out_value |= 1 << gpio_idx
             self.log.trace("On GPIO path `{}', setting DDR mode to {}.".format(gpio_path, ddr_str))
             open(os.path.join(GPIO_SYSFS_BASE_DIR, gpio_path, 'direction'), 'w').write(ddr_str)
 
@@ -196,12 +199,18 @@ class SysFSGPIO(object):
             value = 1
         assert (1<<gpio_idx) & self._use_mask
         assert (1<<gpio_idx) & self._ddr
+        assert int(value) in [0, 1]
+        value = int(value)
         gpio_num = self._base_gpio + gpio_idx
         gpio_path = os.path.join(GPIO_SYSFS_BASE_DIR, 'gpio{}'.format(gpio_num))
         value_path = os.path.join(gpio_path, GPIO_SYSFS_VALUEFILE)
         self.log.trace("Writing value `{}' to `{}'...".format(value, value_path))
         assert os.path.exists(value_path)
         open(value_path, 'w').write('{}'.format(value))
+        if value:
+            self._out_value |= 1 << gpio_idx
+        else:
+            self._out_value &= ~(1 << gpio_idx)
 
     def reset(self, gpio_idx):
         """
@@ -216,19 +225,23 @@ class SysFSGPIO(object):
         """
         Read back a GPIO at given index.
 
-        Note: The GPIO must be in the valid range, and it's DDR value must be
-        low (for "in").
+        Note: The GPIO must be in the valid range. If it's DDR value is
+        low (for "in") then the register value is read from a local variable.
         """
         assert (1<<gpio_idx) & self._use_mask
-        assert (1<<gpio_idx) & (~self._ddr)
-        gpio_num = self._base_gpio + gpio_idx
-        gpio_path = os.path.join(GPIO_SYSFS_BASE_DIR, 'gpio{}'.format(gpio_num))
-        value_path = os.path.join(gpio_path, GPIO_SYSFS_VALUEFILE)
-        assert os.path.exists(value_path)
-        read_value = int(open(value_path, 'r').read().strip())
-        self.log.trace("Reading value {} from `{}'...".format(read_value, value_path))
+        if (1<<gpio_idx) & (~self._ddr):
+            gpio_num = self._base_gpio + gpio_idx
+            gpio_path = os.path.join(GPIO_SYSFS_BASE_DIR, 'gpio{}'.format(gpio_num))
+            value_path = os.path.join(gpio_path, GPIO_SYSFS_VALUEFILE)
+            assert os.path.exists(value_path)
+            read_value = int(open(value_path, 'r').read().strip())
+            self.log.trace("Reading value {} from `{}'...".format(read_value, value_path))
+        else:
+            read_value = 1 if self._out_value & (1 << gpio_idx) else 0
+            self.log.trace("Reading value {} from local var".format(read_value))
         return read_value
 
+
 class GPIOBank:
     """
     Usability / convenience wrapper for GPIO banks accessed by SysFSGPIO
-- 
cgit v1.2.3