dolphin/Source/Core/DolphinQt2/TAS/StickWidget.cpp

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

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

#include <QMouseEvent>
#include <QPainter>

#include <algorithm>

StickWidget::StickWidget(QWidget* parent, u16 max_x, u16 max_y) : QWidget(parent)
{
  m_max_x = max_x;
  m_max_y = max_y;
  m_x = 0;
  m_y = 0;

  setMouseTracking(false);
}

void StickWidget::SetX(u16 x)
{
  m_x = std::min(m_max_x, x);

  update();
}

void StickWidget::SetY(u16 y)
{
  m_y = std::min(m_max_y, y);

  update();
}

void StickWidget::paintEvent(QPaintEvent* event)
{
  QPainter painter(this);

  painter.setBrush(Qt::white);
  painter.drawEllipse(0, 0, width() - 1, height() - 1);

  painter.drawLine(0, height() / 2, width(), height() / 2);
  painter.drawLine(width() / 2, 0, width() / 2, height());

  // convert from value space to widget space
  u16 x = (m_x * width()) / m_max_x;
  u16 y = height() - (m_y * height()) / m_max_y;

  painter.drawLine(width() / 2, height() / 2, x, y);

  painter.setBrush(Qt::blue);
  int wh_avg = (width() + height()) / 2;
  int radius = wh_avg / 30;
  painter.drawEllipse(x - radius, y - radius, radius * 2, radius * 2);
}

void StickWidget::mousePressEvent(QMouseEvent* event)
{
  handleMouseEvent(event);
}

void StickWidget::mouseMoveEvent(QMouseEvent* event)
{
  handleMouseEvent(event);
}

void StickWidget::handleMouseEvent(QMouseEvent* event)
{
  // convert from widget space to value space
  int new_x = ((int)event->x() * m_max_x) / width();
  int new_y = m_max_y - ((int)event->y() * m_max_y) / height();

  m_x = std::max(0, std::min((int)m_max_x, new_x));
  m_y = std::max(0, std::min((int)m_max_y, new_y));

  emit ChangedX(m_x);
  emit ChangedY(m_y);
  update();
}