Add legend node function

pblottiere · pblottiere · commit 81f029d19c59 · 2019-04-09T15:43:11.000+02:00
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
