mainwindow.cpp

#include "mainwindow.h"
#include "ui_mainwindow.h"

#include <QMessageBox>
#include <QPainter>
#include <QPrintDialog>
#include <QPrintPreviewDialog>
#include <QTimer>

#include "tool/pencil.h"
#include "tool/eraser.h"
#include "tool/brush.h"
#include "tool/fillcolor.h"
#include "tool/eyedropper.h"
#include "tool/text.h"


#define DIALOG_CLOSE_MSG tr("Are you sure you want to quit the app? If you haven't saved the project, you will lose your last unsaved work.")


MainWindow::MainWindow(AppContext *context, QWidget *parent)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
{
    this->context = context;

    /*****************************************************************************/
    // zakladni inicializace UI
    this->ui->setupUi(this);
    // setaveni toolbaru
    this->colorPicker = new ColorPicker(this->ui->toolBar);
    this->ui->toolBar->addWidget(this->colorPicker);
    this->ui->toolBar->addAction(this->ui->actionZoom_in);
    this->ui->toolBar->addAction(this->ui->actionZoom_out);
    this->ui->toolBar->addAction(this->ui->actionEye_Dropper);
    this->ui->toolBar->addAction(this->ui->actionPen);
    this->ui->toolBar->addAction(this->ui->actionEraser);
    this->ui->toolBar->addAction(this->ui->actionBrush);
    this->ui->toolBar->addAction(this->ui->actionFill);
    this->ui->toolBar->addAction(this->ui->actionText);
    //this->ui->toolBar->addAction(this->ui->actionCircle);
    //this->ui->toolBar->addAction(this->ui->actionRectangle);
    //this->ui->toolBar->addAction(this->ui->actionPolygon);
    // label pro status bar
    this->statusLabel = new QLabel(this->ui->statusbar);
    this->statusLabel->setAlignment(Qt::AlignLeft);
    this->statusLabel->setText(tr("Project: None"));
    this->statusBar()->addPermanentWidget(this->statusLabel, 1);
    /*****************************************************************************/


    /*****************************************************************************/
    // inicializace kontextu
    this->context->setWorkspace(new Workspace(Workspace_defaultConfig(), this));
    this->context->setToolController(new ToolController(this));
    this->context->setLayerManager(new LayerManager(this));
    this->context->addTool(new PencilTool(this, this->colorPicker));
    this->context->addTool(new EraserTool(this));
    this->context->addTool(new BrushTool(this, this->colorPicker));
    this->context->addTool(new FillColorTool(this, this->colorPicker));
    this->context->addTool(new EyeDropperTool(this, this->colorPicker));
    this->context->addTool(new TextTool(this));
    this->context->selectToolFromList(TOOL_PENCIL);
    /*****************************************************************************/


    /*****************************************************************************/
    // sestaveni celkove pracovni plochy s vyuzitim splitteru
    this->splitter_horizontal = new QSplitter(Qt::Horizontal);
    this->splitter_horizontal->setObjectName("bg-widget");
    // leva strana (pracovani plocha pro upravu bitmapove grafiky)
    this->splitter_horizontal->addWidget(this->context->getWorkspace());

    // prava strany (manazer vrstev + ovladani aktualne vybraneho nastroje)
    this->splitter_vertical = new QSplitter(Qt::Vertical);
    this->splitter_vertical->setObjectName("bg-widget");
    // horni strana (ovladani nastroje)
    this->splitter_vertical->addWidget(this->context->getToolController());
    // dolni strana (manazer vrstev)
    this->splitter_vertical->addWidget(this->context->getLayerManager());

    this->splitter_horizontal->addWidget(this->splitter_vertical);
    this->setCentralWidget(this->splitter_horizontal);
    this->splitter_horizontal->setStretchFactor(0, 3);
    this->splitter_horizontal->setStretchFactor(1, 1);
    /*****************************************************************************/


    /*****************************************************************************/
    // inicializace oken
    this->window_newProject = new NewProject(this->context);
    this->window_openProject = new OpenProject(this->context);
    this->window_importImage = new ImportImage(this->context);
    this->window_exportProject = new ExportProject(this->context);

    this->window_about = new About();

    connect(this->window_newProject, SIGNAL(projectCreated()),
            this, SLOT(updateStatusBar()));
    connect(this->window_openProject, SIGNAL(projectOpened()),
            this, SLOT(updateStatusBar()));

    /*****************************************************************************/


    /*****************************************************************************/
    // vlastni styly
    this->splitter_horizontal->setStyleSheet("QWidget#bg-widget {background: rgb(49, 49, 49);}");
    /*****************************************************************************/


    // finish
    this->updateStatusBar();
    this->highlightToolbar(this->ui->actionPen);
}

MainWindow::~MainWindow()
{
    delete ui;
    delete this->context;

    if(this->colorPicker) delete this->colorPicker;
    if(this->splitter_horizontal) delete this->splitter_horizontal;
    if(this->statusLabel) delete this->statusLabel;

    if(this->window_newProject) delete this->window_newProject;
    if(this->window_openProject) delete this->window_openProject;
    if(this->window_importImage) delete this->window_importImage;
    if(this->window_exportProject) delete this->window_exportProject;
    if(this->window_about) delete this->window_about;
}

void MainWindow::showEvent(QShowEvent *event)
{
    QWidget::showEvent(event);
    QTimer::singleShot(200, this, SLOT(workspace_setDefaultScale()));
}

void MainWindow::closeEvent(QCloseEvent *event)
{
    QMessageBox::StandardButton reply;
    reply = QMessageBox::question(
                this,
                tr("Exit"),
                DIALOG_CLOSE_MSG,
                QMessageBox::Yes|QMessageBox::No);

    if(reply == QMessageBox::Yes) {
        event->accept();
    } else {
        event->ignore();
    }
}

void MainWindow::updateStatusBar()
{
    Project *p = this->context->getProject();
    if(p == NULL) {
        this->statusLabel->setText(tr("Project: None"));
    } else {
        this->statusLabel->setText(QString(tr("Project: %1 (%2)")).arg(p->getName(), p->getPath()));
    }
}

void MainWindow::on_actionNew_project_triggered()
{
    this->window_newProject->show();
}


void MainWindow::on_actionOpen_project_triggered()
{
    this->window_openProject->show();
}


void MainWindow::on_actionSave_project_triggered()
{
    if(this->context == NULL) return;
    if(this->context->getProject() == NULL) return;
    WriteProjectToFile(this->context->getProject());
    QMessageBox::information(this, tr("Project"), tr("Project saved successfully"));
}


void MainWindow::on_actionImport_image_triggered()
{
    this->window_importImage->show();
}


void MainWindow::on_actionExport_image_triggered()
{
    this->window_exportProject->show();
}


void MainWindow::on_actionExit_triggered()
{
    QMessageBox::StandardButton reply;
    reply = QMessageBox::question(
                this,
                tr("Exit"),
                DIALOG_CLOSE_MSG,
                QMessageBox::Yes|QMessageBox::No);

    if(reply == QMessageBox::Yes) {
        this->close();
    }
}


void MainWindow::on_actionPrint_triggered()
{
    if(this->context == NULL) return;
    if(this->context->getProject() == NULL) {
        return;
    }

    QPrinter printer(QPrinter::HighResolution);
    QPrintPreviewDialog printPreviewDialog(&printer, this);
    connect(&printPreviewDialog, &QPrintPreviewDialog::paintRequested,
            this, &MainWindow::printerPaint);
    printPreviewDialog.exec();
}

void MainWindow::printerPaint(QPrinter *printer) {
    if(printer == NULL) return;

    Project *p = this->context->getProject();
    if(p == NULL) {
        return;
    }

    QPainter painter;
    painter.begin(printer);

    QPrinter::Unit unit = QPrinter::DevicePixel;
    double xscale = printer->pageRect(unit).width() / double(p->getSize().width());
    double yscale = printer->pageRect(unit).height() / double(p->getSize().height());
    double scale = qMin(xscale, yscale);
    painter.translate(printer->paperRect(unit).x() + printer->pageRect(unit).width()/2,
                      printer->paperRect(unit).y() + printer->pageRect(unit).height()/2);
    painter.scale(scale, scale);
    painter.translate(-p->getSize().width()/2, -p->getSize().height()/2);

    p->paintEvent(painter, true);
    painter.end();
}


void MainWindow::on_actionAbout_triggered()
{
    this->window_about->show();
}

void MainWindow::on_actionEye_Dropper_triggered()
{
    if(this->context == NULL) return;
    this->context->selectToolFromList(TOOL_EYEDROPPER);
    this->highlightToolbar(this->ui->actionEye_Dropper);
}

void MainWindow::on_actionPen_triggered()
{
    if(this->context == NULL) return;
    this->context->selectToolFromList(TOOL_PENCIL);
    this->highlightToolbar(this->ui->actionPen);
}


void MainWindow::on_actionBrush_triggered()
{
    if(this->context == NULL) return;
    this->context->selectToolFromList(TOOL_BRUSH);
    this->highlightToolbar(this->ui->actionBrush);
}


void MainWindow::on_actionFill_triggered()
{
    if(this->context == NULL) return;
    this->context->selectToolFromList(TOOL_FILLCOLOR);
    this->highlightToolbar(this->ui->actionFill);
}


void MainWindow::on_actionText_triggered()
{
    if(this->context == NULL) return;
    this->context->selectToolFromList(TOOL_TEXT);
    this->highlightToolbar(this->ui->actionText);
}


void MainWindow::on_actionEraser_triggered()
{
    if(this->context == NULL) return;
    this->context->selectToolFromList(TOOL_ERASER);
    this->highlightToolbar(this->ui->actionEraser);
}


void MainWindow::on_actionAdd_layer_triggered()
{
    if(this->context == NULL) return;
    LayerManager *lm = this->context->getLayerManager();
    if(lm) {
        lm->addLayer();
    }
}


void MainWindow::on_actionRemove_layer_triggered()
{
    if(this->context == NULL) return;
    LayerManager *lm = this->context->getLayerManager();
    if(lm) {
        lm->removeLayer();
    }
}


void MainWindow::highlightToolbar(QAction *action)
{
    QWidget *w;
    for(QAction *a :this->ui->toolBar->actions()) {
        if(a != NULL) {
            w = this->ui->toolBar->widgetForAction(a);
            if(w != NULL && w != this->colorPicker) {
                w->setStyleSheet("");
            }
        }
    }
    w = this->ui->toolBar->widgetForAction(action);
    if(w != NULL) {
        w->setStyleSheet("background: rgba(183, 134, 32, 20%); border: 1px solid #b78620;");
    }
}

void MainWindow::on_actionZoom_in_triggered()
{
    if(this->context == NULL) return;
    if(this->context->getWorkspace() == NULL) return;
    this->context->getWorkspace()->zoomIN();
    this->context->getWorkspace()->repaint();
}


void MainWindow::on_actionZoom_out_triggered()
{
    if(this->context == NULL) return;
    if(this->context->getWorkspace() == NULL) return;
    this->context->getWorkspace()->zoomOUT();
    this->context->getWorkspace()->repaint();
}



void MainWindow::on_actionSwitch_Image_Mask_triggered()
{
    if(this->context == NULL) return;

    // zmeni mod (PROJECT_EDIT <-> MASK_EDIT)
    if(this->context->getLayerManager() != NULL &&
            this->context->getProject() != NULL) {
        if(this->context->getProject()->getMode() == PROJECT_EDIT) {
            this->context->getLayerManager()->selectMode(MASK_EDIT);
        } else {
            this->context->getLayerManager()->selectMode(PROJECT_EDIT);
        }
    }
}

void MainWindow::workspace_setDefaultScale()
{
    if(this->context == NULL) return;
    Workspace *w = this->context->getWorkspace();
    if(w == NULL) return;
    w->setDefaultScale();
}