[DM/LED] Add GPIO LED driver

Signed-off-by: GuEe-GUI <[email protected]>

3 files changed, 238 insertions, 0 deletions

diff --git a/components/drivers/led/Kconfig b/components/drivers/led/Kconfig
index 89415363d6..462aa0bac0 100644
--- a/components/drivers/led/Kconfig
+++ b/components/drivers/led/Kconfig
@@ -3,6 +3,13 @@ menuconfig RT_USING_LED
     depends on RT_USING_DM
     default n
 
+config RT_LED_GPIO
+    bool "GPIO connected LEDs Support"
+    depends on RT_USING_LED
+    depends on RT_USING_PINCTRL
+    depends on RT_USING_OFW
+    default n
+
 if RT_USING_LED
     osource "$(SOC_DM_LED_DIR)/Kconfig"
 endif
diff --git a/components/drivers/led/SConscript b/components/drivers/led/SConscript
index 915772d78b..276ca0bd8b 100644
--- a/components/drivers/led/SConscript
+++ b/components/drivers/led/SConscript
@@ -10,6 +10,9 @@ CPPPATH = [cwd + '/../include']
 
 src = ['led.c']
 
+if GetDepend(['RT_LED_GPIO']):
+    src += ['led-gpio.c']
+
 group = DefineGroup('DeviceDrivers', src, depend = [''], CPPPATH = CPPPATH)
 
 Return('group')
diff --git a/components/drivers/led/led-gpio.c b/components/drivers/led/led-gpio.c
new file mode 100644
index 0000000000..0b6bd6875e
--- /dev/null
+++ b/components/drivers/led/led-gpio.c
@@ -0,0 +1,228 @@
+/*
+ * Copyright (c) 2006-2022, RT-Thread Development Team
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Change Logs:
+ * Date           Author       Notes
+ * 2022-3-08      GuEe-GUI     the first version
+ */
+
+#include <rtthread.h>
+#include <rtdevice.h>
+
+#define DBG_TAG "led.gpio"
+#define DBG_LVL DBG_INFO
+#include <rtdbg.h>
+
+struct gpio_led
+{
+    struct rt_led_device parent;
+
+    rt_base_t pin;
+    rt_uint8_t active_val;
+};
+
+#define raw_to_gpio_led(raw) rt_container_of(raw, struct gpio_led, parent);
+
+static rt_err_t gpio_led_set_state(struct rt_led_device *led, enum rt_led_state state)
+{
+    rt_err_t err = RT_EOK;
+    struct gpio_led *gled = raw_to_gpio_led(led);
+
+    rt_pin_mode(gled->pin, PIN_MODE_OUTPUT);
+
+    switch (state)
+    {
+    case RT_LED_S_OFF:
+        rt_pin_write(gled->pin, !gled->active_val);
+        break;
+
+    case RT_LED_S_ON:
+        rt_pin_write(gled->pin, gled->active_val);
+        break;
+
+    case RT_LED_S_TOGGLE:
+        err = led->ops->get_state(led, &state);
+
+        if (!err)
+        {
+            err = led->ops->set_state(led, state == RT_LED_S_OFF ? RT_LED_S_ON : RT_LED_S_OFF);
+        }
+        break;
+
+    default:
+        return -RT_ENOSYS;
+    }
+
+    return err;
+}
+
+static rt_err_t gpio_led_get_state(struct rt_led_device *led, enum rt_led_state *out_state)
+{
+    struct gpio_led *gled = raw_to_gpio_led(led);
+
+    switch (rt_pin_read(gled->pin))
+    {
+    case PIN_LOW:
+        *out_state = RT_LED_S_OFF;
+        break;
+
+    case PIN_HIGH:
+        *out_state = RT_LED_S_ON;
+        break;
+
+    default:
+        return -RT_ERROR;
+    }
+
+    return RT_EOK;
+}
+
+const static struct rt_led_ops gpio_led_ops =
+{
+    .set_state = gpio_led_set_state,
+    .get_state = gpio_led_get_state,
+};
+
+static rt_err_t ofw_append_gpio_led(struct rt_ofw_node *np)
+{
+    rt_err_t err;
+    enum rt_led_state led_state = RT_LED_S_OFF;
+    const char *propname, *state, *trigger;
+    struct gpio_led *gled = rt_malloc(sizeof(*gled));
+
+    if (!gled)
+    {
+        return -RT_ENOMEM;
+    }
+
+    gled->pin = rt_ofw_get_named_pin(np, RT_NULL, 0, RT_NULL, &gled->active_val);
+
+    if (gled->pin < 0)
+    {
+        err = gled->pin;
+
+        goto _fail;
+    }
+
+    gled->parent.ops = &gpio_led_ops;
+
+    if ((err = rt_hw_led_register(&gled->parent)))
+    {
+        goto _fail;
+    }
+
+    if (!rt_ofw_prop_read_string(np, "default-state", &state))
+    {
+        if (!rt_strcmp(state, "on"))
+        {
+            led_state = RT_LED_S_ON;
+        }
+    }
+
+    if ((propname = rt_ofw_get_prop_fuzzy_name(np, "default-trigger$")))
+    {
+        if (!rt_ofw_prop_read_string(np, propname, &trigger))
+        {
+            if (!rt_strcmp(trigger, "heartbeat") ||
+                !rt_strcmp(trigger, "timer"))
+            {
+                led_state = RT_LED_S_BLINK;
+            }
+        }
+    }
+
+    rt_led_set_state(&gled->parent, led_state);
+
+    rt_ofw_data(np) = &gled->parent;
+
+    return RT_EOK;
+
+_fail:
+    rt_free(gled);
+
+    return err;
+}
+
+static rt_err_t gpio_led_probe(struct rt_platform_device *pdev)
+{
+    rt_bool_t pinctrl_apply = RT_FALSE;
+    struct rt_ofw_node *led_np, *np = pdev->parent.ofw_node;
+
+    if (rt_ofw_prop_read_bool(np, "pinctrl-0"))
+    {
+        pinctrl_apply = RT_TRUE;
+        rt_pin_ctrl_confs_apply_by_name(&pdev->parent, RT_NULL);
+    }
+
+    rt_ofw_foreach_available_child_node(np, led_np)
+    {
+        rt_err_t err = ofw_append_gpio_led(led_np);
+
+        if (err == -RT_ENOMEM)
+        {
+            rt_ofw_node_put(led_np);
+
+            return err;
+        }
+        else if (err)
+        {
+            LOG_E("%s: create LED fail", rt_ofw_node_full_name(led_np));
+            continue;
+        }
+
+        if (!pinctrl_apply)
+        {
+            struct rt_device dev_tmp;
+
+            dev_tmp.ofw_node = led_np;
+            rt_pin_ctrl_confs_apply_by_name(&dev_tmp, RT_NULL);
+        }
+    }
+
+    return RT_EOK;
+}
+
+static rt_err_t gpio_led_remove(struct rt_platform_device *pdev)
+{
+    struct gpio_led *gled;
+    struct rt_led_device *led_dev;
+    struct rt_ofw_node *led_np, *np = pdev->parent.ofw_node;
+
+    rt_ofw_foreach_available_child_node(np, led_np)
+    {
+        led_dev = rt_ofw_data(led_np);
+
+        if (!led_dev)
+        {
+            continue;
+        }
+
+        gled = rt_container_of(led_dev, struct gpio_led, parent);
+
+        rt_ofw_data(led_np) = RT_NULL;
+
+        rt_hw_led_unregister(&gled->parent);
+
+        rt_free(gled);
+    }
+
+    return RT_EOK;
+}
+
+static const struct rt_ofw_node_id gpio_led_ofw_ids[] =
+{
+    { .compatible = "gpio-leds" },
+    { /* sentinel */ }
+};
+
+static struct rt_platform_driver gpio_led_driver =
+{
+    .name = "led-gpio",
+    .ids = gpio_led_ofw_ids,
+
+    .probe = gpio_led_probe,
+    .remove = gpio_led_remove,
+};
+RT_PLATFORM_DRIVER_EXPORT(gpio_led_driver);