/*

  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 "framebufferseries.h"

FrameBufferSeries::FrameBufferSeries(FrameBuffer* buffer)

{

    xAsIndex = true;

    _xmin = 0;

    _xmax = 1;

    _buffer = buffer;

    int_index_start = 0;

    int_index_end = 0;

}

void FrameBufferSeries::setXAxis(bool asIndex, double xmin, double xmax)

{

    xAsIndex = asIndex;

    _xmin = xmin;

    _xmax = xmax;

}

size_t FrameBufferSeries::size() const

{

    return int_index_end - int_index_start;

}

    {

        return QPointF(i * (_xmax - _xmin) / _buffer->size() + _xmin, _buffer->sample(i));

    }

}

QRectF FrameBufferSeries::boundingRect() const

{

    if (xAsIndex)

    {

        return _buffer->boundingRect();

    }

    else

    {

        auto rect = _buffer->boundingRect();

        rect.setLeft(_xmin);

        rect.setRight(_xmax);

        return rect;

    }

}

void FrameBufferSeries::setRectOfInterest(const QRectF& rect)

{

    if (xAsIndex)

    {

    }

    else

    {

        double xsize = _xmax - _xmin;

        size_t bsize = _buffer->size();

    }

    int_index_start = std::max(int_index_start, (size_t) 0);

    int_index_end = std::min(_buffer->size(), int_index_end);

}

    snapshotOverlay = NULL;

    connect(&zoomer, &QwtPlotZoomer::zoomed,

            [this](const QRectF &rect)

            {

                onXScaleChanged();

            });

    connect(this, &QwtPlot::itemAttached,

            [this](QwtPlotItem *plotItem, bool on)

            {

                if (symbolSize) updateSymbols();

            });

    // init demo indicator

    QwtText demoText(" DEMO RUNNING ");  // looks better with spaces

    demoText.setColor(QColor("white"));

    demoText.setBackgroundBrush(Qt::darkRed);

    demoText.setBorderRadius(4);

    demoText.setRenderFlags(Qt::AlignLeft | Qt::AlignBottom);

    demoIndicator.setText(demoText);

    demoIndicator.hide();

    demoIndicator.attach(this);

}

Plot::~Plot()

{

    if (snapshotOverlay != NULL) delete snapshotOverlay;

}

void Plot::setYAxis(bool autoScaled, double yAxisMin, double yAxisMax)

{

    this->isAutoScaled = autoScaled;

    if (!autoScaled)

    {

        yMin = yAxisMin;

        yMax = yAxisMax;

    }

    zoomer.zoom(0);

    resetAxes();

}

void Plot::setXAxis(double xMin, double xMax)

{

    _xMin = xMin;

    grid.enableX(show);

    grid.enableY(show);

    replot();

}

void Plot::showMinorGrid(bool show)

{

    grid.enableXMin(show);

    grid.enableYMin(show);

    replot();

}

void Plot::showLegend(bool show)

{

    legend.setVisible(show);

    replot();

}

void Plot::showDemoIndicator(bool show)

{

    demoIndicator.setVisible(show);

    replot();

}

void Plot::unzoom()

{

    zoomer.zoom(0);

}

void Plot::darkBackground(bool enabled)

{

    QColor gridColor;

    if (enabled)

    {

        setCanvasBackground(QBrush(Qt::black));

        gridColor.setHsvF(0, 0, 0.25);

        grid.setPen(gridColor);

        zoomer.setRubberBandPen(QPen(Qt::white));

        zoomer.setTrackerPen(QPen(Qt::white));

        sZoomer.setPickerPen(QPen(Qt::white));

        legend.setTextPen(QPen(Qt::white));

    }

    else

    {

        setCanvasBackground(QBrush(Qt::white));

        gridColor.setHsvF(0, 0, 0.80);

        grid.setPen(gridColor);

        zoomer.setRubberBandPen(QPen(Qt::black));

class Plot : public QwtPlot

{

    Q_OBJECT

    friend class PlotManager;

public:

    enum ShowSymbols

    {

        ShowSymbolsAuto,

        ShowSymbolsShow,

        ShowSymbolsHide

    };

    Plot(QWidget* parent = 0);

    ~Plot();

    static QColor makeColor(unsigned int channelIndex);

public slots:

    void showGrid(bool show = true);

    void showMinorGrid(bool show = true);

    void showLegend(bool show = true);

    void showDemoIndicator(bool show = true);

    void unzoom();

    void darkBackground(bool enabled = true);

    void setYAxis(bool autoScaled, double yMin = 0, double yMax = 1);

    void setXAxis(double xMin, double xMax);

    void setSymbols(ShowSymbols shown);

    /**

     * Displays an animation for snapshot.

     *

     * @param light show a light colored (white) animation or the opposite

     */

    void flashSnapshotOverlay(bool light);

    void setNumOfSamples(unsigned value);

protected:

    /// update the display of symbols depending on `symbolSize`

    void updateSymbols();

private:

    bool isAutoScaled;

    double yMin, yMax;

    double _xMin, _xMax;

    unsigned numOfSamples;

    int symbolSize;

    Zoomer zoomer;

    ScaleZoomer sZoomer;

    QwtPlotGrid grid;

    PlotSnapshotOverlay* snapshotOverlay;

    QwtPlotLegendItem legend;

    QwtPlotTextLabel demoIndicator;

    ShowSymbols showSymbols;

    void resetAxes();

    void resizeEvent(QResizeEvent * event);

    void calcSymbolSize();

private slots:

    void unzoomed();

    void onXScaleChanged();

};

#endif // PLOT_H

/*

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

#include <QMetaEnum>

#include <QtDebug>

#include "qwt_symbol.h"

#include "plot.h"

#include "plotmanager.h"

#include "utils.h"

#include "setting_defines.h"

PlotManager::PlotManager(QWidget* plotArea, ChannelInfoModel* infoModel, QObject *parent) :

    QObject(parent),

    _plotArea(plotArea),

    showGridAction("&Grid", this),

    showMinorGridAction("&Minor Grid", this),

    unzoomAction("&Unzoom", this),

    darkBackgroundAction("&Dark Background", this),

    showLegendAction("&Legend", this),

    showMultiAction("Multi &Plot", this),

    setSymbolsAction("Symbols", this)

{

    _autoScaled = true;

    _yMin = 0;

    _yMax = 1;

    _xAxisAsIndex = true;

    isDemoShown = false;

    _infoModel = infoModel;

    _numOfSamples = 1;

    showSymbols = Plot::ShowSymbolsAuto;

    // initalize layout and single widget

    isMulti = false;

    scrollArea = NULL;

    setupLayout(isMulti);

    addPlotWidget();

    // initialize menu actions

    showGridAction.setToolTip("Show Grid");

    showMinorGridAction.setToolTip("Show Minor Grid");

    unzoomAction.setToolTip("Unzoom the Plot");

    darkBackgroundAction.setToolTip("Enable Dark Plot Background");

    showLegendAction.setToolTip("Display the Legend on Plot");

    showMultiAction.setToolTip("Display All Channels Separately");

    setSymbolsAction.setToolTip("Show/Hide symbols");

    showGridAction.setShortcut(QKeySequence("G"));

    showMinorGridAction.setShortcut(QKeySequence("M"));

    showGridAction.setCheckable(true);

    showMinorGridAction.setCheckable(true);

    darkBackgroundAction.setCheckable(true);

    showLegendAction.setCheckable(true);

    showMultiAction.setCheckable(true);

        connect(_infoModel, &QAbstractItemModel::modelReset,

                [this]()

                {

                    onChannelInfoChanged(_infoModel->index(0, 0), // start

                                         _infoModel->index(_infoModel->rowCount()-1, 0), // end

                                         {}); // roles ignored

                });

    }

}

PlotManager::~PlotManager()

{

    while (curves.size())

    {

        delete curves.takeLast();

    }

    // remove all widgets

    while (plotWidgets.size())

    {

        delete plotWidgets.takeLast();

    }

    if (scrollArea != NULL) delete scrollArea;

}

void PlotManager::onChannelInfoChanged(const QModelIndex &topLeft,

                                       const QModelIndex &bottomRight,

                                       const QVector<int> &roles)

{

    int start = topLeft.row();

    int end = bottomRight.row();

    for (int ci = start; ci <= end; ci++)

    {

        QString name = topLeft.sibling(ci, ChannelInfoModel::COLUMN_NAME).data(Qt::EditRole).toString();

        QColor color = topLeft.sibling(ci, ChannelInfoModel::COLUMN_NAME).data(Qt::ForegroundRole).value<QColor>();

        bool visible = topLeft.sibling(ci, ChannelInfoModel::COLUMN_VISIBILITY).data(Qt::CheckStateRole).toBool();

        curves[ci]->setTitle(name);

        curves[ci]->setPen(color);

        curves[ci]->setVisible(visible);

        curves[ci]->setItemAttribute(QwtPlotItem::Legend, visible);

        // replot only updated widgets

        if (isMulti)

        {

            plotWidgets[ci]->updateSymbols(); // required for color change

            plotWidgets[ci]->updateLegend(curves[ci]);

            plotWidgets[ci]->setVisible(visible);

            if (visible)

            {

                plotWidgets[ci]->replot();

            }

        }

    }

    // replot single widget

    if (!isMulti)

    {

        plotWidgets[0]->updateSymbols();

        plotWidgets[0]->updateLegend();

        replot();

    }

}

void PlotManager::setMulti(bool enabled)

{

    if (enabled == isMulti) return;

    isMulti = enabled;

    // detach all curves

    for (auto curve : curves)

    {

        curve->detach();

    }

    // remove all widgets

    while (plotWidgets.size())

    {

        delete plotWidgets.takeLast();

    }

    // setup new layout

    setupLayout(isMulti);

    if (isMulti)

    {

        // add new widgets and attach

        for (auto curve : curves)

        {

        }

    }

    else

    {

        // add a single widget

        auto plot = addPlotWidget();

        // attach all curves

        for (auto curve : curves)

        {

            curve->attach(plot);

        }

    }

}

void PlotManager::setupLayout(bool multiPlot)

{

    // delete previous layout if it exists

    if (_plotArea->layout() != 0)

    {

        delete _plotArea->layout();

    }

    if (multiPlot)

    {

        // setup a scroll area

        scrollArea = new QScrollArea();

        auto scrolledPlotArea = new QWidget(scrollArea);

        scrollArea->setWidget(scrolledPlotArea);

        scrollArea->setWidgetResizable(true);

        _plotArea->setLayout(new QVBoxLayout());

        _plotArea->layout()->addWidget(scrollArea);

        _plotArea->layout()->setContentsMargins(0,0,0,0);

        layout = new QVBoxLayout(scrolledPlotArea);

    }

public slots:

    /// Enable/Disable multiple plot display

    void setMulti(bool enabled);

    /// Update all plot widgets

    void replot();

    /// Enable display of a "DEMO" label on each plot

    void showDemoIndicator(bool show = true);

    /// Set the Y axis

    void setYAxis(bool autoScaled, double yMin = 0, double yMax = 1);

    /// Set the X axis

    void setXAxis(bool asIndex, double xMin = 0 , double xMax = 1);

    /// Display an animation for snapshot

    void flashSnapshotOverlay();

    /// Should be called to update zoom base

    void setNumOfSamples(unsigned value);

private:

    bool isMulti;

    QWidget* _plotArea;

    QVBoxLayout* layout; ///< layout of the `plotArea`

    QScrollArea* scrollArea;

    QList<QwtPlotCurve*> curves;

    QList<Plot*> plotWidgets;

    ChannelInfoModel* _infoModel;

    bool isDemoShown;

    bool _autoScaled;

    double _yMin;

    double _yMax;

    bool _xAxisAsIndex;

    double _xMin;

    double _xMax;

    unsigned _numOfSamples;

    Plot::ShowSymbols showSymbols;

    // menu actions

    QAction showGridAction;

    QAction showMinorGridAction;

    QAction unzoomAction;

    QAction darkBackgroundAction;

    QAction showLegendAction;

    QAction showMultiAction;

    QAction setSymbolsAction;

    QMenu setSymbolsMenu;

    QAction* setSymbolsAutoAct;

    QAction* setSymbolsShowAct;

    QAction* setSymbolsHideAct;

    void setupLayout(bool multiPlot);

    /// Inserts a new plot widget to the current layout.

    Plot* addPlotWidget();

    /// Returns the plot widget that given curve is attached to

    Plot* plotWidget(unsigned curveIndex);

    /// Common part of overloaded `addCurve` functions

    void _addCurve(QwtPlotCurve* curve);

    void setSymbols(Plot::ShowSymbols shown);

private slots:

    void showGrid(bool show = true);

    void showMinorGrid(bool show = true);

    void showLegend(bool show = true);

    void unzoom();

    void darkBackground(bool enabled = true);

    void onChannelInfoChanged(const QModelIndex & topLeft,

                              const QModelIndex & bottomRight,

                              const QVector<int> & roles = QVector<int> ());

};

#endif // PLOTMANAGER_H