#include "ui_portcontrol.h"

#include <QSerialPortInfo>

#include <QKeySequence>

#include <QtDebug>

#include "utils.h"

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

    QWidget(parent),

    ui(new Ui::PortControl),

    portToolBar("Port"),

    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.addAction(&openAction);

    portToolBar.addWidget(&tbPortList);

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

    // setup buttons

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

                     this, &PortControl::loadPortList);

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

    // TODO: port name coming from combobox is dirty, create a separate layer of signals

    //       that will sanitize this information

    QObject::connect(ui->cbPortList,

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

                     this, &PortControl::selectPort);

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

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

        }

    }

}

void PortControl::enableSkipByte(bool enabled)

{

    ui->pbSkipByte->setDisabled(enabled);

}

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

}

  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;

    QToolBar* toolBar();

private:

    Ui::PortControl *ui;

    QButtonGroup parityButtons;

    QButtonGroup dataBitsButtons;

    QButtonGroup stopBitsButtons;

    QButtonGroup flowControlButtons;

    QToolBar portToolBar;

    QAction openAction;

    QComboBox tbPortList;

    PortList portList;

public slots:

    void loadPortList();

    void loadBaudRateList();

    void togglePort();

    void selectPort(QString portName);

    void enableSkipByte(bool enabled = true);

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

    void portToggled(bool open);

};

#endif // PORTCONTROL_H