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

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

#define BUTTON_PIN 2
#define GROUND_LEVEL 50
#define DINO_WIDTH 10
#define DINO_HEIGHT 15
#define OBSTACLE_WIDTH 5
#define OBSTACLE_HEIGHT 10

int dinoY = GROUND_LEVEL;
int dinoVelocity = 0;
bool jumping = false;
int obstacleX = SCREEN_WIDTH;
int score = 0;

void setup() {
  pinMode(BUTTON_PIN, INPUT_PULLUP);
  Serial.begin(9600);
  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for (;;);
  }
  display.clearDisplay();
  display.display();
}

void loop() {
  display.clearDisplay();

  // Draw ground
  display.drawLine(0, GROUND_LEVEL, SCREEN_WIDTH, GROUND_LEVEL, SSD1306_WHITE);

  // Draw dino
  display.fillRect(10, dinoY - DINO_HEIGHT, DINO_WIDTH, DINO_HEIGHT, SSD1306_WHITE);

  // Draw obstacle
  display.fillRect(obstacleX, GROUND_LEVEL - OBSTACLE_HEIGHT, OBSTACLE_WIDTH, OBSTACLE_HEIGHT, SSD1306_WHITE);

  // Update dino position
  if (jumping) {
    dinoY += dinoVelocity;
    dinoVelocity += 1;
    if (dinoY >= GROUND_LEVEL) {
      dinoY = GROUND_LEVEL;
      jumping = false;
    }
  }

  // Update obstacle position
  obstacleX -= 2;
  if (obstacleX < 0) {
    obstacleX = SCREEN_WIDTH;
    score++;
  }

  // Debug: Print obstacle and dino positions
  Serial.print("Obstacle X: ");
  Serial.print(obstacleX);
  Serial.print(", Dino Y: ");
  Serial.println(dinoY);

  // Check for collision
  if (obstacleX < 10 + DINO_WIDTH && obstacleX + OBSTACLE_WIDTH > 10 && dinoY >= GROUND_LEVEL - OBSTACLE_HEIGHT) {
    Serial.println("Collision detected!");
    gameOver();
  }

  // Display score
  display.setCursor(SCREEN_WIDTH - 30, 10);
  display.print("Score: ");
  display.print(score);

  // Check for jump
  if (digitalRead(BUTTON_PIN) == LOW && !jumping) {
    dinoVelocity = -10;
    jumping = true;
  }

  display.display();
  delay(20);
}

void gameOver() {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(SCREEN_WIDTH / 2 - 30, SCREEN_HEIGHT / 2 - 10);
  display.print("Game Over!");
  display.setCursor(SCREEN_WIDTH / 2 - 30, SCREEN_HEIGHT / 2 + 10);
  display.print("Score: ");
  display.print(score);
  display.display();
  while (true) {
    // Freeze the game
  }
}