1 files changed, 10 insertions, 2 deletions
diff --git a/src/shader/code/vertex.glsl b/src/shader/code/vertex.glsl
index e132e5f..1051d8f 100644
--- a/src/shader/code/vertex.glsl
+++ b/src/shader/code/vertex.glsl
@@ -10,12 +10,20 @@ out VS_OUT {
 uniform uint nX;
 uniform uint nY;
 
-const float displayAmplifier = 10.0;
+const float displayAmplifier = 5.0;
 
 float unit(float x) {
 	return 1.0/(1.0+exp(-x));
 }
 
+float sq(float x) {
+	return x*x;
+}
+
+float norm(vec2 v) {
+	return sqrt(sq(v.x)+sq(v.y));
+}
+
 vec3 getColor(float x) {
 	return x*vec3(1.0,0.0,0.0) + (1-x)*vec3(-0.5,0.0,1.0);
 }
@@ -39,6 +47,6 @@ void main() {
 		1.
 	);
 
-	vs_out.color = getColor(unit(displayAmplifier * VertexPosition.z));
+	vs_out.color = getColor(unit(displayAmplifier * VertexPosition.z * norm(VertexPosition.xy)));
 }
 )";