#include "blade.h"

#define LED_PIN 6
#define LED_COUNT 32

CRGB blade[LED_COUNT];
CHSV blade_color = CHSV(144, 255, 216);

#define BLADE_OFF 0
#define BLADE_ON 1
#define BLADE_IGNITE 2
#define BLADE_RETRACT 3
uint8_t state = BLADE_OFF;

#define IGNITE_MS 200
#define RETRACT_MS 400
uint32_t anim_start;

void setup_blade() {
  FastLED.addLeds<WS2812B, LED_PIN, GRB>(blade, LED_COUNT);
}

void draw_blade() {
  int num_lit = LED_COUNT;

  switch(state) {
    case BLADE_OFF:
      return;

    case BLADE_ON:
      if(blade[0] == blade_color) return;
      break;

    case BLADE_IGNITE:
      num_lit = (millis() - anim_start) * LED_COUNT / IGNITE_MS;
      if(num_lit >= LED_COUNT) {
        state = BLADE_ON;
        num_lit = LED_COUNT;
      }
      break;

    case BLADE_RETRACT:
      num_lit = LED_COUNT - (millis() - anim_start) * LED_COUNT / RETRACT_MS;
      if(num_lit <= 0) {
        state = BLADE_OFF;
        num_lit = 0;
      }
      break;
  }

  for(int i = 0; i < num_lit; i++) {
    blade[i] = CHSV(blade_color.h - i, blade_color.s - i - i, blade_color.v);
  }
  for(int i = num_lit; i < LED_COUNT; i++) {
    blade[i] = CRGB::Black;
  }
  FastLED.show();
}

// good blade colors
// skywalker blue (136, 255, 216)
// blue-white (144, 128, 216)

void change_hue(uint8_t d) {
  blade_color.h += d;
}

void change_sat(uint8_t d) {
  blade_color.s += d;
}

void toggle_blade() {
  if(state == BLADE_OFF) {
    state = BLADE_IGNITE;
    anim_start = millis();
  } else {
    state = BLADE_RETRACT;
    anim_start = millis();
  }
}