84 lines
1.9 KiB
C++
84 lines
1.9 KiB
C++
#include "neuroui.h"
|
|
#include "ui_neuroui.h"
|
|
|
|
NeuroUI::NeuroUI(QWidget *parent) :
|
|
QMainWindow(parent),
|
|
ui(new Ui::NeuroUI)
|
|
{
|
|
ui->setupUi(this);
|
|
|
|
ui->logView->addItem("Ready.");
|
|
}
|
|
|
|
NeuroUI::~NeuroUI()
|
|
{
|
|
if (m_netLearner != nullptr)
|
|
{
|
|
m_netLearner->cancelLearning();
|
|
m_netLearner->wait();
|
|
}
|
|
|
|
delete ui;
|
|
}
|
|
|
|
void NeuroUI::on_runButton_clicked()
|
|
{
|
|
ui->logView->clear();
|
|
ui->errorPlotter->clear();
|
|
|
|
if (m_netLearner == nullptr)
|
|
{
|
|
m_netLearner.reset(new NetLearner);
|
|
|
|
connect(m_netLearner.get(), &NetLearner::logMessage, this, &NeuroUI::logMessage);
|
|
connect(m_netLearner.get(), &NetLearner::progress, this, &NeuroUI::progress);
|
|
|
|
connect(m_netLearner.get(), &NetLearner::started, this, &NeuroUI::netLearnerStarted);
|
|
connect(m_netLearner.get(), &NetLearner::finished, this, &NeuroUI::netLearnerFinished);
|
|
|
|
connect(m_netLearner.get(), &NetLearner::currentNetError, ui->errorPlotter, &ErrorPlotter::addErrorValue);
|
|
|
|
connect(m_netLearner.get(), &NetLearner::sampleImageLoaded, this, &NeuroUI::setImage);
|
|
}
|
|
|
|
m_netLearner->start();
|
|
}
|
|
|
|
void NeuroUI::logMessage(const QString &logMessage)
|
|
{
|
|
if (ui->logView->count() == static_cast<int>(m_logSize))
|
|
{
|
|
delete ui->logView->item(0);
|
|
}
|
|
|
|
ui->logView->addItem(logMessage);
|
|
ui->logView->scrollToBottom();
|
|
}
|
|
|
|
void NeuroUI::netLearnerStarted()
|
|
{
|
|
ui->runButton->setEnabled(false);
|
|
ui->progressBar->setValue(0);
|
|
}
|
|
|
|
void NeuroUI::netLearnerFinished()
|
|
{
|
|
ui->runButton->setEnabled(true);
|
|
ui->progressBar->setValue(0);
|
|
}
|
|
|
|
void NeuroUI::progress(double progress)
|
|
{
|
|
double range = ui->progressBar->maximum() - ui->progressBar->minimum();
|
|
int value = ui->progressBar->minimum() + (int)(progress * range);
|
|
|
|
ui->progressBar->setValue(value);
|
|
}
|
|
|
|
void NeuroUI::setImage(const QImage &image)
|
|
{
|
|
QPixmap pixmap;
|
|
pixmap.convertFromImage(image);
|
|
ui->label->setPixmap(pixmap);
|
|
}
|