1 // SPDX-License-Identifier: GPL-2.0-only
3 * Driver for Allwinner sun4i Pulse Width Modulation Controller
8 #include <linux/bitops.h>
10 #include <linux/delay.h>
11 #include <linux/err.h>
13 #include <linux/jiffies.h>
14 #include <linux/module.h>
16 #include <linux/of_device.h>
17 #include <linux/platform_device.h>
18 #include <linux/pwm.h>
19 #include <linux/slab.h>
20 #include <linux/spinlock.h>
21 #include <linux/time.h>
23 #define PWM_CTRL_REG 0x0
25 #define PWM_CH_PRD_BASE 0x4
26 #define PWM_CH_PRD_OFFSET 0x4
27 #define PWM_CH_PRD(ch) (PWM_CH_PRD_BASE + PWM_CH_PRD_OFFSET * (ch))
29 #define PWMCH_OFFSET 15
30 #define PWM_PRESCAL_MASK GENMASK(3, 0)
31 #define PWM_PRESCAL_OFF 0
33 #define PWM_ACT_STATE BIT(5)
34 #define PWM_CLK_GATING BIT(6)
35 #define PWM_MODE BIT(7)
36 #define PWM_PULSE BIT(8)
37 #define PWM_BYPASS BIT(9)
39 #define PWM_RDY_BASE 28
40 #define PWM_RDY_OFFSET 1
41 #define PWM_RDY(ch) BIT(PWM_RDY_BASE + PWM_RDY_OFFSET * (ch))
43 #define PWM_PRD(prd) (((prd) - 1) << 16)
44 #define PWM_PRD_MASK GENMASK(15, 0)
46 #define PWM_DTY_MASK GENMASK(15, 0)
48 #define PWM_REG_PRD(reg) ((((reg) >> 16) & PWM_PRD_MASK) + 1)
49 #define PWM_REG_DTY(reg) ((reg) & PWM_DTY_MASK)
50 #define PWM_REG_PRESCAL(reg, chan) (((reg) >> ((chan) * PWMCH_OFFSET)) & PWM_PRESCAL_MASK)
52 #define BIT_CH(bit, chan) ((bit) << ((chan) * PWMCH_OFFSET))
54 static const u32 prescaler_table[] = {
70 0, /* Actually 1 but tested separately */
73 struct sun4i_pwm_data {
74 bool has_prescaler_bypass;
78 struct sun4i_pwm_chip {
83 const struct sun4i_pwm_data *data;
84 unsigned long next_period[2];
88 static inline struct sun4i_pwm_chip *to_sun4i_pwm_chip(struct pwm_chip *chip)
90 return container_of(chip, struct sun4i_pwm_chip, chip);
93 static inline u32 sun4i_pwm_readl(struct sun4i_pwm_chip *chip,
96 return readl(chip->base + offset);
99 static inline void sun4i_pwm_writel(struct sun4i_pwm_chip *chip,
100 u32 val, unsigned long offset)
102 writel(val, chip->base + offset);
105 static void sun4i_pwm_get_state(struct pwm_chip *chip,
106 struct pwm_device *pwm,
107 struct pwm_state *state)
109 struct sun4i_pwm_chip *sun4i_pwm = to_sun4i_pwm_chip(chip);
112 unsigned int prescaler;
114 clk_rate = clk_get_rate(sun4i_pwm->clk);
116 val = sun4i_pwm_readl(sun4i_pwm, PWM_CTRL_REG);
118 if ((PWM_REG_PRESCAL(val, pwm->hwpwm) == PWM_PRESCAL_MASK) &&
119 sun4i_pwm->data->has_prescaler_bypass)
122 prescaler = prescaler_table[PWM_REG_PRESCAL(val, pwm->hwpwm)];
127 if (val & BIT_CH(PWM_ACT_STATE, pwm->hwpwm))
128 state->polarity = PWM_POLARITY_NORMAL;
130 state->polarity = PWM_POLARITY_INVERSED;
132 if ((val & BIT_CH(PWM_CLK_GATING | PWM_EN, pwm->hwpwm)) ==
133 BIT_CH(PWM_CLK_GATING | PWM_EN, pwm->hwpwm))
134 state->enabled = true;
136 state->enabled = false;
138 val = sun4i_pwm_readl(sun4i_pwm, PWM_CH_PRD(pwm->hwpwm));
140 tmp = prescaler * NSEC_PER_SEC * PWM_REG_DTY(val);
141 state->duty_cycle = DIV_ROUND_CLOSEST_ULL(tmp, clk_rate);
143 tmp = prescaler * NSEC_PER_SEC * PWM_REG_PRD(val);
144 state->period = DIV_ROUND_CLOSEST_ULL(tmp, clk_rate);
147 static int sun4i_pwm_calculate(struct sun4i_pwm_chip *sun4i_pwm,
148 const struct pwm_state *state,
149 u32 *dty, u32 *prd, unsigned int *prsclr)
151 u64 clk_rate, div = 0;
152 unsigned int pval, prescaler = 0;
154 clk_rate = clk_get_rate(sun4i_pwm->clk);
156 if (sun4i_pwm->data->has_prescaler_bypass) {
157 /* First, test without any prescaler when available */
158 prescaler = PWM_PRESCAL_MASK;
161 * When not using any prescaler, the clock period in nanoseconds
162 * is not an integer so round it half up instead of
163 * truncating to get less surprising values.
165 div = clk_rate * state->period + NSEC_PER_SEC / 2;
166 do_div(div, NSEC_PER_SEC);
167 if (div - 1 > PWM_PRD_MASK)
171 if (prescaler == 0) {
172 /* Go up from the first divider */
173 for (prescaler = 0; prescaler < PWM_PRESCAL_MASK; prescaler++) {
174 if (!prescaler_table[prescaler])
176 pval = prescaler_table[prescaler];
179 div = div * state->period;
180 do_div(div, NSEC_PER_SEC);
181 if (div - 1 <= PWM_PRD_MASK)
185 if (div - 1 > PWM_PRD_MASK)
190 div *= state->duty_cycle;
191 do_div(div, state->period);
198 static int sun4i_pwm_apply(struct pwm_chip *chip, struct pwm_device *pwm,
199 const struct pwm_state *state)
201 struct sun4i_pwm_chip *sun4i_pwm = to_sun4i_pwm_chip(chip);
202 struct pwm_state cstate;
205 unsigned int delay_us;
208 pwm_get_state(pwm, &cstate);
210 if (!cstate.enabled) {
211 ret = clk_prepare_enable(sun4i_pwm->clk);
213 dev_err(chip->dev, "failed to enable PWM clock\n");
218 spin_lock(&sun4i_pwm->ctrl_lock);
219 ctrl = sun4i_pwm_readl(sun4i_pwm, PWM_CTRL_REG);
221 if ((cstate.period != state->period) ||
222 (cstate.duty_cycle != state->duty_cycle)) {
223 u32 period, duty, val;
224 unsigned int prescaler;
226 ret = sun4i_pwm_calculate(sun4i_pwm, state,
227 &duty, &period, &prescaler);
229 dev_err(chip->dev, "period exceeds the maximum value\n");
230 spin_unlock(&sun4i_pwm->ctrl_lock);
232 clk_disable_unprepare(sun4i_pwm->clk);
236 if (PWM_REG_PRESCAL(ctrl, pwm->hwpwm) != prescaler) {
237 /* Prescaler changed, the clock has to be gated */
238 ctrl &= ~BIT_CH(PWM_CLK_GATING, pwm->hwpwm);
239 sun4i_pwm_writel(sun4i_pwm, ctrl, PWM_CTRL_REG);
241 ctrl &= ~BIT_CH(PWM_PRESCAL_MASK, pwm->hwpwm);
242 ctrl |= BIT_CH(prescaler, pwm->hwpwm);
245 val = (duty & PWM_DTY_MASK) | PWM_PRD(period);
246 sun4i_pwm_writel(sun4i_pwm, val, PWM_CH_PRD(pwm->hwpwm));
247 sun4i_pwm->next_period[pwm->hwpwm] = jiffies +
248 usecs_to_jiffies(cstate.period / 1000 + 1);
249 sun4i_pwm->needs_delay[pwm->hwpwm] = true;
252 if (state->polarity != PWM_POLARITY_NORMAL)
253 ctrl &= ~BIT_CH(PWM_ACT_STATE, pwm->hwpwm);
255 ctrl |= BIT_CH(PWM_ACT_STATE, pwm->hwpwm);
257 ctrl |= BIT_CH(PWM_CLK_GATING, pwm->hwpwm);
258 if (state->enabled) {
259 ctrl |= BIT_CH(PWM_EN, pwm->hwpwm);
260 } else if (!sun4i_pwm->needs_delay[pwm->hwpwm]) {
261 ctrl &= ~BIT_CH(PWM_EN, pwm->hwpwm);
262 ctrl &= ~BIT_CH(PWM_CLK_GATING, pwm->hwpwm);
265 sun4i_pwm_writel(sun4i_pwm, ctrl, PWM_CTRL_REG);
267 spin_unlock(&sun4i_pwm->ctrl_lock);
272 if (!sun4i_pwm->needs_delay[pwm->hwpwm]) {
273 clk_disable_unprepare(sun4i_pwm->clk);
277 /* We need a full period to elapse before disabling the channel. */
279 if (sun4i_pwm->needs_delay[pwm->hwpwm] &&
280 time_before(now, sun4i_pwm->next_period[pwm->hwpwm])) {
281 delay_us = jiffies_to_usecs(sun4i_pwm->next_period[pwm->hwpwm] -
283 if ((delay_us / 500) > MAX_UDELAY_MS)
284 msleep(delay_us / 1000 + 1);
286 usleep_range(delay_us, delay_us * 2);
288 sun4i_pwm->needs_delay[pwm->hwpwm] = false;
290 spin_lock(&sun4i_pwm->ctrl_lock);
291 ctrl = sun4i_pwm_readl(sun4i_pwm, PWM_CTRL_REG);
292 ctrl &= ~BIT_CH(PWM_CLK_GATING, pwm->hwpwm);
293 ctrl &= ~BIT_CH(PWM_EN, pwm->hwpwm);
294 sun4i_pwm_writel(sun4i_pwm, ctrl, PWM_CTRL_REG);
295 spin_unlock(&sun4i_pwm->ctrl_lock);
297 clk_disable_unprepare(sun4i_pwm->clk);
302 static const struct pwm_ops sun4i_pwm_ops = {
303 .apply = sun4i_pwm_apply,
304 .get_state = sun4i_pwm_get_state,
305 .owner = THIS_MODULE,
308 static const struct sun4i_pwm_data sun4i_pwm_dual_nobypass = {
309 .has_prescaler_bypass = false,
313 static const struct sun4i_pwm_data sun4i_pwm_dual_bypass = {
314 .has_prescaler_bypass = true,
318 static const struct sun4i_pwm_data sun4i_pwm_single_bypass = {
319 .has_prescaler_bypass = true,
323 static const struct of_device_id sun4i_pwm_dt_ids[] = {
325 .compatible = "allwinner,sun4i-a10-pwm",
326 .data = &sun4i_pwm_dual_nobypass,
328 .compatible = "allwinner,sun5i-a10s-pwm",
329 .data = &sun4i_pwm_dual_bypass,
331 .compatible = "allwinner,sun5i-a13-pwm",
332 .data = &sun4i_pwm_single_bypass,
334 .compatible = "allwinner,sun7i-a20-pwm",
335 .data = &sun4i_pwm_dual_bypass,
337 .compatible = "allwinner,sun8i-h3-pwm",
338 .data = &sun4i_pwm_single_bypass,
343 MODULE_DEVICE_TABLE(of, sun4i_pwm_dt_ids);
345 static int sun4i_pwm_probe(struct platform_device *pdev)
347 struct sun4i_pwm_chip *pwm;
348 struct resource *res;
351 pwm = devm_kzalloc(&pdev->dev, sizeof(*pwm), GFP_KERNEL);
355 pwm->data = of_device_get_match_data(&pdev->dev);
359 res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
360 pwm->base = devm_ioremap_resource(&pdev->dev, res);
361 if (IS_ERR(pwm->base))
362 return PTR_ERR(pwm->base);
364 pwm->clk = devm_clk_get(&pdev->dev, NULL);
365 if (IS_ERR(pwm->clk))
366 return PTR_ERR(pwm->clk);
368 pwm->chip.dev = &pdev->dev;
369 pwm->chip.ops = &sun4i_pwm_ops;
371 pwm->chip.npwm = pwm->data->npwm;
372 pwm->chip.of_xlate = of_pwm_xlate_with_flags;
373 pwm->chip.of_pwm_n_cells = 3;
375 spin_lock_init(&pwm->ctrl_lock);
377 ret = pwmchip_add(&pwm->chip);
379 dev_err(&pdev->dev, "failed to add PWM chip: %d\n", ret);
383 platform_set_drvdata(pdev, pwm);
388 static int sun4i_pwm_remove(struct platform_device *pdev)
390 struct sun4i_pwm_chip *pwm = platform_get_drvdata(pdev);
392 return pwmchip_remove(&pwm->chip);
395 static struct platform_driver sun4i_pwm_driver = {
398 .of_match_table = sun4i_pwm_dt_ids,
400 .probe = sun4i_pwm_probe,
401 .remove = sun4i_pwm_remove,
403 module_platform_driver(sun4i_pwm_driver);
405 MODULE_ALIAS("platform:sun4i-pwm");
407 MODULE_DESCRIPTION("Allwinner sun4i PWM driver");
408 MODULE_LICENSE("GPL v2");