rsoc/board/ports/led_matrix/drv_matrix_led.c

306 lines
7.8 KiB
C
Raw Permalink Normal View History

2024-07-22 20:00:29 +08:00
/*
* Copyright (c) 2006-2023, RT-Thread Development Team
*
* SPDX-License-Identifier: Apache-2.0
*
* Change Logs:
* Date Author Notes
* 2023-05-22 yuanjie first version, function
*/
/**
* WS2812B serial LED data timing flow:
* | T0H | H | 350ns | ±150ns |
* | T0L | L | 800ns | ±150ns |
* | T1H | H | 700ns | ±150ns |
* | T1L | L | 600ns | ±150ns |
* | RES | L | 50us | -- |
* When using TIM peripheral, to meet 800kHz (1250ns) refresh rate:
* - period is: 1250ns
* - logic 0 is: 400ns(H) + 900ns(L)
* - logic 1 is: 900ns(H) + 400ns(L)
*/
#include <rtdevice.h>
#include <board.h>
#include <drv_matrix_led.h>
#include <drv_common.h>
#include <drv_gpio.h>
#ifndef LED_NUM
#define LED_NUM 19 // LED灯珠个数
#endif
#define LED_MATRIX_EN_PIN GET_PIN(F, 2)
TIM_HandleTypeDef htim3;
DMA_HandleTypeDef hdma_tim3_ch2;
rt_align(RT_ALIGN_SIZE) uint8_t led_buffer[LED_NUM * 24 * 2];
extern void HAL_TIM_MspPostInit(TIM_HandleTypeDef *htim);
// 模拟bit码: 2为逻辑0, 7为逻辑1
const uint8_t tile[] = {2, 7};
// 常见颜色定义
const RGBColor_TypeDef DARK = {0, 0, 0};
const RGBColor_TypeDef GREEN = {255, 0, 0};
const RGBColor_TypeDef RED = {0, 255, 0};
const RGBColor_TypeDef BLUE = {0, 0, 255};
const RGBColor_TypeDef WHITE = {255, 255, 255};
const RGBColor_TypeDef LT_RED = {0, 32, 0};
const RGBColor_TypeDef LT_GREEN = {32, 0, 0};
const RGBColor_TypeDef LT_BLUE = {0, 0, 32};
const RGBColor_TypeDef LT_WHITE = {16, 16, 16};
// 灯颜色缓存
RGBColor_TypeDef RGB_Data[LED_NUM] = {0};
void led_matrix_rst();
/**
* @brief This function handles DMA2 stream3 global interrupt.
*/
void DMA1_Stream5_IRQHandler(void)
{
HAL_DMA_IRQHandler(&hdma_tim3_ch2);
}
void HAL_TIM_PWM_PulseFinishedCallback(TIM_HandleTypeDef *htim)
{
if (htim->Channel == HAL_TIM_ACTIVE_CHANNEL_2)
{
__HAL_TIM_SetCompare(htim, TIM_CHANNEL_2,0); //占空比清0
}
}
/**
* @brief matrix Initialization Function
* @param None
* @retval None
*/
static int matrix_init(void)
{
TIM_MasterConfigTypeDef sMasterConfig = {0};
TIM_OC_InitTypeDef sConfigOC = {0};
/* TIM3_CH2 Init */
__HAL_RCC_TIM3_CLK_ENABLE();
htim3.Instance = TIM3;
htim3.Init.Prescaler = 10-1;
htim3.Init.CounterMode = TIM_COUNTERMODE_UP;
htim3.Init.Period = 10-1; // 840kHz
htim3.Init.ClockDivision = TIM_CLOCKDIVISION_DIV1;
htim3.Init.RepetitionCounter = 0;
htim3.Init.AutoReloadPreload = TIM_AUTORELOAD_PRELOAD_DISABLE;
if (HAL_TIM_PWM_Init(&htim3) != HAL_OK)
{
Error_Handler();
}
sMasterConfig.MasterOutputTrigger = TIM_TRGO_RESET;
sMasterConfig.MasterSlaveMode = TIM_MASTERSLAVEMODE_DISABLE;
if (HAL_TIMEx_MasterConfigSynchronization(&htim3, &sMasterConfig) != HAL_OK)
{
Error_Handler();
}
sConfigOC.OCMode = TIM_OCMODE_PWM1;
sConfigOC.Pulse = 0;
sConfigOC.OCPolarity = TIM_OCPOLARITY_HIGH;
sConfigOC.OCFastMode = TIM_OCFAST_DISABLE;
if (HAL_TIM_PWM_ConfigChannel(&htim3, &sConfigOC, TIM_CHANNEL_2) != HAL_OK)
{
Error_Handler();
}
HAL_TIM_MspPostInit(&htim3);
/* TIM3 DMA Init */
__HAL_RCC_DMA1_CLK_ENABLE();
hdma_tim3_ch2.Instance = DMA1_Stream5;
hdma_tim3_ch2.Init.Channel = DMA_CHANNEL_5;
hdma_tim3_ch2.Init.Direction = DMA_MEMORY_TO_PERIPH;
hdma_tim3_ch2.Init.PeriphInc = DMA_PINC_DISABLE;
hdma_tim3_ch2.Init.MemInc = DMA_MINC_ENABLE;
hdma_tim3_ch2.Init.PeriphDataAlignment = DMA_PDATAALIGN_HALFWORD;
hdma_tim3_ch2.Init.MemDataAlignment = DMA_PDATAALIGN_HALFWORD;
hdma_tim3_ch2.Init.Mode = DMA_NORMAL;
hdma_tim3_ch2.Init.Priority = DMA_PRIORITY_HIGH;
hdma_tim3_ch2.Init.FIFOMode = DMA_FIFOMODE_DISABLE;
if (HAL_DMA_Init(&hdma_tim3_ch2) != HAL_OK)
{
Error_Handler();
}
__HAL_LINKDMA(&htim3, hdma[TIM_DMA_ID_CC2], hdma_tim3_ch2);
/* NVIC configuration for DMA transfer complete interrupt */
HAL_NVIC_SetPriority(DMA1_Stream5_IRQn, 0, 0);
HAL_NVIC_EnableIRQ(DMA1_Stream5_IRQn);
rt_pin_mode(LED_MATRIX_EN_PIN, PIN_MODE_OUTPUT);
rt_pin_write(LED_MATRIX_EN_PIN, PIN_LOW);
led_matrix_rst();
return RT_EOK;
}
INIT_APP_EXPORT(matrix_init);
/**
* @brief
* @param[in] ID
*/
void Set_LEDColor(uint16_t LedId, RGBColor_TypeDef Color)
{
RGB_Data[LedId].G = Color.G;
RGB_Data[LedId].R = Color.R;
RGB_Data[LedId].B = Color.B;
}
/**
* @brief TIM发送控制ws2812
* @param[in]
*/
static void TIM_Send_WS2812(uint8_t *rgb_buffer, uint32_t size)
{
// 判断上次DMA有没有传输完成
while (HAL_DMA_GetState(&hdma_tim3_ch2) != HAL_DMA_STATE_READY);
// 发送一个24bit的RGB数据
HAL_TIM_PWM_Start_DMA(&htim3, TIM_CHANNEL_2, (uint32_t *)rgb_buffer, size);
}
/**
* @brief WS2812
* @param[in]
*/
void RGB_Reflash(void)
{
uint8_t dat_b,dat_r,dat_g;
// 将数组颜色转化为24个要发送的字节数据
for (uint16_t i = 0; i < LED_NUM; i++)
{
dat_g = RGB_Data[i].G;
dat_r = RGB_Data[i].R;
dat_b = RGB_Data[i].B;
for (uint16_t j = 0; j < 8; j++) {
led_buffer[(14 + (i * 48))-(j<<1)] = tile[dat_g & 0x01];
led_buffer[(14 + (i * 48))-(j<<1) + 1] = 0;
led_buffer[(30 + (i * 48))-(j<<1)] = tile[dat_r & 0x01];
led_buffer[(30 + (i * 48))-(j<<1) + 1] = 0;
led_buffer[(46 + (i * 48))-(j<<1)] = tile[dat_b & 0x01];
led_buffer[(46 + (i * 48))-(j<<1) + 1] = 0;
dat_g >>=1;
dat_r >>=1;
dat_b >>=1;
}
}
TIM_Send_WS2812(led_buffer, sizeof(led_buffer) / 2);
}
void led_matrix_rst()
{
for (uint32_t i = 0; i < (LED_NUM * 24); i++)
{
led_buffer[ (i<<1) ] = 3;
led_buffer[ (i<<1) + 1] = 0;
}
TIM_Send_WS2812(led_buffer, sizeof(led_buffer) / 2 );
}
MSH_CMD_EXPORT(led_matrix_rst, Test led matrix on board)
void led_matrix_fill(RGBColor_TypeDef Color)
{
rt_memset(RGB_Data, 0x00, sizeof(RGB_Data));
for (uint8_t i = 0; i < LED_NUM; i++)
{
Set_LEDColor(i, Color);
}
RGB_Reflash();
}
void led_matrix_fill_test(uint8_t index)
{
switch (index)
{
case 0:
led_matrix_fill(LT_RED);
break;
case 1:
led_matrix_fill(LT_GREEN);
break;
case 2:
led_matrix_fill(LT_BLUE);
break;
case 3:
led_matrix_fill(LT_WHITE);
break;
default:
break;
}
}
void led_matrix_test1()
{
rt_memset(RGB_Data, 0x00, sizeof(RGB_Data));
Set_LEDColor(0, RED);
Set_LEDColor(1, GREEN);
Set_LEDColor(2, BLUE);
Set_LEDColor(3, RED);
Set_LEDColor(4, GREEN);
Set_LEDColor(5, BLUE);
Set_LEDColor(6, RED);
Set_LEDColor(7, GREEN);
Set_LEDColor(8, BLUE);
Set_LEDColor(9, WHITE);
// led_matrix_rst();
RGB_Reflash();
}
MSH_CMD_EXPORT(led_matrix_test1, Test led matrix on board)
void led_matrix_test2()
{
rt_memset(RGB_Data, 0x00, sizeof(RGB_Data));
Set_LEDColor(0, BLUE);
Set_LEDColor(1, RED);
Set_LEDColor(2, GREEN);
Set_LEDColor(3, BLUE);
Set_LEDColor(4, RED);
Set_LEDColor(5, GREEN);
Set_LEDColor(6, BLUE);
Set_LEDColor(7, RED);
Set_LEDColor(8, GREEN);
Set_LEDColor(9, RED);
Set_LEDColor(14, GREEN);
Set_LEDColor(15, GREEN);
Set_LEDColor(16, BLUE);
Set_LEDColor(17, RED);
Set_LEDColor(18, WHITE);
RGB_Reflash();
}
MSH_CMD_EXPORT(led_matrix_test2, Test led matrix on board)
void led_matrix_test3()
{
for (uint8_t i = 0; i < 4; i++)
{
led_matrix_fill_test(i);
rt_thread_mdelay(1000);
}
led_matrix_rst();
}
MSH_CMD_EXPORT(led_matrix_test3, Test led matrix on board)
void led_matrix_show_color(uint8_t r, uint8_t g, uint8_t b)
{
RGBColor_TypeDef color = {g,r,b};
led_matrix_fill(color);
}