// SPDX-License-Identifier: BSD-3-Clause
/*
 * Clock drivers for Qualcomm sm6125
 *
 * (C) Copyright 2026 Biswapriyo Nath <nathbappai@gmail.com>
 *
 */

#include <clk-uclass.h>
#include <dm.h>
#include <linux/delay.h>
#include <asm/io.h>
#include <linux/bitops.h>
#include <linux/bug.h>
#include <dt-bindings/clock/qcom,gcc-sm6125.h>

#include "clock-qcom.h"

#define GCC_BASE 0x01400000

#define QUPV3_WRAP0_S4_CMD_RCGR 0x1f608
#define SDCC1_APPS_CLK_CMD_RCGR 0x38028
#define SDCC2_APPS_CLK_CMD_RCGR 0x1e00c

#define GCC_GPLL0_MODE 0x0
#define GCC_GPLL3_MODE 0x3000
#define GCC_GPLL4_MODE 0x4000
#define GCC_GPLL5_MODE 0x5000
#define GCC_GPLL6_MODE 0x6000
#define GCC_GPLL7_MODE 0x7000
#define GCC_GPLL8_MODE 0x8000
#define GCC_GPLL9_MODE 0x9000

static const struct freq_tbl ftbl_gcc_qupv3_wrap0_s0_clk_src[] = {
	F(7372800, CFG_CLK_SRC_GPLL0_AUX2, 1, 384, 15625),
	F(14745600, CFG_CLK_SRC_GPLL0_AUX2, 1, 768, 15625),
	F(19200000, CFG_CLK_SRC_CXO, 1, 0, 0),
	F(29491200, CFG_CLK_SRC_GPLL0_AUX2, 1, 1536, 15625),
	F(32000000, CFG_CLK_SRC_GPLL0_AUX2, 1, 8, 75),
	F(48000000, CFG_CLK_SRC_GPLL0_AUX2, 1, 4, 25),
	F(64000000, CFG_CLK_SRC_GPLL0_AUX2, 1, 16, 75),
	F(75000000, CFG_CLK_SRC_GPLL0_AUX2, 4, 0, 0),
	F(80000000, CFG_CLK_SRC_GPLL0_AUX2, 1, 4, 15),
	F(96000000, CFG_CLK_SRC_GPLL0_AUX2, 1, 8, 25),
	F(100000000, CFG_CLK_SRC_GPLL0, 6, 0, 0),
	F(102400000, CFG_CLK_SRC_GPLL0_AUX2, 1, 128, 375),
	F(112000000, CFG_CLK_SRC_GPLL0_AUX2, 1, 28, 75),
	F(117964800, CFG_CLK_SRC_GPLL0_AUX2, 1, 6144, 15625),
	F(120000000, CFG_CLK_SRC_GPLL0_AUX2, 2.5, 0, 0),
	F(128000000, CFG_CLK_SRC_GPLL6, 3, 0, 0),
	{}
};

static const struct freq_tbl ftbl_gcc_sdcc2_apps_clk_src[] = {
	F(400000, CFG_CLK_SRC_CXO, 12, 1, 4),
	F(19200000, CFG_CLK_SRC_CXO, 1, 0, 0),
	F(25000000, CFG_CLK_SRC_GPLL0_AUX2, 12, 0, 0),
	F(50000000, CFG_CLK_SRC_GPLL0_AUX2, 6, 0, 0),
	F(100000000, CFG_CLK_SRC_GPLL0_AUX2, 3, 0, 0),
	{}
};

static const struct pll_vote_clk gpll0_clk = {
	.status = 0,
	.status_bit = BIT(31),
	.ena_vote = 0x79000,
	.vote_bit = BIT(0),
};

static const struct gate_clk sm6125_clks[] = {
	GATE_CLK(GCC_CFG_NOC_USB3_PRIM_AXI_CLK, 0x1a084, BIT(0)),
	GATE_CLK(GCC_QUPV3_WRAP0_CORE_2X_CLK, 0x7900c, BIT(9)),
	GATE_CLK(GCC_QUPV3_WRAP0_CORE_CLK, 0x7900c, BIT(8)),
	GATE_CLK(GCC_QUPV3_WRAP0_S0_CLK, 0x7900c, BIT(10)),
	GATE_CLK(GCC_QUPV3_WRAP0_S1_CLK, 0x7900c, BIT(11)),
	GATE_CLK(GCC_QUPV3_WRAP0_S2_CLK, 0x7900c, BIT(12)),
	GATE_CLK(GCC_QUPV3_WRAP0_S3_CLK, 0x7900c, BIT(13)),
	GATE_CLK(GCC_QUPV3_WRAP0_S4_CLK, 0x7900c, BIT(14)),
	GATE_CLK(GCC_QUPV3_WRAP0_S5_CLK, 0x7900c, BIT(15)),
	GATE_CLK(GCC_QUPV3_WRAP_0_M_AHB_CLK, 0x7900c, BIT(6)),
	GATE_CLK(GCC_QUPV3_WRAP_0_S_AHB_CLK, 0x7900c, BIT(7)),
	GATE_CLK(GCC_SDCC1_AHB_CLK, 0x38008, BIT(0)),
	GATE_CLK(GCC_SDCC1_APPS_CLK, 0x38004, BIT(0)),
	GATE_CLK(GCC_SDCC1_ICE_CORE_CLK, 0x3800c, BIT(0)),
	GATE_CLK(GCC_SDCC2_AHB_CLK, 0x1e008, BIT(0)),
	GATE_CLK(GCC_SDCC2_APPS_CLK, 0x1e004, BIT(0)),
	GATE_CLK(GCC_SYS_NOC_CPUSS_AHB_CLK, 0x79004, BIT(0)),
	GATE_CLK(GCC_SYS_NOC_UFS_PHY_AXI_CLK, 0x45098, BIT(0)),
	GATE_CLK(GCC_SYS_NOC_USB3_PRIM_AXI_CLK, 0x1a080, BIT(0)),
	GATE_CLK(GCC_UFS_PHY_AHB_CLK, 0x45014, BIT(0)),
	GATE_CLK(GCC_UFS_PHY_AXI_CLK, 0x45010, BIT(0)),
	GATE_CLK(GCC_UFS_PHY_ICE_CORE_CLK, 0x45044, BIT(0)),
	GATE_CLK(GCC_UFS_PHY_PHY_AUX_CLK, 0x45078, BIT(0)),
	GATE_CLK(GCC_UFS_PHY_RX_SYMBOL_0_CLK, 0x4501c, BIT(0)),
	GATE_CLK(GCC_UFS_PHY_TX_SYMBOL_0_CLK, 0x45018, BIT(0)),
	GATE_CLK(GCC_UFS_PHY_UNIPRO_CORE_CLK, 0x45040, BIT(0)),
	GATE_CLK(GCC_USB30_PRIM_MASTER_CLK, 0x1a010, BIT(0)),
	GATE_CLK(GCC_USB30_PRIM_MOCK_UTMI_CLK, 0x1a018, BIT(0)),
	GATE_CLK(GCC_USB30_PRIM_SLEEP_CLK, 0x1a014, BIT(0)),
	GATE_CLK(GCC_USB3_PRIM_CLKREF_CLK, 0x80278, BIT(0)),
	GATE_CLK(GCC_USB3_PRIM_PHY_COM_AUX_CLK, 0x1a054, BIT(0)),
	GATE_CLK(GCC_USB3_PRIM_PHY_PIPE_CLK, 0x1a058, BIT(0)),
	GATE_CLK(GCC_AHB2PHY_USB_CLK, 0x1d008, BIT(0)),
	GATE_CLK(GCC_UFS_MEM_CLKREF_CLK, 0x8c000, BIT(0)),
};

static ulong sm6125_set_rate(struct clk *clk, ulong rate)
{
	struct msm_clk_priv *priv = dev_get_priv(clk->dev);
	const struct freq_tbl *freq;

	debug("%s: clk %s rate %lu\n", __func__, sm6125_clks[clk->id].name,
	      rate);

	switch (clk->id) {
	case GCC_QUPV3_WRAP0_S4_CLK:
		freq = qcom_find_freq(ftbl_gcc_qupv3_wrap0_s0_clk_src, rate);
		clk_rcg_set_rate_mnd(priv->base, QUPV3_WRAP0_S4_CMD_RCGR,
				     freq->pre_div, freq->m, freq->n, freq->src,
				     16);
		return 0;
	case GCC_SDCC2_APPS_CLK:
		clk_enable_gpll0(priv->base, &gpll0_clk);
		freq = qcom_find_freq(ftbl_gcc_sdcc2_apps_clk_src, rate);
		WARN(freq->src != CFG_CLK_SRC_GPLL0,
		     "SDCC2_APPS_CLK_SRC not set to GPLL0, requested rate %lu\n",
		     rate);
		clk_rcg_set_rate_mnd(priv->base, SDCC2_APPS_CLK_CMD_RCGR,
				     freq->pre_div, freq->m, freq->n, freq->src,
				     8);
		return freq->freq;
	case GCC_SDCC1_APPS_CLK:
		/* The firmware turns this on for us and always sets it to this rate */
		return 384000000;
	default:
		return rate;
	}
}

static int sm6125_enable(struct clk *clk)
{
	struct msm_clk_priv *priv = dev_get_priv(clk->dev);

	if (priv->data->num_clks < clk->id) {
		debug("%s: unknown clk id %lu\n", __func__, clk->id);
		return 0;
	}

	debug("%s: clk %s\n", __func__, sm6125_clks[clk->id].name);

	switch (clk->id) {
	case GCC_USB30_PRIM_MASTER_CLK:
		qcom_gate_clk_en(priv, GCC_USB3_PRIM_PHY_COM_AUX_CLK);
		qcom_gate_clk_en(priv, GCC_USB3_PRIM_CLKREF_CLK);
		break;
	}

	return qcom_gate_clk_en(priv, clk->id);
}

static const struct qcom_reset_map sm6125_gcc_resets[] = {
	[GCC_QUSB2PHY_PRIM_BCR] = { 0x1c000 },
	[GCC_QUSB2PHY_SEC_BCR] = { 0x1c004 },
	[GCC_UFS_PHY_BCR] = { 0x45000 },
	[GCC_USB30_PRIM_BCR] = { 0x1a000 },
	[GCC_USB_PHY_CFG_AHB2PHY_BCR] = { 0x1d000 },
	[GCC_USB3PHY_PHY_PRIM_SP0_BCR] = { 0x1b008 },
	[GCC_USB3_PHY_PRIM_SP0_BCR] = { 0x1b000 },
	[GCC_CAMSS_MICRO_BCR] = { 0x560ac },
};

static const struct qcom_power_map sm6125_gdscs[] = {
	[USB30_PRIM_GDSC] = { 0x1a004 },
	[UFS_PHY_GDSC] = { 0x45004 },
	[CAMSS_VFE0_GDSC] = { 0x54004 },
	[CAMSS_VFE1_GDSC] = { 0x5403c },
	[CAMSS_TOP_GDSC] = { 0x5607c },
	[CAM_CPP_GDSC] = { 0x560bc },
	[HLOS1_VOTE_TURING_MMU_TBU1_GDSC] = { 0x7d060 },
	[HLOS1_VOTE_MM_SNOC_MMU_TBU_RT_GDSC] = { 0x80074 },
	[HLOS1_VOTE_MM_SNOC_MMU_TBU_NRT_GDSC] = { 0x80084 },
	[HLOS1_VOTE_TURING_MMU_TBU0_GDSC] = { 0x80094 },
};

static const phys_addr_t sm6125_gpll_addrs[] = {
	GCC_BASE + GCC_GPLL0_MODE, GCC_BASE + GCC_GPLL3_MODE,
	GCC_BASE + GCC_GPLL4_MODE, GCC_BASE + GCC_GPLL5_MODE,
	GCC_BASE + GCC_GPLL6_MODE, GCC_BASE + GCC_GPLL7_MODE,
	GCC_BASE + GCC_GPLL8_MODE, GCC_BASE + GCC_GPLL9_MODE,
};

static const phys_addr_t sm6125_rcg_addrs[] = {
	0x0141a01c, // GCC_USB30_PRIM_MASTER_CMD_RCGR
	0x0141a034, // GCC_USB30_PRIM_MOCK_UTMI_CMD_RCGR
	0x0141a060, // GCC_USB3_PRIM_PHY_AUX_CMD_RCGR
	0x01438028, // GCC_SDCC1_APPS_CMD_RCGR
	0x0141e00c, // GCC_SDCC2_APPS_CMD_RCGR
	0x0141f148, // GCC_QUPV3_WRAP0_S0_CMD_RCGR
	0x0141f278, // GCC_QUPV3_WRAP0_S1_CMD_RCGR
	0x0141f3a8, // GCC_QUPV3_WRAP0_S2_CMD_RCGR
	0x0141f4d8, // GCC_QUPV3_WRAP0_S3_CMD_RCGR
	0x0141f608, // GCC_QUPV3_WRAP0_S4_CMD_RCGR
	0x0141f738, // GCC_QUPV3_WRAP0_S5_CMD_RCGR
	0x01445020, // GCC_UFS_PHY_AXI_CMD_RCGR
	0x01445048, // GCC_UFS_PHY_ICE_CORE_CMD_RCGR
	0x01445060, // GCC_UFS_PHY_UNIPRO_CORE_CMD_RCGR
	0x0144507c, // GCC_UFS_PHY_PHY_AUX_CMD_RCGR
};

static const char *const sm6125_rcg_names[] = {
	"GCC_USB30_PRIM_MASTER_CMD_RCGR",
	"GCC_USB30_PRIM_MOCK_UTMI_CMD_RCGR",
	"GCC_USB3_PRIM_PHY_AUX_CMD_RCGR",
	"GCC_SDCC1_APPS_CMD_RCGR",
	"GCC_SDCC2_APPS_CMD_RCGR",
	"GCC_QUPV3_WRAP0_S0_CMD_RCGR",
	"GCC_QUPV3_WRAP0_S1_CMD_RCGR",
	"GCC_QUPV3_WRAP0_S2_CMD_RCGR",
	"GCC_QUPV3_WRAP0_S3_CMD_RCGR",
	"GCC_QUPV3_WRAP0_S4_CMD_RCGR",
	"GCC_QUPV3_WRAP0_S5_CMD_RCGR",
	"GCC_UFS_PHY_AXI_CMD_RCGR",
	"GCC_UFS_PHY_ICE_CORE_CMD_RCGR",
	"GCC_UFS_PHY_UNIPRO_CORE_CMD_RCGR",
	"GCC_UFS_PHY_PHY_AUX_CMD_RCGR",
};

static struct msm_clk_data sm6125_gcc_data = {
	.resets = sm6125_gcc_resets,
	.num_resets = ARRAY_SIZE(sm6125_gcc_resets),
	.clks = sm6125_clks,
	.num_clks = ARRAY_SIZE(sm6125_clks),
	.power_domains = sm6125_gdscs,
	.num_power_domains = ARRAY_SIZE(sm6125_gdscs),

	.enable = sm6125_enable,
	.set_rate = sm6125_set_rate,

	.dbg_pll_addrs = sm6125_gpll_addrs,
	.num_plls = ARRAY_SIZE(sm6125_gpll_addrs),
	.dbg_rcg_addrs = sm6125_rcg_addrs,
	.num_rcgs = ARRAY_SIZE(sm6125_rcg_addrs),
	.dbg_rcg_names = sm6125_rcg_names,
};

static const struct udevice_id gcc_sm6125_of_match[] = {
	{
		.compatible = "qcom,gcc-sm6125",
		.data = (ulong)&sm6125_gcc_data,
	},
	{}
};

U_BOOT_DRIVER(gcc_sm6125) = {
	.name = "gcc_sm6125",
	.id = UCLASS_NOP,
	.of_match = gcc_sm6125_of_match,
	.bind = qcom_cc_bind,
	.flags = DM_FLAG_PRE_RELOC,
};