// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (c) 2019 MediaTek Inc.
 */

#include <linux/clk.h>
#include <linux/pm_runtime.h>
#include "adsp_clk_audio.h"

enum adsp_clk {
	CLK_TOP_ADSP_SEL,
	CLK_TOP_CLK26M,
	CLK_TOP_ADSPPLL,
	CLK_TOP_AUD_LOCAL_BUS_SEL,
	CLK_TOP_MAINPLL_D5_D2,
	ADSP_CLK_NUM
};

struct adsp_clock_attr {
	const char *name;
	struct clk *clock;
};

static struct device *pm_dev;

static struct adsp_clock_attr adsp_clks[ADSP_CLK_NUM] = {
	[CLK_TOP_ADSP_SEL] = {"clk_top_adsp_sel", NULL},
	[CLK_TOP_CLK26M] = {"clk_top_clk26m", NULL},
	[CLK_TOP_ADSPPLL] = {"clk_top_adsppll", NULL},
	[CLK_TOP_AUD_LOCAL_BUS_SEL] = {"clk_top_aud_local_bus_sel", NULL},
	[CLK_TOP_MAINPLL_D5_D2] = {"clk_top_mainpll_d5_d2", NULL},
};

static int adsp_set_top_mux(enum adsp_clk clk)
{
	int ret = 0;

	if (clk >= ADSP_CLK_NUM || clk < 0)
		return -EINVAL;

	ret = clk_set_parent(adsp_clks[CLK_TOP_ADSP_SEL].clock,
		      adsp_clks[clk].clock);
	if (IS_ERR(&ret))
		pr_err("[ADSP] %s clk_set_parent(clk_top_adsp_sel-%x) fail %d\n",
		      __func__, clk, ret);

	return ret;
}

static int adsp_set_local_bus_mux(enum adsp_clk clk)
{
	int ret = 0;

	pr_debug("%s(%x)\n", __func__, clk);

	if (clk >= ADSP_CLK_NUM || clk < 0)
		return -EINVAL;

	ret = clk_set_parent(adsp_clks[CLK_TOP_AUD_LOCAL_BUS_SEL].clock,
		      adsp_clks[clk].clock);
	if (IS_ERR(&ret))
		pr_err("[ADSP] %s clk_set_parent(clk_top_audio_local_bus_sel-%x) fail %d\n",
		      __func__, clk, ret);

	return ret;
}

void adsp_mt_select_clock_mode(enum adsp_clk_mode mode)
{
	switch (mode) {
	case CLK_LOW_POWER:
		adsp_set_top_mux(CLK_TOP_CLK26M);
		break;
	case CLK_HIGH_PERFORM:
	case CLK_DEFAULT_INIT:
		adsp_set_top_mux(CLK_TOP_ADSPPLL);
		break;
	default:
		break;
	}
}

int adsp_mt_enable_clock(void)
{
	int ret = 0;

	ret = clk_prepare_enable(adsp_clks[CLK_TOP_ADSP_SEL].clock);
	if (IS_ERR(&ret)) {
		pr_err("%s(), clk_prepare_enable %s fail, ret %d\n",
			__func__, adsp_clks[CLK_TOP_ADSP_SEL].name, ret);
		return -EINVAL;
	}

	pm_runtime_get_sync(pm_dev);

	return 0;
}

void adsp_mt_disable_clock(void)
{
	pr_debug("%s()\n", __func__);

	pm_runtime_put_sync(pm_dev);
	clk_disable_unprepare(adsp_clks[CLK_TOP_ADSP_SEL].clock);
}


void adsp_mt6985_select_clock_mode(enum adsp_clk_mode mode)
{
	switch (mode) {
	case CLK_LOW_POWER:
		adsp_set_top_mux(CLK_TOP_CLK26M);
		adsp_set_local_bus_mux(CLK_TOP_CLK26M);
		break;
	case CLK_HIGH_PERFORM:
	case CLK_DEFAULT_INIT:
		adsp_set_top_mux(CLK_TOP_ADSPPLL);
		adsp_set_local_bus_mux(CLK_TOP_MAINPLL_D5_D2);
		break;
	default:
		break;
	}
}

int adsp_mt6985_enable_clock(void)
{
	int ret = 0;

	ret = clk_prepare_enable(adsp_clks[CLK_TOP_ADSP_SEL].clock);
	if (IS_ERR(&ret)) {
		pr_err("%s(), clk_prepare_enable %s fail, ret %d\n",
			__func__, adsp_clks[CLK_TOP_ADSP_SEL].name, ret);
		return -EINVAL;
	}

	ret = clk_prepare_enable(adsp_clks[CLK_TOP_AUD_LOCAL_BUS_SEL].clock);
	if (IS_ERR(&ret)) {
		pr_err("%s(), clk_prepare_enable %s fail, ret %d\n",
			__func__, adsp_clks[CLK_TOP_AUD_LOCAL_BUS_SEL].name, ret);
		return -EINVAL;
	}

	pm_runtime_get_sync(pm_dev);

	return 0;
}

void adsp_mt6985_disable_clock(void)
{
	pm_runtime_put_sync(pm_dev);
	clk_disable_unprepare(adsp_clks[CLK_TOP_ADSP_SEL].clock);
	clk_disable_unprepare(adsp_clks[CLK_TOP_AUD_LOCAL_BUS_SEL].clock);
}

/* clock init */
int adsp_clk_probe(struct platform_device *pdev,
			  struct adsp_clk_operations *ops)
{
	int ret = 0;
	size_t i;
	struct device *dev = &pdev->dev;

	for (i = 0; i < ARRAY_SIZE(adsp_clks); i++) {
		adsp_clks[i].clock = devm_clk_get(dev, adsp_clks[i].name);
		if (IS_ERR(adsp_clks[i].clock)) {
			ret = PTR_ERR(adsp_clks[i].clock);
			pr_err("%s devm_clk_get %s fail %d\n", __func__,
				   adsp_clks[i].name, ret);
		}
	}
	pm_dev = &pdev->dev;

	if (IS_ERR(adsp_clks[CLK_TOP_AUD_LOCAL_BUS_SEL].clock)) {
		ops->enable = adsp_mt_enable_clock;
		ops->disable = adsp_mt_disable_clock;
		ops->select = adsp_mt_select_clock_mode;
	} else {
		ops->enable = adsp_mt6985_enable_clock;
		ops->disable = adsp_mt6985_disable_clock;
		ops->select = adsp_mt6985_select_clock_mode;
	}

	return 0;
}

/* clock deinit */
void adsp_clk_remove(void *dev)
{
	pr_debug("%s\n", __func__);
}