// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2019 MediaTek Inc.
 * Author: Stanley Chu <stanley.chu@mediatek.com>
 */

#include <linux/clk.h>
#include <linux/delay.h>
#include <linux/io.h>
#include <linux/module.h>
#include <linux/of_platform.h>
#include <linux/phy/phy.h>
#include <linux/platform_device.h>
#include <linux/pm.h>
#include <linux/pm_domain.h>

#include <ufs-mediatek-sip.h>
#include <ufs-mediatek.h>

#define UFSPHY_CLKS_CNT    2

/* mphy register and offsets */
#define MP_GLB_DIG_8C               0x008C
#define FRC_PLL_ISO_EN              BIT(8)
#define PLL_ISO_EN                  BIT(9)
#define FRC_FRC_PWR_ON              BIT(10)
#define PLL_PWR_ON                  BIT(11)

#define MP_LN_DIG_RX_9C             0xA09C
#define FSM_DIFZ_FRC                BIT(18)

#define MP_LN_DIG_RX_AC             0xA0AC
#define FRC_RX_SQ_EN                BIT(0)
#define RX_SQ_EN                    BIT(1)

#define MP_LN_RX_44                 0xB044
#define FRC_CDR_PWR_ON              BIT(17)
#define CDR_PWR_ON                  BIT(18)
#define FRC_CDR_ISO_EN              BIT(19)
#define CDR_ISO_EN                  BIT(20)

struct ufs_mtk_phy {
	struct device *dev;
	void __iomem *mmio;
	u32 ver;
	struct clk_bulk_data clks[UFSPHY_CLKS_CNT];
};

static inline u32 mphy_readl(struct ufs_mtk_phy *phy, u32 reg)
{
	return readl(phy->mmio + reg);
}

static inline void mphy_writel(struct ufs_mtk_phy *phy, u32 val, u32 reg)
{
	writel(val, phy->mmio + reg);
}

static void mphy_set_bit(struct ufs_mtk_phy *phy, u32 reg, u32 bit)
{
	u32 val;

	val = mphy_readl(phy, reg);
	val |= bit;
	mphy_writel(phy, val, reg);
}

static void mphy_clr_bit(struct ufs_mtk_phy *phy, u32 reg, u32 bit)
{
	u32 val;

	val = mphy_readl(phy, reg);
	val &= ~bit;
	mphy_writel(phy, val, reg);
}

static struct ufs_mtk_phy *get_ufs_mtk_phy(struct phy *generic_phy)
{
	return (struct ufs_mtk_phy *)phy_get_drvdata(generic_phy);
}

static inline bool ufs_mtk_phy_pm_allowed(struct ufs_mtk_phy *phy)
{
	return (phy->ver > 0);
}

static int ufs_mtk_phy_clk_init(struct ufs_mtk_phy *phy)
{
	struct device *dev = phy->dev;
	struct clk_bulk_data *clks = phy->clks;

	clks[0].id = "unipro";
	clks[1].id = "mp";
	return devm_clk_bulk_get(dev, UFSPHY_CLKS_CNT, clks);
}

static void ufs_mtk_phy_set_active(struct ufs_mtk_phy *phy)
{
	/* release DA_MP_PLL_PWR_ON */
	mphy_set_bit(phy, MP_GLB_DIG_8C, PLL_PWR_ON);
	mphy_clr_bit(phy, MP_GLB_DIG_8C, FRC_FRC_PWR_ON);

	/* release DA_MP_PLL_ISO_EN */
	mphy_clr_bit(phy, MP_GLB_DIG_8C, PLL_ISO_EN);
	mphy_clr_bit(phy, MP_GLB_DIG_8C, FRC_PLL_ISO_EN);

	/* release DA_MP_CDR_PWR_ON */
	mphy_set_bit(phy, MP_LN_RX_44, CDR_PWR_ON);
	mphy_clr_bit(phy, MP_LN_RX_44, FRC_CDR_PWR_ON);

	/* release DA_MP_CDR_ISO_EN */
	mphy_clr_bit(phy, MP_LN_RX_44, CDR_ISO_EN);
	mphy_clr_bit(phy, MP_LN_RX_44, FRC_CDR_ISO_EN);

	/* release DA_MP_RX0_SQ_EN */
	mphy_set_bit(phy, MP_LN_DIG_RX_AC, RX_SQ_EN);
	mphy_clr_bit(phy, MP_LN_DIG_RX_AC, FRC_RX_SQ_EN);

	/* delay 1us to wait DIFZ stable */
	udelay(1);

	/* release DIFZ */
	mphy_clr_bit(phy, MP_LN_DIG_RX_9C, FSM_DIFZ_FRC);
}

static void ufs_mtk_phy_set_deep_hibern(struct ufs_mtk_phy *phy)
{
	/* force DIFZ */
	mphy_set_bit(phy, MP_LN_DIG_RX_9C, FSM_DIFZ_FRC);

	/* force DA_MP_RX0_SQ_EN */
	mphy_set_bit(phy, MP_LN_DIG_RX_AC, FRC_RX_SQ_EN);
	mphy_clr_bit(phy, MP_LN_DIG_RX_AC, RX_SQ_EN);

	/* force DA_MP_CDR_ISO_EN */
	mphy_set_bit(phy, MP_LN_RX_44, FRC_CDR_ISO_EN);
	mphy_set_bit(phy, MP_LN_RX_44, CDR_ISO_EN);

	/* force DA_MP_CDR_PWR_ON */
	mphy_set_bit(phy, MP_LN_RX_44, FRC_CDR_PWR_ON);
	mphy_clr_bit(phy, MP_LN_RX_44, CDR_PWR_ON);

	/* force DA_MP_PLL_ISO_EN */
	mphy_set_bit(phy, MP_GLB_DIG_8C, FRC_PLL_ISO_EN);
	mphy_set_bit(phy, MP_GLB_DIG_8C, PLL_ISO_EN);

	/* force DA_MP_PLL_PWR_ON */
	mphy_set_bit(phy, MP_GLB_DIG_8C, FRC_FRC_PWR_ON);
	mphy_clr_bit(phy, MP_GLB_DIG_8C, PLL_PWR_ON);
}

static int ufs_mtk_phy_power_on(struct phy *generic_phy)
{
	struct ufs_mtk_phy *phy = get_ufs_mtk_phy(generic_phy);
	int ret;

	if (phy->ver)
		return 0;

	ret = clk_bulk_prepare_enable(UFSPHY_CLKS_CNT, phy->clks);
	if (ret)
		return ret;

	ufs_mtk_phy_set_active(phy);
	return 0;
}

static int ufs_mtk_phy_power_off(struct phy *generic_phy)
{

	struct ufs_mtk_phy *phy = get_ufs_mtk_phy(generic_phy);

	if (phy->ver)
		return 0;

	ufs_mtk_phy_set_deep_hibern(phy);
	clk_bulk_disable_unprepare(UFSPHY_CLKS_CNT, phy->clks);
	return 0;
}

static const struct phy_ops ufs_mtk_phy_ops = {
	.power_on       = ufs_mtk_phy_power_on,
	.power_off      = ufs_mtk_phy_power_off,
	.owner          = THIS_MODULE,
};

static int ufs_mtk_phy_runtime_suspend(struct device *dev)
{
	struct arm_smccc_res res;
	struct ufs_mtk_phy *phy = dev_get_drvdata(dev);

	if (!ufs_mtk_phy_pm_allowed(phy))
		goto out;

	ufs_mtk_host_pwr_ctrl(HOST_PWR_MPHY, false, res);

out:
	return 0;
}

static int ufs_mtk_phy_runtime_resume(struct device *dev)
{
	struct arm_smccc_res res;
	struct ufs_mtk_phy *phy = dev_get_drvdata(dev);

	if (!ufs_mtk_phy_pm_allowed(phy))
		goto out;

	ufs_mtk_host_pwr_ctrl(HOST_PWR_MPHY, true, res);

out:
	return 0;
}

static int ufs_mtk_phy_system_suspend(struct device *dev)
{
	struct arm_smccc_res res;
	struct ufs_mtk_phy *phy = dev_get_drvdata(dev);

	if (pm_runtime_suspended(dev))
		goto out;

	if (!ufs_mtk_phy_pm_allowed(phy))
		goto out;

	ufs_mtk_host_pwr_ctrl(HOST_PWR_MPHY, false, res);

out:
	return 0;
}

static int ufs_mtk_phy_system_resume(struct device *dev)
{
	struct arm_smccc_res res;
	struct ufs_mtk_phy *phy = dev_get_drvdata(dev);

	if (!ufs_mtk_phy_pm_allowed(phy))
		goto out;

	ufs_mtk_host_pwr_ctrl(HOST_PWR_MPHY, true, res);

out:
	return 0;
}

static int ufs_mtk_phy_init(struct ufs_mtk_phy *phy)
{
	struct device *dev = phy->dev;
	u32 val = 0;
	int ret;

#if IS_ENABLED(CONFIG_UFS_MEDIATEK_INTERNAL)
	struct tag_chipid *chipid;
	/* Get chip id from bootmode */
	chipid = (struct tag_chipid *)ufs_mtk_get_boot_property(dev->of_node,
								"atag,chipid", NULL);

	ret = of_property_read_u32(dev->of_node, "mediatek,pm-forbidden-on-hwver", &val);
	if (!ret && chipid) {
		if (chipid->hw_ver == val) {
			pm_runtime_forbid(dev);
			dev_info(dev, "pm forbidden");
		}
	}
#endif

	ret = of_property_read_u32(dev->of_node, "mphy-ver", &val);
	if (!ret)
		phy->ver = val;

	if (!phy->ver)
		ufs_mtk_phy_clk_init(phy);

	return 0;
}


static int ufs_mtk_phy_probe(struct platform_device *pdev)
{
	struct device *dev = &pdev->dev;
	struct phy *generic_phy;
	struct phy_provider *phy_provider;
	struct ufs_mtk_phy *phy;
	int ret = 0;

	phy = devm_kzalloc(dev, sizeof(*phy), GFP_KERNEL);
	if (!phy)
		return -ENOMEM;

	phy->mmio = devm_platform_ioremap_resource(pdev, 0);
	if (IS_ERR(phy->mmio))
		return PTR_ERR(phy->mmio);

	phy->dev = dev;

	ret = ufs_mtk_phy_init(phy);
	if (ret)
		return ret;

	generic_phy = devm_phy_create(dev, NULL, &ufs_mtk_phy_ops);
	if (IS_ERR(generic_phy))
		return PTR_ERR(generic_phy);

	phy_set_drvdata(generic_phy, phy);
	dev_set_drvdata(dev, phy);

	phy_provider = devm_of_phy_provider_register(dev, of_phy_simple_xlate);
	if (IS_ERR(phy_provider))
		return PTR_ERR(phy_provider);

	return ret;
}

static const struct of_device_id ufs_mtk_phy_of_match[] = {
	{.compatible = "mediatek,mt8183-ufsphy"},
	{},
};
MODULE_DEVICE_TABLE(of, ufs_mtk_phy_of_match);

static const struct dev_pm_ops ufs_mtk_phy_pm_ops = {
	SET_SYSTEM_SLEEP_PM_OPS(ufs_mtk_phy_system_suspend, ufs_mtk_phy_system_resume)
	SET_RUNTIME_PM_OPS(ufs_mtk_phy_runtime_suspend, ufs_mtk_phy_runtime_resume, NULL)
};

static struct platform_driver ufs_mtk_phy_driver = {
	.probe = ufs_mtk_phy_probe,
	.driver = {
		.of_match_table = ufs_mtk_phy_of_match,
		.pm     = &ufs_mtk_phy_pm_ops,
		.name = "ufs_mtk_phy",
	},
};
module_platform_driver(ufs_mtk_phy_driver);

MODULE_DESCRIPTION("Universal Flash Storage (UFS) MediaTek MPHY");
MODULE_AUTHOR("Stanley Chu <stanley.chu@mediatek.com>");
MODULE_LICENSE("GPL v2");