qgsfeatureexpressionvaluesgatherer.h

Go to the documentation of this file.

/***************************************************************************
  qgsfeatureexpressionvaluesgatherer - QgsFeatureExpressionValuesGatherer
 ---------------------
 begin                : 10.3.2017
 copyright            : (C) 2017 by Matthias Kuhn
 email                : [email protected]
 ***************************************************************************
 *                                                                         *
 *   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 QGSFEATUREEXPRESSIONVALUESGATHERER_H
#define QGSFEATUREEXPRESSIONVALUESGATHERER_H
 
#include <QThread>
#include <QMutex>
#include "qgsapplication.h"
#include "qgslogger.h"
#include "qgsvectorlayer.h"
#include "qgsvectorlayerfeatureiterator.h"
 
#define SIP_NO_FILE
 
// just internal guff - definitely not for exposing to public API!
 
class QgsFeatureExpressionValuesGatherer: public QThread
{
    Q_OBJECT
 
  public:
 
    QgsFeatureExpressionValuesGatherer( QgsVectorLayer *layer,
                                        const QString &displayExpression = QString(),
                                        const QgsFeatureRequest &request = QgsFeatureRequest(),
                                        const QStringList &identifierFields = QStringList() )
      : mSource( new QgsVectorLayerFeatureSource( layer ) )
      , mDisplayExpression( displayExpression.isEmpty() ? layer->displayExpression() : displayExpression )
      , mRequest( request )
      , mIdentifierFields( identifierFields )
    {
    }
 
    struct Entry
    {
      Entry() = default;
 
      Entry( const QVariantList &_identifierFields, const QString &_value, const QgsFeature &_feature )
        : identifierFields( _identifierFields )
        , featureId( _feature.isValid() ? _feature.id() : FID_NULL )
        , value( _value )
        , feature( _feature )
      {}
 
      Entry( const QgsFeatureId &_featureId, const QString &_value, const QgsVectorLayer *layer )
        : featureId( _featureId )
        , value( _value )
        , feature( QgsFeature( layer->fields() ) )
      {}
 
      QVariantList identifierFields;
      QgsFeatureId featureId;
      QString value;
      QgsFeature feature;
 
      bool operator()( const Entry &lhs, const Entry &rhs ) const;
    };
 
    static Entry nullEntry( QgsVectorLayer *layer )
    {
      return Entry( QVariantList(), QgsApplication::nullRepresentation(), QgsFeature( layer->fields() ) );
    }
 
    void run() override
    {
      mWasCanceled = false;
 
      mIterator = mSource->getFeatures( mRequest );
 
      mDisplayExpression.prepare( &mExpressionContext );
 
      QgsFeature feature;
      QList<int> attributeIndexes;
      for ( const QString &fieldName : qgis::as_const( mIdentifierFields ) )
        attributeIndexes << mSource->fields().indexOf( fieldName );
 
      while ( mIterator.nextFeature( feature ) )
      {
        mExpressionContext.setFeature( feature );
        QVariantList attributes;
        for ( const int idx : attributeIndexes )
          attributes << feature.attribute( idx );
 
        const QString expressionValue = mDisplayExpression.evaluate( &mExpressionContext ).toString();
 
        mEntries.append( Entry( attributes, expressionValue, feature ) );
 
        QMutexLocker locker( &mCancelMutex );
        if ( mWasCanceled )
          return;
      }
    }
 
    void stop()
    {
      QMutexLocker locker( &mCancelMutex );
      mWasCanceled = true;
    }
 
    bool wasCanceled() const
    {
      QMutexLocker locker( &mCancelMutex );
      return mWasCanceled;
    }
 
    QVector<Entry> entries() const
    {
      return mEntries;
    }
 
    QgsFeatureRequest request() const
    {
      return mRequest;
    }
 
    QVariant data() const
    {
      return mData;
    }
 
    void setData( const QVariant &data )
    {
      mData = data;
    }
 
  protected:
    QVector<Entry> mEntries;
 
  private:
    std::unique_ptr<QgsVectorLayerFeatureSource> mSource;
    QgsExpression mDisplayExpression;
    QgsExpressionContext mExpressionContext;
    QgsFeatureRequest mRequest;
    QgsFeatureIterator mIterator;
    bool mWasCanceled = false;
    mutable QMutex mCancelMutex;
    QStringList mIdentifierFields;
    QVariant mData;
};
 
 
 
#endif // QGSFEATUREEXPRESSIONVALUESGATHERER_H