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

#define SCREEN_WIDTH 128 // OLED顯示寬度，單位為像素
#define SCREEN_HEIGHT 64 // OLED顯示高度，單位為像素

// 創(chuàng)建SSD1306對(duì)象伐债，使用SPI接口
#define OLED_RESET     9 // RES引腳連接到D9
#define OLED_DC       8 // DC引腳連接到D8
#define OLED_CS      10 // CS引腳連接到D10（如果有的話）

// 使用 SPI 接口創(chuàng)建 Adafruit_SSD1306 對(duì)象
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &SPI, OLED_DC, OLED_RESET, OLED_CS);

// 貪吃蛇參數(shù)
int snakeLength = 3;
int snakeX[100] = {20, 19, 18}; // 蛇身坐標(biāo)數(shù)組舶替，初始位置為(20,19),(19,19),(18,19)
int snakeY[100] = {10, 10, 10};
int foodX = 5 * 4; // 食物隨機(jī)位置说墨，確保與4對(duì)齊
int foodY = 5 * 4;

char direction = 'R'; // 初始方向?yàn)橛?
// 搖桿引腳定義
const int joyXPin = A0; // X軸引腳連接到A0
const int joyYPin = A1; // Y軸引腳連接到A1
const int buttonPin = 2; // 按鈕引腳連接到D2

void setup() {
    pinMode(buttonPin, INPUT_PULLUP); // 設(shè)置按鈕引腳為輸入并啟用內(nèi)部上拉電阻
    
    display.begin(SSD1306_SWITCHCAPVCC); // 初始化顯示屏俩滥，不再需要I2C地址
    
    display.clearDisplay();
    display.setTextSize(1);
    display.setTextColor(WHITE);
    
    randomSeed(analogRead(A3)); // 用于生成隨機(jī)數(shù)種子
    
    drawFood(); // 繪制食物初始位置
    
}

void loop() {

    readJoystick(); // 檢測(cè)搖桿狀態(tài)
    
    if (checkCollision()) {
        gameOver();
        return;
    }
    
    moveSnake();
    
    if (checkCollision()) { 
        gameOver(); 
        return; 
    }
    
    display.clearDisplay();

    drawBorder();   // 繪制邊框
    drawSnake();    // 繪制蛇身

    if ((snakeX[0]-foodX > -4 && snakeX[0]-foodX < 4) && (snakeY[0] - foodY > -4 )&&(snakeY[0] - foodY > 4 )) {
        growSnake(); // 吃到食物時(shí)增長(zhǎng)蛇身并重新生成食物
        drawFood();
    } else {
        drawFood(); // 始終繪制食物躬它，即使沒有被吃掉
    }

    display.display();
    
    delay(500); // 控制游戲速度巍实，可以根據(jù)需要調(diào)整延遲時(shí)間 
}

void readJoystick() {
    int xValue = analogRead(joyXPin);
    int yValue = analogRead(joyYPin);
    
    if (xValue < 400) { 
        direction = 'R'; 
    } else if (xValue > 600) { 
        direction = 'L'; 
    } else if (yValue < 400) { 
        direction = 'U'; 
    } else if (yValue > 600) { 
        direction = 'D'; 
    }
}

void moveSnake() {
     for (int i = snakeLength; i > 0; i--) {
         snakeX[i] = snakeX[i-1];
         snakeY[i] = snakeY[i-1];
     }

     switch(direction) {
         case 'U': snakeY[0] -= 4; break;   // 蛇頭移動(dòng)距離改為4像素
         case 'D': snakeY[0] += 4; break;
         case 'L': snakeX[0] -= 4; break;
         case 'R': snakeX[0] += 4; break;
     }

     // 邊界檢查：如果蛇頭超出屏幕邊界滓技，則游戲結(jié)束。
     if (snakeX[0] < 0 || snakeX[0] >= SCREEN_WIDTH || snakeY[0] < 0 || snakeY[0] >= SCREEN_HEIGHT) {
         gameOver();
     }
}

void drawSnake() {
     for(int i=0; i<snakeLength; i++) {
         display.fillRect(snakeX[i], snakeY[i], 4, 4, WHITE); // 繪制蛇身棚潦，每節(jié)占用4x4像素區(qū)域令漂。
     }
}

void drawFood() {
     display.fillRect(foodX, foodY, 4, 4, WHITE); // 繪制食物，每個(gè)食物占用4x4像素區(qū)域丸边。
}

bool checkCollision() {
     for(int i=1; i<snakeLength; i++) {  
         if(snakeX[0] == snakeX[i] && snakeY[0] == snakeY[i]) return true;  
     }
     
     return false;
}

void growSnake() {
     if(snakeLength < sizeof(snakeX)/sizeof(snakeX[0])) {  
         snakeLength++;  
         foodX = random(0,(SCREEN_WIDTH/4))*4;  
         foodY = random(0,(SCREEN_HEIGHT/4))*4;
     }
}

void gameOver() {
     display.clearDisplay();
     display.setCursor(30,30);
     display.print('Game Over!');
     display.display();
     
     while(true); // 游戲結(jié)束后停止程序運(yùn)行叠必，可以重啟或重置。
}

// 新增函數(shù)：繪制邊框
void drawBorder() {
   for(int x=0; x<SCREEN_WIDTH; x++) {
       display.drawPixel(x, 0, WHITE);           // 上邊框
       display.drawPixel(x, SCREEN_HEIGHT-1, WHITE);   // 下邊框
   }
   for(int y=1; y<SCREEN_HEIGHT-1; y++) {
       display.drawPixel(0,y , WHITE);           // 左邊框
       display.drawPixel(SCREEN_WIDTH-1,y , WHITE);   //右邊框   
   }
}