/*

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

#include <QMessageBox>

#include <QCheckBox>

#include <math.h>

#include "color_selector.hpp"

#include "plotcontrolpanel.h"

#include "ui_plotcontrolpanel.h"

#include "setting_defines.h"

/// Confirm if #samples is being set to a value greater than this

/// Used for scale range selection combobox

struct Range

{

    double rmin;

    double rmax;

};

Q_DECLARE_METATYPE(Range);

PlotControlPanel::PlotControlPanel(QWidget *parent) :

    QWidget(parent),

    ui(new Ui::PlotControlPanel),

    resetAct(tr("Reset"), this),

    resetNamesAct(tr("Reset Names"), this),

    resetColorsAct(tr("Reset Colors"), this),

    showAllAct(tr("Show All"), this),

    resetMenu(tr("Reset Menu"), this)

{

    ui->setupUi(this);

    warnNumOfSamples = true;    // TODO: load from settings

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

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

    ui->spXmin->setRange((-1) * std::numeric_limits<double>::max(),

                         std::numeric_limits<double>::max());

    ui->spXmax->setRange((-1) * std::numeric_limits<double>::max(),

                         std::numeric_limits<double>::max());

    // connect signals

    connect(ui->spNumOfSamples, SIGNAL(valueChanged(int)),

            this, SLOT(onNumOfSamples(int)));

    connect(ui->cbAutoScale, &QCheckBox::toggled,

            this, &PlotControlPanel::onAutoScaleChecked);

    connect(ui->spYmax, SIGNAL(valueChanged(double)),

            this, SLOT(onYScaleChanged()));

    connect(ui->spYmin, SIGNAL(valueChanged(double)),

            this, SLOT(onYScaleChanged()));

    connect(ui->cbIndex, &QCheckBox::toggled,

            this, &PlotControlPanel::onIndexChecked);

    connect(ui->spXmax, SIGNAL(valueChanged(double)),

            this, SLOT(onXScaleChanged()));

    connect(ui->spXmin, SIGNAL(valueChanged(double)),

            this, SLOT(onXScaleChanged()));

    connect(ui->spPlotWidth, SIGNAL(valueChanged(int)),

            this, SLOT(onPlotWidthChanged()));

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

    // color selector starts disabled until a channel is selected

    ui->colorSelector->setColor(QColor(0,0,0,0));

    ui->colorSelector->setDisplayMode(color_widgets::ColorPreview::AllAlpha);

    ui->colorSelector->setDisabled(true);

    // reset button

    resetMenu.addAction(&resetNamesAct);

    resetMenu.addAction(&resetColorsAct);

    resetMenu.addAction(&showAllAct);

    resetAct.setMenu(&resetMenu);

    ui->tbReset->setDefaultAction(&resetAct);

}

PlotControlPanel::~PlotControlPanel()

{

    delete ui;

}

unsigned PlotControlPanel::numOfSamples()

{

    return ui->spNumOfSamples->value();

}

void PlotControlPanel::onNumOfSamples(int value)

{

    if (warnNumOfSamples && value > NUMSAMPLES_CONFIRM_AT)

    {

        // ask confirmation

        if (!askNSConfirmation(value))

        {

            // revert to old value

            disconnect(ui->spNumOfSamples, SIGNAL(valueChanged(int)),

                       this, SLOT(onNumOfSamples(int)));

            ui->spNumOfSamples->setValue(_numOfSamples);

            connect(ui->spNumOfSamples, SIGNAL(valueChanged(int)),

                    this, SLOT(onNumOfSamples(int)));

            return;

        }

    }

    _numOfSamples = value;

    emit numOfSamplesChanged(value);

}

bool PlotControlPanel::askNSConfirmation(int value)

{

    auto text = tr("Setting number of samples to a too big value "

                   "(>%1) can seriously impact the performance of "

                   "the application and cause freezes. Are you sure you "

                   "want to change the number of samples to %2?")

        .arg(QString::number(NUMSAMPLES_CONFIRM_AT), QString::number(value));

    // TODO: parent the mainwindow

    QMessageBox mb(QMessageBox::Warning,

                   tr("Confirm Number of Samples"),

                   text,

                   QMessageBox::Apply | QMessageBox::Cancel,

                   this);

    auto cb = new QCheckBox("Don't show this again.");

    connect(cb, &QCheckBox::stateChanged, [this](int state)

            {

                warnNumOfSamples = (state == Qt::Unchecked);

            });

    mb.setCheckBox(cb);

    return mb.exec() == QMessageBox::Apply;

}

void PlotControlPanel::onAutoScaleChecked(bool checked)

{

    if (checked)

    {

        ui->lYmin->setEnabled(false);

        ui->lYmax->setEnabled(false);

        ui->spYmin->setEnabled(false);

        ui->spYmax->setEnabled(false);

        emit yScaleChanged(true); // autoscale

    }

    else

    {

        ui->lYmin->setEnabled(true);

        ui->lYmax->setEnabled(true);

        ui->spYmin->setEnabled(true);

        ui->spYmax->setEnabled(true);

        emit yScaleChanged(false, ui->spYmin->value(), ui->spYmax->value());

    }

}

void PlotControlPanel::onYScaleChanged()

{

    if (!autoScale())

    {

        emit yScaleChanged(false, ui->spYmin->value(), ui->spYmax->value());

    }

}

bool PlotControlPanel::autoScale() const

{

    return ui->cbAutoScale->isChecked();

}

double PlotControlPanel::yMax() const

{

    return ui->spYmax->value();

}

double PlotControlPanel::yMin() const

{

    return ui->spYmin->value();

}

bool PlotControlPanel::xAxisAsIndex() const

{

    return ui->cbIndex->isChecked();

}

double PlotControlPanel::xMax() const

{

    return ui->spXmax->value();

}

double PlotControlPanel::xMin() const

{

    return ui->spXmin->value();

}

void PlotControlPanel::onRangeSelected()

{

    Range r = ui->cbRangePresets->currentData().value<Range>();

    ui->spYmin->setValue(r.rmin);

    ui->spYmax->setValue(r.rmax);

    ui->cbAutoScale->setChecked(false);

}

void PlotControlPanel::onIndexChecked(bool checked)

{

    if (checked)

    {

        ui->lXmin->setEnabled(false);

        ui->lXmax->setEnabled(false);

        ui->spXmin->setEnabled(false);

        ui->spXmax->setEnabled(false);

        emit xScaleChanged(true); // use index

    }

    else

    {

        ui->lXmin->setEnabled(true);

        ui->lXmax->setEnabled(true);

        ui->spXmin->setEnabled(true);

        ui->spXmax->setEnabled(true);

        emit xScaleChanged(false, ui->spXmin->value(), ui->spXmax->value());

    }

    emit plotWidthChanged(plotWidth());

}

void PlotControlPanel::onXScaleChanged()

{

    if (!xAxisAsIndex())

    {

        emit xScaleChanged(false, ui->spXmin->value(), ui->spXmax->value());

        emit plotWidthChanged(plotWidth());

    }

}

double PlotControlPanel::plotWidth() const

{

    double value = ui->spPlotWidth->value();

    if (!xAxisAsIndex())

    {

        // scale by xmin and xmax

        auto xmax = ui->spXmax->value();

        auto xmin = ui->spXmin->value();

        double scale = (xmax - xmin) / _numOfSamples;

        value *= scale;

    }

    return value;

}

void PlotControlPanel::onPlotWidthChanged()

{

    emit plotWidthChanged(plotWidth());

}

void PlotControlPanel::setChannelInfoModel(ChannelInfoModel* model)

{

    ui->tvChannelInfo->setModel(model);

    // channel color selector

    connect(ui->tvChannelInfo->selectionModel(), &QItemSelectionModel::currentRowChanged,

            [this](const QModelIndex &current, const QModelIndex &previous)

            {

                // TODO: duplicate with below lambda

                QColor color(0,0,0,0); // transparent

                if (current.isValid())

                {

                    ui->colorSelector->setEnabled(true);

                    auto model = ui->tvChannelInfo->model();

                    color = model->data(current, Qt::ForegroundRole).value<QColor>();

                }

                else

                {

                    ui->colorSelector->setDisabled(true);

                }

                // temporarily block signals because `setColor` emits `colorChanged`

                bool wasBlocked = ui->colorSelector->blockSignals(true);

                ui->colorSelector->setColor(color);

                ui->colorSelector->blockSignals(wasBlocked);

            });

    connect(ui->tvChannelInfo->selectionModel(), &QItemSelectionModel::selectionChanged,

            [this](const QItemSelection & selected, const QItemSelection & deselected)

            {

                if (!selected.length())

                {

                    ui->colorSelector->setDisabled(true);

                    // temporarily block signals because `setColor` emits `colorChanged`

                    bool wasBlocked = ui->colorSelector->blockSignals(true);

                    ui->colorSelector->setColor(QColor(0,0,0,0));

                    ui->colorSelector->blockSignals(wasBlocked);

                }

            });

    connect(ui->colorSelector, &color_widgets::ColorSelector::colorChanged,

            [this](QColor color)

            {

                auto index = ui->tvChannelInfo->selectionModel()->currentIndex();

                ui->tvChannelInfo->model()->setData(index, color, Qt::ForegroundRole);

            });

    connect(model, &QAbstractItemModel::dataChanged,

            [this](const QModelIndex & topLeft, const QModelIndex & bottomRight, const QVector<int> & roles = QVector<int> ())

            {

                auto current = ui->tvChannelInfo->selectionModel()->currentIndex();

                // no current selection

                if (!current.isValid()) return;

                auto mod = ui->tvChannelInfo->model();

                QColor color = mod->data(current, Qt::ForegroundRole).value<QColor>();

                // temporarily block signals because `setColor` emits `colorChanged`

                bool wasBlocked = ui->colorSelector->blockSignals(true);

                ui->colorSelector->setColor(color);

                ui->colorSelector->blockSignals(wasBlocked);

            });

    // reset actions

    connect(&resetAct, &QAction::triggered, model, &ChannelInfoModel::resetInfos);

    connect(&resetNamesAct, &QAction::triggered, model, &ChannelInfoModel::resetNames);

    connect(&resetColorsAct, &QAction::triggered, model, &ChannelInfoModel::resetColors);

    connect(&showAllAct, &QAction::triggered, model, &ChannelInfoModel::resetVisibility);

}

void PlotControlPanel::saveSettings(QSettings* settings)

{

    settings->beginGroup(SettingGroup_Plot);

    settings->setValue(SG_Plot_NumOfSamples, numOfSamples());

    settings->setValue(SG_Plot_PlotWidth, ui->spPlotWidth->value());

    settings->setValue(SG_Plot_IndexAsX, xAxisAsIndex());

    settings->setValue(SG_Plot_XMax, xMax());

    settings->setValue(SG_Plot_XMin, xMin());

    settings->setValue(SG_Plot_AutoScale, autoScale());

    settings->setValue(SG_Plot_YMax, yMax());

    settings->setValue(SG_Plot_YMin, yMin());

    settings->endGroup();

}

void PlotControlPanel::loadSettings(QSettings* settings)

{

    settings->beginGroup(SettingGroup_Plot);

    ui->spNumOfSamples->setValue(

        settings->value(SG_Plot_NumOfSamples, numOfSamples()).toInt());

    ui->spPlotWidth->setValue(

        settings->value(SG_Plot_PlotWidth, ui->spPlotWidth->value()).toInt());

    ui->cbIndex->setChecked(

        settings->value(SG_Plot_IndexAsX, xAxisAsIndex()).toBool());

    ui->spXmax->setValue(settings->value(SG_Plot_XMax, xMax()).toDouble());

    ui->spXmin->setValue(settings->value(SG_Plot_XMin, xMin()).toDouble());

    ui->cbAutoScale->setChecked(

        settings->value(SG_Plot_AutoScale, autoScale()).toBool());

    ui->spYmax->setValue(settings->value(SG_Plot_YMax, yMax()).toDouble());

    ui->spYmin->setValue(settings->value(SG_Plot_YMin, yMin()).toDouble());

    settings->endGroup();

}

<?xml version="1.0" encoding="UTF-8"?>

<ui version="4.0">

 <class>PlotControlPanel</class>

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

  <property name="geometry">

   <rect>

    <x>0</x>

    <y>0</y>

    <width>706</width>

    <height>195</height>

   </rect>

  </property>

  <property name="windowTitle">

   <string>Form</string>

  </property>

  <layout class="QHBoxLayout" name="horizontalLayout">

   <item>

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

     <property name="sizePolicy">

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

       <horstretch>0</horstretch>

       <verstretch>0</verstretch>

      </sizepolicy>

     </property>

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

      <property name="spacing">

       <number>3</number>

      </property>

      <property name="leftMargin">

       <number>0</number>

      </property>

      <property name="topMargin">

       <number>0</number>

      </property>

      <property name="rightMargin">

       <number>0</number>

      </property>

      <property name="bottomMargin">

       <number>0</number>

      </property>

      <item>

       <widget class="QTableView" name="tvChannelInfo">

        <property name="sizePolicy">

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

          <horstretch>0</horstretch>

          <verstretch>0</verstretch>

         </sizepolicy>

        </property>

        <property name="maximumSize">

         <size>

          <width>300</width>

          <height>170</height>

         </size>

        </property>

        <property name="selectionMode">

         <enum>QAbstractItemView::SingleSelection</enum>

        </property>

        <property name="selectionBehavior">

         <enum>QAbstractItemView::SelectRows</enum>

        </property>

       </widget>

      </item>

      <item>

       <layout class="QHBoxLayout" name="horizontalLayout_2">

        <property name="sizeConstraint">

         <enum>QLayout::SetMaximumSize</enum>

        </property>

        <item>

         <widget class="color_widgets::ColorSelector" name="colorSelector" native="true">

          <property name="sizePolicy">

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

            <horstretch>0</horstretch>

            <verstretch>0</verstretch>

           </sizepolicy>

          </property>

          <property name="minimumSize">

           <size>

            <width>20</width>

            <height>20</height>

           </size>

          </property>

          <property name="maximumSize">

           <size>

            <width>20</width>

            <height>20</height>

           </size>

          </property>

         </widget>

        </item>

        <item>

         <spacer name="horizontalSpacer">

          <property name="orientation">

           <enum>Qt::Horizontal</enum>

          </property>

          <property name="sizeHint" stdset="0">

           <size>

            <width>1</width>

            <height>20</height>

           </size>

          </property>

         </spacer>

        </item>

        <item>

         <widget class="QToolButton" name="tbReset">

          <property name="text">

           <string>Reset</string>

          </property>

          <property name="popupMode">

           <enum>QToolButton::MenuButtonPopup</enum>

          </property>

          <property name="arrowType">

           <enum>Qt::NoArrow</enum>

          </property>

         </widget>

        </item>

       </layout>

      </item>

     </layout>

    </widget>

   </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::FieldsStayAtSizeHint</enum>

     </property>

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

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

       <property name="toolTip">

        <string/>

       </property>

       <property name="text">

        <string>Buffer Size:</string>

       </property>

      </widget>

     </item>

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

      <widget class="QSpinBox" name="spNumOfSamples">

       <property name="minimumSize">

        <size>

         <width>80</width>

         <height>0</height>

        </size>

       </property>

       <property name="toolTip">

        <string>Length of acquisition as number of samples</string>

       </property>

       <property name="keyboardTracking">

        <bool>false</bool>

       </property>

       <property name="minimum">

        <number>2</number>

       </property>

       <property name="maximum">

       </property>

       <property name="value">

        <number>1000</number>

       </property>

      </widget>

     </item>

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

      <widget class="QCheckBox" name="cbIndex">

       <property name="text">

        <string>Index as X AXis</string>

       </property>

       <property name="checked">

        <bool>true</bool>

       </property>

      </widget>

     </item>

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

      <layout class="QHBoxLayout" name="horizontalLayout_4">

       <item>

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

         <property name="enabled">

          <bool>false</bool>

         </property>

         <property name="text">

          <string>Xmin</string>

         </property>

        </widget>

       </item>

       <item>

        <widget class="QDoubleSpinBox" name="spXmin">

         <property name="enabled">

          <bool>false</bool>

         </property>

         <property name="minimumSize">

          <size>

           <width>75</width>

           <height>0</height>

          </size>

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

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

         <property name="enabled">

          <bool>false</bool>

         </property>

         <property name="text">

          <string>Xmax</string>

         </property>

        </widget>

       </item>

       <item>

        <widget class="QDoubleSpinBox" name="spXmax">

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

      </layout>

     </item>

     <item row="3" 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="3" column="1">

      <layout class="QHBoxLayout" name="horizontalLayout_3">

       <item>

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

         <property name="enabled">

          <bool>false</bool>

         </property>

         <property name="text">

          <string>Ymin</string>

         </property>

        </widget>

       </item>

       <item>

        <widget class="QDoubleSpinBox" name="spYmin">

         <property name="enabled">

          <bool>false</bool>

         </property>

         <property name="minimumSize">

          <size>

           <width>75</width>

           <height>0</height>

          </size>

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

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

         <property name="enabled">

          <bool>false</bool>

         </property>

         <property name="text">

          <string>Ymax</string>

         </property>

        </widget>

       </item>

       <item>

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

      </layout>

     </item>

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

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

       <property name="text">

        <string>Select Range Preset:</string>

       </property>

      </widget>

     </item>

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

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

     </item>

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

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

       <property name="toolTip">

        <string/>

       </property>

       <property name="text">

        <string>Plot Width:</string>

       </property>

      </widget>

     </item>

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

      <widget class="QSpinBox" name="spPlotWidth">

       <property name="minimumSize">

        <size>

         <width>80</width>

         <height>0</height>

        </size>

       </property>

       <property name="toolTip">

        <string>Width of X axis as maximum number of samples that are shown in plot</string>

       </property>

       <property name="keyboardTracking">

        <bool>false</bool>

       </property>

       <property name="minimum">

        <number>2</number>

       </property>

       <property name="maximum">

        <number>100000</number>

       </property>

       <property name="value">

        <number>1000</number>

       </property>

      </widget>

     </item>

    </layout>

   </item>

   <item>

    <spacer name="horizontalSpacer_2">

     <property name="orientation">

      <enum>Qt::Horizontal</enum>

     </property>

     <property name="sizeType">

      <enum>QSizePolicy::MinimumExpanding</enum>

     </property>

     <property name="sizeHint" stdset="0">

      <size>

       <width>1</width>

       <height>20</height>

      </size>

     </property>

    </spacer>

   </item>

  </layout>

 </widget>

 <customwidgets>

  <customwidget>

   <class>color_widgets::ColorSelector</class>

   <extends>QWidget</extends>

   <header>color_selector.hpp</header>

   <container>1</container>

  </customwidget>

 </customwidgets>

 <resources/>

 <connections/>

</ui>