LocalContrast/mainwindow.cpp at master · adrogolyub/LocalContrast · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#include "mainwindow.h"
#include "utils.h"
#include <QFileDialog>
#include <QStatusBar>
#include <QButtonGroup>
#include <QMessageBox>

using namespace std::chrono;
using namespace cv;

MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent)
{
    setupUi(this);
    modeGroup = new QButtonGroup(this);
    modeGroup->addButton(rbUniform, (int)(LocalContrastEnhancer::Uniform));
    modeGroup->addButton(rbAHE, (int)LocalContrastEnhancer::AHE);
    modeGroup->addButton(rbCLAHE, (int)LocalContrastEnhancer::CLAHE);
    modeGroup->addButton(rbUS, (int)LocalContrastEnhancer::UnsharpMask);
    _enhancer = new LocalContrastEnhancer(this);
    qRegisterMetaType<cv::Mat>("cv::Mat&");
    connect(_enhancer, &LocalContrastEnhancer::resultReady, this, &MainWindow::onResultReceived);
    connect(forceSlider, &QSlider::valueChanged, this, &MainWindow::updateImage);
    connect(modeGroup, SIGNAL(buttonClicked(int)), this, SLOT(updateImage()));
    setWindowState(Qt::WindowMaximized);
    forceSlider->hide();
}

void MainWindow::disableGUI(bool t)
{
    forceSlider->setEnabled(!t);
    if (!t)
        forceSlider->setFocus(); // because forceSlider loses focus after disabling
    rbAHE->setEnabled(!t);
    rbCLAHE->setEnabled(!t);
    rbUS->setEnabled(!t);
    rbUniform->setEnabled(!t);
    actionOpen->setEnabled(!t);
    actionSave->setEnabled(!t);
}

void MainWindow::on_actionOpen_triggered()
{
    QString fileName = QFileDialog::getOpenFileName(this, tr("Open image"), QString(), tr("Images (*.png *.bmp *.jpg)"));
    if (fileName.isEmpty())
        return;
    _image = imread(fileName.toLatin1().data(), IMREAD_COLOR);
    imageLeft->setPixmap(Utils::matToPixmap(_image));
    Mat image = _image.clone();
    cvtColor(image, image, COLOR_BGR2Lab);
    split(image, _splitLabImage);
    updateImage();
}

void MainWindow::updateImage()
{
    forceSlider->setVisible(!rbUniform->isChecked());
    disableGUI(); // to avoid synchronization issues
    _time = steady_clock::now();
    _enhancer->setMode((LocalContrastEnhancer::Mode)(modeGroup->checkedId()));
    _enhancer->doWork(_splitLabImage[0], forceSlider->value() / (float)forceSlider->maximum());
}

void MainWindow::onResultReceived(Mat &res)
{
    long long elapsed =  duration_cast<milliseconds>(steady_clock::now() - _time).count();
    QMainWindow::statusBar()->showMessage(tr("%1 msec").arg(elapsed));

    vector<Mat> ch;
    ch.push_back(res);
    ch.push_back(_splitLabImage[1]);
    ch.push_back(_splitLabImage[2]);
    merge(ch, res);
    cvtColor(res, res, COLOR_Lab2BGR);
    lastResult = Utils::matToPixmap(res);
    imageRight->setPixmap(lastResult);
    disableGUI(false);
}

void MainWindow::on_actionSave_triggered()
{
    QString path = QFileDialog::getSaveFileName(this, tr("Save image"), QString(), tr("PNG Image (*.png)"));
    if (!path.isEmpty()) {
        if (lastResult.save(path))
            QMessageBox::information(this, tr("Image saved"), tr("Image successfully saved."));
        else
            QMessageBox::critical(this, tr("Failure"), tr("Failed to save image."));
    }
}