1 files changed, 42 insertions, 26 deletions

diff --git a/src/soc/mediatek/mt8173/mtcmos.c b/src/soc/mediatek/mt8173/mtcmos.c
index cdd31f4bc1..00e37edb9b 100644
--- a/src/soc/mediatek/mt8173/mtcmos.c
+++ b/src/soc/mediatek/mt8173/mtcmos.c
@@ -13,10 +13,19 @@
  * GNU General Public License for more details.
  */
 
+#include <stddef.h>
+
 #include <arch/io.h>
 #include <soc/mtcmos.h>
 #include <soc/spm.h>
 
+struct power_domain_data {
+	void *pwr_con;
+	u32 pwr_sta_mask;
+	u32 sram_pdn_mask;
+	u32 sram_ack_mask;
+};
+
 enum {
 	SRAM_ISOINT_B	= 1U << 6,
 	SRAM_CKISO	= 1U << 5,
@@ -28,44 +37,51 @@ enum {
 };
 
 enum {
-	SRAM_PDN           = 0xf << 8,
-	DIS_SRAM_ACK       = 0x1 << 12,
-	AUD_SRAM_ACK       = 0xf << 12,
-};
-
-enum {
-	DIS_PWR_STA_MASK   = 0x1 << 3,
-	AUD_PWR_STA_MASK   = 0x1 << 24,
+	DISP_PWR_STA_MASK	= 0x1 << 3,
+	AUDIO_PWR_STA_MASK	= 0x1 << 24,
 };
 
-static void mtcmos_power_on(u32 *pwr_con, u32 pwr_sta_mask)
+static void mtcmos_power_on(const struct power_domain_data *pd)
 {
-	write32(&mt8173_spm->poweron_config_set,
+	write32(&mtk_spm->poweron_config_set,
 		(SPM_PROJECT_CODE << 16) | (1U << 0));
 
-	setbits_le32(pwr_con, PWR_ON);
-	setbits_le32(pwr_con, PWR_ON_2ND);
+	setbits_le32(pd->pwr_con, PWR_ON);
+	setbits_le32(pd->pwr_con, PWR_ON_2ND);
 
-	while (!(read32(&mt8173_spm->pwr_status) & pwr_sta_mask) ||
-	       !(read32(&mt8173_spm->pwr_status_2nd) & pwr_sta_mask))
+	while (!(read32(&mtk_spm->pwr_status) & pd->pwr_sta_mask) ||
+	       !(read32(&mtk_spm->pwr_status_2nd) & pd->pwr_sta_mask))
 		continue;
 
-	clrbits_le32(pwr_con, PWR_CLK_DIS);
-	clrbits_le32(pwr_con, PWR_ISO);
-	setbits_le32(pwr_con, PWR_RST_B);
-	clrbits_le32(pwr_con, SRAM_PDN);
-}
+	clrbits_le32(pd->pwr_con, PWR_CLK_DIS);
+	clrbits_le32(pd->pwr_con, PWR_ISO);
+	setbits_le32(pd->pwr_con, PWR_RST_B);
+	clrbits_le32(pd->pwr_con, pd->sram_pdn_mask);
 
-void mtcmos_audio_power_on(void)
-{
-	mtcmos_power_on(&mt8173_spm->audio_pwr_con, AUD_PWR_STA_MASK);
-	while (read32(&mt8173_spm->audio_pwr_con) & AUD_SRAM_ACK)
+	while (read32(pd->pwr_con) & pd->sram_ack_mask)
 		continue;
 }
 
 void mtcmos_display_power_on(void)
 {
-	mtcmos_power_on(&mt8173_spm->dis_pwr_con, DIS_PWR_STA_MASK);
-	while (read32(&mt8173_spm->dis_pwr_con) & DIS_SRAM_ACK)
-		continue;
+	static const struct power_domain_data disp = {
+		.pwr_con = &mtk_spm->dis_pwr_con,
+		.pwr_sta_mask = DISP_PWR_STA_MASK,
+		.sram_pdn_mask = DISP_SRAM_PDN_MASK,
+		.sram_ack_mask = DISP_SRAM_ACK_MASK,
+	};
+
+	mtcmos_power_on(&disp);
+}
+
+void mtcmos_audio_power_on(void)
+{
+	static const struct power_domain_data audio = {
+		.pwr_con = &mtk_spm->audio_pwr_con,
+		.pwr_sta_mask = AUDIO_PWR_STA_MASK,
+		.sram_pdn_mask = AUDIO_SRAM_PDN_MASK,
+		.sram_ack_mask = AUDIO_SRAM_ACK_MASK,
+	};
+
+	mtcmos_power_on(&audio);
 }