/****************************** File: ws2812.c *********************************
 * Author      : wizard2001     22.03.2023
 * Version     : V1.0.0
 * Settings    : MounRiver Studio V1.84 MCU CH32V003F4P6 CH32F003_Board  SSOP20
 * Description : Control WS2812 Chain over GPIO
 *               PC.1 - WS2812 Data
 *               PC.2 - LED1 OnBoard
 *               PC.4 - LED2 OnBoard
  * Updates     :
 ******************************************************************************/
//--------------------------- Includes
#include "debug.h"
#include "ws2812.h"


//--------------------------- Variable
WS_LED_t WS_LEDa[Max_LED];                                 // aktueller Wert
WS_LED_t WS_LEDs[Max_LED];                                 // Ziel Wert

//--------------------------- Konstanten

//--------------------------- LED GPIO Init
void LED_init(void)                                        // init OnBoard LEDs
{
  GPIO_InitTypeDef GPIO_InitStructure = {0};
  RCC_APB2PeriphClockCmd(LED_GPIO_CLK, ENABLE);

  GPIO_InitStructure.GPIO_Pin   = LED1 | LED2;             // set output PB8-9
  GPIO_InitStructure.GPIO_Mode  = GPIO_Mode_Out_PP;
  GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
  GPIO_Init(LED_PORT, &GPIO_InitStructure);
}

//--------------------------- GPIO Init
void WS2812_IO_init(void)                                  // init GPIOs
{
  GPIO_InitTypeDef GPIO_InitStructure = {0};
  RCC_APB2PeriphClockCmd(WS_GPIO_CLK, ENABLE);

  GPIO_InitStructure.GPIO_Pin = WS_OUT | Dext;             // set output PB8-9
  GPIO_InitStructure.GPIO_Mode = GPIO_Mode_Out_PP;
  GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
  GPIO_Init(WS_PORT, &GPIO_InitStructure);
}

WS_LED_t rnd_color(void)                                   // random values
{
  WS_LED_t WS_rnd;

  WS_rnd.G = (uint8_t)(rand() % MaxLumiTab);
  printf("Green:%d\r\n",WS_rnd.G);
  WS_rnd.R = (uint8_t)(rand() % MaxLumiTab);
  printf("Red  :%d\r\n",WS_rnd.R);
  WS_rnd.B = (uint8_t)(rand() % MaxLumiTab);
  printf("Blue :%d\r\n",WS_rnd.B);
  return(WS_rnd);
}

void WS_CHAIN_init(void)
{
  for (uint8_t i=0; i<Max_LED; i++) {
    WS_LEDa[i] = rnd_color();
    WS_LEDs[i] = rnd_color();
    printf("LED G: %d, R: %d, B: %d\r\n", WS_LEDa[i].G, WS_LEDa[i].R, WS_LEDa[i].B);
  }  // GRB
  for (uint8_t i=0; i<Max_LED; i++) {
    printf("LED G: %d, R: %d, B: %d\r\n", WS_LEDa[i].G, WS_LEDa[i].R, WS_LEDa[i].B);
  }  // GRB
}

void setHUE(uint8_t n, uint8_t color, uint8_t saturation)  // color spectrum 0..255
{                                                          // saturation 0..255
  uint8_t  section;
  uint8_t   offset;
  uint16_t  fade_u;
  uint16_t  fade_d;

  section = color / 86;                                    // section 0..2
  offset  = color % 86;                                    // offset 0..85

  fade_u = (offset * saturation) >> 5;                     // fading up value
  fade_d = (abs(85-offset) * saturation) >> 5;             // fading down

  if( section ) {
    if( section == 1) {                                    // section 1: 85..169
      WS_LEDa[n].R = 0;
      WS_LEDa[n].B = (uint8_t) (fade_u);
      WS_LEDa[n].G = (uint8_t) (fade_d);
    } else {                                               // section 2; 170..255
      WS_LEDa[n].R = (uint8_t) (fade_u);
      WS_LEDa[n].B = (uint8_t) (fade_d);
      WS_LEDa[n].G = 0;
    }
  } else {                                                 // section 0: 00..84
    WS_LEDa[n].R = (uint8_t) (fade_d);
    WS_LEDa[n].B = 0;
    WS_LEDa[n].G = (uint8_t) (fade_u);
  }
}


void WS2812_dimm(void)                                     // alle LEDs GRB
{                                                          // zwischen ZUfallswerten
                                                           // dimmen
  for (uint8_t i=0; i<Max_LED; i++)
  {
    if (WS_LEDa[i].G == WS_LEDs[i].G)
    {
      WS_LEDs[i].G = (uint8_t)(rand());
    }
    if (WS_LEDa[i].G < WS_LEDs[i].G)  WS_LEDa[i].G++;
                                 else WS_LEDa[i].G--;
    if (WS_LEDa[i].R == WS_LEDs[i].R)
    {
      WS_LEDs[i].R = (uint8_t)(rand());
    }
    if (WS_LEDa[i].R < WS_LEDs[i].R)  WS_LEDa[i].R++;
                                 else WS_LEDa[i].R--;
    if (WS_LEDa[i].B == WS_LEDs[i].B)
    {
      WS_LEDs[i].B = (uint8_t)(rand());
    }
    if (WS_LEDa[i].B < WS_LEDs[i].B)  WS_LEDa[i].B++;
                                 else WS_LEDa[i].B--;
  }
}
//----------------------- Write Sequenz, Zeiten gemessen (SYSCLK_FREQ_24MHz_HSE)
//-- Zeiten nach Website light_ws2812      t1-350ns  t2-900ns t3-1250ns
void WS2812_wr(uint8_t nLED)            // t1-330ns  t2-830ns t3-1250ns
{
  uint32_t mask;                                           // Bitmask
  uint32_t color = 0;
  uint8_t  n;

  for (n = 0; n < nLED; n++)
  {
    mask = 0x800000;
    color = (dim_tab[WS_LEDa[n].G] << 16) |
    		(dim_tab[WS_LEDa[n].R] <<  8) |
    		(dim_tab[WS_LEDa[n].B]);
//    printf("Color:%d\r\n",color);
	while (mask)                                           // 24 Bit
    {
      WS_OUT_H;
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      if (color & mask) {WS_OUT_L;}                        // check bit
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      WS_OUT_L;
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      __asm volatile(" nop ");
      mask >>= 1;                                          //  next Bit
    }
  }
}