// Pin definitions
const int led1Pin = 2;
const int led2Pin = 3;
const int led3Pin = 4;
const int potPin = A0;

// Shift register pins for 7-segment display 1 (tens digit)
const int sr1DataPin = 8;
const int sr1ClockPin = 9;
const int sr1LatchPin = 10;

// Shift register pins for 7-segment display 2 (units digit)
const int sr2DataPin = 11;
const int sr2ClockPin = 12;
const int sr2LatchPin = 13;

// Segment patterns for digits 0-9 (for common anode, 0 = ON, 1 = OFF)
// Segments: A, B, C, D, E, F, G, DP (in that order)
const byte digitPatterns[10] = {
  B00000011,  // 0: All segments except G and DP
  B10011111,  // 1: B, C only
  B00100101,  // 2: A, B, D, E, G
  B00001101,  // 3: A, B, C, D, G
  B10011001,  // 4: B, C, F, G
  B01001001,  // 5: A, C, D, F, G
  B01000001,  // 6: A, C, D, E, F, G
  B00011111,  // 7: A, B, C
  B00000001,  // 8: All segments except DP
  B00001001   // 9: A, B, C, D, F, G
};

int lastMappedValue = -1;
unsigned long lastChangeTime = 0;
const unsigned long minStateTime = 2000;  // 2 seconds in milliseconds

void setup() {
  // Initialize LED pins as outputs
  pinMode(led1Pin, OUTPUT);
  pinMode(led2Pin, OUTPUT);
  pinMode(led3Pin, OUTPUT);
  
  // Initialize shift register pins as outputs
  pinMode(sr1DataPin, OUTPUT);
  pinMode(sr1ClockPin, OUTPUT);
  pinMode(sr1LatchPin, OUTPUT);
  pinMode(sr2DataPin, OUTPUT);
  pinMode(sr2ClockPin, OUTPUT);
  pinMode(sr2LatchPin, OUTPUT);
  
  // Initialize all LEDs to OFF
  digitalWrite(led1Pin, LOW);
  digitalWrite(led2Pin, LOW);
  digitalWrite(led3Pin, LOW);
  
  // Display "00" on the 7-segment displays
  displayDigit(0, sr1DataPin, sr1ClockPin, sr1LatchPin);  // Tens digit
  displayDigit(0, sr2DataPin, sr2ClockPin, sr2LatchPin);  // Units digit
}

void loop() {
  // Read the potentiometer value
  int potValue = analogRead(potPin);
  
  // Map the potentiometer value (0-1023) to a range of 0-99
  int mappedValue = map(potValue, 0, 1023, 0, 99);
  
  // Check if the value has changed and if enough time has passed since the last change
  if (mappedValue != lastMappedValue && (millis() - lastChangeTime >= minStateTime || lastMappedValue == -1)) {
    lastMappedValue = mappedValue;
    lastChangeTime = millis();
    
    // Update the 7-segment displays
    int tensDigit = mappedValue / 10;
    int unitsDigit = mappedValue % 10;
    displayDigit(tensDigit, sr1DataPin, sr1ClockPin, sr1LatchPin);  // Tens digit
    displayDigit(unitsDigit, sr2DataPin, sr2ClockPin, sr2LatchPin);  // Units digit
    
    // Update the LEDs based on the mapped value
    if (mappedValue >= 0 && mappedValue <= 33) {
      digitalWrite(led1Pin, HIGH);
      digitalWrite(led2Pin, LOW);
      digitalWrite(led3Pin, LOW);
    } else if (mappedValue >= 34 && mappedValue <= 66) {
      digitalWrite(led1Pin, HIGH);
      digitalWrite(led2Pin, HIGH);
      digitalWrite(led3Pin, LOW);
    } else if (mappedValue >= 67 && mappedValue <= 99) {
      digitalWrite(led1Pin, HIGH);
      digitalWrite(led2Pin, HIGH);
      digitalWrite(led3Pin, HIGH);
    }
  }
}

void displayDigit(int digit, int dataPin, int clockPin, int latchPin) {
  // Ensure digit is between 0 and 9
  if (digit < 0) digit = 0;
  if (digit > 9) digit = 9;
  
  // Get the pattern for the digit
  byte pattern = digitPatterns[digit];
  
  // Send the pattern to the shift register
  digitalWrite(latchPin, LOW);
  shiftOut(dataPin, clockPin, MSBFIRST, pattern);
  digitalWrite(latchPin, HIGH);
}