tspsom.cpp

#include "tspsom.h"
#include "ui_tspsom.h"
#include <QVector>
#include <QVariant>
#include <QMessageBox>
#include <QtConcurrent/QtConcurrent>

TspSom::TspSom(QWidget *parent)
    : QMainWindow(parent)
    , ui(new Ui::TspSom)
    , citiesMap()
    , som()
{
    ui->setupUi(this);

    //Setting up the plotting area
    ui->graph_qcustomplot->xAxis->setRange(0,1);
    ui->graph_qcustomplot->yAxis->setRange(0,1);



    //Citites graph configuration
    ui->graph_qcustomplot->addGraph();
    ui->graph_qcustomplot->graph(0)->setLineStyle(QCPGraph::lsNone); //Dots
    ui->graph_qcustomplot->graph(0)->setPen(QPen(Qt::black)); //Color black
    ui->graph_qcustomplot->graph(0)->setScatterStyle(QCPScatterStyle(QCPScatterStyle::ssStar, 10)); //Stars

    //SOM graph configuration (a graph with loops needs to be defined as QCPCurve)
    //Qcustomplot takes ownership of the new curve so no direct managing is required.
    somCurve = new QCPCurve(ui->graph_qcustomplot->xAxis, ui->graph_qcustomplot->yAxis);
    somCurve->setPen(QPen(Qt::blue)); //color blue
    somCurve->setScatterStyle(QCPScatterStyle(QCPScatterStyle::ssDisc, 3)); //Filled circles


    //Populate combobox
    ui->initialization_comboBox->addItem("Circular", QVariant(true));
    ui->initialization_comboBox->addItem("Random", QVariant(false));

    //Connect qt signals and slots (more connections are formed in the uic)
    QObject::connect(&som, SIGNAL(trainingIterationDone()), this, SLOT(updatePlot()));
    QObject::connect(&som, SIGNAL(trainingDone(bool)), this, SLOT(enableButtons(bool)));

}

TspSom::~TspSom()
{
    delete ui;
}


/**
 * @brief TspSom::generateMapClicked Qt Slot that handless the "generate map" -button click.
 * Generates the appropriate amount of cites.
 */
void TspSom::generateMapClicked()
{
    int numberOfCities = ui->cities_spinBox->value();
    citiesMap.InitializeRandom(numberOfCities);

    QVector<double> x = citiesMap.getXaxis();
    QVector<double> y = citiesMap.getYaxis();

    ui->graph_qcustomplot->graph(0)->setData(x, y);
    ui->graph_qcustomplot->replot();
}

/**
 * @brief TspSom::initializeSOM_clicked Qt slot that handles the "Initialize SOM" -button click.
 * Initializes the SOM according to parmeters in form.
 */
void TspSom::initializeSOM_clicked()
{
    bool initMode = ui->initialization_comboBox->currentData().toBool();
    int nodes = ui->nodes_spinBox->value();

    if (!initMode) {
        som.InitializeRandom(nodes);
    }else {
        som.InitializeCircular(nodes);
    }

    updatePlot();
}


/**
 * @brief TspSom::trainClicked Qt slot that handless the "Train SOM" button click.
 */
void TspSom::trainClicked(){
    //Show errormessage if citesMap or som isn't initialized.
    if (citiesMap.getPoints().isEmpty() || som.getPoints().isEmpty()){
        QMessageBox errorBox;
        errorBox.setText("Map has to be generated and SOM need to be initialized before training!");
        errorBox.setIcon(QMessageBox::Critical);
        errorBox.exec();
        return;
    }
    int iterations = ui->iterations_spinBox->value();

    //Disable Ui buttons, while training is in progress
    enableButtons(false);

    //Initiates the training process in another thread. The future object is not used here.
    QFuture<void> future = QtConcurrent::run(&som, &SOM::train, citiesMap, iterations);
}

/**
 * @brief updatePlot is a slot that updates the plot in UI
 */
void TspSom::updatePlot(){
    QVector<double> x = som.getXaxis();
    QVector<double> y = som.getYaxis();

    somCurve->setData(x, y);
    ui->graph_qcustomplot->replot();
}

/**
 * @brief TspSom::enableButtons slot that sets the enabled status of buttons.
 * @param b
 */
void TspSom::enableButtons(bool b){
    ui->generateMap_pushButton->setEnabled(b);
    ui->initializeSOM_pushButton->setEnabled(b);
    ui->trainSOM_pushButton->setEnabled(b);
}