]> git.sesse.net Git - movit/blobdiff - footer.comp
We've run with many more GPUs than just those three now.
[movit] / footer.comp
index 3ece901eb00e8de9fe4f1200101a19162b2b08fc..17175d9c717777665c0ec537d91cc95e6524c835 100644 (file)
@@ -1,3 +1,14 @@
+// GLSL is pickier than the C++ preprocessor in if-testing for undefined
+// tokens; do some fixups here to keep it happy.
+
+#ifndef SQUARE_ROOT_TRANSFORMATION
+#define SQUARE_ROOT_TRANSFORMATION 0
+#endif
+
+#ifndef FLIP_ORIGIN
+#define FLIP_ORIGIN 0
+#endif
+
 void main()
 {
        INPUT();
@@ -10,10 +21,23 @@ vec4 tex2D(sampler2D s, vec2 coord)
 
 void cs_output(uvec2 coord, vec4 val)
 {
-       imageStore(outbuf, ivec2(coord), val);
+       cs_output(ivec2(coord), val);
 }
 
 void cs_output(ivec2 coord, vec4 val)
 {
-       imageStore(outbuf, coord, val);
+       // Run the value through any preprocessing steps we might have.
+       CS_OUTPUT_VAL = val;
+       val = CS_POSTPROC(vec2(0.0, 0.0));
+
+#if SQUARE_ROOT_TRANSFORMATION
+       // Make sure we don't give negative values to sqrt.
+       val.rgb = sqrt(max(val.rgb, 0.0));
+#endif
+
+#if FLIP_ORIGIN
+       coord.y = imageSize(tex_outbuf).y - coord.y - 1;
+#endif
+
+       imageStore(tex_outbuf, coord, val);
 }