/*
 * This file is part of the "bluetoothheater" distribution 
 * (https://gitlab.com/mrjones.id.au/bluetoothheater) 
 *
 * Copyright (C) 2018  Ray Jones <ray@mrjones.id.au>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 * 
 */

#include "128x64OLED.h"
#include "display.h"
#include "KeyPad.h"
#include "helpers.h"
#include "Screen3.h"

const int border = 3;
const int radius = 4;
const int Row[] = { 52, 40, 28, 16 } ;
const int Col[] = { border, 70, 100};
const char* Label0 = "<-             ->"; 
const char* Label1[] = { "Thermostat",
                         "Fixed Hz" }; 
const char* Label2[] = { "Prime pump",
                          "OFF",
                          "ON" }; 


CScreen3::CScreen3(C128x64_OLED& display, CScreenManager& mgr) : CScreen(display, mgr) 
{
  _PrimeStop = 0;
  _PrimeCheck = 0;
  _rowSel = 0;
  _colSel = 0;
}


void 
CScreen3::show(const CProtocol& CtlFrame, const CProtocol& HtrFrame)
{
  CScreen::show(CtlFrame, HtrFrame);
  
  CRect extents;

  // show next/prev screen navigation line
  _drawMenuTextCentreJustified(_display.xCentre(), Row[0], _rowSel == 0, Label0);

  // thermostat / fixed mode selection menu
  // highlight active state - depends if line is active whetehr this is an open box, or inverse text
  int col = getThermostatMode() ? 0 : 1;              // follow actual heater settings
  _display.getTextExtents(Label1[col], extents);
  extents.xPos = (col == 0) ? border : _display.width() - extents.width - border;
  extents.yPos = Row[1];
  if(_rowSel == 1) {
    // draw selection box
    _drawSelectionBox(extents.xPos, extents.yPos, Label1[col]);
  }
  else {
    // draw white background, expanded about usual text size
    extents.Expand(1);
    _display.fillRect(extents.xPos, extents.yPos, extents.width, extents.height, WHITE);
  }
  
  if(col == 0 && _rowSel != 1)   
    _display.setTextColor(BLACK);
  _display.setCursor(border, Row[1]);
  _display.print(Label1[0]);
  _display.setTextColor(WHITE);

  if(col == 1 && _rowSel != 1)
    _display.setTextColor(BLACK);
  _display.setCursor(_display.width() - border, Row[1]);
  _display.printRightJustified(Label1[1]);
  _display.setTextColor(WHITE);

  // fuel pump priming menu
  _drawMenuText(Col[0], Row[2], "Prime pump");
  if(_rowSel == 2) {
    _drawMenuText(Col[1], Row[2], _colSel == 1, "OFF");
    if(_colSel != 2) {
      if(!getRunState()) {                    // prevent option if heater is running
        _drawMenuText(Col[2], Row[2], "ON");  // becomes Hz when actually priming 
      }
    }
    else {
      float pumpHz = getPumpHz();
      // recognise if heater has stopped pump, after an initial holdoff upon first starting
      long tDelta = millis() - _PrimeCheck;
      if(_PrimeCheck && tDelta > 0 && pumpHz < 0.1) {
        stopPump();
      }
      // test if time is up, stop priming if so
      tDelta = millis() - _PrimeStop;
      if(_PrimeStop && tDelta > 0) {
        stopPump();
      }

      if(_PrimeStop) {
        char msg[16];
        sprintf(msg, "%.1fHz", pumpHz);
        _drawMenuTextRightJustified(_display.width()-border, Row[2], true, msg);
      }
    }
  }

  _display.display();

}


void 
CScreen3::animate()
{
  // do nothing!!
};


void 
CScreen3::keyHandler(uint8_t event)
{
  if(event & keyPressed) {
    // press CENTRE
    if(event & key_Centre) {
      return;
    }
    // press LEFT 
    if(event & key_Left) {
      switch(_rowSel) {
        case 0: 
          _ScreenManager.prevScreen(); 
          break;
        case 1: 
          _colSel = 0; 
          setThermostatMode(1);
          break;
        case 2: 
          _colSel = 1; 
          break;
        case 3: break;
      }
    }
    // press RIGHT 
    if(event & key_Right) {
      switch(_rowSel) {
        case 0: 
          _ScreenManager.nextScreen(); 
          break;
        case 1: 
          _colSel = 1; 
          setThermostatMode(0);
          break;
        case 2: 
          if(!getRunState()) 
            _colSel = 2; 
          break;
        case 3: break;
      }
    }
    // press UP
    if(event & key_Up) {
      _rowSel++;
      UPPERLIMIT(_rowSel, 2);
      if(_rowSel == 2)
        _colSel = 1;       // select OFF upon entry to priming menu
    }
    // press DOWN
    if(event & key_Down) {
      _rowSel--;
      LOWERLIMIT(_rowSel, 0);
      _colSel = 0;
    }

    // check if fuel priming was selected
    if(_rowSel == 2 && _colSel == 2) {
      reqPumpPrime(true);
      _PrimeStop = millis() + 150000;   // allow 2.5 minutes - much the same as the heater itself cuts out at
      _PrimeCheck = millis() + 3000;    // holdoff upon start before testing for heater shutting off pump
    }
    else {
      stopPump();
    }

    _ScreenManager.reqUpdate();
  }
}

void 
CScreen3::stopPump()
{
  reqPumpPrime(false);
  _PrimeCheck = 0;
  _PrimeStop = 0;
  if(_colSel == 2)
    _colSel = 1;
}