/*

  Copyright © 2015 Hasan Yavuz Özderya

  This file is part of serialplot.

  serialplot is free software: you can redistribute it and/or modify

  it under the terms of the GNU General Public License as published by

  the Free Software Foundation, either version 3 of the License, or

  (at your option) any later version.

  serialplot is distributed in the hope that it will be useful,

  but WITHOUT ANY WARRANTY; without even the implied warranty of

  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License

  along with serialplot.  If not, see <http://www.gnu.org/licenses/>.

*/

#include "mainwindow.h"

#include <QApplication>

#include <QtGlobal>

#include "version.h"

MainWindow* pMainWindow;

void messageHandler(QtMsgType type, const QMessageLogContext &context,

                    const QString &msg)

{

    pMainWindow->messageHandler(type, context, msg);

}

int main(int argc, char *argv[])

{

    QApplication a(argc, argv);

    MainWindow w;

    pMainWindow = &w;

    qInstallMessageHandler(messageHandler);

    // log application information

    qDebug() << "SerialPlot" << VERSION_STRING;

    qDebug() << "Revision" << VERSION_REVISION;

    w.show();

    return a.exec();

}

    numberFormatButtons.addButton(ui->rbFloat,  NumberFormat_float);

    numberFormatButtons.addButton(ui->rbASCII,  NumberFormat_ASCII);

    QObject::connect(&numberFormatButtons, SIGNAL(buttonToggled(int, bool)),

                     this, SLOT(onNumberFormatButtonToggled(int, bool)));

    // init port signals

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

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

    // set limits for axis limit boxes

    ui->spYmin->setRange((-1) * std::numeric_limits<double>::max(),

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

    ui->spYmax->setRange((-1) * std::numeric_limits<double>::max(),

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

    // init data arrays and plot

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

    numOfChannels = 1;

    dataX.resize(numOfSamples);

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

    {

        dataX[i] = i;

    }

    // init channel data and curve list

    for (unsigned int i = 0; i < numOfChannels; i++)

    {

        channelsData.append(DataArray(numOfSamples, 0.0));

        curves.append(new QwtPlotCurve());

        curves[i]->setSamples(dataX, channelsData[i]);

        curves[i]->setPen(makeColor(i));

        curves[i]->attach(ui->plot);

    }

    // init zoomer

    zoomer = new Zoomer(ui->plot->canvas(), false);

    zoomer->setZoomBase();

    QObject::connect(zoomer, &Zoomer::unzoomed,

                     this, &MainWindow::unzoomed);

    // init number format

    if (numberFormatButtons.checkedId() >= 0)

    {

        selectNumberFormat((NumberFormat) numberFormatButtons.checkedId());

    }

    else

    {

        selectNumberFormat(NumberFormat_uint8);

    }

    // Init sps (sample per second) counter

    sampleCount = 0;

    spsLabel.setText("0sps");

    spsLabel.setToolTip("samples per second (total of all channels)");

    ui->statusBar->addPermanentWidget(&spsLabel);

    spsTimer.start(SPS_UPDATE_TIMEOUT * 1000);

    QObject::connect(&spsTimer, &QTimer::timeout,

                     this, &MainWindow::spsTimerTimeout);

    // Init demo mode

    demoCount = 0;

    demoTimer.setInterval(100);

    QObject::connect(&demoTimer, &QTimer::timeout,

                     this, &MainWindow::demoTimerTimeout);

    QObject::connect(ui->actionDemoMode, &QAction::toggled,

                     this, &MainWindow::enableDemo);

    {   // init demo indicator

        QwtText demoText(" DEMO RUNNING ");  // looks better with spaces

        demoText.setColor(QColor("white"));

        demoText.setBackgroundBrush(Qt::darkRed);

        demoText.setBorderRadius(4);

        demoText.setRenderFlags(Qt::AlignLeft | Qt::AlignTop);

        demoIndicator.setText(demoText);

        demoIndicator.hide();

        demoIndicator.attach(ui->plot);

    }

}

MainWindow::~MainWindow()

{

    for (auto curve : curves)

    {

        delete curve;

    }

    if (serialPort.isOpen())

    {

        serialPort.close();

    }

    delete ui;

}

void MainWindow::setupAboutDialog()

{

    Ui_AboutDialog uiAboutDialog;

    uiAboutDialog.setupUi(&aboutDialog);

    QObject::connect(uiAboutDialog.pbAboutQt, &QPushButton::clicked,

                     [](){ QApplication::aboutQt();});

    QString aboutText = uiAboutDialog.lbAbout->text();

    aboutText.replace("$VERSION_STRING$", VERSION_STRING);

    aboutText.replace("$VERSION_REVISION$", VERSION_REVISION);

    uiAboutDialog.lbAbout->setText(aboutText);

}

void MainWindow::onPortToggled(bool open)

{

    // make sure demo mode is disabled

    if (open && isDemoRunning()) enableDemo(false);

    ui->actionDemoMode->setEnabled(!open);

}

void MainWindow::onDataReady()

{

    if (!ui->actionPause->isChecked())

    {

        // a package is a set of channel data like {CHAN0_SAMPLE, CHAN1_SAMPLE...}

        int packageSize = sampleSize * numOfChannels;

        int bytesAvailable = serialPort.bytesAvailable();

        if (bytesAvailable > 0 && skipByteRequested)

        {

            serialPort.read(1);

            skipByteRequested = false;

            bytesAvailable--;

        }

        if (bytesAvailable < packageSize)

        {

            return;

        }

        else

        {

            int numOfPackagesToRead =

                (bytesAvailable - (bytesAvailable % packageSize)) / packageSize;

            QVector<DataArray> channelSamples(numOfChannels);

            for (unsigned int ci = 0; ci < numOfChannels; ci++)

            {

                channelSamples[ci].resize(numOfPackagesToRead);

            }

            int i = 0;

            while(i < numOfPackagesToRead)

            {

                for (unsigned int ci = 0; ci < numOfChannels; ci++)

                {

                    channelSamples[ci].replace(i, (this->*readSample)());

                }

                i++;

            }

            for (unsigned int ci = 0; ci < numOfChannels; ci++)

            {

                addChannelData(ci, channelSamples[ci]);

            }

        }

    }

    else

    {

        serialPort.clear(QSerialPort::Input);

    }

}

void MainWindow::onDataReadyASCII()

{

    while(serialPort.canReadLine())

    {

        QByteArray line = serialPort.readLine();

        // discard data if paused

        if (ui->actionPause->isChecked())

        {

            return;

        }

        line = line.trimmed();

        auto separatedValues = line.split(',');

        if (separatedValues.length() >= int(numOfChannels))

        {

            for (unsigned int ci = 0; ci < numOfChannels; ci++)

            {

                double channelSample = separatedValues[ci].toDouble();

                addChannelData(ci, DataArray({channelSample}));

            }

/*

  Below crude drawing demostrates how color selection occurs for

  given channel index

  0°                     <--Hue Value-->                           360°

  |* . o . + . o . * . o . + . o . * . o . + . o . * . o . + . o . |

  * -> 0-3

  + -> 4-7

  o -> 8-15

  . -> 16-31

 */

QColor MainWindow::makeColor(unsigned int channelIndex)

{

    auto i = channelIndex;

    if (i < 4)

    {

        return QColor::fromHsv(360*i/4, 255, 230);

    }

    else

    {

        double p = floor(log2(i));

        double n = pow(2, p);

        i = i - n;

        return QColor::fromHsv(360*i/n + 360/pow(2,p+1), 255, 230);

    }

}

void MainWindow::onExportCsv()

{

    bool wasPaused = ui->actionPause->isChecked();

    ui->actionPause->setChecked(true); // pause plotting

    QString fileName = QFileDialog::getSaveFileName(this, tr("Export CSV File"));

    if (fileName.isNull())  // user canceled export

    {

        ui->actionPause->setChecked(wasPaused);

    }

    else

    {

        QFile file(fileName);

        if (file.open(QIODevice::WriteOnly | QIODevice::Text))

        {

            QTextStream fileStream(&file);

            for (unsigned int ci = 0; ci < numOfChannels; ci++)

            {

                fileStream << "Channel " << ci;

                if (ci != numOfChannels-1) fileStream << ",";

            }

            fileStream << '\n';

            for (unsigned int i = 0; i < numOfSamples; i++)

            {

                for (unsigned int ci = 0; ci < numOfChannels; ci++)

                {

                    fileStream << channelsData[ci][i];

                    if (ci != numOfChannels-1) fileStream << ",";

                }

                fileStream << '\n';

            }

        }

        else

        {

            qCritical() << "File open error during export: " << file.error();

        }

    }

}

void MainWindow::messageHandler(QtMsgType type,

                                const QMessageLogContext &context,

                                const QString &msg)

{

    QString logString;

    switch (type)

    {

        case QtDebugMsg:

            logString = "[Debug] " + msg;

            break;

        case QtWarningMsg:

            logString = "[Warning] " + msg;

            break;

        case QtCriticalMsg:

            logString = "[Error] " + msg;

            break;

        case QtFatalMsg:

            logString = "[Fatal] " + msg;

            break;

    }

    std::cerr << logString.toStdString() << std::endl;

    {

        ui->statusBar->showMessage(msg, 5000);

    }

}

void MainWindow::unzoomed()

{

    if (ui->cbAutoScale->isChecked())

    {

        ui->plot->setAxisAutoScale(QwtPlot::yLeft);

    }

    else

    {

        ui->plot->setAxisScale(QwtPlot::yLeft, ui->spYmin->value(),

                               ui->spYmax->value());

    }

    ui->plot->setAxisAutoScale(QwtPlot::xBottom);

    ui->plot->replot();

}