search

hey-frnk / STM32_HAL_NeoPixel

SK6812 RGBW NeoPixel using STM32 HAL on NUCLEO-F042K6 STM32F042K6 STM32F0
44 stars 11 forks source link

Need help porting for STM32F407 #7

Open krupis opened 9 months ago

krupis commented 9 months ago

Hello. I am trying to port the library to the STM32F407 MCU. I use WS2812B neopixel LED's.

See my full source code:

/* USER CODE BEGIN Header */
/**
 ******************************************************************************
 * @file           : main.c
 * @brief          : Main program body
 ******************************************************************************
 * @attention
 *
 * Copyright (c) 2024 STMicroelectronics.
 * All rights reserved.
 *
 * This software is licensed under terms that can be found in the LICENSE file
 * in the root directory of this software component.
 * If no LICENSE file comes with this software, it is provided AS-IS.
 *
 ******************************************************************************
 */
/* USER CODE END Header */
/* Includes ------------------------------------------------------------------*/
#include "main.h"

/* Private includes ----------------------------------------------------------*/
/* USER CODE BEGIN Includes */

#ifdef __GNUC__
/* With GCC/RAISONANCE, small printf (option LD Linker->Libraries->Small printf
 set to 'Yes') calls __io_putchar() */
#define PUTCHAR_PROTOTYPE int __io_putchar(int ch)
#else
#define PUTCHAR_PROTOTYPE int fputc(int ch, FILE *f)
#endif /* __GNUC__ */

/* USER CODE END Includes */

/* Private typedef -----------------------------------------------------------*/
/* USER CODE BEGIN PTD */

/* USER CODE END PTD */

/* Private define ------------------------------------------------------------*/
/* USER CODE BEGIN PD */

/* USER CODE END PD */

/* Private macro -------------------------------------------------------------*/
/* USER CODE BEGIN PM */

/* USER CODE END PM */

/* Private variables ---------------------------------------------------------*/
TIM_HandleTypeDef htim2;
DMA_HandleTypeDef hdma_tim2_ch1;

UART_HandleTypeDef huart3;

/* USER CODE BEGIN PV */

PUTCHAR_PROTOTYPE {
    HAL_UART_Transmit(&huart3, (uint8_t*) &ch, 1, 0xFFFF);

    return ch;
}

#define PWM_HI (38)
#define PWM_LO (19)

// LED parameters
#define NUM_BPP (3)
#define NUM_PIXELS (2)
#define NUM_BYTES (NUM_BPP * NUM_PIXELS)

// LED color buffer
uint8_t rgb_arr[NUM_BYTES] = { 0 };

// LED write buffer
#define WR_BUF_LEN (NUM_BPP * 8 * 2)
uint8_t wr_buf[WR_BUF_LEN] = { 0 };
uint_fast8_t wr_buf_p = 0;

static void led_render() {
    if (wr_buf_p != 0 || hdma_tim2_ch1.State != HAL_DMA_STATE_READY) {
        // Ongoing transfer, cancel!
        for (uint8_t i = 0; i < WR_BUF_LEN; ++i)
            wr_buf[i] = 0;
        wr_buf_p = 0;
        HAL_TIM_PWM_Stop_DMA(&htim2, TIM_CHANNEL_1);
        return;
    }
    // Ooh boi the first data buffer half (and the second!)
    for (uint_fast8_t i = 0; i < 8; ++i) {
        wr_buf[i] = PWM_LO << (((rgb_arr[0] << i) & 0x80) > 0);
        wr_buf[i + 8] = PWM_LO << (((rgb_arr[1] << i) & 0x80) > 0);
        wr_buf[i + 16] = PWM_LO << (((rgb_arr[2] << i) & 0x80) > 0);

        wr_buf[i + 24] = PWM_LO << (((rgb_arr[3] << i) & 0x80) > 0);
        wr_buf[i + 32] = PWM_LO << (((rgb_arr[4] << i) & 0x80) > 0);
        wr_buf[i + 40] = PWM_LO << (((rgb_arr[5] << i) & 0x80) > 0);
    }

    HAL_TIM_PWM_Start_DMA(&htim2, TIM_CHANNEL_1, (uint32_t*) wr_buf,WR_BUF_LEN);
    wr_buf_p = 2; // Since we're ready for the next buffer
}

static inline uint8_t scale8(uint8_t x, uint8_t scale) {
    return ((uint16_t) x * scale) >> 8;
}

static void led_set_RGB(uint8_t index, uint8_t r, uint8_t g, uint8_t b) {
    rgb_arr[3 * index] = scale8(g, 0xB0); // g;
    rgb_arr[3 * index + 1] = r;
    rgb_arr[3 * index + 2] = scale8(b, 0xF0); // b;
}

/* USER CODE END PV */

/* Private function prototypes -----------------------------------------------*/
void SystemClock_Config(void);
static void MX_GPIO_Init(void);
static void MX_DMA_Init(void);
static void MX_USART3_UART_Init(void);
static void MX_TIM2_Init(void);
/* USER CODE BEGIN PFP */

/* USER CODE END PFP */

/* Private user code ---------------------------------------------------------*/
/* USER CODE BEGIN 0 */

/* USER CODE END 0 */

/**
  * @brief  The application entry point.
  * @retval int
  */
int main(void)
{
  /* USER CODE BEGIN 1 */

  /* USER CODE END 1 */

  /* MCU Configuration--------------------------------------------------------*/

  /* Reset of all peripherals, Initializes the Flash interface and the Systick. */
  HAL_Init();

  /* USER CODE BEGIN Init */

  /* USER CODE END Init */

  /* Configure the system clock */
  SystemClock_Config();

  /* USER CODE BEGIN SysInit */

  /* USER CODE END SysInit */

  /* Initialize all configured peripherals */
  MX_GPIO_Init();
  MX_DMA_Init();
  MX_USART3_UART_Init();
  MX_TIM2_Init();
  /* USER CODE BEGIN 2 */

  /* USER CODE END 2 */

  /* Infinite loop */
  /* USER CODE BEGIN WHILE */
    while (1) {
    /* USER CODE END WHILE */

    /* USER CODE BEGIN 3 */
        led_set_RGB(0, 250, 0, 0);
        led_set_RGB(1, 0, 100, 0);
        led_render();
        HAL_Delay(1000);

    }
  /* USER CODE END 3 */
}

/**
  * @brief System Clock Configuration
  * @retval None
  */
void SystemClock_Config(void)
{
  RCC_OscInitTypeDef RCC_OscInitStruct = {0};
  RCC_ClkInitTypeDef RCC_ClkInitStruct = {0};

  /** Configure the main internal regulator output voltage
  */
  __HAL_RCC_PWR_CLK_ENABLE();
  __HAL_PWR_VOLTAGESCALING_CONFIG(PWR_REGULATOR_VOLTAGE_SCALE1);

  /** Initializes the RCC Oscillators according to the specified parameters
  * in the RCC_OscInitTypeDef structure.
  */
  RCC_OscInitStruct.OscillatorType = RCC_OSCILLATORTYPE_HSI;
  RCC_OscInitStruct.HSIState = RCC_HSI_ON;
  RCC_OscInitStruct.HSICalibrationValue = RCC_HSICALIBRATION_DEFAULT;
  RCC_OscInitStruct.PLL.PLLState = RCC_PLL_ON;
  RCC_OscInitStruct.PLL.PLLSource = RCC_PLLSOURCE_HSI;
  RCC_OscInitStruct.PLL.PLLM = 8;
  RCC_OscInitStruct.PLL.PLLN = 96;
  RCC_OscInitStruct.PLL.PLLP = RCC_PLLP_DIV2;
  RCC_OscInitStruct.PLL.PLLQ = 4;
  if (HAL_RCC_OscConfig(&RCC_OscInitStruct) != HAL_OK)
  {
    Error_Handler();
  }

  /** Initializes the CPU, AHB and APB buses clocks
  */
  RCC_ClkInitStruct.ClockType = RCC_CLOCKTYPE_HCLK|RCC_CLOCKTYPE_SYSCLK
                              |RCC_CLOCKTYPE_PCLK1|RCC_CLOCKTYPE_PCLK2;
  RCC_ClkInitStruct.SYSCLKSource = RCC_SYSCLKSOURCE_PLLCLK;
  RCC_ClkInitStruct.AHBCLKDivider = RCC_SYSCLK_DIV2;
  RCC_ClkInitStruct.APB1CLKDivider = RCC_HCLK_DIV2;
  RCC_ClkInitStruct.APB2CLKDivider = RCC_HCLK_DIV1;

  if (HAL_RCC_ClockConfig(&RCC_ClkInitStruct, FLASH_LATENCY_1) != HAL_OK)
  {
    Error_Handler();
  }
}

/**
  * @brief TIM2 Initialization Function
  * @param None
  * @retval None
  */
static void MX_TIM2_Init(void)
{

  /* USER CODE BEGIN TIM2_Init 0 */

  /* USER CODE END TIM2_Init 0 */

  TIM_ClockConfigTypeDef sClockSourceConfig = {0};
  TIM_MasterConfigTypeDef sMasterConfig = {0};
  TIM_OC_InitTypeDef sConfigOC = {0};

  /* 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 = 60-1;
  htim2.Init.ClockDivision = TIM_CLOCKDIVISION_DIV1;
  htim2.Init.AutoReloadPreload = TIM_AUTORELOAD_PRELOAD_DISABLE;
  if (HAL_TIM_Base_Init(&htim2) != HAL_OK)
  {
    Error_Handler();
  }
  sClockSourceConfig.ClockSource = TIM_CLOCKSOURCE_INTERNAL;
  if (HAL_TIM_ConfigClockSource(&htim2, &sClockSourceConfig) != HAL_OK)
  {
    Error_Handler();
  }
  if (HAL_TIM_PWM_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_PWM1;
  sConfigOC.Pulse = 0;
  sConfigOC.OCPolarity = TIM_OCPOLARITY_HIGH;
  sConfigOC.OCFastMode = TIM_OCFAST_DISABLE;
  if (HAL_TIM_PWM_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);

}

/**
  * @brief USART3 Initialization Function
  * @param None
  * @retval None
  */
static void MX_USART3_UART_Init(void)
{

  /* USER CODE BEGIN USART3_Init 0 */

  /* USER CODE END USART3_Init 0 */

  /* USER CODE BEGIN USART3_Init 1 */

  /* USER CODE END USART3_Init 1 */
  huart3.Instance = USART3;
  huart3.Init.BaudRate = 115200;
  huart3.Init.WordLength = UART_WORDLENGTH_8B;
  huart3.Init.StopBits = UART_STOPBITS_1;
  huart3.Init.Parity = UART_PARITY_NONE;
  huart3.Init.Mode = UART_MODE_TX_RX;
  huart3.Init.HwFlowCtl = UART_HWCONTROL_NONE;
  huart3.Init.OverSampling = UART_OVERSAMPLING_16;
  if (HAL_UART_Init(&huart3) != HAL_OK)
  {
    Error_Handler();
  }
  /* USER CODE BEGIN USART3_Init 2 */

  /* USER CODE END USART3_Init 2 */

}

/**
  * Enable DMA controller clock
  */
static void MX_DMA_Init(void)
{

  /* DMA controller clock enable */
  __HAL_RCC_DMA1_CLK_ENABLE();

  /* DMA interrupt init */
  /* DMA1_Stream5_IRQn interrupt configuration */
  HAL_NVIC_SetPriority(DMA1_Stream5_IRQn, 0, 0);
  HAL_NVIC_EnableIRQ(DMA1_Stream5_IRQn);

}

/**
  * @brief GPIO Initialization Function
  * @param None
  * @retval None
  */
static void MX_GPIO_Init(void)
{
/* USER CODE BEGIN MX_GPIO_Init_1 */
/* USER CODE END MX_GPIO_Init_1 */

  /* GPIO Ports Clock Enable */
  __HAL_RCC_GPIOA_CLK_ENABLE();
  __HAL_RCC_GPIOD_CLK_ENABLE();

/* USER CODE BEGIN MX_GPIO_Init_2 */
/* USER CODE END MX_GPIO_Init_2 */
}

/* USER CODE BEGIN 4 */

void HAL_TIM_PWM_PulseFinishedHalfCpltCallback(TIM_HandleTypeDef *htim) {
    // DMA buffer set from LED(wr_buf_p) to LED(wr_buf_p + 1)
    if (wr_buf_p < NUM_PIXELS) {
        // We're in. Fill the even buffer
        for (uint_fast8_t i = 0; i < 8; ++i) {
            wr_buf[i] = PWM_LO << (((rgb_arr[3 * wr_buf_p] << i) & 0x80) > 0);
            wr_buf[i + 8] = PWM_LO << (((rgb_arr[3 * wr_buf_p + 1] << i) & 0x80) > 0);
            wr_buf[i + 16] = PWM_LO << (((rgb_arr[3 * wr_buf_p + 2] << i) & 0x80) > 0);
        }
        wr_buf_p++;
    } else if (wr_buf_p < NUM_PIXELS + 2) {
        // Last two transfers are resets. 64 * 1.25 us = 80 us == good enough reset
        // First half reset zero fill
        for (uint8_t i = 0; i < WR_BUF_LEN / 2; ++i){
            wr_buf[i] = 0;
        }
        wr_buf_p++;
    }
}

void HAL_TIM_PWM_PulseFinishedCallback(TIM_HandleTypeDef *htim) {
    // DMA buffer set from LED(wr_buf_p) to LED(wr_buf_p + 1)
    if (wr_buf_p < NUM_PIXELS) {
        // We're in. Fill the odd buffer

        for (uint_fast8_t i = 0; i < 8; ++i) {
            wr_buf[i + 24] = PWM_LO
                    << (((rgb_arr[3 * wr_buf_p] << i) & 0x80) > 0);
            wr_buf[i + 32] = PWM_LO
                    << (((rgb_arr[3 * wr_buf_p + 1] << i) & 0x80) > 0);
            wr_buf[i + 40] = PWM_LO
                    << (((rgb_arr[3 * wr_buf_p + 2] << i) & 0x80) > 0);
        }
        wr_buf_p++;
    } else if (wr_buf_p < NUM_PIXELS + 2) {
        // Second half reset zero fill
        for (uint8_t i = WR_BUF_LEN / 2; i < WR_BUF_LEN; ++i)
            wr_buf[i] = 0;
        ++wr_buf_p;
    } else {
        // We're done. Lean back and until next time!
        wr_buf_p = 0;
        HAL_TIM_PWM_Stop_DMA(&htim2, TIM_CHANNEL_1);
    }
}

/* USER CODE END 4 */

/**
  * @brief  This function is executed in case of error occurrence.
  * @retval None
  */
void Error_Handler(void)
{
  /* USER CODE BEGIN Error_Handler_Debug */
    /* User can add his own implementation to report the HAL error return state */
    __disable_irq();
    while (1) {
    }
  /* USER CODE END Error_Handler_Debug */
}

#ifdef  USE_FULL_ASSERT
/**
  * @brief  Reports the name of the source file and the source line number
  *         where the assert_param error has occurred.
  * @param  file: pointer to the source file name
  * @param  line: assert_param error line source number
  * @retval None
  */
void assert_failed(uint8_t *file, uint32_t line)
{
  /* USER CODE BEGIN 6 */
  /* User can add his own implementation to report the file name and line number,
     ex: printf("Wrong parameters value: file %s on line %d\r\n", file, line) */
  /* USER CODE END 6 */
}
#endif /* USE_FULL_ASSERT */

As you can see in my main.c I am just trying to set LED0 to red and LED1 to green:

        led_set_RGB(0, 250, 0, 0);
        led_set_RGB(1, 0, 100, 0);

My clock setup: image

My TIM2 setup: image

My TIM2 DMA setup: image

I have connected logic analyzer to check how the DATA line looks like and this is the result: image

The above waveform is sent every 1 second which does not make any sense to me. I have even put the breakpoint before the HAL_TIM_PWM_Start_DMA to check how does my wr_buf looks like:

rgb_arr uint8_t [6] 0x20000118  
    rgb_arr[0]  uint8_t 0x0 
    rgb_arr[1]  uint8_t 0xfa    
    rgb_arr[2]  uint8_t 0x0 
    rgb_arr[3]  uint8_t 0x44    
    rgb_arr[4]  uint8_t 0x0 
    rgb_arr[5]  uint8_t 0x0 
wr_buf  uint8_t [48]    0x20000120 <wr_buf> (Default)   
    wr_buf[0]   uint8_t 19 (Decimal)    
    wr_buf[1]   uint8_t 19 (Decimal)    
    wr_buf[2]   uint8_t 19 (Decimal)    
    wr_buf[3]   uint8_t 19 (Decimal)    
    wr_buf[4]   uint8_t 19 (Decimal)    
    wr_buf[5]   uint8_t 19 (Decimal)    
    wr_buf[6]   uint8_t 19 (Decimal)    
    wr_buf[7]   uint8_t 19 (Decimal)    
    wr_buf[8]   uint8_t 38 (Decimal)    
    wr_buf[9]   uint8_t 38 (Decimal)    
    wr_buf[10]  uint8_t 38 (Decimal)    
    wr_buf[11]  uint8_t 38 (Decimal)    
    wr_buf[12]  uint8_t 38 (Decimal)    
    wr_buf[13]  uint8_t 19 (Decimal)    
    wr_buf[14]  uint8_t 38 (Decimal)    
    wr_buf[15]  uint8_t 19 (Decimal)    
    wr_buf[16]  uint8_t 19 (Decimal)    
    wr_buf[17]  uint8_t 19 (Decimal)    
    wr_buf[18]  uint8_t 19 (Decimal)    
    wr_buf[19]  uint8_t 19 (Decimal)    
    wr_buf[20]  uint8_t 19 (Decimal)    
    wr_buf[21]  uint8_t 19 (Decimal)    
    wr_buf[22]  uint8_t 19 (Decimal)    
    wr_buf[23]  uint8_t 19 (Decimal)    
    wr_buf[24]  uint8_t 19 (Decimal)    
    wr_buf[25]  uint8_t 38 (Decimal)    
    wr_buf[26]  uint8_t 19 (Decimal)    
    wr_buf[27]  uint8_t 19 (Decimal)    
    wr_buf[28]  uint8_t 19 (Decimal)    
    wr_buf[29]  uint8_t 38 (Decimal)    
    wr_buf[30]  uint8_t 19 (Decimal)    
    wr_buf[31]  uint8_t 19 (Decimal)    
    wr_buf[32]  uint8_t 19 (Decimal)    
    wr_buf[33]  uint8_t 19 (Decimal)    
    wr_buf[34]  uint8_t 19 (Decimal)    
    wr_buf[35]  uint8_t 19 (Decimal)    
    wr_buf[36]  uint8_t 19 (Decimal)    
    wr_buf[37]  uint8_t 19 (Decimal)    
    wr_buf[38]  uint8_t 19 (Decimal)    
    wr_buf[39]  uint8_t 19 (Decimal)    
    wr_buf[40]  uint8_t 19 (Decimal)    
    wr_buf[41]  uint8_t 19 (Decimal)    
    wr_buf[42]  uint8_t 19 (Decimal)    
    wr_buf[43]  uint8_t 19 (Decimal)    
    wr_buf[44]  uint8_t 19 (Decimal)    
    wr_buf[45]  uint8_t 19 (Decimal)    
    wr_buf[46]  uint8_t 19 (Decimal)    
    wr_buf[47]  uint8_t 19 (Decimal)

But it does not seem to match whatever is being sent as you can see from the waveforms..

nainai323 commented 8 months ago

maybe you need to enable fast mode