162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-or-later
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (C) 2016 Maxime Ripard
462306a36Sopenharmony_ci * Maxime Ripard <maxime.ripard@free-electrons.com>
562306a36Sopenharmony_ci */
662306a36Sopenharmony_ci
762306a36Sopenharmony_ci#include <linux/clk-provider.h>
862306a36Sopenharmony_ci#include <linux/io.h>
962306a36Sopenharmony_ci
1062306a36Sopenharmony_ci#include "ccu_gate.h"
1162306a36Sopenharmony_ci#include "ccu_nk.h"
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_cistruct _ccu_nk {
1462306a36Sopenharmony_ci	unsigned long	n, min_n, max_n;
1562306a36Sopenharmony_ci	unsigned long	k, min_k, max_k;
1662306a36Sopenharmony_ci};
1762306a36Sopenharmony_ci
1862306a36Sopenharmony_cistatic unsigned long ccu_nk_find_best(unsigned long parent, unsigned long rate,
1962306a36Sopenharmony_ci				      struct _ccu_nk *nk)
2062306a36Sopenharmony_ci{
2162306a36Sopenharmony_ci	unsigned long best_rate = 0;
2262306a36Sopenharmony_ci	unsigned int best_k = 0, best_n = 0;
2362306a36Sopenharmony_ci	unsigned int _k, _n;
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_ci	for (_k = nk->min_k; _k <= nk->max_k; _k++) {
2662306a36Sopenharmony_ci		for (_n = nk->min_n; _n <= nk->max_n; _n++) {
2762306a36Sopenharmony_ci			unsigned long tmp_rate = parent * _n * _k;
2862306a36Sopenharmony_ci
2962306a36Sopenharmony_ci			if (tmp_rate > rate)
3062306a36Sopenharmony_ci				continue;
3162306a36Sopenharmony_ci
3262306a36Sopenharmony_ci			if ((rate - tmp_rate) < (rate - best_rate)) {
3362306a36Sopenharmony_ci				best_rate = tmp_rate;
3462306a36Sopenharmony_ci				best_k = _k;
3562306a36Sopenharmony_ci				best_n = _n;
3662306a36Sopenharmony_ci			}
3762306a36Sopenharmony_ci		}
3862306a36Sopenharmony_ci	}
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_ci	nk->k = best_k;
4162306a36Sopenharmony_ci	nk->n = best_n;
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_ci	return best_rate;
4462306a36Sopenharmony_ci}
4562306a36Sopenharmony_ci
4662306a36Sopenharmony_cistatic void ccu_nk_disable(struct clk_hw *hw)
4762306a36Sopenharmony_ci{
4862306a36Sopenharmony_ci	struct ccu_nk *nk = hw_to_ccu_nk(hw);
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_ci	return ccu_gate_helper_disable(&nk->common, nk->enable);
5162306a36Sopenharmony_ci}
5262306a36Sopenharmony_ci
5362306a36Sopenharmony_cistatic int ccu_nk_enable(struct clk_hw *hw)
5462306a36Sopenharmony_ci{
5562306a36Sopenharmony_ci	struct ccu_nk *nk = hw_to_ccu_nk(hw);
5662306a36Sopenharmony_ci
5762306a36Sopenharmony_ci	return ccu_gate_helper_enable(&nk->common, nk->enable);
5862306a36Sopenharmony_ci}
5962306a36Sopenharmony_ci
6062306a36Sopenharmony_cistatic int ccu_nk_is_enabled(struct clk_hw *hw)
6162306a36Sopenharmony_ci{
6262306a36Sopenharmony_ci	struct ccu_nk *nk = hw_to_ccu_nk(hw);
6362306a36Sopenharmony_ci
6462306a36Sopenharmony_ci	return ccu_gate_helper_is_enabled(&nk->common, nk->enable);
6562306a36Sopenharmony_ci}
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_cistatic unsigned long ccu_nk_recalc_rate(struct clk_hw *hw,
6862306a36Sopenharmony_ci					unsigned long parent_rate)
6962306a36Sopenharmony_ci{
7062306a36Sopenharmony_ci	struct ccu_nk *nk = hw_to_ccu_nk(hw);
7162306a36Sopenharmony_ci	unsigned long rate, n, k;
7262306a36Sopenharmony_ci	u32 reg;
7362306a36Sopenharmony_ci
7462306a36Sopenharmony_ci	reg = readl(nk->common.base + nk->common.reg);
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ci	n = reg >> nk->n.shift;
7762306a36Sopenharmony_ci	n &= (1 << nk->n.width) - 1;
7862306a36Sopenharmony_ci	n += nk->n.offset;
7962306a36Sopenharmony_ci	if (!n)
8062306a36Sopenharmony_ci		n++;
8162306a36Sopenharmony_ci
8262306a36Sopenharmony_ci	k = reg >> nk->k.shift;
8362306a36Sopenharmony_ci	k &= (1 << nk->k.width) - 1;
8462306a36Sopenharmony_ci	k += nk->k.offset;
8562306a36Sopenharmony_ci	if (!k)
8662306a36Sopenharmony_ci		k++;
8762306a36Sopenharmony_ci
8862306a36Sopenharmony_ci	rate = parent_rate * n * k;
8962306a36Sopenharmony_ci	if (nk->common.features & CCU_FEATURE_FIXED_POSTDIV)
9062306a36Sopenharmony_ci		rate /= nk->fixed_post_div;
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_ci	return rate;
9362306a36Sopenharmony_ci}
9462306a36Sopenharmony_ci
9562306a36Sopenharmony_cistatic long ccu_nk_round_rate(struct clk_hw *hw, unsigned long rate,
9662306a36Sopenharmony_ci			      unsigned long *parent_rate)
9762306a36Sopenharmony_ci{
9862306a36Sopenharmony_ci	struct ccu_nk *nk = hw_to_ccu_nk(hw);
9962306a36Sopenharmony_ci	struct _ccu_nk _nk;
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_ci	if (nk->common.features & CCU_FEATURE_FIXED_POSTDIV)
10262306a36Sopenharmony_ci		rate *= nk->fixed_post_div;
10362306a36Sopenharmony_ci
10462306a36Sopenharmony_ci	_nk.min_n = nk->n.min ?: 1;
10562306a36Sopenharmony_ci	_nk.max_n = nk->n.max ?: 1 << nk->n.width;
10662306a36Sopenharmony_ci	_nk.min_k = nk->k.min ?: 1;
10762306a36Sopenharmony_ci	_nk.max_k = nk->k.max ?: 1 << nk->k.width;
10862306a36Sopenharmony_ci
10962306a36Sopenharmony_ci	rate = ccu_nk_find_best(*parent_rate, rate, &_nk);
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_ci	if (nk->common.features & CCU_FEATURE_FIXED_POSTDIV)
11262306a36Sopenharmony_ci		rate = rate / nk->fixed_post_div;
11362306a36Sopenharmony_ci
11462306a36Sopenharmony_ci	return rate;
11562306a36Sopenharmony_ci}
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_cistatic int ccu_nk_set_rate(struct clk_hw *hw, unsigned long rate,
11862306a36Sopenharmony_ci			   unsigned long parent_rate)
11962306a36Sopenharmony_ci{
12062306a36Sopenharmony_ci	struct ccu_nk *nk = hw_to_ccu_nk(hw);
12162306a36Sopenharmony_ci	unsigned long flags;
12262306a36Sopenharmony_ci	struct _ccu_nk _nk;
12362306a36Sopenharmony_ci	u32 reg;
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci	if (nk->common.features & CCU_FEATURE_FIXED_POSTDIV)
12662306a36Sopenharmony_ci		rate = rate * nk->fixed_post_div;
12762306a36Sopenharmony_ci
12862306a36Sopenharmony_ci	_nk.min_n = nk->n.min ?: 1;
12962306a36Sopenharmony_ci	_nk.max_n = nk->n.max ?: 1 << nk->n.width;
13062306a36Sopenharmony_ci	_nk.min_k = nk->k.min ?: 1;
13162306a36Sopenharmony_ci	_nk.max_k = nk->k.max ?: 1 << nk->k.width;
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_ci	ccu_nk_find_best(parent_rate, rate, &_nk);
13462306a36Sopenharmony_ci
13562306a36Sopenharmony_ci	spin_lock_irqsave(nk->common.lock, flags);
13662306a36Sopenharmony_ci
13762306a36Sopenharmony_ci	reg = readl(nk->common.base + nk->common.reg);
13862306a36Sopenharmony_ci	reg &= ~GENMASK(nk->n.width + nk->n.shift - 1, nk->n.shift);
13962306a36Sopenharmony_ci	reg &= ~GENMASK(nk->k.width + nk->k.shift - 1, nk->k.shift);
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_ci	reg |= (_nk.k - nk->k.offset) << nk->k.shift;
14262306a36Sopenharmony_ci	reg |= (_nk.n - nk->n.offset) << nk->n.shift;
14362306a36Sopenharmony_ci	writel(reg, nk->common.base + nk->common.reg);
14462306a36Sopenharmony_ci
14562306a36Sopenharmony_ci	spin_unlock_irqrestore(nk->common.lock, flags);
14662306a36Sopenharmony_ci
14762306a36Sopenharmony_ci	ccu_helper_wait_for_lock(&nk->common, nk->lock);
14862306a36Sopenharmony_ci
14962306a36Sopenharmony_ci	return 0;
15062306a36Sopenharmony_ci}
15162306a36Sopenharmony_ci
15262306a36Sopenharmony_ciconst struct clk_ops ccu_nk_ops = {
15362306a36Sopenharmony_ci	.disable	= ccu_nk_disable,
15462306a36Sopenharmony_ci	.enable		= ccu_nk_enable,
15562306a36Sopenharmony_ci	.is_enabled	= ccu_nk_is_enabled,
15662306a36Sopenharmony_ci
15762306a36Sopenharmony_ci	.recalc_rate	= ccu_nk_recalc_rate,
15862306a36Sopenharmony_ci	.round_rate	= ccu_nk_round_rate,
15962306a36Sopenharmony_ci	.set_rate	= ccu_nk_set_rate,
16062306a36Sopenharmony_ci};
16162306a36Sopenharmony_ciEXPORT_SYMBOL_NS_GPL(ccu_nk_ops, SUNXI_CCU);
162