source: vbox/trunk/src/VBox/Frontends/VirtualBox/src/medium/UIFDCreationDialog.cpp@ 82781

/* $Id: UIFDCreationDialog.cpp 79365 2019-06-26 15:57:32Z vboxsync $ */
/** @file
 * VBox Qt GUI - UIFDCreationDialog class implementation.
 */

/*
 * Copyright (C) 2008-2019 Oracle Corporation
 *
 * This file is part of VirtualBox Open Source Edition (OSE), as
 * available from http://www.virtualbox.org. This file is free software;
 * you can redistribute it and/or modify it under the terms of the GNU
 * General Public License (GPL) as published by the Free Software
 * Foundation, in version 2 as it comes in the "COPYING" file of the
 * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
 * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
 */

/* Qt includes */
#include<QCheckBox>
#include<QDialogButtonBox>
#include<QDir>
#include<QGridLayout>
#include<QLabel>
#include<QPushButton>

/* GUI includes */
#include "UIFDCreationDialog.h"
#include "UIFilePathSelector.h"
#include "UIMedium.h"
#include "UIMessageCenter.h"
#include "UICommon.h"

/* COM includes: */
#include "CSystemProperties.h"
#include "CMedium.h"
#include "CMediumFormat.h"


UIFDCreationDialog::UIFDCreationDialog(QWidget *pParent,
                                           const QString &strDefaultFolder,
                                           const QString &strMachineName /* = QString() */)
   : QIWithRetranslateUI<QDialog>(pParent)
    , m_pFilePathselector(0)
    , m_pPathLabel(0)
    , m_pSizeLabel(0)
    , m_pSizeCombo(0)
    , m_pButtonBox(0)
    , m_pFormatCheckBox(0)
    , m_strDefaultFolder(strDefaultFolder)
    , m_strMachineName(strMachineName)
{

    prepare();
    /* Adjust dialog size: */
    adjustSize();

#ifdef VBOX_WS_MAC
    setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);
    setFixedSize(minimumSize());
#endif /* VBOX_WS_MAC */
}

void UIFDCreationDialog::retranslateUi()
{
    if (m_strMachineName.isEmpty())
        setWindowTitle(QString("%1").arg(tr("Floppy Disk Creator")));
    else
        setWindowTitle(QString("%1 - %2").arg(m_strMachineName).arg(tr("Floppy Disk Creator")));
    if (m_pPathLabel)
        m_pPathLabel->setText(tr("File Path:"));
    if (m_pSizeLabel)
        m_pSizeLabel->setText(tr("Size:"));
    if (m_pButtonBox)
        m_pButtonBox->button(QDialogButtonBox::Ok)->setText("Create");
    if (m_pFormatCheckBox)
        m_pFormatCheckBox->setText(tr("Format disk as FAT12"));
    if (m_pSizeCombo)
    {
        //m_pSizeCombo->setItemText(FDSize_2_88M, tr("2.88M"));
        m_pSizeCombo->setItemText(FDSize_1_44M, tr("1.44M"));
        m_pSizeCombo->setItemText(FDSize_1_2M, tr("1.2M"));
        m_pSizeCombo->setItemText(FDSize_720K, tr("720K"));
        m_pSizeCombo->setItemText(FDSize_360K, tr("360K"));
    }
}

void UIFDCreationDialog::prepare()
{

#ifndef VBOX_WS_MAC
    setWindowIcon(QIcon(":/fd_add_32px.png"));
#endif

    setWindowModality(Qt::WindowModal);
    setSizeGripEnabled(false);

    QGridLayout *pMainLayout = new QGridLayout;
    if (!pMainLayout)
        return;
    setLayout(pMainLayout);

    m_pPathLabel = new QLabel;
    if (m_pPathLabel)
    {
        pMainLayout->addWidget(m_pPathLabel, 0, 0, 1, 1);
        m_pPathLabel->setAlignment(Qt::AlignRight|Qt::AlignTrailing|Qt::AlignVCenter);
    }

    m_pFilePathselector = new UIFilePathSelector;
    if (m_pFilePathselector)
    {
        pMainLayout->addWidget(m_pFilePathselector, 0, 1, 1, 2);
        m_pFilePathselector->setMode(UIFilePathSelector::Mode_File_Save);
        QString strFolder = getDefaultFolder();
        m_pFilePathselector->setDefaultPath(strFolder);
        m_pFilePathselector->setPath(strFolder);
    }

    m_pSizeLabel = new QLabel;
    if (m_pSizeLabel)
    {
        pMainLayout->addWidget(m_pSizeLabel, 1, 0, 1, 1);
        m_pSizeLabel->setAlignment(Qt::AlignRight|Qt::AlignTrailing|Qt::AlignVCenter);
    }

    m_pSizeCombo = new QComboBox;
    if (m_pSizeCombo)
    {
        pMainLayout->addWidget(m_pSizeCombo, 1, 1, 1, 1);
        //m_pSizeCombo->insertItem(FDSize_2_88M, "2.88M", 2949120);
        m_pSizeCombo->insertItem(FDSize_1_44M, "1.44M", 1474560);
        m_pSizeCombo->insertItem(FDSize_1_2M, "1.2M", 1228800);
        m_pSizeCombo->insertItem(FDSize_720K, "720K", 737280);
        m_pSizeCombo->insertItem(FDSize_360K, "360K", 368640);
        m_pSizeCombo->setCurrentIndex(FDSize_1_44M);

    }

    m_pFormatCheckBox = new QCheckBox;
    if (m_pFormatCheckBox)
    {
        pMainLayout->addWidget(m_pFormatCheckBox, 2, 1, 1, 1);
        m_pFormatCheckBox->setCheckState(Qt::Checked);
    }

    m_pButtonBox =
        new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel, Qt::Horizontal, this);
    if (m_pButtonBox)
    {
        pMainLayout->addWidget(m_pButtonBox, 3, 0, 1, 3);
        connect(m_pButtonBox, &QDialogButtonBox::accepted, this, &UIFDCreationDialog::accept);
        connect(m_pButtonBox, &QDialogButtonBox::rejected, this, &UIFDCreationDialog::reject);
    }
    retranslateUi();
}

QString UIFDCreationDialog::getDefaultFolder() const
{
    QString strPreferredExtension = UIMediumDefs::getPreferredExtensionForMedium(KDeviceType_Floppy);

    QString strInitialPath = m_strDefaultFolder;
    if (strInitialPath.isEmpty())
        strInitialPath = uiCommon().virtualBox().GetSystemProperties().GetDefaultMachineFolder();

    if (strInitialPath.isEmpty())
        return strInitialPath;

    QString strDiskname = !(m_strMachineName.isEmpty()) ? m_strMachineName : "NewFloppyDisk";
    strDiskname = UICommon::findUniqueFileName(m_strDefaultFolder, strDiskname);

    strInitialPath = QDir(strInitialPath).absoluteFilePath(strDiskname + "." + strPreferredExtension);
    return strInitialPath;
}

void UIFDCreationDialog::accept()
{
    QVector<CMediumFormat>  mediumFormats = UIMediumDefs::getFormatsForDeviceType(KDeviceType_Floppy);

    if (m_pFilePathselector->path().isEmpty() || mediumFormats.isEmpty())
        return;

    CVirtualBox vbox = uiCommon().virtualBox();
    QString strMediumLocation = m_pFilePathselector->path();

    CMedium newMedium = vbox.CreateMedium(mediumFormats[0].GetName(), strMediumLocation,
                                          KAccessMode_ReadWrite, KDeviceType_Floppy);
    if (!vbox.isOk())
    {
        msgCenter().cannotCreateMediumStorage(vbox, strMediumLocation, this);
        return;
    }

    QVector<KMediumVariant> variants(1, KMediumVariant_Fixed);
    /* Decide if formatting the disk is required: */
    if (m_pFormatCheckBox && m_pFormatCheckBox->checkState() == Qt::Checked)
        variants.push_back(KMediumVariant_Formatted);
    CProgress progress = newMedium.CreateBaseStorage(m_pSizeCombo->currentData().toLongLong(), variants);

    if (!newMedium.isOk())
    {
        msgCenter().cannotCreateMediumStorage(newMedium, strMediumLocation, this);
        return;
    }
    /* Show creation progress: */
    msgCenter().showModalProgressDialog(progress, windowTitle(), ":/progress_media_create_90px.png", this);
    if (progress.GetCanceled())
        return;

    if (!progress.isOk() || progress.GetResultCode() != 0)
    {
        msgCenter().cannotCreateHardDiskStorage(progress, strMediumLocation, this);
        return;
    }
    /* Store the id of the newly create medium: */
    m_uMediumID = newMedium.GetId();

    /* Notify UICommon about the new medium: */
    uiCommon().createMedium(UIMedium(newMedium, UIMediumDeviceType_Floppy, KMediumState_Created));

    /* After a successful creation and initilization of the floppy disk we call base class accept
       effectively closing this dialog: */
    QDialog::accept();
}

QUuid UIFDCreationDialog::mediumID() const
{
    return m_uMediumID;
}