Add legend node function

qgis · Apr 9, 2019 · 81f029d · 81f029d
1 parent 3925246
commit 81f029d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/server/services/wms/qgswmsgetlegendgraphics.cpp b/src/server/services/wms/qgswmsgetlegendgraphics.cpp
@@ -228,5 +228,18 @@ namespace QgsWms
 
     return tree.release();
   }
+
+  QgsLayerTreeModelLegendNode *legendNode( const QgsLayerTreeModel &model, const QString &rule )
+  {
+    for ( QgsLayerTreeLayer *layer : model->rootGroup()->findLayers() )
+    {
+      for ( QgsLayerTreeModelLegendNode *node : model->layerLegendNodes( layer ) )
+      {
+        if ( node->data( Qt::DisplayRole ).toString().compare( rule ) == 0 )
+          return node;
+      }
+    }
+    return nullptr;
+  }
 } // namespace QgsWms
 
diff --git a/src/server/services/wms/qgswmsgetlegendgraphics.h b/src/server/services/wms/qgswmsgetlegendgraphics.h
@@ -37,4 +37,6 @@ namespace QgsWms
   QgsLayerTreeModel *legendModel( const QgsWmsRenderContext &context, QgsLayerTree &tree );
 
   QgsLayerTree *layerTree( const QgsWmsRenderContext &context );
+
+  QgsLayerTreeModelLegendNode *legendNode( const QgsLayerTreeModel &model, const QString &rule );
 } // namespace QgsWms