qgsvectorlayerfeaturecounter.h

Go to the documentation of this file.

/***************************************************************************
    qgsvectorlayerfeaturecounter.h
    ---------------------
    begin                : May 2017
    copyright            : (C) 2017 by Matthias Kuhn
    email                : matthias at opengis 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 QGSVECTORLAYERFEATURECOUNTER_H
#define QGSVECTORLAYERFEATURECOUNTER_H
 
#include "qgsfeatureid.h"
#include "qgsrenderer.h"
#include "qgstaskmanager.h"
#include "qgsvectorlayerfeatureiterator.h"

class CORE_EXPORT QgsVectorLayerFeatureCounter : public QgsTask
{
    Q_OBJECT
 
  public:
    QgsVectorLayerFeatureCounter( QgsVectorLayer *layer, const QgsExpressionContext &context = QgsExpressionContext(), bool storeSymbolFids = false );
    ~QgsVectorLayerFeatureCounter() override;

    bool run() override;
 
    void cancel() override;

    QHash<QString, long long> symbolFeatureCountMap() const SIP_SKIP;

    long long featureCount( const QString &legendKey ) const;

    QHash<QString, QgsFeatureIds> symbolFeatureIdMap() const SIP_SKIP;

    QgsFeatureIds featureIds( const QString &symbolkey ) const;
 
  signals:

    void symbolsCounted();
 
  private:
    std::unique_ptr<QgsVectorLayerFeatureSource> mSource;
    std::unique_ptr<QgsFeatureRenderer> mRenderer;
    QgsExpressionContext mExpressionContext;
    QHash<QString, long long> mSymbolFeatureCountMap;
    QHash<QString, QgsFeatureIds> mSymbolFeatureIdMap;
    std::unique_ptr< QgsFeedback > mFeedback;
    bool mWithFids = false;
    long mFeatureCount = 0;
};
 
#endif // QGSVECTORLAYERFEATURECOUNTER_H