162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (c) 2014 MediaTek Inc.
462306a36Sopenharmony_ci * Author: James Liao <jamesjj.liao@mediatek.com>
562306a36Sopenharmony_ci */
662306a36Sopenharmony_ci
762306a36Sopenharmony_ci#include <linux/clk-provider.h>
862306a36Sopenharmony_ci#include <linux/mfd/syscon.h>
962306a36Sopenharmony_ci#include <linux/module.h>
1062306a36Sopenharmony_ci#include <linux/printk.h>
1162306a36Sopenharmony_ci#include <linux/regmap.h>
1262306a36Sopenharmony_ci#include <linux/slab.h>
1362306a36Sopenharmony_ci#include <linux/types.h>
1462306a36Sopenharmony_ci
1562306a36Sopenharmony_ci#include "clk-gate.h"
1662306a36Sopenharmony_ci
1762306a36Sopenharmony_cistruct mtk_clk_gate {
1862306a36Sopenharmony_ci	struct clk_hw	hw;
1962306a36Sopenharmony_ci	struct regmap	*regmap;
2062306a36Sopenharmony_ci	int		set_ofs;
2162306a36Sopenharmony_ci	int		clr_ofs;
2262306a36Sopenharmony_ci	int		sta_ofs;
2362306a36Sopenharmony_ci	u8		bit;
2462306a36Sopenharmony_ci};
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_cistatic inline struct mtk_clk_gate *to_mtk_clk_gate(struct clk_hw *hw)
2762306a36Sopenharmony_ci{
2862306a36Sopenharmony_ci	return container_of(hw, struct mtk_clk_gate, hw);
2962306a36Sopenharmony_ci}
3062306a36Sopenharmony_ci
3162306a36Sopenharmony_cistatic u32 mtk_get_clockgating(struct clk_hw *hw)
3262306a36Sopenharmony_ci{
3362306a36Sopenharmony_ci	struct mtk_clk_gate *cg = to_mtk_clk_gate(hw);
3462306a36Sopenharmony_ci	u32 val;
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_ci	regmap_read(cg->regmap, cg->sta_ofs, &val);
3762306a36Sopenharmony_ci
3862306a36Sopenharmony_ci	return val & BIT(cg->bit);
3962306a36Sopenharmony_ci}
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_cistatic int mtk_cg_bit_is_cleared(struct clk_hw *hw)
4262306a36Sopenharmony_ci{
4362306a36Sopenharmony_ci	return mtk_get_clockgating(hw) == 0;
4462306a36Sopenharmony_ci}
4562306a36Sopenharmony_ci
4662306a36Sopenharmony_cistatic int mtk_cg_bit_is_set(struct clk_hw *hw)
4762306a36Sopenharmony_ci{
4862306a36Sopenharmony_ci	return mtk_get_clockgating(hw) != 0;
4962306a36Sopenharmony_ci}
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_cistatic void mtk_cg_set_bit(struct clk_hw *hw)
5262306a36Sopenharmony_ci{
5362306a36Sopenharmony_ci	struct mtk_clk_gate *cg = to_mtk_clk_gate(hw);
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci	regmap_write(cg->regmap, cg->set_ofs, BIT(cg->bit));
5662306a36Sopenharmony_ci}
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_cistatic void mtk_cg_clr_bit(struct clk_hw *hw)
5962306a36Sopenharmony_ci{
6062306a36Sopenharmony_ci	struct mtk_clk_gate *cg = to_mtk_clk_gate(hw);
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_ci	regmap_write(cg->regmap, cg->clr_ofs, BIT(cg->bit));
6362306a36Sopenharmony_ci}
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_cistatic void mtk_cg_set_bit_no_setclr(struct clk_hw *hw)
6662306a36Sopenharmony_ci{
6762306a36Sopenharmony_ci	struct mtk_clk_gate *cg = to_mtk_clk_gate(hw);
6862306a36Sopenharmony_ci
6962306a36Sopenharmony_ci	regmap_set_bits(cg->regmap, cg->sta_ofs, BIT(cg->bit));
7062306a36Sopenharmony_ci}
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_cistatic void mtk_cg_clr_bit_no_setclr(struct clk_hw *hw)
7362306a36Sopenharmony_ci{
7462306a36Sopenharmony_ci	struct mtk_clk_gate *cg = to_mtk_clk_gate(hw);
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ci	regmap_clear_bits(cg->regmap, cg->sta_ofs, BIT(cg->bit));
7762306a36Sopenharmony_ci}
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_cistatic int mtk_cg_enable(struct clk_hw *hw)
8062306a36Sopenharmony_ci{
8162306a36Sopenharmony_ci	mtk_cg_clr_bit(hw);
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_ci	return 0;
8462306a36Sopenharmony_ci}
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_cistatic void mtk_cg_disable(struct clk_hw *hw)
8762306a36Sopenharmony_ci{
8862306a36Sopenharmony_ci	mtk_cg_set_bit(hw);
8962306a36Sopenharmony_ci}
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_cistatic int mtk_cg_enable_inv(struct clk_hw *hw)
9262306a36Sopenharmony_ci{
9362306a36Sopenharmony_ci	mtk_cg_set_bit(hw);
9462306a36Sopenharmony_ci
9562306a36Sopenharmony_ci	return 0;
9662306a36Sopenharmony_ci}
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_cistatic void mtk_cg_disable_inv(struct clk_hw *hw)
9962306a36Sopenharmony_ci{
10062306a36Sopenharmony_ci	mtk_cg_clr_bit(hw);
10162306a36Sopenharmony_ci}
10262306a36Sopenharmony_ci
10362306a36Sopenharmony_cistatic int mtk_cg_enable_no_setclr(struct clk_hw *hw)
10462306a36Sopenharmony_ci{
10562306a36Sopenharmony_ci	mtk_cg_clr_bit_no_setclr(hw);
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_ci	return 0;
10862306a36Sopenharmony_ci}
10962306a36Sopenharmony_ci
11062306a36Sopenharmony_cistatic void mtk_cg_disable_no_setclr(struct clk_hw *hw)
11162306a36Sopenharmony_ci{
11262306a36Sopenharmony_ci	mtk_cg_set_bit_no_setclr(hw);
11362306a36Sopenharmony_ci}
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_cistatic int mtk_cg_enable_inv_no_setclr(struct clk_hw *hw)
11662306a36Sopenharmony_ci{
11762306a36Sopenharmony_ci	mtk_cg_set_bit_no_setclr(hw);
11862306a36Sopenharmony_ci
11962306a36Sopenharmony_ci	return 0;
12062306a36Sopenharmony_ci}
12162306a36Sopenharmony_ci
12262306a36Sopenharmony_cistatic void mtk_cg_disable_inv_no_setclr(struct clk_hw *hw)
12362306a36Sopenharmony_ci{
12462306a36Sopenharmony_ci	mtk_cg_clr_bit_no_setclr(hw);
12562306a36Sopenharmony_ci}
12662306a36Sopenharmony_ci
12762306a36Sopenharmony_ciconst struct clk_ops mtk_clk_gate_ops_setclr = {
12862306a36Sopenharmony_ci	.is_enabled	= mtk_cg_bit_is_cleared,
12962306a36Sopenharmony_ci	.enable		= mtk_cg_enable,
13062306a36Sopenharmony_ci	.disable	= mtk_cg_disable,
13162306a36Sopenharmony_ci};
13262306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(mtk_clk_gate_ops_setclr);
13362306a36Sopenharmony_ci
13462306a36Sopenharmony_ciconst struct clk_ops mtk_clk_gate_ops_setclr_inv = {
13562306a36Sopenharmony_ci	.is_enabled	= mtk_cg_bit_is_set,
13662306a36Sopenharmony_ci	.enable		= mtk_cg_enable_inv,
13762306a36Sopenharmony_ci	.disable	= mtk_cg_disable_inv,
13862306a36Sopenharmony_ci};
13962306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(mtk_clk_gate_ops_setclr_inv);
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_ciconst struct clk_ops mtk_clk_gate_ops_no_setclr = {
14262306a36Sopenharmony_ci	.is_enabled	= mtk_cg_bit_is_cleared,
14362306a36Sopenharmony_ci	.enable		= mtk_cg_enable_no_setclr,
14462306a36Sopenharmony_ci	.disable	= mtk_cg_disable_no_setclr,
14562306a36Sopenharmony_ci};
14662306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(mtk_clk_gate_ops_no_setclr);
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ciconst struct clk_ops mtk_clk_gate_ops_no_setclr_inv = {
14962306a36Sopenharmony_ci	.is_enabled	= mtk_cg_bit_is_set,
15062306a36Sopenharmony_ci	.enable		= mtk_cg_enable_inv_no_setclr,
15162306a36Sopenharmony_ci	.disable	= mtk_cg_disable_inv_no_setclr,
15262306a36Sopenharmony_ci};
15362306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(mtk_clk_gate_ops_no_setclr_inv);
15462306a36Sopenharmony_ci
15562306a36Sopenharmony_cistatic struct clk_hw *mtk_clk_register_gate(struct device *dev, const char *name,
15662306a36Sopenharmony_ci					 const char *parent_name,
15762306a36Sopenharmony_ci					 struct regmap *regmap, int set_ofs,
15862306a36Sopenharmony_ci					 int clr_ofs, int sta_ofs, u8 bit,
15962306a36Sopenharmony_ci					 const struct clk_ops *ops,
16062306a36Sopenharmony_ci					 unsigned long flags)
16162306a36Sopenharmony_ci{
16262306a36Sopenharmony_ci	struct mtk_clk_gate *cg;
16362306a36Sopenharmony_ci	int ret;
16462306a36Sopenharmony_ci	struct clk_init_data init = {};
16562306a36Sopenharmony_ci
16662306a36Sopenharmony_ci	cg = kzalloc(sizeof(*cg), GFP_KERNEL);
16762306a36Sopenharmony_ci	if (!cg)
16862306a36Sopenharmony_ci		return ERR_PTR(-ENOMEM);
16962306a36Sopenharmony_ci
17062306a36Sopenharmony_ci	init.name = name;
17162306a36Sopenharmony_ci	init.flags = flags | CLK_SET_RATE_PARENT;
17262306a36Sopenharmony_ci	init.parent_names = parent_name ? &parent_name : NULL;
17362306a36Sopenharmony_ci	init.num_parents = parent_name ? 1 : 0;
17462306a36Sopenharmony_ci	init.ops = ops;
17562306a36Sopenharmony_ci
17662306a36Sopenharmony_ci	cg->regmap = regmap;
17762306a36Sopenharmony_ci	cg->set_ofs = set_ofs;
17862306a36Sopenharmony_ci	cg->clr_ofs = clr_ofs;
17962306a36Sopenharmony_ci	cg->sta_ofs = sta_ofs;
18062306a36Sopenharmony_ci	cg->bit = bit;
18162306a36Sopenharmony_ci
18262306a36Sopenharmony_ci	cg->hw.init = &init;
18362306a36Sopenharmony_ci
18462306a36Sopenharmony_ci	ret = clk_hw_register(dev, &cg->hw);
18562306a36Sopenharmony_ci	if (ret) {
18662306a36Sopenharmony_ci		kfree(cg);
18762306a36Sopenharmony_ci		return ERR_PTR(ret);
18862306a36Sopenharmony_ci	}
18962306a36Sopenharmony_ci
19062306a36Sopenharmony_ci	return &cg->hw;
19162306a36Sopenharmony_ci}
19262306a36Sopenharmony_ci
19362306a36Sopenharmony_cistatic void mtk_clk_unregister_gate(struct clk_hw *hw)
19462306a36Sopenharmony_ci{
19562306a36Sopenharmony_ci	struct mtk_clk_gate *cg;
19662306a36Sopenharmony_ci	if (!hw)
19762306a36Sopenharmony_ci		return;
19862306a36Sopenharmony_ci
19962306a36Sopenharmony_ci	cg = to_mtk_clk_gate(hw);
20062306a36Sopenharmony_ci
20162306a36Sopenharmony_ci	clk_hw_unregister(hw);
20262306a36Sopenharmony_ci	kfree(cg);
20362306a36Sopenharmony_ci}
20462306a36Sopenharmony_ci
20562306a36Sopenharmony_ciint mtk_clk_register_gates(struct device *dev, struct device_node *node,
20662306a36Sopenharmony_ci			   const struct mtk_gate *clks, int num,
20762306a36Sopenharmony_ci			   struct clk_hw_onecell_data *clk_data)
20862306a36Sopenharmony_ci{
20962306a36Sopenharmony_ci	int i;
21062306a36Sopenharmony_ci	struct clk_hw *hw;
21162306a36Sopenharmony_ci	struct regmap *regmap;
21262306a36Sopenharmony_ci
21362306a36Sopenharmony_ci	if (!clk_data)
21462306a36Sopenharmony_ci		return -ENOMEM;
21562306a36Sopenharmony_ci
21662306a36Sopenharmony_ci	regmap = device_node_to_regmap(node);
21762306a36Sopenharmony_ci	if (IS_ERR(regmap)) {
21862306a36Sopenharmony_ci		pr_err("Cannot find regmap for %pOF: %pe\n", node, regmap);
21962306a36Sopenharmony_ci		return PTR_ERR(regmap);
22062306a36Sopenharmony_ci	}
22162306a36Sopenharmony_ci
22262306a36Sopenharmony_ci	for (i = 0; i < num; i++) {
22362306a36Sopenharmony_ci		const struct mtk_gate *gate = &clks[i];
22462306a36Sopenharmony_ci
22562306a36Sopenharmony_ci		if (!IS_ERR_OR_NULL(clk_data->hws[gate->id])) {
22662306a36Sopenharmony_ci			pr_warn("%pOF: Trying to register duplicate clock ID: %d\n",
22762306a36Sopenharmony_ci				node, gate->id);
22862306a36Sopenharmony_ci			continue;
22962306a36Sopenharmony_ci		}
23062306a36Sopenharmony_ci
23162306a36Sopenharmony_ci		hw = mtk_clk_register_gate(dev, gate->name, gate->parent_name,
23262306a36Sopenharmony_ci					    regmap,
23362306a36Sopenharmony_ci					    gate->regs->set_ofs,
23462306a36Sopenharmony_ci					    gate->regs->clr_ofs,
23562306a36Sopenharmony_ci					    gate->regs->sta_ofs,
23662306a36Sopenharmony_ci					    gate->shift, gate->ops,
23762306a36Sopenharmony_ci					    gate->flags);
23862306a36Sopenharmony_ci
23962306a36Sopenharmony_ci		if (IS_ERR(hw)) {
24062306a36Sopenharmony_ci			pr_err("Failed to register clk %s: %pe\n", gate->name,
24162306a36Sopenharmony_ci			       hw);
24262306a36Sopenharmony_ci			goto err;
24362306a36Sopenharmony_ci		}
24462306a36Sopenharmony_ci
24562306a36Sopenharmony_ci		clk_data->hws[gate->id] = hw;
24662306a36Sopenharmony_ci	}
24762306a36Sopenharmony_ci
24862306a36Sopenharmony_ci	return 0;
24962306a36Sopenharmony_ci
25062306a36Sopenharmony_cierr:
25162306a36Sopenharmony_ci	while (--i >= 0) {
25262306a36Sopenharmony_ci		const struct mtk_gate *gate = &clks[i];
25362306a36Sopenharmony_ci
25462306a36Sopenharmony_ci		if (IS_ERR_OR_NULL(clk_data->hws[gate->id]))
25562306a36Sopenharmony_ci			continue;
25662306a36Sopenharmony_ci
25762306a36Sopenharmony_ci		mtk_clk_unregister_gate(clk_data->hws[gate->id]);
25862306a36Sopenharmony_ci		clk_data->hws[gate->id] = ERR_PTR(-ENOENT);
25962306a36Sopenharmony_ci	}
26062306a36Sopenharmony_ci
26162306a36Sopenharmony_ci	return PTR_ERR(hw);
26262306a36Sopenharmony_ci}
26362306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(mtk_clk_register_gates);
26462306a36Sopenharmony_ci
26562306a36Sopenharmony_civoid mtk_clk_unregister_gates(const struct mtk_gate *clks, int num,
26662306a36Sopenharmony_ci			      struct clk_hw_onecell_data *clk_data)
26762306a36Sopenharmony_ci{
26862306a36Sopenharmony_ci	int i;
26962306a36Sopenharmony_ci
27062306a36Sopenharmony_ci	if (!clk_data)
27162306a36Sopenharmony_ci		return;
27262306a36Sopenharmony_ci
27362306a36Sopenharmony_ci	for (i = num; i > 0; i--) {
27462306a36Sopenharmony_ci		const struct mtk_gate *gate = &clks[i - 1];
27562306a36Sopenharmony_ci
27662306a36Sopenharmony_ci		if (IS_ERR_OR_NULL(clk_data->hws[gate->id]))
27762306a36Sopenharmony_ci			continue;
27862306a36Sopenharmony_ci
27962306a36Sopenharmony_ci		mtk_clk_unregister_gate(clk_data->hws[gate->id]);
28062306a36Sopenharmony_ci		clk_data->hws[gate->id] = ERR_PTR(-ENOENT);
28162306a36Sopenharmony_ci	}
28262306a36Sopenharmony_ci}
28362306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(mtk_clk_unregister_gates);
28462306a36Sopenharmony_ci
28562306a36Sopenharmony_ciMODULE_LICENSE("GPL");
286