Fix size_curr_geo in resize shader

3 files changed, 3 insertions, 1 deletions

diff --git a/src/render_helpers/shaders/mod.rs b/src/render_helpers/shaders/mod.rs
index 8dc3263f..4c4530c5 100644
--- a/src/render_helpers/shaders/mod.rs
+++ b/src/render_helpers/shaders/mod.rs
@@ -92,6 +92,7 @@ fn compile_resize_program(
             UniformName::new("niri_input_to_curr_geo", UniformType::Matrix3x3),
             UniformName::new("niri_curr_geo_to_prev_geo", UniformType::Matrix3x3),
             UniformName::new("niri_curr_geo_to_next_geo", UniformType::Matrix3x3),
+            UniformName::new("niri_curr_geo_size", UniformType::_2f),
             UniformName::new("niri_geo_to_tex_prev", UniformType::Matrix3x3),
             UniformName::new("niri_geo_to_tex_next", UniformType::Matrix3x3),
             UniformName::new("niri_progress", UniformType::_1f),
diff --git a/src/render_helpers/shaders/resize-epilogue.frag b/src/render_helpers/shaders/resize-epilogue.frag
index b3eafc7c..b857d57e 100644
--- a/src/render_helpers/shaders/resize-epilogue.frag
+++ b/src/render_helpers/shaders/resize-epilogue.frag
@@ -1,7 +1,7 @@
 
 void main() {
     vec3 coords_curr_geo = niri_input_to_curr_geo * vec3(niri_v_coords, 1.0);
-    vec3 size_curr_geo = niri_input_to_curr_geo * vec3(niri_size, 1.0);
+    vec3 size_curr_geo = vec3(niri_curr_geo_size, 1.0);
 
     vec4 color = resize_color(coords_curr_geo, size_curr_geo);
 
diff --git a/src/render_helpers/shaders/resize-prelude.frag b/src/render_helpers/shaders/resize-prelude.frag
index 95587702..86423ce9 100644
--- a/src/render_helpers/shaders/resize-prelude.frag
+++ b/src/render_helpers/shaders/resize-prelude.frag
@@ -8,6 +8,7 @@ uniform vec2 niri_size;
 uniform mat3 niri_input_to_curr_geo;
 uniform mat3 niri_curr_geo_to_prev_geo;
 uniform mat3 niri_curr_geo_to_next_geo;
+uniform vec2 niri_curr_geo_size;
 
 uniform sampler2D niri_tex_prev;
 uniform mat3 niri_geo_to_tex_prev;