1// SPDX-License-Identifier: GPL-2.0-or-later
2/*
3 * Copyright (C) 2013 Freescale Semiconductor, Inc.
4 */
5
6#include <linux/bits.h>
7#include <linux/clk-provider.h>
8#include <linux/err.h>
9#include <linux/io.h>
10#include <linux/slab.h>
11#include "clk.h"
12
13/**
14 * struct clk_fixup_mux - imx integer fixup multiplexer clock
15 * @mux: the parent class
16 * @ops: pointer to clk_ops of parent class
17 * @fixup: a hook to fixup the write value
18 *
19 * The imx fixup multiplexer clock is a subclass of basic clk_mux
20 * with an addtional fixup hook.
21 */
22struct clk_fixup_mux {
23	struct clk_mux mux;
24	const struct clk_ops *ops;
25	void (*fixup)(u32 *val);
26};
27
28static inline struct clk_fixup_mux *to_clk_fixup_mux(struct clk_hw *hw)
29{
30	struct clk_mux *mux = to_clk_mux(hw);
31
32	return container_of(mux, struct clk_fixup_mux, mux);
33}
34
35static u8 clk_fixup_mux_get_parent(struct clk_hw *hw)
36{
37	struct clk_fixup_mux *fixup_mux = to_clk_fixup_mux(hw);
38
39	return fixup_mux->ops->get_parent(&fixup_mux->mux.hw);
40}
41
42static int clk_fixup_mux_set_parent(struct clk_hw *hw, u8 index)
43{
44	struct clk_fixup_mux *fixup_mux = to_clk_fixup_mux(hw);
45	struct clk_mux *mux = to_clk_mux(hw);
46	unsigned long flags;
47	u32 val;
48
49	spin_lock_irqsave(mux->lock, flags);
50
51	val = readl(mux->reg);
52	val &= ~(mux->mask << mux->shift);
53	val |= index << mux->shift;
54	fixup_mux->fixup(&val);
55	writel(val, mux->reg);
56
57	spin_unlock_irqrestore(mux->lock, flags);
58
59	return 0;
60}
61
62static const struct clk_ops clk_fixup_mux_ops = {
63	.determine_rate = clk_hw_determine_rate_no_reparent,
64	.get_parent = clk_fixup_mux_get_parent,
65	.set_parent = clk_fixup_mux_set_parent,
66};
67
68struct clk_hw *imx_clk_hw_fixup_mux(const char *name, void __iomem *reg,
69			      u8 shift, u8 width, const char * const *parents,
70			      int num_parents, void (*fixup)(u32 *val))
71{
72	struct clk_fixup_mux *fixup_mux;
73	struct clk_hw *hw;
74	struct clk_init_data init;
75	int ret;
76
77	if (!fixup)
78		return ERR_PTR(-EINVAL);
79
80	fixup_mux = kzalloc(sizeof(*fixup_mux), GFP_KERNEL);
81	if (!fixup_mux)
82		return ERR_PTR(-ENOMEM);
83
84	init.name = name;
85	init.ops = &clk_fixup_mux_ops;
86	init.parent_names = parents;
87	init.num_parents = num_parents;
88	init.flags = 0;
89
90	fixup_mux->mux.reg = reg;
91	fixup_mux->mux.shift = shift;
92	fixup_mux->mux.mask = BIT(width) - 1;
93	fixup_mux->mux.lock = &imx_ccm_lock;
94	fixup_mux->mux.hw.init = &init;
95	fixup_mux->ops = &clk_mux_ops;
96	fixup_mux->fixup = fixup;
97
98	hw = &fixup_mux->mux.hw;
99
100	ret = clk_hw_register(NULL, hw);
101	if (ret) {
102		kfree(fixup_mux);
103		return ERR_PTR(ret);
104	}
105
106	return hw;
107}
108