/*
  BlinkRGB

  Demonstrates usage of onboard RGB LED on some ESP dev boards.

  Calling digitalWrite(RGB_BUILTIN, HIGH) will use hidden RGB driver.

  RGBLedWrite demonstrates controll of each channel:
  void neopixelWrite(uint8_t pin, uint8_t red_val, uint8_t green_val, uint8_t blue_val)

  WARNING: After using digitalWrite to drive RGB LED it will be impossible to drive the same pin
    with normal HIGH/LOW level
*/
//#define RGB_BRIGHTNESS 64 // Change white brightness (max 255)

// the setup function runs once when you press reset or power the board

#include <Arduino.h>
//#include "driver/rmt.h"

#define FIRMWARE_VERSION 20240421
#define PRODUCT_NAME "LED-Test"

#define SERIAL_SPEED 115200

void setup() {
  // init serial port
  Serial.begin(SERIAL_SPEED);
  while (!Serial);
  delay(1000);

  // And the monkey flips the switch. (Akiva Goldsman)
  Serial.println();
  Serial.println("*** " + String(PRODUCT_NAME) + " ***");
  Serial.println("Firmware: " + String(FIRMWARE_VERSION));

  // initialize the RGB LED
#ifdef RGB_BUILTIN
  pinMode(RGB_BUILTIN, OUTPUT);
#endif
 
  Serial.println("** setup() done, starting loop()");
 }

// the loop function runs over and over again forever
void loop() {
#ifdef RGB_BUILTIN
  Serial.println("digitalWrite to PIN: " + String(RGB_BUILTIN));
  delay(1000);

  Serial.println("  PIN: HIGH");
  digitalWrite(RGB_BUILTIN, HIGH);   // Turn the RGB LED white
  delay(1000);
 
  Serial.println("  PIN: LOW");
  digitalWrite(RGB_BUILTIN, LOW);    // Turn the RGB LED off
  delay(1000);

  Serial.println("neopixelWrite to PIN: " + String(RGB_BUILTIN));

  Serial.println("  PIN: RED");
  neopixelWrite(RGB_BUILTIN, RGB_BRIGHTNESS, 0, 0); // Red
  delay(1000);
  Serial.println("  PIN: GREEN");
  neopixelWrite(RGB_BUILTIN, 0, RGB_BRIGHTNESS, 0); // Green
  delay(1000);
  Serial.println("  PIN: BLUE");
  neopixelWrite(RGB_BUILTIN, 0, 0, RGB_BRIGHTNESS); // Blue
  delay(1000);
  Serial.println("  PIN: BLACK");
  neopixelWrite(RGB_BUILTIN, 0, 0, 0); // Off / black
  delay(1000);
#endif
}

void neopixelWrite(uint8_t pin, uint8_t red_val, uint8_t green_val, uint8_t blue_val){
  rmt_data_t led_data[24];
  static rmt_obj_t* rmt_send = NULL;
  static bool initialized = false;

  uint8_t _pin = pin;
#ifdef RGB_BUILTIN
  if(pin == RGB_BUILTIN){
    _pin = RGB_BUILTIN-SOC_GPIO_PIN_COUNT;
  }
#endif

  if(!initialized){
    if((rmt_send = rmtInit(_pin, RMT_TX_MODE, RMT_MEM_64)) == NULL){
        log_e("RGB LED driver initialization failed!");
        rmt_send = NULL;
        return;
    }
    rmtSetTick(rmt_send, 100);
    initialized = true;
  }

  int color[] = {red_val, green_val, blue_val};  // Color coding is in order RED, GREEN, BLUE
  int i = 0;
  for(int col=0; col<3; col++ ){
    for(int bit=0; bit<8; bit++){
      if((color[col] & (1<<(7-bit)))){
        // HIGH bit
        led_data[i].level0 = 1; // T1H
        led_data[i].duration0 = 8; // 0.8us
        led_data[i].level1 = 0; // T1L
        led_data[i].duration1 = 4; // 0.4us
      }else{
        // LOW bit
        led_data[i].level0 = 1; // T0H
        led_data[i].duration0 = 4; // 0.4us
        led_data[i].level1 = 0; // T0L
        led_data[i].duration1 = 8; // 0.8us
      }
      i++;
    }
  }
  rmtWrite(rmt_send, led_data, 24);
}