qgsvectorlayerdiagramprovider.h

Go to the documentation of this file.

/***************************************************************************
  qgsvectorlayerdiagramprovider.h
  --------------------------------------
  Date                 : September 2015
  Copyright            : (C) 2015 by Martin Dobias
  Email                : wonder dot sk at gmail dot com
 ***************************************************************************
 *                                                                         *
 *   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 QGSVECTORLAYERDIAGRAMPROVIDER_H
#define QGSVECTORLAYERDIAGRAMPROVIDER_H

#include "qgslabelingenginev2.h"
#include "qgslabelfeature.h"

class QgsDiagramLabelFeature : public QgsLabelFeature
{
  public:
    QgsDiagramLabelFeature( QgsFeatureId id, GEOSGeometry* geometry, QSizeF size )
        : QgsLabelFeature( id, geometry, size ) {}

    void setAttributes( const QgsAttributes& attrs ) { mAttributes = attrs; }
    const QgsAttributes& attributes() { return mAttributes; }

  protected:
    QgsAttributes mAttributes;
};


class QgsAbstractFeatureSource;


class CORE_EXPORT QgsVectorLayerDiagramProvider : public QgsAbstractLabelProvider
{
  public:

    explicit QgsVectorLayerDiagramProvider( QgsVectorLayer* layer, bool ownFeatureLoop = true );

    QgsVectorLayerDiagramProvider( const QgsDiagramLayerSettings* diagSettings,
                                   const QgsDiagramRendererV2* diagRenderer,
                                   const QString& layerId,
                                   const QgsFields& fields,
                                   const QgsCoordinateReferenceSystem& crs,
                                   QgsAbstractFeatureSource* source,
                                   bool ownsSource );

    ~QgsVectorLayerDiagramProvider();

    virtual QList<QgsLabelFeature*> labelFeatures( QgsRenderContext& context ) override;

    virtual void drawLabel( QgsRenderContext& context, pal::LabelPosition* label ) const override;

    // new virtual methods

    virtual bool prepare( const QgsRenderContext& context, QStringList& attributeNames );

    virtual void registerFeature( QgsFeature& feature, QgsRenderContext &context, QgsGeometry* obstacleGeometry = nullptr );

  protected:
    void init();
    QgsLabelFeature* registerDiagram( QgsFeature& feat, QgsRenderContext& context, QgsGeometry* obstacleGeometry = nullptr );

  protected:

    QgsDiagramLayerSettings mSettings;
    QgsDiagramRendererV2* mDiagRenderer;
    QString mLayerId;

    // these are needed only if using own renderer loop

    QgsFields mFields;
    QgsCoordinateReferenceSystem mLayerCrs;
    QgsAbstractFeatureSource* mSource;
    bool mOwnsSource;

    QList<QgsLabelFeature*> mFeatures;
};

#endif // QGSVECTORLAYERDIAGRAMPROVIDER_H