qgsdiagramrendererv2.h

Go to the documentation of this file.

/***************************************************************************
    qgsdiagramrendererv2.h
    ---------------------
    begin                : March 2011
    copyright            : (C) 2011 by Marco Hugentobler
    email                : marco dot hugentobler at sourcepole dot ch
 ***************************************************************************
 *                                                                         *
 *   This program 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 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
#ifndef QGSDIAGRAMRENDERERV2_H
#define QGSDIAGRAMRENDERERV2_H

#include <QColor>
#include <QFont>
#include <QList>
#include <QPointF>
#include <QSizeF>
#include <QDomDocument>

#include "qgsfeature.h"
#include "qgsexpressioncontext.h"
#include "qgssymbollayerv2.h"

class QgsDiagram;
class QgsDiagramRendererV2;
class QgsFeature;
class QgsRenderContext;
class QDomElement;
class QgsCoordinateTransform;
class QgsMapToPixel;
class QgsVectorLayer;
class QgsLayerTreeModelLegendNode;
class QgsLayerTreeLayer;

namespace pal { class Layer; }

class CORE_EXPORT QgsDiagramLayerSettings
{
  public:

    //avoid inclusion of QgsPalLabeling
    enum Placement
    {
      AroundPoint = 0, // Point / Polygon
      OverPoint, // Point / Polygon
      Line, // Line / Polygon
      Curved, // Line
      Horizontal, // Polygon
      Free // Polygon
    };

    enum LinePlacementFlags
    {
      OnLine    = 1,
      AboveLine = 2,
      BelowLine = 4,
      MapOrientation = 8
    };

    QgsDiagramLayerSettings();

    QgsDiagramLayerSettings( const QgsDiagramLayerSettings& rh );

    QgsDiagramLayerSettings& operator=( const QgsDiagramLayerSettings& rh );

    ~QgsDiagramLayerSettings();

    //TODO QGIS 3.0 - rename getter to placement()
    Placement getPlacement() const { return placement; }

    void setPlacement( Placement value ) { placement = value; }

    //TODO QGIS 3.0 - make private, rename to mPlacement
    Placement placement;

    unsigned int linePlacementFlags() const { return placementFlags; }

    void setLinePlacementFlags( unsigned int flags ) { placementFlags = flags; }

    // TODO QGIS 3.0 - make private, rename to mPlacementFlags, use QFlags
    unsigned int placementFlags;

    //TODO QGIS 3.0 - rename getter to priority()
    int getPriority() const { return priority; }

    void setPriority( int value ) { priority = value; }

    // TODO QGIS 3.0 - make private, rename to mPriority
    int priority;

    //TODO QGIS 3.0 - rename getter to zIndex()
    double getZIndex() const { return zIndex; }

    void setZIndex( double index ) { zIndex = index; }

    // TODO QGIS 3.0 - rename to mZIndex, make private
    double zIndex;


    bool isObstacle() const { return obstacle; }

    void setIsObstacle( bool isObstacle ) { obstacle = isObstacle; }

    // TODO QGIS 3.0 - rename to mObstacle, make private
    bool obstacle;

    double distance() const { return dist; }

    void setDistance( double distance ) { dist = distance; }

    // TODO QGIS 3.0 - make private, rename to mDistance
    double dist;

    // TODO QGIS 3.0 - rename to renderer()
    QgsDiagramRendererV2* getRenderer() { return renderer; }

    // TODO QGIS 3.0 - rename to renderer()
    const QgsDiagramRendererV2* getRenderer() const { return renderer; }

    void setRenderer( QgsDiagramRendererV2* diagramRenderer );

    // TODO QGIS 3.0 - make private, rename to mRenderer
    QgsDiagramRendererV2* renderer;

    QgsCoordinateTransform* coordinateTransform() { return ct; }

    const QgsCoordinateTransform* coordinateTransform() const { return ct; }

    void setCoordinateTransform( QgsCoordinateTransform* transform );

    // TODO QGIS 3.0 - make private, rename to mCt
    QgsCoordinateTransform* ct;

    Q_DECL_DEPRECATED const QgsMapToPixel* xform;

    Q_DECL_DEPRECATED QgsFields fields;

    int xPosColumn;

    int yPosColumn;

    int showColumn;

    bool showAllDiagrams() const { return showAll; }

    void setShowAllDiagrams( bool showAllDiagrams ) { showAll = showAllDiagrams; }

    // TODO QGIS 3.0 - make private, rename to mShowAll
    bool showAll;

    void readXML( const QDomElement& elem, const QgsVectorLayer* layer );
    void writeXML( QDomElement& layerElem, QDomDocument& doc, const QgsVectorLayer* layer ) const;

    //TODO QGIS 3.0 - remove need for fields parameter
    QSet< QString > referencedFields( const QgsExpressionContext& context = QgsExpressionContext(), const QgsFields& fields = QgsFields() ) const;

};

class CORE_EXPORT QgsDiagramSettings
{
  public:

    enum SizeType
    {
      MM,
      MapUnits
    };

    enum LabelPlacementMethod
    {
      Height,
      XHeight
    };

    enum DiagramOrientation
    {
      Up,
      Down,
      Left,
      Right
    };

    QgsDiagramSettings()
        : enabled( true )
        , sizeType( QgsSymbolV2::MM )
        , lineSizeUnit( QgsSymbolV2::MM )
        , penWidth( 0.0 )
        , labelPlacementMethod( QgsDiagramSettings::Height )
        , diagramOrientation( QgsDiagramSettings::Up )
        , barWidth( 5.0 )
        , transparency( 0 )
        , scaleByArea( true )
        , angleOffset( 90 * 16 ) //top
        , scaleBasedVisibility( false )
        , minScaleDenominator( -1 )
        , maxScaleDenominator( -1 )
        , minimumSize( 0.0 )
    {}
    bool enabled;
    QFont font;
    QList< QColor > categoryColors;
    QList< QString > categoryAttributes;
    QList< QString > categoryLabels;
    QSizeF size; //size

    QgsSymbolV2::OutputUnit sizeType;

    QgsMapUnitScale sizeScale;

    QgsSymbolV2::OutputUnit lineSizeUnit;

    QgsMapUnitScale lineSizeScale;

    QColor backgroundColor;
    QColor penColor;
    double penWidth;
    LabelPlacementMethod labelPlacementMethod;
    DiagramOrientation diagramOrientation;
    double barWidth;
    int transparency; // 0 - 100
    bool scaleByArea;
    int angleOffset;

    bool scaleBasedVisibility;
    //scale range (-1 if no lower / upper bound )
    double minScaleDenominator;
    double maxScaleDenominator;

    double minimumSize;

    void readXML( const QDomElement& elem, const QgsVectorLayer* layer );
    void writeXML( QDomElement& rendererElem, QDomDocument& doc, const QgsVectorLayer* layer ) const;

    QList< QgsLayerTreeModelLegendNode* > legendItems( QgsLayerTreeLayer* nodeLayer ) const;

};

class CORE_EXPORT QgsDiagramInterpolationSettings
{
  public:
    QSizeF lowerSize;
    QSizeF upperSize;
    double lowerValue;
    double upperValue;

    //TODO QGIS 3.0 - don't store index, store field name
    int classificationAttribute;

    QString classificationAttributeExpression;
    bool classificationAttributeIsExpression;
};


class CORE_EXPORT QgsDiagramRendererV2
{
  public:

    QgsDiagramRendererV2();
    virtual ~QgsDiagramRendererV2();

    virtual QgsDiagramRendererV2* clone() const = 0;

    virtual QSizeF sizeMapUnits( const QgsFeature& feature, const QgsRenderContext& c ) const;

    virtual QString rendererName() const = 0;

    virtual QList<QString> diagramAttributes() const = 0;

    //TODO QGIS 3.0 - remove need for fields parameter
    virtual QSet< QString > referencedFields( const QgsExpressionContext& context = QgsExpressionContext(), const QgsFields& fields = QgsFields() ) const;

    void renderDiagram( const QgsFeature& feature, QgsRenderContext& c, QPointF pos ) const;

    void setDiagram( QgsDiagram* d );
    QgsDiagram* diagram() const { return mDiagram; }

    virtual QList<QgsDiagramSettings> diagramSettings() const = 0;

    virtual void readXML( const QDomElement& elem, const QgsVectorLayer* layer ) = 0;
    virtual void writeXML( QDomElement& layerElem, QDomDocument& doc, const QgsVectorLayer* layer ) const = 0;

    virtual QList< QgsLayerTreeModelLegendNode* > legendItems( QgsLayerTreeLayer* nodeLayer ) const;

    bool attributeLegend() const { return mShowAttributeLegend; }

    void setAttributeLegend( bool enabled ) { mShowAttributeLegend = enabled; }

    bool sizeLegend() const { return mShowSizeLegend; }

    void setSizeLegend( bool enabled ) { mShowSizeLegend = enabled; }

    QgsMarkerSymbolV2* sizeLegendSymbol() const { return mSizeLegendSymbol.data(); }

    void setSizeLegendSymbol( QgsMarkerSymbolV2* symbol ) { mSizeLegendSymbol.reset( symbol ); }

  protected:
    QgsDiagramRendererV2( const QgsDiagramRendererV2& other );
    QgsDiagramRendererV2& operator=( const QgsDiagramRendererV2& other );

    virtual bool diagramSettings( const QgsFeature &feature, const QgsRenderContext& c, QgsDiagramSettings& s ) const = 0;

    virtual QSizeF diagramSize( const QgsFeature& features, const QgsRenderContext& c ) const = 0;

    void convertSizeToMapUnits( QSizeF& size, const QgsRenderContext& context ) const;

    static int dpiPaintDevice( const QPainter* );

    //read / write diagram
    void _readXML( const QDomElement& elem, const QgsVectorLayer* layer );
    void _writeXML( QDomElement& rendererElem, QDomDocument& doc, const QgsVectorLayer* layer ) const;

    QgsDiagram* mDiagram;

    bool mShowAttributeLegend;

    bool mShowSizeLegend;

    QScopedPointer< QgsMarkerSymbolV2 > mSizeLegendSymbol;
};

class CORE_EXPORT QgsSingleCategoryDiagramRenderer : public QgsDiagramRendererV2
{
  public:
    QgsSingleCategoryDiagramRenderer();
    ~QgsSingleCategoryDiagramRenderer();

    QgsSingleCategoryDiagramRenderer* clone() const override;

    QString rendererName() const override { return "SingleCategory"; }

    QList<QString> diagramAttributes() const override { return mSettings.categoryAttributes; }

    void setDiagramSettings( const QgsDiagramSettings& s ) { mSettings = s; }

    QList<QgsDiagramSettings> diagramSettings() const override;

    void readXML( const QDomElement& elem, const QgsVectorLayer* layer ) override;
    void writeXML( QDomElement& layerElem, QDomDocument& doc, const QgsVectorLayer* layer ) const override;

    QList< QgsLayerTreeModelLegendNode* > legendItems( QgsLayerTreeLayer* nodeLayer ) const override;

  protected:
    bool diagramSettings( const QgsFeature &feature, const QgsRenderContext& c, QgsDiagramSettings& s ) const override;

    QSizeF diagramSize( const QgsFeature&, const QgsRenderContext& c ) const override;

  private:
    QgsDiagramSettings mSettings;
};

class CORE_EXPORT QgsLinearlyInterpolatedDiagramRenderer : public QgsDiagramRendererV2
{
  public:
    QgsLinearlyInterpolatedDiagramRenderer();
    ~QgsLinearlyInterpolatedDiagramRenderer();

    QgsLinearlyInterpolatedDiagramRenderer* clone() const override;

    QList<QgsDiagramSettings> diagramSettings() const override;

    void setDiagramSettings( const QgsDiagramSettings& s ) { mSettings = s; }

    QList<QString> diagramAttributes() const override;

    virtual QSet< QString > referencedFields( const QgsExpressionContext& context = QgsExpressionContext(), const QgsFields& fields = QgsFields() ) const override;

    QString rendererName() const override { return "LinearlyInterpolated"; }

    void setLowerValue( double val ) { mInterpolationSettings.lowerValue = val; }
    double lowerValue() const { return mInterpolationSettings.lowerValue; }

    void setUpperValue( double val ) { mInterpolationSettings.upperValue = val; }
    double upperValue() const { return mInterpolationSettings.upperValue; }

    void setLowerSize( QSizeF s ) { mInterpolationSettings.lowerSize = s; }
    QSizeF lowerSize() const { return mInterpolationSettings.lowerSize; }

    void setUpperSize( QSizeF s ) { mInterpolationSettings.upperSize = s; }
    QSizeF upperSize() const { return mInterpolationSettings.upperSize; }

    int classificationAttribute() const { return mInterpolationSettings.classificationAttribute; }
    void setClassificationAttribute( int index ) { mInterpolationSettings.classificationAttribute = index; }

    QString classificationAttributeExpression() const { return mInterpolationSettings.classificationAttributeExpression; }
    void setClassificationAttributeExpression( const QString& expression ) { mInterpolationSettings.classificationAttributeExpression = expression; }

    bool classificationAttributeIsExpression() const { return mInterpolationSettings.classificationAttributeIsExpression; }
    void setClassificationAttributeIsExpression( bool isExpression ) { mInterpolationSettings.classificationAttributeIsExpression = isExpression; }

    void readXML( const QDomElement& elem, const QgsVectorLayer* layer ) override;
    void writeXML( QDomElement& layerElem, QDomDocument& doc, const QgsVectorLayer* layer ) const override;

    QList< QgsLayerTreeModelLegendNode* > legendItems( QgsLayerTreeLayer* nodeLayer ) const override;

  protected:
    bool diagramSettings( const QgsFeature &feature, const QgsRenderContext& c, QgsDiagramSettings& s ) const override;

    QSizeF diagramSize( const QgsFeature&, const QgsRenderContext& c ) const override;

  private:
    QgsDiagramSettings mSettings;
    QgsDiagramInterpolationSettings mInterpolationSettings;
};

#endif // QGSDIAGRAMRENDERERV2_H