18c2ecf20Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
28c2ecf20Sopenharmony_ci// Copyright (c) 2018, The Linux Foundation. All rights reserved.
38c2ecf20Sopenharmony_ci
48c2ecf20Sopenharmony_ci#include <linux/kernel.h>
58c2ecf20Sopenharmony_ci#include <linux/module.h>
68c2ecf20Sopenharmony_ci#include <linux/init.h>
78c2ecf20Sopenharmony_ci#include <linux/io.h>
88c2ecf20Sopenharmony_ci#include <linux/delay.h>
98c2ecf20Sopenharmony_ci#include <linux/err.h>
108c2ecf20Sopenharmony_ci#include <linux/clk-provider.h>
118c2ecf20Sopenharmony_ci#include <linux/spinlock.h>
128c2ecf20Sopenharmony_ci
138c2ecf20Sopenharmony_ci#include <asm/krait-l2-accessors.h>
148c2ecf20Sopenharmony_ci
158c2ecf20Sopenharmony_ci#include "clk-krait.h"
168c2ecf20Sopenharmony_ci
178c2ecf20Sopenharmony_ci/* Secondary and primary muxes share the same cp15 register */
188c2ecf20Sopenharmony_cistatic DEFINE_SPINLOCK(krait_clock_reg_lock);
198c2ecf20Sopenharmony_ci
208c2ecf20Sopenharmony_ci#define LPL_SHIFT	8
218c2ecf20Sopenharmony_cistatic void __krait_mux_set_sel(struct krait_mux_clk *mux, int sel)
228c2ecf20Sopenharmony_ci{
238c2ecf20Sopenharmony_ci	unsigned long flags;
248c2ecf20Sopenharmony_ci	u32 regval;
258c2ecf20Sopenharmony_ci
268c2ecf20Sopenharmony_ci	spin_lock_irqsave(&krait_clock_reg_lock, flags);
278c2ecf20Sopenharmony_ci	regval = krait_get_l2_indirect_reg(mux->offset);
288c2ecf20Sopenharmony_ci	regval &= ~(mux->mask << mux->shift);
298c2ecf20Sopenharmony_ci	regval |= (sel & mux->mask) << mux->shift;
308c2ecf20Sopenharmony_ci	if (mux->lpl) {
318c2ecf20Sopenharmony_ci		regval &= ~(mux->mask << (mux->shift + LPL_SHIFT));
328c2ecf20Sopenharmony_ci		regval |= (sel & mux->mask) << (mux->shift + LPL_SHIFT);
338c2ecf20Sopenharmony_ci	}
348c2ecf20Sopenharmony_ci	krait_set_l2_indirect_reg(mux->offset, regval);
358c2ecf20Sopenharmony_ci
368c2ecf20Sopenharmony_ci	/* Wait for switch to complete. */
378c2ecf20Sopenharmony_ci	mb();
388c2ecf20Sopenharmony_ci	udelay(1);
398c2ecf20Sopenharmony_ci
408c2ecf20Sopenharmony_ci	/*
418c2ecf20Sopenharmony_ci	 * Unlock now to make sure the mux register is not
428c2ecf20Sopenharmony_ci	 * modified while switching to the new parent.
438c2ecf20Sopenharmony_ci	 */
448c2ecf20Sopenharmony_ci	spin_unlock_irqrestore(&krait_clock_reg_lock, flags);
458c2ecf20Sopenharmony_ci}
468c2ecf20Sopenharmony_ci
478c2ecf20Sopenharmony_cistatic int krait_mux_set_parent(struct clk_hw *hw, u8 index)
488c2ecf20Sopenharmony_ci{
498c2ecf20Sopenharmony_ci	struct krait_mux_clk *mux = to_krait_mux_clk(hw);
508c2ecf20Sopenharmony_ci	u32 sel;
518c2ecf20Sopenharmony_ci
528c2ecf20Sopenharmony_ci	sel = clk_mux_index_to_val(mux->parent_map, 0, index);
538c2ecf20Sopenharmony_ci	mux->en_mask = sel;
548c2ecf20Sopenharmony_ci	/* Don't touch mux if CPU is off as it won't work */
558c2ecf20Sopenharmony_ci	if (__clk_is_enabled(hw->clk))
568c2ecf20Sopenharmony_ci		__krait_mux_set_sel(mux, sel);
578c2ecf20Sopenharmony_ci
588c2ecf20Sopenharmony_ci	mux->reparent = true;
598c2ecf20Sopenharmony_ci
608c2ecf20Sopenharmony_ci	return 0;
618c2ecf20Sopenharmony_ci}
628c2ecf20Sopenharmony_ci
638c2ecf20Sopenharmony_cistatic u8 krait_mux_get_parent(struct clk_hw *hw)
648c2ecf20Sopenharmony_ci{
658c2ecf20Sopenharmony_ci	struct krait_mux_clk *mux = to_krait_mux_clk(hw);
668c2ecf20Sopenharmony_ci	u32 sel;
678c2ecf20Sopenharmony_ci
688c2ecf20Sopenharmony_ci	sel = krait_get_l2_indirect_reg(mux->offset);
698c2ecf20Sopenharmony_ci	sel >>= mux->shift;
708c2ecf20Sopenharmony_ci	sel &= mux->mask;
718c2ecf20Sopenharmony_ci	mux->en_mask = sel;
728c2ecf20Sopenharmony_ci
738c2ecf20Sopenharmony_ci	return clk_mux_val_to_index(hw, mux->parent_map, 0, sel);
748c2ecf20Sopenharmony_ci}
758c2ecf20Sopenharmony_ci
768c2ecf20Sopenharmony_ciconst struct clk_ops krait_mux_clk_ops = {
778c2ecf20Sopenharmony_ci	.set_parent = krait_mux_set_parent,
788c2ecf20Sopenharmony_ci	.get_parent = krait_mux_get_parent,
798c2ecf20Sopenharmony_ci	.determine_rate = __clk_mux_determine_rate_closest,
808c2ecf20Sopenharmony_ci};
818c2ecf20Sopenharmony_ciEXPORT_SYMBOL_GPL(krait_mux_clk_ops);
828c2ecf20Sopenharmony_ci
838c2ecf20Sopenharmony_ci/* The divider can divide by 2, 4, 6 and 8. But we only really need div-2. */
848c2ecf20Sopenharmony_cistatic long krait_div2_round_rate(struct clk_hw *hw, unsigned long rate,
858c2ecf20Sopenharmony_ci				  unsigned long *parent_rate)
868c2ecf20Sopenharmony_ci{
878c2ecf20Sopenharmony_ci	*parent_rate = clk_hw_round_rate(clk_hw_get_parent(hw), rate * 2);
888c2ecf20Sopenharmony_ci	return DIV_ROUND_UP(*parent_rate, 2);
898c2ecf20Sopenharmony_ci}
908c2ecf20Sopenharmony_ci
918c2ecf20Sopenharmony_cistatic int krait_div2_set_rate(struct clk_hw *hw, unsigned long rate,
928c2ecf20Sopenharmony_ci			       unsigned long parent_rate)
938c2ecf20Sopenharmony_ci{
948c2ecf20Sopenharmony_ci	struct krait_div2_clk *d = to_krait_div2_clk(hw);
958c2ecf20Sopenharmony_ci	unsigned long flags;
968c2ecf20Sopenharmony_ci	u32 val;
978c2ecf20Sopenharmony_ci	u32 mask = BIT(d->width) - 1;
988c2ecf20Sopenharmony_ci
998c2ecf20Sopenharmony_ci	if (d->lpl)
1008c2ecf20Sopenharmony_ci		mask = mask << (d->shift + LPL_SHIFT) | mask << d->shift;
1018c2ecf20Sopenharmony_ci	else
1028c2ecf20Sopenharmony_ci		mask <<= d->shift;
1038c2ecf20Sopenharmony_ci
1048c2ecf20Sopenharmony_ci	spin_lock_irqsave(&krait_clock_reg_lock, flags);
1058c2ecf20Sopenharmony_ci	val = krait_get_l2_indirect_reg(d->offset);
1068c2ecf20Sopenharmony_ci	val &= ~mask;
1078c2ecf20Sopenharmony_ci	krait_set_l2_indirect_reg(d->offset, val);
1088c2ecf20Sopenharmony_ci	spin_unlock_irqrestore(&krait_clock_reg_lock, flags);
1098c2ecf20Sopenharmony_ci
1108c2ecf20Sopenharmony_ci	return 0;
1118c2ecf20Sopenharmony_ci}
1128c2ecf20Sopenharmony_ci
1138c2ecf20Sopenharmony_cistatic unsigned long
1148c2ecf20Sopenharmony_cikrait_div2_recalc_rate(struct clk_hw *hw, unsigned long parent_rate)
1158c2ecf20Sopenharmony_ci{
1168c2ecf20Sopenharmony_ci	struct krait_div2_clk *d = to_krait_div2_clk(hw);
1178c2ecf20Sopenharmony_ci	u32 mask = BIT(d->width) - 1;
1188c2ecf20Sopenharmony_ci	u32 div;
1198c2ecf20Sopenharmony_ci
1208c2ecf20Sopenharmony_ci	div = krait_get_l2_indirect_reg(d->offset);
1218c2ecf20Sopenharmony_ci	div >>= d->shift;
1228c2ecf20Sopenharmony_ci	div &= mask;
1238c2ecf20Sopenharmony_ci	div = (div + 1) * 2;
1248c2ecf20Sopenharmony_ci
1258c2ecf20Sopenharmony_ci	return DIV_ROUND_UP(parent_rate, div);
1268c2ecf20Sopenharmony_ci}
1278c2ecf20Sopenharmony_ci
1288c2ecf20Sopenharmony_ciconst struct clk_ops krait_div2_clk_ops = {
1298c2ecf20Sopenharmony_ci	.round_rate = krait_div2_round_rate,
1308c2ecf20Sopenharmony_ci	.set_rate = krait_div2_set_rate,
1318c2ecf20Sopenharmony_ci	.recalc_rate = krait_div2_recalc_rate,
1328c2ecf20Sopenharmony_ci};
1338c2ecf20Sopenharmony_ciEXPORT_SYMBOL_GPL(krait_div2_clk_ops);
134