#include <GyverOLED.h>
GyverOLED<SSD1306_128x64, OLED_NO_BUFFER> oled;

#include <GyverEncoder.h>
#define CLK 3
#define DT 4
#define SW 2
Encoder enc(CLK, DT, SW, TYPE2);

#include <FastLED.h>
#define LED_PIN 5
#define LED_NUM 3
CRGB leds[LED_NUM];

#include <EEPROM.h>

const byte bitmap_color_noinvert[] PROGMEM = {
  0xFC, 0x06, 0x03, 0x01, 0x01, 0x01, 0x81, 0xC1, 0x61, 0xB1, 0x51, 0x51, 0xD1, 0x91, 0x11, 0x91, 0x51, 0x51, 0xD1, 0xB1, 0x61, 0xC1, 0x81, 0x01, 0x01, 0x01, 0x03, 0x06, 0xFC,
  0xFF, 0x00, 0x00, 0x00, 0xF8, 0x0F, 0x39, 0x68, 0x48, 0x79, 0x03, 0x02, 0x02, 0x01, 0x00, 0x01, 0x03, 0x02, 0x02, 0x79, 0x48, 0x68, 0x39, 0x87, 0xFC, 0x00, 0x00, 0x00, 0xFF,
  0xFF, 0x00, 0x00, 0x00, 0x03, 0x0E, 0x38, 0x60, 0x40, 0xC0, 0x80, 0x80, 0x00, 0x00, 0x00, 0xB8, 0xE8, 0x08, 0x08, 0x08, 0x0C, 0x04, 0x06, 0x03, 0x00, 0x00, 0x00, 0x00, 0xFF,
  0x07, 0x0C, 0x18, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x11, 0x11, 0x11, 0x11, 0x11, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x18, 0x0C, 0x07,
};
const byte bitmap_color_invert[] PROGMEM = {
  0x00, 0xF8, 0xFC, 0xFE, 0xFE, 0xFE, 0x7E, 0x3E, 0x9E, 0x4E, 0xAE, 0xAE, 0x2E, 0x6E, 0xEE, 0x6E, 0xAE, 0xAE, 0x2E, 0x4E, 0x9E, 0x3E, 0x7E, 0xFE, 0xFE, 0xFE, 0xFC, 0xF8, 0x00,
  0x00, 0xFF, 0xFF, 0xFF, 0x07, 0xF0, 0xC6, 0x97, 0xB7, 0x86, 0xFC, 0xFD, 0xFD, 0xFE, 0xFF, 0xFE, 0xFC, 0xFD, 0xFD, 0x86, 0xB7, 0x97, 0xC6, 0x78, 0x03, 0xFF, 0xFF, 0xFF, 0x00,
  0x00, 0xFF, 0xFF, 0xFF, 0xFC, 0xF1, 0xC7, 0x9F, 0xBF, 0x3F, 0x7F, 0x7F, 0xFF, 0xFF, 0xFF, 0x47, 0x17, 0xF7, 0xF7, 0xF7, 0xF3, 0xFB, 0xF9, 0xFC, 0xFF, 0xFF, 0xFF, 0xFF, 0x00,
  0x00, 0x03, 0x07, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0E, 0x0E, 0x0E, 0x0E, 0x0E, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x07, 0x03, 0x00,
};
const byte bitmap_brightness_noinvert[] PROGMEM = {
  0xFC, 0x06, 0x03, 0x01, 0x01, 0x01, 0x01, 0x81, 0x81, 0x01, 0x01, 0x01, 0x01, 0xE1, 0xF1, 0xE1, 0x01, 0x01, 0x01, 0x01, 0x81, 0x81, 0x01, 0x01, 0x01, 0x01, 0x03, 0x06, 0xFC,
  0xFF, 0x00, 0x00, 0x00, 0x40, 0xE0, 0xE0, 0xE1, 0x43, 0x07, 0xF6, 0xF8, 0xFC, 0xFC, 0xFD, 0x04, 0x04, 0x08, 0xF6, 0x07, 0x43, 0xE1, 0xE0, 0xE0, 0x40, 0x00, 0x00, 0x00, 0xFF,
  0xFF, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 0x38, 0x1C, 0x0D, 0x03, 0x07, 0xE7, 0xF7, 0xE4, 0x04, 0x02, 0x0D, 0x1C, 0x38, 0x30, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xFF,
  0x07, 0x0C, 0x18, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x11, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x18, 0x0C, 0x07,
};
const byte bitmap_brightness_invert[] PROGMEM = {
  0x00, 0xF8, 0xFC, 0xFE, 0xFE, 0xFE, 0xFE, 0x7E, 0x7E, 0xFE, 0xFE, 0xFE, 0xFE, 0x1E, 0x0E, 0x1E, 0xFE, 0xFE, 0xFE, 0xFE, 0x7E, 0x7E, 0xFE, 0xFE, 0xFE, 0xFE, 0xFC, 0xF8, 0x00,
  0x00, 0xFF, 0xFF, 0xFF, 0xBF, 0x1F, 0x1F, 0x1E, 0xBC, 0xF8, 0x09, 0x07, 0x03, 0x03, 0x02, 0xFB, 0xFB, 0xF7, 0x09, 0xF8, 0xBC, 0x1E, 0x1F, 0x1F, 0xBF, 0xFF, 0xFF, 0xFF, 0x00,
  0x00, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xCF, 0xC7, 0xE3, 0xF2, 0xFC, 0xF8, 0x18, 0x08, 0x1B, 0xFB, 0xFD, 0xF2, 0xE3, 0xC7, 0xCF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0x00,
  0x00, 0x03, 0x07, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0E, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x07, 0x03, 0x00,
};
const byte bitmap_modes_noinvert[] PROGMEM = {
  0xFC, 0x06, 0x03, 0x01, 0x01, 0x81, 0x81, 0xC1, 0xE1, 0xE1, 0xE1, 0xC1, 0x81, 0x81, 0x81, 0x81, 0x81, 0x81, 0x81, 0x81, 0x81, 0x81, 0x81, 0x81, 0x81, 0x01, 0x03, 0x06, 0xFC,
  0xFF, 0x00, 0x00, 0x00, 0x41, 0x61, 0x61, 0x63, 0x63, 0x63, 0x63, 0x63, 0x61, 0x61, 0x61, 0x61, 0x61, 0xF1, 0xF9, 0xF9, 0xF9, 0xF1, 0x61, 0x61, 0x20, 0x00, 0x00, 0x00, 0xFF,
  0xFF, 0x00, 0x00, 0x00, 0x10, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x3C, 0x7C, 0x7C, 0x7C, 0x3C, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x08, 0x00, 0x00, 0x00, 0xFF,
  0x07, 0x0C, 0x18, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x10, 0x18, 0x0C, 0x07,
};
const byte bitmap_modes_invert[] PROGMEM = {
  0x00, 0xF8, 0xFC, 0xFE, 0xFE, 0x7E, 0x7E, 0x3E, 0x1E, 0x1E, 0x1E, 0x3E, 0x7E, 0x7E, 0x7E, 0x7E, 0x7E, 0x7E, 0x7E, 0x7E, 0x7E, 0x7E, 0x7E, 0x7E, 0x7E, 0xFE, 0xFC, 0xF8, 0x00,
  0x00, 0xFF, 0xFF, 0xFF, 0xBE, 0x9E, 0x9E, 0x9C, 0x9C, 0x9C, 0x9C, 0x9C, 0x9E, 0x9E, 0x9E, 0x9E, 0x9E, 0x0E, 0x06, 0x06, 0x06, 0x0E, 0x9E, 0x9E, 0xDF, 0xFF, 0xFF, 0xFF, 0x00,
  0x00, 0xFF, 0xFF, 0xFF, 0xEF, 0xE7, 0xE7, 0xE7, 0xE7, 0xE7, 0xE7, 0xC3, 0x83, 0x83, 0x83, 0xC3, 0xE7, 0xE7, 0xE7, 0xE7, 0xE7, 0xE7, 0xE7, 0xE7, 0xF7, 0xFF, 0xFF, 0xFF, 0x00,
  0x00, 0x03, 0x07, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x0F, 0x07, 0x03, 0x00,
};

struct {
  byte color; 
  byte brithness; 
  byte modes;
  byte intensity;
} eepromValues;

byte pointer, brithness_marker, color_marker, intensity_marker = 0;
boolean menu, submenu = false;
byte pointer_submenu = 0;

void setup() {
  oled.init();
  oled.setContrast(255);
  oled.clear();
  oled.setScale(1);

  EEPROM.get(0, eepromValues);
  //EEPROM.put(0, eepromValues);

  FastLED.addLeds <WS2812, LED_PIN, GRB>(leds, LED_NUM);
  //FastLED.setBrightness(25*eepromValues.brithness);
  leds[0].setHSV(eepromValues.color, eepromValues.intensity, 25*eepromValues.brithness);
  leds[1].setHSV(eepromValues.color, eepromValues.intensity, 25*eepromValues.brithness);
  leds[2].setHSV(eepromValues.color, eepromValues.intensity, 25*eepromValues.brithness);
  FastLED.show();
}

void loop() {
  byte marker = pointer;
  enc.tick();
  if (menu && submenu == false && enc.isClick()) {
    submenu = true;
    oled.clear();
    brithness_marker = eepromValues.brithness + 1;
    color_marker = eepromValues.color + 1;
    intensity_marker = eepromValues.intensity + 1;
  };
  if (submenu) {
    switch (marker) {
      case 0:
        if (enc.isClick()) {
          switch (pointer_submenu) {
            case 0: pointer_submenu = 1; break;
            case 1: pointer_submenu = 0; break;
          }
        }
        if (pointer_submenu == 0 && enc.isLeft() && eepromValues.color > 0) {
            eepromValues.color-=5;
        }
        if (pointer_submenu == 0 && enc.isRight()&& eepromValues.color < 255) {
            eepromValues.color+=5;
        }
        if (pointer_submenu == 1 && enc.isLeft() && eepromValues.intensity > 0) {
            eepromValues.intensity-=5;
        }
        if (pointer_submenu == 1 && enc.isRight() && eepromValues.intensity < 255) {
            eepromValues.intensity+=5;
        }
        if (eepromValues.color != color_marker or eepromValues.intensity != intensity_marker) {
          if (eepromValues.color != color_marker) {
            oled.setScale(1);
            oled.setCursorXY(38,1);
            oled.print("Цвет");
            oled.roundRect(8,15,90,25,OLED_STROKE);
            oled.roundRect(12,17,88,23,OLED_CLEAR);
            oled.roundRect(12,18,constrain(14+eepromValues.color*10/35,14,86),22);
            if (eepromValues.color > 99) {
              oled.setCursorXY(91,17);
            } else if (eepromValues.color < 10) {
              oled.setCursorXY(95,17);
            } else {
              oled.setCursorXY(93,17);
            }
            oled.print("  ");
            oled.print(eepromValues.color);
            oled.print("  ");
          }
          if (eepromValues.intensity != intensity_marker) {
            oled.setCursorXY(14,33);
            oled.print("Насыщенность");
            oled.roundRect(8,47,90,57,OLED_STROKE);
            oled.roundRect(12,49,88,55,OLED_CLEAR);
            oled.roundRect(12,50,constrain(14+eepromValues.intensity*10/35,14,86),54); //86
            if (eepromValues.intensity > 99) {
              oled.setCursorXY(91,49);
            } else if (eepromValues.intensity < 10) {
              oled.setCursorXY(95,49);
            } else {
              oled.setCursorXY(93,49);
            }
            oled.print("  ");
            oled.print(eepromValues.intensity);
            oled.print("  ");
          }
            leds[0].setHSV(eepromValues.color, eepromValues.intensity, 25*eepromValues.brithness);
            leds[1].setHSV(eepromValues.color, eepromValues.intensity, 25*eepromValues.brithness);
            leds[2].setHSV(eepromValues.color, eepromValues.intensity, 25*eepromValues.brithness);
            FastLED.show();
            oled.setScale(1);
            color_marker = eepromValues.color;
            intensity_marker = eepromValues.intensity;
        }
        break;
      case 1:
        if (enc.isLeft() && eepromValues.brithness > 0) {
            eepromValues.brithness--;
        }
        if (enc.isRight()&& eepromValues.brithness < 10) {
            eepromValues.brithness++;
        }
        if (eepromValues.brithness != brithness_marker) {
            oled.setScale(2);
            oled.setCursorXY(25,3);
            oled.print("Яркость");
            oled.roundRect(14,30,114,40,OLED_STROKE);
            oled.roundRect(18,33,112,39,OLED_CLEAR); // 18,33,110,37
            if (eepromValues.brithness != 0) {
              oled.roundRect(18,33,10+eepromValues.brithness*10,37); // 18,33,110,37
            }
            oled.setScale(1);
            if (eepromValues.brithness == 10) {
              oled.setCursorXY(48,49);
            } else if (eepromValues.brithness == 0) {
              oled.setCursorXY(52,49);
            } else {
              oled.setCursorXY(50,49);
            }
            oled.print(" ");
            oled.print(eepromValues.brithness*10);
            oled.print("% ");
            leds[0].setHSV(eepromValues.color, eepromValues.intensity, 255*eepromValues.brithness/10);
            leds[1].setHSV(eepromValues.color, eepromValues.intensity, 255*eepromValues.brithness/10);
            leds[2].setHSV(eepromValues.color, eepromValues.intensity, 255*eepromValues.brithness/10);
            FastLED.show();
            brithness_marker = eepromValues.brithness;
        }
        break;
      case 2:
        break;
    }
  }
  if (submenu && enc.isHolded()) {
    submenu = false;
    marker = 1;
    pointer = 0;
    oled.clear();
    EEPROM.put(0, eepromValues);
  }
  if (menu && submenu == false) {
    if (enc.isRight() && pointer != 2) {
      pointer++;
    };
    if (enc.isLeft() && pointer != 0) {
      pointer--;
    };
  } else if (menu == false) {
    enc.isRight();
    enc.isLeft();
  };
  if (enc.isClick() && menu == false) {
    menu = true;
    marker = 1;
  };
  if (enc.isHolded() && submenu == false) {
    menu = false;
    oled.clear();
    pointer = 0;
    marker = 0;
  };
  if (marker != pointer && submenu == false) {
    printText();
    oled.drawBitmap(6, 8, bitmap_color_noinvert, 29, 29, BITMAP_NORMAL, BUF_ADD);
    oled.drawBitmap(49, 8, bitmap_brightness_noinvert, 29, 29, BITMAP_NORMAL, BUF_ADD);
    oled.drawBitmap(92, 8, bitmap_modes_noinvert, 29, 29, BITMAP_NORMAL, BUF_ADD);
    switch (pointer) {
      case 0:
        oled.drawBitmap(6, 8, bitmap_color_invert, 29, 29, BITMAP_NORMAL, BUF_ADD);
        break;
      case 1:
        oled.drawBitmap(49, 8, bitmap_brightness_invert, 29, 29, BITMAP_NORMAL, BUF_ADD);
        break;
      case 2:
        oled.drawBitmap(92, 8, bitmap_modes_invert, 29, 29, BITMAP_NORMAL, BUF_ADD);
        break;
    };
  };
};

void printText() {
  oled.setCursorXY(10, 43);
  oled.print("Цвет");
  oled.setCursorXY(44, 43);
  oled.print("Яркость");
  oled.setCursorXY(92, 43);
  oled.print("Режим");
};