/*
 * 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 "ThermostatModeScreen.h"
#include "KeyPad.h"
#include "../Utility/helpers.h"
#include "../Utility/UtilClasses.h"
#include "fonts/Icons.h"
#include "../Utility/BTC_GPIO.h"


///////////////////////////////////////////////////////////////////////////
//
// CThermostatModeScreen
//
// This screen provides control over experimental features
//
///////////////////////////////////////////////////////////////////////////

static const int Line3 = 14;
static const int Line2 = 27;
static const int Line1 = 40;
static const int Column = 40;

CThermostatModeScreen::CThermostatModeScreen(C128x64_OLED& display, CScreenManager& mgr) : CUIEditScreen(display, mgr) 
{
  _initUI();
  _window = 10;
  _thermoMode = 0;
  _cyclicMode.init();
}

void 
CThermostatModeScreen::onSelect()
{
  CUIEditScreen::onSelect();
  _initUI();
  _window = NVstore.getUserSettings().ThermostatWindow;
  _thermoMode = NVstore.getUserSettings().ThermostatMethod;
  _cyclicMode = NVstore.getUserSettings().cyclic;
}

void
CThermostatModeScreen::_initUI()
{
  CUIEditScreen::_initUI();
  _keyRepeat = -1;
}

bool 
CThermostatModeScreen::show()
{
  char msg[20];
  _display.clearDisplay();

  if(!CUIEditScreen::show()) {  // for showing "saving settings"

    _showTitle("Thermostat Mode");
    _drawBitmap(3, 14, ThermostatIconInfo);
    float fTemp = _window;
    if(NVstore.getUserSettings().degF) {
      fTemp = fTemp * 9 / 5;
      sprintf(msg, "%.1f`F", fTemp);
    }
    else {
      sprintf(msg, "%.1f`C", fTemp);
    }
    _printMenuText(Column, Line2, msg, _rowSel == 3);
    const char* modeStr = NULL;
    switch(_thermoMode) {
      case 0: modeStr = "Standard"; break;
      case 1: modeStr = "Deadband"; break;
      case 2: modeStr = "Linear Hz"; break;
      case 3: modeStr = "Ext thermostat"; break;
      case 4: modeStr = "Stop Start"; break;
    }
    if(modeStr)
      _printMenuText(Column, Line3, modeStr, _rowSel == 4);
    if(_cyclicMode.isEnabled()) {
      float fTemp = _cyclicMode.Stop+1;
      if(NVstore.getUserSettings().degF) {
        fTemp = fTemp * 9 / 5;
        sprintf(msg, "\352>%.0f`F", fTemp);
      }
      else {
        sprintf(msg, "\352>%.0f`C", fTemp);
      }
    }
    else {
      strcpy(msg, "OFF");
    }
    _printMenuText(Column, Line1, msg, _rowSel == 1);
    if(_cyclicMode.isEnabled()) {
      float fTemp = _cyclicMode.Start;
      if(NVstore.getUserSettings().degF) {
        fTemp = fTemp * 9 / 5;
        sprintf(msg, "\352<%.0f`F", fTemp);
      }
      else {
        sprintf(msg, "\352<%.0f`C", fTemp);
      }
    }
    else {
      strcpy(msg, "");
    }
    _printMenuText(Column + 42, Line1, msg, _rowSel == 2);
  }

  return true;
}

bool 
CThermostatModeScreen::animate()
{
  if(_saveBusy()) {
    return false;
  }

  int yPos = 53;
  int xPos = _display.xCentre();
  const char* pMsg = NULL;
  switch(_rowSel) {
    case 0:
      _printMenuText(xPos, yPos, " \021  \030Edit  Exit   \020 ", true, eCentreJustify);
      break;
    case 1:
      _display.drawFastHLine(0, 52, 128, WHITE);
      pMsg = "                    Heater shuts down over set point.                    ";
      _scrollMessage(56, pMsg, _scrollChar);
      break;
    case 2:
      _display.drawFastHLine(0, 52, 128, WHITE);
      pMsg = "                    Heater restarts below setpoint.                    ";
      _scrollMessage(56, pMsg, _scrollChar);
      break;
    case 3:
      _display.drawFastHLine(0, 52, 128, WHITE);
      pMsg = "                    User defined window for custom thermostat modes.                    ";
      _scrollMessage(56, pMsg, _scrollChar);
      break;
    case 4:
      _display.drawFastHLine(0, 52, 128, WHITE);
      switch(_thermoMode) {
        case 0:
          pMsg = "                   Use heater's standard thermostat control.                    ";
          break;
        case 1:
          pMsg = "                   The user defined window sets the thermostat's hysteresis.                    ";
          break;
        case 2:
          pMsg = "                   The pump rate is adjusted linearly across the set point window.                    ";
          break;
        case 3:
          pMsg = "                   The heater runs according to GPIO input #2: Open:minimum, Closed:maximum.                    ";
          break;
        case 4:
          pMsg = "                   The heater is stopped then started according to the defined window.                    ";
          break;
      }
      if(pMsg)
        _scrollMessage(56, pMsg, _scrollChar);
      break;
  }
  return true;
}

bool 
CThermostatModeScreen::keyHandler(uint8_t event)
{
  if(CUIEditScreen::keyHandler(event)) {   // potentially handles save confirm stage 
    return true;
  }

  if(event & keyPressed) {
    _keyRepeat = 0;  // unlock hold function
    // press LEFT to select previous screen
    if(event & key_Left) {
      switch(_rowSel) {
        case 0:
          _ScreenManager.prevMenu();
          break;
        case 4:
          _scrollChar = 0;
          // deliberate fall through here
        case 1:
        case 2:
        case 3:
          _adjust(-1);
          break;
      }
    }
    // press RIGHT to select next screen
    if(event & key_Right) {
      switch(_rowSel) {
        case 0:
          _ScreenManager.nextMenu();
          break;
        case 4:
          _scrollChar = 0;
          // deliberate fall through here
        case 1:
        case 2:
        case 3:
          _adjust(+1);
          break;
      }
    }
    if(event & key_Down) {
      if(_rowSel != 0) {
        _scrollChar = 0;
        _rowSel--;
        if(_rowSel == 2 && !_cyclicMode.isEnabled()) 
          _rowSel--;
        LOWERLIMIT(_rowSel, 0);
      }
      else {
        _ScreenManager.selectMenu(CScreenManager::SystemSettingsLoop, CScreenManager::SysVerUI);
      }
    }
    // UP press
    if(event & key_Up) {
      switch(_rowSel) {
        case 0:
        case 1:
        case 2:
        case 3:
          _scrollChar = 0;
          _rowSel++;
          if(_rowSel == 2 && !_cyclicMode.isEnabled()) 
            _rowSel++;
          UPPERLIMIT(_rowSel, 4);
          break;
      }
    }
    // CENTRE press
    if(event & key_Centre) {
      switch(_rowSel) {
        case 0:
//          _ScreenManager.selectMenu(CScreenManager::RootMenuLoop, CScreenManager::SettingsUI);  // force return to main menu
          _ScreenManager.selectMenu(CScreenManager::RootMenuLoop);  // force return to main menu
          break;
        case 1:
        case 2:
        case 3:
        case 4:
          _confirmSave();   // enter save confirm mode
          _rowSel = 0;
          break;
      }
    }
    if(event & keyRepeat) {
      _keyRepeat++;
      if(_rowSel == 3) {
        if(event & key_Right) {
          _adjust(+1);
          _ScreenManager.reqUpdate();
        }
        if(event & key_Left) {
          _adjust(-1);
          _ScreenManager.reqUpdate();
        }
      }
    }
    if(event & keyReleased) {
      _keyRepeat = -1;
    }
    _ScreenManager.reqUpdate();
  }

  return true;
}

void 
CThermostatModeScreen::_adjust(int dir)
{
  switch(_rowSel) {
    case 1:
      _cyclicMode.Stop += dir;
      BOUNDSLIMIT(_cyclicMode.Stop, 0, 9);
      break;
    case 2:
      _cyclicMode.Start += dir;
      BOUNDSLIMIT(_cyclicMode.Start, -20, 0);
      break;
    case 3:   // window
      _window += (dir * 0.1);
      BOUNDSLIMIT(_window, 0.2, 10.0);
      break;
    case 4:   // thermostat mode
      _thermoMode += dir;
#if USE_JTAG == 0
      if(_thermoMode == 3 && !GPIOin.usesExternalThermostat())
        _thermoMode += dir;
#else
      // CANNOT USE GPIO WITH JTAG DEBUG
      if(_thermoMode == 3
        _thermoMode += dir;
#endif
      WRAPLIMITS(_thermoMode, 0, 4);
      break;
  }
}

void
CThermostatModeScreen::_saveNV()
{
  sUserSettings settings = NVstore.getUserSettings();
  settings.ThermostatMethod = _thermoMode;
  settings.ThermostatWindow = _window;
  settings.cyclic = _cyclicMode;
  NVstore.setUserSettings(settings);
  NVstore.save();
}