src/main/drivers/light_ws2811strip.h

   1 /*
   2  * This file is part of Cleanflight and Betaflight.
   3  *
   4  * Cleanflight and Betaflight are free software. You can redistribute
   5  * this software and/or modify this software under the terms of the
   6  * GNU General Public License as published by the Free Software
   7  * Foundation, either version 3 of the License, or (at your option)
   8  * any later version.
   9  *
  10  * Cleanflight and Betaflight are distributed in the hope that they
  11  * will be useful, but WITHOUT ANY WARRANTY; without even the implied
  12  * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13  * See the GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this software.
  17  *
  18  * If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 #pragma once
  22
  23 #include "platform.h"
  24
  25 #include "drivers/io_types.h"
  26
  27 #define WS2811_LED_STRIP_LENGTH    32
  28
  29 #define WS2811_BITS_PER_LED_MAX    32
  30
  31 #if defined(USE_WS2811_SINGLE_COLOUR)
  32 #define WS2811_DATA_BUFFER_SIZE    1
  33 #define WS2811_DMA_BUFFER_SIZE     (WS2811_DATA_BUFFER_SIZE * WS2811_BITS_PER_LED_MAX)
  34 // Do 2 extra iterations of the DMA transfer with the output set to low to generate the > 50us delay.
  35 #define WS2811_DELAY_ITERATIONS    2
  36 #else
  37 #define WS2811_DATA_BUFFER_SIZE    WS2811_LED_STRIP_LENGTH
  38 // for 50us delay
  39 #define WS2811_DELAY_BUFFER_LENGTH 42
  40 // number of bytes needed is #LEDs * 24 bytes + 42 trailing bytes)
  41 #define WS2811_DMA_BUFFER_SIZE     (WS2811_DATA_BUFFER_SIZE * WS2811_BITS_PER_LED_MAX + WS2811_DELAY_BUFFER_LENGTH)
  42 #endif
  43
  44 #ifdef USE_LEDSTRIP_CACHE_MGMT
  45 // WS2811_DMA_BUFFER_SIZE is multiples of uint32_t
  46 // Number of bytes required for buffer
  47 #define WS2811_DMA_BUF_BYTES              (WS2811_DMA_BUFFER_SIZE * sizeof(uint32_t))
  48 // Number of bytes required to cache align buffer
  49 #define WS2811_DMA_BUF_CACHE_ALIGN_BYTES  ((WS2811_DMA_BUF_BYTES + 0x20) & ~0x1f)
  50 // Size of array to create a cache aligned buffer
  51 #define WS2811_DMA_BUF_CACHE_ALIGN_LENGTH (WS2811_DMA_BUF_CACHE_ALIGN_BYTES / sizeof(uint32_t))
  52 extern uint32_t ledStripDMABuffer[WS2811_DMA_BUF_CACHE_ALIGN_LENGTH];
  53 #else
  54 extern uint32_t ledStripDMABuffer[WS2811_DMA_BUFFER_SIZE];
  55 #endif
  56
  57 #define WS2811_TIMER_MHZ           48
  58 #define WS2811_CARRIER_HZ          800000
  59
  60 // Enumeration to match the string options defined in lookupLedStripFormatRGB in settings.c
  61 typedef enum {
  62     LED_GRB,
  63     LED_RGB,
  64     LED_GRBW
  65 } ledStripFormatRGB_e;
  66
  67 void ws2811LedStripInit(ioTag_t ioTag);
  68 void ws2811LedStripEnable(void);
  69
  70 bool ws2811LedStripHardwareInit(ioTag_t ioTag);
  71 void ws2811LedStripDMAEnable(void);
  72
  73 void ws2811UpdateStrip(ledStripFormatRGB_e ledFormat, uint8_t brightness);
  74
  75 void setLedHsv(uint16_t index, const hsvColor_t *color);
  76 void getLedHsv(uint16_t index, hsvColor_t *color);
  77
  78 void scaleLedValue(uint16_t index, const uint8_t scalePercent);
  79 void setLedValue(uint16_t index, const uint8_t value);
  80
  81 void setStripColor(const hsvColor_t *color);
  82 void setStripColors(const hsvColor_t *colors);
  83
  84 void setUsedLedCount(unsigned ledCount);
  85
  86 bool isWS2811LedStripReady(void);
  87
  88 extern volatile bool ws2811LedDataTransferInProgress;
  89
  90 extern uint16_t BIT_COMPARE_1;
  91 extern uint16_t BIT_COMPARE_0;