barchartplotter.h

#ifndef BARCHARTPLOTTER_H
#define BARCHARTPLOTTER_H


#include "plotterbase.h"


namespace QSint
{


class BarChartPlotter : public PlotterBase
{
    Q_OBJECT
public:
    explicit BarChartPlotter(QWidget *parent = 0);


    enum BarChartType
    {
        Stacked,
        Columns,
        Trend
    };

    void setBarType(BarChartType type);
    inline BarChartType barType() const { return m_type; }

    void setBarSize(int min, int max = INT_MAX);
    inline int barSizeMin() const { return m_barsize_min; }
    inline int barSizeMax() const { return m_barsize_max; }

    void setBarScale(double scale);
    inline double barScale() const { return m_scale; }

    void setBarOpacity(double value);
    inline double barOpacity() const { return m_opacity; }


    void setZeroLinePen(const QPen &pen);
    inline const QPen& zeroLinePen() const { return m_zeroLinePen; }

protected:
    virtual void drawContent(QPainter &p);

    virtual void drawSegment(QPainter &p, QRect rect,
                              const QModelIndex &index, double value,
                              bool isHighlighted) const;

    virtual void drawValue(QPainter &p, QRect rect,
                              const QModelIndex &index, double value,
                              bool isHighlighted) const;

    virtual void drawValueText(QPainter &p,
                               const QRect &rect,
                               int flags,
                               bool isHighlighted,
                               const QModelIndex &index,
                               const QString &text) const;

    virtual QRect drawHighlightedValueFrame(QPainter &p,
                                           const QRect &rect,
                                           const QRect &textRect) const;

    int m_barsize_min, m_barsize_max;
    double m_scale;
    double m_opacity;
    BarChartType m_type;

    QPen m_zeroLinePen;

protected:
    class BarPainter
    {
    protected:
        static void drawBarItem(QPainter &p, QRect rect,
                                 const QPen &pen, const QBrush &brush,
                                 const QModelIndex &index,
                                 double value);

        static void drawValueText(QPainter &p, QRect rect, int flags,
                                 const QPen &pen, const QBrush &brush,
                                 const QModelIndex &index,
                                 double value);
    };

    class StackedBarPainter: public BarPainter
    {
    public:
        static void draw(BarChartPlotter *plotter,
                         QPainter &p,
                         int count,
                         int row_count,
                         int p_start,
                         int p_offs,
                         int bar_size);
    };

    class ColumnBarPainter: public BarPainter
    {
    public:
        static void draw(BarChartPlotter *plotter,
                         QPainter &p,
                         int count,
                         int row_count,
                         int p_start,
                         int p_offs,
                         int bar_size);
    };

    class TrendPainter: public BarPainter
    {
    public:
        static void draw(BarChartPlotter *plotter,
                         QPainter &p,
                         int count,
                         int row_count,
                         int p_start,
                         int p_offs,
                         int bar_size);
    };
};


} // namespace

#endif // BARCHARTPLOTTER_H