#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

// Declaration for an SSD1306 display connected to I2C (SDA, SCL pins)
#define OLED_RESET     4 // Reset pin # (or -1 if sharing Arduino reset pin)
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

//Constants
const int screenPin = 4;
const int modePin = 2;
unsigned long DELAY_TIME = 15000;
unsigned long delayStart = 0;

bool delayRunning = false;
bool ledOn = false;

int screenState = 1; 
int modeState = 1;
int counter = 20;

void setup();
void screenA();
void screenB();
void screenOff();
void screenOn();
void loop() ;


void setup() {

  // put your setup code here, to run once:
  Serial.begin(115200);
    // SSD1306_SWITCHCAPVCC = generate display voltage from 3.3V internally
  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3D)) { // Address 0x3D for 128x64
    Serial.println(F("SSD1306 allocation failed"));
    for(;;); // Don't proceed, loop forever
  }

  // Show initial display buffer contents on the screen --
  // the library initializes this with an Adafruit splash screen.
  display.display();
  delay(2000); // Pause for 2 seconds

  // Clear the buffer
  display.clearDisplay();

    pinMode(screenPin, INPUT_PULLUP);
    pinMode(modePin, INPUT_PULLUP);

   screenA();

  ledOn = false;
  // start delay
  delayStart = millis();
  delayRunning = true;

}

void screenA(){
    //display.clearDisplay();  
    display.setTextColor(WHITE);
    display.setTextSize(1);
    display.setCursor(30,32);
    display.print("Screen A On");
    display.display();
    delay(1);
    display.clearDisplay();
    }  

void screenB(){  
    //display.clearDisplay();
    display.setTextColor(WHITE);
    display.setTextSize(1);
    display.setCursor(30,32);
    display.print("Screen B On");
    display.display();
    delay(5000);
    display.clearDisplay();
    }

void screenOff(){
  display.ssd1306_command(SSD1306_DISPLAYOFF);
    }  

void screenOn(){
  display.ssd1306_command(SSD1306_DISPLAYON);
    } 

//void timer()





void loop() {

    int screenState = digitalRead(screenPin);
    int modeVal = digitalRead(modePin);
  
    //print out the value of the pushbutton
   Serial.print("Screensate =");
   Serial.println(screenState);
   Serial.print("Modval =");
   Serial.println(modeVal);

   if (delayRunning && ((millis() - delayStart) >= DELAY_TIME)) {
       delayStart += DELAY_TIME; // this prevents drift in the delays
       screenOff();
   }
   if (screenState == HIGH && modeVal == HIGH){
      screenA();
   }
   if (screenState == LOW && modeVal == HIGH) {
    screenB();
   }
   if (modeVal == LOW) {
      screenOn();
      ledOn = false;
      // start delay
      delayStart = millis();
      delayRunning = true;
   }
   //}
     //if (modeVal == LOW && OnOff == 1) {
    //screenOn();
    //OnOff = 0;
   //}
   //delay(1);
      
}