/*
 * This file is part of the DSLogic-gui project.
 * DSLogic-gui is based on PulseView.
 *
 * Copyright (C) 2012 Joel Holdsworth <joel@airwebreathe.org.uk>
 * Copyright (C) 2013 DreamSourceLab <dreamsourcelab@dreamsourcelab.com>
 *
 * This program 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 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
 */


#include "triggerdock.h"
#include "../sigsession.h"

#include <QObject>
#include <QGridLayout>
#include <QVBoxLayout>
#include <QPainter>
#include <QRegExpValidator>
#include <QMessageBox>

#include "libsigrok4DSLogic/libsigrok.h"

namespace pv {
namespace dock {

TriggerDock::TriggerDock(QWidget *parent, SigSession &session) :
    QWidget(parent),
    _session(session)
{
    int i;

    QFont font("Monaco");
    font.setStyleHint(QFont::Monospace);
    font.setFixedPitch(true);

    simple_radioButton = new QRadioButton("Simple Trigger", this);
    simple_radioButton->setChecked(true);
    adv_radioButton = new QRadioButton("Advanced Trigger", this);

    position_label = new QLabel("Trigger Position: ", this);
    position_spinBox = new QSpinBox(this);
    position_spinBox->setRange(0, 100);
    position_spinBox->setButtonSymbols(QAbstractSpinBox::NoButtons);
    position_slider = new QSlider(Qt::Horizontal, this);
    position_slider->setRange(0, 100);
    connect(position_slider, SIGNAL(valueChanged(int)), position_spinBox, SLOT(setValue(int)));
    connect(position_spinBox, SIGNAL(valueChanged(int)), position_slider, SLOT(setValue(int)));

    stages_label = new QLabel("Total Trigger Stages: ", this);
    stages_label->setDisabled(true);
    stages_comboBox = new QComboBox(this);
    for (i = 1; i <= TriggerStages; i++)
        stages_comboBox->addItem(QString::number(i));
    //stages_comboBox->setCurrentIndex(stages_comboBox->count() - 1);
    stages_comboBox->setDisabled(true);

    stage_tabWidget = new QTabWidget(this);
    stage_tabWidget->setTabPosition(QTabWidget::East);
    stage_tabWidget->setDisabled(true);

    QRegExp value_rx("[10XRFCxrfc ]+");
    QValidator *value_validator = new QRegExpValidator(value_rx, this);
    for (i = 0; i < TriggerStages; i++) {
        QComboBox *_logic_comboBox = new QComboBox(this);
        _logic_comboBox->addItem(tr("Or"));
        _logic_comboBox->addItem(tr("And"));
        _logic_comboBox->setCurrentIndex(1);
        _logic_comboBox_list.push_back(_logic_comboBox);

        QLineEdit *_value0_lineEdit = new QLineEdit("X X X X X X X X X X X X X X X X", this);
        _value0_lineEdit->setFont(font);
        _value0_lineEdit->setValidator(value_validator);
        _value0_lineEdit->setMaxLength(TriggerProbes * 2 - 1);
        _value0_lineEdit->setInputMask("X X X X X X X X X X X X X X X X");
        _value0_lineEdit_list.push_back(_value0_lineEdit);
        QSpinBox *_count0_spinBox = new QSpinBox(this);
        _count0_spinBox->setRange(1, 1 << TriggerCountBits);
        _count0_spinBox->setButtonSymbols(QAbstractSpinBox::NoButtons);
        _count0_spinBox_list.push_back(_count0_spinBox);
        QComboBox *_inv0_comboBox = new QComboBox(this);
        _inv0_comboBox->addItem(tr("=="));
        _inv0_comboBox->addItem(tr("!="));
        _inv0_comboBox_list.push_back(_inv0_comboBox);

        QLineEdit *_value1_lineEdit = new QLineEdit("X X X X X X X X X X X X X X X X", this);
        _value1_lineEdit->setFont(font);
        _value1_lineEdit->setValidator(value_validator);
        _value1_lineEdit->setMaxLength(TriggerProbes * 2 - 1);
        _value1_lineEdit->setInputMask("X X X X X X X X X X X X X X X X");
        _value1_lineEdit_list.push_back(_value1_lineEdit);
        QSpinBox *_count1_spinBox = new QSpinBox(this);
        _count1_spinBox->setRange(1, 1 << TriggerCountBits);
        _count1_spinBox->setButtonSymbols(QAbstractSpinBox::NoButtons);
        _count1_spinBox_list.push_back(_count1_spinBox);
        QComboBox *_inv1_comboBox = new QComboBox(this);
        _inv1_comboBox->addItem(tr("=="));
        _inv1_comboBox->addItem(tr("!="));
        _inv1_comboBox_list.push_back(_inv1_comboBox);

        QLabel *value_exp_label = new QLabel("1 1 1 1 1 1\n5 4 3 2 1 0 9 8 7 6 5 4 3 2 1 0", this);
        QLabel *inv_exp_label = new QLabel("Inv", this);
        QLabel *count_exp_label = new QLabel("Counter", this);
        value_exp_label->setFont(font);

        QVBoxLayout *stage_layout = new QVBoxLayout();
        QGridLayout *stage_glayout = new QGridLayout();
        stage_glayout->addWidget(value_exp_label, 1, 0);
        stage_glayout->addWidget(inv_exp_label, 1, 1);
        stage_glayout->addWidget(count_exp_label, 1, 2);
        stage_glayout->addWidget(_value0_lineEdit, 2, 0);
        stage_glayout->addWidget(_inv0_comboBox, 2, 1);
        stage_glayout->addWidget(_count0_spinBox, 2, 2);
        stage_glayout->addWidget(_logic_comboBox, 2, 3);
        stage_glayout->addWidget(_value1_lineEdit, 3, 0);
        stage_glayout->addWidget(_inv1_comboBox, 3, 1);
        stage_glayout->addWidget(_count1_spinBox, 3, 2);
        stage_layout->addLayout(stage_glayout);
        stage_layout->addSpacing(160);
        stage_layout->addWidget(new QLabel("X: Don't care\n0: Low level\n1: High level\nR: Rising edge\nF: Falling edge\nC: Rising/Falling edge"));
        stage_layout->addStretch(1);

        QGroupBox *_stage_groupBox = new QGroupBox("Stage"+QString::number(i), this);
        _stage_groupBox->setFlat(true);
        _stage_groupBox->setLayout(stage_layout);
        _stage_groupBox_list.push_back(_stage_groupBox);

        stage_tabWidget->addTab((QWidget *)_stage_groupBox, QString::number(i));

        connect(_value0_lineEdit, SIGNAL(editingFinished()), this, SLOT(value_changed()));
        connect(_value1_lineEdit, SIGNAL(editingFinished()), this, SLOT(value_changed()));
        connect(_logic_comboBox, SIGNAL(currentIndexChanged(int)), this, SLOT(logic_changed(int)));
        connect(_inv0_comboBox, SIGNAL(currentIndexChanged(int)), this, SLOT(inv_changed(int)));
        connect(_inv1_comboBox, SIGNAL(currentIndexChanged(int)), this, SLOT(inv_changed(int)));
        connect(_count0_spinBox, SIGNAL(editingFinished()), this, SLOT(count_changed()));
        connect(_count1_spinBox, SIGNAL(editingFinished()), this, SLOT(count_changed()));
    }

    connect(simple_radioButton, SIGNAL(clicked()), this, SLOT(simple_trigger()));
    connect(adv_radioButton, SIGNAL(clicked()), this, SLOT(adv_trigger()));
    connect(stages_comboBox, SIGNAL(currentIndexChanged(int)), this, SLOT(trigger_stages_changed(int)));
    connect(position_slider, SIGNAL(valueChanged(int)), this, SLOT(pos_changed(int)));

    QVBoxLayout *layout = new QVBoxLayout(this);
    QGridLayout *gLayout = new QGridLayout();
    gLayout->addWidget(simple_radioButton, 0, 0);
    gLayout->addWidget(adv_radioButton, 1, 0);
    gLayout->addWidget(position_label, 2, 0);
    gLayout->addWidget(position_spinBox, 2, 1);
    gLayout->addWidget(new QLabel(this), 2, 2);
    gLayout->addWidget(position_slider, 3, 0, 1, 3);
    gLayout->addWidget(stages_label, 4, 0);
    gLayout->addWidget(stages_comboBox, 4, 1);
    gLayout->addWidget(new QLabel(this), 4, 2);
    gLayout->setColumnStretch(2, 1);

    layout->addLayout(gLayout);
    layout->addWidget(stage_tabWidget);
    layout->addStretch(1);
    setLayout(layout);
}

TriggerDock::~TriggerDock()
{
}

void TriggerDock::paintEvent(QPaintEvent *)
{
    QStyleOption opt;
    opt.init(this);
    QPainter p(this);
    style()->drawPrimitive(QStyle::PE_Widget, &opt, &p, this);
}

void TriggerDock::simple_trigger()
{
    int i;
    stages_label->setDisabled(true);
    stages_comboBox->setDisabled(true);
    stage_tabWidget->setDisabled(true);
    for (i = 0; i < TriggerStages; i++) {
        stage_tabWidget->setTabEnabled(i, true);
//        _mu_label_list.at(i)->setDisabled(true);
//        _logic_comboBox_list.at(i)->setDisabled(true);

//        _value0_lineEdit_list.at(i)->setDisabled(true);
//        _count0_spinBox_list.at(i)->setDisabled(true);
//        _inv0_comboBox_list.at(i)->setDisabled(true);

//        _value1_lineEdit_list.at(i)->setDisabled(true);
//        _count1_spinBox_list.at(i)->setDisabled(true);
//        _inv1_comboBox_list.at(i)->setDisabled(true);
    }
    ds_trigger_set_mode(SIMPLE_TRIGGER);
    _session.set_adv_trigger(false);
}

void TriggerDock::adv_trigger()
{
    if (strcmp(_session.get_device()->driver->name, "DSLogic") == 0) {
        widget_enable();
        ds_trigger_set_mode(ADV_TRIGGER);
        _session.set_adv_trigger(true);
    } else {
        QMessageBox msg(this);
        msg.setText("Trigger");
        msg.setInformativeText("Advanced Trigger need DSLogic Hardware Support!");
        msg.setStandardButtons(QMessageBox::Ok);
        msg.setIcon(QMessageBox::Warning);
        msg.exec();
        simple_radioButton->setChecked(true);
    }
}

void TriggerDock::trigger_stages_changed(int index)
{
    widget_enable();
    ds_trigger_set_stage(index);
    value_changed();
    logic_changed(0);
    inv_changed(0);
    count_changed();
}

void TriggerDock::widget_enable()
{
    int i;
    int enable_stages;
    stages_label->setDisabled(false);
    stages_comboBox->setVisible(true);
    stages_comboBox->setDisabled(false);
    stage_tabWidget->setDisabled(false);
    enable_stages = stages_comboBox->currentText().toInt();
    for (i = 0; i < enable_stages; i++) {
        stage_tabWidget->setTabEnabled(i, true);
//        _mu_label_list.at(i)->setVisible(true);
//        _mu_label_list.at(i)->setDisabled(false);
//        _logic_comboBox_list.at(i)->setVisible(true);
//        _logic_comboBox_list.at(i)->setDisabled(false);

//        _value0_lineEdit_list.at(i)->setVisible(true);
//        _value0_lineEdit_list.at(i)->setDisabled(false);
//        _count0_spinBox_list.at(i)->setVisible(true);
//        _count0_spinBox_list.at(i)->setDisabled(false);
//        _inv0_comboBox_list.at(i)->setVisible(true);
//        _inv0_comboBox_list.at(i)->setDisabled(false);

//        _value1_lineEdit_list.at(i)->setVisible(true);
//        _value1_lineEdit_list.at(i)->setDisabled(false);
//        _count1_spinBox_list.at(i)->setVisible(true);
//        _count1_spinBox_list.at(i)->setDisabled(false);
//        _inv1_comboBox_list.at(i)->setVisible(true);
//        _inv1_comboBox_list.at(i)->setDisabled(false);
    }
    for (i = enable_stages; i < TriggerStages; i++) {
          stage_tabWidget->setTabEnabled(i, false);
//        _mu_label_list.at(i)->setVisible(false);
//        _logic_comboBox_list.at(i)->setVisible(false);

//        _value0_lineEdit_list.at(i)->setVisible(false);
//        _count0_spinBox_list.at(i)->setVisible(false);
//        _inv0_comboBox_list.at(i)->setVisible(false);

//        _value1_lineEdit_list.at(i)->setVisible(false);
//        _count1_spinBox_list.at(i)->setVisible(false);
//        _inv1_comboBox_list.at(i)->setVisible(false);
    }
}

void TriggerDock::value_changed()
{
    int i;

    for (i = 0; i < stages_comboBox->currentText().toInt(); i++) {
        _value0_lineEdit_list.at(i)->setText(_value0_lineEdit_list.at(i)->text().toUpper());
        while(_value0_lineEdit_list.at(i)->text().length() < TriggerProbes)
            _value0_lineEdit_list.at(i)->setText("X" + _value0_lineEdit_list.at(i)->text());

        _value1_lineEdit_list.at(i)->setText(_value1_lineEdit_list.at(i)->text().toUpper());
        while(_value1_lineEdit_list.at(i)->text().length() < TriggerProbes)
            _value1_lineEdit_list.at(i)->setText("X" + _value1_lineEdit_list.at(i)->text());

        ds_trigger_stage_set_value(i, TriggerProbes,
                             _value0_lineEdit_list.at(i)->text().toLocal8Bit().data(),
                             _value1_lineEdit_list.at(i)->text().toLocal8Bit().data());
    }
}

void TriggerDock::logic_changed(int index)
{
    (void)index;

    int i;

    for (i = 0; i < stages_comboBox->currentText().toInt(); i++) {
        ds_trigger_stage_set_logic(i, TriggerProbes,
                             _logic_comboBox_list.at(i)->currentIndex());
    }
}

void TriggerDock::inv_changed(int index)
{
    (void)index;

    int i;

    for (i = 0; i < stages_comboBox->currentText().toInt(); i++) {
        ds_trigger_stage_set_inv(i, TriggerProbes,
                             _inv0_comboBox_list.at(i)->currentIndex(),
                             _inv1_comboBox_list.at(i)->currentIndex());
    }
}

void TriggerDock::count_changed()
{
    int i;

    for (i = 0; i < stages_comboBox->currentText().toInt(); i++) {
        ds_trigger_stage_set_count(i, TriggerProbes,
                                   _count0_spinBox_list.at(i)->value() - 1,
                                   _count1_spinBox_list.at(i)->value() - 1);
    }
}

void TriggerDock::pos_changed(int pos)
{
    ds_trigger_set_pos(pos);
}

void TriggerDock::device_change()
{
    if (strcmp(_session.get_device()->driver->name, "DSLogic") != 0) {
        position_spinBox->setDisabled(true);
        position_slider->setDisabled(true);
    } else {
        position_spinBox->setDisabled(false);
        position_slider->setDisabled(false);
    }
}

} // namespace dock
} // namespace pv