#include "Display.h"
#include <TaskManager.h>
#include <logger.h>

Display::Display() : _disp(0), _statusFrame(0), _displaySaveMode(false) {
}

Display::~Display() {
}

void Display::setup(std::shared_ptr<BoardConfig> boardConfig) {
  if (boardConfig->OledReset != 0) {
    pinMode(boardConfig->OledReset, OUTPUT);
    digitalWrite(boardConfig->OledReset, HIGH);
    delay(1);
    digitalWrite(boardConfig->OledReset, LOW);
    delay(10);
    digitalWrite(boardConfig->OledReset, HIGH);
  }
  Wire.begin(boardConfig->OledSda, boardConfig->OledScl);
  _disp = std::shared_ptr<OLEDDisplay>(new SSD1306(&Wire, boardConfig->OledAddr));

  Bitmap bitmap(_disp->getWidth(), _disp->getHeight());
  _disp->display(&bitmap);

  _displayFrameRate.setTimeout(500);
  _displayFrameRate.start();

  _frameTimeout.setTimeout(15 * 1000);
  _displaySaveModeTimer.setTimeout(10 * 1000);
}

void Display::turn180() {
  _disp->flipScreenVertically();
}

void Display::activateDisplaySaveMode() {
  _displaySaveMode = true;
}

void Display::setDisplaySaveTimeout(uint32_t timeout) {
  _displaySaveModeTimer.setTimeout(timeout * 1000);
}

void Display::update() {
  if (_displayFrameRate.check()) {

    if (_frames.size() > 0) {
      std::shared_ptr<DisplayFrame> frame = *_frames.begin();
      Bitmap                        bitmap(_disp.get());
      frame->drawStatusPage(bitmap);
      _disp->display(&bitmap);

      if (!_frameTimeout.isActive()) {
        _frameTimeout.start();
        _displaySaveModeTimer.reset();
      } else if (_frameTimeout.check()) {
        _frames.pop_front();
        _frameTimeout.reset();
      }
    } else {
      if (_disp->isDisplayOn()) {
        Bitmap bitmap(_disp.get());
        _statusFrame->drawStatusPage(bitmap);
        _disp->display(&bitmap);

        if (_displaySaveMode) {
          if (_displaySaveModeTimer.isActive() && _displaySaveModeTimer.check()) {
            _disp->displayOff();
            _displaySaveModeTimer.reset();
          } else if (!_displaySaveModeTimer.isActive()) {
            _displaySaveModeTimer.start();
          }
        }
      }
    }

    _displayFrameRate.start();
  }
}

void Display::addFrame(std::shared_ptr<DisplayFrame> frame) {
  _frames.push_back(frame);
}

void Display::setStatusFrame(std::shared_ptr<StatusFrame> frame) {
  _statusFrame = frame;
}

void Display::showSpashScreen(String firmwareTitle, String version) {
  Bitmap bitmap(_disp.get());
  bitmap.drawString(0, 10, firmwareTitle);
  bitmap.drawString(0, 20, version);
  bitmap.drawString(0, 35, "by Peter Buchegger");
  bitmap.drawString(30, 45, "OE5BPA");
  _disp->display(&bitmap);
}

void Display::showStatusScreen(String header, String text) {
  Bitmap bitmap(_disp.get());
  bitmap.drawString(0, 0, header);
  bitmap.drawStringLF(0, 10, text);
  _disp->display(&bitmap);
}

void TextFrame::drawStatusPage(Bitmap &bitmap) {
  bitmap.drawString(0, 0, _header);
  bitmap.drawStringLF(0, 10, _text);
}