162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright 2018 NXP
462306a36Sopenharmony_ci */
562306a36Sopenharmony_ci
662306a36Sopenharmony_ci#include <linux/clk-provider.h>
762306a36Sopenharmony_ci#include <linux/errno.h>
862306a36Sopenharmony_ci#include <linux/export.h>
962306a36Sopenharmony_ci#include <linux/io.h>
1062306a36Sopenharmony_ci#include <linux/slab.h>
1162306a36Sopenharmony_ci
1262306a36Sopenharmony_ci#include "clk.h"
1362306a36Sopenharmony_ci
1462306a36Sopenharmony_ci#define PCG_PREDIV_SHIFT	16
1562306a36Sopenharmony_ci#define PCG_PREDIV_WIDTH	3
1662306a36Sopenharmony_ci#define PCG_PREDIV_MAX		8
1762306a36Sopenharmony_ci
1862306a36Sopenharmony_ci#define PCG_DIV_SHIFT		0
1962306a36Sopenharmony_ci#define PCG_CORE_DIV_WIDTH	3
2062306a36Sopenharmony_ci#define PCG_DIV_WIDTH		6
2162306a36Sopenharmony_ci#define PCG_DIV_MAX		64
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_ci#define PCG_PCS_SHIFT		24
2462306a36Sopenharmony_ci#define PCG_PCS_MASK		0x7
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_ci#define PCG_CGC_SHIFT		28
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_cistatic unsigned long imx8m_clk_composite_divider_recalc_rate(struct clk_hw *hw,
2962306a36Sopenharmony_ci						unsigned long parent_rate)
3062306a36Sopenharmony_ci{
3162306a36Sopenharmony_ci	struct clk_divider *divider = to_clk_divider(hw);
3262306a36Sopenharmony_ci	unsigned long prediv_rate;
3362306a36Sopenharmony_ci	unsigned int prediv_value;
3462306a36Sopenharmony_ci	unsigned int div_value;
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_ci	prediv_value = readl(divider->reg) >> divider->shift;
3762306a36Sopenharmony_ci	prediv_value &= clk_div_mask(divider->width);
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ci	prediv_rate = divider_recalc_rate(hw, parent_rate, prediv_value,
4062306a36Sopenharmony_ci						NULL, divider->flags,
4162306a36Sopenharmony_ci						divider->width);
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_ci	div_value = readl(divider->reg) >> PCG_DIV_SHIFT;
4462306a36Sopenharmony_ci	div_value &= clk_div_mask(PCG_DIV_WIDTH);
4562306a36Sopenharmony_ci
4662306a36Sopenharmony_ci	return divider_recalc_rate(hw, prediv_rate, div_value, NULL,
4762306a36Sopenharmony_ci				   divider->flags, PCG_DIV_WIDTH);
4862306a36Sopenharmony_ci}
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_cistatic int imx8m_clk_composite_compute_dividers(unsigned long rate,
5162306a36Sopenharmony_ci						unsigned long parent_rate,
5262306a36Sopenharmony_ci						int *prediv, int *postdiv)
5362306a36Sopenharmony_ci{
5462306a36Sopenharmony_ci	int div1, div2;
5562306a36Sopenharmony_ci	int error = INT_MAX;
5662306a36Sopenharmony_ci	int ret = -EINVAL;
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci	*prediv = 1;
5962306a36Sopenharmony_ci	*postdiv = 1;
6062306a36Sopenharmony_ci
6162306a36Sopenharmony_ci	for (div1 = 1; div1 <= PCG_PREDIV_MAX; div1++) {
6262306a36Sopenharmony_ci		for (div2 = 1; div2 <= PCG_DIV_MAX; div2++) {
6362306a36Sopenharmony_ci			int new_error = ((parent_rate / div1) / div2) - rate;
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_ci			if (abs(new_error) < abs(error)) {
6662306a36Sopenharmony_ci				*prediv = div1;
6762306a36Sopenharmony_ci				*postdiv = div2;
6862306a36Sopenharmony_ci				error = new_error;
6962306a36Sopenharmony_ci				ret = 0;
7062306a36Sopenharmony_ci			}
7162306a36Sopenharmony_ci		}
7262306a36Sopenharmony_ci	}
7362306a36Sopenharmony_ci	return ret;
7462306a36Sopenharmony_ci}
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_cistatic long imx8m_clk_composite_divider_round_rate(struct clk_hw *hw,
7762306a36Sopenharmony_ci						unsigned long rate,
7862306a36Sopenharmony_ci						unsigned long *prate)
7962306a36Sopenharmony_ci{
8062306a36Sopenharmony_ci	int prediv_value;
8162306a36Sopenharmony_ci	int div_value;
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_ci	imx8m_clk_composite_compute_dividers(rate, *prate,
8462306a36Sopenharmony_ci						&prediv_value, &div_value);
8562306a36Sopenharmony_ci	rate = DIV_ROUND_UP(*prate, prediv_value);
8662306a36Sopenharmony_ci
8762306a36Sopenharmony_ci	return DIV_ROUND_UP(rate, div_value);
8862306a36Sopenharmony_ci
8962306a36Sopenharmony_ci}
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_cistatic int imx8m_clk_composite_divider_set_rate(struct clk_hw *hw,
9262306a36Sopenharmony_ci					unsigned long rate,
9362306a36Sopenharmony_ci					unsigned long parent_rate)
9462306a36Sopenharmony_ci{
9562306a36Sopenharmony_ci	struct clk_divider *divider = to_clk_divider(hw);
9662306a36Sopenharmony_ci	unsigned long flags;
9762306a36Sopenharmony_ci	int prediv_value;
9862306a36Sopenharmony_ci	int div_value;
9962306a36Sopenharmony_ci	int ret;
10062306a36Sopenharmony_ci	u32 orig, val;
10162306a36Sopenharmony_ci
10262306a36Sopenharmony_ci	ret = imx8m_clk_composite_compute_dividers(rate, parent_rate,
10362306a36Sopenharmony_ci						&prediv_value, &div_value);
10462306a36Sopenharmony_ci	if (ret)
10562306a36Sopenharmony_ci		return -EINVAL;
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_ci	spin_lock_irqsave(divider->lock, flags);
10862306a36Sopenharmony_ci
10962306a36Sopenharmony_ci	orig = readl(divider->reg);
11062306a36Sopenharmony_ci	val = orig & ~((clk_div_mask(divider->width) << divider->shift) |
11162306a36Sopenharmony_ci		       (clk_div_mask(PCG_DIV_WIDTH) << PCG_DIV_SHIFT));
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_ci	val |= (u32)(prediv_value  - 1) << divider->shift;
11462306a36Sopenharmony_ci	val |= (u32)(div_value - 1) << PCG_DIV_SHIFT;
11562306a36Sopenharmony_ci
11662306a36Sopenharmony_ci	if (val != orig)
11762306a36Sopenharmony_ci		writel(val, divider->reg);
11862306a36Sopenharmony_ci
11962306a36Sopenharmony_ci	spin_unlock_irqrestore(divider->lock, flags);
12062306a36Sopenharmony_ci
12162306a36Sopenharmony_ci	return ret;
12262306a36Sopenharmony_ci}
12362306a36Sopenharmony_ci
12462306a36Sopenharmony_cistatic int imx8m_divider_determine_rate(struct clk_hw *hw,
12562306a36Sopenharmony_ci				      struct clk_rate_request *req)
12662306a36Sopenharmony_ci{
12762306a36Sopenharmony_ci	struct clk_divider *divider = to_clk_divider(hw);
12862306a36Sopenharmony_ci	int prediv_value;
12962306a36Sopenharmony_ci	int div_value;
13062306a36Sopenharmony_ci
13162306a36Sopenharmony_ci	/* if read only, just return current value */
13262306a36Sopenharmony_ci	if (divider->flags & CLK_DIVIDER_READ_ONLY) {
13362306a36Sopenharmony_ci		u32 val;
13462306a36Sopenharmony_ci
13562306a36Sopenharmony_ci		val = readl(divider->reg);
13662306a36Sopenharmony_ci		prediv_value = val >> divider->shift;
13762306a36Sopenharmony_ci		prediv_value &= clk_div_mask(divider->width);
13862306a36Sopenharmony_ci		prediv_value++;
13962306a36Sopenharmony_ci
14062306a36Sopenharmony_ci		div_value = val >> PCG_DIV_SHIFT;
14162306a36Sopenharmony_ci		div_value &= clk_div_mask(PCG_DIV_WIDTH);
14262306a36Sopenharmony_ci		div_value++;
14362306a36Sopenharmony_ci
14462306a36Sopenharmony_ci		return divider_ro_determine_rate(hw, req, divider->table,
14562306a36Sopenharmony_ci						 PCG_PREDIV_WIDTH + PCG_DIV_WIDTH,
14662306a36Sopenharmony_ci						 divider->flags, prediv_value * div_value);
14762306a36Sopenharmony_ci	}
14862306a36Sopenharmony_ci
14962306a36Sopenharmony_ci	return divider_determine_rate(hw, req, divider->table,
15062306a36Sopenharmony_ci				      PCG_PREDIV_WIDTH + PCG_DIV_WIDTH,
15162306a36Sopenharmony_ci				      divider->flags);
15262306a36Sopenharmony_ci}
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_cistatic const struct clk_ops imx8m_clk_composite_divider_ops = {
15562306a36Sopenharmony_ci	.recalc_rate = imx8m_clk_composite_divider_recalc_rate,
15662306a36Sopenharmony_ci	.round_rate = imx8m_clk_composite_divider_round_rate,
15762306a36Sopenharmony_ci	.set_rate = imx8m_clk_composite_divider_set_rate,
15862306a36Sopenharmony_ci	.determine_rate = imx8m_divider_determine_rate,
15962306a36Sopenharmony_ci};
16062306a36Sopenharmony_ci
16162306a36Sopenharmony_cistatic u8 imx8m_clk_composite_mux_get_parent(struct clk_hw *hw)
16262306a36Sopenharmony_ci{
16362306a36Sopenharmony_ci	return clk_mux_ops.get_parent(hw);
16462306a36Sopenharmony_ci}
16562306a36Sopenharmony_ci
16662306a36Sopenharmony_cistatic int imx8m_clk_composite_mux_set_parent(struct clk_hw *hw, u8 index)
16762306a36Sopenharmony_ci{
16862306a36Sopenharmony_ci	struct clk_mux *mux = to_clk_mux(hw);
16962306a36Sopenharmony_ci	u32 val = clk_mux_index_to_val(mux->table, mux->flags, index);
17062306a36Sopenharmony_ci	unsigned long flags = 0;
17162306a36Sopenharmony_ci	u32 reg;
17262306a36Sopenharmony_ci
17362306a36Sopenharmony_ci	if (mux->lock)
17462306a36Sopenharmony_ci		spin_lock_irqsave(mux->lock, flags);
17562306a36Sopenharmony_ci
17662306a36Sopenharmony_ci	reg = readl(mux->reg);
17762306a36Sopenharmony_ci	reg &= ~(mux->mask << mux->shift);
17862306a36Sopenharmony_ci	val = val << mux->shift;
17962306a36Sopenharmony_ci	reg |= val;
18062306a36Sopenharmony_ci	/*
18162306a36Sopenharmony_ci	 * write twice to make sure non-target interface
18262306a36Sopenharmony_ci	 * SEL_A/B point the same clk input.
18362306a36Sopenharmony_ci	 */
18462306a36Sopenharmony_ci	writel(reg, mux->reg);
18562306a36Sopenharmony_ci	writel(reg, mux->reg);
18662306a36Sopenharmony_ci
18762306a36Sopenharmony_ci	if (mux->lock)
18862306a36Sopenharmony_ci		spin_unlock_irqrestore(mux->lock, flags);
18962306a36Sopenharmony_ci
19062306a36Sopenharmony_ci	return 0;
19162306a36Sopenharmony_ci}
19262306a36Sopenharmony_ci
19362306a36Sopenharmony_cistatic int
19462306a36Sopenharmony_ciimx8m_clk_composite_mux_determine_rate(struct clk_hw *hw,
19562306a36Sopenharmony_ci				       struct clk_rate_request *req)
19662306a36Sopenharmony_ci{
19762306a36Sopenharmony_ci	return clk_mux_ops.determine_rate(hw, req);
19862306a36Sopenharmony_ci}
19962306a36Sopenharmony_ci
20062306a36Sopenharmony_ci
20162306a36Sopenharmony_cistatic const struct clk_ops imx8m_clk_composite_mux_ops = {
20262306a36Sopenharmony_ci	.get_parent = imx8m_clk_composite_mux_get_parent,
20362306a36Sopenharmony_ci	.set_parent = imx8m_clk_composite_mux_set_parent,
20462306a36Sopenharmony_ci	.determine_rate = imx8m_clk_composite_mux_determine_rate,
20562306a36Sopenharmony_ci};
20662306a36Sopenharmony_ci
20762306a36Sopenharmony_cistruct clk_hw *__imx8m_clk_hw_composite(const char *name,
20862306a36Sopenharmony_ci					const char * const *parent_names,
20962306a36Sopenharmony_ci					int num_parents, void __iomem *reg,
21062306a36Sopenharmony_ci					u32 composite_flags,
21162306a36Sopenharmony_ci					unsigned long flags)
21262306a36Sopenharmony_ci{
21362306a36Sopenharmony_ci	struct clk_hw *hw = ERR_PTR(-ENOMEM), *mux_hw;
21462306a36Sopenharmony_ci	struct clk_hw *div_hw, *gate_hw = NULL;
21562306a36Sopenharmony_ci	struct clk_divider *div = NULL;
21662306a36Sopenharmony_ci	struct clk_gate *gate = NULL;
21762306a36Sopenharmony_ci	struct clk_mux *mux = NULL;
21862306a36Sopenharmony_ci	const struct clk_ops *divider_ops;
21962306a36Sopenharmony_ci	const struct clk_ops *mux_ops;
22062306a36Sopenharmony_ci
22162306a36Sopenharmony_ci	mux = kzalloc(sizeof(*mux), GFP_KERNEL);
22262306a36Sopenharmony_ci	if (!mux)
22362306a36Sopenharmony_ci		goto fail;
22462306a36Sopenharmony_ci
22562306a36Sopenharmony_ci	mux_hw = &mux->hw;
22662306a36Sopenharmony_ci	mux->reg = reg;
22762306a36Sopenharmony_ci	mux->shift = PCG_PCS_SHIFT;
22862306a36Sopenharmony_ci	mux->mask = PCG_PCS_MASK;
22962306a36Sopenharmony_ci	mux->lock = &imx_ccm_lock;
23062306a36Sopenharmony_ci
23162306a36Sopenharmony_ci	div = kzalloc(sizeof(*div), GFP_KERNEL);
23262306a36Sopenharmony_ci	if (!div)
23362306a36Sopenharmony_ci		goto fail;
23462306a36Sopenharmony_ci
23562306a36Sopenharmony_ci	div_hw = &div->hw;
23662306a36Sopenharmony_ci	div->reg = reg;
23762306a36Sopenharmony_ci	if (composite_flags & IMX_COMPOSITE_CORE) {
23862306a36Sopenharmony_ci		div->shift = PCG_DIV_SHIFT;
23962306a36Sopenharmony_ci		div->width = PCG_CORE_DIV_WIDTH;
24062306a36Sopenharmony_ci		divider_ops = &clk_divider_ops;
24162306a36Sopenharmony_ci		mux_ops = &imx8m_clk_composite_mux_ops;
24262306a36Sopenharmony_ci	} else if (composite_flags & IMX_COMPOSITE_BUS) {
24362306a36Sopenharmony_ci		div->shift = PCG_PREDIV_SHIFT;
24462306a36Sopenharmony_ci		div->width = PCG_PREDIV_WIDTH;
24562306a36Sopenharmony_ci		divider_ops = &imx8m_clk_composite_divider_ops;
24662306a36Sopenharmony_ci		mux_ops = &imx8m_clk_composite_mux_ops;
24762306a36Sopenharmony_ci	} else {
24862306a36Sopenharmony_ci		div->shift = PCG_PREDIV_SHIFT;
24962306a36Sopenharmony_ci		div->width = PCG_PREDIV_WIDTH;
25062306a36Sopenharmony_ci		divider_ops = &imx8m_clk_composite_divider_ops;
25162306a36Sopenharmony_ci		mux_ops = &clk_mux_ops;
25262306a36Sopenharmony_ci		if (!(composite_flags & IMX_COMPOSITE_FW_MANAGED))
25362306a36Sopenharmony_ci			flags |= CLK_SET_PARENT_GATE;
25462306a36Sopenharmony_ci	}
25562306a36Sopenharmony_ci
25662306a36Sopenharmony_ci	div->lock = &imx_ccm_lock;
25762306a36Sopenharmony_ci	div->flags = CLK_DIVIDER_ROUND_CLOSEST;
25862306a36Sopenharmony_ci
25962306a36Sopenharmony_ci	/* skip registering the gate ops if M4 is enabled */
26062306a36Sopenharmony_ci	if (!mcore_booted) {
26162306a36Sopenharmony_ci		gate = kzalloc(sizeof(*gate), GFP_KERNEL);
26262306a36Sopenharmony_ci		if (!gate)
26362306a36Sopenharmony_ci			goto fail;
26462306a36Sopenharmony_ci
26562306a36Sopenharmony_ci		gate_hw = &gate->hw;
26662306a36Sopenharmony_ci		gate->reg = reg;
26762306a36Sopenharmony_ci		gate->bit_idx = PCG_CGC_SHIFT;
26862306a36Sopenharmony_ci		gate->lock = &imx_ccm_lock;
26962306a36Sopenharmony_ci	}
27062306a36Sopenharmony_ci
27162306a36Sopenharmony_ci	hw = clk_hw_register_composite(NULL, name, parent_names, num_parents,
27262306a36Sopenharmony_ci			mux_hw, mux_ops, div_hw,
27362306a36Sopenharmony_ci			divider_ops, gate_hw, &clk_gate_ops, flags);
27462306a36Sopenharmony_ci	if (IS_ERR(hw))
27562306a36Sopenharmony_ci		goto fail;
27662306a36Sopenharmony_ci
27762306a36Sopenharmony_ci	return hw;
27862306a36Sopenharmony_ci
27962306a36Sopenharmony_cifail:
28062306a36Sopenharmony_ci	kfree(gate);
28162306a36Sopenharmony_ci	kfree(div);
28262306a36Sopenharmony_ci	kfree(mux);
28362306a36Sopenharmony_ci	return ERR_CAST(hw);
28462306a36Sopenharmony_ci}
28562306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(__imx8m_clk_hw_composite);
286