#include <Adafruit_NeoPixel.h>

#define LED_PIN 23
#define LED_COUNT 112

#define BTN_START 13
#define BTN_HOUR 25
#define BTN_MIN 26
#define BTN_RESET 27

Adafruit_NeoPixel strip(LED_COUNT, LED_PIN, NEO_GRB + NEO_KHZ800);

// LAIKAI
int clockSeconds = 12 * 3600;
int stopwatchSeconds = 0;

bool stopwatchRunning = false;

bool colonState = false;
unsigned long lastTick = 0;

// SPALVOS
uint32_t colorClock;
uint32_t colorGreen;
uint32_t colorRed;

#define SEG_SIZE 4

byte digits[10][7] = {
  {1,1,1,1,1,1,0},
  {0,1,1,0,0,0,0},
  {1,1,0,1,1,0,1},
  {1,1,1,1,0,0,1},
  {0,1,1,0,0,1,1},
  {1,0,1,1,0,1,1},
  {1,0,1,1,1,1,1},
  {1,1,1,0,0,0,0},
  {1,1,1,1,1,1,1},
  {1,1,1,1,0,1,1}
};

// ================= SETUP =================
void setup() {

  pinMode(BTN_START, INPUT_PULLUP);
  pinMode(BTN_HOUR, INPUT_PULLUP);
  pinMode(BTN_MIN, INPUT_PULLUP);
  pinMode(BTN_RESET, INPUT_PULLUP);

  strip.begin();
  strip.setBrightness(80);   // 🔥 FIX
  strip.show();

  colorClock = strip.Color(0, 0, 150);
  colorGreen = strip.Color(0, 150, 0);
  colorRed   = strip.Color(150, 0, 0);
}

// ================= CLEAR =================
void clearAll() {
  for (int i = 0; i < LED_COUNT; i++) {
    strip.setPixelColor(i, 0);
  }
}

// ================= SEGMENT =================
void setSegment(int startLED, int segment, bool on, uint32_t color) {
  int offset = segment * SEG_SIZE;

  for (int i = 0; i < SEG_SIZE; i++) {
    strip.setPixelColor(startLED + offset + i, on ? color : 0);
  }
}

// ================= DIGIT =================
void showDigit(int digit, int pos, uint32_t color) {
  int start = pos * 28;

  for (int i = 0; i < 7; i++) {
    setSegment(start, i, digits[digit][i], color);
  }
}

// ================= COLON =================
void drawColon() {
  int base = 56;

  uint32_t c = colonState ? colorClock : 0;

  for (int i = 0; i < 4; i++) {
    strip.setPixelColor(base + i, c);
  }
}

// ================= CLOCK =================
void showClock() {

  int h = (clockSeconds / 3600) % 24;
  int m = (clockSeconds / 60) % 60;

  showDigit(h / 10, 0, colorClock);
  showDigit(h % 10, 1, colorClock);
  showDigit(m / 10, 3, colorClock);
  showDigit(m % 10, 4, colorClock);

  drawColon();
}

// ================= STOPWATCH =================
void showStopwatch() {

  int m = stopwatchSeconds / 60;
  int s = stopwatchSeconds % 60;

  uint32_t col = (stopwatchSeconds < 60) ? colorGreen : colorRed;

  showDigit(m / 10, 0, col);
  showDigit(m % 10, 1, col);
  showDigit(s / 10, 3, col);
  showDigit(s % 10, 4, col);

  drawColon();
}

// ================= LOOP =================
void loop() {

  // MYGTUKAI
  if (digitalRead(BTN_START) == LOW) {
    stopwatchRunning = !stopwatchRunning;
    delay(200);
  }

  if (digitalRead(BTN_RESET) == LOW) {
    stopwatchSeconds = 0;
    stopwatchRunning = false;
    delay(200);
  }

  if (digitalRead(BTN_HOUR) == LOW) {
    clockSeconds += 3600;
    delay(200);
  }

  if (digitalRead(BTN_MIN) == LOW) {
    clockSeconds += 60;
    delay(200);
  }

  // LAIKAS
  if (millis() - lastTick >= 1000) {
    lastTick = millis();

    colonState = !colonState;

    if (stopwatchRunning) {
      stopwatchSeconds++;

      if (stopwatchSeconds >= 600) {
        stopwatchRunning = false;
        stopwatchSeconds = 0;
      }
    } else {
      clockSeconds++;
    }
  }

  // 🔥 FIX: visada clear prieš draw
  clearAll();

  if (stopwatchRunning) showStopwatch();
  else showClock();

  strip.show();
}