dolphin/Source/Core/DolphinQt/NetPlay/ChunkedProgressDialog.cpp

// Copyright 2017 Dolphin Emulator Project
// SPDX-License-Identifier: GPL-2.0-or-later

#include "DolphinQt/NetPlay/ChunkedProgressDialog.h"

#include <algorithm>
#include <cmath>
#include <functional>

#include <fmt/format.h>

#include <QDialogButtonBox>
#include <QGroupBox>
#include <QLabel>
#include <QProgressBar>
#include <QPushButton>
#include <QVBoxLayout>

#include "Core/NetPlayClient.h"
#include "Core/NetPlayServer.h"

#include "DolphinQt/Settings.h"

static QString GetPlayerNameFromPID(int pid)
{
  QString player_name = QObject::tr("Invalid Player ID");
  auto client = Settings::Instance().GetNetPlayClient();
  if (!client)
    return player_name;

  for (const auto* player : client->GetPlayers())
  {
    if (player->pid == pid)
    {
      player_name = QString::fromStdString(player->name);
      break;
    }
  }
  return player_name;
}

ChunkedProgressDialog::ChunkedProgressDialog(QWidget* parent) : QDialog(parent)
{
  CreateWidgets();
  ConnectWidgets();
  setWindowTitle(tr("Data Transfer"));
  setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint);
}

void ChunkedProgressDialog::CreateWidgets()
{
  m_main_layout = new QVBoxLayout;
  m_progress_box = new QGroupBox;
  m_progress_layout = new QVBoxLayout;
  m_button_box = new QDialogButtonBox(QDialogButtonBox::NoButton);

  m_progress_box->setLayout(m_progress_layout);

  m_main_layout->addWidget(m_progress_box);
  m_main_layout->addWidget(m_button_box);
  setLayout(m_main_layout);
}

void ChunkedProgressDialog::ConnectWidgets()
{
  connect(m_button_box, &QDialogButtonBox::rejected, this, &ChunkedProgressDialog::reject);
}

void ChunkedProgressDialog::show(const QString& title, const u64 data_size,
                                 const std::vector<int>& players)
{
  m_progress_box->setTitle(title);
  m_data_size = data_size;

  for (auto& pair : m_progress_bars)
  {
    m_progress_layout->removeWidget(pair.second);
    pair.second->deleteLater();
  }

  for (auto& pair : m_status_labels)
  {
    m_progress_layout->removeWidget(pair.second);
    pair.second->deleteLater();
  }

  m_progress_bars.clear();
  m_status_labels.clear();

  auto client = Settings::Instance().GetNetPlayClient();
  if (!client)
    return;

  if (Settings::Instance().GetNetPlayServer())
  {
    m_button_box->setStandardButtons(QDialogButtonBox::Cancel);
    QPushButton* cancel_button = m_button_box->button(QDialogButtonBox::Cancel);
    cancel_button->setAutoDefault(false);
    cancel_button->setDefault(false);
  }
  else
  {
    m_button_box->setStandardButtons(QDialogButtonBox::Close);
    QPushButton* close_button = m_button_box->button(QDialogButtonBox::Close);
    close_button->setAutoDefault(false);
    close_button->setDefault(false);
  }

  for (const auto* player : client->GetPlayers())
  {
    if (std::find(players.begin(), players.end(), player->pid) == players.end())
      continue;

    m_progress_bars[player->pid] = new QProgressBar;
    m_status_labels[player->pid] = new QLabel;

    m_progress_layout->addWidget(m_progress_bars[player->pid]);
    m_progress_layout->addWidget(m_status_labels[player->pid]);
  }

  QDialog::show();
}

void ChunkedProgressDialog::SetProgress(const int pid, const u64 progress)
{
  QString player_name = GetPlayerNameFromPID(pid);

  if (!m_status_labels.contains(pid))
    return;

  const float acquired = progress / 1024.0f / 1024.0f;
  const float total = m_data_size / 1024.0f / 1024.0f;
  const int prog = std::lround((static_cast<float>(progress) / m_data_size) * 100.0f);

  m_status_labels[pid]->setText(tr("%1[%2]: %3/%4 MiB")
                                    .arg(player_name, QString::number(pid),
                                         QString::fromStdString(fmt::format("{:.2f}", acquired)),
                                         QString::fromStdString(fmt::format("{:.2f}", total))));
  m_progress_bars[pid]->setValue(prog);
}

void ChunkedProgressDialog::reject()
{
  auto server = Settings::Instance().GetNetPlayServer();

  if (server)
    server->AbortGameStart();

  QDialog::reject();
}