#include <SPI.h>
#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <ezButton.h>
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET 4
Adafruit_SSD1306 oled(SCREEN_WIDTH,SCREEN_HEIGHT,&Wire,-1);
ezButton button1(13);
ezButton button2(33);
ezButton button3(14);
ezButton button4(27);
ezButton button5(26);
int vote1=0,vote2=0,vote3=0,vote4=0,vote5=0;
int flag=0;
int voting_completed=0;


void setup() {
  // put your setup code here, to run once:
  Serial.begin(9600);
  Serial.println("Hello, ESP32!");
  button1.setDebounceTime(25);
  button2.setDebounceTime(25);
  button3.setDebounceTime(25);
  button4.setDebounceTime(25);
  button5.setDebounceTime(25);
  if(!oled.begin(SSD1306_SWITCHCAPVCC,0x3C))
  {
    Serial.println("SSD1306 ALLOCATION FAILED");
    for(;;);
  }
  delay(2000);
  oled.clearDisplay();
  oled.setTextSize(2);
  oled.setTextColor(WHITE);
  oled.setCursor(2,5);
  oled.println("Start");
  oled.setCursor(2,21);
  oled.println("Voting");
  oled.display();
  
  oled.clearDisplay();
  oled.setTextSize(2);
  oled.setTextColor(WHITE);
  oled.setCursor(2,0);
  oled.println("A-BLUE");
  oled.setCursor(2,16);
  oled.println("B-YELLOW");
  oled.setCursor(2,32);
  oled.println("C-GREEN");
  oled.setCursor(2,48);
  oled.println("D-RED");
  oled.setCursor(2,64);
  oled.display();
  delay(2000);

}

void loop() {
  // put your main code here, to run repeatedly:
  delay(10); // this speeds up the simulation
  button1.loop();
  button2.loop();
  button3.loop();
  button4.loop();
  button5.loop();
  if(voting_completed==0){
    if(button1.isPressed()){
      vote1++;
    }
    else if(button2.isPressed()){
      vote2++;
    }
    else if(button3.isPressed()){
      vote3++;
    }
    else if(button4.isPressed()){
      vote4++;

    } 
    else if(button5.isPressed()){
      voting_completed=1;
      vote_count();
      delay(2000);
      Serial.println("voting completed");
    }
  }else {
    determine_winner();
  }
  delay(10);
  }
  void vote_count(){
    oled.clearDisplay();
    oled.setTextSize(2);
    oled.setTextColor(WHITE);
    oled.setCursor(1,0);
    oled.print("A B C D");
    oled.setCursor(1,20);
    oled.print(vote1);

    oled.setCursor(35,20);
    oled.print(vote2);

    oled.setCursor(65,20);
    oled.print(vote3);

    oled.setCursor(100,20);
    oled.print(vote4);
    oled.display();

  }
  //if(flag==0)
  //{oled.clearDisplay();
  //oled.setTextSize(2);
  //oled.setTextColor(WHITE);
  //oled.setCursor(2,0);
  //oled.println("A-BLUE");
  //oled.setCursor(2,16);
  //oled.println("B-YELLOW");
  //oled.setCursor(2,32);
  //oled.println("C-GREEN");
  //oled.setCursor(2,48);
  //oled.println("D-RED");
  //oled.setCursor(2,64);
  //oled.display();\
  //flag=1;
  //delay(10);
  //}
  
  
void determine_winner(){
  oled.clearDisplay();
  oled.setTextSize(1);
  oled.setTextColor(WHITE);
  oled.setCursor(1,10);
  if((vote1>vote2 && vote1>vote3 && vote1>vote4))
  oled.print("A is the winner");
  else if((vote2>vote1 && vote2>vote3 && vote2>vote4))
  oled.print("B is the winner");
  else if((vote3>vote1 && vote3>vote2 && vote3>vote4 ))
  oled.print("C is the winner");
  else if((vote4>vote1 && vote4>vote2 && vote4>vote3))
  oled.print("D is the winner");
  else 
  oled.print("There was a tie Vote again");
  oled.display();

}