/*

  Copyright © 2017 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 <QMessageBox>

#include <QFile>

#include <QTextStream>

#include <QMenu>

#include <QDesktopServices>

#include <QMap>

#include <QtDebug>

#include <qwt_plot.h>

#include <limits.h>

#include <cmath>

#include <iostream>

#include <plot.h>

#include <barplot.h>

#include "framebufferseries.h"

#include "utils.h"

#include "defines.h"

#include "version.h"

#include "setting_defines.h"

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

#include <QtPlugin>

Q_IMPORT_PLUGIN(QWindowsIntegrationPlugin)

#endif

const QMap<int, QString> panelSettingMap({

        {0, "Port"},

        {1, "DataFormat"},

        {2, "Plot"},

        {3, "Commands"}

    });

MainWindow::MainWindow(QWidget *parent) :

    QMainWindow(parent),

    ui(new Ui::MainWindow),

    aboutDialog(this),

    portControl(&serialPort),

    channelMan(1, 1, this),

    secondaryPlot(NULL),

    snapshotMan(this, &channelMan),

    commandPanel(&serialPort),

    dataFormatPanel(&serialPort, &channelMan, &recorder),

    recordPanel(&recorder, &channelMan),

    updateCheckDialog(this)

{

    ui->setupUi(this);

    plotMan = new PlotManager(ui->plotArea, &plotMenu, channelMan.infoModel());

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

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

    ui->tabWidget->insertTab(2, &plotControlPanel, "Plot");

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

    ui->tabWidget->insertTab(4, &recordPanel, "Record");

    ui->tabWidget->setCurrentIndex(0);

    auto tbPortControl = portControl.toolBar();

    addToolBar(tbPortControl);

    addToolBar(recordPanel.toolbar());

    ui->plotToolBar->addAction(snapshotMan.takeSnapshotAction());

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

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

    connect(&commandPanel, &CommandPanel::focusRequested, [this]()

            {

                this->ui->tabWidget->setCurrentWidget(&commandPanel);

                this->ui->tabWidget->showTabs();

            });

    tbPortControl->setObjectName("tbPortControl");

    ui->plotToolBar->setObjectName("tbPlot");

    setupAboutDialog();

    // init view menu

    ui->menuBar->insertMenu(ui->menuSecondary->menuAction(), &plotMenu);

    plotMenu.addSeparator();

    QMenu* tbMenu = plotMenu.addMenu("Toolbars");

    tbMenu->addAction(ui->plotToolBar->toggleViewAction());

    tbMenu->addAction(portControl.toolBar()->toggleViewAction());

    // init UI signals

    // Secondary plot menu signals

    connect(ui->actionBarPlot, &QAction::triggered,

            this, &MainWindow::showBarPlot);

    // Help menu signals

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

              &aboutDialog, &QWidget::show);

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

              &updateCheckDialog, &QWidget::show);

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

                     [](){QDesktopServices::openUrl(QUrl(BUG_REPORT_URL));});

    // File menu signals

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

                     this, &MainWindow::onExportCsv);

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

                     this, &MainWindow::onSaveSettings);

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

                     this, &MainWindow::onLoadSettings);

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

    // plot control signals

    connect(&plotControlPanel, &PlotControlPanel::numOfSamplesChanged,

            this, &MainWindow::onNumOfSamplesChanged);

    connect(&plotControlPanel, &PlotControlPanel::numOfSamplesChanged,

            plotMan, &PlotManager::setNumOfSamples);

    connect(&plotControlPanel, &PlotControlPanel::yScaleChanged,

            plotMan, &PlotManager::setYAxis);

    connect(&plotControlPanel, &PlotControlPanel::xScaleChanged,

            plotMan, &PlotManager::setXAxis);

    connect(&plotControlPanel, &PlotControlPanel::plotWidthChanged,

            plotMan, &PlotManager::setPlotWidth);

    // plot toolbar signals

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

                     this, SLOT(clearPlot()));

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

                     plotMan, &PlotManager::flashSnapshotOverlay);

    // init port signals

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

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

    // init data format and reader

    QObject::connect(&channelMan, &ChannelManager::dataAdded,

                     plotMan, &PlotManager::replot);

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

                     &channelMan, &ChannelManager::pause);

    QObject::connect(&recordPanel, &RecordPanel::recordStarted,

                     &dataFormatPanel, &DataFormatPanel::startRecording);

    QObject::connect(&recordPanel, &RecordPanel::recordStopped,

                     &dataFormatPanel, &DataFormatPanel::stopRecording);

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

                     [this](bool enabled)

                     {

                         if (enabled && !recordPanel.recordPaused())

                         {

                             dataFormatPanel.pause(true);

                         }

                         else

                         {

                             dataFormatPanel.pause(false);

                         }

                     });

    QObject::connect(&recordPanel, &RecordPanel::recordPausedChanged,

                     [this](bool enabled)

                     {

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

                         {

                             dataFormatPanel.pause(false);

                         }

                     });

    connect(&serialPort, &QIODevice::aboutToClose,

            &recordPanel, &RecordPanel::onPortClose);

    // init data arrays and plot

    numOfSamples = plotControlPanel.numOfSamples();

    unsigned numOfChannels = dataFormatPanel.numOfChannels();

    channelMan.setNumOfSamples(numOfSamples);

    channelMan.setNumOfChannels(dataFormatPanel.numOfChannels());

    connect(&dataFormatPanel, &DataFormatPanel::numOfChannelsChanged,

            &channelMan, &ChannelManager::setNumOfChannels);

    connect(&channelMan, &ChannelManager::numOfChannelsChanged,

            this, &MainWindow::onNumOfChannelsChanged);

    plotControlPanel.setChannelInfoModel(channelMan.infoModel());

    // init curve list

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

    {

        plotMan->addCurve(channelMan.channelName(i), channelMan.channelBuffer(i));

    }

    // init scales

    plotMan->setYAxis(plotControlPanel.autoScale(),

                      plotControlPanel.yMin(), plotControlPanel.yMax());

    plotMan->setXAxis(plotControlPanel.xAxisAsIndex(),

                      plotControlPanel.xMin(), plotControlPanel.xMax());

    plotMan->setNumOfSamples(numOfSamples);

    plotMan->setPlotWidth(plotControlPanel.plotWidth());

    // Init sps (sample per second) counter

    spsLabel.setText("0sps");

    spsLabel.setToolTip("samples per second (per channel)");

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

    QObject::connect(&dataFormatPanel,

                     &DataFormatPanel::samplesPerSecondChanged,

                     this, &MainWindow::onSpsChanged);

    // init demo

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

                     this, &MainWindow::enableDemo);

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

                     plotMan, &PlotManager::showDemoIndicator);

    // load default settings

    QSettings settings("serialplot", "serialplot");

    loadAllSettings(&settings);

    // ensure command panel has 1 command if none loaded

    if (!commandPanel.numOfCommands())

    {

        commandPanel.newCommandAction()->trigger();

    }

    // Important: This should be after newCommandAction is triggered

    // (above) we don't want user to be greeted with command panel on

    // the very first run.

    connect(commandPanel.newCommandAction(), &QAction::triggered, [this]()

            {

                this->ui->tabWidget->setCurrentWidget(&commandPanel);

                this->ui->tabWidget->showTabs();

            });

}

MainWindow::~MainWindow()

{

    if (serialPort.isOpen())

    {

        serialPort.close();

    }

    delete plotMan;

    delete ui;

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

}

void MainWindow::closeEvent(QCloseEvent * event)

{

    // save snapshots

    if (!snapshotMan.isAllSaved())

    {

        auto clickedButton = QMessageBox::warning(

            this, "Closing SerialPlot",

            "There are un-saved snapshots. If you close you will loose the data.",

            QMessageBox::Discard, QMessageBox::Cancel);

        if (clickedButton == QMessageBox::Cancel)

        {

            event->ignore();

            return;

        }

    }

    // save settings

    QSettings settings("serialplot", "serialplot");

    saveAllSettings(&settings);

    settings.sync();

<?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>650</height>

   </rect>

  </property>

  <property name="windowTitle">

   <string>SerialPlot</string>

  </property>

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

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

    <item>

     <widget class="QSplitter" name="splitter">

      <property name="sizePolicy">

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

        <horstretch>0</horstretch>

        <verstretch>0</verstretch>

       </sizepolicy>

      </property>

      <property name="orientation">

       <enum>Qt::Horizontal</enum>

      </property>

      <widget class="QWidget" name="plotArea" native="true">

       <property name="sizePolicy">

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

         <horstretch>0</horstretch>

         <verstretch>0</verstretch>

        </sizepolicy>

       </property>

      </widget>

     </widget>

    </item>

    <item>

     <widget class="HidableTabWidget" 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="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>24</height>

    </rect>

   </property>

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

    <property name="title">

     <string>&amp;Help</string>

    </property>

    <addaction name="actionDemoMode"/>

    <addaction name="actionReportBug"/>

    <addaction name="actionCheckUpdate"/>

    <addaction name="actionHelpAbout"/>

   </widget>

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

    <property name="title">

     <string>&amp;File</string>

    </property>

    <addaction name="actionSaveSettings"/>

    <addaction name="actionLoadSettings"/>

    <addaction name="actionExportCsv"/>

    <addaction name="separator"/>

    <addaction name="actionQuit"/>

   </widget>

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

    <property name="title">

     <string>Secondary</string>

    </property>

    <addaction name="actionBarPlot"/>

    <addaction name="separator"/>

    <addaction name="actionVertical"/>

   </widget>

   <addaction name="menuFile"/>

   <addaction name="menuSecondary"/>

   <addaction name="menuHelp"/>

  </widget>

  <widget class="QToolBar" name="plotToolBar">

   <property name="windowTitle">

    <string>Plot Toolbar</string>

   </property>

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

    <iconset theme="player_pause">

     <normaloff>.</normaloff>.</iconset>

   </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="icon">

    <iconset theme="editclear">

     <normaloff>.</normaloff>.</iconset>

   </property>

   <property name="text">

    <string>Clear</string>

   </property>

   <property name="shortcut">

    <string>Ctrl+K</string>

   </property>

  </action>

  <action name="actionHelpAbout">

   <property name="text">

    <string>&amp;About</string>

   </property>

  </action>

  <action name="actionDemoMode">

   <property name="checkable">

    <bool>true</bool>

   </property>

   <property name="text">

    <string>&amp;Demo Mode</string>

   </property>

   <property name="toolTip">

    <string>Toggle Demo Mode</string>

   </property>

  </action>

  <action name="actionExportCsv">

   <property name="text">

    <string>&amp;Export CSV</string>

   </property>

   <property name="toolTip">

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

   </property>

  </action>

  <action name="actionQuit">

   <property name="text">

    <string>&amp;Quit</string>

   </property>

   <property name="shortcut">

    <string>Ctrl+Q</string>

   </property>

  </action>

  <action name="actionReportBug">

   <property name="text">

    <string>&amp;Report a Bug</string>

   </property>

   <property name="toolTip">

    <string>Report a Bug on SerialPlot Website</string>

   </property>

  </action>

  <action name="actionSaveSettings">

   <property name="text">

    <string>&amp;Save Settings</string>

   </property>

   <property name="toolTip">

    <string>Save Settings to a File</string>

   </property>

  </action>

  <action name="actionLoadSettings">

   <property name="text">

    <string>&amp;Load Settings</string>

   </property>

   <property name="toolTip">

    <string>Load Settings from a File</string>

   </property>

  </action>

  <action name="actionCheckUpdate">

   <property name="text">

    <string>&amp;Check Update</string>

   </property>

  </action>

  <action name="actionBarPlot">

   <property name="checkable">

    <bool>true</bool>

   </property>

   <property name="text">

    <string>Bar Plot</string>

   </property>

  </action>

  <action name="actionVertical">

   <property name="checkable">

    <bool>true</bool>

   </property>

   <property name="text">

    <string>Vertical</string>

   </property>

  </action>

  <action name="actionHorizontal">

   <property name="checkable">

    <bool>true</bool>

   </property>

   <property name="text">

    <string>Horizontal</string>

   </property>

  </action>

 </widget>

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

 <customwidgets>

  <customwidget>

   <class>HidableTabWidget</class>

   <extends>QTabWidget</extends>

   <header>hidabletabwidget.h</header>

   <container>1</container>

  </customwidget>

 </customwidgets>

 <resources/>

 <connections/>

</ui>