rockchip: clk: rk3399: add clock support for SCLK_SPI1 and SCLK_SPI5
[platform/kernel/u-boot.git] / drivers / clk / rockchip / clk_rk3399.c
1 /*
2  * (C) Copyright 2015 Google, Inc
3  * (C) 2017 Theobroma Systems Design und Consulting GmbH
4  *
5  * SPDX-License-Identifier:     GPL-2.0
6  */
7
8 #include <common.h>
9 #include <clk-uclass.h>
10 #include <dm.h>
11 #include <dt-structs.h>
12 #include <errno.h>
13 #include <mapmem.h>
14 #include <syscon.h>
15 #include <asm/io.h>
16 #include <asm/arch/clock.h>
17 #include <asm/arch/cru_rk3399.h>
18 #include <asm/arch/hardware.h>
19 #include <dm/lists.h>
20 #include <dt-bindings/clock/rk3399-cru.h>
21
22 DECLARE_GLOBAL_DATA_PTR;
23
24 #if CONFIG_IS_ENABLED(OF_PLATDATA)
25 struct rk3399_clk_plat {
26         struct dtd_rockchip_rk3399_cru dtd;
27 };
28
29 struct rk3399_pmuclk_plat {
30         struct dtd_rockchip_rk3399_pmucru dtd;
31 };
32 #endif
33
34 struct pll_div {
35         u32 refdiv;
36         u32 fbdiv;
37         u32 postdiv1;
38         u32 postdiv2;
39         u32 frac;
40 };
41
42 #define RATE_TO_DIV(input_rate, output_rate) \
43         ((input_rate) / (output_rate) - 1);
44 #define DIV_TO_RATE(input_rate, div)    ((input_rate) / ((div) + 1))
45
46 #define PLL_DIVISORS(hz, _refdiv, _postdiv1, _postdiv2) {\
47         .refdiv = _refdiv,\
48         .fbdiv = (u32)((u64)hz * _refdiv * _postdiv1 * _postdiv2 / OSC_HZ),\
49         .postdiv1 = _postdiv1, .postdiv2 = _postdiv2};
50
51 #if defined(CONFIG_SPL_BUILD)
52 static const struct pll_div gpll_init_cfg = PLL_DIVISORS(GPLL_HZ, 2, 2, 1);
53 static const struct pll_div cpll_init_cfg = PLL_DIVISORS(CPLL_HZ, 1, 2, 2);
54 #else
55 static const struct pll_div ppll_init_cfg = PLL_DIVISORS(PPLL_HZ, 2, 2, 1);
56 #endif
57
58 static const struct pll_div apll_l_1600_cfg = PLL_DIVISORS(1600*MHz, 3, 1, 1);
59 static const struct pll_div apll_l_600_cfg = PLL_DIVISORS(600*MHz, 1, 2, 1);
60
61 static const struct pll_div *apll_l_cfgs[] = {
62         [APLL_L_1600_MHZ] = &apll_l_1600_cfg,
63         [APLL_L_600_MHZ] = &apll_l_600_cfg,
64 };
65
66 enum {
67         /* PLL_CON0 */
68         PLL_FBDIV_MASK                  = 0xfff,
69         PLL_FBDIV_SHIFT                 = 0,
70
71         /* PLL_CON1 */
72         PLL_POSTDIV2_SHIFT              = 12,
73         PLL_POSTDIV2_MASK               = 0x7 << PLL_POSTDIV2_SHIFT,
74         PLL_POSTDIV1_SHIFT              = 8,
75         PLL_POSTDIV1_MASK               = 0x7 << PLL_POSTDIV1_SHIFT,
76         PLL_REFDIV_MASK                 = 0x3f,
77         PLL_REFDIV_SHIFT                = 0,
78
79         /* PLL_CON2 */
80         PLL_LOCK_STATUS_SHIFT           = 31,
81         PLL_LOCK_STATUS_MASK            = 1 << PLL_LOCK_STATUS_SHIFT,
82         PLL_FRACDIV_MASK                = 0xffffff,
83         PLL_FRACDIV_SHIFT               = 0,
84
85         /* PLL_CON3 */
86         PLL_MODE_SHIFT                  = 8,
87         PLL_MODE_MASK                   = 3 << PLL_MODE_SHIFT,
88         PLL_MODE_SLOW                   = 0,
89         PLL_MODE_NORM,
90         PLL_MODE_DEEP,
91         PLL_DSMPD_SHIFT                 = 3,
92         PLL_DSMPD_MASK                  = 1 << PLL_DSMPD_SHIFT,
93         PLL_INTEGER_MODE                = 1,
94
95         /* PMUCRU_CLKSEL_CON0 */
96         PMU_PCLK_DIV_CON_MASK           = 0x1f,
97         PMU_PCLK_DIV_CON_SHIFT          = 0,
98
99         /* PMUCRU_CLKSEL_CON1 */
100         SPI3_PLL_SEL_SHIFT              = 7,
101         SPI3_PLL_SEL_MASK               = 1 << SPI3_PLL_SEL_SHIFT,
102         SPI3_PLL_SEL_24M                = 0,
103         SPI3_PLL_SEL_PPLL               = 1,
104         SPI3_DIV_CON_SHIFT              = 0x0,
105         SPI3_DIV_CON_MASK               = 0x7f,
106
107         /* PMUCRU_CLKSEL_CON2 */
108         I2C_DIV_CON_MASK                = 0x7f,
109         CLK_I2C8_DIV_CON_SHIFT          = 8,
110         CLK_I2C0_DIV_CON_SHIFT          = 0,
111
112         /* PMUCRU_CLKSEL_CON3 */
113         CLK_I2C4_DIV_CON_SHIFT          = 0,
114
115         /* CLKSEL_CON0 */
116         ACLKM_CORE_L_DIV_CON_SHIFT      = 8,
117         ACLKM_CORE_L_DIV_CON_MASK       = 0x1f << ACLKM_CORE_L_DIV_CON_SHIFT,
118         CLK_CORE_L_PLL_SEL_SHIFT        = 6,
119         CLK_CORE_L_PLL_SEL_MASK         = 3 << CLK_CORE_L_PLL_SEL_SHIFT,
120         CLK_CORE_L_PLL_SEL_ALPLL        = 0x0,
121         CLK_CORE_L_PLL_SEL_ABPLL        = 0x1,
122         CLK_CORE_L_PLL_SEL_DPLL         = 0x10,
123         CLK_CORE_L_PLL_SEL_GPLL         = 0x11,
124         CLK_CORE_L_DIV_MASK             = 0x1f,
125         CLK_CORE_L_DIV_SHIFT            = 0,
126
127         /* CLKSEL_CON1 */
128         PCLK_DBG_L_DIV_SHIFT            = 0x8,
129         PCLK_DBG_L_DIV_MASK             = 0x1f << PCLK_DBG_L_DIV_SHIFT,
130         ATCLK_CORE_L_DIV_SHIFT          = 0,
131         ATCLK_CORE_L_DIV_MASK           = 0x1f << ATCLK_CORE_L_DIV_SHIFT,
132
133         /* CLKSEL_CON14 */
134         PCLK_PERIHP_DIV_CON_SHIFT       = 12,
135         PCLK_PERIHP_DIV_CON_MASK        = 0x7 << PCLK_PERIHP_DIV_CON_SHIFT,
136         HCLK_PERIHP_DIV_CON_SHIFT       = 8,
137         HCLK_PERIHP_DIV_CON_MASK        = 3 << HCLK_PERIHP_DIV_CON_SHIFT,
138         ACLK_PERIHP_PLL_SEL_SHIFT       = 7,
139         ACLK_PERIHP_PLL_SEL_MASK        = 1 << ACLK_PERIHP_PLL_SEL_SHIFT,
140         ACLK_PERIHP_PLL_SEL_CPLL        = 0,
141         ACLK_PERIHP_PLL_SEL_GPLL        = 1,
142         ACLK_PERIHP_DIV_CON_SHIFT       = 0,
143         ACLK_PERIHP_DIV_CON_MASK        = 0x1f,
144
145         /* CLKSEL_CON21 */
146         ACLK_EMMC_PLL_SEL_SHIFT         = 7,
147         ACLK_EMMC_PLL_SEL_MASK          = 0x1 << ACLK_EMMC_PLL_SEL_SHIFT,
148         ACLK_EMMC_PLL_SEL_GPLL          = 0x1,
149         ACLK_EMMC_DIV_CON_SHIFT         = 0,
150         ACLK_EMMC_DIV_CON_MASK          = 0x1f,
151
152         /* CLKSEL_CON22 */
153         CLK_EMMC_PLL_SHIFT              = 8,
154         CLK_EMMC_PLL_MASK               = 0x7 << CLK_EMMC_PLL_SHIFT,
155         CLK_EMMC_PLL_SEL_GPLL           = 0x1,
156         CLK_EMMC_PLL_SEL_24M            = 0x5,
157         CLK_EMMC_DIV_CON_SHIFT          = 0,
158         CLK_EMMC_DIV_CON_MASK           = 0x7f << CLK_EMMC_DIV_CON_SHIFT,
159
160         /* CLKSEL_CON23 */
161         PCLK_PERILP0_DIV_CON_SHIFT      = 12,
162         PCLK_PERILP0_DIV_CON_MASK       = 0x7 << PCLK_PERILP0_DIV_CON_SHIFT,
163         HCLK_PERILP0_DIV_CON_SHIFT      = 8,
164         HCLK_PERILP0_DIV_CON_MASK       = 3 << HCLK_PERILP0_DIV_CON_SHIFT,
165         ACLK_PERILP0_PLL_SEL_SHIFT      = 7,
166         ACLK_PERILP0_PLL_SEL_MASK       = 1 << ACLK_PERILP0_PLL_SEL_SHIFT,
167         ACLK_PERILP0_PLL_SEL_CPLL       = 0,
168         ACLK_PERILP0_PLL_SEL_GPLL       = 1,
169         ACLK_PERILP0_DIV_CON_SHIFT      = 0,
170         ACLK_PERILP0_DIV_CON_MASK       = 0x1f,
171
172         /* CLKSEL_CON25 */
173         PCLK_PERILP1_DIV_CON_SHIFT      = 8,
174         PCLK_PERILP1_DIV_CON_MASK       = 0x7 << PCLK_PERILP1_DIV_CON_SHIFT,
175         HCLK_PERILP1_PLL_SEL_SHIFT      = 7,
176         HCLK_PERILP1_PLL_SEL_MASK       = 1 << HCLK_PERILP1_PLL_SEL_SHIFT,
177         HCLK_PERILP1_PLL_SEL_CPLL       = 0,
178         HCLK_PERILP1_PLL_SEL_GPLL       = 1,
179         HCLK_PERILP1_DIV_CON_SHIFT      = 0,
180         HCLK_PERILP1_DIV_CON_MASK       = 0x1f,
181
182         /* CLKSEL_CON26 */
183         CLK_SARADC_DIV_CON_SHIFT        = 8,
184         CLK_SARADC_DIV_CON_MASK         = 0xff << CLK_SARADC_DIV_CON_SHIFT,
185
186         /* CLKSEL_CON27 */
187         CLK_TSADC_SEL_X24M              = 0x0,
188         CLK_TSADC_SEL_SHIFT             = 15,
189         CLK_TSADC_SEL_MASK              = 1 << CLK_TSADC_SEL_SHIFT,
190         CLK_TSADC_DIV_CON_SHIFT         = 0,
191         CLK_TSADC_DIV_CON_MASK          = 0x3ff,
192
193         /* CLKSEL_CON47 & CLKSEL_CON48 */
194         ACLK_VOP_PLL_SEL_SHIFT          = 6,
195         ACLK_VOP_PLL_SEL_MASK           = 0x3 << ACLK_VOP_PLL_SEL_SHIFT,
196         ACLK_VOP_PLL_SEL_CPLL           = 0x1,
197         ACLK_VOP_DIV_CON_SHIFT          = 0,
198         ACLK_VOP_DIV_CON_MASK           = 0x1f << ACLK_VOP_DIV_CON_SHIFT,
199
200         /* CLKSEL_CON49 & CLKSEL_CON50 */
201         DCLK_VOP_DCLK_SEL_SHIFT         = 11,
202         DCLK_VOP_DCLK_SEL_MASK          = 1 << DCLK_VOP_DCLK_SEL_SHIFT,
203         DCLK_VOP_DCLK_SEL_DIVOUT        = 0,
204         DCLK_VOP_PLL_SEL_SHIFT          = 8,
205         DCLK_VOP_PLL_SEL_MASK           = 3 << DCLK_VOP_PLL_SEL_SHIFT,
206         DCLK_VOP_PLL_SEL_VPLL           = 0,
207         DCLK_VOP_DIV_CON_MASK           = 0xff,
208         DCLK_VOP_DIV_CON_SHIFT          = 0,
209
210         /* CLKSEL_CON58 */
211         CLK_SPI_PLL_SEL_WIDTH = 1,
212         CLK_SPI_PLL_SEL_MASK = ((1 < CLK_SPI_PLL_SEL_WIDTH) - 1),
213         CLK_SPI_PLL_SEL_CPLL = 0,
214         CLK_SPI_PLL_SEL_GPLL = 1,
215         CLK_SPI_PLL_DIV_CON_WIDTH = 7,
216         CLK_SPI_PLL_DIV_CON_MASK = ((1 << CLK_SPI_PLL_DIV_CON_WIDTH) - 1),
217
218         CLK_SPI5_PLL_DIV_CON_SHIFT      = 8,
219         CLK_SPI5_PLL_SEL_SHIFT          = 15,
220
221         /* CLKSEL_CON59 */
222         CLK_SPI1_PLL_SEL_SHIFT          = 15,
223         CLK_SPI1_PLL_DIV_CON_SHIFT      = 8,
224         CLK_SPI0_PLL_SEL_SHIFT          = 7,
225         CLK_SPI0_PLL_DIV_CON_SHIFT      = 0,
226
227         /* CLKSEL_CON60 */
228         CLK_SPI4_PLL_SEL_SHIFT          = 15,
229         CLK_SPI4_PLL_DIV_CON_SHIFT      = 8,
230         CLK_SPI2_PLL_SEL_SHIFT          = 7,
231         CLK_SPI2_PLL_DIV_CON_SHIFT      = 0,
232
233         /* CLKSEL_CON61 */
234         CLK_I2C_PLL_SEL_MASK            = 1,
235         CLK_I2C_PLL_SEL_CPLL            = 0,
236         CLK_I2C_PLL_SEL_GPLL            = 1,
237         CLK_I2C5_PLL_SEL_SHIFT          = 15,
238         CLK_I2C5_DIV_CON_SHIFT          = 8,
239         CLK_I2C1_PLL_SEL_SHIFT          = 7,
240         CLK_I2C1_DIV_CON_SHIFT          = 0,
241
242         /* CLKSEL_CON62 */
243         CLK_I2C6_PLL_SEL_SHIFT          = 15,
244         CLK_I2C6_DIV_CON_SHIFT          = 8,
245         CLK_I2C2_PLL_SEL_SHIFT          = 7,
246         CLK_I2C2_DIV_CON_SHIFT          = 0,
247
248         /* CLKSEL_CON63 */
249         CLK_I2C7_PLL_SEL_SHIFT          = 15,
250         CLK_I2C7_DIV_CON_SHIFT          = 8,
251         CLK_I2C3_PLL_SEL_SHIFT          = 7,
252         CLK_I2C3_DIV_CON_SHIFT          = 0,
253
254         /* CRU_SOFTRST_CON4 */
255         RESETN_DDR0_REQ_SHIFT           = 8,
256         RESETN_DDR0_REQ_MASK            = 1 << RESETN_DDR0_REQ_SHIFT,
257         RESETN_DDRPHY0_REQ_SHIFT        = 9,
258         RESETN_DDRPHY0_REQ_MASK         = 1 << RESETN_DDRPHY0_REQ_SHIFT,
259         RESETN_DDR1_REQ_SHIFT           = 12,
260         RESETN_DDR1_REQ_MASK            = 1 << RESETN_DDR1_REQ_SHIFT,
261         RESETN_DDRPHY1_REQ_SHIFT        = 13,
262         RESETN_DDRPHY1_REQ_MASK         = 1 << RESETN_DDRPHY1_REQ_SHIFT,
263 };
264
265 #define VCO_MAX_KHZ     (3200 * (MHz / KHz))
266 #define VCO_MIN_KHZ     (800 * (MHz / KHz))
267 #define OUTPUT_MAX_KHZ  (3200 * (MHz / KHz))
268 #define OUTPUT_MIN_KHZ  (16 * (MHz / KHz))
269
270 /*
271  *  the div restructions of pll in integer mode, these are defined in
272  *  * CRU_*PLL_CON0 or PMUCRU_*PLL_CON0
273  */
274 #define PLL_DIV_MIN     16
275 #define PLL_DIV_MAX     3200
276
277 /*
278  * How to calculate the PLL(from TRM V0.3 Part 1 Page 63):
279  * Formulas also embedded within the Fractional PLL Verilog model:
280  * If DSMPD = 1 (DSM is disabled, "integer mode")
281  * FOUTVCO = FREF / REFDIV * FBDIV
282  * FOUTPOSTDIV = FOUTVCO / POSTDIV1 / POSTDIV2
283  * Where:
284  * FOUTVCO = Fractional PLL non-divided output frequency
285  * FOUTPOSTDIV = Fractional PLL divided output frequency
286  *               (output of second post divider)
287  * FREF = Fractional PLL input reference frequency, (the OSC_HZ 24MHz input)
288  * REFDIV = Fractional PLL input reference clock divider
289  * FBDIV = Integer value programmed into feedback divide
290  *
291  */
292 static void rkclk_set_pll(u32 *pll_con, const struct pll_div *div)
293 {
294         /* All 8 PLLs have same VCO and output frequency range restrictions. */
295         u32 vco_khz = OSC_HZ / 1000 * div->fbdiv / div->refdiv;
296         u32 output_khz = vco_khz / div->postdiv1 / div->postdiv2;
297
298         debug("PLL at %p: fbdiv=%d, refdiv=%d, postdiv1=%d, "
299                            "postdiv2=%d, vco=%u khz, output=%u khz\n",
300                            pll_con, div->fbdiv, div->refdiv, div->postdiv1,
301                            div->postdiv2, vco_khz, output_khz);
302         assert(vco_khz >= VCO_MIN_KHZ && vco_khz <= VCO_MAX_KHZ &&
303                output_khz >= OUTPUT_MIN_KHZ && output_khz <= OUTPUT_MAX_KHZ &&
304                div->fbdiv >= PLL_DIV_MIN && div->fbdiv <= PLL_DIV_MAX);
305
306         /*
307          * When power on or changing PLL setting,
308          * we must force PLL into slow mode to ensure output stable clock.
309          */
310         rk_clrsetreg(&pll_con[3], PLL_MODE_MASK,
311                      PLL_MODE_SLOW << PLL_MODE_SHIFT);
312
313         /* use integer mode */
314         rk_clrsetreg(&pll_con[3], PLL_DSMPD_MASK,
315                      PLL_INTEGER_MODE << PLL_DSMPD_SHIFT);
316
317         rk_clrsetreg(&pll_con[0], PLL_FBDIV_MASK,
318                      div->fbdiv << PLL_FBDIV_SHIFT);
319         rk_clrsetreg(&pll_con[1],
320                      PLL_POSTDIV2_MASK | PLL_POSTDIV1_MASK |
321                      PLL_REFDIV_MASK | PLL_REFDIV_SHIFT,
322                      (div->postdiv2 << PLL_POSTDIV2_SHIFT) |
323                      (div->postdiv1 << PLL_POSTDIV1_SHIFT) |
324                      (div->refdiv << PLL_REFDIV_SHIFT));
325
326         /* waiting for pll lock */
327         while (!(readl(&pll_con[2]) & (1 << PLL_LOCK_STATUS_SHIFT)))
328                 udelay(1);
329
330         /* pll enter normal mode */
331         rk_clrsetreg(&pll_con[3], PLL_MODE_MASK,
332                      PLL_MODE_NORM << PLL_MODE_SHIFT);
333 }
334
335 static int pll_para_config(u32 freq_hz, struct pll_div *div)
336 {
337         u32 ref_khz = OSC_HZ / KHz, refdiv, fbdiv = 0;
338         u32 postdiv1, postdiv2 = 1;
339         u32 fref_khz;
340         u32 diff_khz, best_diff_khz;
341         const u32 max_refdiv = 63, max_fbdiv = 3200, min_fbdiv = 16;
342         const u32 max_postdiv1 = 7, max_postdiv2 = 7;
343         u32 vco_khz;
344         u32 freq_khz = freq_hz / KHz;
345
346         if (!freq_hz) {
347                 printf("%s: the frequency can't be 0 Hz\n", __func__);
348                 return -1;
349         }
350
351         postdiv1 = DIV_ROUND_UP(VCO_MIN_KHZ, freq_khz);
352         if (postdiv1 > max_postdiv1) {
353                 postdiv2 = DIV_ROUND_UP(postdiv1, max_postdiv1);
354                 postdiv1 = DIV_ROUND_UP(postdiv1, postdiv2);
355         }
356
357         vco_khz = freq_khz * postdiv1 * postdiv2;
358
359         if (vco_khz < VCO_MIN_KHZ || vco_khz > VCO_MAX_KHZ ||
360             postdiv2 > max_postdiv2) {
361                 printf("%s: Cannot find out a supported VCO"
362                        " for Frequency (%uHz).\n", __func__, freq_hz);
363                 return -1;
364         }
365
366         div->postdiv1 = postdiv1;
367         div->postdiv2 = postdiv2;
368
369         best_diff_khz = vco_khz;
370         for (refdiv = 1; refdiv < max_refdiv && best_diff_khz; refdiv++) {
371                 fref_khz = ref_khz / refdiv;
372
373                 fbdiv = vco_khz / fref_khz;
374                 if ((fbdiv >= max_fbdiv) || (fbdiv <= min_fbdiv))
375                         continue;
376                 diff_khz = vco_khz - fbdiv * fref_khz;
377                 if (fbdiv + 1 < max_fbdiv && diff_khz > fref_khz / 2) {
378                         fbdiv++;
379                         diff_khz = fref_khz - diff_khz;
380                 }
381
382                 if (diff_khz >= best_diff_khz)
383                         continue;
384
385                 best_diff_khz = diff_khz;
386                 div->refdiv = refdiv;
387                 div->fbdiv = fbdiv;
388         }
389
390         if (best_diff_khz > 4 * (MHz/KHz)) {
391                 printf("%s: Failed to match output frequency %u, "
392                        "difference is %u Hz,exceed 4MHZ\n", __func__, freq_hz,
393                        best_diff_khz * KHz);
394                 return -1;
395         }
396         return 0;
397 }
398
399 #ifdef CONFIG_SPL_BUILD
400 static void rkclk_init(struct rk3399_cru *cru)
401 {
402         u32 aclk_div;
403         u32 hclk_div;
404         u32 pclk_div;
405
406         /*
407          * some cru registers changed by bootrom, we'd better reset them to
408          * reset/default values described in TRM to avoid confusion in kernel.
409          * Please consider these three lines as a fix of bootrom bug.
410          */
411         rk_clrsetreg(&cru->clksel_con[12], 0xffff, 0x4101);
412         rk_clrsetreg(&cru->clksel_con[19], 0xffff, 0x033f);
413         rk_clrsetreg(&cru->clksel_con[56], 0x0003, 0x0003);
414
415         /* configure gpll cpll */
416         rkclk_set_pll(&cru->gpll_con[0], &gpll_init_cfg);
417         rkclk_set_pll(&cru->cpll_con[0], &cpll_init_cfg);
418
419         /* configure perihp aclk, hclk, pclk */
420         aclk_div = GPLL_HZ / PERIHP_ACLK_HZ - 1;
421         assert((aclk_div + 1) * PERIHP_ACLK_HZ == GPLL_HZ && aclk_div < 0x1f);
422
423         hclk_div = PERIHP_ACLK_HZ / PERIHP_HCLK_HZ - 1;
424         assert((hclk_div + 1) * PERIHP_HCLK_HZ ==
425                PERIHP_ACLK_HZ && (hclk_div < 0x4));
426
427         pclk_div = PERIHP_ACLK_HZ / PERIHP_PCLK_HZ - 1;
428         assert((pclk_div + 1) * PERIHP_PCLK_HZ ==
429                PERIHP_ACLK_HZ && (pclk_div < 0x7));
430
431         rk_clrsetreg(&cru->clksel_con[14],
432                      PCLK_PERIHP_DIV_CON_MASK | HCLK_PERIHP_DIV_CON_MASK |
433                      ACLK_PERIHP_PLL_SEL_MASK | ACLK_PERIHP_DIV_CON_MASK,
434                      pclk_div << PCLK_PERIHP_DIV_CON_SHIFT |
435                      hclk_div << HCLK_PERIHP_DIV_CON_SHIFT |
436                      ACLK_PERIHP_PLL_SEL_GPLL << ACLK_PERIHP_PLL_SEL_SHIFT |
437                      aclk_div << ACLK_PERIHP_DIV_CON_SHIFT);
438
439         /* configure perilp0 aclk, hclk, pclk */
440         aclk_div = GPLL_HZ / PERILP0_ACLK_HZ - 1;
441         assert((aclk_div + 1) * PERILP0_ACLK_HZ == GPLL_HZ && aclk_div < 0x1f);
442
443         hclk_div = PERILP0_ACLK_HZ / PERILP0_HCLK_HZ - 1;
444         assert((hclk_div + 1) * PERILP0_HCLK_HZ ==
445                PERILP0_ACLK_HZ && (hclk_div < 0x4));
446
447         pclk_div = PERILP0_ACLK_HZ / PERILP0_PCLK_HZ - 1;
448         assert((pclk_div + 1) * PERILP0_PCLK_HZ ==
449                PERILP0_ACLK_HZ && (pclk_div < 0x7));
450
451         rk_clrsetreg(&cru->clksel_con[23],
452                      PCLK_PERILP0_DIV_CON_MASK | HCLK_PERILP0_DIV_CON_MASK |
453                      ACLK_PERILP0_PLL_SEL_MASK | ACLK_PERILP0_DIV_CON_MASK,
454                      pclk_div << PCLK_PERILP0_DIV_CON_SHIFT |
455                      hclk_div << HCLK_PERILP0_DIV_CON_SHIFT |
456                      ACLK_PERILP0_PLL_SEL_GPLL << ACLK_PERILP0_PLL_SEL_SHIFT |
457                      aclk_div << ACLK_PERILP0_DIV_CON_SHIFT);
458
459         /* perilp1 hclk select gpll as source */
460         hclk_div = GPLL_HZ / PERILP1_HCLK_HZ - 1;
461         assert((hclk_div + 1) * PERILP1_HCLK_HZ ==
462                GPLL_HZ && (hclk_div < 0x1f));
463
464         pclk_div = PERILP1_HCLK_HZ / PERILP1_HCLK_HZ - 1;
465         assert((pclk_div + 1) * PERILP1_HCLK_HZ ==
466                PERILP1_HCLK_HZ && (hclk_div < 0x7));
467
468         rk_clrsetreg(&cru->clksel_con[25],
469                      PCLK_PERILP1_DIV_CON_MASK | HCLK_PERILP1_DIV_CON_MASK |
470                      HCLK_PERILP1_PLL_SEL_MASK,
471                      pclk_div << PCLK_PERILP1_DIV_CON_SHIFT |
472                      hclk_div << HCLK_PERILP1_DIV_CON_SHIFT |
473                      HCLK_PERILP1_PLL_SEL_GPLL << HCLK_PERILP1_PLL_SEL_SHIFT);
474 }
475 #endif
476
477 void rk3399_configure_cpu(struct rk3399_cru *cru,
478                           enum apll_l_frequencies apll_l_freq)
479 {
480         u32 aclkm_div;
481         u32 pclk_dbg_div;
482         u32 atclk_div;
483
484         rkclk_set_pll(&cru->apll_l_con[0], apll_l_cfgs[apll_l_freq]);
485
486         aclkm_div = APLL_HZ / ACLKM_CORE_HZ - 1;
487         assert((aclkm_div + 1) * ACLKM_CORE_HZ == APLL_HZ &&
488                aclkm_div < 0x1f);
489
490         pclk_dbg_div = APLL_HZ / PCLK_DBG_HZ - 1;
491         assert((pclk_dbg_div + 1) * PCLK_DBG_HZ == APLL_HZ &&
492                pclk_dbg_div < 0x1f);
493
494         atclk_div = APLL_HZ / ATCLK_CORE_HZ - 1;
495         assert((atclk_div + 1) * ATCLK_CORE_HZ == APLL_HZ &&
496                atclk_div < 0x1f);
497
498         rk_clrsetreg(&cru->clksel_con[0],
499                      ACLKM_CORE_L_DIV_CON_MASK | CLK_CORE_L_PLL_SEL_MASK |
500                      CLK_CORE_L_DIV_MASK,
501                      aclkm_div << ACLKM_CORE_L_DIV_CON_SHIFT |
502                      CLK_CORE_L_PLL_SEL_ALPLL << CLK_CORE_L_PLL_SEL_SHIFT |
503                      0 << CLK_CORE_L_DIV_SHIFT);
504
505         rk_clrsetreg(&cru->clksel_con[1],
506                      PCLK_DBG_L_DIV_MASK | ATCLK_CORE_L_DIV_MASK,
507                      pclk_dbg_div << PCLK_DBG_L_DIV_SHIFT |
508                      atclk_div << ATCLK_CORE_L_DIV_SHIFT);
509 }
510 #define I2C_CLK_REG_MASK(bus) \
511                         (I2C_DIV_CON_MASK << \
512                         CLK_I2C ##bus## _DIV_CON_SHIFT | \
513                         CLK_I2C_PLL_SEL_MASK << \
514                         CLK_I2C ##bus## _PLL_SEL_SHIFT)
515
516 #define I2C_CLK_REG_VALUE(bus, clk_div) \
517                               ((clk_div - 1) << \
518                                         CLK_I2C ##bus## _DIV_CON_SHIFT | \
519                               CLK_I2C_PLL_SEL_GPLL << \
520                                         CLK_I2C ##bus## _PLL_SEL_SHIFT)
521
522 #define I2C_CLK_DIV_VALUE(con, bus) \
523                         (con >> CLK_I2C ##bus## _DIV_CON_SHIFT) & \
524                                 I2C_DIV_CON_MASK;
525
526 #define I2C_PMUCLK_REG_MASK(bus) \
527                         (I2C_DIV_CON_MASK << \
528                          CLK_I2C ##bus## _DIV_CON_SHIFT)
529
530 #define I2C_PMUCLK_REG_VALUE(bus, clk_div) \
531                                 ((clk_div - 1) << \
532                                 CLK_I2C ##bus## _DIV_CON_SHIFT)
533
534 static ulong rk3399_i2c_get_clk(struct rk3399_cru *cru, ulong clk_id)
535 {
536         u32 div, con;
537
538         switch (clk_id) {
539         case SCLK_I2C1:
540                 con = readl(&cru->clksel_con[61]);
541                 div = I2C_CLK_DIV_VALUE(con, 1);
542                 break;
543         case SCLK_I2C2:
544                 con = readl(&cru->clksel_con[62]);
545                 div = I2C_CLK_DIV_VALUE(con, 2);
546                 break;
547         case SCLK_I2C3:
548                 con = readl(&cru->clksel_con[63]);
549                 div = I2C_CLK_DIV_VALUE(con, 3);
550                 break;
551         case SCLK_I2C5:
552                 con = readl(&cru->clksel_con[61]);
553                 div = I2C_CLK_DIV_VALUE(con, 5);
554                 break;
555         case SCLK_I2C6:
556                 con = readl(&cru->clksel_con[62]);
557                 div = I2C_CLK_DIV_VALUE(con, 6);
558                 break;
559         case SCLK_I2C7:
560                 con = readl(&cru->clksel_con[63]);
561                 div = I2C_CLK_DIV_VALUE(con, 7);
562                 break;
563         default:
564                 printf("do not support this i2c bus\n");
565                 return -EINVAL;
566         }
567
568         return DIV_TO_RATE(GPLL_HZ, div);
569 }
570
571 static ulong rk3399_i2c_set_clk(struct rk3399_cru *cru, ulong clk_id, uint hz)
572 {
573         int src_clk_div;
574
575         /* i2c0,4,8 src clock from ppll, i2c1,2,3,5,6,7 src clock from gpll*/
576         src_clk_div = GPLL_HZ / hz;
577         assert(src_clk_div - 1 < 127);
578
579         switch (clk_id) {
580         case SCLK_I2C1:
581                 rk_clrsetreg(&cru->clksel_con[61], I2C_CLK_REG_MASK(1),
582                              I2C_CLK_REG_VALUE(1, src_clk_div));
583                 break;
584         case SCLK_I2C2:
585                 rk_clrsetreg(&cru->clksel_con[62], I2C_CLK_REG_MASK(2),
586                              I2C_CLK_REG_VALUE(2, src_clk_div));
587                 break;
588         case SCLK_I2C3:
589                 rk_clrsetreg(&cru->clksel_con[63], I2C_CLK_REG_MASK(3),
590                              I2C_CLK_REG_VALUE(3, src_clk_div));
591                 break;
592         case SCLK_I2C5:
593                 rk_clrsetreg(&cru->clksel_con[61], I2C_CLK_REG_MASK(5),
594                              I2C_CLK_REG_VALUE(5, src_clk_div));
595                 break;
596         case SCLK_I2C6:
597                 rk_clrsetreg(&cru->clksel_con[62], I2C_CLK_REG_MASK(6),
598                              I2C_CLK_REG_VALUE(6, src_clk_div));
599                 break;
600         case SCLK_I2C7:
601                 rk_clrsetreg(&cru->clksel_con[63], I2C_CLK_REG_MASK(7),
602                              I2C_CLK_REG_VALUE(7, src_clk_div));
603                 break;
604         default:
605                 printf("do not support this i2c bus\n");
606                 return -EINVAL;
607         }
608
609         return DIV_TO_RATE(GPLL_HZ, src_clk_div);
610 }
611
612 /*
613  * RK3399 SPI clocks have a common divider-width (7 bits) and a single bit
614  * to select either CPLL or GPLL as the clock-parent. The location within
615  * the enclosing CLKSEL_CON (i.e. div_shift and sel_shift) are variable.
616  */
617
618 struct spi_clkreg {
619         uint8_t reg;  /* CLKSEL_CON[reg] register in CRU */
620         uint8_t div_shift;
621         uint8_t sel_shift;
622 };
623
624 /*
625  * The entries are numbered relative to their offset from SCLK_SPI0.
626  *
627  * Note that SCLK_SPI3 (which is configured via PMUCRU and requires different
628  * logic is not supported).
629  */
630 static const struct spi_clkreg spi_clkregs[] = {
631         [0] = { .reg = 59,
632                 .div_shift = CLK_SPI0_PLL_DIV_CON_SHIFT,
633                 .sel_shift = CLK_SPI0_PLL_SEL_SHIFT, },
634         [1] = { .reg = 59,
635                 .div_shift = CLK_SPI1_PLL_DIV_CON_SHIFT,
636                 .sel_shift = CLK_SPI1_PLL_SEL_SHIFT, },
637         [2] = { .reg = 60,
638                 .div_shift = CLK_SPI2_PLL_DIV_CON_SHIFT,
639                 .sel_shift = CLK_SPI2_PLL_SEL_SHIFT, },
640         [3] = { .reg = 60,
641                 .div_shift = CLK_SPI4_PLL_DIV_CON_SHIFT,
642                 .sel_shift = CLK_SPI4_PLL_SEL_SHIFT, },
643         [4] = { .reg = 58,
644                 .div_shift = CLK_SPI5_PLL_DIV_CON_SHIFT,
645                 .sel_shift = CLK_SPI5_PLL_SEL_SHIFT, },
646 };
647
648 static inline u32 extract_bits(u32 val, unsigned width, unsigned shift)
649 {
650         return (val >> shift) & ((1 << width) - 1);
651 }
652
653 static ulong rk3399_spi_get_clk(struct rk3399_cru *cru, ulong clk_id)
654 {
655         const struct spi_clkreg *spiclk = NULL;
656         u32 div, val;
657
658         switch (clk_id) {
659         case SCLK_SPI0 ... SCLK_SPI5:
660                 spiclk = &spi_clkregs[clk_id - SCLK_SPI0];
661                 break;
662
663         default:
664                 error("%s: SPI clk-id %ld not supported\n", __func__, clk_id);
665                 return -EINVAL;
666         }
667
668         val = readl(&cru->clksel_con[spiclk->reg]);
669         div = extract_bits(val, CLK_SPI_PLL_DIV_CON_WIDTH, spiclk->div_shift);
670
671         return DIV_TO_RATE(GPLL_HZ, div);
672 }
673
674 static ulong rk3399_spi_set_clk(struct rk3399_cru *cru, ulong clk_id, uint hz)
675 {
676         const struct spi_clkreg *spiclk = NULL;
677         int src_clk_div;
678
679         src_clk_div = RATE_TO_DIV(GPLL_HZ, hz);
680         assert(src_clk_div < 127);
681
682         switch (clk_id) {
683         case SCLK_SPI1 ... SCLK_SPI5:
684                 spiclk = &spi_clkregs[clk_id - SCLK_SPI0];
685                 break;
686
687         default:
688                 error("%s: SPI clk-id %ld not supported\n", __func__, clk_id);
689                 return -EINVAL;
690         }
691
692         rk_clrsetreg(&cru->clksel_con[spiclk->reg],
693                      ((CLK_SPI_PLL_DIV_CON_MASK << spiclk->div_shift) |
694                        (CLK_SPI_PLL_SEL_GPLL << spiclk->sel_shift)),
695                      ((src_clk_div << spiclk->div_shift) |
696                       (CLK_SPI_PLL_SEL_GPLL << spiclk->sel_shift)));
697
698
699         return DIV_TO_RATE(GPLL_HZ, src_clk_div);
700 }
701
702 static ulong rk3399_vop_set_clk(struct rk3399_cru *cru, ulong clk_id, u32 hz)
703 {
704         struct pll_div vpll_config = {0};
705         int aclk_vop = 198*MHz;
706         void *aclkreg_addr, *dclkreg_addr;
707         u32 div;
708
709         switch (clk_id) {
710         case DCLK_VOP0:
711                 aclkreg_addr = &cru->clksel_con[47];
712                 dclkreg_addr = &cru->clksel_con[49];
713                 break;
714         case DCLK_VOP1:
715                 aclkreg_addr = &cru->clksel_con[48];
716                 dclkreg_addr = &cru->clksel_con[50];
717                 break;
718         default:
719                 return -EINVAL;
720         }
721         /* vop aclk source clk: cpll */
722         div = CPLL_HZ / aclk_vop;
723         assert(div - 1 < 32);
724
725         rk_clrsetreg(aclkreg_addr,
726                      ACLK_VOP_PLL_SEL_MASK | ACLK_VOP_DIV_CON_MASK,
727                      ACLK_VOP_PLL_SEL_CPLL << ACLK_VOP_PLL_SEL_SHIFT |
728                      (div - 1) << ACLK_VOP_DIV_CON_SHIFT);
729
730         /* vop dclk source from vpll, and equals to vpll(means div == 1) */
731         if (pll_para_config(hz, &vpll_config))
732                 return -1;
733
734         rkclk_set_pll(&cru->vpll_con[0], &vpll_config);
735
736         rk_clrsetreg(dclkreg_addr,
737                      DCLK_VOP_DCLK_SEL_MASK | DCLK_VOP_PLL_SEL_MASK|
738                      DCLK_VOP_DIV_CON_MASK,
739                      DCLK_VOP_DCLK_SEL_DIVOUT << DCLK_VOP_DCLK_SEL_SHIFT |
740                      DCLK_VOP_PLL_SEL_VPLL << DCLK_VOP_PLL_SEL_SHIFT |
741                      (1 - 1) << DCLK_VOP_DIV_CON_SHIFT);
742
743         return hz;
744 }
745
746 static ulong rk3399_mmc_get_clk(struct rk3399_cru *cru, uint clk_id)
747 {
748         u32 div, con;
749
750         switch (clk_id) {
751         case SCLK_SDMMC:
752                 con = readl(&cru->clksel_con[16]);
753                 break;
754         case SCLK_EMMC:
755                 con = readl(&cru->clksel_con[21]);
756                 break;
757         default:
758                 return -EINVAL;
759         }
760         div = (con & CLK_EMMC_DIV_CON_MASK) >> CLK_EMMC_DIV_CON_SHIFT;
761
762         if ((con & CLK_EMMC_PLL_MASK) >> CLK_EMMC_PLL_SHIFT
763                         == CLK_EMMC_PLL_SEL_24M)
764                 return DIV_TO_RATE(24*1000*1000, div);
765         else
766                 return DIV_TO_RATE(GPLL_HZ, div);
767 }
768
769 static ulong rk3399_mmc_set_clk(struct rk3399_cru *cru,
770                                 ulong clk_id, ulong set_rate)
771 {
772         int src_clk_div;
773         int aclk_emmc = 198*MHz;
774
775         switch (clk_id) {
776         case SCLK_SDMMC:
777                 /* Select clk_sdmmc source from GPLL by default */
778                 src_clk_div = GPLL_HZ / set_rate;
779
780                 if (src_clk_div > 127) {
781                         /* use 24MHz source for 400KHz clock */
782                         src_clk_div = 24*1000*1000 / set_rate;
783                         rk_clrsetreg(&cru->clksel_con[16],
784                                      CLK_EMMC_PLL_MASK | CLK_EMMC_DIV_CON_MASK,
785                                      CLK_EMMC_PLL_SEL_24M << CLK_EMMC_PLL_SHIFT |
786                                      (src_clk_div - 1) << CLK_EMMC_DIV_CON_SHIFT);
787                 } else {
788                         rk_clrsetreg(&cru->clksel_con[16],
789                                      CLK_EMMC_PLL_MASK | CLK_EMMC_DIV_CON_MASK,
790                                      CLK_EMMC_PLL_SEL_GPLL << CLK_EMMC_PLL_SHIFT |
791                                      (src_clk_div - 1) << CLK_EMMC_DIV_CON_SHIFT);
792                 }
793                 break;
794         case SCLK_EMMC:
795                 /* Select aclk_emmc source from GPLL */
796                 src_clk_div = GPLL_HZ / aclk_emmc;
797                 assert(src_clk_div - 1 < 31);
798
799                 rk_clrsetreg(&cru->clksel_con[21],
800                              ACLK_EMMC_PLL_SEL_MASK | ACLK_EMMC_DIV_CON_MASK,
801                              ACLK_EMMC_PLL_SEL_GPLL << ACLK_EMMC_PLL_SEL_SHIFT |
802                              (src_clk_div - 1) << ACLK_EMMC_DIV_CON_SHIFT);
803
804                 /* Select clk_emmc source from GPLL too */
805                 src_clk_div = GPLL_HZ / set_rate;
806                 assert(src_clk_div - 1 < 127);
807
808                 rk_clrsetreg(&cru->clksel_con[22],
809                              CLK_EMMC_PLL_MASK | CLK_EMMC_DIV_CON_MASK,
810                              CLK_EMMC_PLL_SEL_GPLL << CLK_EMMC_PLL_SHIFT |
811                              (src_clk_div - 1) << CLK_EMMC_DIV_CON_SHIFT);
812                 break;
813         default:
814                 return -EINVAL;
815         }
816         return rk3399_mmc_get_clk(cru, clk_id);
817 }
818
819 #define PMUSGRF_DDR_RGN_CON16 0xff330040
820 static ulong rk3399_ddr_set_clk(struct rk3399_cru *cru,
821                                 ulong set_rate)
822 {
823         struct pll_div dpll_cfg;
824
825         /*  IC ECO bug, need to set this register */
826         writel(0xc000c000, PMUSGRF_DDR_RGN_CON16);
827
828         /*  clk_ddrc == DPLL = 24MHz / refdiv * fbdiv / postdiv1 / postdiv2 */
829         switch (set_rate) {
830         case 200*MHz:
831                 dpll_cfg = (struct pll_div)
832                 {.refdiv = 1, .fbdiv = 50, .postdiv1 = 6, .postdiv2 = 1};
833                 break;
834         case 300*MHz:
835                 dpll_cfg = (struct pll_div)
836                 {.refdiv = 2, .fbdiv = 100, .postdiv1 = 4, .postdiv2 = 1};
837                 break;
838         case 666*MHz:
839                 dpll_cfg = (struct pll_div)
840                 {.refdiv = 2, .fbdiv = 111, .postdiv1 = 2, .postdiv2 = 1};
841                 break;
842         case 800*MHz:
843                 dpll_cfg = (struct pll_div)
844                 {.refdiv = 1, .fbdiv = 100, .postdiv1 = 3, .postdiv2 = 1};
845                 break;
846         case 933*MHz:
847                 dpll_cfg = (struct pll_div)
848                 {.refdiv = 1, .fbdiv = 116, .postdiv1 = 3, .postdiv2 = 1};
849                 break;
850         default:
851                 error("Unsupported SDRAM frequency!,%ld\n", set_rate);
852         }
853         rkclk_set_pll(&cru->dpll_con[0], &dpll_cfg);
854
855         return set_rate;
856 }
857 static ulong rk3399_clk_get_rate(struct clk *clk)
858 {
859         struct rk3399_clk_priv *priv = dev_get_priv(clk->dev);
860         ulong rate = 0;
861
862         switch (clk->id) {
863         case 0 ... 63:
864                 return 0;
865         case SCLK_SDMMC:
866         case SCLK_EMMC:
867                 rate = rk3399_mmc_get_clk(priv->cru, clk->id);
868                 break;
869         case SCLK_I2C1:
870         case SCLK_I2C2:
871         case SCLK_I2C3:
872         case SCLK_I2C5:
873         case SCLK_I2C6:
874         case SCLK_I2C7:
875                 rate = rk3399_i2c_get_clk(priv->cru, clk->id);
876                 break;
877         case SCLK_SPI0...SCLK_SPI5:
878                 rate = rk3399_spi_get_clk(priv->cru, clk->id);
879                 break;
880         case SCLK_UART0:
881         case SCLK_UART2:
882                 return 24000000;
883         case DCLK_VOP0:
884         case DCLK_VOP1:
885                 break;
886         default:
887                 return -ENOENT;
888         }
889
890         return rate;
891 }
892
893 static ulong rk3399_clk_set_rate(struct clk *clk, ulong rate)
894 {
895         struct rk3399_clk_priv *priv = dev_get_priv(clk->dev);
896         ulong ret = 0;
897
898         switch (clk->id) {
899         case 0 ... 63:
900                 return 0;
901         case SCLK_SDMMC:
902         case SCLK_EMMC:
903                 ret = rk3399_mmc_set_clk(priv->cru, clk->id, rate);
904                 break;
905         case SCLK_MAC:
906                 /* nothing to do, as this is an external clock */
907                 ret = rate;
908                 break;
909         case SCLK_I2C1:
910         case SCLK_I2C2:
911         case SCLK_I2C3:
912         case SCLK_I2C5:
913         case SCLK_I2C6:
914         case SCLK_I2C7:
915                 ret = rk3399_i2c_set_clk(priv->cru, clk->id, rate);
916                 break;
917         case SCLK_SPI0...SCLK_SPI5:
918                 ret = rk3399_spi_set_clk(priv->cru, clk->id, rate);
919                 break;
920         case DCLK_VOP0:
921         case DCLK_VOP1:
922                 ret = rk3399_vop_set_clk(priv->cru, clk->id, rate);
923                 break;
924         case SCLK_DDRCLK:
925                 ret = rk3399_ddr_set_clk(priv->cru, rate);
926                 break;
927         default:
928                 return -ENOENT;
929         }
930
931         return ret;
932 }
933
934 static struct clk_ops rk3399_clk_ops = {
935         .get_rate = rk3399_clk_get_rate,
936         .set_rate = rk3399_clk_set_rate,
937 };
938
939 static int rk3399_clk_probe(struct udevice *dev)
940 {
941 #ifdef CONFIG_SPL_BUILD
942         struct rk3399_clk_priv *priv = dev_get_priv(dev);
943
944 #if CONFIG_IS_ENABLED(OF_PLATDATA)
945         struct rk3399_clk_plat *plat = dev_get_platdata(dev);
946
947         priv->cru = map_sysmem(plat->dtd.reg[1], plat->dtd.reg[3]);
948 #endif
949         rkclk_init(priv->cru);
950 #endif
951         return 0;
952 }
953
954 static int rk3399_clk_ofdata_to_platdata(struct udevice *dev)
955 {
956 #if !CONFIG_IS_ENABLED(OF_PLATDATA)
957         struct rk3399_clk_priv *priv = dev_get_priv(dev);
958
959         priv->cru = (struct rk3399_cru *)dev_get_addr(dev);
960 #endif
961         return 0;
962 }
963
964 static int rk3399_clk_bind(struct udevice *dev)
965 {
966         int ret;
967
968         /* The reset driver does not have a device node, so bind it here */
969         ret = device_bind_driver(gd->dm_root, "rk3399_sysreset", "reset", &dev);
970         if (ret)
971                 printf("Warning: No RK3399 reset driver: ret=%d\n", ret);
972
973         return 0;
974 }
975
976 static const struct udevice_id rk3399_clk_ids[] = {
977         { .compatible = "rockchip,rk3399-cru" },
978         { }
979 };
980
981 U_BOOT_DRIVER(clk_rk3399) = {
982         .name           = "rockchip_rk3399_cru",
983         .id             = UCLASS_CLK,
984         .of_match       = rk3399_clk_ids,
985         .priv_auto_alloc_size = sizeof(struct rk3399_clk_priv),
986         .ofdata_to_platdata = rk3399_clk_ofdata_to_platdata,
987         .ops            = &rk3399_clk_ops,
988         .bind           = rk3399_clk_bind,
989         .probe          = rk3399_clk_probe,
990 #if CONFIG_IS_ENABLED(OF_PLATDATA)
991         .platdata_auto_alloc_size = sizeof(struct rk3399_clk_plat),
992 #endif
993 };
994
995 static ulong rk3399_i2c_get_pmuclk(struct rk3399_pmucru *pmucru, ulong clk_id)
996 {
997         u32 div, con;
998
999         switch (clk_id) {
1000         case SCLK_I2C0_PMU:
1001                 con = readl(&pmucru->pmucru_clksel[2]);
1002                 div = I2C_CLK_DIV_VALUE(con, 0);
1003                 break;
1004         case SCLK_I2C4_PMU:
1005                 con = readl(&pmucru->pmucru_clksel[3]);
1006                 div = I2C_CLK_DIV_VALUE(con, 4);
1007                 break;
1008         case SCLK_I2C8_PMU:
1009                 con = readl(&pmucru->pmucru_clksel[2]);
1010                 div = I2C_CLK_DIV_VALUE(con, 8);
1011                 break;
1012         default:
1013                 printf("do not support this i2c bus\n");
1014                 return -EINVAL;
1015         }
1016
1017         return DIV_TO_RATE(PPLL_HZ, div);
1018 }
1019
1020 static ulong rk3399_i2c_set_pmuclk(struct rk3399_pmucru *pmucru, ulong clk_id,
1021                                    uint hz)
1022 {
1023         int src_clk_div;
1024
1025         src_clk_div = PPLL_HZ / hz;
1026         assert(src_clk_div - 1 < 127);
1027
1028         switch (clk_id) {
1029         case SCLK_I2C0_PMU:
1030                 rk_clrsetreg(&pmucru->pmucru_clksel[2], I2C_PMUCLK_REG_MASK(0),
1031                              I2C_PMUCLK_REG_VALUE(0, src_clk_div));
1032                 break;
1033         case SCLK_I2C4_PMU:
1034                 rk_clrsetreg(&pmucru->pmucru_clksel[3], I2C_PMUCLK_REG_MASK(4),
1035                              I2C_PMUCLK_REG_VALUE(4, src_clk_div));
1036                 break;
1037         case SCLK_I2C8_PMU:
1038                 rk_clrsetreg(&pmucru->pmucru_clksel[2], I2C_PMUCLK_REG_MASK(8),
1039                              I2C_PMUCLK_REG_VALUE(8, src_clk_div));
1040                 break;
1041         default:
1042                 printf("do not support this i2c bus\n");
1043                 return -EINVAL;
1044         }
1045
1046         return DIV_TO_RATE(PPLL_HZ, src_clk_div);
1047 }
1048
1049 static ulong rk3399_pwm_get_clk(struct rk3399_pmucru *pmucru)
1050 {
1051         u32 div, con;
1052
1053         /* PWM closk rate is same as pclk_pmu */
1054         con = readl(&pmucru->pmucru_clksel[0]);
1055         div = con & PMU_PCLK_DIV_CON_MASK;
1056
1057         return DIV_TO_RATE(PPLL_HZ, div);
1058 }
1059
1060 static ulong rk3399_pmuclk_get_rate(struct clk *clk)
1061 {
1062         struct rk3399_pmuclk_priv *priv = dev_get_priv(clk->dev);
1063         ulong rate = 0;
1064
1065         switch (clk->id) {
1066         case PCLK_RKPWM_PMU:
1067                 rate = rk3399_pwm_get_clk(priv->pmucru);
1068                 break;
1069         case SCLK_I2C0_PMU:
1070         case SCLK_I2C4_PMU:
1071         case SCLK_I2C8_PMU:
1072                 rate = rk3399_i2c_get_pmuclk(priv->pmucru, clk->id);
1073                 break;
1074         default:
1075                 return -ENOENT;
1076         }
1077
1078         return rate;
1079 }
1080
1081 static ulong rk3399_pmuclk_set_rate(struct clk *clk, ulong rate)
1082 {
1083         struct rk3399_pmuclk_priv *priv = dev_get_priv(clk->dev);
1084         ulong ret = 0;
1085
1086         switch (clk->id) {
1087         case SCLK_I2C0_PMU:
1088         case SCLK_I2C4_PMU:
1089         case SCLK_I2C8_PMU:
1090                 ret = rk3399_i2c_set_pmuclk(priv->pmucru, clk->id, rate);
1091                 break;
1092         default:
1093                 return -ENOENT;
1094         }
1095
1096         return ret;
1097 }
1098
1099 static struct clk_ops rk3399_pmuclk_ops = {
1100         .get_rate = rk3399_pmuclk_get_rate,
1101         .set_rate = rk3399_pmuclk_set_rate,
1102 };
1103
1104 #ifndef CONFIG_SPL_BUILD
1105 static void pmuclk_init(struct rk3399_pmucru *pmucru)
1106 {
1107         u32 pclk_div;
1108
1109         /*  configure pmu pll(ppll) */
1110         rkclk_set_pll(&pmucru->ppll_con[0], &ppll_init_cfg);
1111
1112         /*  configure pmu pclk */
1113         pclk_div = PPLL_HZ / PMU_PCLK_HZ - 1;
1114         rk_clrsetreg(&pmucru->pmucru_clksel[0],
1115                      PMU_PCLK_DIV_CON_MASK,
1116                      pclk_div << PMU_PCLK_DIV_CON_SHIFT);
1117 }
1118 #endif
1119
1120 static int rk3399_pmuclk_probe(struct udevice *dev)
1121 {
1122 #if CONFIG_IS_ENABLED(OF_PLATDATA) || !defined(CONFIG_SPL_BUILD)
1123         struct rk3399_pmuclk_priv *priv = dev_get_priv(dev);
1124 #endif
1125
1126 #if CONFIG_IS_ENABLED(OF_PLATDATA)
1127         struct rk3399_pmuclk_plat *plat = dev_get_platdata(dev);
1128
1129         priv->pmucru = map_sysmem(plat->dtd.reg[1], plat->dtd.reg[3]);
1130 #endif
1131
1132 #ifndef CONFIG_SPL_BUILD
1133         pmuclk_init(priv->pmucru);
1134 #endif
1135         return 0;
1136 }
1137
1138 static int rk3399_pmuclk_ofdata_to_platdata(struct udevice *dev)
1139 {
1140 #if !CONFIG_IS_ENABLED(OF_PLATDATA)
1141         struct rk3399_pmuclk_priv *priv = dev_get_priv(dev);
1142
1143         priv->pmucru = (struct rk3399_pmucru *)dev_get_addr(dev);
1144 #endif
1145         return 0;
1146 }
1147
1148 static const struct udevice_id rk3399_pmuclk_ids[] = {
1149         { .compatible = "rockchip,rk3399-pmucru" },
1150         { }
1151 };
1152
1153 U_BOOT_DRIVER(rockchip_rk3399_pmuclk) = {
1154         .name           = "rockchip_rk3399_pmucru",
1155         .id             = UCLASS_CLK,
1156         .of_match       = rk3399_pmuclk_ids,
1157         .priv_auto_alloc_size = sizeof(struct rk3399_pmuclk_priv),
1158         .ofdata_to_platdata = rk3399_pmuclk_ofdata_to_platdata,
1159         .ops            = &rk3399_pmuclk_ops,
1160         .probe          = rk3399_pmuclk_probe,
1161 #if CONFIG_IS_ENABLED(OF_PLATDATA)
1162         .platdata_auto_alloc_size = sizeof(struct rk3399_pmuclk_plat),
1163 #endif
1164 };