rgb
This commit is contained in:
@@ -109,6 +109,15 @@ bool pq_pop(PacketQueue *q, uint8_t out_packet[PACKET_SIZE]){
|
||||
#define MODE_DEBUG 3
|
||||
#define UART_RX_BUFF_SIZE 64
|
||||
#define QUEUE_SIZ 8
|
||||
|
||||
#define LED_COUNT 8 //Adjust this based on how many leds we have daisy chained
|
||||
#define BYTES_PER_LED 3 //The neopixel needs 3 bytes G R B
|
||||
#define BITS_PER_LED (BYTES_PER_LED * 8)
|
||||
#define LED_BUFFER_SIZE (LED_COUNT * BITS_PER_LED)
|
||||
#define T0H 3
|
||||
#define T1H 4
|
||||
|
||||
uint16_t led_buffer[LED_BUFFER_SIZE]; //Contains the colors that is going to get send
|
||||
/* USER CODE END PM */
|
||||
|
||||
/* Private variables ---------------------------------------------------------*/
|
||||
@@ -191,6 +200,9 @@ void resetReport(void);
|
||||
void sendMessage(void);
|
||||
void findBestParent();
|
||||
void mergeChild();
|
||||
void setRGBcolor(int index, uint8_t r, uint8_t g, uint8_t b);
|
||||
void sendRGBcolor();
|
||||
void RGBProcess();
|
||||
/* USER CODE END PFP */
|
||||
|
||||
/* Private user code ---------------------------------------------------------*/
|
||||
@@ -313,7 +325,7 @@ int main(void)
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
RGBProcess();
|
||||
HAL_Delay(20);
|
||||
/* USER CODE END WHILE */
|
||||
|
||||
@@ -322,6 +334,39 @@ int main(void)
|
||||
/* USER CODE END 3 */
|
||||
}
|
||||
|
||||
void RGBProcess(){
|
||||
uint8_t red = 255;
|
||||
uint8_t blue = 256/2;
|
||||
uint8_t green = 0;
|
||||
for(int i = 0; i < LED_COUNT; i++){
|
||||
setRGBcolor(i, red, blue, green);
|
||||
red += 10;
|
||||
blue+= 10;
|
||||
green += 10;
|
||||
}
|
||||
sendRGBcolor();
|
||||
}
|
||||
//Sets each bit in the buffer that is going to get sent thru PWM DMA
|
||||
void setRGBcolor(int index, uint8_t r, uint8_t g, uint8_t b)
|
||||
{
|
||||
uint8_t colors[3] = {g, r, b};
|
||||
|
||||
for (int c = 0; c < 3; c++) {
|
||||
for (int bit = 0; bit < 8; bit++) {
|
||||
if (colors[c] & (1 << (7 - bit))) {
|
||||
led_buffer[index * 24 + (c * 8 + bit)] = T1H;
|
||||
} else {
|
||||
led_buffer[index * 24 + (c * 8 + bit)] = T0H;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//Sends whats on the buffer thru DMA
|
||||
void sendRGBcolor(){
|
||||
HAL_TIM_PWM_Start_DMA(&htim2, TIM_CHANNEL_1,(uint32_t*)led_buffer,LED_BUFFER_SIZE);
|
||||
}
|
||||
|
||||
void mergeChild(){
|
||||
uint8_t packet[12];
|
||||
if (pq_pop(&huart1q, packet)) {
|
||||
|
||||
@@ -27,50 +27,37 @@
|
||||
TIM_HandleTypeDef htim2;
|
||||
TIM_HandleTypeDef htim3;
|
||||
|
||||
/* TIM2 init function */
|
||||
void MX_TIM2_Init(void)
|
||||
{
|
||||
TIM_OC_InitTypeDef sConfigOC = {0};
|
||||
|
||||
/* USER CODE BEGIN TIM2_Init 0 */
|
||||
htim2.Instance = TIM2;
|
||||
htim2.Init.Prescaler = 0;
|
||||
htim2.Init.CounterMode = TIM_COUNTERMODE_UP;
|
||||
htim2.Init.Period = 9;
|
||||
htim2.Init.ClockDivision = TIM_CLOCKDIVISION_DIV1;
|
||||
htim2.Init.AutoReloadPreload = TIM_AUTORELOAD_PRELOAD_DISABLE;
|
||||
|
||||
/* USER CODE END TIM2_Init 0 */
|
||||
if (HAL_TIM_PWM_Init(&htim2) != HAL_OK) Error_Handler();
|
||||
|
||||
TIM_MasterConfigTypeDef sMasterConfig = {0};
|
||||
TIM_OC_InitTypeDef sConfigOC = {0};
|
||||
sConfigOC.OCMode = TIM_OCMODE_PWM1;
|
||||
sConfigOC.Pulse = 0;
|
||||
sConfigOC.OCPolarity = TIM_OCPOLARITY_HIGH;
|
||||
sConfigOC.OCFastMode = TIM_OCFAST_DISABLE;
|
||||
|
||||
/* USER CODE BEGIN TIM2_Init 1 */
|
||||
|
||||
/* USER CODE END TIM2_Init 1 */
|
||||
htim2.Instance = TIM2;
|
||||
htim2.Init.Prescaler = 0;
|
||||
htim2.Init.CounterMode = TIM_COUNTERMODE_UP;
|
||||
htim2.Init.Period = 4294967295;
|
||||
htim2.Init.ClockDivision = TIM_CLOCKDIVISION_DIV1;
|
||||
htim2.Init.AutoReloadPreload = TIM_AUTORELOAD_PRELOAD_DISABLE;
|
||||
if (HAL_TIM_OC_Init(&htim2) != HAL_OK)
|
||||
{
|
||||
Error_Handler();
|
||||
}
|
||||
sMasterConfig.MasterOutputTrigger = TIM_TRGO_RESET;
|
||||
sMasterConfig.MasterSlaveMode = TIM_MASTERSLAVEMODE_DISABLE;
|
||||
if (HAL_TIMEx_MasterConfigSynchronization(&htim2, &sMasterConfig) != HAL_OK)
|
||||
{
|
||||
Error_Handler();
|
||||
}
|
||||
sConfigOC.OCMode = TIM_OCMODE_FORCED_ACTIVE;
|
||||
sConfigOC.Pulse = 0;
|
||||
sConfigOC.OCPolarity = TIM_OCPOLARITY_HIGH;
|
||||
sConfigOC.OCFastMode = TIM_OCFAST_DISABLE;
|
||||
if (HAL_TIM_OC_ConfigChannel(&htim2, &sConfigOC, TIM_CHANNEL_1) != HAL_OK)
|
||||
{
|
||||
Error_Handler();
|
||||
}
|
||||
/* USER CODE BEGIN TIM2_Init 2 */
|
||||
|
||||
/* USER CODE END TIM2_Init 2 */
|
||||
HAL_TIM_MspPostInit(&htim2);
|
||||
if (HAL_TIM_PWM_ConfigChannel(&htim2, &sConfigOC, TIM_CHANNEL_1) != HAL_OK) Error_Handler();
|
||||
|
||||
HAL_TIM_MspPostInit(&htim2);
|
||||
}
|
||||
|
||||
void HAL_TIM_PWM_PulseFinishedCallback(TIM_HandleTypeDef *htim)
|
||||
{
|
||||
if (htim->Instance == TIM2) {
|
||||
HAL_TIM_PWM_Stop_DMA(&htim2, TIM_CHANNEL_1);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/* TIM3 init function */
|
||||
void MX_TIM3_Init(void)
|
||||
{
|
||||
|
||||
Reference in New Issue
Block a user