Identify tool: Show layer type icons in layer selection mode

qgis · May 21, 2013 · 29871ee · 29871ee
1 parent e5d8307
commit 29871ee
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/app/qgsmaptoolidentifyaction.cpp b/src/app/qgsmaptoolidentifyaction.cpp
@@ -13,6 +13,7 @@
  *                                                                         *
  ***************************************************************************/
 
+#include "qgsapplication.h"
 #include "qgscursors.h"
 #include "qgsdistancearea.h"
 #include "qgsfeature.h"
@@ -251,6 +252,29 @@ void QgsMapToolIdentifyAction::fillLayerSelectionMenu( QMenu& menu )
   {
     QAction* action = new QAction( resultIt.key()->name(), 0 );
     action->setData( resultIt.key()->id() );
+    //add point/line/polygon icon
+    QgsVectorLayer* vl = qobject_cast<QgsVectorLayer*>( resultIt.key() );
+    if ( vl )
+    {
+      switch ( vl->geometryType() )
+      {
+        case QGis::Point:
+          action->setIcon( QgsApplication::getThemeIcon( "/mIconPointLayer.png" ) );
+          break;
+        case QGis::Line:
+          action->setIcon( QgsApplication::getThemeIcon( "/mIconLineLayer.png" ) );
+          break;
+        case QGis::Polygon:
+          action->setIcon( QgsApplication::getThemeIcon( "/mIconPolygonLayer.png" ) );
+          break;
+        default:
+          break;
+      }
+    }
+    else if ( resultIt.key()->type() == QgsMapLayer::RasterLayer )
+    {
+      action->setIcon( QgsApplication::getThemeIcon( "/mIconRaster.png" ) );
+    }
     QObject::connect( action, SIGNAL( hovered() ), this, SLOT( handleMenuHover() ) );
     menu.addAction( action );
   }