#include <Servo.h>

// Define pin assignments
// LEDs
const int led1Pin = 8;
const int led2Pin = 9;
const int led3Pin = 10;

// Servos
const int servo1Pin = 5;
const int servo2Pin = 6;
const int servo3Pin = 7;

// Bar Graph - using analog pins
const int barGraph1 = A1;
const int barGraph2 = A2;
const int barGraph3 = A3;
const int barGraph4 = A4;
const int barGraph5 = A5;
const int barGraph6 = A6;
const int barGraph7 = A7;
const int barGraph8 = A8;

// Create servo objects
Servo servo1;
Servo servo2;
Servo servo3;

// State tracking
int currentState = 0;
unsigned long stateStartTime = 0;
const unsigned long stateDuration = 2000; // 2 seconds in milliseconds

void setup() {
  // Initialize LEDs
  pinMode(led1Pin, OUTPUT);
  pinMode(led2Pin, OUTPUT);
  pinMode(led3Pin, OUTPUT);
  
  // Initialize Bar Graph
  pinMode(barGraph1, OUTPUT);
  pinMode(barGraph2, OUTPUT);
  pinMode(barGraph3, OUTPUT);
  pinMode(barGraph4, OUTPUT);
  pinMode(barGraph5, OUTPUT);
  pinMode(barGraph6, OUTPUT);
  pinMode(barGraph7, OUTPUT);
  pinMode(barGraph8, OUTPUT);
  
  // Initialize Servos
  servo1.attach(servo1Pin);
  servo2.attach(servo2Pin);
  servo3.attach(servo3Pin);
  
  // Set initial state
  resetAll();
  
  // Start the state timing
  stateStartTime = millis();
}

void loop() {
  // Check if it's time to move to the next state
  if (millis() - stateStartTime >= stateDuration) {
    // Update state and reset timer
    currentState = (currentState + 1) % 4; // 4 states (0-3)
    stateStartTime = millis();
    
    // Execute the new state
    switch (currentState) {
      case 0: // Initialization/Reset state
        resetAll();
        break;
      case 1: // State 1
        setState1();
        break;
      case 2: // State 2
        setState2();
        break;
      case 3: // State 3
        setState3();
        break;
    }
  }
}

// Function to reset all components
void resetAll() {
  // Reset Servos
  servo1.write(0);
  servo2.write(0);
  servo3.write(0);
  
  // Turn off all LEDs
  digitalWrite(led1Pin, LOW);
  digitalWrite(led2Pin, LOW);
  digitalWrite(led3Pin, LOW);
  
  // Turn off all bar graph segments
  digitalWrite(barGraph1, LOW);
  digitalWrite(barGraph2, LOW);
  digitalWrite(barGraph3, LOW);
  digitalWrite(barGraph4, LOW);
  digitalWrite(barGraph5, LOW);
  digitalWrite(barGraph6, LOW);
  digitalWrite(barGraph7, LOW);
  digitalWrite(barGraph8, LOW);
}

// Function for State 1
void setState1() {
  // Reset all components first to ensure clean state
  resetAll();
  
  // Move Servo1 to 60°
  servo1.write(60);
  
  // Turn on LED1
  digitalWrite(led1Pin, HIGH);
  
  // Light up first two segments of bar graph
  digitalWrite(barGraph1, HIGH);
  digitalWrite(barGraph2, HIGH);
}

// Function for State 2
void setState2() {
  // Reset all components first to ensure clean state
  resetAll();
  
  // Move Servo2 to 120°
  servo2.write(120);
  
  // Turn on LED2
  digitalWrite(led2Pin, HIGH);
  
  // Light up next three segments of bar graph
  digitalWrite(barGraph3, HIGH);
  digitalWrite(barGraph4, HIGH);
  digitalWrite(barGraph5, HIGH);
}

// Function for State 3
void setState3() {
  // Reset all components first to ensure clean state
  resetAll();
  
  // Move Servo3 to 180°
  servo3.write(180);
  
  // Turn on LED3
  digitalWrite(led3Pin, HIGH);
  
  // Light up final three segments of bar graph
  digitalWrite(barGraph6, HIGH);
  digitalWrite(barGraph7, HIGH);
  digitalWrite(barGraph8, HIGH);
}