first commit for chrg

2025-08-16 22:58:22 +08:00
commit 52a3ed5862
2306 changed files with 1021208 additions and 0 deletions
@@ -0,0 +1,212 @@
+/*
+ * Copyright (c) 2006-2022, RT-Thread Development Team
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Change Logs:
+ * Date           Author       Notes
+ * 2022-12-06     GuEe-GUI     first version
+ */
+
+#include "dev_spi_dm.h"
+
+#define DBG_TAG "spi.bus"
+#define DBG_LVL DBG_INFO
+#include <rtdbg.h>
+
+extern rt_err_t rt_spidev_device_init(struct rt_spi_device *dev, const char *name);
+
+static struct rt_bus spi_bus;
+
+void spi_bus_scan_devices(struct rt_spi_bus *bus)
+{
+#ifdef RT_USING_OFW
+    if (bus->parent.ofw_node)
+    {
+        struct rt_ofw_node *np = bus->parent.ofw_node, *spi_dev_np;
+
+        rt_ofw_foreach_available_child_node(np, spi_dev_np)
+        {
+            struct rt_spi_device *spi_dev;
+
+            if (!rt_ofw_prop_read_bool(spi_dev_np, "compatible"))
+            {
+                continue;
+            }
+
+            if ((bus->mode & RT_SPI_BUS_MODE_SPI) == RT_SPI_BUS_MODE_SPI)
+            {
+                spi_dev = rt_calloc(1, sizeof(struct rt_spi_device));
+            }
+            else if ((bus->mode & RT_SPI_BUS_MODE_QSPI) == RT_SPI_BUS_MODE_QSPI)
+            {
+                spi_dev = rt_calloc(1, sizeof(struct rt_qspi_device));
+            }
+            else
+            {
+                LOG_E("Unknown bus mode = %x", bus->mode);
+                RT_ASSERT(0);
+            }
+
+            if (!spi_dev)
+            {
+                rt_ofw_node_put(spi_dev_np);
+                LOG_E("Not memory to create spi device: %s",
+                        rt_ofw_node_full_name(spi_dev_np));
+
+                return;
+            }
+
+            spi_dev->parent.ofw_node = spi_dev_np;
+            spi_dev->parent.type = RT_Device_Class_Unknown;
+            spi_dev->name = rt_ofw_node_name(spi_dev_np);
+            spi_dev->bus = bus;
+
+            rt_dm_dev_set_name(&spi_dev->parent, rt_ofw_node_full_name(spi_dev_np));
+
+            if (spi_device_ofw_parse(spi_dev))
+            {
+                continue;
+            }
+
+            rt_spi_device_register(spi_dev);
+        }
+    }
+#endif /* RT_USING_OFW */
+}
+
+rt_err_t rt_spi_driver_register(struct rt_spi_driver *driver)
+{
+    RT_ASSERT(driver != RT_NULL);
+
+    driver->parent.bus = &spi_bus;
+
+    return rt_driver_register(&driver->parent);
+}
+
+rt_err_t rt_spi_device_register(struct rt_spi_device *device)
+{
+    RT_ASSERT(device != RT_NULL);
+
+    return rt_bus_add_device(&spi_bus, &device->parent);
+}
+
+static rt_bool_t spi_match(rt_driver_t drv, rt_device_t dev)
+{
+    const struct rt_spi_device_id *id;
+    struct rt_spi_driver *driver = rt_container_of(drv, struct rt_spi_driver, parent);
+    struct rt_spi_device *device = rt_container_of(dev, struct rt_spi_device, parent);
+
+    if ((id = driver->ids))
+    {
+        for (; id->name[0]; ++id)
+        {
+            if (!rt_strcmp(id->name, device->name))
+            {
+                device->id = id;
+                device->ofw_id = RT_NULL;
+
+                return RT_TRUE;
+            }
+        }
+    }
+
+#ifdef RT_USING_OFW
+    device->ofw_id = rt_ofw_node_match(device->parent.ofw_node, driver->ofw_ids);
+
+    if (device->ofw_id)
+    {
+        device->id = RT_NULL;
+
+        return RT_TRUE;
+    }
+#endif
+
+    return RT_FALSE;
+}
+
+static rt_err_t spi_probe(rt_device_t dev)
+{
+    rt_err_t err;
+    struct rt_spi_bus *bus;
+    struct rt_spi_driver *driver = rt_container_of(dev->drv, struct rt_spi_driver, parent);
+    struct rt_spi_device *device = rt_container_of(dev, struct rt_spi_device, parent);
+
+    if (!device->bus)
+    {
+        return -RT_EINVAL;
+    }
+
+    err = driver->probe(device);
+
+    if (err)
+    {
+        return err;
+    }
+
+    bus = device->bus;
+
+    if (bus->cs_pins[0] >= 0)
+    {
+        device->cs_pin = bus->cs_pins[device->chip_select[0]];
+
+        rt_pin_mode(device->cs_pin, PIN_MODE_OUTPUT);
+    }
+    else
+    {
+        device->cs_pin = PIN_NONE;
+    }
+
+    /* Driver not register SPI device to system */
+    if (device->parent.type == RT_Device_Class_Unknown)
+    {
+        rt_spidev_device_init(device, rt_dm_dev_get_name(&device->parent));
+    }
+
+    return err;
+}
+
+static rt_err_t spi_remove(rt_device_t dev)
+{
+    struct rt_spi_driver *driver = rt_container_of(dev->drv, struct rt_spi_driver, parent);
+    struct rt_spi_device *device = rt_container_of(dev, struct rt_spi_device, parent);
+
+    if (driver && driver->remove)
+    {
+        driver->remove(device);
+    }
+    rt_free(device);
+
+    return RT_EOK;
+}
+
+static rt_err_t spi_shutdown(rt_device_t dev)
+{
+    struct rt_spi_driver *driver = rt_container_of(dev->drv, struct rt_spi_driver, parent);
+    struct rt_spi_device *device = rt_container_of(dev, struct rt_spi_device, parent);
+
+    if (driver && driver->shutdown)
+    {
+        driver->shutdown(device);
+    }
+    rt_free(device);
+
+    return RT_EOK;
+}
+
+static struct rt_bus spi_bus =
+{
+    .name = "spi",
+    .match = spi_match,
+    .probe = spi_probe,
+    .remove = spi_remove,
+    .shutdown = spi_shutdown,
+};
+
+static int spi_bus_init(void)
+{
+    rt_bus_register(&spi_bus);
+
+    return 0;
+}
+INIT_CORE_EXPORT(spi_bus_init);