main_window.cpp

#include <QtCore/QDebug>
#include <QMessageBox>
#include <QFileDialog>
#include <QDropEvent>
#include <QUrl>
#include <QtConcurrentRun>
#include <QMouseEvent>
#include <QSettings>
#include <QProgressDialog>

#include "main_window.h"
#include "tab_page.h"
#include "resize_dialog.h"

const char FORMAT_LIST[] = "Images (*.png *.jpg *.jpeg *.bmp *.pbm *.pgm *.ppm *.tiff *.xbm *.xpm *.nef *.cr2 *.crw *.raf *.dng *.mos *.kdc *.dcr)";

static const QString LOADDIR_KEY("init-file-choose-dir");

MainWindow::MainWindow(QWidget* parent):
    QMainWindow(parent)
{
    ui.setupUi(this);
    setTabActionsEnabled(false);

    connect(ui.actionAbout_Qt, SIGNAL(triggered()),
            this, SLOT(aboutQT()));
    connect(ui.actionAbout, SIGNAL(triggered()),
            this, SLOT(about()));
    connect(ui.action_Open, SIGNAL(triggered()),
            this, SLOT(chooseFiles()));
    connect(ui.action_Resize, SIGNAL(triggered()),
            this, SLOT(resizeDialog()));
    connect(ui.actionZoom_In, SIGNAL(triggered()),
            this, SLOT(zoomIn()));
    connect(ui.actionZoom_Out, SIGNAL(triggered()),
            this, SLOT(zoomOut()));
    connect(ui.action_Close, SIGNAL(triggered()),
            this, SLOT(closeCurrentTab()));
    connect(ui.actionClose_All, SIGNAL(triggered()),
            this, SLOT(closeAllTabs()));
    connect(ui.tabWidget, SIGNAL(tabCloseRequested(int)),
            this, SLOT(closeTab(int)));
    connect(ui.tabWidget, SIGNAL(currentChanged(int)),
            this, SLOT(displayCurrentTab()));
}

void MainWindow::setTabIcon(int index)
{
    static QIcon icon(":/icon/icons/image.svg");
    ui.tabWidget->setTabIcon(index, icon);
}

void MainWindow::closeAllTabs()
{
    for (int i=0; i<ui.tabWidget->count(); ++i) {
        auto page = ui.tabWidget->widget(i);
        delete page;
    }
    ui.tabWidget->clear();
}

void MainWindow::closeCurrentTab()
{
    closeTab(ui.tabWidget->currentIndex());
}

void MainWindow::setTabActionsEnabled(bool flag)
{
    ui.actionZoom_In->setEnabled(flag);
    ui.actionZoom_Out->setEnabled(flag);
    ui.action_Close->setEnabled(flag);
    ui.action_Resize->setEnabled(flag);
}

void MainWindow::mouseDoubleClickEvent(QMouseEvent* event)
{
    chooseFiles();
}

void MainWindow::zoomIn()
{
    TabPage* page = qobject_cast<TabPage*>(ui.tabWidget->currentWidget());
    page->zoomIn();
}

void MainWindow::zoomOut()
{
    TabPage* page = qobject_cast<TabPage*>(ui.tabWidget->currentWidget());
    page->zoomOut();
}

void MainWindow::resizeDialog()
{
    TabPage* page = qobject_cast<TabPage*>(ui.tabWidget->currentWidget());

    ResizeDialog dialog(page->getHeight(), page->getWidth(), this);
    connect(&dialog, 
            SIGNAL(dialogFinished(int, int, bool, bool, const QString&,
                    int)),
            this, SLOT(resizeOpts(int, int, bool, bool, const QString&,
                    int)));
    dialog.exec();
}

void MainWindow::saveImage(int index, int height, int width, bool sameDir,
        const QString& outdir, int quality) 
{
    TabPage* page = 
            qobject_cast<TabPage*>(ui.tabWidget->widget(index));
    auto dir = sameDir ? QFileInfo(page->getFullpath()).path() : outdir;
    page->save(height, width, dir, quality);
}

void MainWindow::resizeOpts(int height, int width, bool applyToAll,
        bool sameDir, const QString& outdir, int quality)
{
    if (applyToAll) {
        auto count = ui.tabWidget->count();
        QProgressDialog progress(tr("Resizing..."), tr("Abort"), 0, count,
                this);
        progress.setWindowModality(Qt::WindowModal);
        for (int i=0; i<count; ++i) {
            progress.setValue(i);
            if (progress.wasCanceled())
                break;
            saveImage(i, height, width, sameDir, outdir, quality);
        }
        progress.setValue(count);
    } else {
        QProgressDialog progress(tr("Resizing..."), tr("Abort"), 0, 1,
                this);
        progress.setWindowModality(Qt::WindowModal);

        saveImage(ui.tabWidget->currentIndex(), height, width, sameDir,
                outdir, quality);
        progress.setValue(1);
    }
}

void MainWindow::dragEnterEvent(QDragEnterEvent* event)
{
    if (event->mimeData()->hasFormat("text/uri-list")) {
        event->acceptProposedAction();
    }
}

void MainWindow::dropEvent(QDropEvent* event)
{
    qDebug() << "Drop event";
    auto mimeData = event->mimeData();
    if (mimeData->hasUrls()) {
        QStringList pathList;
        auto urlList = mimeData->urls();
        for (int i=0; i<urlList.size(); ++i) {
            pathList.append(urlList.at(i).toLocalFile());
        }
        addTabs(pathList);
    }
}

void MainWindow::displayCurrentTab()
{
    if (ui.tabWidget->count() == 0) {
        return;
    }
    TabPage* page = qobject_cast<TabPage*>(ui.tabWidget->currentWidget());
    Q_ASSERT(page != nullptr);
    page->loadImage();
    setTabActionsEnabled(true);
}


void MainWindow::closeTab(int index)
{
    auto page = ui.tabWidget->widget(index);
    Q_ASSERT(page != nullptr);
    ui.tabWidget->removeTab(index);
    delete page;
    if (ui.tabWidget->count() == 0) {
        setTabActionsEnabled(false);
    }
}
void MainWindow::loadErrorHandler(const QString& fullpath)
{
    QMessageBox::information(this, tr("Error!"),
            tr("Cannot load %1.").arg(fullpath));
    TabPage* page = qobject_cast<TabPage*>(sender());
    auto idx = ui.tabWidget->indexOf(page);
    closeTab(idx);
}


void MainWindow::aboutQT()
{
    QMessageBox::aboutQt(this);
}

void MainWindow::about()
{
    QMessageBox::about(this, "ImgResize", tr("Resize your images."));
}

void MainWindow::addTabs(const QStringList& pathList)
{
    auto it = pathList.begin();
    while (it != pathList.end()) {
        auto page = new TabPage(*it);
        lastTabIdx = ui.tabWidget->addTab(page,
                QFileInfo(*it).fileName());
        connect(page, SIGNAL(loadError(const QString&)),
                this, SLOT(loadErrorHandler(const QString&)));
        setTabIcon(lastTabIdx);
        qDebug() << "Tab added: " << *it;
        ++it;
    }
    ui.tabWidget->setCurrentIndex(lastTabIdx);
}

void MainWindow::chooseFiles()
{
    QSettings settings;
    auto initdir = settings.value(LOADDIR_KEY, ".").toString();
    auto files = QFileDialog::getOpenFileNames(
            this,
            tr("Select one or more files to open"),
            initdir,
            FORMAT_LIST);
    addTabs(files);
    if (files.size() > 0) {
        settings.setValue(LOADDIR_KEY, files.first());
    }
}