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

// OLED
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

// LEDs
int ledX = 18;
int ledY = 19;
int ledTotal = 5;

// Valores
float F13 = 0.504;
float F13x = 0.252;
float F13y = 0.436;

float F23 = 1.008;
float F23x = 0.504;
float F23y = -0.873;

float Fx = 0.756;
float Fy = -0.437;
float Ftotal = 0.874;

void pantalla(String linea1, String linea2, String linea3, String linea4) {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(WHITE);

  display.setCursor(0,0);
  display.println(linea1);

  display.setCursor(0,15);
  display.println(linea2);

  display.setCursor(0,30);
  display.println(linea3);

  display.setCursor(0,45);
  display.println(linea4);

  display.display();
}

void setup() {
  Serial.begin(115200);

  pinMode(ledX, OUTPUT);
  pinMode(ledY, OUTPUT);
  pinMode(ledTotal, OUTPUT);

  display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
}

void loop() {

  // 🔹 1. DATOS
  pantalla("LEY DE COULOMB", "q1=2uC q2=-4uC", "q3=7uC r=0.5m", "");
  delay(2500);

  // 🔹 2. F13
  pantalla("F13 =", "0.504 N", "Dir: 60 grados", "Repulsiva");
  delay(2500);

  // 🔹 3. COMPONENTES F13
  pantalla("F13x=0.252N", "F13y=0.436N", "", "");
  delay(2500);

  // 🔹 4. F23
  pantalla("F23 =", "1.008 N", "Dir: -60 grados", "Atractiva");
  delay(2500);

  // 🔹 5. COMPONENTES F23
  pantalla("F23x=0.504N", "F23y=-0.873N", "", "");
  delay(2500);

  // 🔹 6. RESULTANTE
  pantalla("Fx=0.756N", "Fy=-0.437N", "F=0.874N", "Ang=-30");
  delay(3000);

  // LEDs según resultado
  digitalWrite(ledX, Fx > 0 ? HIGH : LOW);
  digitalWrite(ledY, Fy < 0 ? HIGH : LOW);
  digitalWrite(ledTotal, HIGH);

  delay(3000);

  // apagar LEDs
  digitalWrite(ledX, LOW);
  digitalWrite(ledY, LOW);
  digitalWrite(ledTotal, LOW);

  delay(2000);
}