#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);

unsigned long ledPreviousMillis = 0;
unsigned long displayPreviousMillis = 0;
const long ledInterval = 1000; // interval at which to blink LED (milliseconds)
const long displayInterval = 500; // interval at which to update display (milliseconds)

void setup() {
  pinMode(LED_BUILTIN, OUTPUT);
  // Initialize display
  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for (;;) ;
  }
  display.display(); // Show initial display buffer contents on the screen
  delay(2000); // Pause for 2 seconds
  display.clearDisplay(); // Clear the buffer

  // Other setup code
}

void loop() {
  unsigned long currentMillis = millis();

  // Task 1: Update LED state every 'ledInterval' milliseconds
  if (currentMillis - ledPreviousMillis >= ledInterval) {
    ledPreviousMillis = currentMillis;
    toggleLED();
  }

  // Task 2: Update OLED display every 'displayInterval' milliseconds
  if (currentMillis - displayPreviousMillis >= displayInterval) {
    displayPreviousMillis = currentMillis;
    updateDisplay();
  }
}

void toggleLED() {
  static bool ledState = LOW;
  digitalWrite(LED_BUILTIN, ledState);
  ledState = !ledState;
}

void updateDisplay() {
  // Update the OLED display content
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);
  display.println("Hello from SSD1306!");
  display.display();
}