#include <QCloseEvent>

#include <QEventLoop>

#include <QProgressBar>

#include <QTime>

#include <QTimer>

#include <QVBoxLayout>

#include "UIProgressDialog.h"

#include "QIDialogButtonBox.h"

#include "QILabel.h"

#include "UISpecialControls.h"

#include "VBoxGlobal.h"

#ifdef Q_WS_MAC

# include "VBoxUtils-darwin.h"

#endif /* Q_WS_MAC */

#include "CProgress.h"

const char *UIProgressDialog::m_spcszOpDescTpl = "%1 ... (%2/%3)";

UIProgressDialog::UIProgressDialog(CProgress &progress,

const QString &strTitle,

QPixmap *pImage /* = 0 */,

bool fSheetOnDarwin /* = false */,

int cMinDuration /* = 2000 */,

QWidget *pParent /* = 0 */)

: QIWithRetranslateUI2<QIDialog>(pParent, Qt::MSWindowsFixedSizeDialogHint | Qt::WindowTitleHint)

, m_progress(progress)

, m_pImageLbl(0)

, m_fCancelEnabled(false)

, m_cOperations(m_progress.GetOperationCount())

, m_iCurrentOperation(m_progress.GetOperation() + 1)

, m_fEnded(false)

{

/* Setup dialog: */

setModal(true);

setWindowTitle(QString("%1: %2").arg(strTitle, m_progress.GetDescription()));

setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);

/* Create main layout: */

QHBoxLayout *pMainLayout = new QHBoxLayout(this);

#ifdef Q_WS_MAC

/* Check if Mac Sheet is allowed: */

if (fSheetOnDarwin && vboxGlobal().isSheetWindowAllowed(pParent))

{

vboxGlobal().setSheetWindowUsed(pParent, true);

setWindowFlags(Qt::Sheet);

}

::darwinSetHidesAllTitleButtons(this);

::darwinSetShowsResizeIndicator(this, false);

if (pImage)

pMainLayout->setContentsMargins(30, 15, 30, 15);

else

pMainLayout->setContentsMargins(6, 6, 6, 6);

#else /* Q_WS_MAC */

NOREF(fSheetOnDarwin);

#endif /* !Q_WS_MAC */

/* Create image: */

if (pImage)

{

m_pImageLbl = new QLabel(this);

m_pImageLbl->setPixmap(*pImage);

pMainLayout->addWidget(m_pImageLbl);

}

/* Create description: */

m_pDescriptionLbl = new QILabel(this);

if (m_cOperations > 1)

m_pDescriptionLbl->setText(QString(m_spcszOpDescTpl)

.arg(m_progress.GetOperationDescription())

.arg(m_iCurrentOperation).arg(m_cOperations));

else

m_pDescriptionLbl->setText(QString("%1 ...")

.arg(m_progress.GetOperationDescription()));

/* Create progress-bar: */

m_pProgressBar = new QProgressBar(this);

m_pProgressBar->setMaximum(100);

m_pProgressBar->setValue(0);

/* Create cancel button: */

m_fCancelEnabled = m_progress.GetCancelable();

m_pCancelBtn = new UIMiniCancelButton(this);

m_pCancelBtn->setEnabled(m_fCancelEnabled);

m_pCancelBtn->setFocusPolicy(Qt::ClickFocus);

connect(m_pCancelBtn, SIGNAL(clicked()), this, SLOT(sltCancelOperation()));

/* Create estimation label: */

m_pEtaLbl = new QILabel(this);

/* Create proggress layout: */

QHBoxLayout *pProgressLayout = new QHBoxLayout;

pProgressLayout->setMargin(0);

pProgressLayout->addWidget(m_pProgressBar, 0, Qt::AlignVCenter);

pProgressLayout->addWidget(m_pCancelBtn, 0, Qt::AlignVCenter);

/* Create description layout: */

QVBoxLayout *pDescriptionLayout = new QVBoxLayout;

pDescriptionLayout->setMargin(0);

pDescriptionLayout->addStretch(1);

pDescriptionLayout->addWidget(m_pDescriptionLbl, 0, Qt::AlignHCenter);

pDescriptionLayout->addLayout(pProgressLayout);

pDescriptionLayout->addWidget(m_pEtaLbl, 0, Qt::AlignLeft | Qt::AlignVCenter);

pDescriptionLayout->addStretch(1);

pMainLayout->addLayout(pDescriptionLayout);

/* Translate finally: */

retranslateUi();

/* The progress dialog will be shown automatically after

QTimer::singleShot(cMinDuration, this, SLOT(sltShowDialog()));

}

UIProgressDialog::~UIProgressDialog()

{

#ifdef Q_WS_MAC

/* Check if Mac Sheet was used: */

if ((windowFlags() & Qt::Sheet) == Qt::Sheet)

vboxGlobal().setSheetWindowUsed(parentWidget(), false);

#endif /* Q_WS_MAC */

}

void UIProgressDialog::retranslateUi()

{

m_strCancel = tr("Canceling...");

m_pCancelBtn->setText(tr("&Cancel"));

m_pCancelBtn->setToolTip(tr("Cancel the current operation"));

}

int UIProgressDialog::run(int cRefreshInterval)

{

if (m_progress.isOk())

{

/* Start refresh timer */

int id = startTimer(cRefreshInterval);

/* Set busy cursor.

#ifndef Q_WS_MAC

if (m_fCancelEnabled)

QApplication::setOverrideCursor(QCursor(Qt::BusyCursor));

else

QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));

#endif /* Q_WS_MAC */

/* Enter the modal loop, but don't show the window immediately */

exec(false);

/* Kill refresh timer */

killTimer(id);

#ifndef Q_WS_MAC

/* Reset the busy cursor */

QApplication::restoreOverrideCursor();

#endif /* Q_WS_MAC */

return result();

}

return Rejected;

}

void UIProgressDialog::reject()

{

if (m_fCancelEnabled)

sltCancelOperation();

}

void UIProgressDialog::timerEvent(QTimerEvent* /* pEvent */)

{

/* We should hide progress-dialog

if (m_fEnded && !isHidden())

{

hide();

return;

}

else if (m_fEnded)

return;

if (!m_fEnded && (!m_progress.isOk() || m_progress.GetCompleted()))

{

/* Progress finished: */

if (m_progress.isOk())

{

m_pProgressBar->setValue(100);

done(Accepted);

}

/* Progress is not valid: */

else

done(Rejected);

/* Request to exit loop: */

m_fEnded = true;

return;

}

if (!m_progress.GetCanceled())

{

/* Update the progress dialog: */

/* First ETA */

long newTime = m_progress.GetTimeRemaining();

long seconds;

long minutes;

long hours;

long days;

seconds = newTime < 0 ? 0 : newTime;

minutes = seconds / 60;

seconds -= minutes * 60;

hours = minutes / 60;

minutes -= hours * 60;

days = hours / 24;

hours -= days * 24;

QString strDays = VBoxGlobal::daysToString(days);

QString strHours = VBoxGlobal::hoursToString(hours);

QString strMinutes = VBoxGlobal::minutesToString(minutes);

QString strSeconds = VBoxGlobal::secondsToString(seconds);

QString strTwoComp = tr("%1, %2 remaining", "You may wish to translate this more like \"Time remaining: %1, %2\"");

QString strOneComp = tr("%1 remaining", "You may wish to translate this more like \"Time remaining: %1\"");

if (days > 1 && hours > 0)

m_pEtaLbl->setText(strTwoComp.arg(strDays).arg(strHours));

else if (days > 1)

m_pEtaLbl->setText(strOneComp.arg(strDays));

else if (days > 0 && hours > 0)

m_pEtaLbl->setText(strTwoComp.arg(strDays).arg(strHours));

else if (days > 0 && minutes > 5)

m_pEtaLbl->setText(strTwoComp.arg(strDays).arg(strMinutes));

else if (days > 0)

m_pEtaLbl->setText(strOneComp.arg(strDays));

else if (hours > 2)

m_pEtaLbl->setText(strOneComp.arg(strHours));

else if (hours > 0 && minutes > 0)

m_pEtaLbl->setText(strTwoComp.arg(strHours).arg(strMinutes));

else if (hours > 0)

m_pEtaLbl->setText(strOneComp.arg(strHours));

else if (minutes > 2)

m_pEtaLbl->setText(strOneComp.arg(strMinutes));

else if (minutes > 0 && seconds > 5)

m_pEtaLbl->setText(strTwoComp.arg(strMinutes).arg(strSeconds));

else if (minutes > 0)

m_pEtaLbl->setText(strOneComp.arg(strMinutes));

else if (seconds > 5)

m_pEtaLbl->setText(strOneComp.arg(strSeconds));

else if (seconds > 0)

m_pEtaLbl->setText(tr("A few seconds remaining"));

else

m_pEtaLbl->clear();

/* Then operation text if changed: */

ulong newOp = m_progress.GetOperation() + 1;

if (newOp != m_iCurrentOperation)

{

m_iCurrentOperation = newOp;

m_pDescriptionLbl->setText(QString(m_spcszOpDescTpl)

.arg(m_progress.GetOperationDescription())

.arg(m_iCurrentOperation).arg(m_cOperations));

}

m_pProgressBar->setValue(m_progress.GetPercent());

/* Then cancel button: */

m_fCancelEnabled = m_progress.GetCancelable();

m_pCancelBtn->setEnabled(m_fCancelEnabled);

}

else

m_pEtaLbl->setText(m_strCancel);

}

void UIProgressDialog::closeEvent(QCloseEvent *pEvent)

{

if (m_fCancelEnabled)

sltCancelOperation();

else

pEvent->ignore();

}

void UIProgressDialog::sltShowDialog()

{

/* We should not show progress-dialog

if (!m_fEnded)

show();

}

void UIProgressDialog::sltCancelOperation()

{

m_pCancelBtn->setEnabled(false);

m_progress.Cancel();

}