/*

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

#include "utils.h"

#include "version.h"

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

#include <QtPlugin>

Q_IMPORT_PLUGIN(QWindowsIntegrationPlugin)

#endif

struct Range

{

    double rmin;

    double rmax;

};

Q_DECLARE_METATYPE(Range);

MainWindow::MainWindow(QWidget *parent) :

    QMainWindow(parent),

    ui(new Ui::MainWindow),

    aboutDialog(this),

    portControl(&serialPort),

    commandPanel(&serialPort),

    dataFormatPanel(&serialPort, &channelBuffers),

    snapshotMan(this, &channelBuffers)

{

    ui->setupUi(this);

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

    ui->tabWidget->insertTab(1, &dataFormatPanel, "Data Format");

    ui->tabWidget->insertTab(3, &commandPanel, "Commands");

    ui->tabWidget->setCurrentIndex(0);

    addToolBar(portControl.toolBar());

    ui->menuBar->insertMenu(ui->menuHelp->menuAction(), snapshotMan.menu());

    setupAboutDialog();

    // init view menu

    for (auto a : ui->plot->menuActions())

    {

        ui->menuView->addAction(a);

    }

    // init UI signals

    // menu signals

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

              &aboutDialog, &QWidget::show);

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

                     this, &MainWindow::onExportCsv);

    ui->actionQuit->setShortcutContext(Qt::ApplicationShortcut);

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

                     this, &MainWindow::close);

    // port control signals

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

                     this, &MainWindow::onPortToggled);

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

                     &dataFormatPanel, &DataFormatPanel::requestSkipByte);

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

    QObject::connect(snapshotMan.takeSnapshotAction(), &QAction::triggered,

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

    // 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 format and reader

    QObject::connect(&dataFormatPanel, &DataFormatPanel::dataAdded,

                     ui->plot, &QwtPlot::replot);

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

                     &dataFormatPanel, &DataFormatPanel::pause);

    // init data arrays and plot

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

    unsigned numOfChannels = dataFormatPanel.numOfChannels();

    QObject::connect(&dataFormatPanel,

                     &DataFormatPanel::numOfChannelsChanged,

                     this,

                     &MainWindow::onNumOfChannelsChanged);

    // 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(Plot::makeColor(i));

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

    }

    // init auto scale

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

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

    // init scale range preset list

    for (int nbits = 8; nbits <= 24; nbits++) // signed binary formats

    {

        int rmax = pow(2, nbits-1)-1;

        int rmin = -rmax-1;

        Range r = {double(rmin),  double(rmax)};

        ui->cbRangePresets->addItem(

            QString().sprintf("Signed %d bits %d to +%d", nbits, rmin, rmax),

            QVariant::fromValue(r));

    }

    for (int nbits = 8; nbits <= 24; nbits++) // unsigned binary formats

    {

        int rmax = pow(2, nbits)-1;

        ui->cbRangePresets->addItem(

            QString().sprintf("Unsigned %d bits %d to +%d", nbits, 0, rmax),

            QVariant::fromValue(Range{0, double(rmax)}));

    }

    ui->cbRangePresets->addItem("-1 to +1", QVariant::fromValue(Range{-1, +1}));

    ui->cbRangePresets->addItem("0 to +1", QVariant::fromValue(Range{0, +1}));

    ui->cbRangePresets->addItem("-100 to +100", QVariant::fromValue(Range{-100, +100}));

    ui->cbRangePresets->addItem("0 to +100", QVariant::fromValue(Range{0, +100}));

    QObject::connect(ui->cbRangePresets, SIGNAL(activated(int)),

                     this, SLOT(onRangeSelected()));

    // Init sps (sample per second) counter

    spsLabel.setText("0sps");

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

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

    QObject::connect(&dataFormatPanel,

                     &DataFormatPanel::samplesPerSecondChanged,

                     this, &MainWindow::onSpsChanged);

    // init demo

    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::onPortError(QSerialPort::SerialPortError error)

{

    switch(error)

    {

        case QSerialPort::NoError :

            break;

        case QSerialPort::ResourceError :

            qWarning() << "Port error: resource unavaliable; most likely device removed.";

            if (serialPort.isOpen())

            {

                qWarning() << "Closing port on resource error: " << serialPort.portName();

                portControl.togglePort();

            }

            portControl.loadPortList();

            break;

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

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

            <property name="text">

             <string>Number Of Samples:</string>

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

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

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

            <property name="enabled">

             <bool>false</bool>

            </property>

            <property name="text">

             <string>Ymin</string>

            </property>

           </widget>

          </item>

          <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="toolTip">

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

            </property>

            <property name="value">

             <double>0.000000000000000</double>

            </property>

           </widget>

          </item>

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

           <widget class="QComboBox" name="cbRangePresets"/>

          </item>

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

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

            <property name="text">

             <string>Select Range Preset:</string>

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

      </widget>

     </widget>

    </item>

   </layout>

  </widget>

  <widget class="QMenuBar" name="menuBar">

   <property name="geometry">

    <rect>

     <x>0</x>

     <y>0</y>

     <width>653</width>

     <height>27</height>

    </rect>

   </property>

   <widget class="QMenu" name="menuHelp">

    <property name="title">

     <string>Help</string>

    </property>

    <addaction name="actionDemoMode"/>

    <addaction name="actionHelpAbout"/>

   </widget>

   <widget class="QMenu" name="menuFile">

    <property name="title">

     <string>File</string>

    </property>

    <addaction name="actionExportCsv"/>

    <addaction name="separator"/>

    <addaction name="actionQuit"/>

   </widget>

   <widget class="QMenu" name="menuView">

    <property name="title">

     <string>View</string>

    </property>

   </widget>

   <addaction name="menuFile"/>

   <addaction name="menuView"/>

   <addaction name="menuHelp"/>

  </widget>

   <attribute name="toolBarArea">

    <enum>TopToolBarArea</enum>

   </attribute>

   <attribute name="toolBarBreak">

    <bool>false</bool>

   </attribute>

   <addaction name="actionPause"/>

   <addaction name="actionClear"/>

  </widget>

  <widget class="QStatusBar" name="statusBar"/>

  <action name="actionPause">

   <property name="checkable">

    <bool>true</bool>

   </property>

   <property name="checked">

    <bool>false</bool>

   </property>

   <property name="text">

    <string>Pause</string>

   </property>

   <property name="toolTip">

    <string>Pause Plotting</string>

   </property>

   <property name="shortcut">

    <string>P</string>

   </property>

  </action>

  <action name="actionClear">

   <property name="text">

    <string>Clear</string>

   </property>

   <property name="shortcut">

    <string>Ctrl+K</string>

   </property>

  </action>

  <action name="actionHelpAbout">

   <property name="text">

    <string>About</string>

   </property>

  </action>

  <action name="actionDemoMode">

   <property name="checkable">

    <bool>true</bool>

   </property>

   <property name="text">

    <string>Demo Mode</string>

   </property>

   <property name="toolTip">

    <string>Toggle Demo Mode</string>

   </property>

  </action>

  <action name="actionExportCsv">

   <property name="text">

    <string>Export CSV</string>

   </property>

   <property name="toolTip">

    <string>Export plot data to CSV</string>

   </property>

  </action>

  <action name="actionQuit">

   <property name="text">

    <string>Quit</string>

   </property>

   <property name="shortcut">

    <string>Ctrl+Q</string>

   </property>

  </action>

 </widget>

 <layoutdefault spacing="6" margin="11"/>

 <customwidgets>

  <customwidget>

   <class>Plot</class>

   <extends>QWidget</extends>

   <header>plot.h</header>

   <container>1</container>

  </customwidget>

  <customwidget>

   <class>HidableTabWidget</class>

   <extends>QTabWidget</extends>

   <header>hidabletabwidget.h</header>

   <container>1</container>

  </customwidget>

 </customwidgets>

 <resources/>

 <connections/>

</ui>

/*

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

#include "ui_portcontrol.h"

#include <QSerialPortInfo>

#include <QKeySequence>

#include <QLabel>

#include <QtDebug>

#include "utils.h"

#define TBPORTLIST_MINWIDTH (200)

PortControl::PortControl(QSerialPort* port, QWidget* parent) :

    QWidget(parent),

    ui(new Ui::PortControl),

    openAction("Open", this)

{

    ui->setupUi(this);

    serialPort = port;

    // setup the toolbar

    openAction.setCheckable(true);

    openAction.setShortcut(QKeySequence("F2"));

    openAction.setToolTip("Open Port (F2)");

    QObject::connect(&openAction, &QAction::triggered,

                     this, &PortControl::openActionTriggered);

    portToolBar.addWidget(&tbPortList);

    portToolBar.addAction(&openAction);

    // setup port selection widgets

    tbPortList.setMinimumWidth(TBPORTLIST_MINWIDTH);

    tbPortList.setModel(&portList);

    ui->cbPortList->setModel(&portList);

    QObject::connect(ui->cbPortList,

                     SELECT<int>::OVERLOAD_OF(&QComboBox::activated),

                     this, &PortControl::onCbPortListActivated);

    QObject::connect(&tbPortList,

                     SELECT<int>::OVERLOAD_OF(&QComboBox::activated),

                     this, &PortControl::onTbPortListActivated);

    QObject::connect(ui->cbPortList,

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

                     this, &PortControl::selectPort);

    QObject::connect(&tbPortList,

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

                     this, &PortControl::selectPort);

    // setup buttons

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

                     this, &PortControl::loadPortList);

    ui->pbOpenPort->setDefaultAction(&openAction);

    // setup baud rate selection widget

    QObject::connect(ui->cbBaudRate,

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

                     this, &PortControl::selectBaudRate);

    // setup parity selection buttons

    parityButtons.addButton(ui->rbNoParity, (int) QSerialPort::NoParity);

    parityButtons.addButton(ui->rbEvenParity, (int) QSerialPort::EvenParity);

    parityButtons.addButton(ui->rbOddParity, (int) QSerialPort::OddParity);

    QObject::connect(&parityButtons,

                     SELECT<int>::OVERLOAD_OF(&QButtonGroup::buttonClicked),

                     this, &PortControl::selectParity);

    // setup data bits selection buttons

    dataBitsButtons.addButton(ui->rb8Bits, (int) QSerialPort::Data8);

    dataBitsButtons.addButton(ui->rb7Bits, (int) QSerialPort::Data7);

    dataBitsButtons.addButton(ui->rb6Bits, (int) QSerialPort::Data6);

    dataBitsButtons.addButton(ui->rb5Bits, (int) QSerialPort::Data5);

    QObject::connect(&dataBitsButtons,

                     SELECT<int>::OVERLOAD_OF(&QButtonGroup::buttonClicked),

                     this, &PortControl::selectDataBits);

    // setup stop bits selection buttons

    stopBitsButtons.addButton(ui->rb1StopBit, (int) QSerialPort::OneStop);

    stopBitsButtons.addButton(ui->rb2StopBit, (int) QSerialPort::TwoStop);

    QObject::connect(&stopBitsButtons,

                     SELECT<int>::OVERLOAD_OF(&QButtonGroup::buttonClicked),

                     this, &PortControl::selectStopBits);

    // setup flow control selection buttons

    flowControlButtons.addButton(ui->rbNoFlowControl,

                                 (int) QSerialPort::NoFlowControl);

    flowControlButtons.addButton(ui->rbHardwareControl,

                                 (int) QSerialPort::HardwareControl);

    flowControlButtons.addButton(ui->rbSoftwareControl,

                                 (int) QSerialPort::SoftwareControl);

    QObject::connect(&flowControlButtons,

                     SELECT<int>::OVERLOAD_OF(&QButtonGroup::buttonClicked),

                     this, &PortControl::selectFlowControl);

    // init skip byte button

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

                     [this](){emit skipByteRequested();});

    loadPortList();

    loadBaudRateList();

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

}

PortControl::~PortControl()

{

    delete ui;

}

void PortControl::loadPortList()

{

    QString currentSelection = ui->cbPortList->currentData(PortNameRole).toString();

    portList.loadPortList();

    int index = portList.indexOf(currentSelection);

    if (index >= 0)

    {

        ui->cbPortList->setCurrentIndex(index);

        tbPortList.setCurrentIndex(index);

    }

}

void PortControl::loadBaudRateList()

{

    ui->cbBaudRate->clear();

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

    {

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

    }

}

void PortControl::selectBaudRate(QString baudRate)

{

    if (serialPort->isOpen())

    {

        if (!serialPort->setBaudRate(baudRate.toInt()))

        {

            qCritical() << "Can't set baud rate!";

        }

    }

}

void PortControl::selectParity(int parity)

{

    if (serialPort->isOpen())

    {

        if(!serialPort->setParity((QSerialPort::Parity) parity))

        {

            qCritical() << "Can't set parity option!";

        }

    }

}

void PortControl::selectDataBits(int dataBits)

{

    if (serialPort->isOpen())

    {

        if(!serialPort->setDataBits((QSerialPort::DataBits) dataBits))

        {

            qCritical() << "Can't set numer of data bits!";

        }

    }

}

void PortControl::selectStopBits(int stopBits)

{

    if (serialPort->isOpen())

    {

        if(!serialPort->setStopBits((QSerialPort::StopBits) stopBits))

        {

            qCritical() << "Can't set number of stop bits!";

        }

    }

}

void PortControl::selectFlowControl(int flowControl)

{

    if (serialPort->isOpen())

    {

        if(!serialPort->setFlowControl((QSerialPort::FlowControl) flowControl))

        {

            qCritical() << "Can't set flow control option!";

        }

    }

}

void PortControl::togglePort()

{

    if (serialPort->isOpen())

    {

        serialPort->close();

        qDebug() << "Closed port:" << serialPort->portName();

        emit portToggled(false);

    }

    else

    {

        // we get the port name from the edit text, which may not be

        // in the portList if user hasn't pressed Enter

        // Also note that, portText may not be the `portName`

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

        QString portName;

        int portIndex = portList.indexOf(portText);

        if (portIndex < 0) // not in list, add to model and update the selections

        {