187e8a7c91730afe56ab52ca04ee39dc1b0b0002
[platform/kernel/u-boot.git] / drivers / i2c / muxes / i2c-mux-uclass.c
1 /*
2  * Copyright (c) 2015 Google, Inc
3  * Written by Simon Glass <sjg@chromium.org>
4  *
5  * SPDX-License-Identifier:     GPL-2.0+
6  */
7
8 #include <common.h>
9 #include <dm.h>
10 #include <errno.h>
11 #include <i2c.h>
12 #include <dm/lists.h>
13 #include <dm/root.h>
14
15 DECLARE_GLOBAL_DATA_PTR;
16
17 /**
18  * struct i2c_mux: Information the uclass stores about an I2C mux
19  *
20  * @selected:   Currently selected mux, or -1 for none
21  * @i2c_bus: I2C bus to use for communcation
22  */
23 struct i2c_mux {
24         int selected;
25         struct udevice *i2c_bus;
26 };
27
28 /**
29  * struct i2c_mux_bus: Information about each bus the mux controls
30  *
31  * @channel: Channel number used to select this bus
32  */
33 struct i2c_mux_bus {
34         uint channel;
35 };
36
37 /* Find out the mux channel number */
38 static int i2c_mux_child_post_bind(struct udevice *dev)
39 {
40         struct i2c_mux_bus *plat = dev_get_parent_platdata(dev);
41         int channel;
42
43         channel = fdtdec_get_int(gd->fdt_blob, dev_of_offset(dev), "reg", -1);
44         if (channel < 0)
45                 return -EINVAL;
46         plat->channel = channel;
47
48         return 0;
49 }
50
51 /* Find the I2C buses selected by this mux */
52 static int i2c_mux_post_bind(struct udevice *mux)
53 {
54         ofnode node;
55         int ret;
56
57         debug("%s: %s\n", __func__, mux->name);
58         /*
59          * There is no compatible string in the sub-nodes, so we must manually
60          * bind these
61          */
62         dev_for_each_subnode(node, mux) {
63                 struct udevice *dev;
64                 const char *name;
65
66                 name = ofnode_get_name(node);
67                 ret = device_bind_driver_to_node(mux, "i2c_mux_bus_drv", name,
68                                                  node, &dev);
69                 debug("   - bind ret=%d, %s\n", ret, dev ? dev->name : NULL);
70                 if (ret)
71                         return ret;
72         }
73
74         return 0;
75 }
76
77 /* Set up the mux ready for use */
78 static int i2c_mux_post_probe(struct udevice *mux)
79 {
80         struct i2c_mux *priv = dev_get_uclass_priv(mux);
81         int ret;
82
83         debug("%s: %s\n", __func__, mux->name);
84         priv->selected = -1;
85
86         /* if parent is of i2c uclass already, we'll take that, otherwise
87          * look if we find an i2c-parent phandle
88          */
89         if (UCLASS_I2C == device_get_uclass_id(mux->parent)) {
90                 priv->i2c_bus = dev_get_parent(mux);
91                 debug("%s: bus=%p/%s\n", __func__, priv->i2c_bus,
92                       priv->i2c_bus->name);
93                 return 0;
94         }
95
96         ret = uclass_get_device_by_phandle(UCLASS_I2C, mux, "i2c-parent",
97                                            &priv->i2c_bus);
98         if (ret)
99                 return ret;
100         debug("%s: bus=%p/%s\n", __func__, priv->i2c_bus, priv->i2c_bus->name);
101
102         return 0;
103 }
104
105 int i2c_mux_select(struct udevice *dev)
106 {
107         struct i2c_mux_bus *plat = dev_get_parent_platdata(dev);
108         struct udevice *mux = dev->parent;
109         struct i2c_mux_ops *ops = i2c_mux_get_ops(mux);
110
111         if (!ops->select)
112                 return -ENOSYS;
113
114         return ops->select(mux, dev, plat->channel);
115 }
116
117 int i2c_mux_deselect(struct udevice *dev)
118 {
119         struct i2c_mux_bus *plat = dev_get_parent_platdata(dev);
120         struct udevice *mux = dev->parent;
121         struct i2c_mux_ops *ops = i2c_mux_get_ops(mux);
122
123         if (!ops->deselect)
124                 return -ENOSYS;
125
126         return ops->deselect(mux, dev, plat->channel);
127 }
128
129 static int i2c_mux_bus_set_bus_speed(struct udevice *dev, unsigned int speed)
130 {
131         struct udevice *mux = dev->parent;
132         struct i2c_mux *priv = dev_get_uclass_priv(mux);
133         int ret, ret2;
134
135         ret = i2c_mux_select(dev);
136         if (ret)
137                 return ret;
138         ret = dm_i2c_set_bus_speed(priv->i2c_bus, speed);
139         ret2 = i2c_mux_deselect(dev);
140
141         return ret ? ret : ret2;
142 }
143
144 static int i2c_mux_bus_probe(struct udevice *dev, uint chip_addr,
145                              uint chip_flags)
146 {
147         struct udevice *mux = dev->parent;
148         struct i2c_mux *priv = dev_get_uclass_priv(mux);
149         struct dm_i2c_ops *ops = i2c_get_ops(priv->i2c_bus);
150         int ret, ret2;
151
152         debug("%s: %s, bus %s\n", __func__, dev->name, priv->i2c_bus->name);
153         if (!ops->probe_chip)
154                 return -ENOSYS;
155         ret = i2c_mux_select(dev);
156         if (ret)
157                 return ret;
158         ret = ops->probe_chip(priv->i2c_bus, chip_addr, chip_flags);
159         ret2 = i2c_mux_deselect(dev);
160
161         return ret ? ret : ret2;
162 }
163
164 static int i2c_mux_bus_xfer(struct udevice *dev, struct i2c_msg *msg,
165                             int nmsgs)
166 {
167         struct udevice *mux = dev->parent;
168         struct i2c_mux *priv = dev_get_uclass_priv(mux);
169         struct dm_i2c_ops *ops = i2c_get_ops(priv->i2c_bus);
170         int ret, ret2;
171
172         debug("%s: %s, bus %s\n", __func__, dev->name, priv->i2c_bus->name);
173         if (!ops->xfer)
174                 return -ENOSYS;
175         ret = i2c_mux_select(dev);
176         if (ret)
177                 return ret;
178         ret = ops->xfer(priv->i2c_bus, msg, nmsgs);
179         ret2 = i2c_mux_deselect(dev);
180
181         return ret ? ret : ret2;
182 }
183
184 static const struct dm_i2c_ops i2c_mux_bus_ops = {
185         .xfer           = i2c_mux_bus_xfer,
186         .probe_chip     = i2c_mux_bus_probe,
187         .set_bus_speed  = i2c_mux_bus_set_bus_speed,
188 };
189
190 U_BOOT_DRIVER(i2c_mux_bus) = {
191         .name           = "i2c_mux_bus_drv",
192         .id             = UCLASS_I2C,
193         .ops    = &i2c_mux_bus_ops,
194 };
195
196 UCLASS_DRIVER(i2c_mux) = {
197         .id             = UCLASS_I2C_MUX,
198         .name           = "i2c_mux",
199         .post_bind      = i2c_mux_post_bind,
200         .post_probe     = i2c_mux_post_probe,
201         .per_device_auto_alloc_size = sizeof(struct i2c_mux),
202         .per_child_platdata_auto_alloc_size = sizeof(struct i2c_mux_bus),
203         .child_post_bind = i2c_mux_child_post_bind,
204 };