// Copyright 2018 Dolphin Emulator Project
// Licensed under GPLv2+
// Refer to the license.txt file included.

#include "DolphinQt2/TAS/GCTASInputWindow.h"
#include "Common/CommonTypes.h"
#include "DolphinQt2/TAS/Shared.h"
#include "InputCommon/GCPadStatus.h"

#include <QCheckBox>
#include <QGroupBox>
#include <QHBoxLayout>
#include <QVBoxLayout>

#include <algorithm>

GCTASInputWindow::GCTASInputWindow(QWidget* parent) : QDialog(parent)
{
  auto* main_stick_box = CreateStickInputs(this, tr("Main Stick ALT+F/G"), &m_x_main_stick_byte,
                                           &m_y_main_stick_byte, 255, 255, Qt::Key_F, Qt::Key_G);
  auto* c_stick_box = CreateStickInputs(this, tr("C Stick ALT+H/J"), &m_x_c_stick_byte,
                                        &m_y_c_stick_byte, 255, 255, Qt::Key_H, Qt::Key_J);

  auto* top_layout = new QHBoxLayout;
  top_layout->addWidget(main_stick_box);
  top_layout->addWidget(c_stick_box);

  auto* l_trigger_layout = new QHBoxLayout;
  m_l_trigger_byte = CreateTriggerInputs(this, l_trigger_layout, Qt::Key_N, Qt::Horizontal);

  auto* l_trigger_box = new QGroupBox(tr("Left Trigger ALT+N"));
  l_trigger_box->setLayout(l_trigger_layout);

  auto* r_trigger_layout = new QHBoxLayout;
  m_r_trigger_byte = CreateTriggerInputs(this, r_trigger_layout, Qt::Key_M, Qt::Horizontal);

  auto* r_trigger_box = new QGroupBox(tr("Right Trigger ALT+M"));
  r_trigger_box->setLayout(r_trigger_layout);

  m_a_button = new QCheckBox(QStringLiteral("&A"));
  m_b_button = new QCheckBox(QStringLiteral("&B"));
  m_x_button = new QCheckBox(QStringLiteral("&X"));
  m_y_button = new QCheckBox(QStringLiteral("&Y"));
  m_z_button = new QCheckBox(QStringLiteral("&Z"));
  m_l_button = new QCheckBox(QStringLiteral("&L"));
  m_r_button = new QCheckBox(QStringLiteral("&R"));
  m_start_button = new QCheckBox(QStringLiteral("&START"));
  m_left_button = new QCheckBox(QStringLiteral("L&eft"));
  m_up_button = new QCheckBox(QStringLiteral("&Up"));
  m_down_button = new QCheckBox(QStringLiteral("&Down"));
  m_right_button = new QCheckBox(QStringLiteral("R&ight"));

  auto* buttons_layout1 = new QHBoxLayout;
  buttons_layout1->addWidget(m_a_button);
  buttons_layout1->addWidget(m_b_button);
  buttons_layout1->addWidget(m_x_button);
  buttons_layout1->addWidget(m_y_button);
  buttons_layout1->addWidget(m_z_button);
  buttons_layout1->addWidget(m_l_button);
  buttons_layout1->addWidget(m_r_button);

  auto* buttons_layout2 = new QHBoxLayout;
  buttons_layout2->addWidget(m_start_button);
  buttons_layout2->addWidget(m_left_button);
  buttons_layout2->addWidget(m_up_button);
  buttons_layout2->addWidget(m_down_button);
  buttons_layout2->addWidget(m_right_button);

  auto* buttons_layout = new QVBoxLayout;
  buttons_layout->setSizeConstraint(QLayout::SetFixedSize);
  buttons_layout->addLayout(buttons_layout1);
  buttons_layout->addLayout(buttons_layout2);

  auto* buttons_box = new QGroupBox(tr("Buttons"));
  buttons_box->setLayout(buttons_layout);

  auto* layout = new QVBoxLayout;
  layout->addLayout(top_layout);
  layout->addWidget(l_trigger_box);
  layout->addWidget(r_trigger_box);
  layout->addWidget(buttons_box);

  setLayout(layout);
}

void GCTASInputWindow::GetValues(GCPadStatus* pad)
{
  if (!isVisible())
    return;

  SetButton(m_a_button, pad, PAD_BUTTON_A);
  SetButton(m_b_button, pad, PAD_BUTTON_B);
  SetButton(m_x_button, pad, PAD_BUTTON_X);
  SetButton(m_y_button, pad, PAD_BUTTON_Y);
  SetButton(m_z_button, pad, PAD_TRIGGER_Z);
  SetButton(m_l_button, pad, PAD_TRIGGER_L);
  SetButton(m_r_button, pad, PAD_TRIGGER_R);
  SetButton(m_left_button, pad, PAD_BUTTON_LEFT);
  SetButton(m_up_button, pad, PAD_BUTTON_UP);
  SetButton(m_down_button, pad, PAD_BUTTON_DOWN);
  SetButton(m_right_button, pad, PAD_BUTTON_RIGHT);
  SetButton(m_start_button, pad, PAD_BUTTON_START);

  if (m_a_button->isChecked())
    pad->analogA = 0xFF;
  else
    pad->analogA = 0x00;

  if (m_b_button->isChecked())
    pad->analogB = 0xFF;
  else
    pad->analogB = 0x00;

  pad->triggerLeft = m_l_trigger_byte->value();
  pad->triggerRight = m_r_trigger_byte->value();

  pad->stickX = m_x_main_stick_byte->value();
  pad->stickY = m_y_main_stick_byte->value();
  pad->substickX = m_x_c_stick_byte->value();
  pad->substickY = m_y_c_stick_byte->value();
}