|
| 1 | +/* |
| 2 | + * Copyright (c) 2023-2024 Chen Xingyu <[email protected]> |
| 3 | + * SPDX-License-Identifier: Apache-2.0 |
| 4 | + */ |
| 5 | + |
| 6 | +#define DT_DRV_COMPAT sophgo_cvi_pwm |
| 7 | + |
| 8 | +#include <zephyr/arch/cpu.h> |
| 9 | +#include <zephyr/device.h> |
| 10 | +#include <zephyr/drivers/pwm.h> |
| 11 | +#if defined(CONFIG_PINCTRL) |
| 12 | +#include <zephyr/drivers/pinctrl.h> |
| 13 | +#endif /* CONFIG_PINCTRL */ |
| 14 | + |
| 15 | +#define HLPERIOD(base, n) (base + 0x000 + (n) * 8) |
| 16 | +#define PERIOD(base, n) (base + 0x004 + (n) * 8) |
| 17 | +#define PWMCONFIG(base) (base + 0x040) |
| 18 | +#define PWMSTART(base) (base + 0x044) |
| 19 | +#define PWMDONE(base) (base + 0x048) |
| 20 | +#define PWMUPDATE(base) (base + 0x04c) |
| 21 | +#define PCOUNT(base, n) (base + 0x050 + (n) * 4) |
| 22 | +#define PULSECOUNT(base, n) (base + 0x060 + (n) * 4) |
| 23 | +#define SHIFTCOUNT(base, n) (base + 0x080 + (n) * 4) |
| 24 | +#define SHIFTSTART(base) (base + 0x090) |
| 25 | +#define PWM_OE(base) (base + 0x0d0) |
| 26 | + |
| 27 | +/* PWMCONFIG */ |
| 28 | +#define CFG_POLARITY(n) BIT(n + 0) |
| 29 | +#define CFG_PWMMODE(n) BIT(n + 8) |
| 30 | +#define CFG_SHIFTMODE BIT(16) |
| 31 | + |
| 32 | +#define PWM_CH_MAX 4 |
| 33 | + |
| 34 | +struct pwm_cvi_config { |
| 35 | + mm_reg_t base; |
| 36 | + uint32_t clk_pwm; |
| 37 | +#if defined(CONFIG_PINCTRL) |
| 38 | + const struct pinctrl_dev_config *pcfg; |
| 39 | +#endif /* CONFIG_PINCTRL */ |
| 40 | +}; |
| 41 | + |
| 42 | +static int pwm_cvi_set_cycles(const struct device *dev, uint32_t channel, uint32_t period_cycles, |
| 43 | + uint32_t pulse_cycles, pwm_flags_t flags) |
| 44 | +{ |
| 45 | + const struct pwm_cvi_config *cfg = dev->config; |
| 46 | + uint32_t regval; |
| 47 | + |
| 48 | + if (channel > PWM_CH_MAX) { |
| 49 | + return -EINVAL; |
| 50 | + } |
| 51 | + |
| 52 | + if (period_cycles > cfg->clk_pwm) { |
| 53 | + return -EINVAL; |
| 54 | + } |
| 55 | + |
| 56 | + if (pulse_cycles >= period_cycles) { |
| 57 | + pulse_cycles = period_cycles - 1; |
| 58 | + } |
| 59 | + |
| 60 | + /* Configure output */ |
| 61 | + regval = sys_read32(PWM_OE(cfg->base)); |
| 62 | + regval |= BIT(channel); |
| 63 | + sys_write32(regval, PWM_OE(cfg->base)); |
| 64 | + |
| 65 | + /* Set polarity and mode */ |
| 66 | + regval = sys_read32(PWMCONFIG(cfg->base)); |
| 67 | + if (flags & PWM_POLARITY_INVERTED) { |
| 68 | + regval &= ~CFG_POLARITY(channel); /* active low */ |
| 69 | + } else { |
| 70 | + regval |= CFG_POLARITY(channel); /* active high */ |
| 71 | + } |
| 72 | + regval &= ~CFG_PWMMODE(channel); /* continuous mode */ |
| 73 | + sys_write32(regval, PWMCONFIG(cfg->base)); |
| 74 | + |
| 75 | + /* Set period and pulse */ |
| 76 | + sys_write32(period_cycles, PERIOD(cfg->base, channel)); |
| 77 | + sys_write32(pulse_cycles, HLPERIOD(cfg->base, channel)); |
| 78 | + |
| 79 | + if (sys_read32(PWMSTART(cfg->base)) & BIT(channel)) { |
| 80 | + /* Update channel */ |
| 81 | + regval = sys_read32(PWMUPDATE(cfg->base)); |
| 82 | + regval |= BIT(channel); |
| 83 | + sys_write32(regval, PWMUPDATE(cfg->base)); |
| 84 | + regval &= ~BIT(channel); |
| 85 | + sys_write32(regval, PWMUPDATE(cfg->base)); |
| 86 | + } else { |
| 87 | + /* Start channel */ |
| 88 | + regval = sys_read32(PWMSTART(cfg->base)); |
| 89 | + regval &= ~BIT(channel); |
| 90 | + sys_write32(regval, PWMSTART(cfg->base)); |
| 91 | + regval |= BIT(channel); |
| 92 | + sys_write32(regval, PWMSTART(cfg->base)); |
| 93 | + } |
| 94 | + |
| 95 | + return 0; |
| 96 | +} |
| 97 | + |
| 98 | +static int pwm_cvi_get_cycles_per_sec(const struct device *dev, uint32_t channel, uint64_t *cycles) |
| 99 | +{ |
| 100 | + const struct pwm_cvi_config *cfg = dev->config; |
| 101 | + |
| 102 | + if (channel > PWM_CH_MAX) { |
| 103 | + return -EINVAL; |
| 104 | + } |
| 105 | + |
| 106 | + *cycles = cfg->clk_pwm; |
| 107 | + |
| 108 | + return 0; |
| 109 | +} |
| 110 | + |
| 111 | +static int pwm_cvi_init(const struct device *dev) |
| 112 | +{ |
| 113 | +#if defined(CONFIG_PINCTRL) |
| 114 | + const struct pwm_cvi_config *cfg = dev->config; |
| 115 | + int ret; |
| 116 | + |
| 117 | + ret = pinctrl_apply_state(cfg->pcfg, PINCTRL_STATE_DEFAULT); |
| 118 | + if (ret < 0) { |
| 119 | + return ret; |
| 120 | + } |
| 121 | +#endif /* CONFIG_PINCTRL */ |
| 122 | + |
| 123 | + return 0; |
| 124 | +} |
| 125 | + |
| 126 | +static DEVICE_API(pwm, pwm_cvi_api) = { |
| 127 | + .set_cycles = pwm_cvi_set_cycles, |
| 128 | + .get_cycles_per_sec = pwm_cvi_get_cycles_per_sec, |
| 129 | +}; |
| 130 | + |
| 131 | +#define PWM_CVI_INST(n) \ |
| 132 | + IF_ENABLED(CONFIG_PINCTRL, (PINCTRL_DT_INST_DEFINE(n);)) \ |
| 133 | + \ |
| 134 | + static const struct pwm_cvi_config pwm_cvi_cfg_##n = { \ |
| 135 | + .base = DT_INST_REG_ADDR(n), \ |
| 136 | + .clk_pwm = DT_INST_PROP(n, clock_frequency), \ |
| 137 | + IF_ENABLED(CONFIG_PINCTRL, (.pcfg = PINCTRL_DT_INST_DEV_CONFIG_GET(n),)) \ |
| 138 | + }; \ |
| 139 | + \ |
| 140 | + DEVICE_DT_INST_DEFINE(n, &pwm_cvi_init, NULL, NULL, &pwm_cvi_cfg_##n, PRE_KERNEL_1, \ |
| 141 | + CONFIG_PWM_INIT_PRIORITY, &pwm_cvi_api); |
| 142 | + |
| 143 | +DT_INST_FOREACH_STATUS_OKAY(PWM_CVI_INST) |
0 commit comments