Openbmc dev snapshot.

1 files changed, 153 insertions, 0 deletions

diff --git a/common/recipes-utils/openbmc-gpio/files/openbmc_gpio_util.py b/common/recipes-utils/openbmc-gpio/files/openbmc_gpio_util.py
new file mode 100644
index 0000000..e4b40b0
--- /dev/null
+++ b/common/recipes-utils/openbmc-gpio/files/openbmc_gpio_util.py
@@ -0,0 +1,153 @@
+#!/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 soc_gpio_table import soc_gpio_table
+
+import openbmc_gpio
+import openbmc_gpio_table
+
+import argparse
+import logging
+import sys
+
+
+def _get_gpio_table():
+    gpio = openbmc_gpio_table.SocGPIOTable(soc_gpio_table)
+    return gpio
+
+
+def dump_func(args):
+    gpio = _get_gpio_table()
+    gpio.dump_functions()
+    return 0
+
+
+def config_func(args):
+    gpio = _get_gpio_table()
+    try:
+        gpio.config_function(args.function)
+    except openbmc_gpio_table.NotSmartEnoughException as e:
+        print('The code is not smart enough to set function "%s": %s\n'
+              'Please set the function manually.'
+              % (args.function, str(e)))
+        print('The current HW setting for this function is:')
+        gpio.dump_function(args.function)
+        return -1
+    except Exception as e:
+        print('Failed to set function "%s": %s\n'
+              'Please set the function manually.'
+              % (args.function, str(e)))
+        print('The current HW setting for this function is:')
+        gpio.dump_function(args.function)
+        logging.exception('Exception:')
+        return -2
+
+    print('Function "%s" is set' % args.function)
+    return 0
+
+
+def export_func(args):
+    openbmc_gpio.gpio_export(args.gpio, args.shadow)
+
+
+def set_func(args):
+    openbmc_gpio.gpio_set(args.gpio, args.value,
+                          change_direction=False if args.keep else True)
+
+
+def get_func(args):
+    val = openbmc_gpio.gpio_get(
+        args.gpio, change_direction=False if args.keep else True)
+    print('%d' % val)
+
+
+def info_func(args):
+    res = openbmc_gpio.gpio_info(args.gpio)
+    print('GPIO info for %s:' % args.gpio)
+    print('Path: %s\nShadow: %s\nDirection: %s\nValue: %s'
+          % (res['path'], res['shadow'], res['direction'], res['value']))
+
+
+def main():
+    ap = argparse.ArgumentParser()
+    ap.add_argument('--debug', action='store_true',
+                    help='Enable debug messages')
+
+    subparser = ap.add_subparsers()
+
+    dump_parser = subparser.add_parser(
+        'dump', help='Dump the current HW GPIO settings')
+    dump_parser.set_defaults(func=dump_func)
+
+    config_parser = subparser.add_parser(
+        'config', help='Configure one HW pin to a function')
+    config_parser.add_argument(
+        'function', help='The function name to set')
+    config_parser.set_defaults(func=config_func)
+
+    export_parser = subparser.add_parser(
+        'export', help='Export a GPIO directory')
+    export_parser.add_argument(
+        'gpio', help='The GPIO name, i.e. "A4", or "GPIOD2"')
+    export_parser.add_argument(
+        'shadow', default=None,
+        help='The shadow name given to this GPIO')
+    export_parser.set_defaults(func=export_func)
+
+
+    set_parser = subparser.add_parser(
+        'set', help='Set a value for a GPIO')
+    set_parser.add_argument(
+        'gpio', help='The GPIO name or number')
+    set_parser.add_argument(
+        'value', type=int, choices=[0, 1],
+        help='The value to set')
+    set_parser.add_argument(
+        '-k', '--keep', action='store_true',
+        help='Keep the GPIO direction')
+    set_parser.set_defaults(func=set_func)
+
+    get_parser = subparser.add_parser(
+        'get', help='Get a GPIO\'s value')
+    get_parser.add_argument(
+        'gpio', help='The GPIO name or number')
+    get_parser.add_argument(
+        '-k', '--keep', action='store_true',
+        help='Keep the GPIO direction')
+    get_parser.set_defaults(func=get_func)
+
+    info_parser = subparser.add_parser(
+        'info', help='Get a GPIO\'s info')
+    info_parser.add_argument(
+        'gpio', help='The GPIO name or number')
+    info_parser.set_defaults(func=info_func)
+
+    args = ap.parse_args()
+
+    logging.basicConfig(level=logging.DEBUG if args.debug else logging.INFO,
+                        format='%(asctime)s: %(message)s')
+
+    return args.func(args)
+
+
+rc = main()
+sys.exit(rc)