1 files changed, 76 insertions, 0 deletions
diff --git a/common/recipes-utils/openbmc-gpio/files/openbmc_gpio_setup.py b/common/recipes-utils/openbmc-gpio/files/openbmc_gpio_setup.py
new file mode 100644
index 0000000..7c126d9
--- /dev/null
+++ b/common/recipes-utils/openbmc-gpio/files/openbmc_gpio_setup.py
@@ -0,0 +1,76 @@
+#!/usr/bin/python -tt
+# Copyright 2015-present Facebook. All rights reserved.
+#
+# This program file is free software; you can redistribute it and/or modify it
+# under the terms of the GNU General Public License as published by the
+# Free Software Foundation; version 2 of the License.
+#
+# This program is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+# for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program in a file named COPYING; if not, write to the
+# Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor,
+# Boston, MA 02110-1301 USA
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+from __future__ import unicode_literals
+
+from board_gpio_table import board_gpio_table
+from soc_gpio_table import soc_gpio_table
+
+import openbmc_gpio
+import openbmc_gpio_table
+
+import logging
+import sys
+
+
+def setup_board_gpio(soc_gpio_table, board_gpio_table, validate=True):
+    soc = openbmc_gpio_table.SocGPIOTable(soc_gpio_table)
+    gpio_configured = []
+    for gpio in board_gpio_table:
+        try:
+            soc.config_function(gpio.gpio, write_through=False)
+            gpio_configured.append(gpio.gpio)
+        except openbmc_gpio_table.ConfigUnknownFunction as e:
+            # not multiple-function GPIO pin
+            pass
+        except openbmc_gpio_table.NotSmartEnoughException as e:
+            logging.error('Failed to configure "%s" for "%s": %s'
+                          % (gpio.gpio, gpio.shadow, str(e)))
+
+    soc.write_to_hw()
+
+    if validate:
+        all_functions = set(soc.get_active_functions(refresh=True))
+        for gpio in gpio_configured:
+            if gpio not in all_functions:
+                raise Exception('Failed to configure function "%s"' % gpio)
+
+    for gpio in board_gpio_table:
+        openbmc_gpio.gpio_export(gpio.gpio, gpio.shadow)
+        if gpio.value == openbmc_gpio_table.GPIO_INPUT:
+            continue
+        elif gpio.value == openbmc_gpio_table.GPIO_OUT_HIGH:
+            openbmc_gpio.gpio_set(gpio.gpio, 1)
+        elif gpio.value == openbmc_gpio_table.GPIO_OUT_LOW:
+            openbmc_gpio.gpio_set(gpio.gpio, 0)
+        else:
+            raise Exception('Invalid value "%s"' % gpio.value)
+
+def main():
+    print('Setting up GPIOs ... ', end='')
+    sys.stdout.flush()
+    openbmc_gpio.setup_shadow()
+    setup_board_gpio(soc_gpio_table, board_gpio_table)
+    print('Done')
+    sys.stdout.flush()
+    return 0
+
+if __name__ == '__main__':
+    sys.exit(main())