1 files changed, 15 insertions, 5 deletions
diff --git a/src/shader/code/vertex.glsl b/src/shader/code/vertex.glsl
index 7851292..8958600 100644
--- a/src/shader/code/vertex.glsl
+++ b/src/shader/code/vertex.glsl
@@ -10,6 +10,8 @@ out VS_OUT {
 uniform uint nX;
 uniform uint nY;
 
+uniform bool fluidQuality;
+
 const float velocityDisplayAmplifier = 3.0;
 const int   qualityDisplayRestrictor = 6;
 
@@ -66,11 +68,19 @@ void main() {
 	} else if ( isWallFrontier(material) ) {
 		vs_out.color = vec3(0.0, 0.0, 0.0);
 	} else {
-		vs_out.color = mix(
-			vec3(0.0, 1.0, 0.0),
-			vec3(1.0, 0.0, 0.0),
-			restrictedQuality(VertexPosition.y)
-		);
+		if ( fluidQuality ) {
+			vs_out.color = mix(
+				vec3(0.0, 1.0, 0.0),
+				vec3(1.0, 0.0, 0.0),
+				restrictedQuality(VertexPosition.y)
+			);
+		} else {
+			vs_out.color = mix(
+				vec3(-0.5, 0.0, 1.0),
+				vec3( 1.0, 0.0, 0.0),
+				velocityDisplayAmplifier * norm(VertexPosition.xy)
+			);
+		}
 	}
 }
 )";