[FEATURE] Active layer feature locator filter

Searches for matching attributes in any field from the current active layer.
qgis · Jun 1, 2017 · 55ed6f9 · 55ed6f9
1 parent 5077e12
commit 55ed6f9
Show file tree

Hide file tree

Showing 5 changed files with 102 additions and 4 deletions.
diff --git a/src/app/locator/qgsinbuiltlocatorfilters.cpp b/src/app/locator/qgsinbuiltlocatorfilters.cpp
@@ -25,6 +25,7 @@
 #include "qgsmaplayermodel.h"
 #include "qgscomposition.h"
 #include "qgslayoutmanager.h"
+#include "qgsmapcanvas.h"
 #include <QToolButton>
 
 QgsLayerTreeLocatorFilter::QgsLayerTreeLocatorFilter( QObject *parent )
@@ -163,3 +164,80 @@ void QgsActionLocatorFilter::searchActions( const QString &string, QWidget *pare
     }
   }
 }
+
+QgsActiveLayerFeaturesLocatorFilter::QgsActiveLayerFeaturesLocatorFilter( QObject *parent )
+  : QgsLocatorFilter( parent )
+{
+  setUseWithoutPrefix( false );
+}
+
+void QgsActiveLayerFeaturesLocatorFilter::fetchResults( const QString &string, const QgsLocatorContext &, QgsFeedback *feedback )
+{
+  if ( string.length() < 3 )
+    return;
+
+  QgsVectorLayer *layer = qobject_cast< QgsVectorLayer *>( QgisApp::instance()->activeLayer() );
+  if ( !layer )
+    return;
+
+  int i = 0;
+  int found = 0;
+  QgsExpression dispExpression( layer->displayExpression() );
+  QgsExpressionContext context;
+  context.appendScopes( QgsExpressionContextUtils::globalProjectLayerScopes( layer ) );
+  dispExpression.prepare( &context );
+
+  Q_FOREACH ( const QgsField &field, layer->fields() )
+  {
+    if ( feedback->isCanceled() )
+      return;
+
+    QString exp = QStringLiteral( "%1 ILIKE '%%2%'" ).arg( QgsExpression::quotedColumnRef( field.name() ),
+                  string );
+    QgsFeatureRequest req;
+    req.setFlags( QgsFeatureRequest::NoGeometry );
+    QStringList attrs;
+    attrs << field.name();
+    attrs.append( dispExpression.referencedColumns().toList() );
+    req.setSubsetOfAttributes( attrs, layer->fields() );
+    req.setFilterExpression( exp );
+    req.setLimit( 30 );
+    QgsFeature f;
+    QgsFeatureIterator it = layer->getFeatures( req );
+    while ( it.nextFeature( f ) )
+    {
+      if ( feedback->isCanceled() )
+        return;
+
+      QgsLocatorResult result;
+      result.filter = this;
+
+      context.setFeature( f );
+      result.displayString = dispExpression.evaluate( &context ).toString();
+      if ( result.displayString.isEmpty() )
+        result.displayString = f.attribute( i ).toString();
+      else
+        result.description = f.attribute( i ).toString();
+
+      result.userData = f.id();
+      result.icon = QgsMapLayerModel::iconForLayer( layer );
+      result.score = static_cast< double >( string.length() ) / f.attribute( i ).toString().size();
+      emit resultFetched( result );
+
+      found++;
+      if ( found >= 30 )
+        return;
+    }
+    i++;
+  }
+}
+
+void QgsActiveLayerFeaturesLocatorFilter::triggerResult( const QgsLocatorResult &result )
+{
+  QgsFeatureId id = result.userData.toLongLong();
+  QgsVectorLayer *layer = qobject_cast< QgsVectorLayer *>( QgisApp::instance()->activeLayer() );
+  if ( !layer )
+    return;
+
+  QgisApp::instance()->mapCanvas()->zoomToFeatureIds( layer, QgsFeatureIds() << id );
+}
diff --git a/src/app/locator/qgsinbuiltlocatorfilters.h b/src/app/locator/qgsinbuiltlocatorfilters.h
@@ -77,6 +77,23 @@ class QgsActionLocatorFilter : public QgsLocatorFilter
 
 };
 
+class QgsActiveLayerFeaturesLocatorFilter : public QgsLocatorFilter
+{
+    Q_OBJECT
+
+  public:
+
+    QgsActiveLayerFeaturesLocatorFilter( QObject *parent = nullptr );
+    virtual QString name() const override { return QStringLiteral( "features" ); }
+    virtual QString displayName() const override { return tr( "Active Layer Features" ); }
+    virtual Priority priority() const override { return Medium; }
+    QString prefix() const override { return QStringLiteral( "f" ); }
+
+    void fetchResults( const QString &string, const QgsLocatorContext &context, QgsFeedback *feedback ) override;
+    void triggerResult( const QgsLocatorResult &result ) override;
+};
+
+
 #endif // QGSINBUILTLOCATORFILTERS_H
 
 
diff --git a/src/app/qgisapp.cpp b/src/app/qgisapp.cpp
@@ -2672,7 +2672,7 @@ void QgisApp::createStatusBar()
                 << mSnappingToolBar;
 
   mLocatorWidget->locator()->registerFilter( new QgsActionLocatorFilter( actionObjects ) );
-
+  mLocatorWidget->locator()->registerFilter( new QgsActiveLayerFeaturesLocatorFilter() );
 }
 
 void QgisApp::setIconSizes( int size )

diff --git a/src/app/qgisapp.h b/src/app/qgisapp.h
@@ -621,6 +621,9 @@ class APP_EXPORT QgisApp : public QMainWindow, private Ui::MainWindow
     //! Process the list of URIs that have been dropped in QGIS
     void handleDropUriList( const QgsMimeDataUtils::UriList &lst );
 
+    //! Returns the active map layer.
+    QgsMapLayer *activeLayer();
+
   public slots:
     void layerTreeViewDoubleClicked( const QModelIndex &index );
     //! Make sure the insertion point for new layers is up-to-date with the current item in layer tree view
@@ -1139,8 +1142,7 @@ class APP_EXPORT QgisApp : public QMainWindow, private Ui::MainWindow
     void hideDeselectedLayers();
     //reimplements method from base (gui) class
     void showSelectedLayers();
-    //! Return pointer to the active layer
-    QgsMapLayer *activeLayer();
+
     //! Open the help contents in a browser
     void helpContents();
     //! Open the API documentation in a browser

diff --git a/src/gui/locator/qgslocator.cpp b/src/gui/locator/qgslocator.cpp
@@ -63,7 +63,8 @@ void QgsLocator::registerFilter( QgsLocatorFilter *filter )
   if ( !filter->prefix().isEmpty() )
   {
     if ( filter->name() == QStringLiteral( "actions" ) || filter->name() == QStringLiteral( "processing_alg" )
-         || filter->name() == QStringLiteral( "layertree" ) || filter->name() == QStringLiteral( "layouts" ) )
+         || filter->name() == QStringLiteral( "layertree" ) || filter->name() == QStringLiteral( "layouts" )
+         || filter->name() == QStringLiteral( "features" ) )
     {
       //inbuilt filter, no prefix check
       mPrefixedFilters.insert( filter->prefix(), filter );