#include <Adafruit_SSD1306.h>
#include "ArduinoTrace.h"

#define OLED_RESET -1  // Reset pin (usually not connected)
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64

const int circleRadius = 20;
int circleX = SCREEN_WIDTH / 2;
int circleY = SCREEN_HEIGHT / 2;
bool circleColor = WHITE;

int squareX = 1;
int squareY = 0;

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

unsigned long lastCircleChange = 0;
unsigned long lastSquareMove = 0;

void setup() {
  Serial.begin(115200);
  display.begin(SSD1306_SWITCHCAPVCC, 0x3C);  // Set display address (check datasheet if needed)
  display.display();
  delay(2000); // Wait for OLED to stabilize
}

void loop() {
  // Update circle color and redraw every 500ms
  if (millis() - lastCircleChange >= 500) {
    circleColor = !circleColor;
    lastCircleChange = millis();
  }

  // Move square around the screen every 60ms
  if (millis() - lastSquareMove >= 60) {
    // Check for edge collisions and update square position
    if (squareX + 4 >= SCREEN_WIDTH && squareY + 4 < SCREEN_HEIGHT ) { // Right edge
      squareX = SCREEN_WIDTH - 4;
      squareY++;
    } else if (squareY + 4 >= SCREEN_HEIGHT && squareX > 0) { // Bottom edge
      squareY = SCREEN_HEIGHT - 4;
      squareX--;
    } else if (squareX <= 0) { // Left edge
      squareX = 1;
      squareY--;
    } else if (squareY <= 0) { // Top edge
      squareY = 0;
      squareX++;
    } else {
      squareX += 2;
      //BREAK();
    }
    lastSquareMove = millis();
    Serial.println(squareX);
  }

  // Clear the display
  display.clearDisplay();

  // Draw the circle
  display.fillCircle(circleX, circleY, circleRadius, circleColor);

  // Draw the square
  display.drawRect(squareX, squareY, 4, 4, WHITE);

  // Display the updated content
  display.display();
}