1 /*
2  * (C) Copyright 2017 Rockchip Electronics Co., Ltd
3  *
4  * SPDX-License-Identifier:     GPL-2.0
5  */
6 #include <common.h>
7 #include <clk.h>
8 #include <dm.h>
9 #include <dt-structs.h>
10 #include <errno.h>
11 #include <ram.h>
12 #include <regmap.h>
13 #include <syscon.h>
14 #include <asm/io.h>
15 #include <asm/arch/clock.h>
16 #include <asm/arch/cru_rk322x.h>
17 #include <asm/arch/grf_rk322x.h>
18 #include <asm/arch/hardware.h>
19 #include <asm/arch/sdram_rk322x.h>
20 #include <asm/arch/timer.h>
21 #include <asm/arch/uart.h>
22 #include <asm/arch/sdram_common.h>
23 #include <asm/types.h>
24 #include <linux/err.h>
25 
26 DECLARE_GLOBAL_DATA_PTR;
27 struct chan_info {
28 	struct rk322x_ddr_pctl *pctl;
29 	struct rk322x_ddr_phy *phy;
30 	struct rk322x_service_sys *msch;
31 };
32 
33 struct dram_info {
34 	struct chan_info chan[1];
35 	struct ram_info info;
36 	struct clk ddr_clk;
37 	struct rk322x_cru *cru;
38 	struct rk322x_grf *grf;
39 };
40 
41 struct rk322x_sdram_params {
42 #if CONFIG_IS_ENABLED(OF_PLATDATA)
43 		struct dtd_rockchip_rk3228_dmc of_plat;
44 #endif
45 		struct rk322x_sdram_channel ch[1];
46 		struct rk322x_pctl_timing pctl_timing;
47 		struct rk322x_phy_timing phy_timing;
48 		struct rk322x_base_params base;
49 		int num_channels;
50 		struct regmap *map;
51 };
52 
53 #ifdef CONFIG_TPL_BUILD
54 /*
55  * [7:6]  bank(n:n bit bank)
56  * [5:4]  row(13+n)
57  * [3]    cs(0:1 cs, 1:2 cs)
58  * [2:1]  bank(n:n bit bank)
59  * [0]    col(10+n)
60  */
61 const char ddr_cfg_2_rbc[] = {
62 	((0 << 6) | (0 << 4) | (0 << 3) | (1 << 2) | 1),
63 	((0 << 6) | (1 << 4) | (0 << 3) | (1 << 2) | 1),
64 	((0 << 6) | (2 << 4) | (0 << 3) | (1 << 2) | 1),
65 	((0 << 6) | (3 << 4) | (0 << 3) | (1 << 2) | 1),
66 	((0 << 6) | (1 << 4) | (0 << 3) | (1 << 2) | 2),
67 	((0 << 6) | (2 << 4) | (0 << 3) | (1 << 2) | 2),
68 	((0 << 6) | (3 << 4) | (0 << 3) | (1 << 2) | 2),
69 	((0 << 6) | (0 << 4) | (0 << 3) | (1 << 2) | 0),
70 	((0 << 6) | (1 << 4) | (0 << 3) | (1 << 2) | 0),
71 	((0 << 6) | (2 << 4) | (0 << 3) | (1 << 2) | 0),
72 	((0 << 6) | (3 << 4) | (0 << 3) | (1 << 2) | 0),
73 	((0 << 6) | (2 << 4) | (0 << 3) | (0 << 2) | 1),
74 	((1 << 6) | (1 << 4) | (0 << 3) | (0 << 2) | 2),
75 	((1 << 6) | (1 << 4) | (0 << 3) | (0 << 2) | 1),
76 	((0 << 6) | (3 << 4) | (1 << 3) | (1 << 2) | 1),
77 	((0 << 6) | (3 << 4) | (1 << 3) | (1 << 2) | 0),
78 };
79 
80 static void copy_to_reg(u32 *dest, const u32 *src, u32 n)
81 {
82 	int i;
83 
84 	for (i = 0; i < n / sizeof(u32); i++) {
85 		writel(*src, dest);
86 		src++;
87 		dest++;
88 	}
89 }
90 
91 void phy_pctrl_reset(struct rk322x_cru *cru,
92 		     struct rk322x_ddr_phy *ddr_phy)
93 {
94 	rk_clrsetreg(&cru->cru_softrst_con[5], 1 << DDRCTRL_PSRST_SHIFT |
95 			1 << DDRCTRL_SRST_SHIFT | 1 << DDRPHY_PSRST_SHIFT |
96 			1 << DDRPHY_SRST_SHIFT,
97 			1 << DDRCTRL_PSRST_SHIFT | 1 << DDRCTRL_SRST_SHIFT |
98 			1 << DDRPHY_PSRST_SHIFT | 1 << DDRPHY_SRST_SHIFT);
99 
100 	rockchip_udelay(10);
101 
102 	rk_clrreg(&cru->cru_softrst_con[5], 1 << DDRPHY_PSRST_SHIFT |
103 						  1 << DDRPHY_SRST_SHIFT);
104 	rockchip_udelay(10);
105 
106 	rk_clrreg(&cru->cru_softrst_con[5], 1 << DDRCTRL_PSRST_SHIFT |
107 						  1 << DDRCTRL_SRST_SHIFT);
108 	rockchip_udelay(10);
109 
110 	clrbits_le32(&ddr_phy->ddrphy_reg[0],
111 		     SOFT_RESET_MASK << SOFT_RESET_SHIFT);
112 	rockchip_udelay(10);
113 	setbits_le32(&ddr_phy->ddrphy_reg[0],
114 		     SOFT_DERESET_ANALOG);
115 	rockchip_udelay(5);
116 	setbits_le32(&ddr_phy->ddrphy_reg[0],
117 		     SOFT_DERESET_DIGITAL);
118 
119 	rockchip_udelay(1);
120 }
121 
122 void phy_dll_bypass_set(struct rk322x_ddr_phy *ddr_phy, u32 freq)
123 {
124 	u32 tmp;
125 
126 	setbits_le32(&ddr_phy->ddrphy_reg[0x13], 0x10);
127 	setbits_le32(&ddr_phy->ddrphy_reg[0x26], 0x10);
128 	setbits_le32(&ddr_phy->ddrphy_reg[0x36], 0x10);
129 	setbits_le32(&ddr_phy->ddrphy_reg[0x46], 0x10);
130 	setbits_le32(&ddr_phy->ddrphy_reg[0x56], 0x10);
131 
132 	clrbits_le32(&ddr_phy->ddrphy_reg[0x14], 0x8);
133 	clrbits_le32(&ddr_phy->ddrphy_reg[0x27], 0x8);
134 	clrbits_le32(&ddr_phy->ddrphy_reg[0x37], 0x8);
135 	clrbits_le32(&ddr_phy->ddrphy_reg[0x47], 0x8);
136 	clrbits_le32(&ddr_phy->ddrphy_reg[0x57], 0x8);
137 
138 	if (freq <= 400)
139 		setbits_le32(&ddr_phy->ddrphy_reg[0xa4], 0x1f);
140 	else
141 		clrbits_le32(&ddr_phy->ddrphy_reg[0xa4], 0x1f);
142 
143 	if (freq <= 680)
144 		tmp = 3;
145 	else
146 		tmp = 2;
147 
148 	writel(tmp, &ddr_phy->ddrphy_reg[0x28]);
149 	writel(tmp, &ddr_phy->ddrphy_reg[0x38]);
150 	writel(tmp, &ddr_phy->ddrphy_reg[0x48]);
151 	writel(tmp, &ddr_phy->ddrphy_reg[0x58]);
152 }
153 
154 static void send_command(struct rk322x_ddr_pctl *pctl,
155 			 u32 rank, u32 cmd, u32 arg)
156 {
157 	writel((START_CMD | (rank << 20) | arg | cmd), &pctl->mcmd);
158 	rockchip_udelay(1);
159 	while (readl(&pctl->mcmd) & START_CMD)
160 		;
161 }
162 
163 static void memory_init(struct chan_info *chan,
164 			struct rk322x_sdram_params *sdram_params)
165 {
166 	struct rk322x_ddr_pctl *pctl = chan->pctl;
167 	u32 dramtype = sdram_params->base.dramtype;
168 
169 	if (dramtype == DDR3) {
170 		send_command(pctl, 3, DESELECT_CMD, 0);
171 		rockchip_udelay(1);
172 		send_command(pctl, 3, PREA_CMD, 0);
173 		send_command(pctl, 3, MRS_CMD,
174 			     (0x02 & BANK_ADDR_MASK) << BANK_ADDR_SHIFT |
175 			     (sdram_params->phy_timing.mr[2] & CMD_ADDR_MASK) <<
176 			     CMD_ADDR_SHIFT);
177 
178 		send_command(pctl, 3, MRS_CMD,
179 			     (0x03 & BANK_ADDR_MASK) << BANK_ADDR_SHIFT |
180 			     (sdram_params->phy_timing.mr[3] & CMD_ADDR_MASK) <<
181 			     CMD_ADDR_SHIFT);
182 
183 		send_command(pctl, 3, MRS_CMD,
184 			     (0x01 & BANK_ADDR_MASK) << BANK_ADDR_SHIFT |
185 			     (sdram_params->phy_timing.mr[1] & CMD_ADDR_MASK) <<
186 			     CMD_ADDR_SHIFT);
187 
188 		send_command(pctl, 3, MRS_CMD,
189 			     (0x00 & BANK_ADDR_MASK) << BANK_ADDR_SHIFT |
190 			     ((sdram_params->phy_timing.mr[0] |
191 			       DDR3_DLL_RESET) &
192 			     CMD_ADDR_MASK) << CMD_ADDR_SHIFT);
193 
194 		send_command(pctl, 3, ZQCL_CMD, 0);
195 	} else {
196 		send_command(pctl, 3, MRS_CMD,
197 			     (0x63 & LPDDR23_MA_MASK) << LPDDR23_MA_SHIFT |
198 			     (0 & LPDDR23_OP_MASK) <<
199 			     LPDDR23_OP_SHIFT);
200 		rockchip_udelay(10);
201 		send_command(pctl, 3, MRS_CMD,
202 			     (0x10 & LPDDR23_MA_MASK) << LPDDR23_MA_SHIFT |
203 			     (0xff & LPDDR23_OP_MASK) <<
204 			     LPDDR23_OP_SHIFT);
205 		rockchip_udelay(1);
206 		send_command(pctl, 3, MRS_CMD,
207 			     (0x10 & LPDDR23_MA_MASK) << LPDDR23_MA_SHIFT |
208 			     (0xff & LPDDR23_OP_MASK) <<
209 			     LPDDR23_OP_SHIFT);
210 		rockchip_udelay(1);
211 		send_command(pctl, 3, MRS_CMD,
212 			     (1 & LPDDR23_MA_MASK) << LPDDR23_MA_SHIFT |
213 			     (sdram_params->phy_timing.mr[1] &
214 			      LPDDR23_OP_MASK) << LPDDR23_OP_SHIFT);
215 		send_command(pctl, 3, MRS_CMD,
216 			     (2 & LPDDR23_MA_MASK) << LPDDR23_MA_SHIFT |
217 			     (sdram_params->phy_timing.mr[2] &
218 			      LPDDR23_OP_MASK) << LPDDR23_OP_SHIFT);
219 		send_command(pctl, 3, MRS_CMD,
220 			     (3 & LPDDR23_MA_MASK) << LPDDR23_MA_SHIFT |
221 			     (sdram_params->phy_timing.mr[3] &
222 			      LPDDR23_OP_MASK) << LPDDR23_OP_SHIFT);
223 		if (dramtype == LPDDR3)
224 			send_command(pctl, 3, MRS_CMD, (11 & LPDDR23_MA_MASK) <<
225 				     LPDDR23_MA_SHIFT |
226 				     (sdram_params->phy_timing.mr11 &
227 				      LPDDR23_OP_MASK) << LPDDR23_OP_SHIFT);
228 	}
229 }
230 
231 static u32 data_training(struct chan_info *chan)
232 {
233 	struct rk322x_ddr_phy *ddr_phy = chan->phy;
234 	struct rk322x_ddr_pctl *pctl = chan->pctl;
235 	u32 value;
236 	u32 bw = (readl(&ddr_phy->ddrphy_reg[0]) >> 4) & 0xf;
237 	u32 ret;
238 
239 	/* disable auto refresh */
240 	value = readl(&pctl->trefi) | (1 << 31);
241 	writel(1 << 31, &pctl->trefi);
242 
243 	clrsetbits_le32(&ddr_phy->ddrphy_reg[2], 0x30,
244 			DQS_SQU_CAL_SEL_CS0);
245 	setbits_le32(&ddr_phy->ddrphy_reg[2], DQS_SQU_CAL_START);
246 
247 	rockchip_udelay(30);
248 	ret = readl(&ddr_phy->ddrphy_reg[0xff]);
249 
250 	clrbits_le32(&ddr_phy->ddrphy_reg[2],
251 		     DQS_SQU_CAL_START);
252 
253 	/*
254 	 * since data training will take about 20us, so send some auto
255 	 * refresh(about 7.8us) to complement the lost time
256 	 */
257 	send_command(pctl, 3, PREA_CMD, 0);
258 	send_command(pctl, 3, REF_CMD, 0);
259 
260 	writel(value, &pctl->trefi);
261 
262 	if (ret & 0x10) {
263 		ret = -1;
264 	} else {
265 		ret = (ret & 0xf) ^ bw;
266 		ret = (ret == 0) ? 0 : -1;
267 	}
268 	return ret;
269 }
270 
271 static void move_to_config_state(struct rk322x_ddr_pctl *pctl)
272 {
273 	unsigned int state;
274 
275 	while (1) {
276 		state = readl(&pctl->stat) & PCTL_STAT_MASK;
277 		switch (state) {
278 		case LOW_POWER:
279 			writel(WAKEUP_STATE, &pctl->sctl);
280 			while ((readl(&pctl->stat) & PCTL_STAT_MASK)
281 				!= ACCESS)
282 				;
283 			/*
284 			 * If at low power state, need wakeup first, and then
285 			 * enter the config, so fallthrough
286 			 */
287 		case ACCESS:
288 			/* fallthrough */
289 		case INIT_MEM:
290 			writel(CFG_STATE, &pctl->sctl);
291 			while ((readl(&pctl->stat) & PCTL_STAT_MASK) != CONFIG)
292 				;
293 			break;
294 		case CONFIG:
295 			return;
296 		default:
297 			break;
298 		}
299 	}
300 }
301 
302 static void move_to_access_state(struct rk322x_ddr_pctl *pctl)
303 {
304 	unsigned int state;
305 
306 	while (1) {
307 		state = readl(&pctl->stat) & PCTL_STAT_MASK;
308 		switch (state) {
309 		case LOW_POWER:
310 			writel(WAKEUP_STATE, &pctl->sctl);
311 			while ((readl(&pctl->stat) & PCTL_STAT_MASK) != ACCESS)
312 				;
313 			break;
314 		case INIT_MEM:
315 			writel(CFG_STATE, &pctl->sctl);
316 			while ((readl(&pctl->stat) & PCTL_STAT_MASK) != CONFIG)
317 				;
318 			/* fallthrough */
319 		case CONFIG:
320 			writel(GO_STATE, &pctl->sctl);
321 			while ((readl(&pctl->stat) & PCTL_STAT_MASK) != ACCESS)
322 				;
323 			break;
324 		case ACCESS:
325 			return;
326 		default:
327 			break;
328 		}
329 	}
330 }
331 
332 static void move_to_lowpower_state(struct rk322x_ddr_pctl *pctl)
333 {
334 	unsigned int state;
335 
336 	while (1) {
337 		state = readl(&pctl->stat) & PCTL_STAT_MASK;
338 		switch (state) {
339 		case INIT_MEM:
340 			writel(CFG_STATE, &pctl->sctl);
341 			while ((readl(&pctl->stat) & PCTL_STAT_MASK) != CONFIG)
342 				;
343 			/* fallthrough */
344 		case CONFIG:
345 			writel(GO_STATE, &pctl->sctl);
346 			while ((readl(&pctl->stat) & PCTL_STAT_MASK) != ACCESS)
347 				;
348 			break;
349 		case ACCESS:
350 			writel(SLEEP_STATE, &pctl->sctl);
351 			while ((readl(&pctl->stat) & PCTL_STAT_MASK) !=
352 			       LOW_POWER)
353 				;
354 			break;
355 		case LOW_POWER:
356 			return;
357 		default:
358 			break;
359 		}
360 	}
361 }
362 
363 /* pctl should in low power mode when call this function */
364 static void phy_softreset(struct dram_info *dram)
365 {
366 	struct rk322x_ddr_phy *ddr_phy = dram->chan[0].phy;
367 	struct rk322x_grf *grf = dram->grf;
368 
369 	writel(GRF_DDRPHY_BUFFEREN_CORE_EN, &grf->soc_con[0]);
370 	clrbits_le32(&ddr_phy->ddrphy_reg[0], 0x3 << 2);
371 	rockchip_udelay(1);
372 	setbits_le32(&ddr_phy->ddrphy_reg[0], 1 << 2);
373 	rockchip_udelay(5);
374 	setbits_le32(&ddr_phy->ddrphy_reg[0], 1 << 3);
375 	writel(GRF_DDRPHY_BUFFEREN_CORE_DIS, &grf->soc_con[0]);
376 }
377 
378 /* bw: 2: 32bit, 1:16bit */
379 static void set_bw(struct dram_info *dram, u32 bw)
380 {
381 	struct rk322x_ddr_pctl *pctl = dram->chan[0].pctl;
382 	struct rk322x_ddr_phy *ddr_phy = dram->chan[0].phy;
383 	struct rk322x_grf *grf = dram->grf;
384 
385 	if (bw == 1) {
386 		setbits_le32(&pctl->ppcfg, 1);
387 		clrbits_le32(&ddr_phy->ddrphy_reg[0], 0xc << 4);
388 		writel(GRF_MSCH_NOC_16BIT_EN, &grf->soc_con[0]);
389 		clrbits_le32(&ddr_phy->ddrphy_reg[0x46], 0x8);
390 		clrbits_le32(&ddr_phy->ddrphy_reg[0x56], 0x8);
391 	} else {
392 		clrbits_le32(&pctl->ppcfg, 1);
393 		setbits_le32(&ddr_phy->ddrphy_reg[0], 0xf << 4);
394 		writel(GRF_DDR_32BIT_EN | GRF_MSCH_NOC_32BIT_EN,
395 		       &grf->soc_con[0]);
396 		setbits_le32(&ddr_phy->ddrphy_reg[0x46], 0x8);
397 		setbits_le32(&ddr_phy->ddrphy_reg[0x56], 0x8);
398 	}
399 }
400 
401 static void pctl_cfg(struct rk322x_ddr_pctl *pctl,
402 		     struct rk322x_sdram_params *sdram_params,
403 		     struct rk322x_grf *grf)
404 {
405 	u32 burst_len;
406 	u32 bw;
407 	u32 dramtype = sdram_params->base.dramtype;
408 
409 	if (sdram_params->ch[0].bw == 2)
410 		bw = GRF_DDR_32BIT_EN | GRF_MSCH_NOC_32BIT_EN;
411 	else
412 		bw = GRF_MSCH_NOC_16BIT_EN;
413 
414 	writel(DFI_INIT_START | DFI_DATA_BYTE_DISABLE_EN, &pctl->dfistcfg0);
415 	writel(DFI_DRAM_CLK_SR_EN | DFI_DRAM_CLK_DPD_EN, &pctl->dfistcfg1);
416 	writel(DFI_PARITY_INTR_EN | DFI_PARITY_EN, &pctl->dfistcfg2);
417 	writel(0x51010, &pctl->dfilpcfg0);
418 
419 	writel(1, &pctl->dfitphyupdtype0);
420 	writel(0x0d, &pctl->dfitphyrdlat);
421 	writel(0, &pctl->dfitphywrdata);
422 
423 	writel(0, &pctl->dfiupdcfg);
424 	copy_to_reg(&pctl->togcnt1u, &sdram_params->pctl_timing.togcnt1u,
425 		    sizeof(struct rk322x_pctl_timing));
426 	if (dramtype == DDR3) {
427 		writel((1 << 3) | (1 << 11),
428 		       &pctl->dfiodtcfg);
429 		writel(7 << 16, &pctl->dfiodtcfg1);
430 		writel((readl(&pctl->tcl) - 1) / 2 - 1, &pctl->dfitrddataen);
431 		writel((readl(&pctl->tcwl) - 1) / 2 - 1, &pctl->dfitphywrlat);
432 		writel(500, &pctl->trsth);
433 		writel(0 << MDDR_LPDDR2_CLK_STOP_IDLE_SHIFT | DDR3_EN |
434 		       DDR2_DDR3_BL_8 | (6 - 4) << TFAW_SHIFT | PD_EXIT_SLOW |
435 		       1 << PD_TYPE_SHIFT | 0 << PD_IDLE_SHIFT,
436 		       &pctl->mcfg);
437 		writel(bw | GRF_DDR3_EN, &grf->soc_con[0]);
438 	} else {
439 		if (sdram_params->phy_timing.bl & PHT_BL_8)
440 			burst_len = MDDR_LPDDR2_BL_8;
441 		else
442 			burst_len = MDDR_LPDDR2_BL_4;
443 
444 		writel(readl(&pctl->tcl) / 2 - 1, &pctl->dfitrddataen);
445 		writel(readl(&pctl->tcwl) / 2 - 1, &pctl->dfitphywrlat);
446 		writel(0, &pctl->trsth);
447 		if (dramtype == LPDDR2) {
448 			writel(0 << MDDR_LPDDR2_CLK_STOP_IDLE_SHIFT |
449 			       LPDDR2_S4 | LPDDR2_EN | burst_len |
450 			       (6 - 4) << TFAW_SHIFT | PD_EXIT_FAST |
451 			       1 << PD_TYPE_SHIFT | 0 << PD_IDLE_SHIFT,
452 			       &pctl->mcfg);
453 			writel(0, &pctl->dfiodtcfg);
454 			writel(0, &pctl->dfiodtcfg1);
455 		} else {
456 			writel(0 << MDDR_LPDDR2_CLK_STOP_IDLE_SHIFT |
457 			       LPDDR2_S4 | LPDDR3_EN | burst_len |
458 			       (6 - 4) << TFAW_SHIFT | PD_EXIT_FAST |
459 			       1 << PD_TYPE_SHIFT | 0 << PD_IDLE_SHIFT,
460 			       &pctl->mcfg);
461 			writel((1 << 3) | (1 << 2), &pctl->dfiodtcfg);
462 			writel((7 << 16) | 4, &pctl->dfiodtcfg1);
463 		}
464 		writel(bw | GRF_LPDDR2_3_EN, &grf->soc_con[0]);
465 	}
466 	setbits_le32(&pctl->scfg, 1);
467 }
468 
469 static void phy_cfg(struct chan_info *chan,
470 		    struct rk322x_sdram_params *sdram_params)
471 {
472 	struct rk322x_ddr_phy *ddr_phy = chan->phy;
473 	struct rk322x_service_sys *axi_bus = chan->msch;
474 	struct rk322x_msch_timings *noc_timing = &sdram_params->base.noc_timing;
475 	struct rk322x_phy_timing *phy_timing = &sdram_params->phy_timing;
476 	struct rk322x_pctl_timing *pctl_timing = &sdram_params->pctl_timing;
477 	u32 cmd_drv, clk_drv, dqs_drv, dqs_odt;
478 
479 	writel(noc_timing->ddrtiming, &axi_bus->ddrtiming);
480 	writel(noc_timing->ddrmode, &axi_bus->ddrmode);
481 	writel(noc_timing->readlatency, &axi_bus->readlatency);
482 	writel(noc_timing->activate, &axi_bus->activate);
483 	writel(noc_timing->devtodev, &axi_bus->devtodev);
484 
485 	switch (sdram_params->base.dramtype) {
486 	case DDR3:
487 		writel(PHY_DDR3 | phy_timing->bl, &ddr_phy->ddrphy_reg[1]);
488 		break;
489 	case LPDDR2:
490 		writel(PHY_LPDDR2 | phy_timing->bl, &ddr_phy->ddrphy_reg[1]);
491 		break;
492 	default:
493 		writel(PHY_LPDDR2 | phy_timing->bl, &ddr_phy->ddrphy_reg[1]);
494 		break;
495 	}
496 
497 	writel(phy_timing->cl_al, &ddr_phy->ddrphy_reg[0xb]);
498 	writel(pctl_timing->tcwl, &ddr_phy->ddrphy_reg[0xc]);
499 
500 	cmd_drv = PHY_RON_RTT_34OHM;
501 	clk_drv = PHY_RON_RTT_45OHM;
502 	dqs_drv = PHY_RON_RTT_34OHM;
503 	if (sdram_params->base.dramtype == LPDDR2)
504 		dqs_odt = PHY_RON_RTT_DISABLE;
505 	else
506 		dqs_odt = PHY_RON_RTT_225OHM;
507 
508 	writel(cmd_drv, &ddr_phy->ddrphy_reg[0x11]);
509 	clrsetbits_le32(&ddr_phy->ddrphy_reg[0x12], (0x1f << 3), cmd_drv << 3);
510 	writel(clk_drv, &ddr_phy->ddrphy_reg[0x16]);
511 	writel(clk_drv, &ddr_phy->ddrphy_reg[0x18]);
512 
513 	writel(dqs_drv, &ddr_phy->ddrphy_reg[0x20]);
514 	writel(dqs_drv, &ddr_phy->ddrphy_reg[0x2f]);
515 	writel(dqs_drv, &ddr_phy->ddrphy_reg[0x30]);
516 	writel(dqs_drv, &ddr_phy->ddrphy_reg[0x3f]);
517 	writel(dqs_drv, &ddr_phy->ddrphy_reg[0x40]);
518 	writel(dqs_drv, &ddr_phy->ddrphy_reg[0x4f]);
519 	writel(dqs_drv, &ddr_phy->ddrphy_reg[0x50]);
520 	writel(dqs_drv, &ddr_phy->ddrphy_reg[0x5f]);
521 
522 	writel(dqs_odt, &ddr_phy->ddrphy_reg[0x21]);
523 	writel(dqs_odt, &ddr_phy->ddrphy_reg[0x2e]);
524 	writel(dqs_odt, &ddr_phy->ddrphy_reg[0x31]);
525 	writel(dqs_odt, &ddr_phy->ddrphy_reg[0x3e]);
526 	writel(dqs_odt, &ddr_phy->ddrphy_reg[0x41]);
527 	writel(dqs_odt, &ddr_phy->ddrphy_reg[0x4e]);
528 	writel(dqs_odt, &ddr_phy->ddrphy_reg[0x51]);
529 	writel(dqs_odt, &ddr_phy->ddrphy_reg[0x5e]);
530 }
531 
532 void dram_cfg_rbc(struct chan_info *chan,
533 		  struct rk322x_sdram_params *sdram_params)
534 {
535 	char noc_config;
536 	int i = 0;
537 	struct rk322x_sdram_channel *config = &sdram_params->ch[0];
538 	struct rk322x_service_sys *axi_bus = chan->msch;
539 
540 	move_to_config_state(chan->pctl);
541 
542 	if ((config->rank == 2) && (config->cs1_row == config->cs0_row)) {
543 		if ((config->col + config->bw) == 12) {
544 			i = 14;
545 			goto finish;
546 		} else if ((config->col + config->bw) == 11) {
547 			i = 15;
548 			goto finish;
549 		}
550 	}
551 	noc_config = ((config->cs0_row - 13) << 4) | ((config->bk - 2) << 2) |
552 				(config->col + config->bw - 11);
553 	for (i = 0; i < 11; i++) {
554 		if (noc_config == ddr_cfg_2_rbc[i])
555 			break;
556 	}
557 
558 	if (i < 11)
559 		goto finish;
560 
561 	noc_config = ((config->bk - 2) << 6) | ((config->cs0_row - 13) << 4) |
562 				(config->col + config->bw - 11);
563 
564 	for (i = 11; i < 14; i++) {
565 		if (noc_config == ddr_cfg_2_rbc[i])
566 			break;
567 	}
568 	if (i < 14)
569 		goto finish;
570 	else
571 		i = 0;
572 
573 finish:
574 	writel(i, &axi_bus->ddrconf);
575 	move_to_access_state(chan->pctl);
576 }
577 
578 static void dram_all_config(const struct dram_info *dram,
579 			    struct rk322x_sdram_params *sdram_params)
580 {
581 	struct rk322x_sdram_channel *info = &sdram_params->ch[0];
582 	u32 sys_reg = 0;
583 
584 	sys_reg |= sdram_params->base.dramtype << SYS_REG_DDRTYPE_SHIFT;
585 	sys_reg |= (1 - 1) << SYS_REG_NUM_CH_SHIFT;
586 	sys_reg |= info->row_3_4 << SYS_REG_ROW_3_4_SHIFT(0);
587 	sys_reg |= 1 << SYS_REG_CHINFO_SHIFT(0);
588 	sys_reg |= (info->rank - 1) << SYS_REG_RANK_SHIFT(0);
589 	sys_reg |= (info->col - 9) << SYS_REG_COL_SHIFT(0);
590 	sys_reg |= info->bk == 3 ? 0 : 1 << SYS_REG_BK_SHIFT(0);
591 	sys_reg |= (info->cs0_row - 13) << SYS_REG_CS0_ROW_SHIFT(0);
592 	sys_reg |= (info->cs1_row - 13) << SYS_REG_CS1_ROW_SHIFT(0);
593 	sys_reg |= (2 >> info->bw) << SYS_REG_BW_SHIFT(0);
594 	sys_reg |= (2 >> info->dbw) << SYS_REG_DBW_SHIFT(0);
595 
596 	writel(sys_reg, &dram->grf->os_reg[2]);
597 }
598 
599 #define TEST_PATTEN	0x5aa5f00f
600 
601 static int dram_cap_detect(struct dram_info *dram,
602 			   struct rk322x_sdram_params *sdram_params)
603 {
604 	u32 bw, row, col, addr;
605 	u32 ret = 0;
606 	struct rk322x_service_sys *axi_bus = dram->chan[0].msch;
607 
608 	if (sdram_params->base.dramtype == DDR3)
609 		sdram_params->ch[0].dbw = 1;
610 	else
611 		sdram_params->ch[0].dbw = 2;
612 
613 	move_to_config_state(dram->chan[0].pctl);
614 	/* bw detect */
615 	set_bw(dram, 2);
616 	if (data_training(&dram->chan[0]) == 0) {
617 		bw = 2;
618 	} else {
619 		bw = 1;
620 		set_bw(dram, 1);
621 		move_to_lowpower_state(dram->chan[0].pctl);
622 		phy_softreset(dram);
623 		move_to_config_state(dram->chan[0].pctl);
624 		if (data_training(&dram->chan[0])) {
625 			printf("BW detect error\n");
626 			ret = -EINVAL;
627 		}
628 	}
629 	sdram_params->ch[0].bw = bw;
630 	sdram_params->ch[0].bk = 3;
631 
632 	if (bw == 2)
633 		writel(6, &axi_bus->ddrconf);
634 	else
635 		writel(3, &axi_bus->ddrconf);
636 	move_to_access_state(dram->chan[0].pctl);
637 	for (col = 11; col >= 9; col--) {
638 		writel(0, CONFIG_SYS_SDRAM_BASE);
639 		addr = CONFIG_SYS_SDRAM_BASE +
640 			(1 << (col + bw - 1));
641 		writel(TEST_PATTEN, addr);
642 		if ((readl(addr) == TEST_PATTEN) &&
643 		    (readl(CONFIG_SYS_SDRAM_BASE) == 0))
644 			break;
645 	}
646 	if (col == 8) {
647 		printf("Col detect error\n");
648 		ret = -EINVAL;
649 		goto out;
650 	} else {
651 		sdram_params->ch[0].col = col;
652 	}
653 
654 	writel(10, &axi_bus->ddrconf);
655 
656 	/* Detect row*/
657 	for (row = 16; row >= 12; row--) {
658 		writel(0, CONFIG_SYS_SDRAM_BASE);
659 		addr = CONFIG_SYS_SDRAM_BASE + (1u << (row + 11 + 3 - 1));
660 		writel(TEST_PATTEN, addr);
661 		if ((readl(addr) == TEST_PATTEN) &&
662 		    (readl(CONFIG_SYS_SDRAM_BASE) == 0))
663 			break;
664 	}
665 	if (row == 11) {
666 		printf("Row detect error\n");
667 		ret = -EINVAL;
668 	} else {
669 		sdram_params->ch[0].cs1_row = row;
670 		sdram_params->ch[0].row_3_4 = 0;
671 		sdram_params->ch[0].cs0_row = row;
672 	}
673 	/* cs detect */
674 	writel(0, CONFIG_SYS_SDRAM_BASE);
675 	writel(TEST_PATTEN, CONFIG_SYS_SDRAM_BASE + (1u << 30));
676 	writel(~TEST_PATTEN, CONFIG_SYS_SDRAM_BASE + (1u << 30) + 4);
677 	if ((readl(CONFIG_SYS_SDRAM_BASE + (1u << 30)) == TEST_PATTEN) &&
678 	    (readl(CONFIG_SYS_SDRAM_BASE) == 0))
679 		sdram_params->ch[0].rank = 2;
680 	else
681 		sdram_params->ch[0].rank = 1;
682 out:
683 	return ret;
684 }
685 
686 static int sdram_init(struct dram_info *dram,
687 		      struct rk322x_sdram_params *sdram_params)
688 {
689 	int ret;
690 
691 	ret = clk_set_rate(&dram->ddr_clk,
692 			   sdram_params->base.ddr_freq * MHz * 2);
693 	if (ret < 0) {
694 		printf("Could not set DDR clock\n");
695 		return ret;
696 	}
697 
698 	phy_pctrl_reset(dram->cru, dram->chan[0].phy);
699 	phy_dll_bypass_set(dram->chan[0].phy, sdram_params->base.ddr_freq);
700 	pctl_cfg(dram->chan[0].pctl, sdram_params, dram->grf);
701 	phy_cfg(&dram->chan[0], sdram_params);
702 	writel(POWER_UP_START, &dram->chan[0].pctl->powctl);
703 	while (!(readl(&dram->chan[0].pctl->powstat) & POWER_UP_DONE))
704 		;
705 	memory_init(&dram->chan[0], sdram_params);
706 	move_to_access_state(dram->chan[0].pctl);
707 	ret = dram_cap_detect(dram, sdram_params);
708 	if (ret)
709 		goto out;
710 	dram_cfg_rbc(&dram->chan[0], sdram_params);
711 	dram_all_config(dram, sdram_params);
712 out:
713 	return ret;
714 }
715 
716 static int rk322x_dmc_ofdata_to_platdata(struct udevice *dev)
717 {
718 #if !CONFIG_IS_ENABLED(OF_PLATDATA)
719 	struct rk322x_sdram_params *params = dev_get_platdata(dev);
720 	const void *blob = gd->fdt_blob;
721 	int node = dev_of_offset(dev);
722 	int ret;
723 
724 	params->num_channels = 1;
725 
726 	ret = fdtdec_get_int_array(blob, node, "rockchip,pctl-timing",
727 				   (u32 *)&params->pctl_timing,
728 				   sizeof(params->pctl_timing) / sizeof(u32));
729 	if (ret) {
730 		printf("%s: Cannot read rockchip,pctl-timing\n", __func__);
731 		return -EINVAL;
732 	}
733 	ret = fdtdec_get_int_array(blob, node, "rockchip,phy-timing",
734 				   (u32 *)&params->phy_timing,
735 				   sizeof(params->phy_timing) / sizeof(u32));
736 	if (ret) {
737 		printf("%s: Cannot read rockchip,phy-timing\n", __func__);
738 		return -EINVAL;
739 	}
740 	ret = fdtdec_get_int_array(blob, node, "rockchip,sdram-params",
741 				   (u32 *)&params->base,
742 				   sizeof(params->base) / sizeof(u32));
743 	if (ret) {
744 		printf("%s: Cannot read rockchip,sdram-params\n", __func__);
745 		return -EINVAL;
746 	}
747 	ret = regmap_init_mem(dev, &params->map);
748 	if (ret)
749 		return ret;
750 #endif
751 
752 	return 0;
753 }
754 #endif /* CONFIG_TPL_BUILD */
755 
756 #if CONFIG_IS_ENABLED(OF_PLATDATA)
757 static int conv_of_platdata(struct udevice *dev)
758 {
759 	struct rk322x_sdram_params *plat = dev_get_platdata(dev);
760 	struct dtd_rockchip_rk322x_dmc *of_plat = &plat->of_plat;
761 	int ret;
762 
763 	memcpy(&plat->pctl_timing, of_plat->rockchip_pctl_timing,
764 	       sizeof(plat->pctl_timing));
765 	memcpy(&plat->phy_timing, of_plat->rockchip_phy_timing,
766 	       sizeof(plat->phy_timing));
767 	memcpy(&plat->base, of_plat->rockchip_sdram_params, sizeof(plat->base));
768 
769 	plat->num_channels = 1;
770 	ret = regmap_init_mem_platdata(dev, of_plat->reg,
771 				       ARRAY_SIZE(of_plat->reg) / 2,
772 				       &plat->map);
773 	if (ret)
774 		return ret;
775 
776 	return 0;
777 }
778 #endif
779 
780 static int rk322x_dmc_probe(struct udevice *dev)
781 {
782 #ifdef CONFIG_TPL_BUILD
783 	struct rk322x_sdram_params *plat = dev_get_platdata(dev);
784 	int ret;
785 	struct udevice *dev_clk;
786 #endif
787 	struct dram_info *priv = dev_get_priv(dev);
788 
789 	priv->grf = syscon_get_first_range(ROCKCHIP_SYSCON_GRF);
790 #ifdef CONFIG_TPL_BUILD
791 #if CONFIG_IS_ENABLED(OF_PLATDATA)
792 	ret = conv_of_platdata(dev);
793 	if (ret)
794 		return ret;
795 #endif
796 
797 	priv->chan[0].msch = syscon_get_first_range(ROCKCHIP_SYSCON_MSCH);
798 	priv->chan[0].pctl = regmap_get_range(plat->map, 0);
799 	priv->chan[0].phy = regmap_get_range(plat->map, 1);
800 	ret = rockchip_get_clk(&dev_clk);
801 	if (ret)
802 		return ret;
803 	priv->ddr_clk.id = CLK_DDR;
804 	ret = clk_request(dev_clk, &priv->ddr_clk);
805 	if (ret)
806 		return ret;
807 
808 	priv->cru = rockchip_get_cru();
809 	if (IS_ERR(priv->cru))
810 		return PTR_ERR(priv->cru);
811 	ret = sdram_init(priv, plat);
812 	if (ret)
813 		return ret;
814 #else
815 	priv->info.base = CONFIG_SYS_SDRAM_BASE;
816 	priv->info.size = rockchip_sdram_size(
817 			(phys_addr_t)&priv->grf->os_reg[2]);
818 #endif
819 
820 	return 0;
821 }
822 
823 static int rk322x_dmc_get_info(struct udevice *dev, struct ram_info *info)
824 {
825 	struct dram_info *priv = dev_get_priv(dev);
826 
827 	*info = priv->info;
828 
829 	return 0;
830 }
831 
832 static struct ram_ops rk322x_dmc_ops = {
833 	.get_info = rk322x_dmc_get_info,
834 };
835 
836 static const struct udevice_id rk322x_dmc_ids[] = {
837 	{ .compatible = "rockchip,rk3228-dmc" },
838 	{ }
839 };
840 
841 U_BOOT_DRIVER(dmc_rk322x) = {
842 	.name = "rockchip_rk322x_dmc",
843 	.id = UCLASS_RAM,
844 	.of_match = rk322x_dmc_ids,
845 	.ops = &rk322x_dmc_ops,
846 #ifdef CONFIG_TPL_BUILD
847 	.ofdata_to_platdata = rk322x_dmc_ofdata_to_platdata,
848 #endif
849 	.probe = rk322x_dmc_probe,
850 	.priv_auto_alloc_size = sizeof(struct dram_info),
851 #ifdef CONFIG_TPL_BUILD
852 	.platdata_auto_alloc_size = sizeof(struct rk322x_sdram_params),
853 #endif
854 };
855 
856