src/main/drivers/light_ws2811strip_stm32f4xx.c

   1 /*
   2  * This file is part of Cleanflight.
   3  *
   4  * Cleanflight is free software: you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation, either version 3 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * Cleanflight is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with Cleanflight.  If not, see <http://www.gnu.org/licenses/>.
  16  */
  17
  18 #include <stdbool.h>
  19 #include <stdint.h>
  20
  21 #include "platform.h"
  22
  23 #include "common/color.h"
  24 #include "light_ws2811strip.h"
  25 #include "dma.h"
  26 #include "nvic.h"
  27 #include "io.h"
  28 #include "system.h"
  29 #include "rcc.h"
  30 #include "timer.h"
  31
  32 #ifdef LED_STRIP
  33
  34 #if !defined(WS2811_PIN)
  35 #define WS2811_PIN                      PA0
  36 #define WS2811_TIMER                    TIM5
  37 #define WS2811_DMA_HANDLER_IDENTIFER    DMA1_ST2_HANDLER
  38 #define WS2811_DMA_STREAM               DMA1_Stream2
  39 #define WS2811_DMA_FLAG                 DMA_FLAG_TCIF2
  40 #define WS2811_DMA_IT                   DMA_IT_TCIF2
  41 #define WS2811_DMA_CHANNEL              DMA_Channel_6
  42 #define WS2811_DMA_IRQ                  DMA1_Stream2_IRQn
  43 #define WS2811_TIMER_CHANNEL            TIM_Channel_1
  44 #endif
  45
  46 static IO_t ws2811IO = IO_NONE;
  47 static uint16_t timDMASource = 0;
  48 bool ws2811Initialised = false;
  49
  50 void ws2811DMAHandler(DMA_Stream_TypeDef *stream)
  51 {
  52     if (DMA_GetFlagStatus(stream, WS2811_DMA_FLAG)) {
  53         ws2811LedDataTransferInProgress = 0;
  54         DMA_ClearITPendingBit(stream, WS2811_DMA_IT);
  55         DMA_Cmd(stream, DISABLE);
  56         TIM_DMACmd(WS2811_TIMER, timDMASource, DISABLE);
  57     }
  58 }
  59
  60 void ws2811LedStripHardwareInit(void)
  61 {
  62     TIM_TimeBaseInitTypeDef  TIM_TimeBaseStructure;
  63     TIM_OCInitTypeDef  TIM_OCInitStructure;
  64     DMA_InitTypeDef DMA_InitStructure;
  65
  66     uint16_t prescalerValue;
  67
  68     RCC_ClockCmd(timerRCC(WS2811_TIMER), ENABLE);
  69
  70     ws2811IO = IOGetByTag(IO_TAG(WS2811_PIN));
  71     /* GPIOA Configuration: TIM5 Channel 1 as alternate function push-pull */
  72         IOInit(ws2811IO, OWNER_LED_STRIP, RESOURCE_OUTPUT, 0);
  73     IOConfigGPIOAF(ws2811IO, IO_CONFIG(GPIO_Mode_AF, GPIO_Speed_50MHz, GPIO_OType_PP, GPIO_PuPd_UP), timerGPIOAF(WS2811_TIMER));
  74
  75     // Stop timer
  76     TIM_Cmd(WS2811_TIMER, DISABLE);
  77
  78     /* Compute the prescaler value */
  79     prescalerValue = (uint16_t)(SystemCoreClock / 2 / 84000000) - 1;
  80
  81     /* Time base configuration */
  82     TIM_TimeBaseStructure.TIM_Period = 104; // 800kHz
  83     TIM_TimeBaseStructure.TIM_Prescaler = prescalerValue;
  84     TIM_TimeBaseStructure.TIM_ClockDivision = TIM_CKD_DIV1;
  85     TIM_TimeBaseStructure.TIM_CounterMode = TIM_CounterMode_Up;
  86     TIM_TimeBaseInit(WS2811_TIMER, &TIM_TimeBaseStructure);
  87
  88     /* PWM1 Mode configuration: Channel1 */
  89     TIM_OCInitStructure.TIM_OCMode = TIM_OCMode_PWM1;
  90     TIM_OCInitStructure.TIM_OCIdleState = TIM_OCIdleState_Reset;
  91     TIM_OCInitStructure.TIM_OCNIdleState = TIM_OCNIdleState_Set;
  92     TIM_OCInitStructure.TIM_OCPolarity = TIM_OCNPolarity_High;
  93     TIM_OCInitStructure.TIM_OCNPolarity = TIM_OCNPolarity_High;
  94     TIM_OCInitStructure.TIM_OutputState = TIM_OutputState_Enable;
  95     TIM_OCInitStructure.TIM_OutputNState = TIM_OutputNState_Disable;
  96     TIM_OCInitStructure.TIM_Pulse = 0;
  97
  98     uint32_t channelAddress = 0;
  99     switch (WS2811_TIMER_CHANNEL) {
 100         case TIM_Channel_1:
 101             TIM_OC1Init(WS2811_TIMER, &TIM_OCInitStructure);
 102             timDMASource = TIM_DMA_CC1;
 103             channelAddress = (uint32_t)(&WS2811_TIMER->CCR1);
 104             TIM_OC1PreloadConfig(WS2811_TIMER, TIM_OCPreload_Enable);
 105             break;
 106         case TIM_Channel_2:
 107             TIM_OC2Init(WS2811_TIMER, &TIM_OCInitStructure);
 108             timDMASource = TIM_DMA_CC2;
 109             channelAddress = (uint32_t)(&WS2811_TIMER->CCR2);
 110             TIM_OC2PreloadConfig(WS2811_TIMER, TIM_OCPreload_Enable);
 111             break;
 112         case TIM_Channel_3:
 113             TIM_OC3Init(WS2811_TIMER, &TIM_OCInitStructure);
 114             timDMASource = TIM_DMA_CC3;
 115             channelAddress = (uint32_t)(&WS2811_TIMER->CCR3);
 116             TIM_OC3PreloadConfig(WS2811_TIMER, TIM_OCPreload_Enable);
 117             break;
 118         case TIM_Channel_4:
 119             TIM_OC4Init(WS2811_TIMER, &TIM_OCInitStructure);
 120             timDMASource = TIM_DMA_CC4;
 121             channelAddress = (uint32_t)(&WS2811_TIMER->CCR4);
 122             TIM_OC4PreloadConfig(WS2811_TIMER, TIM_OCPreload_Enable);
 123             break;
 124     }
 125
 126     TIM_CtrlPWMOutputs(WS2811_TIMER, ENABLE);
 127     TIM_ARRPreloadConfig(WS2811_TIMER, ENABLE);
 128
 129     TIM_CCxCmd(WS2811_TIMER, WS2811_TIMER_CHANNEL, TIM_CCx_Enable);
 130     TIM_Cmd(WS2811_TIMER, ENABLE);
 131
 132     dmaSetHandler(WS2811_DMA_HANDLER_IDENTIFER, ws2811DMAHandler);
 133
 134     /* configure DMA */
 135     DMA_Cmd(WS2811_DMA_STREAM, DISABLE);
 136     DMA_DeInit(WS2811_DMA_STREAM);
 137     DMA_StructInit(&DMA_InitStructure);
 138     DMA_InitStructure.DMA_Channel = WS2811_DMA_CHANNEL;
 139     DMA_InitStructure.DMA_PeripheralBaseAddr = channelAddress;
 140     DMA_InitStructure.DMA_Memory0BaseAddr = (uint32_t)ledStripDMABuffer;
 141     DMA_InitStructure.DMA_DIR = DMA_DIR_MemoryToPeripheral;
 142     DMA_InitStructure.DMA_BufferSize = WS2811_DMA_BUFFER_SIZE;
 143     DMA_InitStructure.DMA_PeripheralInc = DMA_PeripheralInc_Disable;
 144     DMA_InitStructure.DMA_MemoryInc = DMA_MemoryInc_Enable;
 145     DMA_InitStructure.DMA_PeripheralDataSize = DMA_PeripheralDataSize_Word;
 146     DMA_InitStructure.DMA_MemoryDataSize = DMA_MemoryDataSize_Word;
 147     DMA_InitStructure.DMA_Mode = DMA_Mode_Normal;
 148     DMA_InitStructure.DMA_Priority = DMA_Priority_VeryHigh;
 149     DMA_InitStructure.DMA_FIFOMode = DMA_FIFOMode_Enable;
 150     DMA_InitStructure.DMA_FIFOThreshold = DMA_FIFOThreshold_1QuarterFull;
 151     DMA_InitStructure.DMA_MemoryBurst = DMA_MemoryBurst_Single;
 152     DMA_InitStructure.DMA_PeripheralBurst = DMA_PeripheralBurst_Single;
 153
 154
 155     RCC_AHB1PeriphClockCmd(RCC_AHB1Periph_DMA1, ENABLE);
 156     DMA_Init(WS2811_DMA_STREAM, &DMA_InitStructure);
 157
 158     DMA_ITConfig(WS2811_DMA_STREAM, DMA_IT_TC, ENABLE);
 159     DMA_ClearITPendingBit(WS2811_DMA_STREAM, WS2811_DMA_IT);
 160
 161     NVIC_InitTypeDef NVIC_InitStructure;
 162
 163     NVIC_InitStructure.NVIC_IRQChannel = WS2811_DMA_IRQ;
 164     NVIC_InitStructure.NVIC_IRQChannelPreemptionPriority = NVIC_PRIORITY_BASE(NVIC_PRIO_WS2811_DMA);
 165     NVIC_InitStructure.NVIC_IRQChannelSubPriority = NVIC_PRIORITY_SUB(NVIC_PRIO_WS2811_DMA);
 166     NVIC_InitStructure.NVIC_IRQChannelCmd = ENABLE;
 167     NVIC_Init(&NVIC_InitStructure);
 168
 169     ws2811Initialised = true;
 170     setStripColor(&hsv_white);
 171     ws2811UpdateStrip();
 172 }
 173
 174 void ws2811LedStripDMAEnable(void)
 175 {
 176     if (!ws2811Initialised)
 177         return;
 178
 179     DMA_SetCurrDataCounter(WS2811_DMA_STREAM, WS2811_DMA_BUFFER_SIZE);  // load number of bytes to be transferred
 180     TIM_SetCounter(WS2811_TIMER, 0);
 181     DMA_Cmd(WS2811_DMA_STREAM, ENABLE);
 182     TIM_DMACmd(WS2811_TIMER, timDMASource, ENABLE);
 183 }
 184
 185 #endif