1 files changed, 7 insertions, 4 deletions
diff --git a/glsl/shader_2d_wired.frag.glsl b/glsl/shader_2d_wired.frag.glsl
index 0671223..569b950 100644
--- a/glsl/shader_2d_wired.frag.glsl
+++ b/glsl/shader_2d_wired.frag.glsl
@@ -15,16 +15,19 @@
  */
 
 #version 450
+#extension GL_ARB_separate_shader_objects : enable
+
+layout(location = 0) in flat int in_instance_index;
 
 layout(location = 0) out vec4 out_color;
 
-layout(set = 1, binding = 0) uniform UBOModel2D
+layout(set = 1, binding = 1) uniform UBOColors
 {
-  vec3 color;
-} ubo_model_2d;
+  vec3 colors[128];
+} ubo_colors;
 
 void
 main()
 {
-  out_color = vec4(ubo_model_2d.color, 1.0);
+  out_color = vec4(ubo_colors.colors[in_instance_index], 1.0);
 }