/*
 * 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/>.
 * 
 */


///////////////////////////////////////////////////////////////////////////
//
// CSetTimerScreen
//
// This screen allows the timers to be adjusted
//
///////////////////////////////////////////////////////////////////////////

#include "SetTimerScreen.h"
#include "KeyPad.h"
#include "../Utility/helpers.h"
#include "../../lib/RTClib/RTClib.h"
#include "../RTC/TimerManager.h"
#include "fonts/Arial.h"
#include "../Utility/NVStorage.h"

const char* briefDOW[] = { "S", "M", "T", "W", "T", "F", "S" };

float calcPumpHz(int desired) {
  const sHeaterTuning& tuning = NVstore.getHeaterTuning();

  float ratio = float(desired - tuning.Tmin) / float(tuning.Tmax - tuning.Tmin);
  float offset = ratio * float(tuning.Pmax - tuning.Pmin);
  float PumpHz = tuning.Pmin + offset;
  return PumpHz / 10.f;  // tuning is saved as Hz x10
}

CSetTimerScreen::CSetTimerScreen(C128x64_OLED& display, CScreenManager& mgr, int instance) : CUIEditScreen(display, mgr) 
{
  _initUI();
  _ConflictTime = 0;
  _conflictID = 0;
  _timerID = instance;
}

void 
CSetTimerScreen::onSelect()
{
  CUIEditScreen::onSelect();
  _initUI();
  NVstore.getTimerInfo(_timerID, _timerInfo);
}

bool 
CSetTimerScreen::show()
{
  if(CUIEditScreen::show()) {
    return true;
  }

  CScreen::show();

  _display.clearDisplay();

  char str[20];
  int xPos, yPos;

  if(_rowSel == 0) {
    NVstore.getTimerInfo(_timerID, _timerInfo);   // ensure actual data when on base menu bar
  }
  sprintf(str, "Set Timer #%d", _timerID + 1);
  _showTitle(str);

  if(_ConflictTime) {
    long tDelta = millis() - _ConflictTime;
    if(tDelta > 0) 
      _ConflictTime = 0;
    sprintf(str, " with Timer %d ", _conflictID);
    _showConflict(str);
  }
  else {
    // start
    xPos = 18;
    yPos = 15;
    _printMenuText(xPos, yPos, "On", false, eRightJustify);
    _printMenuText(xPos+17, yPos, ":");
    xPos += 6;
    sprintf(str, "%02d", _timerInfo.start.hour);
    _printMenuText(xPos, yPos, str, _rowSel==1 && _colSel==0);
    xPos += 17;
    sprintf(str, "%02d", _timerInfo.start.min);
    _printMenuText(xPos, yPos, str, _rowSel==1 && _colSel==1);

    // stop
    xPos = 82;
    yPos = 15;
    _printMenuText(xPos, yPos, "Off", false, eRightJustify);
    _printMenuText(xPos+17, yPos, ":");
    xPos += 6;
    sprintf(str, "%02d", _timerInfo.stop.hour);
    _printMenuText(xPos, yPos, str, _rowSel==1 && _colSel==2);
    xPos += 17;
    sprintf(str, "%02d", _timerInfo.stop.min);
    _printMenuText(xPos, yPos, str, _rowSel==1 && _colSel==3);
    
    yPos = 39;
    {
      if(_timerInfo.temperature) {
        CTransientFont AF(_display, &arialItalic_7ptFontInfo);
        sprintf(str, "( %.1fHz )", calcPumpHz(_timerInfo.temperature));
        _printMenuText(64, yPos, str, false, eLeftJustify);
      }
    }

    // control
    const char* msg;
    _printEnabledTimers();
    
    xPos = _display.width() - border;
    yPos = 28;
    if(_timerInfo.repeat)
      msg = "Repeat";
    else
      msg = "Once";
    _printMenuText(xPos, yPos, msg, _rowSel==1 && _colSel==5, eRightJustify);

    xPos = 18;
    yPos = 41;
    float fTemp = _timerInfo.temperature;
    if(fTemp == 0) {
      strcpy(str, "Current set ");
      strcat(str, NVstore.getUserSettings().degF ? "`F" : "`C");
      _printMenuText(_display.xCentre(), yPos, str, _rowSel==1 && _colSel==6, eCentreJustify);
    }
    else {
      if(NVstore.getUserSettings().degF) {
        fTemp = fTemp * 9 / 5 + 32;
        sprintf(str, "%.0f`F", fTemp);
      }
      else {
        sprintf(str, "%.0f`C", fTemp);
      }
    _printMenuText(59, yPos, str, _rowSel==1 && _colSel==6, eRightJustify);
    }



    // navigation line
    yPos = 53;
    xPos = _display.xCentre();
    if(_rowSel == 2) {
      _display.drawFastHLine(0, 53, 128, WHITE);
      _printMenuText(_display.xCentre(), 57, "\033\032 Sel         \030\031 Adj", false, eCentreJustify);
      _printMenuText(_display.xCentre(), 57, "Done", false, eCentreJustify);
    }
    else if(_rowSel == 1) {
      _display.drawFastHLine(0, 53, 128, WHITE);
      _printMenuText(_display.xCentre(), 57, "\033\032 Sel         \030\031 Adj", false, eCentreJustify);
      _printMenuText(_display.xCentre(), 57, "Save", false, eCentreJustify);
    }
    else {
      _printMenuText(xPos, yPos, " \021     Exit     \020 ", _rowSel==0, eCentreJustify);
    }
  }

  return true;
}


bool 
CSetTimerScreen::keyHandler(uint8_t event)
{
  static bool bHeld = false;
  // handle initial key press
  if(event & keyPressed) {
    _repeatCount = 0;
    bHeld = false;
    // press CENTRE
    if(event & key_Centre) {
      // ON KEY RELEASE
    }
    // press LEFT - navigate fields, or screens
    if(event & key_Left) {
      switch(_rowSel) {
        case 0:
          _ScreenManager.prevMenu(); 
          break;
        case 1:
          // select previous field
          _colSel--;
          WRAPLOWERLIMIT(_colSel, 0, 6);
          break;
        case 2:
          // select previous day
          _colSel--;
          WRAPLOWERLIMIT(_colSel, 0, 6);
          break;
      }
    }
    // press RIGHT - navigate fields, or screens
    if(event & key_Right) {
      switch(_rowSel) {
        case 0:
          _ScreenManager.nextMenu(); 
          break;
        case 1:
          // select next field
          _colSel++;
          WRAPUPPERLIMIT(_colSel, 6, 0);
          break;
        case 2:
          // select next day
          _colSel++;
          WRAPUPPERLIMIT(_colSel, 6, 0);
          break;
      }
    }
  }

  // handle held down keys
  if(event & keyRepeat) {
    _repeatCount++;
    bHeld = true;
    if(_rowSel == 1) {
      if(event & key_Centre) {
        _ScreenManager.reqUpdate();
        _rowSel = 0;
        _colSel = 0;
      }
      if(_colSel < 4) {
        // fast repeat of hour/minute adjustments by holding up or down keys
        if(event & key_Down) _adjust(-1);
        if(event & key_Up) _adjust(+1);
      }
      else if(_colSel == 4) {
        if(event & (key_Up | key_Down)) {
          // enable per day programming by holding up or down
          _timerInfo.enabled &= 0x7f;   // strip next day flag
          _rowSel = 2;
          _colSel = 0;
        }
      }
    }
  }

  if(event & keyReleased) {

    if(!bHeld) {
      if(event & key_Centre) {
        if(_rowSel == 0) {
          _ScreenManager.selectMenu(CScreenManager::RootMenuLoop);  // exit: return to clock screen
        }
        else if(_rowSel == 2) {   // exit from per day settings
          _rowSel = 1;
          _colSel = 4;
        }
        else {  // in config fields, save new settings
          // test if the setting conflict with an already defined timer
          _conflictID = CTimerManager::conflictTest(_timerInfo);  
          if(_conflictID) {
            _timerInfo.enabled = 0;   // cancel enabled status
            _ConflictTime = millis() + 1500;
            _ScreenManager.reqUpdate();
            _rowSel = 1;
            _colSel = 4;   // select enable/disable 
          }
          else {
            _enableStoringMessage();
            _rowSel = 0;
            _colSel = 0;
          }
          CTimerManager::setTimer(_timerInfo);
        }
      }
  
      int maskDOW = 0x01 << _colSel;
      // released DOWN - can only leave adjustment by using OK (centre button)
      if(event & key_Down) {
        // adjust selected item
        switch(_rowSel) {
          case 1:
            if(!(_colSel == 4 && (_timerInfo.enabled & 0x7F) != 0)) {
              _adjust(-1);
            }
            else {
              // bump into per day setup
              _rowSel = 2;
              _colSel = 0;
            }
            break;
          case 2:
            // adjust selected item
            _timerInfo.enabled ^= maskDOW;
            _timerInfo.enabled &= 0x7f;
            break;
        }
      }
      // released UP 
      if(event & key_Up) {
        switch(_rowSel) {
          case 0:
            _rowSel = 1;
            _colSel = 0;
            break;
          case 1:
            // prevent accidentally losing per day settings
            if(!(_colSel == 4 && (_timerInfo.enabled & 0x7F) != 0)) {
              _adjust(+1);   // adjust selected item, unless in per day mode
            }
            else {
              // bump into per day setup
              _rowSel = 2;   
              _colSel = 0;
            }
            break;
          case 2:
            // adjust selected item
            _timerInfo.enabled ^= maskDOW;
            _timerInfo.enabled &= 0x7f;
            break;
        }
      }
    }
  }

  _ScreenManager.reqUpdate();
  return true;
}


void 
CSetTimerScreen::_adjust(int dir)
{
  int maskDOW = 0x01 << _colSel;  // if doing Day of Week - (_rowSel == 2)  

  switch(_colSel) {
    case 0:
      _timerInfo.start.hour += dir;
      WRAPLIMITS(_timerInfo.start.hour, 0, 23);
      break;
    case 1:
      _timerInfo.start.min += dir;
      WRAPLIMITS(_timerInfo.start.min, 0, 59);
      break;
    case 2:
      _timerInfo.stop.hour += dir;
      WRAPLIMITS(_timerInfo.stop.hour, 0, 23);
      break;
    case 3:
      _timerInfo.stop.min += dir;
      WRAPLIMITS(_timerInfo.stop.min, 0, 59);
      break;
    case 4:
      if(_rowSel == 1) {
        _timerInfo.enabled &= 0x80;      // ensure specific day flags are cleared
        _timerInfo.enabled ^= 0x80;      // toggle next day flag
      }
      if(_rowSel == 2) {
        _timerInfo.enabled &= 0x7f;      // ensure next day flag is cleared
        _timerInfo.enabled ^= maskDOW;   // toggle flag for day of week
      }
      break;
    case 5:
      _timerInfo.repeat = !_timerInfo.repeat;
      break;
    case 6:
      if(_timerInfo.temperature <= 8 && dir < 0)
        _timerInfo.temperature = 0;
      else {
        _timerInfo.temperature += dir;
        BOUNDSLIMIT(_timerInfo.temperature, 8, 35);
      }
      break;
  }
}

void
CSetTimerScreen::_printEnabledTimers()
{
  const int dayWidth = 10;
  int xPos = border;
  int yPos = 28;

  if(_timerInfo.enabled == 0x00 && _rowSel != 2) {
    _printMenuText(xPos, yPos, "Disabled", _colSel==4);
  }
  else if(_timerInfo.enabled & 0x80) {
    if(_rowSel==1 && _colSel==4)
      _printMenuText(xPos, yPos, "Enabled", true);
    else 
      _printInverted(xPos, yPos, "Enabled", true);
  }
  else {
    if(_rowSel==1 && _colSel==4) {
      CRect extents;
      extents.width = 7 * dayWidth + 2;
      extents.height = 11;
      extents.xPos = border;
      extents.yPos = yPos-2;
      extents.Expand(border);
      _display.drawRoundRect(extents.xPos, extents.yPos, extents.width, extents.height, radius, WHITE);
    }
    xPos = border+3;  // back step 7 day entries
    int xSel = xPos + _colSel * dayWidth;  // note location of selection now (xPos gets changed)
    // print days, inverse if enabled
    for(int i=0; i<7; i++) {
      int dayMask = 0x01 << i;
      if(_timerInfo.enabled & dayMask) {
        _display.fillRect(xPos-2, yPos-2, 9, 11, WHITE);
      }
      _printInverted(xPos, yPos, briefDOW[i], _timerInfo.enabled & dayMask);
      _display.drawPixel(xPos-2, yPos-2, BLACK);
      _display.drawPixel(xPos-2, yPos+8, BLACK);
      _display.drawPixel(xPos+6, yPos-2, BLACK);
      _display.drawPixel(xPos+6, yPos+8, BLACK);
      xPos += dayWidth;
    }
    // draw selection loop afterwards - writing text otherwise obliterates it
    if(_rowSel==2) {
      CRect extents;
      extents.xPos = xSel-1-border;
      extents.yPos = yPos-1-border;
      extents.width = 13;
      extents.height = 15;
      _display.drawRoundRect(extents.xPos, extents.yPos, extents.width, extents.height, 3, WHITE);
    }
  }
}

void
CSetTimerScreen::_showConflict(const char* str)      
{
  CTransientFont AF(_display, &arial_8ptBoldFontInfo);
  _display.fillRect(19, 22, 90, 36, WHITE);
  _printInverted(_display.xCentre(), 39, str, true, eCentreJustify);
  _printInverted(_display.xCentre(), 28, "Conflicts", true, eCentreJustify);
}