blob: 2a97aa47dfdc2477046e2ab19e514fddb8658926 [file] [log] [blame]
/*
* Copyright (c) 2018, Cue Health Inc
*
* SPDX-License-Identifier: Apache-2.0
*/
#define DT_DRV_COMPAT nordic_nrf_pwm
#include <nrfx_pwm.h>
#include <zephyr/drivers/pwm.h>
#include <zephyr/pm/device.h>
#include <zephyr/drivers/pinctrl.h>
#include <soc.h>
#include <hal/nrf_gpio.h>
#include <stdbool.h>
#include <zephyr/linker/devicetree_regions.h>
#include <zephyr/cache.h>
#include <zephyr/mem_mgmt/mem_attr.h>
#include <zephyr/logging/log.h>
LOG_MODULE_REGISTER(pwm_nrfx, CONFIG_PWM_LOG_LEVEL);
#define PWM_NRFX_CH_POLARITY_MASK BIT(15)
#define PWM_NRFX_CH_COMPARE_MASK BIT_MASK(15)
#define PWM_NRFX_CH_VALUE(compare_value, inverted) \
(compare_value | (inverted ? 0 : PWM_NRFX_CH_POLARITY_MASK))
struct pwm_nrfx_config {
nrfx_pwm_config_t initial_config;
nrf_pwm_sequence_t seq;
const struct pinctrl_dev_config *pcfg;
uint32_t clock_freq;
#ifdef CONFIG_DCACHE
uint32_t mem_attr;
#endif
};
struct pwm_nrfx_data {
nrfx_pwm_t pwm;
uint32_t period_cycles;
/* Bit mask indicating channels that need the PWM generation. */
uint8_t pwm_needed;
uint8_t prescaler;
bool stop_requested;
};
#if NRF_ERRATA_STATIC_CHECK(52, 109)
/* Forward-declare pwm_nrfx_<inst>_data structs to be able to access nrfx_pwm_t needed for the
* workaround.
*/
#define _PWM_DATA_STRUCT_NAME_GET(inst) pwm_nrfx_##inst##_data
#define _PWM_DATA_STRUCT_DECLARE(inst) static struct pwm_nrfx_data _PWM_DATA_STRUCT_NAME_GET(inst);
DT_INST_FOREACH_STATUS_OKAY(_PWM_DATA_STRUCT_DECLARE);
/* Create an array of pointers to all active PWM instances to loop over them in an EGU interrupt
* handler.
*/
#define _PWM_DATA_STRUCT_PWM_PTR_COMMA_GET(inst) &_PWM_DATA_STRUCT_NAME_GET(inst).pwm,
static nrfx_pwm_t *pwm_instances[] = {
DT_INST_FOREACH_STATUS_OKAY(_PWM_DATA_STRUCT_PWM_PTR_COMMA_GET)
};
/* Define an interrupt handler for the EGU instance used by the workaround which calls
* nrfx_pwm_nrf52_anomaly_109_handler for all active PWM instances.
*/
void anomaly_109_egu_handler(void)
{
for (int i = 0; i < ARRAY_SIZE(pwm_instances); i++) {
nrfx_pwm_nrf52_anomaly_109_handler(pwm_instances[i]);
}
}
#define ANOMALY_109_EGU_IRQ_CONNECT(idx) _EGU_IRQ_CONNECT(idx)
#define _EGU_IRQ_CONNECT(idx) \
IRQ_CONNECT(DT_IRQN(DT_NODELABEL(egu##idx)), \
DT_IRQ(DT_NODELABEL(egu##idx), priority), \
anomaly_109_egu_handler, 0, 0)
#else
#define ANOMALY_109_EGU_IRQ_CONNECT(idx)
#endif
/* Ensure the pwm_needed bit mask can accommodate all available channels. */
#if (NRF_PWM_CHANNEL_COUNT > 8)
#error "Current implementation supports maximum 8 channels."
#endif
static uint16_t *seq_values_ptr_get(const struct device *dev)
{
const struct pwm_nrfx_config *config = dev->config;
return (uint16_t *)config->seq.values.p_raw;
}
static void pwm_handler(nrfx_pwm_event_type_t event, void *p_context)
{
ARG_UNUSED(event);
ARG_UNUSED(p_context);
}
static bool pwm_period_check_and_set(const struct device *dev,
uint32_t channel, uint32_t period_cycles)
{
const struct pwm_nrfx_config *config = dev->config;
struct pwm_nrfx_data *data = dev->data;
uint8_t prescaler;
uint32_t countertop;
/* If the currently configured period matches the requested one,
* nothing more needs to be done.
*/
if (period_cycles == data->period_cycles) {
return true;
}
/* If any other channel is driven by the PWM peripheral, the period
* that is currently set cannot be changed, as this would influence
* the output for that channel.
*/
if ((data->pwm_needed & ~BIT(channel)) != 0) {
LOG_ERR("Incompatible period.");
return false;
}
/* Try to find a prescaler that will allow setting the requested period
* after prescaling as the countertop value for the PWM peripheral.
*/
prescaler = 0;
countertop = period_cycles;
do {
if (countertop <= PWM_COUNTERTOP_COUNTERTOP_Msk) {
data->period_cycles = period_cycles;
data->prescaler = prescaler;
nrf_pwm_configure(data->pwm.p_reg,
data->prescaler,
config->initial_config.count_mode,
(uint16_t)countertop);
return true;
}
countertop >>= 1;
++prescaler;
} while (prescaler <= PWM_PRESCALER_PRESCALER_Msk);
LOG_ERR("Prescaler for period_cycles %u not found.", period_cycles);
return false;
}
static bool channel_psel_get(uint32_t channel, uint32_t *psel, struct pwm_nrfx_data *data)
{
*psel = nrf_pwm_pin_get(data->pwm.p_reg, (uint8_t)channel);
return (((*psel & PWM_PSEL_OUT_CONNECT_Msk) >> PWM_PSEL_OUT_CONNECT_Pos)
== PWM_PSEL_OUT_CONNECT_Connected);
}
static int stop_pwm(const struct device *dev)
{
struct pwm_nrfx_data *data = dev->data;
/* Don't wait here for the peripheral to actually stop. Instead,
* ensure it is stopped before starting the next playback.
*/
nrfx_pwm_stop(&data->pwm, false);
return 0;
}
static int pwm_nrfx_set_cycles(const struct device *dev, uint32_t channel,
uint32_t period_cycles, uint32_t pulse_cycles,
pwm_flags_t flags)
{
/* We assume here that period_cycles will always be 16MHz
* peripheral clock. Since pwm_nrfx_get_cycles_per_sec() function might
* be removed, see ISSUE #6958.
* TODO: Remove this comment when issue has been resolved.
*/
const struct pwm_nrfx_config *config = dev->config;
struct pwm_nrfx_data *data = dev->data;
uint16_t compare_value;
bool inverted = (flags & PWM_POLARITY_INVERTED);
bool needs_pwm = false;
if (channel >= NRF_PWM_CHANNEL_COUNT) {
LOG_ERR("Invalid channel: %u.", channel);
return -EINVAL;
}
/* If this PWM is in center-aligned mode, pulse and period lengths
* are effectively doubled by the up-down count, so halve them here
* to compensate.
*/
if (config->initial_config.count_mode == NRF_PWM_MODE_UP_AND_DOWN) {
period_cycles /= 2;
pulse_cycles /= 2;
}
if (pulse_cycles == 0) {
/* Constantly inactive (duty 0%). */
compare_value = 0;
} else if (pulse_cycles >= period_cycles) {
/* Constantly active (duty 100%). */
/* This value is always greater than or equal to COUNTERTOP. */
compare_value = PWM_NRFX_CH_COMPARE_MASK;
needs_pwm = IS_ENABLED(NRF_PWM_HAS_IDLEOUT) &&
IS_ENABLED(CONFIG_PWM_NRFX_NO_GLITCH_DUTY_100);
} else {
/* PWM generation needed. Check if the requested period matches
* the one that is currently set, or the PWM peripheral can be
* reconfigured accordingly.
*/
if (!pwm_period_check_and_set(dev, channel, period_cycles)) {
return -EINVAL;
}
compare_value = (uint16_t)(pulse_cycles >> data->prescaler);
needs_pwm = true;
}
seq_values_ptr_get(dev)[channel] = PWM_NRFX_CH_VALUE(compare_value, inverted);
#ifdef CONFIG_DCACHE
if (config->mem_attr & DT_MEM_CACHEABLE) {
sys_cache_data_flush_range(seq_values_ptr_get(dev), config->seq.length);
}
#endif
LOG_DBG("channel %u, pulse %u, period %u, prescaler: %u.",
channel, pulse_cycles, period_cycles, data->prescaler);
/* If this channel does not need to be driven by the PWM peripheral
* because its state is to be constant (duty 0% or 100%), set properly
* the GPIO configuration for its output pin. This will provide
* the correct output state for this channel when the PWM peripheral
* is stopped.
*/
if (!needs_pwm) {
uint32_t psel;
if (channel_psel_get(channel, &psel, data)) {
uint32_t out_level = (pulse_cycles == 0) ? 0 : 1;
if (inverted) {
out_level ^= 1;
}
nrf_gpio_pin_write(psel, out_level);
}
data->pwm_needed &= ~BIT(channel);
} else {
data->pwm_needed |= BIT(channel);
}
/* If the PWM generation is not needed for any channel (all are set
* to constant inactive or active state), stop the PWM peripheral.
* Otherwise, request a playback of the defined sequence so that
* the PWM peripheral loads `seq_values` into its internal compare
* registers and drives its outputs accordingly.
*/
if (data->pwm_needed == 0) {
int ret = stop_pwm(dev);
if (ret < 0) {
LOG_ERR("PWM stop failed: %d", ret);
return ret;
}
data->stop_requested = true;
} else {
if (data->stop_requested) {
data->stop_requested = false;
/* After a stop is requested, the PWM peripheral stops
* pulse generation at the end of the current period,
* and till that moment, it ignores any start requests,
* so ensure here that it is stopped.
*/
while (!nrfx_pwm_stopped_check(&data->pwm)) {
}
}
/* It is sufficient to play the sequence once without looping.
* The PWM generation will continue with the loaded values
* until another playback is requested (new values will be
* loaded then) or the PWM peripheral is stopped.
*/
nrfx_pwm_simple_playback(&data->pwm, &config->seq, 1,
NRFX_PWM_FLAG_NO_EVT_FINISHED);
}
return 0;
}
static int pwm_nrfx_get_cycles_per_sec(const struct device *dev, uint32_t channel,
uint64_t *cycles)
{
const struct pwm_nrfx_config *config = dev->config;
*cycles = config->clock_freq;
return 0;
}
static DEVICE_API(pwm, pwm_nrfx_drv_api_funcs) = {
.set_cycles = pwm_nrfx_set_cycles,
.get_cycles_per_sec = pwm_nrfx_get_cycles_per_sec,
};
static int pwm_resume(const struct device *dev)
{
const struct pwm_nrfx_config *config = dev->config;
struct pwm_nrfx_data *data = dev->data;
uint8_t initially_inverted = 0;
(void)pinctrl_apply_state(config->pcfg, PINCTRL_STATE_DEFAULT);
for (size_t i = 0; i < NRF_PWM_CHANNEL_COUNT; i++) {
uint32_t psel;
if (channel_psel_get(i, &psel, data)) {
/* Mark channels as inverted according to what initial
* state of their outputs has been set by pinctrl (high
* idle state means that the channel is inverted).
*/
initially_inverted |= nrf_gpio_pin_out_read(psel) ?
BIT(i) : 0;
}
}
for (size_t i = 0; i < NRF_PWM_CHANNEL_COUNT; i++) {
bool inverted = initially_inverted & BIT(i);
seq_values_ptr_get(dev)[i] = PWM_NRFX_CH_VALUE(0, inverted);
}
return 0;
}
static int pwm_suspend(const struct device *dev)
{
const struct pwm_nrfx_config *config = dev->config;
struct pwm_nrfx_data *data = dev->data;
int ret = stop_pwm(dev);
if (ret < 0) {
LOG_ERR("PWM stop failed: %d", ret);
return ret;
}
while (!nrfx_pwm_stopped_check(&data->pwm)) {
}
memset(dev->data, 0, sizeof(struct pwm_nrfx_data));
(void)pinctrl_apply_state(config->pcfg, PINCTRL_STATE_SLEEP);
return 0;
}
static int pwm_nrfx_pm_action(const struct device *dev,
enum pm_device_action action)
{
if (action == PM_DEVICE_ACTION_RESUME) {
return pwm_resume(dev);
} else if (IS_ENABLED(CONFIG_PM_DEVICE) && (action == PM_DEVICE_ACTION_SUSPEND)) {
return pwm_suspend(dev);
} else {
return -ENOTSUP;
}
return 0;
}
static int pwm_nrfx_init(const struct device *dev)
{
const struct pwm_nrfx_config *config = dev->config;
struct pwm_nrfx_data *data = dev->data;
int err;
ANOMALY_109_EGU_IRQ_CONNECT(NRFX_PWM_NRF52_ANOMALY_109_EGU_INSTANCE);
if (IS_ENABLED(CONFIG_PM_DEVICE_RUNTIME)) {
(void)pinctrl_apply_state(config->pcfg, PINCTRL_STATE_SLEEP);
}
err = nrfx_pwm_init(&data->pwm, &config->initial_config, pwm_handler, dev->data);
if (err < 0) {
LOG_ERR("Failed to initialize device: %s", dev->name);
return err;
}
return pm_device_driver_init(dev, pwm_nrfx_pm_action);
}
#define PWM_MEM_REGION(inst) DT_PHANDLE(DT_DRV_INST(inst), memory_regions)
#define PWM_MEMORY_SECTION(inst) \
COND_CODE_1(DT_NODE_HAS_PROP(DT_DRV_INST(inst), memory_regions), \
(__attribute__((__section__(LINKER_DT_NODE_REGION_NAME( \
PWM_MEM_REGION(inst)))))), \
())
#define PWM_GET_MEM_ATTR(inst) \
COND_CODE_1(DT_NODE_HAS_PROP(DT_DRV_INST(inst), memory_regions), \
(DT_PROP_OR(PWM_MEM_REGION(inst), zephyr_memory_attr, 0)), (0))
#define PWM_NRFX_DEFINE(inst) \
NRF_DT_CHECK_NODE_HAS_PINCTRL_SLEEP(DT_DRV_INST(inst)); \
NRF_DT_CHECK_NODE_HAS_REQUIRED_MEMORY_REGIONS(DT_DRV_INST(inst)); \
static struct pwm_nrfx_data pwm_nrfx_##inst##_data = { \
.pwm = NRFX_PWM_INSTANCE(DT_INST_REG_ADDR(inst)), \
}; \
static uint16_t pwm_##inst##_seq_values[NRF_PWM_CHANNEL_COUNT] \
PWM_MEMORY_SECTION(inst); \
PINCTRL_DT_INST_DEFINE(inst); \
static const struct pwm_nrfx_config pwm_nrfx_##inst##_config = { \
.initial_config = { \
.skip_gpio_cfg = true, \
.skip_psel_cfg = true, \
.base_clock = NRF_PWM_CLK_1MHz, \
.count_mode = (DT_INST_PROP(inst, center_aligned) \
? NRF_PWM_MODE_UP_AND_DOWN \
: NRF_PWM_MODE_UP), \
.top_value = 1000, \
.load_mode = NRF_PWM_LOAD_INDIVIDUAL, \
.step_mode = NRF_PWM_STEP_TRIGGERED, \
}, \
.seq.values.p_raw = pwm_##inst##_seq_values, \
.seq.length = NRF_PWM_CHANNEL_COUNT, \
.pcfg = PINCTRL_DT_INST_DEV_CONFIG_GET(inst), \
.clock_freq = COND_CODE_1(DT_INST_CLOCKS_HAS_IDX(inst, 0), \
(DT_PROP(DT_INST_CLOCKS_CTLR(inst), clock_frequency)), \
(16ul * 1000ul * 1000ul)), \
IF_ENABLED(CONFIG_DCACHE, \
(.mem_attr = PWM_GET_MEM_ATTR(inst),)) \
}; \
static int pwm_nrfx_init##inst(const struct device *dev) \
{ \
IRQ_CONNECT(DT_INST_IRQN(inst), DT_INST_IRQ(inst, priority), \
nrfx_pwm_irq_handler, &pwm_nrfx_##inst##_data.pwm, 0); \
return pwm_nrfx_init(dev); \
}; \
PM_DEVICE_DT_INST_DEFINE(inst, pwm_nrfx_pm_action); \
DEVICE_DT_INST_DEINIT_DEFINE(inst, \
pwm_nrfx_init##inst, NULL, \
PM_DEVICE_DT_INST_GET(inst), \
&pwm_nrfx_##inst##_data, \
&pwm_nrfx_##inst##_config, \
POST_KERNEL, CONFIG_PWM_INIT_PRIORITY, \
&pwm_nrfx_drv_api_funcs)
DT_INST_FOREACH_STATUS_OKAY(PWM_NRFX_DEFINE)