#include <Wire.h>
#include <RTClib.h>

// --- SETTINGS & CONFIGURATION ---
#define TEST_MODE false       // Set to false to use actual RTC time
#define SERIAL_DEBUG false     // Set to false to disable all Serial functions for final use
float simSpeed = 30.0;        
int fakeHour = 23;           
int fakeMinute = 58;         
float fakeSecond = 0;        

// --- PIN DEFINITIONS ---
#define DIR_PIN 15
#define M_STEP_PIN 2
#define MM_STEP_PIN 0
#define H_STEP_PIN 4
#define HH_STEP_PIN 16
const int Reset_Pin = 33;     
const int power_fail_pin = 14; 
const int STEPS_PER_UNIT = 20;

// --- CONFIGURABLE ANALOG BUTTON THRESHOLDS ---
int b1value = 100;
int b2value = 200;
int b3value = 300;
int b4value = 400; 

// State Tracking
int Reset_Button_value;
int Saved_M = 0, Saved_MM = 0, Saved_H = 0, Saved_HH = 0;
unsigned long lastUpdateMicros = 0;

RTC_DS1307 rtc;

void stepMotor(int pin, int steps) {
  for (int i = 0; i < steps; i++) {
    digitalWrite(pin, HIGH);
    delayMicroseconds(500);
    digitalWrite(pin, LOW);
    delayMicroseconds(500);
  }
}

void setup() {
  if (SERIAL_DEBUG) {
    Serial.begin(115200);
  }

  int outputPins[] = {M_STEP_PIN, MM_STEP_PIN, H_STEP_PIN, HH_STEP_PIN, DIR_PIN};
  for(int pin : outputPins) pinMode(pin, OUTPUT);
  
  pinMode(power_fail_pin, INPUT); 
  digitalWrite(DIR_PIN, HIGH);

  Wire.begin();
  if (!rtc.begin()) {
    if (SERIAL_DEBUG) Serial.println("RTC not found");
  }
  lastUpdateMicros = micros();
}

void loop() {
  // --- 1. POWER LOSS CHECK ---
  if (digitalRead(power_fail_pin) == LOW) {
    if (SERIAL_DEBUG) Serial.println("Power Loss Detected");
    return; 
  }

  // --- 2. ANALOG BUTTON LOGIC ---
  Reset_Button_value = analogRead(Reset_Pin);
  
  if (Reset_Button_value > b1value - 20 && Reset_Button_value < b1value + 20){
    stepMotor(HH_STEP_PIN, 20);
    delay(1000);
  }
  if (Reset_Button_value > b2value - 20 && Reset_Button_value < b2value + 20){
    stepMotor(H_STEP_PIN, 20);
    delay(1000);
  }
  if (Reset_Button_value > b3value - 20 && Reset_Button_value < b3value + 20){
    stepMotor(MM_STEP_PIN, 20);
    delay(1000);
  }
  if (Reset_Button_value > b4value - 20 && Reset_Button_value < b4value + 20){
    stepMotor(M_STEP_PIN, 20);
    delay(1000);
  }

  // --- 3. TIME TRACKING ---
  int currH, currM;
  if (TEST_MODE) {
    unsigned long delta = micros() - lastUpdateMicros;
    lastUpdateMicros = micros();
    fakeSecond += (delta / 1000000.0) * simSpeed;
    if (fakeSecond >= 60.0) { fakeSecond = 0; fakeMinute++; }
    if (fakeMinute >= 60) { fakeMinute = 0; fakeHour++; }
    if (fakeHour >= 24) { fakeHour = 0; }
    currH = fakeHour;
    currM = fakeMinute;
  } else {
    DateTime now = rtc.now();
    currH = now.hour();
    currM = now.minute();
  }

  // Calculate Digits
  int d_M  = currM % 10;
  int d_MM = currM / 10;
  int d_H  = currH % 10;
  int d_HH = currH / 10;

  // --- 4. STEPPER EXECUTION ---
  if (Saved_M != d_M) {
    stepMotor(M_STEP_PIN, STEPS_PER_UNIT);
    Saved_M = (Saved_M + 1) % 10;
  }
  if (Saved_MM != d_MM) {
    int steps = (Saved_MM == 5) ? (STEPS_PER_UNIT * 5) : STEPS_PER_UNIT; 
    stepMotor(MM_STEP_PIN, steps);
    Saved_MM = (Saved_MM + 1) % 6;
  }
  if (Saved_H != currH) {
    int steps = (Saved_H == 23) ? (STEPS_PER_UNIT * 7) : STEPS_PER_UNIT;
    stepMotor(H_STEP_PIN, steps);
    Saved_H = (Saved_H + 1) % 24;
  }
  if (Saved_HH != d_HH) {
    int steps = (Saved_HH == 2) ? (STEPS_PER_UNIT * 8) : STEPS_PER_UNIT;
    stepMotor(HH_STEP_PIN, steps);
    Saved_HH = (Saved_HH + 1) % 3;
  }

  if (SERIAL_DEBUG) {
    Serial.print("Analog Value: ");
    Serial.println(Reset_Button_value);
  }
}