1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause
3  *
4  * Copyright 2016 Michal Meloun <mmel@FreeBSD.org>
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26  * SUCH DAMAGE.
27  */
28 
29 #include <sys/cdefs.h>
30 __FBSDID("$FreeBSD$");
31 
32 #include <sys/param.h>
33 #include <sys/conf.h>
34 #include <sys/bus.h>
35 #include <sys/kernel.h>
36 #include <sys/systm.h>
37 
38 #include <machine/bus.h>
39 
40 #include <dev/extres/clk/clk.h>
41 
42 #include <arm64/freescale/imx/clk/imx_clk_mux.h>
43 
44 #include "clkdev_if.h"
45 
46 #define	WR4(_clk, off, val)						\
47 	CLKDEV_WRITE_4(clknode_get_device(_clk), off, val)
48 #define	RD4(_clk, off, val)						\
49 	CLKDEV_READ_4(clknode_get_device(_clk), off, val)
50 #define	MD4(_clk, off, clr, set )					\
51 	CLKDEV_MODIFY_4(clknode_get_device(_clk), off, clr, set)
52 #define	DEVICE_LOCK(_clk)						\
53 	CLKDEV_DEVICE_LOCK(clknode_get_device(_clk))
54 #define	DEVICE_UNLOCK(_clk)						\
55 	CLKDEV_DEVICE_UNLOCK(clknode_get_device(_clk))
56 
57 static int imx_clk_mux_init(struct clknode *clk, device_t dev);
58 static int imx_clk_mux_set_mux(struct clknode *clk, int idx);
59 
60 struct imx_clk_mux_sc {
61 	uint32_t	offset;
62 	uint32_t	shift;
63 	uint32_t	mask;
64 	int		mux_flags;
65 };
66 
67 static clknode_method_t imx_clk_mux_methods[] = {
68 	/* Device interface */
69 	CLKNODEMETHOD(clknode_init, 	imx_clk_mux_init),
70 	CLKNODEMETHOD(clknode_set_mux, 	imx_clk_mux_set_mux),
71 	CLKNODEMETHOD_END
72 };
73 DEFINE_CLASS_1(imx_clk_mux, imx_clk_mux_class, imx_clk_mux_methods,
74    sizeof(struct imx_clk_mux_sc), clknode_class);
75 
76 static int
77 imx_clk_mux_init(struct clknode *clk, device_t dev)
78 {
79 	uint32_t reg;
80 	struct imx_clk_mux_sc *sc;
81 	int rv;
82 
83 	sc = clknode_get_softc(clk);
84 
85 	DEVICE_LOCK(clk);
86 	rv = RD4(clk, sc->offset, &reg);
87 	DEVICE_UNLOCK(clk);
88 	if (rv != 0) {
89 		return (rv);
90 	}
91 	reg = (reg >> sc->shift) & sc->mask;
92 	clknode_init_parent_idx(clk, reg);
93 	return(0);
94 }
95 
96 static int
97 imx_clk_mux_set_mux(struct clknode *clk, int idx)
98 {
99 	uint32_t reg;
100 	struct imx_clk_mux_sc *sc;
101 	int rv;
102 
103 	sc = clknode_get_softc(clk);
104 
105 	DEVICE_LOCK(clk);
106 	rv = MD4(clk, sc->offset, sc->mask << sc->shift,
107 	    ((idx & sc->mask) << sc->shift));
108 	if (rv != 0) {
109 		DEVICE_UNLOCK(clk);
110 		return (rv);
111 	}
112 	RD4(clk, sc->offset, &reg);
113 	DEVICE_UNLOCK(clk);
114 
115 	return(0);
116 }
117 
118 int
119 imx_clk_mux_register(struct clkdom *clkdom, struct imx_clk_mux_def *clkdef)
120 {
121 	struct clknode *clk;
122 	struct imx_clk_mux_sc *sc;
123 
124 	clk = clknode_create(clkdom, &imx_clk_mux_class, &clkdef->clkdef);
125 	if (clk == NULL)
126 		return (1);
127 
128 	sc = clknode_get_softc(clk);
129 	sc->offset = clkdef->offset;
130 	sc->shift = clkdef->shift;
131 	sc->mask =  (1 << clkdef->width) - 1;
132 	sc->mux_flags = clkdef->mux_flags;
133 
134 	clknode_register(clkdom, clk);
135 	return (0);
136 }
137