#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);
}