#include <string.h>

// **************************************************************************
// Demonstrate the Pimoroni RTC module (RV3028)
// Assumes that a Pico Display Pack (a 1.14inch IPS LCD screen with four
// useful buttons) is installed, and that the RV3028 I2C module has
// sda, scl and int on GPIO 20, 21 and 22
// Displays a clock showing date and time (no code yet to set the time,
// so do it on a separate machine), or a countdown timer.
// Button A selects the timer setting function, and runs the timer once set
// Button B selects the clock (the selection on startup)
// Button X increments the timer delay in seconds
// Button Y decrements it
// (There are on-screen reminders of the active buttons)
// **************************************************************************

// To use PicoExplorer rather than PicoDisplay, uncomment the following line
// #define USE_PICO_EXPLORER 1
// This:
// - Includes pico_explorer.hpp rather than display.hpp
// - Replaces all PicoDisplay references with PicoExplorer
// - Leaves out the .set_led() calls in flash_led()
#ifdef USE_PICO_EXPLORER
#include "pico_explorer.hpp"
#else
#include "pico_display.hpp"
#endif
#include "breakout_rtc.hpp"

#include "drivers/st7789/st7789.hpp"
#include "libraries/pico_graphics/pico_graphics.hpp"
#include "drivers/button/button.hpp"
#include "drivers/rgbled/rgbled.hpp"

#define MODE_DISP_CLOCK 0
#define MODE_DISP_TIMER 1
#define MODE_SET_TIMER  2

#define DEFAULT_TIMER_COUNT 5

using namespace pimoroni;

#ifdef USE_PICO_EXPLORER
uint16_t screen_width = PicoExplorer::WIDTH;
uint16_t screen_height = PicoExplorer::HEIGHT;

Button button_a(PicoExplorer::A);
Button button_b(PicoExplorer::B);
Button button_x(PicoExplorer::X);
Button button_y(PicoExplorer::Y);

#else
uint16_t screen_width = PicoDisplay::WIDTH;
uint16_t screen_height = PicoDisplay::HEIGHT;

Button button_a(PicoDisplay::A);
Button button_b(PicoDisplay::B);
Button button_x(PicoDisplay::X);
Button button_y(PicoDisplay::Y);

RGBLED led(PicoDisplay::LED_R, PicoDisplay::LED_G, PicoDisplay::LED_B);
#endif

ST7789 st7789(screen_width, screen_height, ROTATE_0, false, get_spi_pins(BG_SPI_FRONT));
PicoGraphics_PenRGB332 graphics(st7789.width, st7789.height, nullptr);

BreakoutRTC rtc;

#define LOW_COUNT_MOD 40
#define HIGH_COUNT_MOD 20

bool repeat_count_reached(uint16_t curr_count) {
  // Check whether the current counter means that a key has repeated
  if(curr_count <= 10*LOW_COUNT_MOD) {
    return (0 == (curr_count % LOW_COUNT_MOD));
  }
  else {
    return (0 == (curr_count % HIGH_COUNT_MOD));
  }
}

#define FLASH_MOD 20
void flash_led(uint32_t curr_count) {
  // Flash the LED based on the current loop counter
  // curr_count=0 will turn LED off
#ifndef USE_PICO_EXPLORER
  if((curr_count % FLASH_MOD) < (FLASH_MOD / 2)) {
    // value less than half modded number - LED off
    led.set_rgb(0, 0, 0);
  }
  else {
    // value more than half modded number - LED on
    led.set_rgb(128, 128, 128);
  }
#endif
}

int main() {
  Pen WHITE = graphics.create_pen(255, 255, 255);
  Pen BG = graphics.create_pen(55, 65, 75);
  Pen RED = graphics.create_pen(255, 0, 0);
  Pen GREEN = graphics.create_pen(0, 255, 0);

  rtc.init();
  // rtc.setup(false);

  // Make sure we have 24-hour time (smaller display!)
  if(rtc.is_12_hour())
    rtc.set_24_hour();

  // Use these variables to make the buttons single-shot
  // Counts number of loops pressed, 0 if not pressed
  // Only for x and y - a and b are single-shot
  uint16_t a_pressed = 0;
  uint16_t b_pressed = 0;
  uint16_t x_pressed = 0;
  uint16_t y_pressed = 0;

  struct pt {
    float      x;
    float      y;
    uint8_t    r;
    float     dx;
    float     dy;
    uint16_t pen;
  };

  uint32_t i = 0;
  char buf[256];
  uint8_t display_mode = MODE_DISP_CLOCK;
  uint16_t timer_count = DEFAULT_TIMER_COUNT;

  while(true) {

    if(a_pressed == 0 && button_a.read()) {
      a_pressed = 1;
      if(display_mode == MODE_DISP_CLOCK) {
        // We were displaying clock = set up timer
        display_mode = MODE_SET_TIMER;
      }
      else if(display_mode == MODE_SET_TIMER) {
        // We were setting up the timer - run it
        rtc.set_timer(false, 1, timer_count, true, true, false);
        display_mode = MODE_DISP_TIMER;
      }
      else if(display_mode == MODE_DISP_TIMER) {
        // We were running the timer - go back to setting up
        if(0 == (timer_count = rtc.get_timer_count())) {
          timer_count = DEFAULT_TIMER_COUNT;
        }
        rtc.disable_timer_interrupt();
        rtc.disable_timer();
        rtc.clear_timer_interrupt_flag();
        display_mode = MODE_SET_TIMER;
      }
    }
    else if(a_pressed >= 1 && !button_a.read()) {
      a_pressed = 0;
    }

    if(b_pressed == 0 && button_b.read()) {
      b_pressed = 1;
      if((display_mode == MODE_DISP_TIMER)
          || (display_mode == MODE_SET_TIMER)) {
        // We were setting or displaying timer - revert to clock
        rtc.disable_timer_interrupt();
        rtc.disable_timer();
        rtc.clear_timer_interrupt_flag();
        display_mode = MODE_DISP_CLOCK;
        timer_count = DEFAULT_TIMER_COUNT;
      }
    }
    else if(b_pressed >= 1 && !button_b.read()) {
      b_pressed = 0;
    }

    if(x_pressed == 0 && button_x.read()) {
      x_pressed = 1;
      if(display_mode == MODE_SET_TIMER) {
        // Setting timer - Increment count
        timer_count++;
      }
    }
    else if(x_pressed >= 1 && button_x.read()) {
      // Button still pressed - check if has reached repeat count
      if(repeat_count_reached(x_pressed++)) {
        timer_count++;
      }
    }
    else if(x_pressed >= 1 && !button_x.read()) {
      x_pressed = 0;
    }

    if(y_pressed == 0 && button_y.read()) {
      y_pressed = 1;
      if(display_mode == MODE_SET_TIMER) {
        // Setting timer - Decrement count
        if (timer_count >= 1) timer_count--;
      }
    }
    else if(y_pressed >= 1 && button_y.read()) {
      // Button still pressed - check if has reached repeat count
      if(repeat_count_reached(y_pressed++)) {
        if(timer_count >= 1)
          timer_count--;
      }
    }
    else if(y_pressed >= 1 && !button_y.read()) {
      y_pressed = 0;
    }

    Rect text_box(5, 5, screen_width-10, screen_height-10);
    graphics.set_pen(BG);
    graphics.rectangle(text_box);
    // text_box.deflate(10);
    graphics.set_clip(text_box);
    graphics.set_pen(WHITE);
    switch(display_mode) {
      case MODE_DISP_CLOCK:
        // Show the clock face
        flash_led(0);
        if(rtc.update_time()) {
          graphics.text("Set Timer",
              Point(text_box.x, text_box.y+2), 230, 1);
          graphics.set_pen(GREEN);
          graphics.text(rtc.string_date(),
              Point(text_box.x, text_box.y+20), 230, 4);
          graphics.set_pen(RED);
          graphics.text(rtc.string_time(),
              Point(text_box.x, text_box.y+60), 230, 6);
          graphics.set_pen(WHITE);
          graphics.text("Clock",
              Point(text_box.x, text_box.y+screen_height-20), 230, 1);
        }
        else {
          sprintf(buf, "Time: rtc.updateTime() ret err");
          graphics.text(buf,
              Point(text_box.x, text_box.y), 30, 2);
        }
        break;

      case MODE_DISP_TIMER:
        graphics.text("Set Timer",
            Point(text_box.x, text_box.y+2), 230, 1);
        if(rtc.read_timer_interrupt_flag()) {
          // Go periodic time interupt - say loop ended
          graphics.set_pen(RED);
          sprintf(buf, "%s", "Timer complete");
          graphics.text(buf,
              Point(text_box.x, text_box.y+30), 230, 4);
          graphics.set_pen(WHITE);
          flash_led(i);
        }
        else {
          sprintf(buf, "%s %d", "Timer running", rtc.get_timer_count());
          graphics.text(buf,
              Point(text_box.x, text_box.y+30), 230, 3);
        }
        graphics.text("Clock",
            Point(text_box.x, text_box.y+screen_height-20), 230, 1);
        break;

      case MODE_SET_TIMER:
        flash_led(0);
        graphics.text("Run Timer",
            Point(text_box.x, text_box.y+2), 230, 1);
        graphics.text("+ Time",
            Point(text_box.x+screen_width-42, text_box.y+2), 230, 1);
        sprintf(buf, "Time %d secs", timer_count);
        graphics.text(buf,
            Point(text_box.x, text_box.y+30), 230, 3);
        graphics.text("Clock",
            Point(text_box.x, text_box.y+screen_height-20), 230, 1);
        graphics.text("- Time",
            Point(text_box.x+screen_width-42,
              text_box.y+screen_height-20), 230, 1);
        break;
    }

    graphics.remove_clip();

    // update screen
    st7789.update(&graphics);

    i++;
  }

  return 0;
}