/*

  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 <qwt_plot.h>

#include <limits.h>

#include <cmath>

#include <iostream>

#include "utils.h"

#include "version.h"

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

    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(&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 = 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 number format

    if (numberFormatButtons.checkedId() >= 0)

    {

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

    }

}

void MainWindow::onAutoScaleChecked(bool checked)

{

    if (checked)

    {

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

        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->setAxisScale(QwtPlot::yLeft, ui->spYmin->value(),

                               ui->spYmax->value());

    }

}

void MainWindow::onYScaleChanged()

{

    ui->plot->setAxisScale(QwtPlot::yLeft, 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));

    return double(data);

}

bool MainWindow::isDemoRunning()

{

    return ui->actionDemoMode->isChecked();

}

void MainWindow::demoTimerTimeout()

{

    demoCount++;

    if (demoCount > 100) demoCount = 0;

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

    {

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

        {

            DataArray data(1);

            data.replace(0, (ci + 1)*demoCount);

            addChannelData(ci, data);

        }

    }

}

void MainWindow::enableDemo(bool enabled)

{

    if (enabled)

    {

        if (!serialPort.isOpen())

        {

            demoTimer.start();

            ui->actionDemoMode->setChecked(true);

        }

        else

        {

            ui->actionDemoMode->setChecked(false);

        }

    }

    else

    {

        demoTimer.stop();

        ui->actionDemoMode->setChecked(false);

    }

}

/*

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

            {

<?xml version="1.0" encoding="UTF-8"?>

<ui version="4.0">

 <class>MainWindow</class>

 <widget class="QMainWindow" name="MainWindow">

  <property name="geometry">

   <rect>

    <x>0</x>

    <y>0</y>

    <width>653</width>

    <height>534</height>

   </rect>

  </property>

  <property name="windowTitle">

   <string>SerialPlot</string>

  </property>

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

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

    <item>

     <widget class="QwtPlot" name="plot">

      <property name="sizePolicy">

       <sizepolicy hsizetype="Preferred" vsizetype="Expanding">

        <horstretch>0</horstretch>

        <verstretch>0</verstretch>

       </sizepolicy>

      </property>

     </widget>

    </item>

    <item>

     <widget class="QTabWidget" name="tabWidget">

      <property name="sizePolicy">

       <sizepolicy hsizetype="Expanding" vsizetype="Minimum">

        <horstretch>0</horstretch>

        <verstretch>0</verstretch>

       </sizepolicy>

      </property>

      <property name="minimumSize">

       <size>

        <width>0</width>

        <height>0</height>

       </size>

      </property>

      <property name="maximumSize">

       <size>

        <width>16777215</width>

        <height>16777215</height>

       </size>

      </property>

      <property name="currentIndex">

       <number>0</number>

      </property>

      <property name="movable">

       <bool>false</bool>

      </property>

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

       <attribute name="title">

        <string>Data Format</string>

       </attribute>

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

        <item>

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

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

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

            <property name="text">

             <string>Number Of Channels:</string>

            </property>

           </widget>

          </item>

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

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

            <property name="minimumSize">

             <size>

              <width>60</width>

              <height>0</height>

             </size>

            </property>

            <property name="keyboardTracking">

             <bool>false</bool>

            </property>

            <property name="minimum">

             <number>1</number>

            </property>

            <property name="maximum">

             <number>32</number>

            </property>

           </widget>

          </item>

         </layout>

        </item>

        <item>

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

          <property name="orientation">

           <enum>Qt::Vertical</enum>

          </property>

         </widget>

        </item>

        <item>

         <widget class="QGroupBox" name="groupBox">

          <property name="title">

           <string>Number Format:</string>

          </property>

          <layout class="QGridLayout" name="gridLayout_2">

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

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

             <property name="text">

              <string>uint8</string>

             </property>

             <property name="checked">

              <bool>true</bool>

             </property>

            </widget>

           </item>

             </property>

             <property name="text">

             </property>

            </widget>

           </item>

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

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

             <property name="text">

              <string>int16</string>

             </property>

            </widget>

           </item>

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

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

             <property name="text">

              <string>int32</string>

             </property>

            </widget>

           </item>

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

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

             <property name="text">

              <string>float</string>

             </property>

            </widget>

           </item>

             <property name="text">

             </property>

            </widget>

           </item>

          </layout>

         </widget>

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

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

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

            <property name="enabled">

             <bool>false</bool>

            </property>

            <property name="maximumSize">

             <size>

              <width>75</width>

              <height>16777215</height>

             </size>

            </property>

            <property name="value">

             <double>0.000000000000000</double>

            </property>

           </widget>

          </item>

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

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

            <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="0" column="0">

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

            <property name="text">

             <string>Number Of Samples:</string>

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

#

# 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/>.

#

#-------------------------------------------------

#

# Project created by QtCreator 2015-03-04T08:20:06

#

#-------------------------------------------------

QT       += core gui serialport

greaterThan(QT_MAJOR_VERSION, 4): QT += widgets

TARGET = serialplot

TEMPLATE = app

CONFIG += qwt

SOURCES += main.cpp\

        mainwindow.cpp \

    customcheckablebutton.cpp \

    portcontrol.cpp

HEADERS  += mainwindow.h \

    utils.h \

    customcheckablebutton.h \

FORMS    += mainwindow.ui \

    about_dialog.ui \

    portcontrol.ui

INCLUDEPATH += qmake/

CONFIG += c++11