Render custom bed textures in png format on prusa beds

1 files changed, 18 insertions, 4 deletions

diff --git a/resources/shaders/printbed.fs b/resources/shaders/printbed.fs
index be14347c2..d1316ca2f 100644
--- a/resources/shaders/printbed.fs
+++ b/resources/shaders/printbed.fs
@@ -5,16 +5,30 @@ const vec3 back_color_light = vec3(0.365, 0.365, 0.365);
 
 uniform sampler2D texture;
 uniform bool transparent_background;
+uniform bool svg_source;
 
 varying vec2 tex_coords;
 
-void main()
+vec4 svg_color()
 {
+    // takes foreground from texture
+    vec4 fore_color = texture2D(texture, tex_coords);
+
     // calculates radial gradient
     vec3 back_color = vec3(mix(back_color_light, back_color_dark, smoothstep(0.0, 0.5, length(abs(tex_coords.xy) - vec2(0.5)))));
 
-    vec4 fore_color = texture2D(texture, tex_coords);
-
     // blends foreground with background
-    gl_FragColor = vec4(mix(back_color, fore_color.rgb, fore_color.a), transparent_background ? fore_color.a : 1.0);
+    return vec4(mix(back_color, fore_color.rgb, fore_color.a), transparent_background ? fore_color.a : 1.0);
+}
+
+vec4 non_svg_color()
+{
+    // takes foreground from texture
+    vec4 color = texture2D(texture, tex_coords);
+    return vec4(color.rgb, transparent_background ? color.a * 0.25 : color.a);
+}
+
+void main()
+{
+    gl_FragColor = svg_source ? svg_color() : non_svg_color();
 }
\ No newline at end of file