{

    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;

        case QSerialPort::DeviceNotFoundError:

            qCritical() << "Device doesn't exists: " << serialPort.portName();

            break;

        case QSerialPort::PermissionError:

            qCritical() << "Permission denied. Either you don't have \

required privileges or device is already opened by another process.";

            break;

        case QSerialPort::OpenError:

            qWarning() << "Device is already opened!";

            break;

        case QSerialPort::NotOpenError:

            qCritical() << "Device is not open!";

            break;

        case QSerialPort::ParityError:

            qCritical() << "Parity error detected.";

            break;

        case QSerialPort::FramingError:

            qCritical() << "Framing error detected.";

            break;

        case QSerialPort::BreakConditionError:

            qCritical() << "Break condition is detected.";

            break;

        case QSerialPort::WriteError:

            qCritical() << "An error occurred while writing data.";

            break;

        case QSerialPort::ReadError:

            qCritical() << "An error occurred while reading data.";

            break;

        case QSerialPort::UnsupportedOperationError:

            qCritical() << "Operation is not supported.";

            break;

        case QSerialPort::TimeoutError:

            qCritical() << "A timeout error occurred.";

            break;

        case QSerialPort::UnknownError:

            qCritical() << "Unknown error! Error: " << serialPort.errorString();

            break;

        default:

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

            break;

    }

}

void MainWindow::clearPlot()

{

    for (unsigned ci = 0; ci < channelMan.numOfChannels(); ci++)

    {

        channelMan.channelBuffer(ci)->clear();

    }

    plotMan->replot();

}

void MainWindow::onNumOfSamplesChanged(int value)

{

    numOfSamples = value;

    channelMan.setNumOfSamples(value);

    plotMan->replot();

}

void MainWindow::onNumOfChannelsChanged(unsigned value)

{

    unsigned int oldNum = plotMan->numOfCurves();

    unsigned numOfChannels = value;

    if (numOfChannels > oldNum)

    {

        // add new channels

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

        {

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

        }

    }

    else if(numOfChannels < oldNum)

    {

        plotMan->removeCurves(oldNum - numOfChannels);

    }

    plotMan->replot();

}

void MainWindow::onChannelNameChanged(unsigned channel, QString name)

{

    // This slot is triggered also when a new channel is added, in

    // this case curve list doesn't contain said channel. No worries,

    // since `onNumOfChannelsChanged` slot will update curve list.

    if (channel < plotMan->numOfCurves()) // check if channel exists in curve list

    {

        plotMan->setTitle(channel, name);

    }

}

void MainWindow::onSpsChanged(unsigned sps)

{

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

}

bool MainWindow::isDemoRunning()

{

    return ui->actionDemoMode->isChecked();

}

void MainWindow::enableDemo(bool enabled)

{

    if (enabled)

    {

        if (!serialPort.isOpen())

        {

            dataFormatPanel.enableDemo(true);

        }

        else

        {

            ui->actionDemoMode->setChecked(false);

        }

    }

    else

    {

        dataFormatPanel.enableDemo(false);

        ui->actionDemoMode->setChecked(false);

    }

}

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

    {

        Snapshot* snapshot = snapshotMan.makeSnapshot();

        snapshot->save(fileName);

        delete snapshot;

    }

}

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;

    }

    if (ui != NULL) ui->ptLog->appendPlainText(logString);

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

    if (type != QtDebugMsg && ui != NULL)

    {

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

    }

}

void MainWindow::saveSettings(QSettings* settings)

{

    // save window geometry

    settings->beginGroup("MainWindow");

    settings->setValue("size", size());

    settings->setValue("pos", pos());

    settings->endGroup();

}

void MainWindow::loadSettings(QSettings* settings)

{

    // load window geometry

    settings->beginGroup("MainWindow");

    resize(settings->value("size", size()).toSize());

    move(settings->value("pos", pos()).toPoint());

    settings->endGroup();

}

void MainWindow::onSaveSettings()

{

    QString fileName = QFileDialog::getSaveFileName(

        this, tr("Save Settings"), QString(), "INI (*.ini)");

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

    {

        QSettings settings(fileName, QSettings::IniFormat);

        saveSettings(&settings);

    }

}

void MainWindow::onLoadSettings()

{

    QString fileName = QFileDialog::getOpenFileName(

        this, tr("Load Settings"), QString(), "INI (*.ini)");

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

    {

        QSettings settings(fileName, QSettings::IniFormat);

        loadSettings(&settings);

    }

}

/*

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

    portToolBar("Port Toolbar"),

    openAction("Open", this),

    loadPortListAction("↺", this)

{

    ui->setupUi(this);

    serialPort = port;

    // setup actions

    openAction.setCheckable(true);

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

    openAction.setToolTip("Open Port");

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

                     this, &PortControl::openActionTriggered);

    loadPortListAction.setToolTip("Reload port list");

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

                     [this](bool checked){loadPortList();});

    // setup toolbar

    portToolBar.addWidget(&tbPortList);

    portToolBar.addAction(&loadPortListAction);

    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

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

    ui->pbReloadPorts->setDefaultAction(&loadPortListAction);

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

    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

        {

            portList.appendRow(new PortListItem(portText));

            ui->cbPortList->setCurrentIndex(portList.rowCount()-1);

            tbPortList.setCurrentIndex(portList.rowCount()-1);

            portName = portText;

        }

        else

        {

            // get the port name from the data field

            portName = static_cast<PortListItem*>(portList.item(portIndex))->portName();

        }

        serialPort->setPortName(ui->cbPortList->currentData(PortNameRole).toString());

        // open port

        if (serialPort->open(QIODevice::ReadWrite))

        {

            // set port settings

            selectBaudRate(ui->cbBaudRate->currentText());

            selectParity((QSerialPort::Parity) parityButtons.checkedId());

            selectDataBits((QSerialPort::DataBits) dataBitsButtons.checkedId());

            selectStopBits((QSerialPort::StopBits) stopBitsButtons.checkedId());

            selectFlowControl((QSerialPort::FlowControl) flowControlButtons.checkedId());

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

            emit portToggled(true);

        }

    }

    openAction.setChecked(serialPort->isOpen());

}

void PortControl::selectPort(QString portName)

{

    // portName may be coming from combobox

    portName = portName.split(" ")[0];

    // has selection actually changed

    if (portName != serialPort->portName())

    {

        // if another port is already open, close it by toggling

        if (serialPort->isOpen())

        {

            togglePort();

            // open new selection by toggling

            togglePort();

        }

    }

}

QToolBar* PortControl::toolBar()

{

    return &portToolBar;

}

void PortControl::openActionTriggered(bool checked)

{

    togglePort();

}

void PortControl::onCbPortListActivated(int index)

{

    tbPortList.setCurrentIndex(index);

}

void PortControl::onTbPortListActivated(int index)

{

    ui->cbPortList->setCurrentIndex(index);

}

/*

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

*/

#ifndef PORTCONTROL_H

#define PORTCONTROL_H

#include <QWidget>

#include <QButtonGroup>

#include <QSerialPort>

#include <QStringList>

#include <QToolBar>

#include <QAction>

#include <QComboBox>

#include "portlist.h"

namespace Ui {

class PortControl;

}

class PortControl : public QWidget

{

    Q_OBJECT

public:

    explicit PortControl(QSerialPort* port, QWidget* parent = 0);

    ~PortControl();

    QSerialPort* serialPort;

private:

    Ui::PortControl *ui;

    QButtonGroup parityButtons;

    QButtonGroup dataBitsButtons;

    QButtonGroup stopBitsButtons;

    QButtonGroup flowControlButtons;

    QToolBar portToolBar;

    QAction openAction;

    QAction loadPortListAction;

    QComboBox tbPortList;

    PortList portList;

public slots:

    void loadPortList();

    void loadBaudRateList();

    void togglePort();

    void selectPort(QString portName);

    void selectBaudRate(QString baudRate);

    void selectParity(int parity); // parity must be one of QSerialPort::Parity

    void selectDataBits(int dataBits); // bits must be one of QSerialPort::DataBits

    void selectStopBits(int stopBits); // stopBits must be one of QSerialPort::StopBits

    void selectFlowControl(int flowControl); // flowControl must be one of QSerialPort::FlowControl

private slots:

    void openActionTriggered(bool checked);

    void onCbPortListActivated(int index);

    void onTbPortListActivated(int index);

signals:

    void portToggled(bool open);

};

#endif // PORTCONTROL_H

/*

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

#include <QIcon>

#include <QtDebug>

#include "portlist.h"

PortListItem::PortListItem(QString name, QString description, quint16 vid, quint16 pid)

{

    construct(name, description, vid, pid);

}

PortListItem::PortListItem(QSerialPortInfo* portInfo)

{

    if (portInfo->hasProductIdentifier())

    {

        construct(portInfo->portName(),

                  portInfo->description(),

                  portInfo->vendorIdentifier(),

                  portInfo->productIdentifier());

    }

    else

    {

        construct(portInfo->portName());

    }

}

void PortListItem::construct(QString name, QString description, quint16 vid, quint16 pid)

{

    QString text = name;

    if (!description.isEmpty())

    {

        text += QString(" ") + description;

    }

    if (vid && pid)

    {

        text += QString("[%1:").arg(vid, 4, 16, QChar('0'));

        text += QString("%1]").arg(pid, 4, 16, QChar('0'));

        setIcon(QIcon(":/usb_icon.png"));

    }

    else if (name.contains("rfcomm"))

    {

        setIcon(QIcon(":/bluetooth_icon.png"));

    }

    else

    {

        setIcon(QIcon(":/rs232_icon.png"));

    }

    setText(text);

    setData(name, PortNameRole);

}

QString PortListItem::portName()

{

    return data(PortNameRole).toString();

}

PortList::PortList(QObject* parent) :

    QStandardItemModel(parent)

{

    loadPortList();

    // we have to use macro based notation to be able to disconnect

    QObject::connect(this, SIGNAL(rowsInserted(const QModelIndex &, int, int)),

                     this, SLOT(onRowsInserted(QModelIndex, int, int)));

}

void PortList::loadPortList()

{

    clear();

    disconnect(this, SLOT(onRowsInserted(QModelIndex,int,int)));

    for (auto portInfo : QSerialPortInfo::availablePorts())

    {

        appendRow(new PortListItem(&portInfo));

    }

    for (auto portName : userEnteredPorts)

    {

        appendRow(new PortListItem(portName));

    }

    QObject::connect(this, SIGNAL(rowsInserted(const QModelIndex &, int, int)),

                     this, SLOT(onRowsInserted(QModelIndex, int, int)));

}

int PortList::indexOf(QString portText)

{

    for (int i = 0; i < rowCount(); i++)

    {

        if (item(i)->text() == portText)

        {

            return i;

        }

    }

    return -1; // not found

}

void PortList::onRowsInserted(QModelIndex parent, int start, int end)

{

    PortListItem* newItem = static_cast<PortListItem*>(item(start));

    QString portName = newItem->text();

    newItem->setData(portName, PortNameRole);

    userEnteredPorts << portName;

}

/*

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

*/

#ifndef PORTLIST_H

#define PORTLIST_H

#include <QStandardItemModel>

#include <QStandardItem>

#include <QList>

#include <QSerialPortInfo>

enum PortListRoles

{

    PortNameRole = Qt::UserRole+1  // portName as QString

};

class PortListItem : public QStandardItem

{

public:

    PortListItem(QString name, QString description="", quint16 vid=0, quint16 pid=0);

    PortListItem(QSerialPortInfo* portInfo);

    QString portName(); // returns only the port name

private:

    // common constructor

    void construct(QString name, QString description="", quint16 vid=0, quint16 pid=0);

};

class PortList : public QStandardItemModel

{

    Q_OBJECT

public:

    PortList(QObject* parent=0);

    void loadPortList();

    int indexOf(QString portText); // return -1 if not found

private:

    QStringList userEnteredPorts;

private slots:

    void onRowsInserted(QModelIndex parent, int start, int end);

};

#endif // PORTLIST_H