/*
 * Copyright (C) EdgeTX
 *
 * Based on code named
 *   opentx - https://github.com/opentx/opentx
 *   th9x - http://code.google.com/p/th9x
 *   er9x - http://code.google.com/p/er9x
 *   gruvin9x - http://code.google.com/p/gruvin9x
 *
 * License GPLv2: http://www.gnu.org/licenses/gpl-2.0.html
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * 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.
 */

#include "opentx_types.h"
#include "board.h"
#include "keys.h"

#include "hal/adc_driver.h"

#if defined(LUA)
#include "lua/lua_api.h"
#endif

bool trimsAsButtons = false;

void setTrimsAsButtons(bool val) { trimsAsButtons = val; }

bool getTrimsAsButtons()
{
  bool lua = false;
#if defined(LUA)
  lua = isLuaStandaloneRunning();
#endif
  return (trimsAsButtons || lua);
}

uint32_t readKeys()
{
  uint32_t result = 0;

#ifndef BOOT
  if (getTrimsAsButtons()) {
#endif
    if (TRIMS_GPIO_REG_LHL & TRIMS_GPIO_PIN_LHL)
       result |= 1 << KEY_RADIO;
     if (TRIMS_GPIO_REG_LHR & TRIMS_GPIO_PIN_LHR)
       result |= 1 << KEY_MODEL;
     if (TRIMS_GPIO_REG_LVD & TRIMS_GPIO_PIN_LVD)
       result |= 1 << KEY_TELEM;
     if (TRIMS_GPIO_REG_LVU & TRIMS_GPIO_PIN_LVU)
       result |= 1 << KEY_PGUP;
     if (TRIMS_GPIO_REG_RVD & TRIMS_GPIO_PIN_RVD)
       result |= 1 << KEY_DOWN;
     if (TRIMS_GPIO_REG_RVU & TRIMS_GPIO_PIN_RVU)
       result |= 1 << KEY_UP;
     if (TRIMS_GPIO_REG_RHL & TRIMS_GPIO_PIN_RHL)
       result |= 1 << KEY_LEFT;
     if (TRIMS_GPIO_REG_RHR & TRIMS_GPIO_PIN_RHR)
       result |= 1 << KEY_RIGHT;
#ifndef BOOT
  }
#endif

  // Enter and Exit are always supported
  if (TRIMS_GPIO_REG_RPRESS & TRIMS_GPIO_PIN_RPRESS)
    result |= 1 << KEY_ENTER;
  if (TRIMS_GPIO_REG_LPRESS & TRIMS_GPIO_PIN_LPRESS)
    result |= 1 << KEY_EXIT;

  return result;
}

uint32_t readTrims()
{
  uint32_t result = 0;

  if(getTrimsAsButtons()) return result;
  if (TRIMS_GPIO_REG_LHL & TRIMS_GPIO_PIN_LHL)
    result |= 1 << (TRM_LH_DWN - TRM_BASE);
  if (TRIMS_GPIO_REG_LHR & TRIMS_GPIO_PIN_LHR)
    result |= 1 << (TRM_LH_UP - TRM_BASE);
  if (TRIMS_GPIO_REG_LVD & TRIMS_GPIO_PIN_LVD)
    result |= 1 << (TRM_LV_DWN - TRM_BASE);
  if (TRIMS_GPIO_REG_LVU & TRIMS_GPIO_PIN_LVU)
    result |= 1 << (TRM_LV_UP - TRM_BASE);
  if (TRIMS_GPIO_REG_RVD & TRIMS_GPIO_PIN_RVD)
    result |= 1 << (TRM_RV_DWN - TRM_BASE);
  if (TRIMS_GPIO_REG_RVU & TRIMS_GPIO_PIN_RVU)
    result |= 1 << (TRM_RV_UP - TRM_BASE);
  if (TRIMS_GPIO_REG_RHL & TRIMS_GPIO_PIN_RHL)
    result |= 1 << (TRM_RH_DWN - TRM_BASE);
  if (TRIMS_GPIO_REG_RHR & TRIMS_GPIO_PIN_RHR)
    result |= 1 << (TRM_RH_UP - TRM_BASE);

  return result;
}

bool trimDown(uint8_t idx)
{
  return readTrims() & (1 << idx);
}

bool keyDown()
{
  return readKeys() || readTrims();
}

#if !defined(BOOT)
uint32_t switchState(uint8_t index)
{
  uint16_t value = getAnalogValue(SWITCH_FIRST + index / 3);
  uint8_t position;

  if (value < 1024)
    position = 0;
  else if (value > 3 * 1024)
    position = 2;
  else
    position = 1;

  return position == (index % 3);
}
#endif

void monitorInit()
{
  GPIO_InitTypeDef GPIO_InitStructure;
  GPIO_InitStructure.GPIO_Mode = GPIO_Mode_IN;
  GPIO_InitStructure.GPIO_Speed = GPIO_Speed_2MHz;
  GPIO_InitStructure.GPIO_OType = GPIO_OType_PP;
  GPIO_InitStructure.GPIO_PuPd = GPIO_PuPd_DOWN;

  GPIO_InitStructure.GPIO_Pin = VBUS_MONITOR_PIN;
  GPIO_Init(GPIOJ, &GPIO_InitStructure);
}

void keysInit()
{
  GPIO_InitTypeDef GPIO_InitStructure;
  GPIO_InitStructure.GPIO_Mode = GPIO_Mode_IN;
  GPIO_InitStructure.GPIO_Speed = GPIO_Speed_2MHz;
  GPIO_InitStructure.GPIO_OType = GPIO_OType_PP;
  GPIO_InitStructure.GPIO_PuPd = GPIO_PuPd_DOWN;

  GPIO_InitStructure.GPIO_Pin = KEYS_GPIOB_PINS;
  GPIO_Init(GPIOB, &GPIO_InitStructure);

  GPIO_InitStructure.GPIO_Pin = KEYS_GPIOC_PINS;
  GPIO_Init(GPIOC, &GPIO_InitStructure);

  GPIO_InitStructure.GPIO_Pin = KEYS_GPIOD_PINS;
  GPIO_Init(GPIOD, &GPIO_InitStructure);

  GPIO_InitStructure.GPIO_Pin = KEYS_GPIOG_PINS;
  GPIO_Init(GPIOG, &GPIO_InitStructure);

  GPIO_InitStructure.GPIO_Pin = KEYS_GPIOH_PINS;
  GPIO_Init(GPIOH, &GPIO_InitStructure);

  GPIO_InitStructure.GPIO_Pin = KEYS_GPIOJ_PINS;
  GPIO_Init(GPIOJ, &GPIO_InitStructure);
  setTrimsAsButtons(false);
}