wgpu_naga_bridge/
lib.rs

1//! Conversions between [`naga`] and [`wgpu_types`].
2
3#![no_std]
4
5use naga::valid::Capabilities as Caps;
6use wgpu_types as wgt;
7
8/// Map [`wgt::Features`] and [`wgt::DownlevelFlags`] to [`naga::valid::Capabilities`].
9pub fn features_to_naga_capabilities(
10    features: wgt::Features,
11    downlevel: wgt::DownlevelFlags,
12) -> Caps {
13    let mut caps = Caps::empty();
14    caps.set(
15        Caps::IMMEDIATES,
16        features.contains(wgt::Features::IMMEDIATES),
17    );
18    caps.set(Caps::FLOAT64, features.contains(wgt::Features::SHADER_F64));
19    caps.set(
20        Caps::SHADER_FLOAT16,
21        features.contains(wgt::Features::SHADER_F16),
22    );
23    caps.set(
24        Caps::SHADER_FLOAT16_IN_FLOAT32,
25        downlevel.contains(wgt::DownlevelFlags::SHADER_F16_IN_F32),
26    );
27    caps.set(
28        Caps::PRIMITIVE_INDEX,
29        features.contains(wgt::Features::PRIMITIVE_INDEX),
30    );
31    caps.set(
32        Caps::TEXTURE_AND_SAMPLER_BINDING_ARRAY,
33        features.contains(wgt::Features::TEXTURE_BINDING_ARRAY),
34    );
35    caps.set(
36        Caps::BUFFER_BINDING_ARRAY,
37        features.contains(wgt::Features::BUFFER_BINDING_ARRAY),
38    );
39    caps.set(
40        Caps::STORAGE_TEXTURE_BINDING_ARRAY,
41        features.contains(wgt::Features::TEXTURE_BINDING_ARRAY)
42            && features.contains(wgt::Features::STORAGE_RESOURCE_BINDING_ARRAY),
43    );
44    caps.set(
45        Caps::STORAGE_BUFFER_BINDING_ARRAY,
46        features.contains(wgt::Features::BUFFER_BINDING_ARRAY)
47            && features.contains(wgt::Features::STORAGE_RESOURCE_BINDING_ARRAY),
48    );
49    caps.set(
50        Caps::TEXTURE_AND_SAMPLER_BINDING_ARRAY_NON_UNIFORM_INDEXING,
51        features
52            .contains(wgt::Features::SAMPLED_TEXTURE_AND_STORAGE_BUFFER_ARRAY_NON_UNIFORM_INDEXING),
53    );
54    caps.set(
55        Caps::BUFFER_BINDING_ARRAY_NON_UNIFORM_INDEXING,
56        features.contains(wgt::Features::UNIFORM_BUFFER_BINDING_ARRAYS),
57    );
58    caps.set(
59        Caps::STORAGE_TEXTURE_BINDING_ARRAY_NON_UNIFORM_INDEXING,
60        features.contains(wgt::Features::STORAGE_TEXTURE_ARRAY_NON_UNIFORM_INDEXING),
61    );
62    caps.set(
63        Caps::STORAGE_BUFFER_BINDING_ARRAY_NON_UNIFORM_INDEXING,
64        features
65            .contains(wgt::Features::SAMPLED_TEXTURE_AND_STORAGE_BUFFER_ARRAY_NON_UNIFORM_INDEXING),
66    );
67    caps.set(
68        Caps::ACCELERATION_STRUCTURE_BINDING_ARRAY,
69        features.contains(wgt::Features::ACCELERATION_STRUCTURE_BINDING_ARRAY),
70    );
71    caps.set(
72        Caps::STORAGE_TEXTURE_16BIT_NORM_FORMATS,
73        features.contains(wgt::Features::TEXTURE_FORMAT_16BIT_NORM),
74    );
75    caps.set(Caps::MULTIVIEW, features.contains(wgt::Features::MULTIVIEW));
76    caps.set(
77        Caps::EARLY_DEPTH_TEST,
78        features.contains(wgt::Features::SHADER_EARLY_DEPTH_TEST),
79    );
80    caps.set(
81        Caps::SHADER_INT64,
82        features.contains(wgt::Features::SHADER_INT64),
83    );
84    caps.set(
85        Caps::SHADER_INT64_ATOMIC_MIN_MAX,
86        features.intersects(
87            wgt::Features::SHADER_INT64_ATOMIC_MIN_MAX | wgt::Features::SHADER_INT64_ATOMIC_ALL_OPS,
88        ),
89    );
90    caps.set(
91        Caps::SHADER_INT64_ATOMIC_ALL_OPS,
92        features.contains(wgt::Features::SHADER_INT64_ATOMIC_ALL_OPS),
93    );
94    caps.set(
95        Caps::TEXTURE_ATOMIC,
96        features.contains(wgt::Features::TEXTURE_ATOMIC),
97    );
98    caps.set(
99        Caps::TEXTURE_INT64_ATOMIC,
100        features.contains(wgt::Features::TEXTURE_INT64_ATOMIC),
101    );
102    caps.set(
103        Caps::SHADER_FLOAT32_ATOMIC,
104        features.contains(wgt::Features::SHADER_FLOAT32_ATOMIC),
105    );
106    caps.set(
107        Caps::MULTISAMPLED_SHADING,
108        downlevel.contains(wgt::DownlevelFlags::MULTISAMPLED_SHADING),
109    );
110    caps.set(
111        Caps::DUAL_SOURCE_BLENDING,
112        features.contains(wgt::Features::DUAL_SOURCE_BLENDING),
113    );
114    caps.set(
115        Caps::CLIP_DISTANCES,
116        features.contains(wgt::Features::CLIP_DISTANCES),
117    );
118    caps.set(
119        Caps::CUBE_ARRAY_TEXTURES,
120        downlevel.contains(wgt::DownlevelFlags::CUBE_ARRAY_TEXTURES),
121    );
122    caps.set(
123        Caps::SUBGROUP,
124        features.intersects(wgt::Features::SUBGROUP | wgt::Features::SUBGROUP_VERTEX),
125    );
126    caps.set(
127        Caps::SUBGROUP_BARRIER,
128        features.intersects(wgt::Features::SUBGROUP_BARRIER),
129    );
130    caps.set(
131        Caps::RAY_QUERY,
132        features.intersects(wgt::Features::EXPERIMENTAL_RAY_QUERY),
133    );
134    caps.set(
135        Caps::SUBGROUP_VERTEX_STAGE,
136        features.contains(wgt::Features::SUBGROUP_VERTEX),
137    );
138    caps.set(
139        Caps::RAY_HIT_VERTEX_POSITION,
140        features.intersects(wgt::Features::EXPERIMENTAL_RAY_HIT_VERTEX_RETURN),
141    );
142    caps.set(
143        Caps::TEXTURE_EXTERNAL,
144        features.intersects(wgt::Features::EXTERNAL_TEXTURE),
145    );
146    caps.set(
147        Caps::SHADER_BARYCENTRICS,
148        features.intersects(wgt::Features::SHADER_BARYCENTRICS),
149    );
150    caps.set(
151        Caps::MESH_SHADER,
152        features.intersects(wgt::Features::EXPERIMENTAL_MESH_SHADER),
153    );
154    caps.set(
155        Caps::MESH_SHADER_POINT_TOPOLOGY,
156        features.intersects(wgt::Features::EXPERIMENTAL_MESH_SHADER_POINTS),
157    );
158    caps.set(
159        Caps::COOPERATIVE_MATRIX,
160        features.intersects(wgt::Features::EXPERIMENTAL_COOPERATIVE_MATRIX),
161    );
162    caps.set(
163        Caps::PER_VERTEX,
164        features.intersects(wgt::Features::SHADER_PER_VERTEX),
165    );
166    caps.set(
167        Caps::DRAW_INDEX,
168        features.intersects(wgt::Features::SHADER_DRAW_INDEX),
169    );
170    caps.set(
171        Caps::MEMORY_DECORATION_COHERENT,
172        features.contains(wgt::Features::MEMORY_DECORATION_COHERENT),
173    );
174    caps.set(
175        Caps::MEMORY_DECORATION_VOLATILE,
176        features.contains(wgt::Features::MEMORY_DECORATION_VOLATILE),
177    );
178    caps
179}
180
181/// Create a [`naga::valid::Validator`] configured for the given feature set.
182pub fn create_validator(
183    features: wgt::Features,
184    downlevel: wgt::DownlevelFlags,
185    flags: naga::valid::ValidationFlags,
186) -> naga::valid::Validator {
187    let caps = features_to_naga_capabilities(features, downlevel);
188    naga::valid::Validator::new(flags, caps)
189}
190
191/// Map a [`wgt::TextureFormat`] to the corresponding [`naga::StorageFormat`], if any.
192pub fn map_storage_format_to_naga(format: wgt::TextureFormat) -> Option<naga::StorageFormat> {
193    use naga::StorageFormat as Sf;
194    use wgt::TextureFormat as Tf;
195
196    Some(match format {
197        Tf::R8Unorm => Sf::R8Unorm,
198        Tf::R8Snorm => Sf::R8Snorm,
199        Tf::R8Uint => Sf::R8Uint,
200        Tf::R8Sint => Sf::R8Sint,
201
202        Tf::R16Uint => Sf::R16Uint,
203        Tf::R16Sint => Sf::R16Sint,
204        Tf::R16Float => Sf::R16Float,
205        Tf::Rg8Unorm => Sf::Rg8Unorm,
206        Tf::Rg8Snorm => Sf::Rg8Snorm,
207        Tf::Rg8Uint => Sf::Rg8Uint,
208        Tf::Rg8Sint => Sf::Rg8Sint,
209
210        Tf::R32Uint => Sf::R32Uint,
211        Tf::R32Sint => Sf::R32Sint,
212        Tf::R32Float => Sf::R32Float,
213        Tf::Rg16Uint => Sf::Rg16Uint,
214        Tf::Rg16Sint => Sf::Rg16Sint,
215        Tf::Rg16Float => Sf::Rg16Float,
216        Tf::Rgba8Unorm => Sf::Rgba8Unorm,
217        Tf::Rgba8Snorm => Sf::Rgba8Snorm,
218        Tf::Rgba8Uint => Sf::Rgba8Uint,
219        Tf::Rgba8Sint => Sf::Rgba8Sint,
220        Tf::Bgra8Unorm => Sf::Bgra8Unorm,
221
222        Tf::Rgb10a2Uint => Sf::Rgb10a2Uint,
223        Tf::Rgb10a2Unorm => Sf::Rgb10a2Unorm,
224        Tf::Rg11b10Ufloat => Sf::Rg11b10Ufloat,
225
226        Tf::R64Uint => Sf::R64Uint,
227        Tf::Rg32Uint => Sf::Rg32Uint,
228        Tf::Rg32Sint => Sf::Rg32Sint,
229        Tf::Rg32Float => Sf::Rg32Float,
230        Tf::Rgba16Uint => Sf::Rgba16Uint,
231        Tf::Rgba16Sint => Sf::Rgba16Sint,
232        Tf::Rgba16Float => Sf::Rgba16Float,
233
234        Tf::Rgba32Uint => Sf::Rgba32Uint,
235        Tf::Rgba32Sint => Sf::Rgba32Sint,
236        Tf::Rgba32Float => Sf::Rgba32Float,
237
238        Tf::R16Unorm => Sf::R16Unorm,
239        Tf::R16Snorm => Sf::R16Snorm,
240        Tf::Rg16Unorm => Sf::Rg16Unorm,
241        Tf::Rg16Snorm => Sf::Rg16Snorm,
242        Tf::Rgba16Unorm => Sf::Rgba16Unorm,
243        Tf::Rgba16Snorm => Sf::Rgba16Snorm,
244
245        _ => return None,
246    })
247}
248
249/// Map a [`naga::StorageFormat`] to the corresponding [`wgt::TextureFormat`].
250pub fn map_storage_format_from_naga(format: naga::StorageFormat) -> wgt::TextureFormat {
251    use naga::StorageFormat as Sf;
252    use wgt::TextureFormat as Tf;
253
254    match format {
255        Sf::R8Unorm => Tf::R8Unorm,
256        Sf::R8Snorm => Tf::R8Snorm,
257        Sf::R8Uint => Tf::R8Uint,
258        Sf::R8Sint => Tf::R8Sint,
259
260        Sf::R16Uint => Tf::R16Uint,
261        Sf::R16Sint => Tf::R16Sint,
262        Sf::R16Float => Tf::R16Float,
263        Sf::Rg8Unorm => Tf::Rg8Unorm,
264        Sf::Rg8Snorm => Tf::Rg8Snorm,
265        Sf::Rg8Uint => Tf::Rg8Uint,
266        Sf::Rg8Sint => Tf::Rg8Sint,
267
268        Sf::R32Uint => Tf::R32Uint,
269        Sf::R32Sint => Tf::R32Sint,
270        Sf::R32Float => Tf::R32Float,
271        Sf::Rg16Uint => Tf::Rg16Uint,
272        Sf::Rg16Sint => Tf::Rg16Sint,
273        Sf::Rg16Float => Tf::Rg16Float,
274        Sf::Rgba8Unorm => Tf::Rgba8Unorm,
275        Sf::Rgba8Snorm => Tf::Rgba8Snorm,
276        Sf::Rgba8Uint => Tf::Rgba8Uint,
277        Sf::Rgba8Sint => Tf::Rgba8Sint,
278        Sf::Bgra8Unorm => Tf::Bgra8Unorm,
279
280        Sf::Rgb10a2Uint => Tf::Rgb10a2Uint,
281        Sf::Rgb10a2Unorm => Tf::Rgb10a2Unorm,
282        Sf::Rg11b10Ufloat => Tf::Rg11b10Ufloat,
283
284        Sf::R64Uint => Tf::R64Uint,
285        Sf::Rg32Uint => Tf::Rg32Uint,
286        Sf::Rg32Sint => Tf::Rg32Sint,
287        Sf::Rg32Float => Tf::Rg32Float,
288        Sf::Rgba16Uint => Tf::Rgba16Uint,
289        Sf::Rgba16Sint => Tf::Rgba16Sint,
290        Sf::Rgba16Float => Tf::Rgba16Float,
291
292        Sf::Rgba32Uint => Tf::Rgba32Uint,
293        Sf::Rgba32Sint => Tf::Rgba32Sint,
294        Sf::Rgba32Float => Tf::Rgba32Float,
295
296        Sf::R16Unorm => Tf::R16Unorm,
297        Sf::R16Snorm => Tf::R16Snorm,
298        Sf::Rg16Unorm => Tf::Rg16Unorm,
299        Sf::Rg16Snorm => Tf::Rg16Snorm,
300        Sf::Rgba16Unorm => Tf::Rgba16Unorm,
301        Sf::Rgba16Snorm => Tf::Rgba16Snorm,
302    }
303}
304
305/// Map a [`naga::ShaderStage`] to the corresponding [`wgt::ShaderStages`] flag.
306pub fn map_naga_stage(stage: naga::ShaderStage) -> wgt::ShaderStages {
307    match stage {
308        naga::ShaderStage::Vertex => wgt::ShaderStages::VERTEX,
309        naga::ShaderStage::Fragment => wgt::ShaderStages::FRAGMENT,
310        naga::ShaderStage::Compute => wgt::ShaderStages::COMPUTE,
311        naga::ShaderStage::Task => wgt::ShaderStages::TASK,
312        naga::ShaderStage::Mesh => wgt::ShaderStages::MESH,
313        naga::ShaderStage::RayGeneration => wgt::ShaderStages::RAY_GENERATION,
314        naga::ShaderStage::AnyHit => wgt::ShaderStages::ANY_HIT,
315        naga::ShaderStage::ClosestHit => wgt::ShaderStages::CLOSEST_HIT,
316        naga::ShaderStage::Miss => wgt::ShaderStages::MISS,
317    }
318}
wgpu_naga_bridge/lib.rs

wgpu_naga_bridge/
lib.rs
