1 files changed, 21 insertions, 6 deletions
diff --git a/stockton-render/src/draw/ui/data/stockton.vert b/stockton-render/src/draw/ui/data/stockton.vert
index 1dd9477..7447fec 100644
--- a/stockton-render/src/draw/ui/data/stockton.vert
+++ b/stockton-render/src/draw/ui/data/stockton.vert
@@ -1,11 +1,26 @@
 #version 450
 
-vec2 positions[3] = vec2[](
-    vec2(0.0, -0.5),
-    vec2(0.5, 0.5),
-    vec2(-0.5, 0.5)
-);
+layout (push_constant) uniform PushConsts {
+    vec2 screen_size;
+} push;
+
+layout(location = 0) in vec2 pos;
+layout (location = 1) in vec2 uv;
+layout (location = 2) in int col; // rgba of u8s
+
+out gl_PerVertex {
+    vec4 gl_Position;
+};
+layout (location = 1) out vec2 frag_uv;
+layout (location = 2) out int frag_col;
 
 void main() {
-    gl_Position = vec4(positions[gl_VertexIndex], 0.0, 1.0);
+    gl_Position = vec4(
+        ((pos.x / push.screen_size.x) * 2.0) - 1.0,
+        ((pos.y / push.screen_size.y) * 2.0) - 1.0,
+        0.0,
+        1.0
+    );
+    frag_uv = uv;
+    frag_col = col;
 }