debug_printf: glsl-out

gfx-rs · Oct 16, 2023 · bd404ff · bd404ff
1 parent 54f02a5
commit bd404ff
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 2 deletions.
diff --git a/src/back/glsl/features.rs b/src/back/glsl/features.rs
@@ -43,6 +43,8 @@ bitflags::bitflags! {
         const IMAGE_SIZE = 1 << 20;
         /// Dual source blending
         const DUAL_SOURCE_BLENDING = 1 << 21;
+        /// Debug Printf
+        const DEBUG_PRINTF = 1 << 22;
     }
 }
 
@@ -235,6 +237,10 @@ impl FeaturesManager {
             // https://registry.khronos.org/OpenGL/extensions/EXT/EXT_blend_func_extended.txt
             writeln!(out, "#extension GL_EXT_blend_func_extended : require")?;
         }
+        if self.0.contains(Features::DEBUG_PRINTF) {
+            // https://github.com/KhronosGroup/GLSL/blob/master/extensions/ext/GLSL_EXT_debug_printf.txt
+            writeln!(out, "#extension GL_EXT_debug_printf : enable")?;
+        }
 
         Ok(())
     }
@@ -407,6 +413,22 @@ impl<'a, W> Writer<'a, W> {
             ..
         } = self;
 
+        for block in module
+            .functions
+            .iter()
+            .map(|(_, f)| &f.body)
+            .chain(std::iter::once(&entry_point.function.body))
+        {
+            for statement in block.iter() {
+                match *statement {
+                    crate::Statement::DebugPrintf { .. } => {
+                        features.request(Features::DEBUG_PRINTF)
+                    }
+                    _ => {}
+                }
+            }
+        }
+
         // Loop trough all expressions in both functions and the entry point
         // to check for needed features
         for (expressions, info) in module

diff --git a/src/back/glsl/mod.rs b/src/back/glsl/mod.rs
@@ -2238,8 +2238,14 @@ impl<'a, W: Write> Writer<'a, W> {
                 writeln!(self.out, ");")?;
             }
             Statement::RayQuery { .. } => unreachable!(),
-            Statement::DebugPrintf { .. } => {
-                return Err(Error::Custom("debugPrintf is not implemented".to_string()));
+            Statement::DebugPrintf {
+                ref format,
+                ref arguments,
+            } => {
+                write!(self.out, "{level}")?;
+                write!(self.out, "debugPrintfEXT(\"{format}\",")?;
+                self.write_slice(arguments, |this, _, arg| this.write_expr(*arg, ctx))?;
+                writeln!(self.out, ");")?
             }
         }