1 /*
2  * Copyright (c) 2006-2022, RT-Thread Development Team
3  *
4  * SPDX-License-Identifier: Apache-2.0
5  *
6  * Change Logs:
7  * Date           Author       Notes
8  * 2022-12-06     GuEe-GUI     first version
9  */
10 
11 #include <rtdevice.h>
12 
13 #define DBG_TAG "dev.i2c.bus"
14 #define DBG_LVL DBG_INFO
15 #include <rtdbg.h>
16 
17 static struct rt_bus i2c_bus;
18 
i2c_bus_scan_clients(struct rt_i2c_bus_device * bus)19 void i2c_bus_scan_clients(struct rt_i2c_bus_device *bus)
20 {
21 #ifdef RT_USING_OFW
22     if (bus->parent.ofw_node)
23     {
24         struct rt_ofw_node *np = bus->parent.ofw_node, *child_np, *i2c_client_np;
25 
26         rt_ofw_foreach_available_child_node(np, child_np)
27         {
28             rt_uint32_t client_addr;
29             struct rt_i2c_client *client;
30 
31             if (rt_ofw_prop_read_bool(child_np, "compatible"))
32             {
33                 i2c_client_np = child_np;
34             }
35             else
36             {
37                 /* Maybe in i2c-mux */
38                 i2c_client_np = rt_ofw_get_next_child(child_np, RT_NULL);
39 
40                 if (!rt_ofw_prop_read_bool(i2c_client_np, "compatible"))
41                 {
42                     continue;
43                 }
44             }
45 
46             client = rt_calloc(1, sizeof(*client));
47 
48             if (!client)
49             {
50                 rt_ofw_node_put(i2c_client_np);
51                 LOG_E("Not memory to create i2c client: %s",
52                         rt_ofw_node_full_name(i2c_client_np));
53 
54                 return;
55             }
56 
57             rt_ofw_prop_read_u32(i2c_client_np, "reg", &client_addr);
58 
59             client->parent.ofw_node = i2c_client_np;
60             client->name = rt_ofw_node_name(i2c_client_np);
61             client->bus = bus;
62             client->client_addr = client_addr;
63 
64             rt_i2c_device_register(client);
65 
66             if (i2c_client_np != child_np)
67             {
68                 rt_ofw_node_put(i2c_client_np);
69             }
70         }
71     }
72 #endif /* RT_USING_OFW */
73 }
74 
rt_i2c_driver_register(struct rt_i2c_driver * driver)75 rt_err_t rt_i2c_driver_register(struct rt_i2c_driver *driver)
76 {
77     RT_ASSERT(driver != RT_NULL);
78 
79     driver->parent.bus = &i2c_bus;
80 
81     return rt_driver_register(&driver->parent);
82 }
83 
rt_i2c_device_register(struct rt_i2c_client * client)84 rt_err_t rt_i2c_device_register(struct rt_i2c_client *client)
85 {
86     RT_ASSERT(client != RT_NULL);
87 
88     return rt_bus_add_device(&i2c_bus, &client->parent);
89 }
90 
i2c_match(rt_driver_t drv,rt_device_t dev)91 static rt_bool_t i2c_match(rt_driver_t drv, rt_device_t dev)
92 {
93     const struct rt_i2c_device_id *id;
94     struct rt_i2c_driver *driver = rt_container_of(drv, struct rt_i2c_driver, parent);
95     struct rt_i2c_client *client = rt_container_of(dev, struct rt_i2c_client, parent);
96 
97     if ((id = driver->ids))
98     {
99         for (; id->name[0]; ++id)
100         {
101             if (!rt_strcmp(id->name, client->name))
102             {
103                 client->id = id;
104                 client->ofw_id = RT_NULL;
105 
106                 return RT_TRUE;
107             }
108         }
109     }
110 
111 #ifdef RT_USING_OFW
112     client->ofw_id = rt_ofw_node_match(client->parent.ofw_node, driver->ofw_ids);
113 
114     if (client->ofw_id)
115     {
116         client->id = RT_NULL;
117 
118         return RT_TRUE;
119     }
120 #endif
121 
122     return RT_FALSE;
123 }
124 
i2c_probe(rt_device_t dev)125 static rt_err_t i2c_probe(rt_device_t dev)
126 {
127     rt_err_t err;
128     struct rt_i2c_driver *driver = rt_container_of(dev->drv, struct rt_i2c_driver, parent);
129     struct rt_i2c_client *client = rt_container_of(dev, struct rt_i2c_client, parent);
130 
131     if (!client->bus)
132     {
133         return -RT_EINVAL;
134     }
135 
136     err = driver->probe(client);
137 
138     return err;
139 }
140 
i2c_remove(rt_device_t dev)141 static rt_err_t i2c_remove(rt_device_t dev)
142 {
143     struct rt_i2c_driver *driver = rt_container_of(dev->drv, struct rt_i2c_driver, parent);
144     struct rt_i2c_client *client = rt_container_of(dev, struct rt_i2c_client, parent);
145 
146     if (driver && driver->remove)
147     {
148         driver->remove(client);
149     }
150 
151     return RT_EOK;
152 }
153 
i2c_shutdown(rt_device_t dev)154 static rt_err_t i2c_shutdown(rt_device_t dev)
155 {
156     struct rt_i2c_driver *driver = rt_container_of(dev->drv, struct rt_i2c_driver, parent);
157     struct rt_i2c_client *client = rt_container_of(dev, struct rt_i2c_client, parent);
158 
159     if (driver && driver->shutdown)
160     {
161         driver->shutdown(client);
162     }
163 
164     return RT_EOK;
165 }
166 
167 static struct rt_bus i2c_bus =
168 {
169     .name = "i2c",
170     .match = i2c_match,
171     .probe = i2c_probe,
172     .remove = i2c_remove,
173     .shutdown = i2c_shutdown,
174 };
175 
i2c_bus_init(void)176 static int i2c_bus_init(void)
177 {
178     rt_bus_register(&i2c_bus);
179 
180     return 0;
181 }
182 INIT_CORE_EXPORT(i2c_bus_init);
183