//todo: test that current time increments properly on real rtc
  //implement tone adjustment for manual mode and brightness adjustment for both
  //implement pwm output for brightness and tone

#include <SPI.h>           
#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <Fonts/FreeSansBold18pt7b.h>

#define SCREEN_WIDTH 128 //oled stuff
#define SCREEN_HEIGHT 64
#define OLED_RESET 4
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

#include "RTClib.h" //real time clock stuff
RTC_DS1307 rtc; //switch this when uploading to real arduino


int pot;
int button = 2;
int automatic_mode;
int manual_mode;
int tone_adjust;
int brightness_adjust;
int brightness_output;

int set_hour; //variables to use when changing current time on oled
int set_minute;
int set_second;

int wake_hour = 0; //variables to store times for display on oled
int wake_minute = 0;
int sleep_hour = 0;
int sleep_minute = 0;

int depth = 0; //variables to store current state when navigating menu
int screen = 0;

//timing variables
long CT; //current time
long WT; //wake time
long ST; //sleep time
long L; //cycle length
long X; //time since cycle start
float prog; //progress along cycle as a percentage
long blue; //blue light percent output by automatic mode(out of 100)
long bright; //brightness percent output by automatic mode (out of 100)



void setup() {

  Serial.begin(115200);
  pinMode(A1, INPUT);
  pinMode(button, INPUT_PULLUP); //button pin
  pinMode(8, INPUT_PULLUP); //switch pins
  pinMode(9, INPUT_PULLUP); //switch pins
  display.begin(SSD1306_SWITCHCAPVCC, 0x3C); //initialize display
  display.clearDisplay(); //initial display settings
  display.setTextSize(1);          
  display.setTextColor(WHITE);

  if (! rtc.begin()) { //initialize real time clock
    Serial.println("Couldn't find RTC");
    Serial.flush();
    abort();
  }  

}

void loop() {
  DateTime now = rtc.now(); //update real time clock functions
  pot = (analogRead(A1)); //potentiometer pin
  set_hour = now.hour(); //rtc variables for oled
  set_minute = now.minute();
  set_second = now.second();
  display.fillRect(0,0,128,128,BLACK); //clears screen 
  automatic_mode = digitalRead(8); //checks switch position
  manual_mode = digitalRead(9); //checks switch position

  CT = 60 * now.hour() + 60 * now.minute() + now.second(); //timing variable math
  ST = 60 * sleep_hour + 60 * sleep_minute;
  WT = 60 * wake_hour  + 60 * wake_minute;
  L = ST - WT; 
  X = CT - WT;
  prog = (float) X / L;

  //blue light timing functions
  if((CT >= WT - (L/14)) && (CT < WT + (3*L/14))){
    blue = 80L;
  }
  else if((CT >= WT + (3*L/14)) && ((CT < WT + (L/2)))){
    blue = 80L + (3L * 105L / 14L) - 105L * prog;
  }
  else if((CT >= WT + (L/2)) && (CT < WT + (9.4*L/14))){
    blue = 50L +(700L / 6L) - (700L/3) * prog;
  }
  else if((CT >= WT + (9.4*L/14)) && (CT < ST)){
    blue = 10L;
  }
  else {
    blue = 0;
  }

  //brightness timing functions
  if((CT >= WT - (L/14)) && (CT < WT)){
    bright = (840L*1L/14L) + 840L*prog;
  }
  else if((CT >= WT) && (CT < WT + (5*L/14))){
    bright = 60L + (112L*prog);
  }
  else if((CT >= WT + (5*L/14)) && (CT < WT + (8*L/14))){
    bright = 100L;
  }
  else if((CT >= WT + (8*L/14)) && (CT < WT + (11*L/14))){
    bright = 100L + (8L*140L/14L) - (140L*prog);
  }
  else if((CT >= WT + (11*L/14)) && (CT < WT + (13*L/14))){
    bright = 70L + (210L*11L/14L) - (210L*prog);
  }
  else if((CT >= WT + (13*L/14)) && (CT < ST)){//changed wt to st
    bright = 40L + (560L*13L/14L) - (560L*prog);
  }
  else {
    bright = 0;
  }


  //display menu code

  if(automatic_mode == 0){ //checks if the three position switch is set to automatic mode

    if(digitalRead(button) == LOW && depth <= 1){ //increment menu position if button is pressed
      depth++;
      delay(100);
    }
    else if(digitalRead(button) == LOW && depth >= 2){
      depth = 0;
      delay(100);
    }

    if(depth == 0){ //if you arent currently adjusting a time setting, lets potentiometer control which page is selected
    screen = map(pot, 0, 1023, 0, 3);
    }

    if(screen == 0){ //if "current time" screen is selected
    display.setFont();
    display.setTextSize(1);
    display.setCursor(10,0);
    display.print("  Current Time  >"); //prints a label indicating this is the current time screen
    if(now.hour() < 10){display.setCursor(30,40);}
    else{display.setCursor(20,40);}
    display.setFont(&FreeSansBold18pt7b);
    display.print(now.hour(), DEC); //prints the current time
    display.print(':');
    if(now.minute() < 10){
      display.print("0");
    }
    display.print(now.minute(),DEC);

   if(depth == 1){ //if the button has been pressed once on this screen to select the hour's number
      set_hour = map(pot, 0, 1023, 0, 24); //make the hours number of the current time adjustable with the potentiometer
      
      if(now.second() % 2 == 0){ // makes underline flash every second
      if(now.hour() < 10) {display.drawFastHLine(30, 50, 18, WHITE);}//draw an underline under the selected number
      else{display.drawFastHLine(20, 50, 36, WHITE);}
      }
      
    }
     if(depth == 2){
      set_minute = map(pot, 0, 1023, 0, 60); //make the minutes number adjustable

      if(now.second() % 2 == 0){
      if(now.hour() < 10) {display.drawFastHLine(58, 50, 36, WHITE);}//draw an underline under the selected number
      else{display.drawFastHLine(68, 50, 36, WHITE);}
      }
    }
  }

 if(screen == 1){ //if "wake time" screen is selected
    display.setFont();
    display.setTextSize(1);
    display.setCursor(10,0);
    display.print("<    Wake Time    >"); //prints a label indicating this is the current time screen
    if(wake_hour < 10){display.setCursor(30,40);}
    else{display.setCursor(20,40);}
    display.setFont(&FreeSansBold18pt7b);
    display.print(wake_hour); //prints the current time
    display.print(':');
    if(wake_minute < 10){
      display.print("0");
    }
    display.print(wake_minute);

   if(depth == 1){ //if the button has been pressed once on this screen to select the hour's number
      wake_hour = map(pot, 0, 1023, 0, 24); //make the hours number of the current time adjustable with the potentiometer
      
      if(now.second() % 2 == 0){ // makes underline flash every second
      if(wake_hour < 10) {display.drawFastHLine(30, 50, 18, WHITE);}//draw an underline under the selected number
      else{display.drawFastHLine(20, 50, 36, WHITE);}
      }
      
    }
     if(depth == 2){
      wake_minute = map(pot, 0, 1023, 0, 60); //make the minutes number adjustable

      if(now.second() % 2 == 0){
      if(wake_hour < 10) {display.drawFastHLine(58, 50, 36, WHITE);}//draw an underline under the selected number
      else{display.drawFastHLine(68, 50, 36, WHITE);}
      }
    }
  }

    if(screen >= 2){ //if "sleep time" screen is selected
    display.setFont();
    display.setTextSize(1);
    display.setCursor(10,0);
    display.print("<    Sleep Time"); //prints a label indicating this is the current time screen
    if(sleep_hour < 10){display.setCursor(30,40);}
    else{display.setCursor(20,40);}
    display.setFont(&FreeSansBold18pt7b);
    display.print(sleep_hour); //prints the current time
    display.print(':');
    if(sleep_minute < 10){
      display.print("0");
    }
    display.print(sleep_minute);

   if(depth == 1){ //if the button has been pressed once on this screen to select the hour's number
      sleep_hour = map(pot, 0, 1023, 0, 24); //make the hours number of the current time adjustable with the potentiometer
      
      if(now.second() % 2 == 0){ // makes underline flash every second
      if(sleep_hour < 10) {display.drawFastHLine(30, 50, 18, WHITE);}//draw an underline under the selected number
      else{display.drawFastHLine(20, 50, 36, WHITE);}
      }
      
    }
     if(depth == 2){
      sleep_minute = map(pot, 0, 1023, 0, 60); //make the minutes number adjustable

      if(now.second() % 2 == 0){
      if(sleep_hour < 10) {display.drawFastHLine(58, 50, 36, WHITE);}//draw an underline under the selected number
      else{display.drawFastHLine(68, 50, 36, WHITE);}
      }
    }
  }
   
  }

  if(manual_mode == 0){ //if switch is in manual mode
  display.setCursor(32,0);
  display.setTextSize(1);
  display.setFont();
  display.print("Manual Mode");
  }

  rtc.adjust(DateTime(2024,3,23,set_hour,set_minute,now.second())); //update clock
  display.display(); //update display

 // Serial.print("Minutes:");
  //Serial.print(now.minute());
  //Serial.print("  Seconds:");
  //Serial.println(now.second());

  Serial.print("Lower Bound:");
  Serial.print(0);
  Serial.print("  ");
  Serial.print("Upper Bound:");
  Serial.print(100);
  Serial.print("  ");
  Serial.print("brightness%:");
  Serial.print(bright);
  Serial.print("   ");
  Serial.print("blue%:");
  Serial.println(blue);


}