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

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

// PIN
#define BTN_ADD 2
#define BTN_RESET 3
#define BUZZER 8

int counter = 0;

void setup() {
  pinMode(BTN_ADD, INPUT_PULLUP);
    pinMode(BTN_RESET, INPUT_PULLUP);
      pinMode(BUZZER, OUTPUT);

        // OLED start
          display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
            display.clearDisplay();
              display.setTextSize(2);
                display.setTextColor(SSD1306_WHITE);
                  display.setCursor(0, 0);
                    display.println("Ready!");
                      display.display();
                      }

                      void loop() {
                        
                        if (digitalRead(BTN_ADD) == LOW) {
                              counter++;
                                  beep();

                                      showNumber();
                                          delay(300);
                                            }

                                              
                                                if (digitalRead(BTN_RESET) == LOW) {
                                                    counter = 0;
                                                        beep();

                                                            showNumber();
                                                                delay(300);
                                                                  }
                                                                  }

                                                                  void beep() {
                                                                    digitalWrite(BUZZER, HIGH);
                                                                      delay(100);
                                                                        digitalWrite(BUZZER, LOW);
                                                                        }

                                                                        void showNumber() {
                                                                          display.clearDisplay();
                                                                            display.setTextSize(2);
                                                                              display.setCursor(0, 0);
                                                                                display.print("Count:");
                                                                                  display.println(counter);
                                                                                    display.display();
                                                                                    }