// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (c) 2016 MediaTek Inc.
 * Author: Tiffany Lin <tiffany.lin@mediatek.com>
 */

#include <linux/clk.h>
#include <linux/interrupt.h>
#include <linux/of_address.h>
#include <linux/of_platform.h>
#include <linux/pm_runtime.h>

#include "mtk_vcodec_dec_pm.h"
#include "mtk_vcodec_util.h"

int mtk_vcodec_init_dec_pm(struct mtk_vcodec_dev *mtkdev)
{
	struct platform_device *pdev;
	struct mtk_vcodec_pm *pm;
	struct mtk_vcodec_clk *dec_clk;
	struct mtk_vcodec_clk_info *clk_info;
	int i = 0, ret = 0;

	pdev = mtkdev->plat_dev;
	pm = &mtkdev->pm;
	pm->mtkdev = mtkdev;
	dec_clk = &pm->vdec_clk;
	pm->dev = &pdev->dev;

	dec_clk->clk_num =
		of_property_count_strings(pdev->dev.of_node, "clock-names");
	if (dec_clk->clk_num > 0) {
		dec_clk->clk_info = devm_kcalloc(&pdev->dev,
			dec_clk->clk_num, sizeof(*clk_info),
			GFP_KERNEL);
		if (!dec_clk->clk_info)
			return -ENOMEM;
	} else {
		mtk_v4l2_err("Failed to get vdec clock count");
		return -EINVAL;
	}

	for (i = 0; i < dec_clk->clk_num; i++) {
		clk_info = &dec_clk->clk_info[i];
		ret = of_property_read_string_index(pdev->dev.of_node,
			"clock-names", i, &clk_info->clk_name);
		if (ret) {
			mtk_v4l2_err("Failed to get clock name id = %d", i);
			return ret;
		}
		clk_info->vcodec_clk = devm_clk_get(&pdev->dev,
			clk_info->clk_name);
		if (IS_ERR(clk_info->vcodec_clk)) {
			mtk_v4l2_err("devm_clk_get (%d)%s fail", i,
				clk_info->clk_name);
			return PTR_ERR(clk_info->vcodec_clk);
		}
	}

	pm_runtime_enable(&pdev->dev);

	return ret;
}

void mtk_vcodec_release_dec_pm(struct mtk_vcodec_dev *dev)
{
	pm_runtime_disable(dev->pm.dev);
}

static void mtk_vcodec_dec_pw_on(struct mtk_vcodec_dev *vdec_dev,
	int comp_idx)
{
	struct mtk_vcodec_dev *comp_dev;
	struct mtk_vcodec_pm *pm;
	int ret;

	comp_dev = mtk_vcodec_get_hw_dev(vdec_dev, comp_idx);
	if (!comp_dev) {
		mtk_v4l2_err("Failed to get hw dev\n");
		return;
	}
	pm = &comp_dev->pm;
	ret = pm_runtime_resume_and_get(pm->dev);
	if (ret < 0)
		mtk_v4l2_err("pm_runtime_resume_and_get fail %d", ret);
}

static void mtk_vcodec_dec_pw_off(struct mtk_vcodec_dev *vdec_dev,
	int comp_idx)
{
	struct mtk_vcodec_dev *comp_dev;
	struct mtk_vcodec_pm *pm;

	comp_dev = mtk_vcodec_get_hw_dev(vdec_dev, comp_idx);
	if (!comp_dev) {
		mtk_v4l2_err("Failed to get hw dev\n");
		return;
	}
	pm = &comp_dev->pm;
	pm_runtime_put_sync(pm->dev);
}

static void mtk_vcodec_dec_clock_on(struct mtk_vcodec_dev *vdec_dev,
	int comp_idx)
{
	struct mtk_vcodec_dev *comp_dev;
	struct mtk_vcodec_pm *pm;
	struct mtk_vcodec_clk *dec_clk;
	int ret, i;

	comp_dev = mtk_vcodec_get_hw_dev(vdec_dev, comp_idx);
	if (!comp_dev) {
		mtk_v4l2_err("Failed to get hw dev\n");
		return;
	}
	pm = &comp_dev->pm;
	dec_clk = &pm->vdec_clk;

	for (i = 0; i < dec_clk->clk_num; i++) {
		ret = clk_prepare_enable(dec_clk->clk_info[i].vcodec_clk);
		if (ret) {
			mtk_v4l2_err("clk_prepare_enable %d %s fail %d", i,
				dec_clk->clk_info[i].clk_name, ret);
			goto error;
		}
	}
	return;

error:
	for (i -= 1; i >= 0; i--)
		clk_disable_unprepare(dec_clk->clk_info[i].vcodec_clk);
}

static void mtk_vcodec_dec_clock_off(struct mtk_vcodec_dev *vdec_dev,
	int comp_idx)
{
	struct mtk_vcodec_dev *comp_dev;
	struct mtk_vcodec_pm *pm;
	struct mtk_vcodec_clk *dec_clk;
	int i;

	comp_dev = mtk_vcodec_get_hw_dev(vdec_dev, comp_idx);
	if (!comp_dev) {
		mtk_v4l2_err("Failed to get hw dev\n");
		return;
	}
	pm = &comp_dev->pm;
	dec_clk = &pm->vdec_clk;

	for (i = dec_clk->clk_num - 1; i >= 0; i--)
		clk_disable_unprepare(dec_clk->clk_info[i].vcodec_clk);
}

static void mtk_vcodec_dec_enable_irq(struct mtk_vcodec_dev *vdec_dev,
	int comp_idx)
{
	struct mtk_vcodec_dev *comp_dev;

	comp_dev = mtk_vcodec_get_hw_dev(vdec_dev, comp_idx);
	if (!comp_dev) {
		mtk_v4l2_err("Failed to get hw dev\n");
		return;
	}

	enable_irq(comp_dev->dec_irq);
}

static void mtk_vcodec_dec_disable_irq(struct mtk_vcodec_dev *vdec_dev,
	int comp_idx)
{
	struct mtk_vcodec_dev *comp_dev;

	comp_dev = mtk_vcodec_get_hw_dev(vdec_dev, comp_idx);
	if (!comp_dev) {
		mtk_v4l2_err("Failed to get hw dev\n");
		return;
	}

	disable_irq(comp_dev->dec_irq);
}

void mtk_vcodec_dec_enable_hardware(struct mtk_vcodec_ctx *ctx,
	int comp_idx)
{
	mutex_lock(&ctx->dev->dec_mutex[comp_idx]);
	if (VDEC_LAT_ARCH(ctx->dev->vdec_pdata->hw_arch) &&
		comp_idx == MTK_VDEC_CORE) {
		mtk_vcodec_dec_pw_on(ctx->dev, MTK_VDEC_LAT0);
		mtk_vcodec_dec_clock_on(ctx->dev, MTK_VDEC_LAT0);
	}
	mtk_vcodec_dec_pw_on(ctx->dev, comp_idx);
	mtk_vcodec_dec_clock_on(ctx->dev, comp_idx);
	mtk_vcodec_dec_enable_irq(ctx->dev, comp_idx);
}

void mtk_vcodec_dec_disable_hardware(struct mtk_vcodec_ctx *ctx,
	int comp_idx)
{
	mtk_vcodec_dec_disable_irq(ctx->dev, comp_idx);
	mtk_vcodec_dec_clock_off(ctx->dev, comp_idx);
	mtk_vcodec_dec_pw_off(ctx->dev, comp_idx);
	if (VDEC_LAT_ARCH(ctx->dev->vdec_pdata->hw_arch) &&
		comp_idx == MTK_VDEC_CORE) {
		mtk_vcodec_dec_clock_off(ctx->dev, MTK_VDEC_LAT0);
		mtk_vcodec_dec_pw_off(ctx->dev, MTK_VDEC_LAT0);
	}
	mutex_unlock(&ctx->dev->dec_mutex[comp_idx]);
}