Qt: Implement NetPlay

1 files changed, 575 insertions, 0 deletions

diff --git a/Source/Core/DolphinQt2/NetPlay/NetPlayDialog.cpp b/Source/Core/DolphinQt2/NetPlay/NetPlayDialog.cpp
new file mode 100644
index 0000000000..e0c16eedd3
--- /dev/null
+++ b/Source/Core/DolphinQt2/NetPlay/NetPlayDialog.cpp
@@ -0,0 +1,575 @@
+// Copyright 2017 Dolphin Emulator Project
+// Licensed under GPLv2+
+// Refer to the license.txt file included.
+
+#include "DolphinQt2/NetPlay/NetPlayDialog.h"
+
+#include <QApplication>
+#include <QCheckBox>
+#include <QClipboard>
+#include <QComboBox>
+#include <QGridLayout>
+#include <QGroupBox>
+#include <QHBoxLayout>
+#include <QLabel>
+#include <QLineEdit>
+#include <QListWidget>
+#include <QMessageBox>
+#include <QProgressDialog>
+#include <QPushButton>
+#include <QSpinBox>
+#include <QTextEdit>
+
+#include <sstream>
+
+#include "Common/CommonPaths.h"
+#include "Common/TraversalClient.h"
+#include "Core/ConfigManager.h"
+#include "Core/Core.h"
+#include "Core/NetPlayServer.h"
+#include "DolphinQt2/GameList/GameList.h"
+#include "DolphinQt2/NetPlay/GameListDialog.h"
+#include "DolphinQt2/NetPlay/MD5Dialog.h"
+#include "DolphinQt2/NetPlay/PadMappingDialog.h"
+#include "DolphinQt2/QtUtils/QueueOnObject.h"
+#include "DolphinQt2/QtUtils/RunOnObject.h"
+#include "DolphinQt2/Settings.h"
+#include "VideoCommon/VideoConfig.h"
+
+NetPlayDialog::NetPlayDialog(QWidget* parent)
+    : QDialog(parent), m_game_list_model(Settings::Instance().GetGameListModel())
+{
+  setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint);
+
+  setWindowTitle(tr("Dolphin NetPlay"));
+
+  m_pad_mapping = new PadMappingDialog(this);
+  m_md5_dialog = new MD5Dialog(this);
+
+  CreateChatLayout();
+  CreatePlayersLayout();
+  CreateMainLayout();
+  ConnectWidgets();
+}
+
+void NetPlayDialog::CreateMainLayout()
+{
+  m_main_layout = new QGridLayout;
+  m_game_button = new QPushButton;
+  m_md5_box = new QComboBox;
+  m_start_button = new QPushButton(tr("Start"));
+  m_buffer_size_box = new QSpinBox;
+  m_save_sd_box = new QCheckBox(tr("Write save / SD-Card data"));
+  m_load_wii_box = new QCheckBox(tr("Load Wii Save"));
+  m_record_input_box = new QCheckBox(tr("Record inputs"));
+  m_buffer_label = new QLabel(tr("Buffer:"));
+  m_quit_button = new QPushButton(tr("Quit"));
+
+  m_game_button->setDefault(false);
+  m_game_button->setAutoDefault(false);
+
+  for (const QString& text :
+       {tr("MD5 Check:"), tr("Current game"), tr("Other game"), tr("SD card")})
+    m_md5_box->addItem(text);
+
+  m_main_layout->addWidget(m_game_button, 0, 0);
+  m_main_layout->addWidget(m_md5_box, 0, 1);
+  m_main_layout->addWidget(m_chat_box, 1, 0);
+  m_main_layout->addWidget(m_players_box, 1, 1);
+
+  auto* options_widget = new QHBoxLayout;
+
+  options_widget->addWidget(m_start_button);
+  options_widget->addWidget(m_buffer_label);
+  options_widget->addWidget(m_buffer_size_box);
+  options_widget->addWidget(m_save_sd_box);
+  options_widget->addWidget(m_load_wii_box);
+  options_widget->addWidget(m_record_input_box);
+  options_widget->addWidget(m_quit_button);
+  m_main_layout->addLayout(options_widget, 2, 0, 1, -1, Qt::AlignRight);
+
+  setLayout(m_main_layout);
+}
+
+void NetPlayDialog::CreateChatLayout()
+{
+  m_chat_box = new QGroupBox(tr("Chat"));
+  m_chat_edit = new QTextEdit;
+  m_chat_type_edit = new QLineEdit;
+  m_chat_send_button = new QPushButton(tr("Send"));
+
+  m_chat_send_button->setDefault(false);
+  m_chat_send_button->setAutoDefault(false);
+
+  m_chat_edit->setReadOnly(true);
+
+  auto* layout = new QGridLayout;
+
+  layout->addWidget(m_chat_edit, 0, 0, 1, -1);
+  layout->addWidget(m_chat_type_edit, 1, 0);
+  layout->addWidget(m_chat_send_button, 1, 1);
+
+  m_chat_box->setLayout(layout);
+}
+
+void NetPlayDialog::CreatePlayersLayout()
+{
+  m_players_box = new QGroupBox(tr("Players"));
+  m_room_box = new QComboBox;
+  m_hostcode_label = new QLabel;
+  m_hostcode_action_button = new QPushButton(tr("Copy"));
+  m_players_list = new QListWidget;
+  m_kick_button = new QPushButton(tr("Kick Player"));
+  m_assign_ports_button = new QPushButton(tr("Assign Controller Ports"));
+
+  auto* layout = new QGridLayout;
+
+  layout->addWidget(m_room_box, 0, 0);
+  layout->addWidget(m_hostcode_label, 0, 1);
+  layout->addWidget(m_hostcode_action_button, 0, 2);
+  layout->addWidget(m_players_list, 1, 0, 1, -1);
+  layout->addWidget(m_kick_button, 2, 0, 1, -1);
+  layout->addWidget(m_assign_ports_button, 3, 0, 1, -1);
+
+  m_players_box->setLayout(layout);
+}
+
+void NetPlayDialog::ConnectWidgets()
+{
+  // Players
+  connect(m_room_box, static_cast<void (QComboBox::*)(int)>(&QComboBox::currentIndexChanged), this,
+          &NetPlayDialog::UpdateGUI);
+  connect(m_hostcode_action_button, &QPushButton::clicked, [this] {
+    if (m_is_copy_button_retry && m_room_box->currentIndex() == 0)
+      g_TraversalClient->ReconnectToServer();
+    else
+      QApplication::clipboard()->setText(m_hostcode_label->text());
+  });
+  connect(m_players_list, &QListWidget::itemSelectionChanged, [this] {
+    int row = m_players_list->currentRow();
+    m_kick_button->setEnabled(row > 0 &&
+                              !m_players_list->currentItem()->data(Qt::UserRole).isNull());
+  });
+  connect(m_kick_button, &QPushButton::clicked, [this] {
+    auto id = m_players_list->currentItem()->data(Qt::UserRole).toInt();
+    Settings::Instance().GetNetPlayServer()->KickPlayer(id);
+  });
+  connect(m_assign_ports_button, &QPushButton::clicked, [this] {
+    m_pad_mapping->exec();
+
+    Settings::Instance().GetNetPlayServer()->SetPadMapping(m_pad_mapping->GetGCPadArray());
+    Settings::Instance().GetNetPlayServer()->SetWiimoteMapping(m_pad_mapping->GetWiimoteArray());
+  });
+
+  // Chat
+  connect(m_chat_send_button, &QPushButton::clicked, this, &NetPlayDialog::OnChat);
+  connect(m_chat_type_edit, &QLineEdit::returnPressed, this, &NetPlayDialog::OnChat);
+
+  // Other
+  connect(m_buffer_size_box, static_cast<void (QSpinBox::*)(int)>(&QSpinBox::valueChanged),
+          [this](int value) {
+            if (Settings::Instance().GetNetPlayServer() != nullptr)
+              Settings::Instance().GetNetPlayServer()->AdjustPadBufferSize(value);
+          });
+
+  connect(m_start_button, &QPushButton::clicked, this, &NetPlayDialog::OnStart);
+  connect(m_quit_button, &QPushButton::clicked, this, &NetPlayDialog::reject);
+  connect(m_md5_box, static_cast<void (QComboBox::*)(int)>(&QComboBox::currentIndexChanged), this,
+          &NetPlayDialog::OnMD5Combo);
+
+  connect(m_game_button, &QPushButton::clicked, [this] {
+    GameListDialog gld(this);
+    if (gld.exec() == QDialog::Accepted)
+    {
+      auto unique_id = gld.GetSelectedUniqueID();
+      Settings::Instance().GetNetPlayServer()->ChangeGame(unique_id.toStdString());
+    }
+  });
+
+  connect(this, &NetPlayDialog::EmulationStopped, this, [this] {
+    if (isVisible())
+      GameStatusChanged(false);
+  });
+}
+
+void NetPlayDialog::OnChat()
+{
+  QueueOnObject(this, [this] {
+    auto msg = m_chat_type_edit->text().toStdString();
+    Settings::Instance().GetNetPlayClient()->SendChatMessage(msg);
+    m_chat_type_edit->clear();
+
+    DisplayMessage(QStringLiteral("%1: %2").arg(QString::fromStdString(m_nickname),
+                                                QString::fromStdString(msg)),
+                   "blue");
+  });
+}
+
+void NetPlayDialog::OnStart()
+{
+  if (!Settings::Instance().GetNetPlayClient()->DoAllPlayersHaveGame())
+  {
+    if (QMessageBox::question(this, tr("Warning"),
+                              tr("Not all players have the game. Do you really want to start?")) ==
+        QMessageBox::No)
+      return;
+  }
+
+  NetSettings settings;
+
+  // Copy all relevant settings
+  SConfig& instance = SConfig::GetInstance();
+  settings.m_CPUthread = instance.bCPUThread;
+  settings.m_CPUcore = instance.iCPUCore;
+  settings.m_EnableCheats = instance.bEnableCheats;
+  settings.m_SelectedLanguage = instance.SelectedLanguage;
+  settings.m_OverrideGCLanguage = instance.bOverrideGCLanguage;
+  settings.m_ProgressiveScan = instance.bProgressive;
+  settings.m_PAL60 = instance.bPAL60;
+  settings.m_DSPHLE = instance.bDSPHLE;
+  settings.m_DSPEnableJIT = instance.m_DSPEnableJIT;
+  settings.m_WriteToMemcard = m_save_sd_box->isChecked();
+  settings.m_CopyWiiSave = m_load_wii_box->isChecked();
+  settings.m_OCEnable = instance.m_OCEnable;
+  settings.m_OCFactor = instance.m_OCFactor;
+  settings.m_EXIDevice[0] = instance.m_EXIDevice[0];
+  settings.m_EXIDevice[1] = instance.m_EXIDevice[1];
+
+  Settings::Instance().GetNetPlayServer()->SetNetSettings(settings);
+  Settings::Instance().GetNetPlayServer()->StartGame();
+}
+
+void NetPlayDialog::OnMD5Combo(int index)
+{
+  std::string identifier;
+
+  switch (index)
+  {
+  case 0:
+    return;
+  case 1:  // Current game
+    identifier = m_current_game;
+    break;
+  case 2:  // Other game
+  {
+    GameListDialog gld(this);
+
+    if (gld.exec() == QDialog::Accepted)
+    {
+      identifier = gld.GetSelectedUniqueID().toStdString();
+      break;
+    }
+    else
+    {
+      m_md5_box->setCurrentIndex(0);
+      return;
+    }
+  }
+  case 3:  // SD Card
+    identifier = WII_SDCARD;
+    break;
+  }
+
+  Settings::Instance().GetNetPlayServer()->ComputeMD5(identifier);
+}
+
+void NetPlayDialog::reject()
+{
+  if (QMessageBox::question(this, tr("Confirmation"),
+                            tr("Are you sure you want to quit NetPlay?")) == QMessageBox::Yes)
+  {
+    QDialog::reject();
+  }
+}
+
+void NetPlayDialog::show(std::string nickname, bool use_traversal)
+{
+  m_nickname = nickname;
+  m_use_traversal = use_traversal;
+
+  m_room_box->clear();
+  m_chat_edit->clear();
+  m_chat_type_edit->clear();
+
+  bool is_hosting = Settings::Instance().GetNetPlayServer() != nullptr;
+
+  if (is_hosting)
+  {
+    if (use_traversal)
+      m_room_box->addItem(tr("Room ID"));
+
+    for (const auto& iface : Settings::Instance().GetNetPlayServer()->GetInterfaceSet())
+      m_room_box->addItem(QString::fromStdString(iface));
+  }
+
+  m_start_button->setHidden(!is_hosting);
+  m_save_sd_box->setHidden(!is_hosting);
+  m_load_wii_box->setHidden(!is_hosting);
+  m_buffer_size_box->setHidden(!is_hosting);
+  m_buffer_label->setHidden(!is_hosting);
+  m_kick_button->setHidden(!is_hosting);
+  m_assign_ports_button->setHidden(!is_hosting);
+  m_md5_box->setHidden(!is_hosting);
+  m_room_box->setHidden(!is_hosting);
+  m_hostcode_label->setHidden(!is_hosting);
+  m_hostcode_action_button->setHidden(!is_hosting);
+  m_game_button->setEnabled(is_hosting);
+  m_kick_button->setEnabled(false);
+
+  QDialog::show();
+  UpdateGUI();
+}
+
+void NetPlayDialog::UpdateGUI()
+{
+  // Update player list
+  std::vector<int> player_ids;
+  std::string tmp;
+
+  Settings::Instance().GetNetPlayClient()->GetPlayerList(tmp, player_ids);
+
+  std::istringstream ss(tmp);
+
+  int row = m_players_list->currentRow();
+  unsigned int i = 0;
+
+  m_players_list->clear();
+
+  while (std::getline(ss, tmp))
+  {
+    auto text = QString::fromStdString(tmp);
+    if (!text.isEmpty())
+    {
+      QListWidgetItem* item = new QListWidgetItem(text);
+
+      if (player_ids.size() > i && !text.startsWith(QStringLiteral("Ping:")) &&
+          !text.startsWith(QStringLiteral("Status:")))
+      {
+        item->setData(Qt::UserRole, player_ids[i]);
+        i++;
+      }
+      m_players_list->addItem(item);
+    }
+  }
+
+  if (row != -1)
+    m_players_list->setCurrentRow(row, QItemSelectionModel::SelectCurrent);
+
+  // Update Room ID / IP label
+  if (m_use_traversal && m_room_box->currentIndex() == 0)
+  {
+    switch (g_TraversalClient->m_State)
+    {
+    case TraversalClient::Connecting:
+      m_hostcode_label->setText(tr("..."));
+      m_hostcode_action_button->setEnabled(false);
+      break;
+    case TraversalClient::Connected:
+      m_hostcode_label->setText(QString::fromStdString(
+          std::string(g_TraversalClient->m_HostId.data(), g_TraversalClient->m_HostId.size())));
+      m_hostcode_action_button->setEnabled(true);
+      m_hostcode_action_button->setText(tr("Copy"));
+      m_is_copy_button_retry = false;
+      break;
+    case TraversalClient::Failure:
+      m_hostcode_label->setText(tr("Error"));
+      m_hostcode_action_button->setText(tr("Retry"));
+      m_hostcode_action_button->setEnabled(true);
+      m_is_copy_button_retry = true;
+      break;
+    }
+  }
+  else if (Settings::Instance().GetNetPlayServer())
+  {
+    m_hostcode_label->setText(
+        QString::fromStdString(Settings::Instance().GetNetPlayServer()->GetInterfaceHost(
+            m_room_box->currentText().toStdString())));
+    m_hostcode_action_button->setText(tr("Copy"));
+    m_hostcode_action_button->setEnabled(true);
+  }
+}
+
+// NetPlayUI methods
+
+void NetPlayDialog::BootGame(const std::string& filename)
+{
+  emit Boot(QString::fromStdString(filename));
+}
+
+void NetPlayDialog::StopGame()
+{
+  emit Stop();
+}
+
+void NetPlayDialog::Update()
+{
+  QueueOnObject(this, [this] { UpdateGUI(); });
+}
+
+void NetPlayDialog::DisplayMessage(const QString& msg, const std::string& color, int duration)
+{
+  QueueOnObject(m_chat_edit, [this, color, msg] {
+    m_chat_edit->append(
+        QStringLiteral("<font color='%1'>%2</font>").arg(QString::fromStdString(color), msg));
+  });
+
+  if (g_ActiveConfig.bShowNetPlayMessages && Core::IsRunning())
+  {
+    u32 osd_color;
+
+    // Convert the color string to a OSD color
+    if (color == "red")
+      osd_color = OSD::Color::RED;
+    else if (color == "cyan")
+      osd_color = OSD::Color::CYAN;
+    else if (color == "green")
+      osd_color = OSD::Color::GREEN;
+    else
+      osd_color = OSD::Color::YELLOW;
+
+    OSD::AddTypedMessage(OSD::MessageType::NetPlayBuffer, msg.toStdString(), OSD::Duration::NORMAL,
+                         osd_color);
+  }
+}
+
+void NetPlayDialog::AppendChat(const std::string& msg)
+{
+  DisplayMessage(QString::fromStdString(msg), "");
+}
+
+void NetPlayDialog::OnMsgChangeGame(const std::string& title)
+{
+  QString qtitle = QString::fromStdString(title);
+  QueueOnObject(this, [this, qtitle, title] {
+    m_game_button->setText(qtitle);
+    m_current_game = title;
+  });
+  DisplayMessage(tr("Game changed to \"%1\"").arg(qtitle), "pink");
+}
+
+void NetPlayDialog::GameStatusChanged(bool running)
+{
+  QueueOnObject(this, [this, running] {
+    if (Settings::Instance().GetNetPlayServer() != nullptr)
+    {
+      m_start_button->setEnabled(!running);
+      m_game_button->setEnabled(!running);
+      m_load_wii_box->setEnabled(!running);
+      m_save_sd_box->setEnabled(!running);
+      m_assign_ports_button->setEnabled(!running);
+    }
+
+    m_record_input_box->setEnabled(!running);
+  });
+}
+
+void NetPlayDialog::OnMsgStartGame()
+{
+  DisplayMessage(tr("Started game"), "green");
+  GameStatusChanged(true);
+
+  QueueOnObject(this, [this] {
+    Settings::Instance().GetNetPlayClient()->StartGame(FindGame(m_current_game));
+  });
+}
+
+void NetPlayDialog::OnMsgStopGame()
+{
+  DisplayMessage(tr("Stopped game"), "red");
+  GameStatusChanged(false);
+}
+
+void NetPlayDialog::OnPadBufferChanged(u32 buffer)
+{
+  QueueOnObject(this, [this, buffer] { m_buffer_size_box->setValue(buffer); });
+  DisplayMessage(tr("Pad size changed to %1").arg(buffer), "gray");
+}
+
+void NetPlayDialog::OnDesync(u32 frame, const std::string& player)
+{
+  DisplayMessage(tr("Possible desync detected: %1 might have desynced at frame %2")
+                     .arg(QString::fromStdString(player), QString::number(frame)),
+                 "red", OSD::Duration::VERY_LONG);
+}
+
+void NetPlayDialog::OnConnectionLost()
+{
+  DisplayMessage(tr("Lost connection to NetPlay server..."), "red");
+}
+
+void NetPlayDialog::OnTraversalError(int error)
+{
+  QueueOnObject(this, [this, error] {
+    switch (error)
+    {
+    case TraversalClient::BadHost:
+      QMessageBox::critical(this, tr("Traversal Error"), tr("Couldn't look up central server"));
+      QDialog::reject();
+      break;
+    case TraversalClient::VersionTooOld:
+      QMessageBox::critical(this, tr("Traversal Error"),
+                            tr("Dolphin is too old for traversal server"));
+      QDialog::reject();
+      break;
+    case TraversalClient::ServerForgotAboutUs:
+    case TraversalClient::SocketSendError:
+    case TraversalClient::ResendTimeout:
+      UpdateGUI();
+      break;
+    }
+  });
+}
+
+bool NetPlayDialog::IsRecording()
+{
+  return RunOnObject(this, [this] { return m_record_input_box->isChecked(); });
+}
+
+std::string NetPlayDialog::FindGame(const std::string& game)
+{
+  return RunOnObject(this, [this, game] {
+    for (int i = 0; i < m_game_list_model->rowCount(QModelIndex()); i++)
+    {
+      if (m_game_list_model->GetUniqueID(i).toStdString() == game)
+        return m_game_list_model->GetPath(i).toStdString();
+    }
+    return std::string("");
+  });
+}
+
+void NetPlayDialog::ShowMD5Dialog(const std::string& file_identifier)
+{
+  QueueOnObject(this, [this, file_identifier] {
+    m_md5_box->setEnabled(false);
+    m_md5_box->setCurrentIndex(0);
+
+    if (m_md5_dialog->isVisible())
+      m_md5_dialog->close();
+
+    m_md5_dialog->show(QString::fromStdString(file_identifier));
+  });
+}
+
+void NetPlayDialog::SetMD5Progress(int pid, int progress)
+{
+  QueueOnObject(this, [this, pid, progress] {
+    if (m_md5_dialog->isVisible())
+      m_md5_dialog->SetProgress(pid, progress);
+  });
+}
+
+void NetPlayDialog::SetMD5Result(int pid, const std::string& result)
+{
+  QueueOnObject(this, [this, pid, result] {
+    m_md5_dialog->SetResult(pid, result);
+    m_md5_box->setEnabled(true);
+  });
+}
+
+void NetPlayDialog::AbortMD5()
+{
+  QueueOnObject(this, [this] {
+    m_md5_dialog->close();
+    m_md5_box->setEnabled(true);
+  });
+}