parent
2093a48bed
commit
85590593ce
@ -0,0 +1,64 @@ |
||||
#include "ws2812.h" |
||||
#include "utils/timebase.h" |
||||
|
||||
#define LONG_DELAY() for (volatile uint32_t __j = 4; __j > 0; __j--) |
||||
#define SHORT_DELAY() for (volatile uint32_t __j = 1; __j > 0; __j--) |
||||
|
||||
static inline |
||||
__attribute__((always_inline)) |
||||
void ws2812_byte(GPIO_TypeDef *GPIOx, uint16_t pin, uint8_t b) |
||||
{ |
||||
for (register volatile uint8_t i = 0; i < 8; i++) { |
||||
GPIOx->BSRR = pin; // set pin high
|
||||
|
||||
// duty cycle determines bit value
|
||||
if (b & 0x80) { |
||||
LONG_DELAY(); |
||||
GPIOx->BRR = pin; // set pin low
|
||||
SHORT_DELAY(); |
||||
} else { |
||||
SHORT_DELAY(); |
||||
GPIOx->BRR = pin; // set pin low
|
||||
LONG_DELAY(); |
||||
} |
||||
|
||||
b <<= 1; // shift to next bit
|
||||
} |
||||
} |
||||
|
||||
/** Set many RGBs */ |
||||
void ws2812_send(GPIO_TypeDef *GPIOx, uint16_t pin, uint32_t *rgbs, uint32_t count) |
||||
{ |
||||
__disable_irq(); |
||||
|
||||
for (int i = 0; i < count; i++) { |
||||
uint32_t rgb = *rgbs++; |
||||
ws2812_byte(GPIOx, pin, rgb_g(rgb)); |
||||
ws2812_byte(GPIOx, pin, rgb_r(rgb)); |
||||
ws2812_byte(GPIOx, pin, rgb_b(rgb)); |
||||
} |
||||
|
||||
__enable_irq(); |
||||
|
||||
ws2812_flip(); // show
|
||||
} |
||||
|
||||
/** Send a single color. */ |
||||
void ws2812_single(GPIO_TypeDef *GPIOx, uint16_t pin, uint32_t rgb) |
||||
{ |
||||
__disable_irq(); |
||||
|
||||
ws2812_byte(GPIOx, pin, rgb_g(rgb)); |
||||
ws2812_byte(GPIOx, pin, rgb_r(rgb)); |
||||
ws2812_byte(GPIOx, pin, rgb_b(rgb)); |
||||
|
||||
__enable_irq(); |
||||
} |
||||
|
||||
/**
|
||||
* @brief Wait the necessary time for the colors sent using ws2812_single() to show. |
||||
*/ |
||||
void ws2812_flip() |
||||
{ |
||||
delay_us(50); // show
|
||||
} |
||||
@ -0,0 +1,92 @@ |
||||
#ifndef MPORK_WS2812_H |
||||
#define MPORK_WS2812_H |
||||
|
||||
/* Includes ------------------------------------------------------------------*/ |
||||
|
||||
#include <common.h> |
||||
|
||||
/* Exported types ------------------------------------------------------------*/ |
||||
/* Exported constants --------------------------------------------------------*/ |
||||
|
||||
// PB8 - WS2812B data line
|
||||
#define COLORLED_GPIO GPIOC |
||||
#define COLORLED_PIN GPIO15 |
||||
|
||||
#define RGB_RED rgb(255, 0, 0) |
||||
#define RGB_ORANGE rgb(255, 110, 0) |
||||
#define RGB_YELLOW rgb(255, 255, 0) |
||||
#define RGB_LIME rgb(160, 255, 0) |
||||
#define RGB_GREEN rgb( 0, 255, 0) |
||||
#define RGB_CYAN rgb( 0, 255, 120) |
||||
#define RGB_BLUE rgb( 0, 0, 255) |
||||
#define RGB_MAGENTA rgb(255, 0, 255) |
||||
#define RGB_WHITE rgb(255, 255, 255) |
||||
#define RGB_BLACK rgb( 0, 0, 0) |
||||
|
||||
/* Exported macros -----------------------------------------------------------*/ |
||||
|
||||
/**
|
||||
* @brief Compose an RGB color. |
||||
* @param r, g, b - components 0xFF |
||||
* @returns integer 0xRRGGBB |
||||
*/ |
||||
#define rgb(r, g, b) (((0xFF & (r)) << 16) | ((0xFF & (g)) << 8) | (0xFF & (b))) |
||||
|
||||
/* Get components */ |
||||
#define rgb_r(rgb) (((rgb) >> 16) & 0xFF) |
||||
#define rgb_g(rgb) (((rgb) >> 8) & 0xFF) |
||||
#define rgb_b(rgb) ((rgb) & 0xFF) |
||||
|
||||
/* Exported functions --------------------------------------------------------*/ |
||||
|
||||
/**
|
||||
* @brief Struct for easy manipulation of RGB colors. |
||||
* |
||||
* Set components in the xrgb.r (etc.) and you will get |
||||
* the hex in xrgb.num. |
||||
*/ |
||||
typedef union { |
||||
|
||||
/** Struct for access to individual color components */ |
||||
struct __attribute__((packed)) { |
||||
uint8_t b; |
||||
uint8_t g; |
||||
uint8_t r; |
||||
}; |
||||
|
||||
/** RGB color as a single uint32_t */ |
||||
uint32_t num; |
||||
|
||||
} ws2812_rgb_t; |
||||
|
||||
|
||||
/**
|
||||
* @brief Macro to compose the RGB struct. |
||||
* |
||||
* You can also use {.num = 0xFF0000} to set the hex directly. |
||||
*/ |
||||
#define WS2812_RGB(r, g, b) {.num = (((r) & 0xFF) << 16) | (((g) & 0xFF) << 8) | ((b) & 0xFF)} |
||||
|
||||
/**
|
||||
* @brief Send a sequence of colors, then display it. |
||||
* @param GPIOx |
||||
* @param pin |
||||
* @param rgbs : array of colors |
||||
* @param count : array length |
||||
*/ |
||||
void ws2812_send(GPIO_TypeDef *GPIOx, uint16_t pin, uint32_t *rgbs, uint32_t count); |
||||
|
||||
/**
|
||||
* @brief Send a single color. |
||||
* @param GPIOx |
||||
* @param pin |
||||
* @param rgb : color to show |
||||
*/ |
||||
void ws2812_single(GPIO_TypeDef *GPIOx, uint16_t pin, uint32_t rgb); |
||||
|
||||
/**
|
||||
* @brief Wait the necessary time for the colors sent using ws2812_single() to show. |
||||
*/ |
||||
void ws2812_flip(); |
||||
|
||||
#endif //MPORK_WS2812_H
|
||||
Loading…
Reference in new issue