#pragma once

#include "../resources.h"
#include "stm32f0xx_hal_gpio.h"
#include <stm32f0xx_hal.h>

using namespace stereo_mix;

const uint8_t kNumChannels = 4;

static const uint16_t kGpioPins[] = { GPIO_PIN_7, GPIO_PIN_15, GPIO_PIN_13, GPIO_PIN_14 };
static GPIO_TypeDef* kGpioPorts[] = { GPIOB, GPIOB, GPIOF, GPIOF };

class Leds {
  public:
  Leds()
  {
    GPIO_InitTypeDef gpioInit;
    for (size_t i = 0; i < kNumChannels; i++) {
      gpioInit.Mode = GPIO_MODE_OUTPUT_PP;
      gpioInit.Pin = kGpioPins[i];
      gpioInit.Pull = GPIO_NOPULL;
      gpioInit.Speed = GPIO_SPEED_FREQ_MEDIUM;
      HAL_GPIO_Init(kGpioPorts[i], &gpioInit);

      intensities[i] = 0;
      blinking[i] = false;
    }
  }
  void Write()
  {
    pwm_counter++;
    pwm_counter &= 0x1ff; // equals to if(pwm_counter > 512) pwm_counter = 0;
    blink_counter++;
    blink_counter &= 0x7FFF;
    for (size_t i = 0; i < kNumChannels; i++) {
      bool in_blink_phase = blink_counter < 16383 || !blinking[i];
      if (intensities[0] && lut_led_gamma[intensities[i]] >= pwm_counter && in_blink_phase) {
        kGpioPorts[i]->BSRR |= kGpioPins[i];
      } else {
        kGpioPorts[i]->BRR |= kGpioPins[i];
      }
    }
  }
  void set_intensity(uint8_t channel, uint8_t intensity)
  {
    if (channel >= kNumChannels)
      return;
    intensities[channel] = intensity;
  }
  void set_blinking(uint8_t channel, bool blink)
  {
    if (channel >= kNumChannels)
      return;
    blinking[channel] = blink;
  }

  private:
  uint16_t pwm_counter;
  uint16_t blink_counter;
  uint8_t intensities[kNumChannels];
  bool blinking[kNumChannels];
};