ce935656f064ce8c6001d04b454effaaccd13b52
[platform/kernel/u-boot.git] / drivers / pinctrl / rockchip / pinctrl-rockchip-core.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2019 Rockchip Electronics Co., Ltd
4  */
5
6 #include <common.h>
7 #include <dm.h>
8 #include <dm/pinctrl.h>
9 #include <regmap.h>
10 #include <syscon.h>
11 #include <fdtdec.h>
12
13 #include "pinctrl-rockchip.h"
14
15 #define MAX_ROCKCHIP_PINS_ENTRIES       30
16 #define MAX_ROCKCHIP_GPIO_PER_BANK      32
17 #define RK_FUNC_GPIO                    0
18
19 static int rockchip_verify_config(struct udevice *dev, u32 bank, u32 pin)
20 {
21         struct rockchip_pinctrl_priv *priv = dev_get_priv(dev);
22         struct rockchip_pin_ctrl *ctrl = priv->ctrl;
23
24         if (bank >= ctrl->nr_banks) {
25                 debug("pin conf bank %d >= nbanks %d\n", bank, ctrl->nr_banks);
26                 return -EINVAL;
27         }
28
29         if (pin >= MAX_ROCKCHIP_GPIO_PER_BANK) {
30                 debug("pin conf pin %d >= %d\n", pin,
31                       MAX_ROCKCHIP_GPIO_PER_BANK);
32                 return -EINVAL;
33         }
34
35         return 0;
36 }
37
38 static void rockchip_get_recalced_mux(struct rockchip_pin_bank *bank, int pin,
39                                       int *reg, u8 *bit, int *mask)
40 {
41         struct rockchip_pinctrl_priv *priv = bank->priv;
42         struct rockchip_pin_ctrl *ctrl = priv->ctrl;
43         struct rockchip_mux_recalced_data *data;
44         int i;
45
46         for (i = 0; i < ctrl->niomux_recalced; i++) {
47                 data = &ctrl->iomux_recalced[i];
48                 if (data->num == bank->bank_num &&
49                     data->pin == pin)
50                         break;
51         }
52
53         if (i >= ctrl->niomux_recalced)
54                 return;
55
56         *reg = data->reg;
57         *mask = data->mask;
58         *bit = data->bit;
59 }
60
61 static bool rockchip_get_mux_route(struct rockchip_pin_bank *bank, int pin,
62                                    int mux, u32 *reg, u32 *value)
63 {
64         struct rockchip_pinctrl_priv *priv = bank->priv;
65         struct rockchip_pin_ctrl *ctrl = priv->ctrl;
66         struct rockchip_mux_route_data *data;
67         int i;
68
69         for (i = 0; i < ctrl->niomux_routes; i++) {
70                 data = &ctrl->iomux_routes[i];
71                 if (data->bank_num == bank->bank_num &&
72                     data->pin == pin && data->func == mux)
73                         break;
74         }
75
76         if (i >= ctrl->niomux_routes)
77                 return false;
78
79         *reg = data->route_offset;
80         *value = data->route_val;
81
82         return true;
83 }
84
85 static int rockchip_get_mux_data(int mux_type, int pin, u8 *bit, int *mask)
86 {
87         int offset = 0;
88
89         if (mux_type & IOMUX_WIDTH_4BIT) {
90                 if ((pin % 8) >= 4)
91                         offset = 0x4;
92                 *bit = (pin % 4) * 4;
93                 *mask = 0xf;
94         } else if (mux_type & IOMUX_WIDTH_3BIT) {
95                 /*
96                  * pin0 ~ pin4 are at first register, and
97                  * pin5 ~ pin7 are at second register.
98                  */
99                 if ((pin % 8) >= 5)
100                         offset = 0x4;
101                 *bit = (pin % 8 % 5) * 3;
102                 *mask = 0x7;
103         } else {
104                 *bit = (pin % 8) * 2;
105                 *mask = 0x3;
106         }
107
108         return offset;
109 }
110
111 static int rockchip_get_mux(struct rockchip_pin_bank *bank, int pin)
112 {
113         struct rockchip_pinctrl_priv *priv = bank->priv;
114         int iomux_num = (pin / 8);
115         struct regmap *regmap;
116         unsigned int val;
117         int reg, ret, mask, mux_type;
118         u8 bit;
119
120         if (iomux_num > 3)
121                 return -EINVAL;
122
123         if (bank->iomux[iomux_num].type & IOMUX_UNROUTED) {
124                 debug("pin %d is unrouted\n", pin);
125                 return -EINVAL;
126         }
127
128         if (bank->iomux[iomux_num].type & IOMUX_GPIO_ONLY)
129                 return RK_FUNC_GPIO;
130
131         regmap = (bank->iomux[iomux_num].type & IOMUX_SOURCE_PMU)
132                                 ? priv->regmap_pmu : priv->regmap_base;
133
134         /* get basic quadrupel of mux registers and the correct reg inside */
135         mux_type = bank->iomux[iomux_num].type;
136         reg = bank->iomux[iomux_num].offset;
137         reg += rockchip_get_mux_data(mux_type, pin, &bit, &mask);
138
139         if (bank->recalced_mask & BIT(pin))
140                 rockchip_get_recalced_mux(bank, pin, &reg, &bit, &mask);
141
142         ret = regmap_read(regmap, reg, &val);
143         if (ret)
144                 return ret;
145
146         return ((val >> bit) & mask);
147 }
148
149 static int rockchip_pinctrl_get_gpio_mux(struct udevice *dev, int banknum,
150                                          int index)
151 {       struct rockchip_pinctrl_priv *priv = dev_get_priv(dev);
152         struct rockchip_pin_ctrl *ctrl = priv->ctrl;
153
154         return rockchip_get_mux(&ctrl->pin_banks[banknum], index);
155 }
156
157 static int rockchip_verify_mux(struct rockchip_pin_bank *bank,
158                                int pin, int mux)
159 {
160         int iomux_num = (pin / 8);
161
162         if (iomux_num > 3)
163                 return -EINVAL;
164
165         if (bank->iomux[iomux_num].type & IOMUX_UNROUTED) {
166                 debug("pin %d is unrouted\n", pin);
167                 return -EINVAL;
168         }
169
170         if (bank->iomux[iomux_num].type & IOMUX_GPIO_ONLY) {
171                 if (mux != IOMUX_GPIO_ONLY) {
172                         debug("pin %d only supports a gpio mux\n", pin);
173                         return -ENOTSUPP;
174                 }
175         }
176
177         return 0;
178 }
179
180 /*
181  * Set a new mux function for a pin.
182  *
183  * The register is divided into the upper and lower 16 bit. When changing
184  * a value, the previous register value is not read and changed. Instead
185  * it seems the changed bits are marked in the upper 16 bit, while the
186  * changed value gets set in the same offset in the lower 16 bit.
187  * All pin settings seem to be 2 bit wide in both the upper and lower
188  * parts.
189  * @bank: pin bank to change
190  * @pin: pin to change
191  * @mux: new mux function to set
192  */
193 static int rockchip_set_mux(struct rockchip_pin_bank *bank, int pin, int mux)
194 {
195         struct rockchip_pinctrl_priv *priv = bank->priv;
196         int iomux_num = (pin / 8);
197         struct regmap *regmap;
198         int reg, ret, mask, mux_type;
199         u8 bit;
200         u32 data, route_reg, route_val;
201
202         ret = rockchip_verify_mux(bank, pin, mux);
203         if (ret < 0)
204                 return ret;
205
206         if (bank->iomux[iomux_num].type & IOMUX_GPIO_ONLY)
207                 return 0;
208
209         debug("setting mux of GPIO%d-%d to %d\n", bank->bank_num, pin, mux);
210
211         regmap = (bank->iomux[iomux_num].type & IOMUX_SOURCE_PMU)
212                                 ? priv->regmap_pmu : priv->regmap_base;
213
214         /* get basic quadrupel of mux registers and the correct reg inside */
215         mux_type = bank->iomux[iomux_num].type;
216         reg = bank->iomux[iomux_num].offset;
217         reg += rockchip_get_mux_data(mux_type, pin, &bit, &mask);
218
219         if (bank->recalced_mask & BIT(pin))
220                 rockchip_get_recalced_mux(bank, pin, &reg, &bit, &mask);
221
222         if (bank->route_mask & BIT(pin)) {
223                 if (rockchip_get_mux_route(bank, pin, mux, &route_reg,
224                                            &route_val)) {
225                         ret = regmap_write(regmap, route_reg, route_val);
226                         if (ret)
227                                 return ret;
228                 }
229         }
230
231         if (mux_type & IOMUX_WRITABLE_32BIT) {
232                 regmap_read(regmap, reg, &data);
233                 data &= ~(mask << bit);
234         } else {
235                 data = (mask << (bit + 16));
236         }
237
238         data |= (mux & mask) << bit;
239         ret = regmap_write(regmap, reg, data);
240
241         return ret;
242 }
243
244 static int rockchip_perpin_drv_list[DRV_TYPE_MAX][8] = {
245         { 2, 4, 8, 12, -1, -1, -1, -1 },
246         { 3, 6, 9, 12, -1, -1, -1, -1 },
247         { 5, 10, 15, 20, -1, -1, -1, -1 },
248         { 4, 6, 8, 10, 12, 14, 16, 18 },
249         { 4, 7, 10, 13, 16, 19, 22, 26 }
250 };
251
252 static int rockchip_set_drive_perpin(struct rockchip_pin_bank *bank,
253                                      int pin_num, int strength)
254 {
255         struct rockchip_pinctrl_priv *priv = bank->priv;
256         struct rockchip_pin_ctrl *ctrl = priv->ctrl;
257         struct regmap *regmap;
258         int reg, ret, i;
259         u32 data, rmask_bits, temp;
260         u8 bit;
261         /* Where need to clean the special mask for rockchip_perpin_drv_list */
262         int drv_type = bank->drv[pin_num / 8].drv_type & (~DRV_TYPE_IO_MASK);
263
264         debug("setting drive of GPIO%d-%d to %d\n", bank->bank_num,
265               pin_num, strength);
266
267         ctrl->drv_calc_reg(bank, pin_num, &regmap, &reg, &bit);
268
269         ret = -EINVAL;
270         for (i = 0; i < ARRAY_SIZE(rockchip_perpin_drv_list[drv_type]); i++) {
271                 if (rockchip_perpin_drv_list[drv_type][i] == strength) {
272                         ret = i;
273                         break;
274                 } else if (rockchip_perpin_drv_list[drv_type][i] < 0) {
275                         ret = rockchip_perpin_drv_list[drv_type][i];
276                         break;
277                 }
278         }
279
280         if (ret < 0) {
281                 debug("unsupported driver strength %d\n", strength);
282                 return ret;
283         }
284
285         switch (drv_type) {
286         case DRV_TYPE_IO_1V8_3V0_AUTO:
287         case DRV_TYPE_IO_3V3_ONLY:
288                 rmask_bits = ROCKCHIP_DRV_3BITS_PER_PIN;
289                 switch (bit) {
290                 case 0 ... 12:
291                         /* regular case, nothing to do */
292                         break;
293                 case 15:
294                         /*
295                          * drive-strength offset is special, as it is spread
296                          * over 2 registers, the bit data[15] contains bit 0
297                          * of the value while temp[1:0] contains bits 2 and 1
298                          */
299                         data = (ret & 0x1) << 15;
300                         temp = (ret >> 0x1) & 0x3;
301
302                         data |= BIT(31);
303                         ret = regmap_write(regmap, reg, data);
304                         if (ret)
305                                 return ret;
306
307                         temp |= (0x3 << 16);
308                         reg += 0x4;
309                         ret = regmap_write(regmap, reg, temp);
310
311                         return ret;
312                 case 18 ... 21:
313                         /* setting fully enclosed in the second register */
314                         reg += 4;
315                         bit -= 16;
316                         break;
317                 default:
318                         debug("unsupported bit: %d for pinctrl drive type: %d\n",
319                               bit, drv_type);
320                         return -EINVAL;
321                 }
322                 break;
323         case DRV_TYPE_IO_DEFAULT:
324         case DRV_TYPE_IO_1V8_OR_3V0:
325         case DRV_TYPE_IO_1V8_ONLY:
326                 rmask_bits = ROCKCHIP_DRV_BITS_PER_PIN;
327                 break;
328         default:
329                 debug("unsupported pinctrl drive type: %d\n",
330                       drv_type);
331                 return -EINVAL;
332         }
333
334         if (bank->drv[pin_num / 8].drv_type & DRV_TYPE_WRITABLE_32BIT) {
335                 regmap_read(regmap, reg, &data);
336                 data &= ~(((1 << rmask_bits) - 1) << bit);
337         } else {
338                 /* enable the write to the equivalent lower bits */
339                 data = ((1 << rmask_bits) - 1) << (bit + 16);
340         }
341
342         data |= (ret << bit);
343         ret = regmap_write(regmap, reg, data);
344         return ret;
345 }
346
347 static int rockchip_pull_list[PULL_TYPE_MAX][4] = {
348         {
349                 PIN_CONFIG_BIAS_DISABLE,
350                 PIN_CONFIG_BIAS_PULL_UP,
351                 PIN_CONFIG_BIAS_PULL_DOWN,
352                 PIN_CONFIG_BIAS_BUS_HOLD
353         },
354         {
355                 PIN_CONFIG_BIAS_DISABLE,
356                 PIN_CONFIG_BIAS_PULL_DOWN,
357                 PIN_CONFIG_BIAS_DISABLE,
358                 PIN_CONFIG_BIAS_PULL_UP
359         },
360 };
361
362 static int rockchip_set_pull(struct rockchip_pin_bank *bank,
363                              int pin_num, int pull)
364 {
365         struct rockchip_pinctrl_priv *priv = bank->priv;
366         struct rockchip_pin_ctrl *ctrl = priv->ctrl;
367         struct regmap *regmap;
368         int reg, ret, i, pull_type;
369         u8 bit;
370         u32 data;
371
372         debug("setting pull of GPIO%d-%d to %d\n", bank->bank_num,
373               pin_num, pull);
374
375         ctrl->pull_calc_reg(bank, pin_num, &regmap, &reg, &bit);
376
377         switch (ctrl->type) {
378         case RK3036:
379         case RK3128:
380                 data = BIT(bit + 16);
381                 if (pull == PIN_CONFIG_BIAS_DISABLE)
382                         data |= BIT(bit);
383                 ret = regmap_write(regmap, reg, data);
384                 break;
385         case RV1108:
386         case RK3188:
387         case RK3288:
388         case RK3368:
389         case RK3399:
390                 /*
391                  * Where need to clean the special mask for
392                  * rockchip_pull_list.
393                  */
394                 pull_type = bank->pull_type[pin_num / 8] & (~PULL_TYPE_IO_MASK);
395                 ret = -EINVAL;
396                 for (i = 0; i < ARRAY_SIZE(rockchip_pull_list[pull_type]);
397                         i++) {
398                         if (rockchip_pull_list[pull_type][i] == pull) {
399                                 ret = i;
400                                 break;
401                         }
402                 }
403
404                 if (ret < 0) {
405                         debug("unsupported pull setting %d\n", pull);
406                         return ret;
407                 }
408
409                 if (bank->pull_type[pin_num / 8] & PULL_TYPE_WRITABLE_32BIT) {
410                         regmap_read(regmap, reg, &data);
411                         data &= ~(((1 << ROCKCHIP_PULL_BITS_PER_PIN) - 1) << bit);
412                 } else {
413                         /* enable the write to the equivalent lower bits */
414                         data = ((1 << ROCKCHIP_PULL_BITS_PER_PIN) - 1) << (bit + 16);
415                 }
416
417                 data |= (ret << bit);
418                 ret = regmap_write(regmap, reg, data);
419                 break;
420         default:
421                 debug("unsupported pinctrl type\n");
422                 return -EINVAL;
423         }
424
425         return ret;
426 }
427
428 static int rockchip_set_schmitt(struct rockchip_pin_bank *bank,
429                                 int pin_num, int enable)
430 {
431         struct rockchip_pinctrl_priv *priv = bank->priv;
432         struct rockchip_pin_ctrl *ctrl = priv->ctrl;
433         struct regmap *regmap;
434         int reg, ret;
435         u8 bit;
436         u32 data;
437
438         debug("setting input schmitt of GPIO%d-%d to %d\n", bank->bank_num,
439               pin_num, enable);
440
441         ret = ctrl->schmitt_calc_reg(bank, pin_num, &regmap, &reg, &bit);
442         if (ret)
443                 return ret;
444
445         /* enable the write to the equivalent lower bits */
446         data = BIT(bit + 16) | (enable << bit);
447
448         return regmap_write(regmap, reg, data);
449 }
450
451 /*
452  * Pinconf_ops handling
453  */
454 static bool rockchip_pinconf_pull_valid(struct rockchip_pin_ctrl *ctrl,
455                                         unsigned int pull)
456 {
457         switch (ctrl->type) {
458         case RK3036:
459         case RK3128:
460                 return (pull == PIN_CONFIG_BIAS_PULL_PIN_DEFAULT ||
461                         pull == PIN_CONFIG_BIAS_DISABLE);
462         case RV1108:
463         case RK3188:
464         case RK3288:
465         case RK3368:
466         case RK3399:
467                 return (pull != PIN_CONFIG_BIAS_PULL_PIN_DEFAULT);
468         }
469
470         return false;
471 }
472
473 /* set the pin config settings for a specified pin */
474 static int rockchip_pinconf_set(struct rockchip_pin_bank *bank,
475                                 u32 pin, u32 param, u32 arg)
476 {
477         struct rockchip_pinctrl_priv *priv = bank->priv;
478         struct rockchip_pin_ctrl *ctrl = priv->ctrl;
479         int rc;
480
481         switch (param) {
482         case PIN_CONFIG_BIAS_DISABLE:
483                 rc =  rockchip_set_pull(bank, pin, param);
484                 if (rc)
485                         return rc;
486                 break;
487
488         case PIN_CONFIG_BIAS_PULL_UP:
489         case PIN_CONFIG_BIAS_PULL_DOWN:
490         case PIN_CONFIG_BIAS_PULL_PIN_DEFAULT:
491         case PIN_CONFIG_BIAS_BUS_HOLD:
492                 if (!rockchip_pinconf_pull_valid(ctrl, param))
493                         return -ENOTSUPP;
494
495                 if (!arg)
496                         return -EINVAL;
497
498                 rc = rockchip_set_pull(bank, pin, param);
499                 if (rc)
500                         return rc;
501                 break;
502
503         case PIN_CONFIG_DRIVE_STRENGTH:
504                 if (!ctrl->drv_calc_reg)
505                         return -ENOTSUPP;
506
507                 rc = rockchip_set_drive_perpin(bank, pin, arg);
508                 if (rc < 0)
509                         return rc;
510                 break;
511
512         case PIN_CONFIG_INPUT_SCHMITT_ENABLE:
513                 if (!ctrl->schmitt_calc_reg)
514                         return -ENOTSUPP;
515
516                 rc = rockchip_set_schmitt(bank, pin, arg);
517                 if (rc < 0)
518                         return rc;
519                 break;
520
521         default:
522                 break;
523         }
524
525         return 0;
526 }
527
528 static const struct pinconf_param rockchip_conf_params[] = {
529         { "bias-disable", PIN_CONFIG_BIAS_DISABLE, 0 },
530         { "bias-bus-hold", PIN_CONFIG_BIAS_BUS_HOLD, 0 },
531         { "bias-pull-up", PIN_CONFIG_BIAS_PULL_UP, 1 },
532         { "bias-pull-down", PIN_CONFIG_BIAS_PULL_DOWN, 1 },
533         { "drive-strength", PIN_CONFIG_DRIVE_STRENGTH, 0 },
534         { "input-enable", PIN_CONFIG_INPUT_ENABLE, 1 },
535         { "input-disable", PIN_CONFIG_INPUT_ENABLE, 0 },
536         { "input-schmitt-disable", PIN_CONFIG_INPUT_SCHMITT_ENABLE, 0 },
537         { "input-schmitt-enable", PIN_CONFIG_INPUT_SCHMITT_ENABLE, 1 },
538 };
539
540 static int rockchip_pinconf_prop_name_to_param(const char *property,
541                                                u32 *default_value)
542 {
543         const struct pinconf_param *p, *end;
544
545         p = rockchip_conf_params;
546         end = p + sizeof(rockchip_conf_params) / sizeof(struct pinconf_param);
547
548         /* See if this pctldev supports this parameter */
549         for (; p < end; p++) {
550                 if (!strcmp(property, p->property)) {
551                         *default_value = p->default_value;
552                         return p->param;
553                 }
554         }
555
556         *default_value = 0;
557         return -EPERM;
558 }
559
560 static int rockchip_pinctrl_set_state(struct udevice *dev,
561                                       struct udevice *config)
562 {
563         struct rockchip_pinctrl_priv *priv = dev_get_priv(dev);
564         struct rockchip_pin_ctrl *ctrl = priv->ctrl;
565         u32 cells[MAX_ROCKCHIP_PINS_ENTRIES * 4];
566         u32 bank, pin, mux, conf, arg, default_val;
567         int ret, count, i;
568         const char *prop_name;
569         const void *value;
570         int prop_len, param;
571         const u32 *data;
572         ofnode node;
573 #ifdef CONFIG_OF_LIVE
574         const struct device_node *np;
575         struct property *pp;
576 #else
577         int property_offset, pcfg_node;
578         const void *blob = gd->fdt_blob;
579 #endif
580         data = dev_read_prop(config, "rockchip,pins", &count);
581         if (count < 0) {
582                 debug("%s: bad array size %d\n", __func__, count);
583                 return -EINVAL;
584         }
585
586         count /= sizeof(u32);
587         if (count > MAX_ROCKCHIP_PINS_ENTRIES * 4) {
588                 debug("%s: unsupported pins array count %d\n",
589                       __func__, count);
590                 return -EINVAL;
591         }
592
593         for (i = 0; i < count; i++)
594                 cells[i] = fdt32_to_cpu(data[i]);
595
596         for (i = 0; i < (count >> 2); i++) {
597                 bank = cells[4 * i + 0];
598                 pin = cells[4 * i + 1];
599                 mux = cells[4 * i + 2];
600                 conf = cells[4 * i + 3];
601
602                 ret = rockchip_verify_config(dev, bank, pin);
603                 if (ret)
604                         return ret;
605
606                 ret = rockchip_set_mux(&ctrl->pin_banks[bank], pin, mux);
607                 if (ret)
608                         return ret;
609
610                 node = ofnode_get_by_phandle(conf);
611                 if (!ofnode_valid(node))
612                         return -ENODEV;
613 #ifdef CONFIG_OF_LIVE
614                 np = ofnode_to_np(node);
615                 for (pp = np->properties; pp; pp = pp->next) {
616                         prop_name = pp->name;
617                         prop_len = pp->length;
618                         value = pp->value;
619 #else
620                 pcfg_node = ofnode_to_offset(node);
621                 fdt_for_each_property_offset(property_offset, blob, pcfg_node) {
622                         value = fdt_getprop_by_offset(blob, property_offset,
623                                                       &prop_name, &prop_len);
624                         if (!value)
625                                 return -ENOENT;
626 #endif
627                         param = rockchip_pinconf_prop_name_to_param(prop_name,
628                                                                     &default_val);
629                         if (param < 0)
630                                 break;
631
632                         if (prop_len >= sizeof(fdt32_t))
633                                 arg = fdt32_to_cpu(*(fdt32_t *)value);
634                         else
635                                 arg = default_val;
636
637                         ret = rockchip_pinconf_set(&ctrl->pin_banks[bank], pin,
638                                                    param, arg);
639                         if (ret) {
640                                 debug("%s: rockchip_pinconf_set fail: %d\n",
641                                       __func__, ret);
642                                 return ret;
643                         }
644                 }
645         }
646
647         return 0;
648 }
649
650 const struct pinctrl_ops rockchip_pinctrl_ops = {
651         .set_state                      = rockchip_pinctrl_set_state,
652         .get_gpio_mux                   = rockchip_pinctrl_get_gpio_mux,
653 };
654
655 /* retrieve the soc specific data */
656 static struct rockchip_pin_ctrl *rockchip_pinctrl_get_soc_data(struct udevice *dev)
657 {
658         struct rockchip_pinctrl_priv *priv = dev_get_priv(dev);
659         struct rockchip_pin_ctrl *ctrl =
660                         (struct rockchip_pin_ctrl *)dev_get_driver_data(dev);
661         struct rockchip_pin_bank *bank;
662         int grf_offs, pmu_offs, drv_grf_offs, drv_pmu_offs, i, j;
663
664         grf_offs = ctrl->grf_mux_offset;
665         pmu_offs = ctrl->pmu_mux_offset;
666         drv_pmu_offs = ctrl->pmu_drv_offset;
667         drv_grf_offs = ctrl->grf_drv_offset;
668         bank = ctrl->pin_banks;
669
670         for (i = 0; i < ctrl->nr_banks; ++i, ++bank) {
671                 int bank_pins = 0;
672
673                 bank->priv = priv;
674                 bank->pin_base = ctrl->nr_pins;
675                 ctrl->nr_pins += bank->nr_pins;
676
677                 /* calculate iomux and drv offsets */
678                 for (j = 0; j < 4; j++) {
679                         struct rockchip_iomux *iom = &bank->iomux[j];
680                         struct rockchip_drv *drv = &bank->drv[j];
681                         int inc;
682
683                         if (bank_pins >= bank->nr_pins)
684                                 break;
685
686                         /* preset iomux offset value, set new start value */
687                         if (iom->offset >= 0) {
688                                 if (iom->type & IOMUX_SOURCE_PMU)
689                                         pmu_offs = iom->offset;
690                                 else
691                                         grf_offs = iom->offset;
692                         } else { /* set current iomux offset */
693                                 iom->offset = (iom->type & IOMUX_SOURCE_PMU) ?
694                                                         pmu_offs : grf_offs;
695                         }
696
697                         /* preset drv offset value, set new start value */
698                         if (drv->offset >= 0) {
699                                 if (iom->type & IOMUX_SOURCE_PMU)
700                                         drv_pmu_offs = drv->offset;
701                                 else
702                                         drv_grf_offs = drv->offset;
703                         } else { /* set current drv offset */
704                                 drv->offset = (iom->type & IOMUX_SOURCE_PMU) ?
705                                                 drv_pmu_offs : drv_grf_offs;
706                         }
707
708                         debug("bank %d, iomux %d has iom_offset 0x%x drv_offset 0x%x\n",
709                               i, j, iom->offset, drv->offset);
710
711                         /*
712                          * Increase offset according to iomux width.
713                          * 4bit iomux'es are spread over two registers.
714                          */
715                         inc = (iom->type & (IOMUX_WIDTH_4BIT |
716                                             IOMUX_WIDTH_3BIT)) ? 8 : 4;
717                         if (iom->type & IOMUX_SOURCE_PMU)
718                                 pmu_offs += inc;
719                         else
720                                 grf_offs += inc;
721
722                         /*
723                          * Increase offset according to drv width.
724                          * 3bit drive-strenth'es are spread over two registers.
725                          */
726                         if ((drv->drv_type == DRV_TYPE_IO_1V8_3V0_AUTO) ||
727                             (drv->drv_type == DRV_TYPE_IO_3V3_ONLY))
728                                 inc = 8;
729                         else
730                                 inc = 4;
731
732                         if (iom->type & IOMUX_SOURCE_PMU)
733                                 drv_pmu_offs += inc;
734                         else
735                                 drv_grf_offs += inc;
736
737                         bank_pins += 8;
738                 }
739
740                 /* calculate the per-bank recalced_mask */
741                 for (j = 0; j < ctrl->niomux_recalced; j++) {
742                         int pin = 0;
743
744                         if (ctrl->iomux_recalced[j].num == bank->bank_num) {
745                                 pin = ctrl->iomux_recalced[j].pin;
746                                 bank->recalced_mask |= BIT(pin);
747                         }
748                 }
749
750                 /* calculate the per-bank route_mask */
751                 for (j = 0; j < ctrl->niomux_routes; j++) {
752                         int pin = 0;
753
754                         if (ctrl->iomux_routes[j].bank_num == bank->bank_num) {
755                                 pin = ctrl->iomux_routes[j].pin;
756                                 bank->route_mask |= BIT(pin);
757                         }
758                 }
759         }
760
761         return ctrl;
762 }
763
764 int rockchip_pinctrl_probe(struct udevice *dev)
765 {
766         struct rockchip_pinctrl_priv *priv = dev_get_priv(dev);
767         struct rockchip_pin_ctrl *ctrl;
768         struct udevice *syscon;
769         struct regmap *regmap;
770         int ret = 0;
771
772         /* get rockchip grf syscon phandle */
773         ret = uclass_get_device_by_phandle(UCLASS_SYSCON, dev, "rockchip,grf",
774                                            &syscon);
775         if (ret) {
776                 debug("unable to find rockchip,grf syscon device (%d)\n", ret);
777                 return ret;
778         }
779
780         /* get grf-reg base address */
781         regmap = syscon_get_regmap(syscon);
782         if (!regmap) {
783                 debug("unable to find rockchip grf regmap\n");
784                 return -ENODEV;
785         }
786         priv->regmap_base = regmap;
787
788         /* option: get pmu-reg base address */
789         ret = uclass_get_device_by_phandle(UCLASS_SYSCON, dev, "rockchip,pmu",
790                                            &syscon);
791         if (!ret) {
792                 /* get pmugrf-reg base address */
793                 regmap = syscon_get_regmap(syscon);
794                 if (!regmap) {
795                         debug("unable to find rockchip pmu regmap\n");
796                         return -ENODEV;
797                 }
798                 priv->regmap_pmu = regmap;
799         }
800
801         ctrl = rockchip_pinctrl_get_soc_data(dev);
802         if (!ctrl) {
803                 debug("driver data not available\n");
804                 return -EINVAL;
805         }
806
807         priv->ctrl = ctrl;
808         return 0;
809 }