#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 23
int currentImage = 0;
unsigned long lastDebounceTime = 0; 
unsigned long debounceDelay = 50; 
bool lastButtonState = HIGH; 
bool buttonState = HIGH;

void setup() {
  pinMode(BUTTON_PIN, INPUT_PULLUP);
  Serial.begin(115200);

  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) { // ตรวจสอบที่นี่ว่า I2C Address ถูกต้อง
    Serial.println(F("SSD1306 allocation failed"));
    for(;;);
  }

  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0,0);
  display.print("Press the button to change image");
  display.display();
}

void drawImage1() {
  display.clearDisplay();
  display.setCursor(0,0);
  display.print("Image 1");
  display.drawRect(10, 20, 50, 30, SSD1306_WHITE); // วาดรูปสี่เหลี่ยม
  display.display();
}

void drawImage2() {
  display.clearDisplay();
  display.setCursor(0,0);
  display.print("Image 2");
  display.drawCircle(64, 32, 20, SSD1306_WHITE); // วาดรูปวงกลม
  display.display();
}

void drawImage3() {
  display.clearDisplay();
  display.setCursor(0,0);
  display.print("Image 3");
  display.drawLine(0, 0, 127, 63, SSD1306_WHITE); // วาดเส้นตรง
  display.display();
}

void loop() {
  bool reading = digitalRead(BUTTON_PIN);

  if (reading != lastButtonState) {
    lastDebounceTime = millis();
  }

  if ((millis() - lastDebounceTime) > debounceDelay) {
    if (reading != buttonState) {
      buttonState = reading;

      if (buttonState == LOW) {
        currentImage = (currentImage + 1) % 3;
        if(currentImage == 0) {
          drawImage1();
        } else if(currentImage == 1) {
          drawImage2();
        } else if(currentImage == 2) {
          drawImage3();
        }
      }
    }
  }

  lastButtonState = reading;
}