18c2ecf20Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
28c2ecf20Sopenharmony_ci/*
38c2ecf20Sopenharmony_ci * Copyright 2018 NXP
48c2ecf20Sopenharmony_ci */
58c2ecf20Sopenharmony_ci
68c2ecf20Sopenharmony_ci#include <linux/clk-provider.h>
78c2ecf20Sopenharmony_ci#include <linux/errno.h>
88c2ecf20Sopenharmony_ci#include <linux/export.h>
98c2ecf20Sopenharmony_ci#include <linux/io.h>
108c2ecf20Sopenharmony_ci#include <linux/slab.h>
118c2ecf20Sopenharmony_ci
128c2ecf20Sopenharmony_ci#include "clk.h"
138c2ecf20Sopenharmony_ci
148c2ecf20Sopenharmony_ci#define PCG_PREDIV_SHIFT	16
158c2ecf20Sopenharmony_ci#define PCG_PREDIV_WIDTH	3
168c2ecf20Sopenharmony_ci#define PCG_PREDIV_MAX		8
178c2ecf20Sopenharmony_ci
188c2ecf20Sopenharmony_ci#define PCG_DIV_SHIFT		0
198c2ecf20Sopenharmony_ci#define PCG_CORE_DIV_WIDTH	3
208c2ecf20Sopenharmony_ci#define PCG_DIV_WIDTH		6
218c2ecf20Sopenharmony_ci#define PCG_DIV_MAX		64
228c2ecf20Sopenharmony_ci
238c2ecf20Sopenharmony_ci#define PCG_PCS_SHIFT		24
248c2ecf20Sopenharmony_ci#define PCG_PCS_MASK		0x7
258c2ecf20Sopenharmony_ci
268c2ecf20Sopenharmony_ci#define PCG_CGC_SHIFT		28
278c2ecf20Sopenharmony_ci
288c2ecf20Sopenharmony_cistatic unsigned long imx8m_clk_composite_divider_recalc_rate(struct clk_hw *hw,
298c2ecf20Sopenharmony_ci						unsigned long parent_rate)
308c2ecf20Sopenharmony_ci{
318c2ecf20Sopenharmony_ci	struct clk_divider *divider = to_clk_divider(hw);
328c2ecf20Sopenharmony_ci	unsigned long prediv_rate;
338c2ecf20Sopenharmony_ci	unsigned int prediv_value;
348c2ecf20Sopenharmony_ci	unsigned int div_value;
358c2ecf20Sopenharmony_ci
368c2ecf20Sopenharmony_ci	prediv_value = readl(divider->reg) >> divider->shift;
378c2ecf20Sopenharmony_ci	prediv_value &= clk_div_mask(divider->width);
388c2ecf20Sopenharmony_ci
398c2ecf20Sopenharmony_ci	prediv_rate = divider_recalc_rate(hw, parent_rate, prediv_value,
408c2ecf20Sopenharmony_ci						NULL, divider->flags,
418c2ecf20Sopenharmony_ci						divider->width);
428c2ecf20Sopenharmony_ci
438c2ecf20Sopenharmony_ci	div_value = readl(divider->reg) >> PCG_DIV_SHIFT;
448c2ecf20Sopenharmony_ci	div_value &= clk_div_mask(PCG_DIV_WIDTH);
458c2ecf20Sopenharmony_ci
468c2ecf20Sopenharmony_ci	return divider_recalc_rate(hw, prediv_rate, div_value, NULL,
478c2ecf20Sopenharmony_ci				   divider->flags, PCG_DIV_WIDTH);
488c2ecf20Sopenharmony_ci}
498c2ecf20Sopenharmony_ci
508c2ecf20Sopenharmony_cistatic int imx8m_clk_composite_compute_dividers(unsigned long rate,
518c2ecf20Sopenharmony_ci						unsigned long parent_rate,
528c2ecf20Sopenharmony_ci						int *prediv, int *postdiv)
538c2ecf20Sopenharmony_ci{
548c2ecf20Sopenharmony_ci	int div1, div2;
558c2ecf20Sopenharmony_ci	int error = INT_MAX;
568c2ecf20Sopenharmony_ci	int ret = -EINVAL;
578c2ecf20Sopenharmony_ci
588c2ecf20Sopenharmony_ci	*prediv = 1;
598c2ecf20Sopenharmony_ci	*postdiv = 1;
608c2ecf20Sopenharmony_ci
618c2ecf20Sopenharmony_ci	for (div1 = 1; div1 <= PCG_PREDIV_MAX; div1++) {
628c2ecf20Sopenharmony_ci		for (div2 = 1; div2 <= PCG_DIV_MAX; div2++) {
638c2ecf20Sopenharmony_ci			int new_error = ((parent_rate / div1) / div2) - rate;
648c2ecf20Sopenharmony_ci
658c2ecf20Sopenharmony_ci			if (abs(new_error) < abs(error)) {
668c2ecf20Sopenharmony_ci				*prediv = div1;
678c2ecf20Sopenharmony_ci				*postdiv = div2;
688c2ecf20Sopenharmony_ci				error = new_error;
698c2ecf20Sopenharmony_ci				ret = 0;
708c2ecf20Sopenharmony_ci			}
718c2ecf20Sopenharmony_ci		}
728c2ecf20Sopenharmony_ci	}
738c2ecf20Sopenharmony_ci	return ret;
748c2ecf20Sopenharmony_ci}
758c2ecf20Sopenharmony_ci
768c2ecf20Sopenharmony_cistatic long imx8m_clk_composite_divider_round_rate(struct clk_hw *hw,
778c2ecf20Sopenharmony_ci						unsigned long rate,
788c2ecf20Sopenharmony_ci						unsigned long *prate)
798c2ecf20Sopenharmony_ci{
808c2ecf20Sopenharmony_ci	int prediv_value;
818c2ecf20Sopenharmony_ci	int div_value;
828c2ecf20Sopenharmony_ci
838c2ecf20Sopenharmony_ci	imx8m_clk_composite_compute_dividers(rate, *prate,
848c2ecf20Sopenharmony_ci						&prediv_value, &div_value);
858c2ecf20Sopenharmony_ci	rate = DIV_ROUND_UP(*prate, prediv_value);
868c2ecf20Sopenharmony_ci
878c2ecf20Sopenharmony_ci	return DIV_ROUND_UP(rate, div_value);
888c2ecf20Sopenharmony_ci
898c2ecf20Sopenharmony_ci}
908c2ecf20Sopenharmony_ci
918c2ecf20Sopenharmony_cistatic int imx8m_clk_composite_divider_set_rate(struct clk_hw *hw,
928c2ecf20Sopenharmony_ci					unsigned long rate,
938c2ecf20Sopenharmony_ci					unsigned long parent_rate)
948c2ecf20Sopenharmony_ci{
958c2ecf20Sopenharmony_ci	struct clk_divider *divider = to_clk_divider(hw);
968c2ecf20Sopenharmony_ci	unsigned long flags;
978c2ecf20Sopenharmony_ci	int prediv_value;
988c2ecf20Sopenharmony_ci	int div_value;
998c2ecf20Sopenharmony_ci	int ret;
1008c2ecf20Sopenharmony_ci	u32 orig, val;
1018c2ecf20Sopenharmony_ci
1028c2ecf20Sopenharmony_ci	ret = imx8m_clk_composite_compute_dividers(rate, parent_rate,
1038c2ecf20Sopenharmony_ci						&prediv_value, &div_value);
1048c2ecf20Sopenharmony_ci	if (ret)
1058c2ecf20Sopenharmony_ci		return -EINVAL;
1068c2ecf20Sopenharmony_ci
1078c2ecf20Sopenharmony_ci	spin_lock_irqsave(divider->lock, flags);
1088c2ecf20Sopenharmony_ci
1098c2ecf20Sopenharmony_ci	orig = readl(divider->reg);
1108c2ecf20Sopenharmony_ci	val = orig & ~((clk_div_mask(divider->width) << divider->shift) |
1118c2ecf20Sopenharmony_ci		       (clk_div_mask(PCG_DIV_WIDTH) << PCG_DIV_SHIFT));
1128c2ecf20Sopenharmony_ci
1138c2ecf20Sopenharmony_ci	val |= (u32)(prediv_value  - 1) << divider->shift;
1148c2ecf20Sopenharmony_ci	val |= (u32)(div_value - 1) << PCG_DIV_SHIFT;
1158c2ecf20Sopenharmony_ci
1168c2ecf20Sopenharmony_ci	if (val != orig)
1178c2ecf20Sopenharmony_ci		writel(val, divider->reg);
1188c2ecf20Sopenharmony_ci
1198c2ecf20Sopenharmony_ci	spin_unlock_irqrestore(divider->lock, flags);
1208c2ecf20Sopenharmony_ci
1218c2ecf20Sopenharmony_ci	return ret;
1228c2ecf20Sopenharmony_ci}
1238c2ecf20Sopenharmony_ci
1248c2ecf20Sopenharmony_cistatic const struct clk_ops imx8m_clk_composite_divider_ops = {
1258c2ecf20Sopenharmony_ci	.recalc_rate = imx8m_clk_composite_divider_recalc_rate,
1268c2ecf20Sopenharmony_ci	.round_rate = imx8m_clk_composite_divider_round_rate,
1278c2ecf20Sopenharmony_ci	.set_rate = imx8m_clk_composite_divider_set_rate,
1288c2ecf20Sopenharmony_ci};
1298c2ecf20Sopenharmony_ci
1308c2ecf20Sopenharmony_cistatic u8 imx8m_clk_composite_mux_get_parent(struct clk_hw *hw)
1318c2ecf20Sopenharmony_ci{
1328c2ecf20Sopenharmony_ci	return clk_mux_ops.get_parent(hw);
1338c2ecf20Sopenharmony_ci}
1348c2ecf20Sopenharmony_ci
1358c2ecf20Sopenharmony_cistatic int imx8m_clk_composite_mux_set_parent(struct clk_hw *hw, u8 index)
1368c2ecf20Sopenharmony_ci{
1378c2ecf20Sopenharmony_ci	struct clk_mux *mux = to_clk_mux(hw);
1388c2ecf20Sopenharmony_ci	u32 val = clk_mux_index_to_val(mux->table, mux->flags, index);
1398c2ecf20Sopenharmony_ci	unsigned long flags = 0;
1408c2ecf20Sopenharmony_ci	u32 reg;
1418c2ecf20Sopenharmony_ci
1428c2ecf20Sopenharmony_ci	if (mux->lock)
1438c2ecf20Sopenharmony_ci		spin_lock_irqsave(mux->lock, flags);
1448c2ecf20Sopenharmony_ci
1458c2ecf20Sopenharmony_ci	reg = readl(mux->reg);
1468c2ecf20Sopenharmony_ci	reg &= ~(mux->mask << mux->shift);
1478c2ecf20Sopenharmony_ci	val = val << mux->shift;
1488c2ecf20Sopenharmony_ci	reg |= val;
1498c2ecf20Sopenharmony_ci	/*
1508c2ecf20Sopenharmony_ci	 * write twice to make sure non-target interface
1518c2ecf20Sopenharmony_ci	 * SEL_A/B point the same clk input.
1528c2ecf20Sopenharmony_ci	 */
1538c2ecf20Sopenharmony_ci	writel(reg, mux->reg);
1548c2ecf20Sopenharmony_ci	writel(reg, mux->reg);
1558c2ecf20Sopenharmony_ci
1568c2ecf20Sopenharmony_ci	if (mux->lock)
1578c2ecf20Sopenharmony_ci		spin_unlock_irqrestore(mux->lock, flags);
1588c2ecf20Sopenharmony_ci
1598c2ecf20Sopenharmony_ci	return 0;
1608c2ecf20Sopenharmony_ci}
1618c2ecf20Sopenharmony_ci
1628c2ecf20Sopenharmony_cistatic int
1638c2ecf20Sopenharmony_ciimx8m_clk_composite_mux_determine_rate(struct clk_hw *hw,
1648c2ecf20Sopenharmony_ci				       struct clk_rate_request *req)
1658c2ecf20Sopenharmony_ci{
1668c2ecf20Sopenharmony_ci	return clk_mux_ops.determine_rate(hw, req);
1678c2ecf20Sopenharmony_ci}
1688c2ecf20Sopenharmony_ci
1698c2ecf20Sopenharmony_ci
1708c2ecf20Sopenharmony_cistatic const struct clk_ops imx8m_clk_composite_mux_ops = {
1718c2ecf20Sopenharmony_ci	.get_parent = imx8m_clk_composite_mux_get_parent,
1728c2ecf20Sopenharmony_ci	.set_parent = imx8m_clk_composite_mux_set_parent,
1738c2ecf20Sopenharmony_ci	.determine_rate = imx8m_clk_composite_mux_determine_rate,
1748c2ecf20Sopenharmony_ci};
1758c2ecf20Sopenharmony_ci
1768c2ecf20Sopenharmony_cistruct clk_hw *imx8m_clk_hw_composite_flags(const char *name,
1778c2ecf20Sopenharmony_ci					const char * const *parent_names,
1788c2ecf20Sopenharmony_ci					int num_parents, void __iomem *reg,
1798c2ecf20Sopenharmony_ci					u32 composite_flags,
1808c2ecf20Sopenharmony_ci					unsigned long flags)
1818c2ecf20Sopenharmony_ci{
1828c2ecf20Sopenharmony_ci	struct clk_hw *hw = ERR_PTR(-ENOMEM), *mux_hw;
1838c2ecf20Sopenharmony_ci	struct clk_hw *div_hw, *gate_hw;
1848c2ecf20Sopenharmony_ci	struct clk_divider *div = NULL;
1858c2ecf20Sopenharmony_ci	struct clk_gate *gate = NULL;
1868c2ecf20Sopenharmony_ci	struct clk_mux *mux = NULL;
1878c2ecf20Sopenharmony_ci	const struct clk_ops *divider_ops;
1888c2ecf20Sopenharmony_ci	const struct clk_ops *mux_ops;
1898c2ecf20Sopenharmony_ci
1908c2ecf20Sopenharmony_ci	mux = kzalloc(sizeof(*mux), GFP_KERNEL);
1918c2ecf20Sopenharmony_ci	if (!mux)
1928c2ecf20Sopenharmony_ci		goto fail;
1938c2ecf20Sopenharmony_ci
1948c2ecf20Sopenharmony_ci	mux_hw = &mux->hw;
1958c2ecf20Sopenharmony_ci	mux->reg = reg;
1968c2ecf20Sopenharmony_ci	mux->shift = PCG_PCS_SHIFT;
1978c2ecf20Sopenharmony_ci	mux->mask = PCG_PCS_MASK;
1988c2ecf20Sopenharmony_ci	mux->lock = &imx_ccm_lock;
1998c2ecf20Sopenharmony_ci
2008c2ecf20Sopenharmony_ci	div = kzalloc(sizeof(*div), GFP_KERNEL);
2018c2ecf20Sopenharmony_ci	if (!div)
2028c2ecf20Sopenharmony_ci		goto fail;
2038c2ecf20Sopenharmony_ci
2048c2ecf20Sopenharmony_ci	div_hw = &div->hw;
2058c2ecf20Sopenharmony_ci	div->reg = reg;
2068c2ecf20Sopenharmony_ci	if (composite_flags & IMX_COMPOSITE_CORE) {
2078c2ecf20Sopenharmony_ci		div->shift = PCG_DIV_SHIFT;
2088c2ecf20Sopenharmony_ci		div->width = PCG_CORE_DIV_WIDTH;
2098c2ecf20Sopenharmony_ci		divider_ops = &clk_divider_ops;
2108c2ecf20Sopenharmony_ci		mux_ops = &imx8m_clk_composite_mux_ops;
2118c2ecf20Sopenharmony_ci	} else if (composite_flags & IMX_COMPOSITE_BUS) {
2128c2ecf20Sopenharmony_ci		div->shift = PCG_PREDIV_SHIFT;
2138c2ecf20Sopenharmony_ci		div->width = PCG_PREDIV_WIDTH;
2148c2ecf20Sopenharmony_ci		divider_ops = &imx8m_clk_composite_divider_ops;
2158c2ecf20Sopenharmony_ci		mux_ops = &imx8m_clk_composite_mux_ops;
2168c2ecf20Sopenharmony_ci	} else {
2178c2ecf20Sopenharmony_ci		div->shift = PCG_PREDIV_SHIFT;
2188c2ecf20Sopenharmony_ci		div->width = PCG_PREDIV_WIDTH;
2198c2ecf20Sopenharmony_ci		divider_ops = &imx8m_clk_composite_divider_ops;
2208c2ecf20Sopenharmony_ci		mux_ops = &clk_mux_ops;
2218c2ecf20Sopenharmony_ci		if (!(composite_flags & IMX_COMPOSITE_FW_MANAGED))
2228c2ecf20Sopenharmony_ci			flags |= CLK_SET_PARENT_GATE;
2238c2ecf20Sopenharmony_ci	}
2248c2ecf20Sopenharmony_ci
2258c2ecf20Sopenharmony_ci	div->lock = &imx_ccm_lock;
2268c2ecf20Sopenharmony_ci	div->flags = CLK_DIVIDER_ROUND_CLOSEST;
2278c2ecf20Sopenharmony_ci
2288c2ecf20Sopenharmony_ci	gate = kzalloc(sizeof(*gate), GFP_KERNEL);
2298c2ecf20Sopenharmony_ci	if (!gate)
2308c2ecf20Sopenharmony_ci		goto fail;
2318c2ecf20Sopenharmony_ci
2328c2ecf20Sopenharmony_ci	gate_hw = &gate->hw;
2338c2ecf20Sopenharmony_ci	gate->reg = reg;
2348c2ecf20Sopenharmony_ci	gate->bit_idx = PCG_CGC_SHIFT;
2358c2ecf20Sopenharmony_ci	gate->lock = &imx_ccm_lock;
2368c2ecf20Sopenharmony_ci
2378c2ecf20Sopenharmony_ci	hw = clk_hw_register_composite(NULL, name, parent_names, num_parents,
2388c2ecf20Sopenharmony_ci			mux_hw, mux_ops, div_hw,
2398c2ecf20Sopenharmony_ci			divider_ops, gate_hw, &clk_gate_ops, flags);
2408c2ecf20Sopenharmony_ci	if (IS_ERR(hw))
2418c2ecf20Sopenharmony_ci		goto fail;
2428c2ecf20Sopenharmony_ci
2438c2ecf20Sopenharmony_ci	return hw;
2448c2ecf20Sopenharmony_ci
2458c2ecf20Sopenharmony_cifail:
2468c2ecf20Sopenharmony_ci	kfree(gate);
2478c2ecf20Sopenharmony_ci	kfree(div);
2488c2ecf20Sopenharmony_ci	kfree(mux);
2498c2ecf20Sopenharmony_ci	return ERR_CAST(hw);
2508c2ecf20Sopenharmony_ci}
2518c2ecf20Sopenharmony_ciEXPORT_SYMBOL_GPL(imx8m_clk_hw_composite_flags);
252