/*

  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 "ui_mainwindow.h"

#include <QByteArray>

#include <QApplication>

#include <QFileDialog>

#include <QFile>

#include <QTextStream>

#include <QtDebug>

#include <QtEndian>

#include <qwt_plot.h>

#include <limits.h>

#include <cmath>

#include <iostream>

#include "utils.h"

#include "version.h"

#include "floatswap.h"

#if defined(Q_OS_WIN) && defined(QT_STATIC)

#include <QtPlugin>

Q_IMPORT_PLUGIN(QWindowsIntegrationPlugin)

#endif

MainWindow::MainWindow(QWidget *parent) :

    QMainWindow(parent),

    ui(new Ui::MainWindow),

    portControl(&serialPort)

{

    ui->setupUi(this);

    ui->tabWidget->insertTab(0, &portControl, "Port");

    ui->tabWidget->setCurrentIndex(0);

    addToolBar(portControl.toolBar());

    setupAboutDialog();

    // init UI signals

    // menu signals

    QObject::connect(ui->actionHelpAbout, &QAction::triggered,

              &aboutDialog, &QWidget::show);

    QObject::connect(ui->actionExportCsv, &QAction::triggered,

                     this, &MainWindow::onExportCsv);

    QObject::connect(ui->actionQuit, &QAction::triggered,

                     this, &MainWindow::close);

    QObject::connect(ui->actionGrid, &QAction::toggled, [this](bool show)

                     {

                         ui->plot->showGrid(show);

                         ui->actionMinorGrid->setEnabled(show);

                     });

    ui->actionMinorGrid->setEnabled(ui->actionGrid->isChecked());

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

                     ui->plot, &Plot::showMinorGrid);

    QObject::connect(ui->actionUnzoom, &QAction::triggered,

                     ui->plot, &Plot::unzoom);

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

                     ui->plot, &Plot::darkBackground);

    ui->plot->darkBackground(ui->actionDarkBackground->isChecked());

    // port control signals

    QObject::connect(&portControl, &PortControl::portToggled,

                     this, &MainWindow::onPortToggled);

    QObject::connect(&portControl, &PortControl::skipByteRequested,

                     this, &MainWindow::skipByte);

    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()));

    QObject::connect(ui->actionClear, SIGNAL(triggered(bool)),

                     this, SLOT(clearPlot()));

    // setup number of channels spinbox

    QObject::connect(ui->spNumOfChannels,

                     SELECT<int>::OVERLOAD_OF(&QSpinBox::valueChanged),

                     this, &MainWindow::onNumOfChannelsChanged);

    // setup number format buttons

    numberFormatButtons.addButton(ui->rbUint8,  NumberFormat_uint8);

    numberFormatButtons.addButton(ui->rbUint16, NumberFormat_uint16);

    numberFormatButtons.addButton(ui->rbUint32, NumberFormat_uint32);

    numberFormatButtons.addButton(ui->rbInt8,   NumberFormat_int8);

    numberFormatButtons.addButton(ui->rbInt16,  NumberFormat_int16);

    numberFormatButtons.addButton(ui->rbInt32,  NumberFormat_int32);

    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 = ui->spNumOfChannels->value();

    // init channel data and curve list

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

    {

        channelBuffers.append(new FrameBuffer(numOfSamples));

        curves.append(new QwtPlotCurve());

        curves[i]->setSamples(channelBuffers[i]);

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

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

    }

    // init auto scale

    ui->plot->setAxis(ui->cbAutoScale->isChecked(),

                      ui->spYmin->value(), ui->spYmax->value());

    // init grid

    ui->plot->showGrid(ui->actionGrid->isChecked());

    ui->plot->showMinorGrid(ui->actionMinorGrid->isChecked());

    // 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)

    {

        // also deletes respective FrameBuffer

        delete curve;

    }

    if (serialPort.isOpen())

    {

        serialPort.close();

    }

    delete ui;

    ui = NULL; // we check if ui is deleted in messageHandler

}

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())

    {

        serialPort.clear(QSerialPort::Input);

        return;

    }

    // 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--;

    }

        case QSerialPort::UnknownError:

            qCritical() << "Unknown error!";

            break;

        default:

            qCritical() << "Unhandled port error: " << error;

            break;

    }

}

void MainWindow::skipByte()

{

    skipByteRequested = true;

}

void MainWindow::addChannelData(unsigned int channel, double* data, unsigned size)

{

    channelBuffers[channel]->addSamples(data, size);

    sampleCount += size;

}

void MainWindow::clearPlot()

{

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

    {

        channelBuffers[ci]->clear();

    }

    ui->plot->replot();

}

void MainWindow::onNumOfSamplesChanged(int value)

{

    numOfSamples = value;

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

    {

        channelBuffers[ci]->resize(numOfSamples);

    }

    ui->plot->replot();

}

void MainWindow::onNumOfChannelsChanged(int value)

{

    unsigned int oldNum = this->numOfChannels;

    this->numOfChannels = value;

    if (numOfChannels > oldNum)

    {

        // add new channels

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

        {

            channelBuffers.append(new FrameBuffer(numOfSamples));

            curves.append(new QwtPlotCurve());

            curves.last()->setSamples(channelBuffers.last());

            curves.last()->setPen(makeColor(curves.length()-1));

            curves.last()->attach(ui->plot);

        }

    }

    else if(numOfChannels < oldNum)

    {

        // remove channels

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

        {

            // also deletes owned FrameBuffer

            delete curves.takeLast();

            channelBuffers.removeLast();

        }

    }

}

void MainWindow::onAutoScaleChecked(bool checked)

{

    if (checked)

    {

        ui->plot->setAxis(true);

        ui->lYmin->setEnabled(false);

        ui->lYmax->setEnabled(false);

        ui->spYmin->setEnabled(false);

        ui->spYmax->setEnabled(false);

    }

    else

    {

        ui->lYmin->setEnabled(true);

        ui->lYmax->setEnabled(true);

        ui->spYmin->setEnabled(true);

        ui->spYmax->setEnabled(true);

        ui->plot->setAxis(false,  ui->spYmin->value(), ui->spYmax->value());

    }

}

void MainWindow::onYScaleChanged()

{

    ui->plot->setAxis(false,  ui->spYmin->value(), ui->spYmax->value());

}

void MainWindow::onNumberFormatButtonToggled(int numberFormatId, bool checked)

{

    if (checked) selectNumberFormat((NumberFormat) numberFormatId);

}

void MainWindow::selectNumberFormat(NumberFormat numberFormatId)

{

    numberFormat = numberFormatId;

    switch(numberFormat)

    {

        case NumberFormat_uint8:

            sampleSize = 1;

            readSample = &MainWindow::readSampleAs<quint8>;

            break;

        case NumberFormat_int8:

            sampleSize = 1;

            readSample = &MainWindow::readSampleAs<qint8>;

            break;

        case NumberFormat_uint16:

            sampleSize = 2;

            readSample = &MainWindow::readSampleAs<quint16>;

            break;

        case NumberFormat_int16:

            sampleSize = 2;

            readSample = &MainWindow::readSampleAs<qint16>;

            break;

        case NumberFormat_uint32:

            sampleSize = 4;

            readSample = &MainWindow::readSampleAs<quint32>;

            break;

        case NumberFormat_int32:

            sampleSize = 4;

            readSample = &MainWindow::readSampleAs<qint32>;

            break;

        case NumberFormat_float:

            sampleSize = 4;

            readSample = &MainWindow::readSampleAs<float>;

            break;

        case NumberFormat_ASCII:

            sampleSize = 0;    // these two members should not be used

            readSample = NULL; // in this mode

            break;

    }

    if (numberFormat == NumberFormat_ASCII)

    {

        QObject::disconnect(&(this->serialPort), &QSerialPort::readyRead, 0, 0);

        QObject::connect(&(this->serialPort), &QSerialPort::readyRead,

                         this, &MainWindow::onDataReadyASCII);

        portControl.enableSkipByte();

    }

    else

    {

        QObject::disconnect(&(this->serialPort), &QSerialPort::readyRead, 0, 0);

        QObject::connect(&(this->serialPort), &QSerialPort::readyRead,

                         this, &MainWindow::onDataReady);

        portControl.enableSkipByte(false);

    }

}

template<typename T> double MainWindow::readSampleAs()

{

    T data;

    this->serialPort.read((char*) &data, sizeof(data));

    if (ui->rbLittleE->isChecked())

    {

        data = qFromLittleEndian(data);

    }

    else

    {

        data = qFromBigEndian(data);

    }

    return double(data);

}

bool MainWindow::isDemoRunning()

{

    return ui->actionDemoMode->isChecked();

}

void MainWindow::spsTimerTimeout()

{

    spsLabel.setText(QString::number(sampleCount/SPS_UPDATE_TIMEOUT) + "sps");

    sampleCount = 0;

}

void MainWindow::demoTimerTimeout()

{

    const double period = 100;

    demoCount++;

    if (demoCount > 100) demoCount = 0;

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

#include <QString>

#include <QVector>

#include <QList>

#include <QSerialPort>

#include <QSignalMapper>

#include <QTimer>

#include <QColor>

#include <QtGlobal>

#include <qwt_plot_curve.h>

#include <qwt_plot_textlabel.h>

#include "portcontrol.h"

#include "ui_about_dialog.h"

#include "framebuffer.h"

namespace Ui {

class MainWindow;

}

class MainWindow : public QMainWindow

{

    Q_OBJECT

public:

    explicit MainWindow(QWidget *parent = 0);

    ~MainWindow();

    void messageHandler(QtMsgType type, const QMessageLogContext &context,

                        const QString &msg);

private:

    enum NumberFormat

    {

        NumberFormat_uint8,

        NumberFormat_uint16,

        NumberFormat_uint32,

        NumberFormat_int8,

        NumberFormat_int16,

        NumberFormat_int32,

        NumberFormat_float,

        NumberFormat_ASCII

    };

    Ui::MainWindow *ui;

    QButtonGroup numberFormatButtons;

    QDialog aboutDialog;

    void setupAboutDialog();

    QSerialPort serialPort;

    PortControl portControl;

    unsigned int numOfSamples;

    unsigned int numOfChannels;

    QList<QwtPlotCurve*> curves;

    typedef QVector<double> DataArray;

    // Note: FrameBuffer s are owned by their respective QwtPlotCurve s.

    QList<FrameBuffer*> channelBuffers;

    // `data` contains i th channels data

    void addChannelData(unsigned int channel, double* data, unsigned size);

    NumberFormat numberFormat;

    unsigned int sampleSize; // number of bytes in the selected number format

    double (MainWindow::*readSample)();

    // note that serialPort should already have enough bytes present

    template<typename T> double readSampleAs();

    bool skipByteRequested;

    const int SPS_UPDATE_TIMEOUT = 1;  // second

    QLabel spsLabel;

    unsigned int sampleCount;

    QTimer spsTimer;

    // demo

    QTimer demoTimer;

    int demoCount;

    bool isDemoRunning();

    QwtPlotTextLabel demoIndicator;

    QColor makeColor(unsigned int channelIndex);

private slots:

    void onPortToggled(bool open);

    void onDataReady();      // used with binary number formats

    void onDataReadyASCII(); // used with ASCII number format

    void onPortError(QSerialPort::SerialPortError error);

    void skipByte();

    void onNumOfSamplesChanged(int value);

    void onAutoScaleChecked(bool checked);

    void onYScaleChanged();

    void onNumOfChannelsChanged(int value);

    void onNumberFormatButtonToggled(int numberFormatId, bool checked);

    void selectNumberFormat(NumberFormat numberFormatId);

    void clearPlot();

    void spsTimerTimeout();

    void demoTimerTimeout();

    void enableDemo(bool enabled);

    void onExportCsv();

};

#endif // MAINWINDOW_H

              <verstretch>0</verstretch>

             </sizepolicy>

            </property>

            <property name="title">

             <string>Byte Order:</string>

            </property>

            <layout class="QVBoxLayout" name="verticalLayout">

             <item>

              <widget class="QRadioButton" name="rbLittleE">

               <property name="toolTip">

                <string>least significant byte first</string>

               </property>

               <property name="text">

                <string>Little Endian</string>

               </property>

               <property name="checked">

                <bool>true</bool>

               </property>

              </widget>

             </item>

             <item>

              <widget class="QRadioButton" name="rbBigE">

               <property name="toolTip">

                <string>most significant byte first</string>

               </property>

               <property name="text">

                <string>Big Endian</string>

               </property>

              </widget>

             </item>

            </layout>

           </widget>

          </item>

          <item>

           <spacer name="verticalSpacer">

            <property name="orientation">

             <enum>Qt::Vertical</enum>

            </property>

            <property name="sizeHint" stdset="0">

             <size>

              <width>20</width>

              <height>1</height>

             </size>

            </property>

           </spacer>

          </item>

         </layout>

        </item>

        <item>

         <spacer name="horizontalSpacer_3">

          <property name="orientation">

           <enum>Qt::Horizontal</enum>

          </property>

          <property name="sizeType">

           <enum>QSizePolicy::MinimumExpanding</enum>

          </property>

          <property name="sizeHint" stdset="0">

           <size>

            <width>1</width>

            <height>20</height>

           </size>

          </property>

         </spacer>

        </item>

       </layout>

      </widget>

      <widget class="QWidget" name="tabPlot">

       <attribute name="title">

        <string>Plot</string>

       </attribute>

       <layout class="QHBoxLayout" name="horizontalLayout_4">

        <item>

         <spacer name="horizontalSpacer_2">

          <property name="orientation">

           <enum>Qt::Horizontal</enum>

          </property>

          <property name="sizeHint" stdset="0">

           <size>

            <width>370</width>

            <height>20</height>

           </size>

          </property>

         </spacer>

        </item>

        <item>

         <widget class="Line" name="line">

          <property name="orientation">

           <enum>Qt::Vertical</enum>

          </property>

         </widget>

        </item>

        <item>

         <layout class="QFormLayout" name="formLayout_2">

          <property name="fieldGrowthPolicy">

           <enum>QFormLayout::AllNonFixedFieldsGrow</enum>

          </property>

            </property>

           </widget>

          </item>

          <item row="0" column="1">

           <widget class="QSpinBox" name="spNumOfSamples">

            <property name="toolTip">

             <string>length of X axis</string>

            </property>

            <property name="keyboardTracking">

             <bool>false</bool>

            </property>

            <property name="minimum">

             <number>2</number>

            </property>

            <property name="maximum">

             <number>10000</number>

            </property>

            <property name="value">

             <number>1000</number>

            </property>

           </widget>

          </item>

          <item row="1" column="0">

           <widget class="QCheckBox" name="cbAutoScale">

            <property name="text">

             <string>Auto Scale Y Axis</string>

            </property>

            <property name="checked">

             <bool>true</bool>

            </property>

           </widget>

          </item>

          <item row="2" column="0">

           <widget class="QLabel" name="lYmax">

            <property name="enabled">

             <bool>false</bool>

            </property>

            <property name="text">

             <string>Ymax</string>

            </property>

           </widget>

          </item>

          <item row="2" column="1">

           <widget class="QDoubleSpinBox" name="spYmax">

            <property name="enabled">

             <bool>false</bool>

            </property>

            <property name="maximumSize">

             <size>

              <width>75</width>

              <height>16777215</height>

             </size>

            </property>

            <property name="toolTip">

             <string>upper limit of Y axis</string>

            </property>

            <property name="maximum">

             <double>1000.000000000000000</double>

            </property>

            <property name="value">

             <double>1000.000000000000000</double>

            </property>

           </widget>

          </item>

         </layout>

        </item>

       </layout>

      </widget>

      <widget class="QWidget" name="tabLog">

       <attribute name="title">

        <string>Log</string>

       </attribute>

       <attribute name="toolTip">

        <string>Error and Warning Messages</string>

       </attribute>

       <layout class="QHBoxLayout" name="horizontalLayout">

        <property name="leftMargin">

         <number>4</number>

        </property>

        <property name="topMargin">

         <number>4</number>

        </property>

        <property name="rightMargin">

         <number>4</number>

        </property>

        <property name="bottomMargin">

         <number>4</number>

        </property>

        <item>

         <widget class="QPlainTextEdit" name="ptLog">

          <property name="readOnly">

           <bool>true</bool>

          </property>

         </widget>

        </item>

       </layout>