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

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

String text;
int point;

int buttonpins[] = {8, 9, 10};

void setup() {
  Serial.begin(9600);
  for (int i = 0; i < 3; i++) {
    pinMode(buttonpins[i], INPUT);
  }

  display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
  display.clearDisplay();
  display.setTextSize(2);
  display.setTextColor(WHITE);
  display.setCursor(64, 16);
  display.print("Hi");
  display.display();

  Serial.setTimeout(10);
  writeoled();
  Serial.println("Time to play with buttons");
  Serial.println("Press button-1 (BLUE) for scrolling LEFT to RIGHT ");
  Serial.println("Press button-2 (GREEN) for scrolling RIGHT to LEFT ");
  Serial.println("Press button-3 (PINK) for scrolling UP to DOWN ");
}

void loop() {
  int buttonpressed = checkButtonPressed();
  if (buttonpressed != 0) {
    Serial.print("Button ");
    Serial.print(buttonpressed);
    Serial.println(" pressed.");

    if (buttonpressed == 1) {
      display.startscrollright(0x00, 0x0F);
      delay(5000);
      display.stopscroll();
      display.clearDisplay();
      display.setCursor(0, 0);
      display.print(text);
      display.display();
    } 
    else if (buttonpressed == 2) {
      display.startscrollleft(0x00, 0x0F);
      delay(5000);
      display.stopscroll();
      display.clearDisplay();
      display.setCursor(0, 0);
      display.print(text);
      display.display();
    } 
    else if (buttonpressed == 3) {
      int t1=0;
      int tpast=0;
      t1=millis();
      while(tpast<5000){
        for (int count = 0; count < 4; count++) {
          point = 16 * count;
          display.setCursor(0, point);
          display.print(text);
          display.display();
          delay(300);  
          display.clearDisplay();  
        }
        tpast=millis()-t1;
      }
      display.clearDisplay();
      display.setCursor(0, 0);
      display.print(text);
      display.display();
    }
  }
}

void writeoled() {
  Serial.println("Enter string:");
  while (Serial.available() == 0) {
  }
  text = Serial.readString();
  display.clearDisplay();
  display.setCursor(0, 0);
  display.print(text);
  display.display();
  Serial.println("data received");
}

int checkButtonPressed() {
  for (int i = 0; i < 3; i++) {
    if (digitalRead(buttonpins[i]) == HIGH) {
      return i + 1;  
    }
  }
  return 0;  
}