Also use ramp based legend nodes for mesh layers

qgis · Dec 16, 2020 · 7fac048 · 7fac048
1 parent bffd8eb
commit 7fac048
Showing 1 changed file with 22 additions and 4 deletions.
diff --git a/src/core/qgsmaplayerlegend.cpp b/src/core/qgsmaplayerlegend.cpp
@@ -28,6 +28,7 @@
 #include "qgssymbollayerutils.h"
 #include "qgspointcloudrenderer.h"
 #include "qgsrasterrenderer.h"
+#include "qgscolorramplegendnode.h"
 
 QgsMapLayerLegend::QgsMapLayerLegend( QObject *parent )
   : QObject( parent )
@@ -471,11 +472,28 @@ QList<QgsLayerTreeModelLegendNode *> QgsDefaultMeshLayerLegend::createLayerTreeM
   if ( indexScalar > -1 )
   {
     QgsMeshRendererScalarSettings settings = rendererSettings.scalarSettings( indexScalar );
-    QgsLegendColorList items;
+    const QgsColorRampShader shader = settings.colorRampShader();
-    settings.colorRampShader().legendSymbologyItems( items );
+    switch ( shader.colorRampType() )
-    for ( const QPair< QString, QColor > &item : qgis::as_const( items ) )
     {
-      nodes << new QgsRasterSymbolLegendNode( nodeLayer, item.second, item.first );
+      case QgsColorRampShader::Interpolated:
+        // for interpolated shaders we use a ramp legend node
+        nodes << new QgsColorRampLegendNode( nodeLayer, shader.sourceColorRamp()->clone(),
+                                             QString::number( shader.minimumValue() ),
+                                             QString::number( shader.maximumValue() ) );
+        break;
+
+      case QgsColorRampShader::Discrete:
+      case QgsColorRampShader::Exact:
+      {
+        // for all others we use itemised lists
+        QgsLegendColorList items;
+        settings.colorRampShader().legendSymbologyItems( items );
+        for ( const QPair< QString, QColor > &item : items )
+        {
+          nodes << new QgsRasterSymbolLegendNode( nodeLayer, item.second, item.first );
+        }
+        break;
+      }
     }
   }