#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <math.h>

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET    -1
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

// Animation control
unsigned long lastSwitch = 0;
int animIndex = 0;
const int animCount = 5;
const int titleDuration = 4000;
const int animDuration = 5000;

// Animation variables
int scrollX = SCREEN_WIDTH;
int ballX = 0, ballY = 0, dx = 2, dy = 2;
int wavePhase = 0;
int starY = SCREEN_HEIGHT;
int pulseSize = 0;
bool pulseGrow = true;

bool showingTitle = true;
unsigned long titleStart = 0;

void setup() {
  Wire.begin();
  display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
  display.clearDisplay();
  display.display();
  titleStart = millis();
}

void loop() {
  unsigned long now = millis();

  if (animIndex >= animCount) {
    showSignature();
    return;
  }

  if (showingTitle && now - titleStart < titleDuration) {
    showTitle(animIndex);
    return;
  }

  if (showingTitle && now - titleStart >= titleDuration) {
    showingTitle = false;
    lastSwitch = now;
    resetAnimation(animIndex);
  }

  if (!showingTitle && now - lastSwitch > animDuration) {
    animIndex++;
    showingTitle = true;
    titleStart = millis();
    display.clearDisplay();
  }

  if (!showingTitle) {
    switch (animIndex) {
      case 0: animateScrollingText(); break;
      case 1: animateBouncingBall(); break;
      case 2: animateWave(); break;
      case 3: animateFallingStar(); break;
      case 4: animatePulseCircle(); break;
    }
  }

  delay(30);
}

// ------------------ Title Screen ------------------
void showTitle(int index) {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(10, 25);
  switch (index) {
    case 0: display.println(" Scrolling Text"); break;
    case 1: display.println(" Bouncing Ball"); break;
    case 2: display.println(" Sine Wave"); break;
    case 3: display.println(" Falling Star"); break;
    case 4: display.println(" Pulsing Circle"); break;
  }
  display.display();
}

// ------------------ Final Signature ------------------
void showSignature() {
  display.clearDisplay();
  display.setTextSize(2);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(30, 25);
  display.println("Arvind");
  display.display();
}

// ------------------ Reset Animation Variables ------------------
void resetAnimation(int index) {
  scrollX = SCREEN_WIDTH;
  ballX = 0; ballY = 0; dx = 2; dy = 2;
  wavePhase = 0;
  starY = SCREEN_HEIGHT;
  pulseSize = 0;
  pulseGrow = true;
}

// ------------------ Animation 1: Scrolling Text ------------------
void animateScrollingText() {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(scrollX, 30);
  display.println("Enper Education");
  display.display();
  scrollX -= 2;
  if (scrollX < -100) scrollX = SCREEN_WIDTH;
}

// ------------------ Animation 2: Bouncing Ball ------------------
void animateBouncingBall() {
  display.clearDisplay();
  display.fillCircle(ballX, ballY, 5, SSD1306_WHITE);
  display.display();

  ballX += dx;
  ballY += dy;

  if (ballX <= 0 || ballX >= SCREEN_WIDTH - 5) dx = -dx;
  if (ballY <= 0 || ballY >= SCREEN_HEIGHT - 5) dy = -dy;
}

// ------------------ Animation 3: Sine Wave ------------------
void animateWave() {
  display.clearDisplay();
  for (int x = 0; x < SCREEN_WIDTH; x++) {
    int y = 32 + 20 * sin((x + wavePhase) * 0.1);
    display.drawPixel(x, y, SSD1306_WHITE);
  }
  wavePhase += 2;
  display.display();
}

// ------------------ Animation 4: Falling Star ------------------
void animateFallingStar() {
  display.clearDisplay();
  display.fillTriangle(64, starY, 60, starY + 10, 68, starY + 10, SSD1306_WHITE);
  display.display();
  starY -= 2;
  if (starY < -10) starY = SCREEN_HEIGHT;
}

// ------------------ Animation 5: Pulsing Circle ------------------
void animatePulseCircle() {
  display.clearDisplay();
  display.fillCircle(SCREEN_WIDTH / 2, SCREEN_HEIGHT / 2, pulseSize, SSD1306_WHITE);
  display.display();

  if (pulseGrow) {
    pulseSize++;
    if (pulseSize > 20) pulseGrow = false;
  } else {
    pulseSize--;
    if (pulseSize < 2) pulseGrow = true;
  }
}