#include <WS2812.h>
#include <CapacitiveSensor.h>

#define LEDCount 37
#define outputPin 11
WS2812 LED(LEDCount);
cRGB value;
CapacitiveSensor capv_btn1 = CapacitiveSensor(4, 2);
bool capv_btn1_state = false;
int capv_btn1_avg = 0;

void setup() {
  //Serial.begin(230400);
  Serial.begin(115200);
  LED.setOutput(outputPin);
  capv_btn1.set_CS_AutocaL_Millis(0xFFFFFFFF);
}

void loop() {
  //int capv_btn1_v = capv_btn1.capacitiveSensor(1);
  capv_btn1_avg = (capv_btn1.capacitiveSensor(1) + capv_btn1_avg * 7) / 8;
  int capv_btn1_v = capv_btn1_avg;
  if (capv_btn1_v > 60 && !capv_btn1_state) {
    Serial.write(0x17);
    Serial.write(0xff);
    capv_btn1_avg += 600;
    capv_btn1_state = true;
  }
  else if (capv_btn1_v < 25 && capv_btn1_state) {
    Serial.write(0x17);
    Serial.write(0x00);
    capv_btn1_state = false;
  }

  if (Serial.available() > 0) {
    byte command = Serial.read();
    if (command == 0x11) {
      LED.sync();
      //Serial.write(0x1a);
    }
    else if (command == 0x12) {
      while (Serial.available() < 4) {
      }
      int pos = Serial.read();
      value.r = Serial.read();
      value.g = Serial.read();
      value.b = Serial.read();
      LED.set_crgb_at(pos, value);
      //Serial.write(0x1b);
    }
    else {
      //Serial.write(0x1e);
    }
  }
}