bench: glsl, spirv, and bincode inputs

Cargo.toml

@@ -63,6 +63,7 @@ pp-rs = { version = "0.2.1", optional = true }
 hexf-parse = { version = "0.2.1", optional = true }
 
 [dev-dependencies]
+bincode = "1"
 criterion = { version = "0.3", features = ["html_reports"] }
 diff = "0.1"
 ron = "0.7"

benches/criterion.rs

@@ -1,7 +1,7 @@
 use criterion::*;
-use std::{fs, path::PathBuf};
+use std::{fs, path::PathBuf, slice};
 
-fn gather_inputs(folder: &str, extension: &str) -> Vec<String> {
+fn gather_inputs(folder: &str, extension: &str) -> Vec<Box<[u8]>> {
     let mut list = Vec::new();
     let read_dir = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
         .join(folder)
@@ -11,8 +11,8 @@ fn gather_inputs(folder: &str, extension: &str) -> Vec<String> {
         match file_entry {
             Ok(entry) => match entry.path().extension() {
                 Some(ostr) if &*ostr == extension => {
-                    let input = fs::read_to_string(entry.path()).unwrap_or_default();
-                    list.push(input);
+                    let input = fs::read(entry.path()).unwrap_or_default();
+                    list.push(input.into_boxed_slice());
                 }
                 _ => continue,
             },
@@ -25,17 +25,71 @@ fn gather_inputs(folder: &str, extension: &str) -> Vec<String> {
     list
 }
 
+fn parse_glsl(stage: naga::ShaderStage, inputs: &[Box<[u8]>]) {
+    let mut parser = naga::front::glsl::Parser::default();
+    let options = naga::front::glsl::Options {
+        stage,
+        defines: Default::default(),
+    };
+    for input in inputs.iter() {
+        let string = std::str::from_utf8(input).unwrap();
+        parser.parse(&options, string).unwrap();
+    }
+}
+
 fn frontends(c: &mut Criterion) {
-    let mut group = c.benchmark_group("tests/in");
+    let mut group = c.benchmark_group("front");
+    #[cfg(all(feature = "serialize", feature = "deserialize"))]
+    group.bench_function("bin", |b| {
+        let inputs_wgsl = gather_inputs("tests/in", "wgsl");
+        let mut parser = naga::front::wgsl::Parser::new();
+        let inputs_bin = inputs_wgsl
+            .iter()
+            .map(|input| {
+                let string = std::str::from_utf8(input).unwrap();
+                let module = parser.parse(string).unwrap();
+                bincode::serialize(&module).unwrap()
+            })
+            .collect::<Vec<_>>();
+        b.iter(move || {
+            for input in inputs_bin.iter() {
+                bincode::deserialize::<naga::Module>(input).unwrap();
+            }
+        });
+    });
+    #[cfg(feature = "wgsl-in")]
     group.bench_function("wgsl", |b| {
         let inputs = gather_inputs("tests/in", "wgsl");
         let mut parser = naga::front::wgsl::Parser::new();
         b.iter(move || {
             for input in inputs.iter() {
-                parser.parse(input).unwrap();
+                let string = std::str::from_utf8(input).unwrap();
+                parser.parse(string).unwrap();
             }
         });
     });
+    #[cfg(feature = "spv-in")]
+    group.bench_function("spv", |b| {
+        let inputs = gather_inputs("tests/in/spv", "spv");
+        b.iter(move || {
+            let options = naga::front::spv::Options::default();
+            for input in inputs.iter() {
+                let spv =
+                    unsafe { slice::from_raw_parts(input.as_ptr() as *const u32, input.len() / 4) };
+                let parser = naga::front::spv::Parser::new(spv.iter().cloned(), &options);
+                parser.parse().unwrap();
+            }
+        });
+    });
+    #[cfg(feature = "glsl-in")]
+    group.bench_function("glsl", |b| {
+        let vert = gather_inputs("tests/in/glsl", "vert");
+        b.iter(move || parse_glsl(naga::ShaderStage::Vertex, &vert));
+        let frag = gather_inputs("tests/in/glsl", "frag");
+        b.iter(move || parse_glsl(naga::ShaderStage::Vertex, &frag));
+        //let comp = gather_inputs("tests/in/glsl", "comp");
+        //b.iter(move || parse_glsl(naga::ShaderStage::Compute, &comp));
+    });
 }
 
 fn validation(_c: &mut Criterion) {}