mlaz commented on a change in pull request #1102: Feature/stm32 pwm enabled URL: https://github.com/apache/mynewt-core/pull/1102#discussion_r188975398
########## File path: hw/drivers/pwm/pwm_stm32/src/pwm_stm32.c ########## @@ -0,0 +1,626 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ +#include "pwm_stm32/pwm_stm32.h" + +#include <bsp.h> +#include <hal/hal_bsp.h> +#include <mcu/cmsis_nvic.h> +#include <os/os.h> +#include <pwm/pwm.h> +#include <stm32_common/stm32_hal.h> + +#include <assert.h> +#include <stdint.h> +#include <string.h> + +#define STM32_PWM_CH_MAX 4 +#define STM32_PWM_CH_IDLE 0x0000 +#define STM32_PWM_CH_MODE_ENA LL_TIM_OCMODE_PWM2 +#define STM32_PWM_CH_MODE_DIS LL_TIM_OCMODE_ACTIVE + + +typedef struct { + uint32_t n_cycles; + user_handler_t cycle_handler; + user_handler_t seq_end_handler; + void *cycle_data; + void *seq_end_data; +} stm32_pwm_dev_cfg_t; + +typedef struct { + TIM_TypeDef *timx; + uint32_t cycle; + uint16_t pin[STM32_PWM_CH_MAX]; + uint16_t irq; + stm32_pwm_dev_cfg_t cfg; +} stm32_pwm_dev_t; + +static stm32_pwm_dev_t stm32_pwm_dev[PWM_CNT]; + +static inline bool +stm32_pwm_ch_is_active(const stm32_pwm_dev_t *pwm, int ch) +{ + return MCU_AFIO_PIN_NONE != pwm->pin[ch]; +} + +static uint32_t +stm32_pwm_ch(int ch) +{ + switch (ch) { + case 0: return LL_TIM_CHANNEL_CH1; + case 1: return LL_TIM_CHANNEL_CH2; + case 2: return LL_TIM_CHANNEL_CH3; + case 3: return LL_TIM_CHANNEL_CH4; + } + assert(0); + return 0; +} + +static void +stm32_pwm_active_ch_set_mode(stm32_pwm_dev_t *pwm, uint32_t mode) { + for (int i=0; i < STM32_PWM_CH_MAX; ++i) { + if (stm32_pwm_ch_is_active(pwm, i)) { + LL_TIM_OC_SetMode(pwm->timx, stm32_pwm_ch(i), mode); + } + } +} + +static void +stm32_pwm_ch_set_compare(TIM_TypeDef *tim, int ch, uint32_t value) +{ + switch (ch) { + case 0: + LL_TIM_OC_SetCompareCH1(tim, value); + break; + case 1: + LL_TIM_OC_SetCompareCH2(tim, value); + break; + case 2: + LL_TIM_OC_SetCompareCH3(tim, value); + break; + case 3: + LL_TIM_OC_SetCompareCH4(tim, value); + break; + } +} + +static void +stm32_pwm_ch_unconfigure(const stm32_pwm_dev_t *pwm, uint32_t id) +{ + uint32_t ch = stm32_pwm_ch(id); + + LL_TIM_CC_DisableChannel(pwm->timx, ch); + LL_TIM_OC_SetMode(pwm->timx, ch, 0); + LL_TIM_OC_SetPolarity(pwm->timx, ch, 0); + LL_TIM_OC_DisablePreload(pwm->timx, ch); +} + +static void +stm32_pwm_isr(stm32_pwm_dev_t *pwm) +{ + uint32_t sr = pwm->timx->SR; + pwm->timx->SR = ~sr; + + if (pwm->cfg.cycle_handler) { + pwm->cfg.cycle_handler(pwm->cfg.cycle_data); + } + + if (pwm->cfg.n_cycles) { + if (!pwm->cycle) { + + LL_TIM_DisableCounter(pwm->timx); + LL_TIM_SetCounter(pwm->timx, 0); + + if (pwm->cfg.seq_end_handler) { + pwm->cfg.seq_end_handler(pwm->cfg.seq_end_data); + } + } else { + if (1 == pwm->cycle) { + /* prep output pins for shutdown */ + stm32_pwm_active_ch_set_mode(pwm, STM32_PWM_CH_MODE_DIS); + } + --pwm->cycle; + } + } +} + +static void +stm32_pwm_isr_0(void) +{ + stm32_pwm_isr(&stm32_pwm_dev[0]); +} + +static void +stm32_pwm_isr_1(void) +{ + stm32_pwm_isr(&stm32_pwm_dev[1]); +} + +static void +stm32_pwm_isr_2(void) +{ + stm32_pwm_isr(&stm32_pwm_dev[2]); +} + +static int +stm32_pwm_enable(struct pwm_dev *dev) +{ + stm32_pwm_dev_t *pwm; + + assert(dev); + assert(dev->pwm_instance_id < PWM_CNT); + + pwm = &stm32_pwm_dev[dev->pwm_instance_id]; + pwm->cycle = pwm->cfg.n_cycles; + + stm32_pwm_active_ch_set_mode(pwm, STM32_PWM_CH_MODE_ENA); + + LL_TIM_GenerateEvent_UPDATE(pwm->timx); + LL_TIM_EnableCounter(pwm->timx); + + return STM32_PWM_ERR_OK; +} + +static int +stm32_pwm_disable(struct pwm_dev *dev) +{ + stm32_pwm_dev_t *pwm; + + assert(dev); + assert(dev->pwm_instance_id < PWM_CNT); + + pwm = &stm32_pwm_dev[dev->pwm_instance_id]; + + LL_TIM_DisableCounter(pwm->timx); + LL_TIM_SetCounter(pwm->timx, 0); + + return STM32_PWM_ERR_OK; +} + +static bool +stm32_pwm_is_enabled(struct pwm_dev *dev) +{ + stm32_pwm_dev_t *pwm; + + assert(dev); + assert(dev->pwm_instance_id < PWM_CNT); + + pwm = &stm32_pwm_dev[dev->pwm_instance_id]; + + return LL_TIM_IsEnabledCounter(pwm->timx); +} + +static int +stm32_pwm_open(struct os_dev *odev, uint32_t wait, void *arg) +{ + struct pwm_dev *dev; + int rc; + + dev = (struct pwm_dev *)odev; + assert(dev); + + if (os_started()) { + rc = os_mutex_pend(&dev->pwm_lock, wait); + if (OS_OK != rc) { + return rc; + } + } + + if (odev->od_flags & OS_DEV_F_STATUS_OPEN) { + os_mutex_release(&dev->pwm_lock); + rc = OS_EBUSY; + return rc; + } + + return STM32_PWM_ERR_OK; +} + +static int +stm32_pwm_close(struct os_dev *odev) +{ + struct pwm_dev *dev; + stm32_pwm_dev_t *pwm; + + dev = (struct pwm_dev *)odev; + assert(dev); + + stm32_pwm_disable(dev); + pwm = &stm32_pwm_dev[dev->pwm_instance_id]; + + for (int i=0; i < STM32_PWM_CH_MAX; ++i) { + stm32_pwm_ch_set_compare(pwm->timx, i, STM32_PWM_CH_IDLE); + if (stm32_pwm_ch_is_active(pwm, i)) { + hal_gpio_init_af(MCU_AFIO_PIN_PAD(pwm->pin[i]), 0, HAL_GPIO_PULL_NONE, 0); + } + pwm->pin[i] = MCU_AFIO_PIN_NONE; + stm32_pwm_ch_unconfigure(pwm, i); + } + + if (os_started()) { + os_mutex_release(&dev->pwm_lock); + } + + return STM32_PWM_ERR_OK; +} + +static int +stm32_pwm_ch_confgigure(struct pwm_dev *dev, uint8_t cnum, struct pwm_chan_cfg *cfg) +{ + stm32_pwm_dev_t *pwm; + uint32_t ch; + + assert(dev); + assert(dev->pwm_instance_id < PWM_CNT); + assert(dev->pwm_chan_count <= STM32_PWM_CH_MAX); + if (cnum >= dev->pwm_chan_count) { + return STM32_PWM_ERR_CHAN; + } + + pwm = &stm32_pwm_dev[dev->pwm_instance_id]; + ch = stm32_pwm_ch(cnum); + + LL_TIM_CC_DisableChannel(pwm->timx, ch); + + if (stm32_pwm_ch_is_active(pwm, cnum)) { + if (hal_gpio_init_af(MCU_AFIO_PIN_PAD(cfg->pin), 0, HAL_GPIO_PULL_NONE, 0)) { + return STM32_PWM_ERR_GPIO; + } + } + + if (MCU_AFIO_PIN_NONE != cfg->pin ) { + LL_TIM_OC_SetMode(pwm->timx, ch, STM32_PWM_CH_MODE_ENA); + LL_TIM_OC_SetPolarity(pwm->timx, ch, cfg->inverted ? LL_TIM_OCPOLARITY_HIGH : LL_TIM_OCPOLARITY_LOW); + LL_TIM_OC_EnablePreload(pwm->timx, ch); + + if (hal_gpio_init_af(MCU_AFIO_PIN_PAD(cfg->pin), MCU_AFIO_PIN_AF(cfg->pin), HAL_GPIO_PULL_NONE, 0)) { + return STM32_PWM_ERR_GPIO; + } + pwm->pin[cnum] = cfg->pin; + + LL_TIM_CC_EnableChannel(pwm->timx, ch); + } else { + stm32_pwm_ch_unconfigure(pwm, cnum); + pwm->pin[cnum] = MCU_AFIO_PIN_NONE; + } + + return STM32_PWM_ERR_OK; +} + + +static int +stm32_pwm_ch_set_duty_cycle(struct pwm_dev *dev, uint8_t cnum, uint16_t fraction) +{ + stm32_pwm_dev_t *pwm; + + assert(dev); + assert(dev->pwm_instance_id < PWM_CNT); + if (cnum >= dev->pwm_chan_count) { + return STM32_PWM_ERR_CHAN; + } + + pwm = &stm32_pwm_dev[dev->pwm_instance_id]; + + stm32_pwm_ch_set_compare(pwm->timx, cnum, fraction); + + return STM32_PWM_ERR_OK; +} + +static int +stm32_pwm_set_frequency(struct pwm_dev *dev, uint32_t freq_hz) +{ + stm32_pwm_dev_t *pwm; + uint32_t id; + uint32_t timer_clock; + uint32_t div, div1, div2; + + assert(dev); + assert(dev->pwm_instance_id < PWM_CNT); + + if (!freq_hz) { + return STM32_PWM_ERR_FREQ; + } + + id = dev->pwm_instance_id; + pwm = &stm32_pwm_dev[id]; + + timer_clock = stm32_hal_timer_get_freq(pwm->timx); + assert(timer_clock); + + div = timer_clock / freq_hz; + if (!div) { + return STM32_PWM_ERR_FREQ; + } + + div1 = div >> 16; + div2 = div / (div1 + 1); + + if (div1 > div2) { + uint32_t tmp = div1; + div1 = div2; + div2 = tmp; + } + div2 -= 1; + + LL_TIM_SetPrescaler(pwm->timx, div1); + LL_TIM_SetAutoReload(pwm->timx, div2); + + return STM32_PWM_ERR_OK; +} + +static int +stm32_pwm_configure(struct pwm_dev *dev, struct pwm_dev_cfg *cfg) +{ + stm32_pwm_dev_t *pwm; + uint32_t prio; + + assert(dev); + assert(dev->pwm_instance_id < PWM_CNT); Review comment: Not sure this is over verifying. The only way this value to be corrupted is driver bad implementation, no? ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services
