simfigure.h

Go to the documentation of this file.

#ifndef SIMFIGURE_H
#define SIMFIGURE_H

#include <QFrame>
#include <QList>
#include <QMap>
#include <QPen>
#include <QBrush>

//#include "simfigure_enums.h"

class QwtPlot;
class QwtPlotGrid;
class QwtPlotItem;
class QwtPlotShapeItem;
class QwtPlotCurve;
class QwtPlotLegendItem;
class QString;
class QwtPlotPicker;
class QwtPlotZoomer;


namespace Ui {
class SimFigure;
}


class SimFigure : public QFrame
{
    Q_OBJECT

public:

    enum class AxisType { Default, 
                          LogX,    
                          LogY,    
                          LogLog   
                        };

    enum class LineType { None,      
                          Solid,     
                          Dotted,    
                          Dashed,    
                          DashDotted 
                        };

    enum class Marker { None,            
                        Asterisk,        
                        Circle,          
                        Plus,            
                        Triangle,        
                        DownTriangle,    
                        RightTriangle,   
                        LeftTriangle,    
                        Box,             
                        Ex               
                      };

    enum class Location { Bottom,       
                          Top,          
                          Left,         
                          Right,        
                          TopLeft,      
                          TopRight,     
                          BottomLeft,   
                          BottomRight,  
                          North,        
                          South,        
                          East,         
                          West,         
                          NorthWest,    
                          NorthEast,    
                          SouthWest,    
                          SouthEast     
                        };

    enum class FileType {
        PNG, 
        BMP, 
        PDF, 
        PS,  
        SVG  
    };

    explicit SimFigure(QWidget *parent = nullptr);
    ~SimFigure();

    int plot(QVector<double> &, QVector<double> &, LineType lt = LineType::Solid, QColor col = Qt::red, Marker mk = Marker::None);
    int scatter(QVector<double> &, QVector<double> &, QColor col = Qt::blue, Marker mk = Marker::Circle);


    void clear(void);
    void cla(void);

    void grid(bool mayor=true, bool minor=true);

    void legend(QList<QString> labels, Location loc=Location::South);
    void moveLegend(Location loc);
    void showLegend(bool = true);
    bool legendVisible(void);

    void select(int);
    void clearSelection(void);

    SimFigure::AxisType axisType(void);
    void setAxisType( AxisType type);

    QString xLabel();
    QString yLabel();
    int labelFontSize();
    QString title();
    int titleFontSize();

    void setXLabel(QString lbl);
    void setYLabel(QString lbl);
    void setXLim(double xmin, double xmax);
    void setYLim(double ymin, double ymax);

    void setXlimits(double xmin, double xmax) {setXLim(xmin,xmax);};

    void setYlimits(double ymin, double ymax) {setYLim(ymin,ymax);};
    void setLabelFontSize(int);
    void setTitle(QString title);
    void setTitleFontSize(int);

    double minX() { return m_xmin; };
    double maxX() { return m_xmax; };
    double minY() { return m_ymin; };
    double maxY() { return m_ymax; };

    int      lineWidth(int ID);
    void     setLineWidth(int ID, int wd);

    double   lineWidthF(int ID);
    void     setLineWidthF(int ID, double wd);

    SimFigure::LineType lineStyle(int ID);
    void     setLineStyle(int ID, LineType lt=LineType::Solid, Marker mk=Marker::None);

    QColor   lineColor(int ID);
    void     setLineColor(int ID, QColor color);

    SimFigure::Marker marker(int ID);
    int      markerSize(int ID);
    void     setMarker(int ID, Marker mk, int size=10);

    void saveToFile(QString filename, SimFigure::FileType type=SimFigure::FileType::PNG, QSizeF size=QSizeF(300,200), int res=85);
    void exportToFile(QString filename, SimFigure::FileType type=SimFigure::FileType::PNG, QSizeF size=QSizeF(300,200), int res=85);

private slots:
    void axisTypeChanged(void);

public slots:
    void on_picker_activated (bool on);
    void on_picker_selected (const QPolygon &polygon);
    void on_picker_appended (const QPoint &pos);
    void on_picker_moved (const QPoint &pos);
    void on_picker_removed (const QPoint &pos);
    void on_picker_changed (const QPolygon &selection);
    void showAxisControls(bool show);
    void fit_data();

signals:
    void curve_selected(int ID);

protected:
    void select(QwtPlotItem *);
    void setLineStyle(QwtPlotCurve *, LineType lt);
    void setLineColor(QwtPlotCurve *, QColor color);
    void setMarker(QwtPlotCurve *curve, Marker mk, int size);
    QwtPlotItem* itemAt( const QPoint& pos ) const;
    void rescale(void);
    void refreshGrid(void);

private:
    Ui::SimFigure *ui;
    QwtPlot       *m_plot;
    QwtPlotGrid   *m_grid;
    QwtPlotPicker *m_picker;
    QwtPlotZoomer *m_zoomer = nullptr;
    QwtPlotLegendItem  *m_legend;
    QMap<QwtPlotCurve *, int> m_plotInvMap;
    QMap<QwtPlotItem *, int>  m_itemInvMap;

    QVector<QwtPlotCurve *> m_curves;

    AxisType m_axisType;
    double  m_xmin = 1.e20;
    double  m_xmax = 1.e-20;
    double  m_ymin = 1.e20;
    double  m_ymax = 1.e-20;

    double  m_data_xmin = 1.e20;
    double  m_data_xmax = 1.e-20;
    double  m_data_ymin = 1.e20;
    double  m_data_ymax = 1.e-20;

    struct SELECTION {
        QPen         pen;
        QBrush       brush;
        int          plotID = -1;
        QwtPlotItem *object = nullptr;
    } lastSelection;

    bool  m_showMajorGrid = true;
    bool  m_showMinorGrid = true;
};

#endif // SIMFIGURE_H