qgspointdistancerenderer.h

Go to the documentation of this file.

/***************************************************************************
                              qgspointdistancerenderer.cpp
                              ----------------------------
  begin                : January 26, 2010
  copyright            : (C) 2010 by Marco Hugentobler
  email                : marco at hugis dot net
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   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 QGSPOINTDISTANCERENDERER_H
#define QGSPOINTDISTANCERENDERER_H

#include "qgis_core.h"
#include "qgis.h"
#include "qgsrenderer.h"
#include <QFont>

class QgsSpatialIndex;

class CORE_EXPORT QgsPointDistanceRenderer: public QgsFeatureRenderer
{
  public:

    struct GroupedFeature
    {

        GroupedFeature( const QgsFeature &feature, QgsMarkerSymbol *symbol SIP_TRANSFER, bool isSelected, const QString &label = QString() )
          : feature( feature )
          , isSelected( isSelected )
          , label( label )
          , mSymbol( symbol )
        {}

        QgsFeature feature;

        QgsMarkerSymbol *symbol() const { return mSymbol.get(); }

        bool isSelected;

        QString label;

      private:
        std::shared_ptr< QgsMarkerSymbol > mSymbol;
    };

    typedef QList< QgsPointDistanceRenderer::GroupedFeature > ClusteredGroup;

    QgsPointDistanceRenderer( const QString &rendererName, const QString &labelAttributeName = QString() );

    void toSld( QDomDocument &doc, QDomElement &element, const QgsStringMap &props = QgsStringMap() ) const override;
    bool renderFeature( QgsFeature &feature, QgsRenderContext &context, int layer = -1, bool selected = false, bool drawVertexMarker = false ) override;
    QSet<QString> usedAttributes( const QgsRenderContext &context ) const override;
    QgsFeatureRenderer::Capabilities capabilities() override;
    QgsSymbolList symbols( QgsRenderContext &context ) override;
    QgsSymbol *symbolForFeature( QgsFeature &feature, QgsRenderContext &context ) override;
    QgsSymbol *originalSymbolForFeature( QgsFeature &feat, QgsRenderContext &context ) override;
    QgsSymbolList symbolsForFeature( QgsFeature &feat, QgsRenderContext &context ) override;
    QgsSymbolList originalSymbolsForFeature( QgsFeature &feat, QgsRenderContext &context ) override;
    QSet< QString > legendKeysForFeature( QgsFeature &feature, QgsRenderContext &context ) override;
    bool willRenderFeature( QgsFeature &feat, QgsRenderContext &context ) override;
    void startRender( QgsRenderContext &context, const QgsFields &fields ) override;
    void stopRender( QgsRenderContext &context ) override;
    QgsLegendSymbolList legendSymbolItems() const override;
    void setEmbeddedRenderer( QgsFeatureRenderer *r SIP_TRANSFER ) override;
    const QgsFeatureRenderer *embeddedRenderer() const override;
    void setLegendSymbolItem( const QString &key, QgsSymbol *symbol SIP_TRANSFER ) override;
    bool legendSymbolItemsCheckable() const override;
    bool legendSymbolItemChecked( const QString &key ) override;
    void checkLegendSymbolItem( const QString &key, bool state ) override;
    QString filter( const QgsFields &fields = QgsFields() ) override;

    void setLabelAttributeName( const QString &name ) { mLabelAttributeName = name; }

    QString labelAttributeName() const { return mLabelAttributeName; }

    void setLabelFont( const QFont &font ) { mLabelFont = font; }

    QFont labelFont() const { return mLabelFont;}

    void setMinimumLabelScale( double scale ) { mMinLabelScale = scale; }

    double minimumLabelScale() const { return mMinLabelScale; }

    void setLabelColor( const QColor &color ) { mLabelColor = color;}

    QColor labelColor() const { return mLabelColor; }

    void setTolerance( double distance ) { mTolerance = distance; }

    double tolerance() const { return mTolerance; }

    void setToleranceUnit( QgsUnitTypes::RenderUnit unit ) { mToleranceUnit = unit; }

    QgsUnitTypes::RenderUnit toleranceUnit() const { return mToleranceUnit; }

    void setToleranceMapUnitScale( const QgsMapUnitScale &scale ) { mToleranceMapUnitScale = scale; }

    const QgsMapUnitScale &toleranceMapUnitScale() const { return mToleranceMapUnitScale; }

  protected:

    std::unique_ptr< QgsFeatureRenderer > mRenderer;

    QString mLabelAttributeName;

    int mLabelIndex;

    double mTolerance;
    QgsUnitTypes::RenderUnit mToleranceUnit;
    QgsMapUnitScale mToleranceMapUnitScale;

    QFont mLabelFont;
    QColor mLabelColor;
    bool mDrawLabels;
    double mMinLabelScale = 0;

    QList<ClusteredGroup> mClusteredGroups;

    QMap<QgsFeatureId, int> mGroupIndex;

    QMap<QgsFeatureId, QgsPointXY > mGroupLocations;

    QgsSpatialIndex *mSpatialIndex = nullptr;

    void drawLabels( QPointF centerPoint, QgsSymbolRenderContext &context, const QList<QPointF> &labelShifts, const ClusteredGroup &group );

  private:

    virtual void drawGroup( QPointF centerPoint, QgsRenderContext &context, const ClusteredGroup &group ) = 0 SIP_FORCE;

    QgsRectangle searchRect( const QgsPointXY &p, double distance ) const;

    void printGroupInfo() const;

    QString getLabel( const QgsFeature &feature ) const;

    void drawGroup( const ClusteredGroup &group, QgsRenderContext &context );

    QgsMarkerSymbol *firstSymbolForFeature( QgsFeature &feature, QgsRenderContext &context );

    QgsExpressionContextScope *createGroupScope( const ClusteredGroup &group ) const;

};

#endif // QGSPOINTDISTANCERENDERER_H