#include "mainwindow.h"

#include "ui_mainwindow.h"

#include <QSerialPortInfo>

#include <QtDebug>

#include <qwt_plot.h>

#include <limits.h>

#include "utils.h"

MainWindow::MainWindow(QWidget *parent) :

    QMainWindow(parent),

    ui(new Ui::MainWindow)

{

    ui->setupUi(this);

    // init UI signals

    QObject::connect(ui->pbReloadPorts, &QPushButton::clicked,

                     this, &MainWindow::loadPortList);

    QObject::connect(ui->pbOpenPort, &QPushButton::clicked,

                     this, &MainWindow::togglePort);

    QObject::connect(this, &MainWindow::portToggled,

                     this, &MainWindow::onPortToggled);

    QObject::connect(ui->cbPortList,

                     SELECT<const QString&>::OVERLOAD_OF(&QComboBox::activated),

                     this, &MainWindow::selectPort);

    QObject::connect(ui->cbBaudRate,

                     SELECT<const QString&>::OVERLOAD_OF(&QComboBox::activated),

                     this, &MainWindow::selectBaudRate);

    QObject::connect(ui->spNumOfSamples, SELECT<int>::OVERLOAD_OF(&QSpinBox::valueChanged),

                     this, &MainWindow::onNumOfSamplesChanged);

    QObject::connect(ui->cbAutoScale, &QCheckBox::toggled,

                     this, &MainWindow::onAutoScaleChecked);

    QObject::connect(ui->spYmin, SIGNAL(valueChanged(double)),

                     this, SLOT(onYScaleChanged()));

    QObject::connect(ui->spYmax, SIGNAL(valueChanged(double)),

                     this, SLOT(onYScaleChanged()));

    // init port signals

    QObject::connect(&(this->serialPort), SIGNAL(error(QSerialPort::SerialPortError)),

                     this, SLOT(onPortError(QSerialPort::SerialPortError)));

    loadPortList();

    loadBaudRateList();

    ui->cbBaudRate->setCurrentIndex(ui->cbBaudRate->findText("9600"));

    // set limits for axis limit boxes

    ui->spYmin->setRange(std::numeric_limits<double>::min(),

                         std::numeric_limits<double>::max());

    ui->spYmax->setRange(std::numeric_limits<double>::min(),

                         std::numeric_limits<double>::max());

    // init plot

    numOfSamples = ui->spNumOfSamples->value();

    dataArray.resize(numOfSamples);

    dataX.resize(numOfSamples);

    for (int i = 0; i < dataX.size(); i++)

    {

        dataX[i] = i;

    }

    curve.setSamples(dataX, dataArray);

    curve.attach(ui->plot);

}

MainWindow::~MainWindow()

{

    if (serialPort.isOpen())

    {

        serialPort.close();

    }

    delete ui;

}

void MainWindow::loadPortList()

{

    QString currentSelection = ui->cbPortList->currentText();

    ui->cbPortList->clear();

    for (auto port : QSerialPortInfo::availablePorts())

    {

        ui->cbPortList->addItem(port.portName());

    }

    // find current selection in the new list, maybe it doesn't exist anymore?

    int currentSelectionIndex = ui->cbPortList->findText(currentSelection);

    if (currentSelectionIndex >= 0)

    {

        ui->cbPortList->setCurrentIndex(currentSelectionIndex);

    }

    else // our port doesn't exist anymore, close port if it's open

    {

        if (serialPort.isOpen()) togglePort();

    }

}

void MainWindow::loadBaudRateList()

{

    ui->cbBaudRate->clear();

    for (auto baudRate : QSerialPortInfo::standardBaudRates())

    {

        ui->cbBaudRate->addItem(QString::number(baudRate));

    }

}

void MainWindow::togglePort()

{

    if (serialPort.isOpen())

    {

        serialPort.close();

        qDebug() << "Port closed, " << serialPort.portName();

        emit portToggled(false);

    }

    else

    {

        serialPort.setPortName(ui->cbPortList->currentText());

        // open port

        if (serialPort.open(QIODevice::ReadWrite))

        {

            qDebug() << "Port opened, " << serialPort.portName();

            emit portToggled(true);

            // set baud rate

            if (!serialPort.setBaudRate(ui->cbBaudRate->currentText().toInt()))

            {

                qDebug() << "Set baud rate failed during port opening: "

                         << serialPort.error();

            }

        }

        else

        {

            qDebug() << "Port open error: " << serialPort.error();

        }

    }

}