Add method to build layer tree

pblottiere · pblottiere · commit 46746a371362 · 2019-04-09T15:43:11.000+02:00
diff --git a/src/server/services/wms/qgswmsgetlegendgraphics.cpp b/src/server/services/wms/qgswmsgetlegendgraphics.cpp
@@ -18,6 +18,10 @@
  *   (at your option) any later version.                                   *
  *                                                                         *
  ***************************************************************************/
+#include "qgslayertree.h"
+#include "qgsvectorlayer.h"
+#include "qgsvectorlayerfeaturecounter.h"
+
 #include "qgswmsutils.h"
 #include "qgswmsserviceexception.h"
 #include "qgswmsgetlegendgraphics.h"
@@ -124,5 +128,50 @@ namespace QgsWms
                                     QStringLiteral( "BBOX parameter cannot be combined with RULE." ) );
     }
   }
+
+  QgsLayerTreeModel *legendModel( const QgsWmsRenderContext &context, QgsLayerTree &tree )
+  {
+    const QgsWmsParameters parameters = context.parameters();
+
+    std::unique_ptr<QgsLayerTreeModel> model( new QgsLayerTreeModel( &tree ) );
+    return model.release();
+  }
+
+  QgsLayerTree *layerTree( const QgsWmsRenderContext &context )
+  {
+    std::unique_ptr<QgsLayerTree> tree;
+
+    QList<QgsVectorLayerFeatureCounter *> counters;
+    for ( QgsMapLayer *ml : context.layersToRender() )
+    {
+      QgsLayerTreeLayer *lt = tree->addLayer( ml );
+
+      // name
+      if ( !ml->title().isEmpty() )
+        lt->setName( ml->title() );
+
+      //show feature count
+      const bool showFeatureCount = context.parameters().showFeatureCountAsBool();
+      const QString property = QStringLiteral( "showFeatureCount" );
+      lt->setCustomProperty( property, showFeatureCount );
+
+      if ( ml->type() != QgsMapLayerType::VectorLayer || !showFeatureCount )
+        continue;
+
+      QgsVectorLayer *vl = qobject_cast<QgsVectorLayer *>( ml );
+      QgsVectorLayerFeatureCounter *counter = vl->countSymbolFeatures();
+      if ( !counter )
+        continue;
+
+      counters.append( counter );
+    }
+
+    for ( QgsVectorLayerFeatureCounter *counter : counters )
+    {
+      counter->waitForFinished();
+    }
+
+    return tree.release();
+  }
 } // namespace QgsWms
 
diff --git a/src/server/services/wms/qgswmsgetlegendgraphics.h b/src/server/services/wms/qgswmsgetlegendgraphics.h
@@ -18,6 +18,9 @@
  *   (at your option) any later version.                                   *
  *                                                                         *
  ***************************************************************************/
+#include "qgslayertreemodel.h"
+
+#include "qgswmsrendercontext.h"
 
 namespace QgsWms
 {
@@ -30,8 +33,8 @@ namespace QgsWms
                                QgsServerResponse &response );
 
   void checkParameters( const QgsWmsParameters &parameters );
-} // namespace QgsWms
-
-
 
+  QgsLayerTreeModel *legendModel( const QgsWmsRenderContext &context, QgsLayerTree &tree );
 
+  QgsLayerTree *layerTree( const QgsWmsRenderContext &context );
+} // namespace QgsWms
