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

// Определение размеров дисплея
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

// Определение клавиатуры
const byte ROWS = 4; // Число строк
const byte COLS = 4; // Число столбцов
char keys[ROWS][COLS] = {
  {'1', '2', '3', 'C'},
  {'4', '5', '6', '/'},
  {'7', '8', '9', '*'},
  {'0', '.', '=', '-'}
};

byte rowPins[ROWS] = {9, 8, 7, 6}; // Подключение строк
byte colPins[COLS] = {5, 4, 3, 2}; // Подключение столбцов

Keypad keypad = Keypad(makeKeymap(keys), rowPins, colPins, ROWS, COLS);

float result = 0; // Переменная для результата
char operation = ' '; // Переменная для операции
bool newOperation = true; // Флаг для новой операции

void setup() {
  display.begin(SSD1306_SWITCHCAPVCC, 0x3C); // Инициализация дисплея
  display.clearDisplay();
  display.setTextSize(2);
  display.setTextColor(WHITE);
  display.setCursor(0, 0);
  display.println("Calculator");
  display.display();
  delay(2000); // Задержка для отображения заголовка
}

void loop() {
  char key = keypad.getKey();

  if (key) {
    display.clearDisplay();
    display.setCursor(0, 0);

    if (key >= '0' && key <= '9') {
      // Обработка чисел
      if (newOperation) {
        result = key - '0'; // Присваиваем начальное число
        newOperation = false;
      } else {
        result = result * 10 + (key - '0'); // Обновляем результат
      }
    } else if (key == 'C') {
      // Очистка
      result = 0;
      operation = ' ';
      newOperation = true;
    } else if (key == '=') {
      // Вычисление результата
      display.clearDisplay();
      display.setCursor(0, 0);
      display.println("Result:");
      display.println(result);
      display.display();
      delay(2000);
      result = 0; // Сброс результата
      newOperation = true;
    } else {
      // Обработка операций
      if (!newOperation) {
        operation = key; // Сохраняем операцию
        newOperation = true; // Устанавливаем флаг для новой операции
      }
    }

    // Отображение результата
    display.clearDisplay();
    display.setCursor(0, 0);
    display.print("Result: ");
    display.println(result);
    display.display();
  }
}