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

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define BUZZER_PIN 25
#define BUTTON_PIN 15

// Project Data
const String version = "v1.9.5";
const String author = "AndiBond";

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);
Preferences preferences;

// --- BITMAPS ---
// Dino 16x16
const unsigned char dino_frame1[] PROGMEM = { 0x00, 0x00, 0x07, 0xf0, 0x07, 0xf8, 0x07, 0xfc, 0x07, 0xfc, 0x07, 0xf0, 0x07, 0xf0, 0x47, 0xe0, 0x47, 0xe0, 0xc7, 0xe0, 0xff, 0xe0, 0xff, 0xe0, 0x7f, 0xe0, 0x3f, 0xe0, 0x1c, 0x20, 0x18, 0x00 };
const unsigned char dino_frame2[] PROGMEM = { 0x00, 0x00, 0x07, 0xf0, 0x07, 0xf8, 0x07, 0xfc, 0x07, 0xfc, 0x07, 0xf0, 0x07, 0xf0, 0x47, 0xe0, 0x47, 0xe0, 0xc7, 0xe0, 0xff, 0xe0, 0xff, 0xe0, 0x7f, 0xe0, 0x3f, 0xe0, 0x08, 0x70, 0x00, 0x30 };

// SMALL CACTUS 8x10
const unsigned char cactus_mini_bmp[] PROGMEM = {
  0x18, 0x18, 0x18, 0x98, 0xdb, 0xff, 0xdb, 0x18, 0x18, 0x18
};

// Heart 8x7
const unsigned char heart_bonus_bmp[] PROGMEM = { 0x66, 0xff, 0xff, 0xff, 0x7e, 0x3c, 0x18 };

// Game parameters
int dinoY = 39;
int velocity = 0, gravity = 2;
bool isJumping = false, legToggle = false;
int score = 0, highScore = 0, lives = 3;
unsigned long invincibilityUntil = 0;

// Obstacles
int obstacleX = 128, obstacleY = 39;

// Bonuses
int bonusX = -20, bonusY = 30;
int heartsCollected = 0;
int level = 1, gameSpeed = 7;

void setup() {
  pinMode(BUTTON_PIN, INPUT_PULLUP);
  pinMode(BUZZER_PIN, OUTPUT);
  randomSeed(analogRead(0)); 
  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) { for(;;); }
  preferences.begin("dino-game", false);
  highScore = preferences.getInt("highscore", 0);
  showStartScreen();
}

void loop() {
  if (digitalRead(BUTTON_PIN) == LOW && !isJumping) {
    velocity = -12; isJumping = true; tone(BUZZER_PIN, 600, 80);
  }

  if (isJumping) {
    dinoY += velocity; velocity += gravity;
    if (dinoY >= 39) { dinoY = 39; isJumping = false; }
  } else { legToggle = !legToggle; }

  level = (score / 10) + 1;
  gameSpeed = 6 + level;
  obstacleX -= gameSpeed;
  bonusX -= gameSpeed;

  if (obstacleX < -20) {
    score++;
    obstacleX = 128 + random(0, 40);
    if (level >= 2) {
      int r = random(0, 10);
      if (r < 3) obstacleY = 25;
      else if (r < 5) obstacleY = 35;
      else obstacleY = 45;
    } else {
      obstacleY = 45;
    }

    if (score % 5 == 0 && bonusX < -10) {
      bonusX = obstacleX + 60;
      bonusY = random(20, 40);
    }
  }

  if (bonusX > 15 && bonusX < 35 && dinoY < bonusY + 7 && dinoY + 16 > bonusY) {
    bonusX = -20;
    heartsCollected++;
    tone(BUZZER_PIN, 1500, 50);
    if (heartsCollected >= 5) {
      heartsCollected = 0;
      if (lives < 3) { lives++; tone(BUZZER_PIN, 2000, 150); }
    }
  }

  if (millis() > invincibilityUntil && obstacleX < 32 && obstacleX + 8 > 20) {
    if (dinoY < obstacleY + 10 && dinoY + 16 > obstacleY) {
      lives--;
      tone(BUZZER_PIN, 150, 300);
      invincibilityUntil = millis() + 1000;
      obstacleX = 140;
      if (lives <= 0) handleGameOver();
    }
  }

  drawGame();
  delay(35); 
}

void drawGame() {
  display.clearDisplay();
  display.setTextColor(SSD1306_WHITE);
  display.setTextSize(1); // Принудительно ставим стандартный четкий шрифт
  
  // Статус-бар (Координаты подправлены для четкости)
  display.setCursor(0, 0);  display.print("HP:"); display.print(lives);
  display.setCursor(32, 0); display.print("H:");  display.print(heartsCollected);
  display.setCursor(62, 0); display.print("S:");  display.print(score);
  display.setCursor(94, 0); display.print("HI:"); display.print(highScore);
  
  display.drawLine(0, 10, 128, 10, SSD1306_WHITE); // Линия под баром
  display.drawLine(0, 55, 128, 55, SSD1306_WHITE); // Земля

  if (bonusX > -10) display.drawBitmap(bonusX, bonusY, heart_bonus_bmp, 8, 7, SSD1306_WHITE);

  // Дино
  if (millis() > invincibilityUntil || (millis() / 100) % 2 == 0) {
    display.drawBitmap(20, dinoY, (isJumping ? dino_frame1 : (legToggle ? dino_frame1 : dino_frame2)), 16, 16, SSD1306_WHITE);
  }

  // Маленький кактус 8x10
  display.drawBitmap(obstacleX, obstacleY, cactus_mini_bmp, 8, 10, SSD1306_WHITE);
  
  display.display();
}

void handleGameOver() {
  for(int i = 0; i < 10; i++) { display.setRotation(random(0, 4)); display.display(); delay(20); }
  display.setRotation(0);
  if (score > highScore) { highScore = score; preferences.putInt("highscore", score); }
  
  display.clearDisplay();
  display.setTextColor(SSD1306_WHITE);
  display.setTextSize(2); // Крупный шрифт для заголовка
  display.setCursor(10, 5); display.print("GAME OVER");
  
  display.setTextSize(1); // Возвращаемся к мелкому для статистики
  display.setCursor(30, 30); display.print("Final Score: "); display.print(score);
  display.setCursor(30, 42); display.print("Best Score:  "); display.print(highScore);
  display.setCursor(30, 54); display.print("v"); display.print(version);
  display.display();
  
  delay(3000);
  lives = 3; score = 0; heartsCollected = 0; obstacleX = 128; dinoY = 39;
  showStartScreen();
}

void showStartScreen() {
  display.clearDisplay();
  display.setTextColor(SSD1306_WHITE);
  display.setTextSize(1);
  display.setCursor(20, 10); display.print("DINO RUN "); display.print(version);
  display.setCursor(20, 25); display.print("Created by:");
  display.setCursor(20, 35); display.print(author);
  display.setCursor(20, 50); display.print("PRESS BUTTON");
  display.display();
  while(digitalRead(BUTTON_PIN) == HIGH) delay(10);
  tone(BUZZER_PIN, 800, 100); delay(200);
}