Add Naga variant to ShaderSource

wgpu/src/backend/direct.rs

@@ -1103,6 +1103,8 @@ impl crate::Context for Context {
  wgc::pipeline::ShaderModuleSource::Naga(module)
  }
  ShaderSource::Wgsl(ref code) => wgc::pipeline::ShaderModuleSource::Wgsl(Borrowed(code)),
+ #[cfg(feature = "naga")]
+ ShaderSource::Naga(ref module) => wgc::pipeline::ShaderModuleSource::Naga(module.take()),
  };
  let (id, error) = wgc::gfx_select!(
  device.id => global.device_create_shader_module(device.id, &descriptor, source, PhantomData)

wgpu/src/backend/web.rs

@@ -1265,6 +1265,22 @@ impl crate::Context for Context {
  web_sys::GpuShaderModuleDescriptor::new(wgsl_text.as_str())
  }
  crate::ShaderSource::Wgsl(ref code) => web_sys::GpuShaderModuleDescriptor::new(code),
+ #[cfg(feature = "naga")]
+ crate::ShaderSource::Naga(ref cell) => {
+ use naga::{back, valid};
+
+ let module = cell.take();
+ let mut validator = valid::Validator::new(
+ valid::ValidationFlags::all(),
+ valid::Capabilities::all(),
+ );
+ let module_info = validator.validate(&module).unwrap();
+
+ let writer_flags = naga::back::wgsl::WriterFlags::empty();
+ let wgsl_text =
+ back::wgsl::write_string(&module, &module_info, writer_flags).unwrap();
+ web_sys::GpuShaderModuleDescriptor::new(wgsl_text.as_str())
+ }
  };
  if let Some(label) = desc.label {
  descriptor.label(label);

wgpu/src/lib.rs

@@ -812,6 +812,10 @@ pub enum ShaderSource<'a> {
  },
  /// WGSL module as a string slice.
  Wgsl(Cow<'a, str>),
+ /// Naga module.
+ #[cfg(feature = "naga")]
+ #[cfg_attr(docsrs, doc(cfg(feature = "naga")))]
+ Naga(std::cell::Cell<naga::Module>),
 }
 
 /// Descriptor for use with [`Device::create_shader_module`].