PlotCurve.h

Go to the documentation of this file.

#ifndef PLOT_CURVE_H
#define PLOT_CURVE_H
#include <QDebug>

#include <qwt_plot.h>

#include <QAction>
#include <QWidget>
#include <QPushButton>
#include <QMenu>

Q_DECLARE_METATYPE(QwtPlot::Axis)
class AxisButton : public QPushButton {
    Q_OBJECT
public:
    AxisButton(QWidget *parent = 0) : QPushButton(parent),
        yLeftAxisAction("Left Y-axis", this), yRightAxisAction("Right Y-axis", this)
    {
        yLeftAxisAction.setData(QVariant::fromValue(QwtPlot::yLeft));
        yRightAxisAction.setData(QVariant::fromValue(QwtPlot::yRight));
        setFlat(true);
        m.addAction(&yLeftAxisAction);
        m.addAction(&yRightAxisAction);
        setMenu(&m);

        connect(&m, SIGNAL(triggered(QAction*)),
                this, SLOT(axisSelected(QAction*)));
        axisSelected(&yLeftAxisAction);
    }

    void setCurrentAxis(QwtPlot::Axis axis){
        if (axis == axisOfAction(&yLeftAxisAction)) {
            axisSelected(&yLeftAxisAction);
        } else if (axis == axisOfAction(&yRightAxisAction)) {
            axisOfAction(&yRightAxisAction);
        } else {
            return;
        }
    }

private:
    QwtPlot::Axis axisOfAction(QAction *action) const {
        return action->data().value<QwtPlot::Axis>();
    }

private slots:
    void axisSelected(QAction *action){
        if (text() == action->text()) return;

        setText(action->text());
        emit axisChanged(axisOfAction(action));
    }
signals:
    void axisChanged(QwtPlot::Axis axis);

private:
    QMenu m;

    QAction yLeftAxisAction;
    QAction yRightAxisAction;
};

#include <qwt_legend_item.h>
#include "qtcolorpicker.h"

#include <QVBoxLayout>
#include <QLabel>
#include <QPalette>
class LegendItem : public QFrame {
    Q_OBJECT
public:
    LegendItem(QWidget *parent = 0) : QFrame (parent){
        setFrameStyle(QFrame::Panel | QFrame::Raised);
        label.setTextFormat(Qt::PlainText);
        label.setContentsMargins(6,0,0,0);
        cPick.setFlat(true);
        cPick.setStandardColors();
        cPick.setColorDialogEnabled(true);
        setup();

        // connections
        connect(&cPick, SIGNAL(colorChanged(QColor)),
                this, SLOT(slot_colorChanged(QColor)));
        connect(&m_axisButton, SIGNAL(axisChanged(QwtPlot::Axis)),
                this, SIGNAL(axisChanged(QwtPlot::Axis)));
    }

    void setText(QString title){
        label.setText(title);
    }

    QColor currentColor() const {
        return cPick.currentColor();
    }

    void setCurrentColor(QColor color){
        cPick.setCurrentColor(color);
    }

    QColor color(int index) const {
        return cPick.color(index);
    }

    void setCurrentAxis(QwtPlot::Axis axis){
        m_axisButton.setCurrentAxis(axis);
    }

    QVBoxLayout *layout(){
        Q_ASSERT(dynamic_cast<QVBoxLayout*>(QFrame::layout()));
        return dynamic_cast<QVBoxLayout*>(QFrame::layout());
    }

private:
    void setup(){
        setLayout(new QVBoxLayout);
        layout()->setSpacing(0);
        layout()->setContentsMargins(0,0,0,0);

        QHBoxLayout *h = new QHBoxLayout();
        h->addWidget(&label);
        h->addWidget(&cPick);

        layout()->addLayout(h);
        layout()->addWidget(&m_axisButton);
    }

private slots:
    void slot_colorChanged(const QColor &color){
        QPalette palette;
        palette.setColor(QPalette::ButtonText, color);
        palette.setColor(QPalette::WindowText, color);
        setPalette(palette);
        emit colorChanged(color);
    }

signals:
    void colorChanged(const QColor &color);
    void axisChanged(QwtPlot::Axis);

private:
    QLabel label;
    QtColorPicker cPick;
    AxisButton m_axisButton;
};

#include <qwt_plot_curve.h>
#include "cachedplotcurvedatamodel.h"

#include <QPen>
#include <QObject>
class Plot;

class PlotCurve : public QObject, public QwtPlotCurve
{
    Q_OBJECT
  //=====================================================================================
  // Constructors and destructor
  //=====================================================================================
  public:
    PlotCurve();
    PlotCurve( const QwtText &title );
    PlotCurve( const QString &title );
    ~PlotCurve();

    QWidget * legendItem() const {
        return new LegendItem();
    }

    void updateLegend(QwtLegend *legend) const {
        if (!legend) return;
        Q_ASSERT(testItemAttribute(QwtPlotItem::Legend));

        LegendItem *lgdItem = dynamic_cast<LegendItem*>(legend->find(this));
        //qDebug() << "Found: " << lgdItem << this << legend->legendItems();

        //als er geen LegendItem in de Legend steekt, maken we er nu een, met alle connecties etc.
        if (!lgdItem){
            lgdItem = dynamic_cast<LegendItem*>(legendItem());
            Q_ASSERT(lgdItem);

            // Setup connections
            // Qt::QueuedConnection is nodig, want indien het niey Queued is,
            // dan wordt deze functie onmiddelijk geëxecuteerd na lgdItem->setCurrentColor,
            // wat resulteert in dubbele legenditem omdat de vorige nog niet geinsert was.
            connect(lgdItem, SIGNAL(colorChanged(QColor)),
                    this, SLOT(setColor(QColor)), Qt::QueuedConnection);
            connect(lgdItem, SIGNAL(axisChanged(QwtPlot::Axis)),
                    this, SLOT(setAxis(QwtPlot::Axis)), Qt::QueuedConnection);
            
            //setup LegendItem color
            QList<QColor> usedColors;
            foreach(QWidget *wItem, legend->legendItems()){
                LegendItem *item = dynamic_cast<LegendItem*>(wItem);
                Q_ASSERT(item);
                usedColors << item->currentColor();
            }

            Q_ASSERT(lgdItem->color(0).isValid());
            lgdItem->setCurrentColor(lgdItem->color(0));

            int i = 0;
            while(lgdItem->color(i).isValid()){
                if (usedColors.contains(lgdItem->color(i))) {
                    i++;
                    continue;
                } else {
                    lgdItem->setCurrentColor(lgdItem->color(i));
                    break;
                }
            } //by now, a color should be set, or the default used.

            // insert into Legend
            legend->insert(this, lgdItem);
            //qDebug() << "Inserted: " << lgdItem << this << legend->legendItems();
        }

        Q_ASSERT(lgdItem);
        Q_ASSERT(lgdItem->updatesEnabled());

        lgdItem->setText(title().text());
    }

  //=====================================================================================
  // Signals
  //=====================================================================================
signals:
    void signal_curveAttached(PlotCurve *curve);

  //=====================================================================================
  // Slots
  //=====================================================================================
public slots:
    void itemChanged(){
        QwtPlotCurve::itemChanged();
    }

    void setColor(QColor color){
        QPen penCopy(pen());
        penCopy.setColor(color);
        setPen(penCopy);
    }

    void setAxis(QwtPlot::Axis axis){
        if (axis == QwtPlot::xTop || axis == QwtPlot::xBottom){
            setXAxis(axis);
            return;
        }
        if (axis == QwtPlot::yLeft || axis == QwtPlot::yRight){
            setYAxis(axis);
            return;
        }
        Q_ASSERT(false);
    }

  //=====================================================================================
  // Methods
  //=====================================================================================
  public:
    void attach( Plot* plot );
    int closestPoint( const QPoint& pos, double* dist ) const;
    void setData(const CachedPlotCurveDataModel &dataModel){
        qDebug() << "Setting data to plot: " << this;
        QwtPlotCurve::setData(dataModel);
        CachedPlotCurveDataModel *model = dynamic_cast<CachedPlotCurveDataModel*>(&data());
        Q_ASSERT(model);
        connect(model, SIGNAL(signal_dataChanged()), this, SLOT(itemChanged()));
    }

  protected:
    // Code just for monitoring
    //virtual void drawCurve( QPainter* p, int style, const QwtScaleMap& xMap,
    //                        const QwtScaleMap& yMap, int from, int to ) const;

  //=====================================================================================
  // Members
  //=====================================================================================
};

#endif