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

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64

#define OLED_RESET    -1 // Reset pin # (or -1 if sharing Arduino reset pin)

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

#define DVD_WIDTH 12
#define DVD_HEIGHT 8
#define DVD_SPEED 2

int dvdX = 0;
int dvdY = 0;
int dvdDX = DVD_SPEED;
int dvdDY = DVD_SPEED;

void setup() {
  Serial.begin(115200);

  // Initialize display
  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for(;;);
  }

  // Clear the buffer
  display.clearDisplay();
  display.display();
}

void loop() {
  // Clear the display
  display.clearDisplay();

  // Draw the DVD logo at current position
  display.fillRect(dvdX, dvdY, DVD_WIDTH, DVD_HEIGHT, WHITE);
  display.display();

  // Move the DVD logo
  dvdX += dvdDX;
  dvdY += dvdDY;

  // Check for collisions with screen edges
  if (dvdX <= 0 || dvdX >= SCREEN_WIDTH - DVD_WIDTH) {
    dvdDX = -dvdDX;
    dvdX += dvdDX;
  }
  if (dvdY <= 0 || dvdY >= SCREEN_HEIGHT - DVD_HEIGHT) {
    dvdDY = -dvdDY;
    dvdY += dvdDY;
  }

  // Delay to control animation speed
  delay(0);
}