1 files changed, 5 insertions, 1 deletions
diff --git a/libs/hwui/shaders/drawLinearGradient.vert b/libs/hwui/shaders/drawLinearGradient.vert
index 963dc87..f5c669b 100644
--- a/libs/hwui/shaders/drawLinearGradient.vert
+++ b/libs/hwui/shaders/drawLinearGradient.vert
@@ -4,13 +4,17 @@ attribute vec4 position;
 
 uniform float gradientLength;
 uniform vec2 gradient;
+uniform vec2 start;
 uniform mat4 transform;
+uniform mat4 screenSpace;
 
 varying float index;
 
 void main(void) {
+    vec4 location = screenSpace * position;
+    index = dot(location.xy - start, gradient) * gradientLength;
+
     gl_Position = transform * position;
-    index = dot(gl_Position.xy, gradient) * gradientLength;
 }
 
 );