162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (c) 2013, The Linux Foundation. All rights reserved.
462306a36Sopenharmony_ci */
562306a36Sopenharmony_ci
662306a36Sopenharmony_ci#include <linux/kernel.h>
762306a36Sopenharmony_ci#include <linux/bitops.h>
862306a36Sopenharmony_ci#include <linux/err.h>
962306a36Sopenharmony_ci#include <linux/delay.h>
1062306a36Sopenharmony_ci#include <linux/export.h>
1162306a36Sopenharmony_ci#include <linux/clk-provider.h>
1262306a36Sopenharmony_ci#include <linux/regmap.h>
1362306a36Sopenharmony_ci
1462306a36Sopenharmony_ci#include "clk-branch.h"
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_cistatic bool clk_branch_in_hwcg_mode(const struct clk_branch *br)
1762306a36Sopenharmony_ci{
1862306a36Sopenharmony_ci	u32 val;
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_ci	if (!br->hwcg_reg)
2162306a36Sopenharmony_ci		return false;
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_ci	regmap_read(br->clkr.regmap, br->hwcg_reg, &val);
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_ci	return !!(val & BIT(br->hwcg_bit));
2662306a36Sopenharmony_ci}
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_cistatic bool clk_branch_check_halt(const struct clk_branch *br, bool enabling)
2962306a36Sopenharmony_ci{
3062306a36Sopenharmony_ci	bool invert = (br->halt_check == BRANCH_HALT_ENABLE);
3162306a36Sopenharmony_ci	u32 val;
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_ci	regmap_read(br->clkr.regmap, br->halt_reg, &val);
3462306a36Sopenharmony_ci
3562306a36Sopenharmony_ci	val &= BIT(br->halt_bit);
3662306a36Sopenharmony_ci	if (invert)
3762306a36Sopenharmony_ci		val = !val;
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ci	return !!val == !enabling;
4062306a36Sopenharmony_ci}
4162306a36Sopenharmony_ci
4262306a36Sopenharmony_cistatic bool clk_branch2_check_halt(const struct clk_branch *br, bool enabling)
4362306a36Sopenharmony_ci{
4462306a36Sopenharmony_ci	u32 val;
4562306a36Sopenharmony_ci	u32 mask;
4662306a36Sopenharmony_ci	bool invert = (br->halt_check == BRANCH_HALT_ENABLE);
4762306a36Sopenharmony_ci
4862306a36Sopenharmony_ci	mask = CBCR_NOC_FSM_STATUS;
4962306a36Sopenharmony_ci	mask |= CBCR_CLK_OFF;
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_ci	regmap_read(br->clkr.regmap, br->halt_reg, &val);
5262306a36Sopenharmony_ci
5362306a36Sopenharmony_ci	if (enabling) {
5462306a36Sopenharmony_ci		val &= mask;
5562306a36Sopenharmony_ci		return (val & CBCR_CLK_OFF) == (invert ? CBCR_CLK_OFF : 0) ||
5662306a36Sopenharmony_ci			FIELD_GET(CBCR_NOC_FSM_STATUS, val) == FSM_STATUS_ON;
5762306a36Sopenharmony_ci	}
5862306a36Sopenharmony_ci	return (val & CBCR_CLK_OFF) == (invert ? 0 : CBCR_CLK_OFF);
5962306a36Sopenharmony_ci}
6062306a36Sopenharmony_ci
6162306a36Sopenharmony_cistatic int clk_branch_wait(const struct clk_branch *br, bool enabling,
6262306a36Sopenharmony_ci		bool (check_halt)(const struct clk_branch *, bool))
6362306a36Sopenharmony_ci{
6462306a36Sopenharmony_ci	bool voted = br->halt_check & BRANCH_VOTED;
6562306a36Sopenharmony_ci	const char *name = clk_hw_get_name(&br->clkr.hw);
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_ci	/*
6862306a36Sopenharmony_ci	 * Skip checking halt bit if we're explicitly ignoring the bit or the
6962306a36Sopenharmony_ci	 * clock is in hardware gated mode
7062306a36Sopenharmony_ci	 */
7162306a36Sopenharmony_ci	if (br->halt_check == BRANCH_HALT_SKIP || clk_branch_in_hwcg_mode(br))
7262306a36Sopenharmony_ci		return 0;
7362306a36Sopenharmony_ci
7462306a36Sopenharmony_ci	if (br->halt_check == BRANCH_HALT_DELAY || (!enabling && voted)) {
7562306a36Sopenharmony_ci		udelay(10);
7662306a36Sopenharmony_ci	} else if (br->halt_check == BRANCH_HALT_ENABLE ||
7762306a36Sopenharmony_ci		   br->halt_check == BRANCH_HALT ||
7862306a36Sopenharmony_ci		   (enabling && voted)) {
7962306a36Sopenharmony_ci		int count = 200;
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ci		while (count-- > 0) {
8262306a36Sopenharmony_ci			if (check_halt(br, enabling))
8362306a36Sopenharmony_ci				return 0;
8462306a36Sopenharmony_ci			udelay(1);
8562306a36Sopenharmony_ci		}
8662306a36Sopenharmony_ci		WARN(1, "%s status stuck at 'o%s'", name,
8762306a36Sopenharmony_ci				enabling ? "ff" : "n");
8862306a36Sopenharmony_ci		return -EBUSY;
8962306a36Sopenharmony_ci	}
9062306a36Sopenharmony_ci	return 0;
9162306a36Sopenharmony_ci}
9262306a36Sopenharmony_ci
9362306a36Sopenharmony_cistatic int clk_branch_toggle(struct clk_hw *hw, bool en,
9462306a36Sopenharmony_ci		bool (check_halt)(const struct clk_branch *, bool))
9562306a36Sopenharmony_ci{
9662306a36Sopenharmony_ci	struct clk_branch *br = to_clk_branch(hw);
9762306a36Sopenharmony_ci	int ret;
9862306a36Sopenharmony_ci
9962306a36Sopenharmony_ci	if (en) {
10062306a36Sopenharmony_ci		ret = clk_enable_regmap(hw);
10162306a36Sopenharmony_ci		if (ret)
10262306a36Sopenharmony_ci			return ret;
10362306a36Sopenharmony_ci	} else {
10462306a36Sopenharmony_ci		clk_disable_regmap(hw);
10562306a36Sopenharmony_ci	}
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_ci	return clk_branch_wait(br, en, check_halt);
10862306a36Sopenharmony_ci}
10962306a36Sopenharmony_ci
11062306a36Sopenharmony_cistatic int clk_branch_enable(struct clk_hw *hw)
11162306a36Sopenharmony_ci{
11262306a36Sopenharmony_ci	return clk_branch_toggle(hw, true, clk_branch_check_halt);
11362306a36Sopenharmony_ci}
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_cistatic void clk_branch_disable(struct clk_hw *hw)
11662306a36Sopenharmony_ci{
11762306a36Sopenharmony_ci	clk_branch_toggle(hw, false, clk_branch_check_halt);
11862306a36Sopenharmony_ci}
11962306a36Sopenharmony_ci
12062306a36Sopenharmony_ciconst struct clk_ops clk_branch_ops = {
12162306a36Sopenharmony_ci	.enable = clk_branch_enable,
12262306a36Sopenharmony_ci	.disable = clk_branch_disable,
12362306a36Sopenharmony_ci	.is_enabled = clk_is_enabled_regmap,
12462306a36Sopenharmony_ci};
12562306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(clk_branch_ops);
12662306a36Sopenharmony_ci
12762306a36Sopenharmony_cistatic int clk_branch2_enable(struct clk_hw *hw)
12862306a36Sopenharmony_ci{
12962306a36Sopenharmony_ci	return clk_branch_toggle(hw, true, clk_branch2_check_halt);
13062306a36Sopenharmony_ci}
13162306a36Sopenharmony_ci
13262306a36Sopenharmony_cistatic void clk_branch2_disable(struct clk_hw *hw)
13362306a36Sopenharmony_ci{
13462306a36Sopenharmony_ci	clk_branch_toggle(hw, false, clk_branch2_check_halt);
13562306a36Sopenharmony_ci}
13662306a36Sopenharmony_ci
13762306a36Sopenharmony_ciconst struct clk_ops clk_branch2_ops = {
13862306a36Sopenharmony_ci	.enable = clk_branch2_enable,
13962306a36Sopenharmony_ci	.disable = clk_branch2_disable,
14062306a36Sopenharmony_ci	.is_enabled = clk_is_enabled_regmap,
14162306a36Sopenharmony_ci};
14262306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(clk_branch2_ops);
14362306a36Sopenharmony_ci
14462306a36Sopenharmony_ciconst struct clk_ops clk_branch2_aon_ops = {
14562306a36Sopenharmony_ci	.enable = clk_branch2_enable,
14662306a36Sopenharmony_ci	.is_enabled = clk_is_enabled_regmap,
14762306a36Sopenharmony_ci};
14862306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(clk_branch2_aon_ops);
14962306a36Sopenharmony_ci
15062306a36Sopenharmony_ciconst struct clk_ops clk_branch_simple_ops = {
15162306a36Sopenharmony_ci	.enable = clk_enable_regmap,
15262306a36Sopenharmony_ci	.disable = clk_disable_regmap,
15362306a36Sopenharmony_ci	.is_enabled = clk_is_enabled_regmap,
15462306a36Sopenharmony_ci};
15562306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(clk_branch_simple_ops);
156