arm64: dts: qcom: sm8550: add TRNG node
[linux-modified.git] / sound / soc / mediatek / mt8195 / mt8195-afe-clk.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * mt8195-afe-clk.c  --  Mediatek 8195 afe clock ctrl
4  *
5  * Copyright (c) 2021 MediaTek Inc.
6  * Author: Bicycle Tsai <bicycle.tsai@mediatek.com>
7  *         Trevor Wu <trevor.wu@mediatek.com>
8  */
9
10 #include <linux/clk.h>
11
12 #include "mt8195-afe-common.h"
13 #include "mt8195-afe-clk.h"
14 #include "mt8195-reg.h"
15 #include "mt8195-audsys-clk.h"
16
17 static const char *aud_clks[MT8195_CLK_NUM] = {
18         /* xtal */
19         [MT8195_CLK_XTAL_26M] = "clk26m",
20         /* divider */
21         [MT8195_CLK_TOP_APLL1] = "apll1_ck",
22         [MT8195_CLK_TOP_APLL2] = "apll2_ck",
23         [MT8195_CLK_TOP_APLL12_DIV0] = "apll12_div0",
24         [MT8195_CLK_TOP_APLL12_DIV1] = "apll12_div1",
25         [MT8195_CLK_TOP_APLL12_DIV2] = "apll12_div2",
26         [MT8195_CLK_TOP_APLL12_DIV3] = "apll12_div3",
27         [MT8195_CLK_TOP_APLL12_DIV9] = "apll12_div9",
28         /* mux */
29         [MT8195_CLK_TOP_A1SYS_HP_SEL] = "a1sys_hp_sel",
30         [MT8195_CLK_TOP_AUD_INTBUS_SEL] = "aud_intbus_sel",
31         [MT8195_CLK_TOP_AUDIO_H_SEL] = "audio_h_sel",
32         [MT8195_CLK_TOP_AUDIO_LOCAL_BUS_SEL] = "audio_local_bus_sel",
33         [MT8195_CLK_TOP_DPTX_M_SEL] = "dptx_m_sel",
34         [MT8195_CLK_TOP_I2SO1_M_SEL] = "i2so1_m_sel",
35         [MT8195_CLK_TOP_I2SO2_M_SEL] = "i2so2_m_sel",
36         [MT8195_CLK_TOP_I2SI1_M_SEL] = "i2si1_m_sel",
37         [MT8195_CLK_TOP_I2SI2_M_SEL] = "i2si2_m_sel",
38         /* clock gate */
39         [MT8195_CLK_INFRA_AO_AUDIO_26M_B] = "infra_ao_audio_26m_b",
40         [MT8195_CLK_SCP_ADSP_AUDIODSP] = "scp_adsp_audiodsp",
41         /* afe clock gate */
42         [MT8195_CLK_AUD_AFE] = "aud_afe",
43         [MT8195_CLK_AUD_APLL1_TUNER] = "aud_apll1_tuner",
44         [MT8195_CLK_AUD_APLL2_TUNER] = "aud_apll2_tuner",
45         [MT8195_CLK_AUD_APLL] = "aud_apll",
46         [MT8195_CLK_AUD_APLL2] = "aud_apll2",
47         [MT8195_CLK_AUD_DAC] = "aud_dac",
48         [MT8195_CLK_AUD_ADC] = "aud_adc",
49         [MT8195_CLK_AUD_DAC_HIRES] = "aud_dac_hires",
50         [MT8195_CLK_AUD_A1SYS_HP] = "aud_a1sys_hp",
51         [MT8195_CLK_AUD_ADC_HIRES] = "aud_adc_hires",
52         [MT8195_CLK_AUD_ADDA6_ADC] = "aud_adda6_adc",
53         [MT8195_CLK_AUD_ADDA6_ADC_HIRES] = "aud_adda6_adc_hires",
54         [MT8195_CLK_AUD_I2SIN] = "aud_i2sin",
55         [MT8195_CLK_AUD_TDM_IN] = "aud_tdm_in",
56         [MT8195_CLK_AUD_I2S_OUT] = "aud_i2s_out",
57         [MT8195_CLK_AUD_TDM_OUT] = "aud_tdm_out",
58         [MT8195_CLK_AUD_HDMI_OUT] = "aud_hdmi_out",
59         [MT8195_CLK_AUD_ASRC11] = "aud_asrc11",
60         [MT8195_CLK_AUD_ASRC12] = "aud_asrc12",
61         [MT8195_CLK_AUD_A1SYS] = "aud_a1sys",
62         [MT8195_CLK_AUD_A2SYS] = "aud_a2sys",
63         [MT8195_CLK_AUD_PCMIF] = "aud_pcmif",
64         [MT8195_CLK_AUD_MEMIF_UL1] = "aud_memif_ul1",
65         [MT8195_CLK_AUD_MEMIF_UL2] = "aud_memif_ul2",
66         [MT8195_CLK_AUD_MEMIF_UL3] = "aud_memif_ul3",
67         [MT8195_CLK_AUD_MEMIF_UL4] = "aud_memif_ul4",
68         [MT8195_CLK_AUD_MEMIF_UL5] = "aud_memif_ul5",
69         [MT8195_CLK_AUD_MEMIF_UL6] = "aud_memif_ul6",
70         [MT8195_CLK_AUD_MEMIF_UL8] = "aud_memif_ul8",
71         [MT8195_CLK_AUD_MEMIF_UL9] = "aud_memif_ul9",
72         [MT8195_CLK_AUD_MEMIF_UL10] = "aud_memif_ul10",
73         [MT8195_CLK_AUD_MEMIF_DL2] = "aud_memif_dl2",
74         [MT8195_CLK_AUD_MEMIF_DL3] = "aud_memif_dl3",
75         [MT8195_CLK_AUD_MEMIF_DL6] = "aud_memif_dl6",
76         [MT8195_CLK_AUD_MEMIF_DL7] = "aud_memif_dl7",
77         [MT8195_CLK_AUD_MEMIF_DL8] = "aud_memif_dl8",
78         [MT8195_CLK_AUD_MEMIF_DL10] = "aud_memif_dl10",
79         [MT8195_CLK_AUD_MEMIF_DL11] = "aud_memif_dl11",
80 };
81
82 struct mt8195_afe_tuner_cfg {
83         unsigned int id;
84         int apll_div_reg;
85         unsigned int apll_div_shift;
86         unsigned int apll_div_maskbit;
87         unsigned int apll_div_default;
88         int ref_ck_sel_reg;
89         unsigned int ref_ck_sel_shift;
90         unsigned int ref_ck_sel_maskbit;
91         unsigned int ref_ck_sel_default;
92         int tuner_en_reg;
93         unsigned int tuner_en_shift;
94         unsigned int tuner_en_maskbit;
95         int upper_bound_reg;
96         unsigned int upper_bound_shift;
97         unsigned int upper_bound_maskbit;
98         unsigned int upper_bound_default;
99         spinlock_t ctrl_lock; /* lock for apll tuner ctrl*/
100         int ref_cnt;
101 };
102
103 static struct mt8195_afe_tuner_cfg mt8195_afe_tuner_cfgs[MT8195_AUD_PLL_NUM] = {
104         [MT8195_AUD_PLL1] = {
105                 .id = MT8195_AUD_PLL1,
106                 .apll_div_reg = AFE_APLL_TUNER_CFG,
107                 .apll_div_shift = 4,
108                 .apll_div_maskbit = 0xf,
109                 .apll_div_default = 0x7,
110                 .ref_ck_sel_reg = AFE_APLL_TUNER_CFG,
111                 .ref_ck_sel_shift = 1,
112                 .ref_ck_sel_maskbit = 0x3,
113                 .ref_ck_sel_default = 0x2,
114                 .tuner_en_reg = AFE_APLL_TUNER_CFG,
115                 .tuner_en_shift = 0,
116                 .tuner_en_maskbit = 0x1,
117                 .upper_bound_reg = AFE_APLL_TUNER_CFG,
118                 .upper_bound_shift = 8,
119                 .upper_bound_maskbit = 0xff,
120                 .upper_bound_default = 0x3,
121         },
122         [MT8195_AUD_PLL2] = {
123                 .id = MT8195_AUD_PLL2,
124                 .apll_div_reg = AFE_APLL_TUNER_CFG1,
125                 .apll_div_shift = 4,
126                 .apll_div_maskbit = 0xf,
127                 .apll_div_default = 0x7,
128                 .ref_ck_sel_reg = AFE_APLL_TUNER_CFG1,
129                 .ref_ck_sel_shift = 1,
130                 .ref_ck_sel_maskbit = 0x3,
131                 .ref_ck_sel_default = 0x1,
132                 .tuner_en_reg = AFE_APLL_TUNER_CFG1,
133                 .tuner_en_shift = 0,
134                 .tuner_en_maskbit = 0x1,
135                 .upper_bound_reg = AFE_APLL_TUNER_CFG1,
136                 .upper_bound_shift = 8,
137                 .upper_bound_maskbit = 0xff,
138                 .upper_bound_default = 0x3,
139         },
140         [MT8195_AUD_PLL3] = {
141                 .id = MT8195_AUD_PLL3,
142                 .apll_div_reg = AFE_EARC_APLL_TUNER_CFG,
143                 .apll_div_shift = 4,
144                 .apll_div_maskbit = 0x3f,
145                 .apll_div_default = 0x3,
146                 .ref_ck_sel_reg = AFE_EARC_APLL_TUNER_CFG,
147                 .ref_ck_sel_shift = 24,
148                 .ref_ck_sel_maskbit = 0x3,
149                 .ref_ck_sel_default = 0x0,
150                 .tuner_en_reg = AFE_EARC_APLL_TUNER_CFG,
151                 .tuner_en_shift = 0,
152                 .tuner_en_maskbit = 0x1,
153                 .upper_bound_reg = AFE_EARC_APLL_TUNER_CFG,
154                 .upper_bound_shift = 12,
155                 .upper_bound_maskbit = 0xff,
156                 .upper_bound_default = 0x4,
157         },
158         [MT8195_AUD_PLL4] = {
159                 .id = MT8195_AUD_PLL4,
160                 .apll_div_reg = AFE_SPDIFIN_APLL_TUNER_CFG,
161                 .apll_div_shift = 4,
162                 .apll_div_maskbit = 0x3f,
163                 .apll_div_default = 0x7,
164                 .ref_ck_sel_reg = AFE_SPDIFIN_APLL_TUNER_CFG1,
165                 .ref_ck_sel_shift = 8,
166                 .ref_ck_sel_maskbit = 0x1,
167                 .ref_ck_sel_default = 0,
168                 .tuner_en_reg = AFE_SPDIFIN_APLL_TUNER_CFG,
169                 .tuner_en_shift = 0,
170                 .tuner_en_maskbit = 0x1,
171                 .upper_bound_reg = AFE_SPDIFIN_APLL_TUNER_CFG,
172                 .upper_bound_shift = 12,
173                 .upper_bound_maskbit = 0xff,
174                 .upper_bound_default = 0x4,
175         },
176         [MT8195_AUD_PLL5] = {
177                 .id = MT8195_AUD_PLL5,
178                 .apll_div_reg = AFE_LINEIN_APLL_TUNER_CFG,
179                 .apll_div_shift = 4,
180                 .apll_div_maskbit = 0x3f,
181                 .apll_div_default = 0x3,
182                 .ref_ck_sel_reg = AFE_LINEIN_APLL_TUNER_CFG,
183                 .ref_ck_sel_shift = 24,
184                 .ref_ck_sel_maskbit = 0x1,
185                 .ref_ck_sel_default = 0,
186                 .tuner_en_reg = AFE_LINEIN_APLL_TUNER_CFG,
187                 .tuner_en_shift = 0,
188                 .tuner_en_maskbit = 0x1,
189                 .upper_bound_reg = AFE_LINEIN_APLL_TUNER_CFG,
190                 .upper_bound_shift = 12,
191                 .upper_bound_maskbit = 0xff,
192                 .upper_bound_default = 0x4,
193         },
194 };
195
196 static struct mt8195_afe_tuner_cfg *mt8195_afe_found_apll_tuner(unsigned int id)
197 {
198         if (id >= MT8195_AUD_PLL_NUM)
199                 return NULL;
200
201         return &mt8195_afe_tuner_cfgs[id];
202 }
203
204 static int mt8195_afe_init_apll_tuner(unsigned int id)
205 {
206         struct mt8195_afe_tuner_cfg *cfg = mt8195_afe_found_apll_tuner(id);
207
208         if (!cfg)
209                 return -EINVAL;
210
211         cfg->ref_cnt = 0;
212         spin_lock_init(&cfg->ctrl_lock);
213
214         return 0;
215 }
216
217 static int mt8195_afe_setup_apll_tuner(struct mtk_base_afe *afe,
218                                        unsigned int id)
219 {
220         const struct mt8195_afe_tuner_cfg *cfg = mt8195_afe_found_apll_tuner(id);
221
222         if (!cfg)
223                 return -EINVAL;
224
225         regmap_update_bits(afe->regmap, cfg->apll_div_reg,
226                            cfg->apll_div_maskbit << cfg->apll_div_shift,
227                            cfg->apll_div_default << cfg->apll_div_shift);
228
229         regmap_update_bits(afe->regmap, cfg->ref_ck_sel_reg,
230                            cfg->ref_ck_sel_maskbit << cfg->ref_ck_sel_shift,
231                            cfg->ref_ck_sel_default << cfg->ref_ck_sel_shift);
232
233         regmap_update_bits(afe->regmap, cfg->upper_bound_reg,
234                            cfg->upper_bound_maskbit << cfg->upper_bound_shift,
235                            cfg->upper_bound_default << cfg->upper_bound_shift);
236
237         return 0;
238 }
239
240 static int mt8195_afe_enable_tuner_clk(struct mtk_base_afe *afe,
241                                        unsigned int id)
242 {
243         struct mt8195_afe_private *afe_priv = afe->platform_priv;
244
245         switch (id) {
246         case MT8195_AUD_PLL1:
247                 mt8195_afe_enable_clk(afe, afe_priv->clk[MT8195_CLK_AUD_APLL]);
248                 mt8195_afe_enable_clk(afe, afe_priv->clk[MT8195_CLK_AUD_APLL1_TUNER]);
249                 break;
250         case MT8195_AUD_PLL2:
251                 mt8195_afe_enable_clk(afe, afe_priv->clk[MT8195_CLK_AUD_APLL2]);
252                 mt8195_afe_enable_clk(afe, afe_priv->clk[MT8195_CLK_AUD_APLL2_TUNER]);
253                 break;
254         default:
255                 break;
256         }
257
258         return 0;
259 }
260
261 static int mt8195_afe_disable_tuner_clk(struct mtk_base_afe *afe,
262                                         unsigned int id)
263 {
264         struct mt8195_afe_private *afe_priv = afe->platform_priv;
265
266         switch (id) {
267         case MT8195_AUD_PLL1:
268                 mt8195_afe_disable_clk(afe, afe_priv->clk[MT8195_CLK_AUD_APLL1_TUNER]);
269                 mt8195_afe_disable_clk(afe, afe_priv->clk[MT8195_CLK_AUD_APLL]);
270                 break;
271         case MT8195_AUD_PLL2:
272                 mt8195_afe_disable_clk(afe, afe_priv->clk[MT8195_CLK_AUD_APLL2_TUNER]);
273                 mt8195_afe_disable_clk(afe, afe_priv->clk[MT8195_CLK_AUD_APLL2]);
274                 break;
275         default:
276                 break;
277         }
278
279         return 0;
280 }
281
282 static int mt8195_afe_enable_apll_tuner(struct mtk_base_afe *afe,
283                                         unsigned int id)
284 {
285         struct mt8195_afe_tuner_cfg *cfg = mt8195_afe_found_apll_tuner(id);
286         unsigned long flags;
287         int ret;
288
289         if (!cfg)
290                 return -EINVAL;
291
292         ret = mt8195_afe_setup_apll_tuner(afe, id);
293         if (ret)
294                 return ret;
295
296         ret = mt8195_afe_enable_tuner_clk(afe, id);
297         if (ret)
298                 return ret;
299
300         spin_lock_irqsave(&cfg->ctrl_lock, flags);
301
302         cfg->ref_cnt++;
303         if (cfg->ref_cnt == 1)
304                 regmap_update_bits(afe->regmap,
305                                    cfg->tuner_en_reg,
306                                    cfg->tuner_en_maskbit << cfg->tuner_en_shift,
307                                    1 << cfg->tuner_en_shift);
308
309         spin_unlock_irqrestore(&cfg->ctrl_lock, flags);
310
311         return 0;
312 }
313
314 static int mt8195_afe_disable_apll_tuner(struct mtk_base_afe *afe,
315                                          unsigned int id)
316 {
317         struct mt8195_afe_tuner_cfg *cfg = mt8195_afe_found_apll_tuner(id);
318         unsigned long flags;
319         int ret;
320
321         if (!cfg)
322                 return -EINVAL;
323
324         spin_lock_irqsave(&cfg->ctrl_lock, flags);
325
326         cfg->ref_cnt--;
327         if (cfg->ref_cnt == 0)
328                 regmap_update_bits(afe->regmap,
329                                    cfg->tuner_en_reg,
330                                    cfg->tuner_en_maskbit << cfg->tuner_en_shift,
331                                    0 << cfg->tuner_en_shift);
332         else if (cfg->ref_cnt < 0)
333                 cfg->ref_cnt = 0;
334
335         spin_unlock_irqrestore(&cfg->ctrl_lock, flags);
336
337         ret = mt8195_afe_disable_tuner_clk(afe, id);
338         if (ret)
339                 return ret;
340
341         return 0;
342 }
343
344 int mt8195_afe_get_mclk_source_clk_id(int sel)
345 {
346         switch (sel) {
347         case MT8195_MCK_SEL_26M:
348                 return MT8195_CLK_XTAL_26M;
349         case MT8195_MCK_SEL_APLL1:
350                 return MT8195_CLK_TOP_APLL1;
351         case MT8195_MCK_SEL_APLL2:
352                 return MT8195_CLK_TOP_APLL2;
353         default:
354                 return -EINVAL;
355         }
356 }
357
358 int mt8195_afe_get_mclk_source_rate(struct mtk_base_afe *afe, int apll)
359 {
360         struct mt8195_afe_private *afe_priv = afe->platform_priv;
361         int clk_id = mt8195_afe_get_mclk_source_clk_id(apll);
362
363         if (clk_id < 0) {
364                 dev_dbg(afe->dev, "invalid clk id\n");
365                 return 0;
366         }
367
368         return clk_get_rate(afe_priv->clk[clk_id]);
369 }
370
371 int mt8195_afe_get_default_mclk_source_by_rate(int rate)
372 {
373         return ((rate % 8000) == 0) ?
374                 MT8195_MCK_SEL_APLL1 : MT8195_MCK_SEL_APLL2;
375 }
376
377 int mt8195_afe_init_clock(struct mtk_base_afe *afe)
378 {
379         struct mt8195_afe_private *afe_priv = afe->platform_priv;
380         int i, ret;
381
382         mt8195_audsys_clk_register(afe);
383
384         afe_priv->clk =
385                 devm_kcalloc(afe->dev, MT8195_CLK_NUM, sizeof(*afe_priv->clk),
386                              GFP_KERNEL);
387         if (!afe_priv->clk)
388                 return -ENOMEM;
389
390         for (i = 0; i < MT8195_CLK_NUM; i++) {
391                 afe_priv->clk[i] = devm_clk_get(afe->dev, aud_clks[i]);
392                 if (IS_ERR(afe_priv->clk[i])) {
393                         dev_dbg(afe->dev, "%s(), devm_clk_get %s fail, ret %ld\n",
394                                 __func__, aud_clks[i],
395                                 PTR_ERR(afe_priv->clk[i]));
396                         return PTR_ERR(afe_priv->clk[i]);
397                 }
398         }
399
400         /* initial tuner */
401         for (i = 0; i < MT8195_AUD_PLL_NUM; i++) {
402                 ret = mt8195_afe_init_apll_tuner(i);
403                 if (ret) {
404                         dev_dbg(afe->dev, "%s(), init apll_tuner%d failed",
405                                 __func__, (i + 1));
406                         return -EINVAL;
407                 }
408         }
409
410         return 0;
411 }
412
413 int mt8195_afe_enable_clk(struct mtk_base_afe *afe, struct clk *clk)
414 {
415         int ret;
416
417         if (clk) {
418                 ret = clk_prepare_enable(clk);
419                 if (ret) {
420                         dev_dbg(afe->dev, "%s(), failed to enable clk\n",
421                                 __func__);
422                         return ret;
423                 }
424         } else {
425                 dev_dbg(afe->dev, "NULL clk\n");
426         }
427         return 0;
428 }
429 EXPORT_SYMBOL_GPL(mt8195_afe_enable_clk);
430
431 void mt8195_afe_disable_clk(struct mtk_base_afe *afe, struct clk *clk)
432 {
433         if (clk)
434                 clk_disable_unprepare(clk);
435         else
436                 dev_dbg(afe->dev, "NULL clk\n");
437 }
438 EXPORT_SYMBOL_GPL(mt8195_afe_disable_clk);
439
440 int mt8195_afe_prepare_clk(struct mtk_base_afe *afe, struct clk *clk)
441 {
442         int ret;
443
444         if (clk) {
445                 ret = clk_prepare(clk);
446                 if (ret) {
447                         dev_dbg(afe->dev, "%s(), failed to prepare clk\n",
448                                 __func__);
449                         return ret;
450                 }
451         } else {
452                 dev_dbg(afe->dev, "NULL clk\n");
453         }
454         return 0;
455 }
456
457 void mt8195_afe_unprepare_clk(struct mtk_base_afe *afe, struct clk *clk)
458 {
459         if (clk)
460                 clk_unprepare(clk);
461         else
462                 dev_dbg(afe->dev, "NULL clk\n");
463 }
464
465 int mt8195_afe_enable_clk_atomic(struct mtk_base_afe *afe, struct clk *clk)
466 {
467         int ret;
468
469         if (clk) {
470                 ret = clk_enable(clk);
471                 if (ret) {
472                         dev_dbg(afe->dev, "%s(), failed to clk enable\n",
473                                 __func__);
474                         return ret;
475                 }
476         } else {
477                 dev_dbg(afe->dev, "NULL clk\n");
478         }
479         return 0;
480 }
481
482 void mt8195_afe_disable_clk_atomic(struct mtk_base_afe *afe, struct clk *clk)
483 {
484         if (clk)
485                 clk_disable(clk);
486         else
487                 dev_dbg(afe->dev, "NULL clk\n");
488 }
489
490 int mt8195_afe_set_clk_rate(struct mtk_base_afe *afe, struct clk *clk,
491                             unsigned int rate)
492 {
493         int ret;
494
495         if (clk) {
496                 ret = clk_set_rate(clk, rate);
497                 if (ret) {
498                         dev_dbg(afe->dev, "%s(), failed to set clk rate\n",
499                                 __func__);
500                         return ret;
501                 }
502         }
503
504         return 0;
505 }
506
507 int mt8195_afe_set_clk_parent(struct mtk_base_afe *afe, struct clk *clk,
508                               struct clk *parent)
509 {
510         int ret;
511
512         if (clk && parent) {
513                 ret = clk_set_parent(clk, parent);
514                 if (ret) {
515                         dev_dbg(afe->dev, "%s(), failed to set clk parent\n",
516                                 __func__);
517                         return ret;
518                 }
519         }
520
521         return 0;
522 }
523
524 static unsigned int get_top_cg_reg(unsigned int cg_type)
525 {
526         switch (cg_type) {
527         case MT8195_TOP_CG_A1SYS_TIMING:
528         case MT8195_TOP_CG_A2SYS_TIMING:
529         case MT8195_TOP_CG_26M_TIMING:
530                 return ASYS_TOP_CON;
531         default:
532                 return 0;
533         }
534 }
535
536 static unsigned int get_top_cg_mask(unsigned int cg_type)
537 {
538         switch (cg_type) {
539         case MT8195_TOP_CG_A1SYS_TIMING:
540                 return ASYS_TOP_CON_A1SYS_TIMING_ON;
541         case MT8195_TOP_CG_A2SYS_TIMING:
542                 return ASYS_TOP_CON_A2SYS_TIMING_ON;
543         case MT8195_TOP_CG_26M_TIMING:
544                 return ASYS_TOP_CON_26M_TIMING_ON;
545         default:
546                 return 0;
547         }
548 }
549
550 static unsigned int get_top_cg_on_val(unsigned int cg_type)
551 {
552         switch (cg_type) {
553         case MT8195_TOP_CG_A1SYS_TIMING:
554         case MT8195_TOP_CG_A2SYS_TIMING:
555         case MT8195_TOP_CG_26M_TIMING:
556                 return get_top_cg_mask(cg_type);
557         default:
558                 return 0;
559         }
560 }
561
562 static unsigned int get_top_cg_off_val(unsigned int cg_type)
563 {
564         switch (cg_type) {
565         case MT8195_TOP_CG_A1SYS_TIMING:
566         case MT8195_TOP_CG_A2SYS_TIMING:
567         case MT8195_TOP_CG_26M_TIMING:
568                 return 0;
569         default:
570                 return get_top_cg_mask(cg_type);
571         }
572 }
573
574 static int mt8195_afe_enable_top_cg(struct mtk_base_afe *afe, unsigned int cg_type)
575 {
576         unsigned int reg = get_top_cg_reg(cg_type);
577         unsigned int mask = get_top_cg_mask(cg_type);
578         unsigned int val = get_top_cg_on_val(cg_type);
579
580         regmap_update_bits(afe->regmap, reg, mask, val);
581         return 0;
582 }
583
584 static int mt8195_afe_disable_top_cg(struct mtk_base_afe *afe, unsigned int cg_type)
585 {
586         unsigned int reg = get_top_cg_reg(cg_type);
587         unsigned int mask = get_top_cg_mask(cg_type);
588         unsigned int val = get_top_cg_off_val(cg_type);
589
590         regmap_update_bits(afe->regmap, reg, mask, val);
591         return 0;
592 }
593
594 int mt8195_afe_enable_reg_rw_clk(struct mtk_base_afe *afe)
595 {
596         struct mt8195_afe_private *afe_priv = afe->platform_priv;
597         int i;
598         static const unsigned int clk_array[] = {
599                 MT8195_CLK_SCP_ADSP_AUDIODSP, /* bus clock for infra */
600                 MT8195_CLK_TOP_AUDIO_H_SEL, /* clock for ADSP bus */
601                 MT8195_CLK_TOP_AUDIO_LOCAL_BUS_SEL, /* bus clock for DRAM access */
602                 MT8195_CLK_TOP_AUD_INTBUS_SEL, /* bus clock for AFE SRAM access */
603                 MT8195_CLK_INFRA_AO_AUDIO_26M_B, /* audio 26M clock */
604                 MT8195_CLK_AUD_AFE, /* AFE HW master switch */
605                 MT8195_CLK_AUD_A1SYS_HP, /* AFE HW clock*/
606                 MT8195_CLK_AUD_A1SYS, /* AFE HW clock */
607         };
608
609         for (i = 0; i < ARRAY_SIZE(clk_array); i++)
610                 mt8195_afe_enable_clk(afe, afe_priv->clk[clk_array[i]]);
611
612         return 0;
613 }
614
615 int mt8195_afe_disable_reg_rw_clk(struct mtk_base_afe *afe)
616 {
617         struct mt8195_afe_private *afe_priv = afe->platform_priv;
618         int i;
619         static const unsigned int clk_array[] = {
620                 MT8195_CLK_AUD_A1SYS,
621                 MT8195_CLK_AUD_A1SYS_HP,
622                 MT8195_CLK_AUD_AFE,
623                 MT8195_CLK_INFRA_AO_AUDIO_26M_B,
624                 MT8195_CLK_TOP_AUD_INTBUS_SEL,
625                 MT8195_CLK_TOP_AUDIO_LOCAL_BUS_SEL,
626                 MT8195_CLK_TOP_AUDIO_H_SEL,
627                 MT8195_CLK_SCP_ADSP_AUDIODSP,
628         };
629
630         for (i = 0; i < ARRAY_SIZE(clk_array); i++)
631                 mt8195_afe_disable_clk(afe, afe_priv->clk[clk_array[i]]);
632
633         return 0;
634 }
635
636 static int mt8195_afe_enable_afe_on(struct mtk_base_afe *afe)
637 {
638         regmap_update_bits(afe->regmap, AFE_DAC_CON0, 0x1, 0x1);
639         return 0;
640 }
641
642 static int mt8195_afe_disable_afe_on(struct mtk_base_afe *afe)
643 {
644         regmap_update_bits(afe->regmap, AFE_DAC_CON0, 0x1, 0x0);
645         return 0;
646 }
647
648 static int mt8195_afe_enable_timing_sys(struct mtk_base_afe *afe)
649 {
650         struct mt8195_afe_private *afe_priv = afe->platform_priv;
651         int i;
652         static const unsigned int clk_array[] = {
653                 MT8195_CLK_AUD_A1SYS,
654                 MT8195_CLK_AUD_A2SYS,
655         };
656         static const unsigned int cg_array[] = {
657                 MT8195_TOP_CG_A1SYS_TIMING,
658                 MT8195_TOP_CG_A2SYS_TIMING,
659                 MT8195_TOP_CG_26M_TIMING,
660         };
661
662         for (i = 0; i < ARRAY_SIZE(clk_array); i++)
663                 mt8195_afe_enable_clk(afe, afe_priv->clk[clk_array[i]]);
664
665         for (i = 0; i < ARRAY_SIZE(cg_array); i++)
666                 mt8195_afe_enable_top_cg(afe, cg_array[i]);
667
668         return 0;
669 }
670
671 static int mt8195_afe_disable_timing_sys(struct mtk_base_afe *afe)
672 {
673         struct mt8195_afe_private *afe_priv = afe->platform_priv;
674         int i;
675         static const unsigned int clk_array[] = {
676                 MT8195_CLK_AUD_A2SYS,
677                 MT8195_CLK_AUD_A1SYS,
678         };
679         static const unsigned int cg_array[] = {
680                 MT8195_TOP_CG_26M_TIMING,
681                 MT8195_TOP_CG_A2SYS_TIMING,
682                 MT8195_TOP_CG_A1SYS_TIMING,
683         };
684
685         for (i = 0; i < ARRAY_SIZE(cg_array); i++)
686                 mt8195_afe_disable_top_cg(afe, cg_array[i]);
687
688         for (i = 0; i < ARRAY_SIZE(clk_array); i++)
689                 mt8195_afe_disable_clk(afe, afe_priv->clk[clk_array[i]]);
690
691         return 0;
692 }
693
694 int mt8195_afe_enable_main_clock(struct mtk_base_afe *afe)
695 {
696         mt8195_afe_enable_timing_sys(afe);
697
698         mt8195_afe_enable_afe_on(afe);
699
700         mt8195_afe_enable_apll_tuner(afe, MT8195_AUD_PLL1);
701         mt8195_afe_enable_apll_tuner(afe, MT8195_AUD_PLL2);
702
703         return 0;
704 }
705
706 int mt8195_afe_disable_main_clock(struct mtk_base_afe *afe)
707 {
708         mt8195_afe_disable_apll_tuner(afe, MT8195_AUD_PLL2);
709         mt8195_afe_disable_apll_tuner(afe, MT8195_AUD_PLL1);
710
711         mt8195_afe_disable_afe_on(afe);
712
713         mt8195_afe_disable_timing_sys(afe);
714
715         return 0;
716 }