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

#define SCREEN_WIDTH 128 // OLED display width, in pixels
#define SCREEN_HEIGHT 64 // OLED display height, in pixels
#define BUTTON_PIN 23
#define BUTTON_PIN2 16
#define BUTTON_PIN3 15
// Declaration for SSD1306 display connected using I2C
#define OLED_RESET     -1 // Reset pin
#define SCREEN_ADDRESS 0x3C
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET); 
void setup()
{
  Serial.begin(9600);

  pinMode(19, OUTPUT);
  pinMode(18, OUTPUT);
  pinMode(17, OUTPUT);

  pinMode(BUTTON_PIN, INPUT);
  pinMode(BUTTON_PIN2, INPUT);
  pinMode(BUTTON_PIN3, INPUT);


  if(!display.begin(SSD1306_SWITCHCAPVCC, SCREEN_ADDRESS)) {
    Serial.println(F("SSD1306 allocation failed"));
    for(;;); // Don't proceed, loop forever
  }

  

}

void loop() {

  digitalWrite(19, HIGH);
  digitalWrite(18, HIGH);
  digitalWrite(17, HIGH);

  int buttonState = digitalRead(BUTTON_PIN);
  int buttonState2 = digitalRead(BUTTON_PIN2);
  int buttonState3 = digitalRead(BUTTON_PIN3);
  
 if(buttonState == 1 && buttonState2 == 1 && buttonState3 == 1 ){
 display.setTextColor(WHITE);
  display.setCursor(0,10);
  display.setTextSize(1.5);
  display.println("Red On!");
  display.setCursor(0,25);
  display.setTextSize(1.5);
  display.println("Green On!");
  display.setCursor(0,40);
  display.setTextSize(1.5);
  display.println("Yellow On!");
  display.display();
  display.clearDisplay();
 }

 else if(buttonState == 0 && buttonState2 == 0 && buttonState3 == 0 ){
 display.setTextColor(WHITE);
  display.setCursor(0,10);
  display.setTextSize(1.5);
  display.println("Red Off!");
  display.setCursor(0,25);
  display.setTextSize(1.5);
  display.println("Green Off!");
  display.setCursor(0,40);
  display.setTextSize(1.5);
  display.println("Yellow Off!");
  display.display();
  display.clearDisplay();
 }  

 else if(buttonState == 1 && buttonState2 == 0 && buttonState3 == 0 ){
 display.setTextColor(WHITE);
  display.setCursor(0,10);
  display.setTextSize(1.5);
  display.println("Red On!");
  display.setCursor(0,25);
  display.setTextSize(1.5);
  display.println("Green Off!");
  display.setCursor(0,40);
  display.setTextSize(1.5);
  display.println("Yellow Off!");
  display.display();
  display.clearDisplay();
 }

 else if(buttonState == 0 && buttonState2 == 1 && buttonState3 == 0 ){
 display.setTextColor(WHITE);
  display.setCursor(0,10);
  display.setTextSize(1.5);
  display.println("Red Off!");
  display.setCursor(0,25);
  display.setTextSize(1.5);
  display.println("Green On!");
  display.setCursor(0,40);
  display.setTextSize(1.5);
  display.println("Yellow Off!");
  display.display();
  display.clearDisplay();
 }  

 else if(buttonState == 0 && buttonState2 == 0 && buttonState3 == 1 ){
 display.setTextColor(WHITE);
  display.setCursor(0,10);
  display.setTextSize(1.5);
  display.println("Red Off!");
  display.setCursor(0,25);
  display.setTextSize(1.5);
  display.println("Green Off!");
  display.setCursor(0,40);
  display.setTextSize(1.5);
  display.println("Yellow On!");
  display.display();
  display.clearDisplay();
 }  

 else if(buttonState == 1 && buttonState2 == 1 && buttonState3 == 0 ){
 display.setTextColor(WHITE);
  display.setCursor(0,10);
  display.setTextSize(1.5);
  display.println("Red On!");
  display.setCursor(0,25);
  display.setTextSize(1.5);
  display.println("Green On!");
  display.setCursor(0,40);
  display.setTextSize(1.5);
  display.println("Yellow Off!");
  display.display();
  display.clearDisplay();
 }  


 else if(buttonState == 0 && buttonState2 == 1 && buttonState3 == 1 ){
 display.setTextColor(WHITE);
  display.setCursor(0,10);
  display.setTextSize(1.5);
  display.println("Red Off!");
  display.setCursor(0,25);
  display.setTextSize(1.5);
  display.println("Green ON!");
  display.setCursor(0,40);
  display.setTextSize(1.5);
  display.println("Yellow ON!");
  display.display();
  display.clearDisplay();
 }  

else if(buttonState == 1 && buttonState2 == 0 && buttonState3 == 1 ){
 display.setTextColor(WHITE);
  display.setCursor(0,10);
  display.setTextSize(1.5);
  display.println("Red ON!");
  display.setCursor(0,25);
  display.setTextSize(1.5);
  display.println("Green Off!");
  display.setCursor(0,40);
  display.setTextSize(1.5);
  display.println("Yellow ON!");
  display.display();
  display.clearDisplay();
 }  
 else display.clearDisplay();
 
}