1 files changed, 18 insertions, 7 deletions
diff --git a/source/blender/draw/modes/shaders/edit_mesh_overlay_mesh_analysis_vert.glsl b/source/blender/draw/modes/shaders/edit_mesh_overlay_mesh_analysis_vert.glsl
index 7065ce3df7c..b89a3f407f9 100644
--- a/source/blender/draw/modes/shaders/edit_mesh_overlay_mesh_analysis_vert.glsl
+++ b/source/blender/draw/modes/shaders/edit_mesh_overlay_mesh_analysis_vert.glsl
@@ -1,14 +1,25 @@
 
 in vec3 pos;
-in vec4 weight_color;
+in float weight;
 
-#ifdef FACE_COLOR
-flat out vec4 weightColor;
-#endif
+uniform sampler1D weightTex;
 
-#ifdef VERTEX_COLOR
 out vec4 weightColor;
-#endif
+
+vec3 weight_to_rgb(float t)
+{
+  if (t < 0.0) {
+    /* Minimum color, grey */
+    return vec3(0.25, 0.25, 0.25);
+  }
+  else if (t > 1.0) {
+    /* Error color */
+    return vec3(1.0, 0.0, 1.0);
+  }
+  else {
+    return texture(weightTex, t).rgb;
+  }
+}
 
 void main()
 {
@@ -16,7 +27,7 @@ void main()
 
   vec3 world_pos = point_object_to_world(pos);
   gl_Position = point_world_to_ndc(world_pos);
-  weightColor = vec4(weight_color.rgb, 1.0);
+  weightColor = vec4(weight_to_rgb(weight), 1.0);
 
 #ifdef USE_WORLD_CLIP_PLANES
   world_clip_planes_calc_clip_distance(world_pos);