// Pin definitions
#define LED_PIN 2
#define RGB_RED_PIN 3
#define RGB_GREEN_PIN 4
#define RGB_BLUE_PIN 5

#define SR1_DATA_PIN 6
#define SR1_CLOCK_PIN 7
#define SR1_LATCH_PIN 8

#define SR2_DATA_PIN 9
#define SR2_CLOCK_PIN 10
#define SR2_LATCH_PIN 11

// Variables for RGB LED colors
const byte COLORS[][3] = {
  {255, 0, 0},    // Red
  {0, 255, 0},    // Green
  {0, 0, 255},    // Blue
  {255, 255, 0},  // Yellow
  {0, 255, 255},  // Cyan
  {255, 0, 255},  // Magenta
  {255, 255, 255} // White
};
const int NUM_COLORS = 7;
int currentColor = 0;

// Variables for 7-segment display
int counter = 0;
// For common anode 7-segment display (0 = ON, 1 = OFF)
const byte SEGMENT_MAP[] = {
  0b11000000,  // 0
  0b11111001,  // 1
  0b10100100,  // 2
  0b10110000,  // 3
  0b10011001,  // 4
  0b10010010,  // 5
  0b10000010,  // 6
  0b11111000,  // 7
  0b10000000,  // 8
  0b10010000   // 9
};

// Timing variables
unsigned long previousMillis = 0;
const long interval = 2000;  // 2 seconds
bool ledState = LOW;  // LED starts in the OFF state

void setup() {
  // Initialize pins
  pinMode(LED_PIN, OUTPUT);
  pinMode(RGB_RED_PIN, OUTPUT);
  pinMode(RGB_GREEN_PIN, OUTPUT);
  pinMode(RGB_BLUE_PIN, OUTPUT);
  
  pinMode(SR1_DATA_PIN, OUTPUT);
  pinMode(SR1_CLOCK_PIN, OUTPUT);
  pinMode(SR1_LATCH_PIN, OUTPUT);
  
  pinMode(SR2_DATA_PIN, OUTPUT);
  pinMode(SR2_CLOCK_PIN, OUTPUT);
  pinMode(SR2_LATCH_PIN, OUTPUT);
  
  // Initialize LED state (starts OFF)
  digitalWrite(LED_PIN, ledState);
  
  // Initialize RGB LED (starts with Red)
  setRGBColor(COLORS[currentColor]);
  
  // Initialize 7-segment displays (starts at 00)
  updateSevenSegment();
}

void loop() {
  unsigned long currentMillis = millis();
  
  if (currentMillis - previousMillis >= interval) {
    previousMillis = currentMillis;
    
    // Toggle LED state (2 seconds on, 2 seconds off)
    ledState = !ledState;
    digitalWrite(LED_PIN, ledState);
    
    // Update RGB LED color
    currentColor = (currentColor + 1) % NUM_COLORS;
    setRGBColor(COLORS[currentColor]);
    
    // Update counter and 7-segment displays
    counter = (counter + 1) % 100;
    updateSevenSegment();
  }
}

void setRGBColor(const byte* color) {
  analogWrite(RGB_RED_PIN, color[0]);
  analogWrite(RGB_GREEN_PIN, color[1]);
  analogWrite(RGB_BLUE_PIN, color[2]);
}

void updateSevenSegment() {
  byte tensDigit = counter / 10;
  byte unitsDigit = counter % 10;
  
  // Update tens digit (sevseg1)
  digitalWrite(SR1_LATCH_PIN, LOW);
  shiftOut(SR1_DATA_PIN, SR1_CLOCK_PIN, MSBFIRST, SEGMENT_MAP[tensDigit]);
  digitalWrite(SR1_LATCH_PIN, HIGH);
  
  // Update units digit (sevseg2)
  digitalWrite(SR2_LATCH_PIN, LOW);
  shiftOut(SR2_DATA_PIN, SR2_CLOCK_PIN, MSBFIRST, SEGMENT_MAP[unitsDigit]);
  digitalWrite(SR2_LATCH_PIN, HIGH);
}