#include <Arduino.h>
#include "keypad.h"
#include "pins.h"

CKeyPad::CKeyPad()
{
  // pin scanning
  _debouncedPins = 0;
  _prevPins = 0;
  _lastDebounceTime = millis();
  _debounceDelay = 50;
  // handler
  _lastKey = 0;
  _lastHoldTime = 0;
  _holdTimeout = 0;
  _keyCallback = NULL;
}

void
CKeyPad::init(int Lkey, int Rkey, int Ckey, int Ukey, int Dkey)
{
  _pins[0] = Lkey;
  _pins[1] = Rkey;
  _pins[2] = Ckey;
  _pins[3] = Ukey;
  _pins[4] = Dkey;
  for(int i=0; i<5; i++) 
    pinMode(_pins[i], INPUT);
}

uint8_t 
CKeyPad::scanPins()
{ 
  
  uint8_t newPins = 0;
  if(digitalRead(_pins[0]) == LOW)  newPins |= key_Left;
  if(digitalRead(_pins[1]) == LOW)  newPins |= key_Right;
  if(digitalRead(_pins[2]) == LOW)  newPins |= key_Centre;
  if(digitalRead(_pins[3]) == LOW)  newPins |= key_Up;
  if(digitalRead(_pins[4]) == LOW)  newPins |= key_Down;

  if(newPins != _prevPins) {
    _lastDebounceTime = millis();
    _prevPins = newPins;
  }

  long elapsed = millis() - _lastDebounceTime;
  if (elapsed > _debounceDelay) {
    // whatever the reading is at, it's been there for longer than the debounce
    // delay, so take it as the actual current state:
    _debouncedPins = newPins;
  }

  return _debouncedPins;
}

void
CKeyPad::setCallback(void (*callback)(uint8_t event))
{
  _keyCallback = callback;
}

uint8_t
CKeyPad::update()
{
  uint8_t newKey = scanPins();

  // determine edge events
  uint8_t keyChange = newKey ^ _lastKey;
  uint8_t Press = keyChange & newKey;     // bits set upon intial press, ONLY
  uint8_t Release = keyChange & ~newKey;  // bits set upon intial release, ONLY
  uint8_t Repeat = 0;  

  _lastKey = newKey;

  if(Press) {
#ifdef DBG_KEYPAD
    DebugPort.println("PRESS");
#endif
    _lastHoldTime = millis();
    _holdTimeout = 350;                 // initial hold delay
  }

  if(Release) {
#ifdef DBG_KEYPAD
    DebugPort.println("RELEASE");
#endif
    _holdTimeout = 0;                   // cancel repeat
  }

  long tDelta = millis() - _lastHoldTime;
  if(_holdTimeout && (tDelta > _holdTimeout)) {
#ifdef DBG_KEYPAD
    DebugPort.println("REPEAT");
#endif
    _holdTimeout = 150;                 // repeat delay
    _lastHoldTime += _holdTimeout;
    Repeat = newKey;
  }

  if(Press) {
    if(_keyCallback != NULL) 
      _keyCallback(keyPressed | Press | newKey);
    return keyPressed | Press;
  }

  if(Release) {
    if(_keyCallback != NULL) 
      _keyCallback(keyReleased | Release);
    return (keyReleased | Release);
  }

  if(Repeat) {
    if(_keyCallback != NULL) 
      _keyCallback(keyRepeat | Repeat);
    return (keyRepeat | Repeat);
  }

  return 0;
}