wgpu_hal/vulkan/
conv.rs

1use alloc::vec::Vec;
2
3use ash::vk;
4
5impl super::PrivateCapabilities {
6    pub fn map_texture_format(&self, format: wgt::TextureFormat) -> vk::Format {
7        use ash::vk::Format as F;
8        use wgt::TextureFormat as Tf;
9        use wgt::{AstcBlock, AstcChannel};
10        match format {
11            Tf::R8Unorm => F::R8_UNORM,
12            Tf::R8Snorm => F::R8_SNORM,
13            Tf::R8Uint => F::R8_UINT,
14            Tf::R8Sint => F::R8_SINT,
15            Tf::R16Uint => F::R16_UINT,
16            Tf::R16Sint => F::R16_SINT,
17            Tf::R16Unorm => F::R16_UNORM,
18            Tf::R16Snorm => F::R16_SNORM,
19            Tf::R16Float => F::R16_SFLOAT,
20            Tf::Rg8Unorm => F::R8G8_UNORM,
21            Tf::Rg8Snorm => F::R8G8_SNORM,
22            Tf::Rg8Uint => F::R8G8_UINT,
23            Tf::Rg8Sint => F::R8G8_SINT,
24            Tf::Rg16Unorm => F::R16G16_UNORM,
25            Tf::Rg16Snorm => F::R16G16_SNORM,
26            Tf::R32Uint => F::R32_UINT,
27            Tf::R32Sint => F::R32_SINT,
28            Tf::R32Float => F::R32_SFLOAT,
29            Tf::Rg16Uint => F::R16G16_UINT,
30            Tf::Rg16Sint => F::R16G16_SINT,
31            Tf::Rg16Float => F::R16G16_SFLOAT,
32            Tf::Rgba8Unorm => F::R8G8B8A8_UNORM,
33            Tf::Rgba8UnormSrgb => F::R8G8B8A8_SRGB,
34            Tf::Bgra8UnormSrgb => F::B8G8R8A8_SRGB,
35            Tf::Rgba8Snorm => F::R8G8B8A8_SNORM,
36            Tf::Bgra8Unorm => F::B8G8R8A8_UNORM,
37            Tf::Rgba8Uint => F::R8G8B8A8_UINT,
38            Tf::Rgba8Sint => F::R8G8B8A8_SINT,
39            Tf::Rgb10a2Uint => F::A2B10G10R10_UINT_PACK32,
40            Tf::Rgb10a2Unorm => F::A2B10G10R10_UNORM_PACK32,
41            Tf::Rg11b10Ufloat => F::B10G11R11_UFLOAT_PACK32,
42            Tf::R64Uint => F::R64_UINT,
43            Tf::Rg32Uint => F::R32G32_UINT,
44            Tf::Rg32Sint => F::R32G32_SINT,
45            Tf::Rg32Float => F::R32G32_SFLOAT,
46            Tf::Rgba16Uint => F::R16G16B16A16_UINT,
47            Tf::Rgba16Sint => F::R16G16B16A16_SINT,
48            Tf::Rgba16Unorm => F::R16G16B16A16_UNORM,
49            Tf::Rgba16Snorm => F::R16G16B16A16_SNORM,
50            Tf::Rgba16Float => F::R16G16B16A16_SFLOAT,
51            Tf::Rgba32Uint => F::R32G32B32A32_UINT,
52            Tf::Rgba32Sint => F::R32G32B32A32_SINT,
53            Tf::Rgba32Float => F::R32G32B32A32_SFLOAT,
54            Tf::Depth32Float => F::D32_SFLOAT,
55            Tf::Depth32FloatStencil8 => F::D32_SFLOAT_S8_UINT,
56            Tf::Depth24Plus => {
57                if self.texture_d24 {
58                    F::X8_D24_UNORM_PACK32
59                } else {
60                    F::D32_SFLOAT
61                }
62            }
63            Tf::Depth24PlusStencil8 => {
64                if self.texture_d24_s8 {
65                    F::D24_UNORM_S8_UINT
66                } else {
67                    F::D32_SFLOAT_S8_UINT
68                }
69            }
70            Tf::Stencil8 => {
71                if self.texture_s8 {
72                    F::S8_UINT
73                } else if self.texture_d24_s8 {
74                    F::D24_UNORM_S8_UINT
75                } else {
76                    F::D32_SFLOAT_S8_UINT
77                }
78            }
79            Tf::Depth16Unorm => F::D16_UNORM,
80            Tf::NV12 => F::G8_B8R8_2PLANE_420_UNORM,
81            Tf::P010 => F::G10X6_B10X6R10X6_2PLANE_420_UNORM_3PACK16,
82            Tf::Rgb9e5Ufloat => F::E5B9G9R9_UFLOAT_PACK32,
83            Tf::Bc1RgbaUnorm => F::BC1_RGBA_UNORM_BLOCK,
84            Tf::Bc1RgbaUnormSrgb => F::BC1_RGBA_SRGB_BLOCK,
85            Tf::Bc2RgbaUnorm => F::BC2_UNORM_BLOCK,
86            Tf::Bc2RgbaUnormSrgb => F::BC2_SRGB_BLOCK,
87            Tf::Bc3RgbaUnorm => F::BC3_UNORM_BLOCK,
88            Tf::Bc3RgbaUnormSrgb => F::BC3_SRGB_BLOCK,
89            Tf::Bc4RUnorm => F::BC4_UNORM_BLOCK,
90            Tf::Bc4RSnorm => F::BC4_SNORM_BLOCK,
91            Tf::Bc5RgUnorm => F::BC5_UNORM_BLOCK,
92            Tf::Bc5RgSnorm => F::BC5_SNORM_BLOCK,
93            Tf::Bc6hRgbUfloat => F::BC6H_UFLOAT_BLOCK,
94            Tf::Bc6hRgbFloat => F::BC6H_SFLOAT_BLOCK,
95            Tf::Bc7RgbaUnorm => F::BC7_UNORM_BLOCK,
96            Tf::Bc7RgbaUnormSrgb => F::BC7_SRGB_BLOCK,
97            Tf::Etc2Rgb8Unorm => F::ETC2_R8G8B8_UNORM_BLOCK,
98            Tf::Etc2Rgb8UnormSrgb => F::ETC2_R8G8B8_SRGB_BLOCK,
99            Tf::Etc2Rgb8A1Unorm => F::ETC2_R8G8B8A1_UNORM_BLOCK,
100            Tf::Etc2Rgb8A1UnormSrgb => F::ETC2_R8G8B8A1_SRGB_BLOCK,
101            Tf::Etc2Rgba8Unorm => F::ETC2_R8G8B8A8_UNORM_BLOCK,
102            Tf::Etc2Rgba8UnormSrgb => F::ETC2_R8G8B8A8_SRGB_BLOCK,
103            Tf::EacR11Unorm => F::EAC_R11_UNORM_BLOCK,
104            Tf::EacR11Snorm => F::EAC_R11_SNORM_BLOCK,
105            Tf::EacRg11Unorm => F::EAC_R11G11_UNORM_BLOCK,
106            Tf::EacRg11Snorm => F::EAC_R11G11_SNORM_BLOCK,
107            Tf::Astc { block, channel } => match channel {
108                AstcChannel::Unorm => match block {
109                    AstcBlock::B4x4 => F::ASTC_4X4_UNORM_BLOCK,
110                    AstcBlock::B5x4 => F::ASTC_5X4_UNORM_BLOCK,
111                    AstcBlock::B5x5 => F::ASTC_5X5_UNORM_BLOCK,
112                    AstcBlock::B6x5 => F::ASTC_6X5_UNORM_BLOCK,
113                    AstcBlock::B6x6 => F::ASTC_6X6_UNORM_BLOCK,
114                    AstcBlock::B8x5 => F::ASTC_8X5_UNORM_BLOCK,
115                    AstcBlock::B8x6 => F::ASTC_8X6_UNORM_BLOCK,
116                    AstcBlock::B8x8 => F::ASTC_8X8_UNORM_BLOCK,
117                    AstcBlock::B10x5 => F::ASTC_10X5_UNORM_BLOCK,
118                    AstcBlock::B10x6 => F::ASTC_10X6_UNORM_BLOCK,
119                    AstcBlock::B10x8 => F::ASTC_10X8_UNORM_BLOCK,
120                    AstcBlock::B10x10 => F::ASTC_10X10_UNORM_BLOCK,
121                    AstcBlock::B12x10 => F::ASTC_12X10_UNORM_BLOCK,
122                    AstcBlock::B12x12 => F::ASTC_12X12_UNORM_BLOCK,
123                },
124                AstcChannel::UnormSrgb => match block {
125                    AstcBlock::B4x4 => F::ASTC_4X4_SRGB_BLOCK,
126                    AstcBlock::B5x4 => F::ASTC_5X4_SRGB_BLOCK,
127                    AstcBlock::B5x5 => F::ASTC_5X5_SRGB_BLOCK,
128                    AstcBlock::B6x5 => F::ASTC_6X5_SRGB_BLOCK,
129                    AstcBlock::B6x6 => F::ASTC_6X6_SRGB_BLOCK,
130                    AstcBlock::B8x5 => F::ASTC_8X5_SRGB_BLOCK,
131                    AstcBlock::B8x6 => F::ASTC_8X6_SRGB_BLOCK,
132                    AstcBlock::B8x8 => F::ASTC_8X8_SRGB_BLOCK,
133                    AstcBlock::B10x5 => F::ASTC_10X5_SRGB_BLOCK,
134                    AstcBlock::B10x6 => F::ASTC_10X6_SRGB_BLOCK,
135                    AstcBlock::B10x8 => F::ASTC_10X8_SRGB_BLOCK,
136                    AstcBlock::B10x10 => F::ASTC_10X10_SRGB_BLOCK,
137                    AstcBlock::B12x10 => F::ASTC_12X10_SRGB_BLOCK,
138                    AstcBlock::B12x12 => F::ASTC_12X12_SRGB_BLOCK,
139                },
140                AstcChannel::Hdr => match block {
141                    AstcBlock::B4x4 => F::ASTC_4X4_SFLOAT_BLOCK_EXT,
142                    AstcBlock::B5x4 => F::ASTC_5X4_SFLOAT_BLOCK_EXT,
143                    AstcBlock::B5x5 => F::ASTC_5X5_SFLOAT_BLOCK_EXT,
144                    AstcBlock::B6x5 => F::ASTC_6X5_SFLOAT_BLOCK_EXT,
145                    AstcBlock::B6x6 => F::ASTC_6X6_SFLOAT_BLOCK_EXT,
146                    AstcBlock::B8x5 => F::ASTC_8X5_SFLOAT_BLOCK_EXT,
147                    AstcBlock::B8x6 => F::ASTC_8X6_SFLOAT_BLOCK_EXT,
148                    AstcBlock::B8x8 => F::ASTC_8X8_SFLOAT_BLOCK_EXT,
149                    AstcBlock::B10x5 => F::ASTC_10X5_SFLOAT_BLOCK_EXT,
150                    AstcBlock::B10x6 => F::ASTC_10X6_SFLOAT_BLOCK_EXT,
151                    AstcBlock::B10x8 => F::ASTC_10X8_SFLOAT_BLOCK_EXT,
152                    AstcBlock::B10x10 => F::ASTC_10X10_SFLOAT_BLOCK_EXT,
153                    AstcBlock::B12x10 => F::ASTC_12X10_SFLOAT_BLOCK_EXT,
154                    AstcBlock::B12x12 => F::ASTC_12X12_SFLOAT_BLOCK_EXT,
155                },
156            },
157        }
158    }
159}
160
161pub fn map_vk_surface_formats(
162    sf: vk::SurfaceFormatKHR,
163) -> Option<(wgt::TextureFormat, wgt::SurfaceColorSpace)> {
164    use ash::vk::Format as F;
165    use wgt::TextureFormat as Tf;
166    // Format list we care about pulled from https://vulkan.gpuinfo.org/listsurfaceformats.php.
167    let format = match sf.format {
168        F::B8G8R8A8_UNORM => Tf::Bgra8Unorm,
169        F::B8G8R8A8_SRGB => Tf::Bgra8UnormSrgb,
170        F::R8G8B8A8_SNORM => Tf::Rgba8Snorm,
171        F::R8G8B8A8_UNORM => Tf::Rgba8Unorm,
172        F::R8G8B8A8_SRGB => Tf::Rgba8UnormSrgb,
173        F::R16G16B16A16_SNORM => Tf::Rgba16Snorm,
174        F::R16G16B16A16_UNORM => Tf::Rgba16Unorm,
175        F::R16G16B16A16_SFLOAT => Tf::Rgba16Float,
176        F::A2B10G10R10_UNORM_PACK32 => Tf::Rgb10a2Unorm,
177        _ => return None,
178    };
179    let color_space = map_vk_color_space(sf.color_space)?;
180    Some((format, color_space))
181}
182
183pub fn map_vk_color_space(color_space: vk::ColorSpaceKHR) -> Option<wgt::SurfaceColorSpace> {
184    use wgt::SurfaceColorSpace as Scs;
185    Some(match color_space {
186        vk::ColorSpaceKHR::SRGB_NONLINEAR => Scs::Srgb,
187        vk::ColorSpaceKHR::EXTENDED_SRGB_LINEAR_EXT => Scs::ExtendedSrgbLinear,
188        vk::ColorSpaceKHR::EXTENDED_SRGB_NONLINEAR_EXT => Scs::ExtendedSrgb,
189        vk::ColorSpaceKHR::DISPLAY_P3_NONLINEAR_EXT => Scs::DisplayP3,
190        vk::ColorSpaceKHR::HDR10_ST2084_EXT => Scs::Bt2100Pq,
191        vk::ColorSpaceKHR::HDR10_HLG_EXT => Scs::Bt2100Hlg,
192        _ => return None,
193    })
194}
195
196pub fn map_surface_color_space(color_space: wgt::SurfaceColorSpace) -> vk::ColorSpaceKHR {
197    use wgt::SurfaceColorSpace as Scs;
198    match color_space {
199        Scs::Auto => unreachable!("wgpu-core resolves `Auto` before configuring the surface"),
200        Scs::Srgb => vk::ColorSpaceKHR::SRGB_NONLINEAR,
201        Scs::ExtendedSrgbLinear => vk::ColorSpaceKHR::EXTENDED_SRGB_LINEAR_EXT,
202        Scs::ExtendedSrgb => vk::ColorSpaceKHR::EXTENDED_SRGB_NONLINEAR_EXT,
203        Scs::DisplayP3 => vk::ColorSpaceKHR::DISPLAY_P3_NONLINEAR_EXT,
204        Scs::Bt2100Pq => vk::ColorSpaceKHR::HDR10_ST2084_EXT,
205        Scs::Bt2100Hlg => vk::ColorSpaceKHR::HDR10_HLG_EXT,
206        Scs::ExtendedDisplayP3 => {
207            unreachable!("`ExtendedDisplayP3` is never reported in the Vulkan surface capabilities")
208        }
209    }
210}
211
212impl crate::Attachment<'_, super::TextureView> {
213    pub(super) fn make_attachment_key(&self, ops: crate::AttachmentOps) -> super::AttachmentKey {
214        super::AttachmentKey {
215            format: self.view.raw_format,
216            layout: derive_image_layout(self.usage, self.view.format),
217            ops,
218        }
219    }
220}
221
222impl crate::ColorAttachment<'_, super::TextureView> {
223    pub(super) unsafe fn make_vk_clear_color(&self) -> vk::ClearColorValue {
224        let cv = &self.clear_value;
225        match self.target.view.format.sample_type(None, None).unwrap() {
226            wgt::TextureSampleType::Float { .. } => vk::ClearColorValue {
227                float32: [cv.r as f32, cv.g as f32, cv.b as f32, cv.a as f32],
228            },
229            wgt::TextureSampleType::Sint => vk::ClearColorValue {
230                int32: [cv.r as i32, cv.g as i32, cv.b as i32, cv.a as i32],
231            },
232            wgt::TextureSampleType::Uint => vk::ClearColorValue {
233                uint32: [cv.r as u32, cv.g as u32, cv.b as u32, cv.a as u32],
234            },
235            wgt::TextureSampleType::Depth => unreachable!(),
236        }
237    }
238}
239
240pub fn derive_image_layout(usage: wgt::TextureUses, format: wgt::TextureFormat) -> vk::ImageLayout {
241    // Note: depth textures are always sampled with RODS layout
242    let is_color = !format.is_depth_stencil_format();
243    match usage {
244        wgt::TextureUses::UNINITIALIZED => vk::ImageLayout::UNDEFINED,
245        wgt::TextureUses::COPY_SRC => vk::ImageLayout::TRANSFER_SRC_OPTIMAL,
246        wgt::TextureUses::COPY_DST => vk::ImageLayout::TRANSFER_DST_OPTIMAL,
247        wgt::TextureUses::RESOURCE if is_color => vk::ImageLayout::SHADER_READ_ONLY_OPTIMAL,
248        wgt::TextureUses::COLOR_TARGET => vk::ImageLayout::COLOR_ATTACHMENT_OPTIMAL,
249        wgt::TextureUses::DEPTH_STENCIL_WRITE => vk::ImageLayout::DEPTH_STENCIL_ATTACHMENT_OPTIMAL,
250        _ => {
251            if usage == wgt::TextureUses::PRESENT {
252                vk::ImageLayout::PRESENT_SRC_KHR
253            } else if is_color {
254                vk::ImageLayout::GENERAL
255            } else {
256                vk::ImageLayout::DEPTH_STENCIL_READ_ONLY_OPTIMAL
257            }
258        }
259    }
260}
261
262pub fn map_texture_usage(usage: wgt::TextureUses) -> vk::ImageUsageFlags {
263    let mut flags = vk::ImageUsageFlags::empty();
264    if usage.contains(wgt::TextureUses::COPY_SRC) {
265        flags |= vk::ImageUsageFlags::TRANSFER_SRC;
266    }
267    if usage.contains(wgt::TextureUses::COPY_DST) {
268        flags |= vk::ImageUsageFlags::TRANSFER_DST;
269    }
270    if usage.contains(wgt::TextureUses::RESOURCE) {
271        flags |= vk::ImageUsageFlags::SAMPLED;
272    }
273    if usage.contains(wgt::TextureUses::COLOR_TARGET) {
274        flags |= vk::ImageUsageFlags::COLOR_ATTACHMENT;
275    }
276    if usage
277        .intersects(wgt::TextureUses::DEPTH_STENCIL_READ | wgt::TextureUses::DEPTH_STENCIL_WRITE)
278    {
279        flags |= vk::ImageUsageFlags::DEPTH_STENCIL_ATTACHMENT;
280    }
281    if usage.intersects(
282        wgt::TextureUses::STORAGE_READ_ONLY
283            | wgt::TextureUses::STORAGE_WRITE_ONLY
284            | wgt::TextureUses::STORAGE_READ_WRITE
285            | wgt::TextureUses::STORAGE_ATOMIC,
286    ) {
287        flags |= vk::ImageUsageFlags::STORAGE;
288    }
289    if usage.contains(wgt::TextureUses::TRANSIENT) {
290        flags |= vk::ImageUsageFlags::TRANSIENT_ATTACHMENT;
291    }
292    flags
293}
294
295pub fn map_texture_usage_to_barrier(
296    usage: wgt::TextureUses,
297) -> (vk::PipelineStageFlags, vk::AccessFlags) {
298    let mut stages = vk::PipelineStageFlags::empty();
299    let mut access = vk::AccessFlags::empty();
300    let shader_stages = vk::PipelineStageFlags::VERTEX_SHADER
301        | vk::PipelineStageFlags::FRAGMENT_SHADER
302        | vk::PipelineStageFlags::COMPUTE_SHADER;
303
304    if usage.contains(wgt::TextureUses::COPY_SRC) {
305        stages |= vk::PipelineStageFlags::TRANSFER;
306        access |= vk::AccessFlags::TRANSFER_READ;
307    }
308    if usage.contains(wgt::TextureUses::COPY_DST) {
309        stages |= vk::PipelineStageFlags::TRANSFER;
310        access |= vk::AccessFlags::TRANSFER_WRITE;
311    }
312    if usage.contains(wgt::TextureUses::RESOURCE) {
313        stages |= shader_stages;
314        access |= vk::AccessFlags::SHADER_READ;
315    }
316    if usage.contains(wgt::TextureUses::COLOR_TARGET) {
317        stages |= vk::PipelineStageFlags::COLOR_ATTACHMENT_OUTPUT;
318        access |= vk::AccessFlags::COLOR_ATTACHMENT_READ | vk::AccessFlags::COLOR_ATTACHMENT_WRITE;
319    }
320    if usage.intersects(wgt::TextureUses::DEPTH_STENCIL_READ) {
321        stages |= vk::PipelineStageFlags::EARLY_FRAGMENT_TESTS
322            | vk::PipelineStageFlags::LATE_FRAGMENT_TESTS;
323        access |= vk::AccessFlags::DEPTH_STENCIL_ATTACHMENT_READ;
324    }
325    if usage.intersects(wgt::TextureUses::DEPTH_STENCIL_WRITE) {
326        stages |= vk::PipelineStageFlags::EARLY_FRAGMENT_TESTS
327            | vk::PipelineStageFlags::LATE_FRAGMENT_TESTS;
328        access |= vk::AccessFlags::DEPTH_STENCIL_ATTACHMENT_READ
329            | vk::AccessFlags::DEPTH_STENCIL_ATTACHMENT_WRITE;
330    }
331    if usage.intersects(
332        wgt::TextureUses::STORAGE_READ_ONLY
333            | wgt::TextureUses::STORAGE_READ_WRITE
334            | wgt::TextureUses::STORAGE_ATOMIC,
335    ) {
336        stages |= shader_stages;
337        access |= vk::AccessFlags::SHADER_READ;
338    }
339    if usage.intersects(
340        wgt::TextureUses::STORAGE_WRITE_ONLY
341            | wgt::TextureUses::STORAGE_READ_WRITE
342            | wgt::TextureUses::STORAGE_ATOMIC,
343    ) {
344        stages |= shader_stages;
345        access |= vk::AccessFlags::SHADER_WRITE;
346    }
347
348    if usage == wgt::TextureUses::UNINITIALIZED || usage == wgt::TextureUses::PRESENT {
349        (
350            vk::PipelineStageFlags::TOP_OF_PIPE,
351            vk::AccessFlags::empty(),
352        )
353    } else {
354        (stages, access)
355    }
356}
357
358pub fn map_vk_image_usage(usage: vk::ImageUsageFlags) -> wgt::TextureUses {
359    let mut bits = wgt::TextureUses::empty();
360    if usage.contains(vk::ImageUsageFlags::TRANSFER_SRC) {
361        bits |= wgt::TextureUses::COPY_SRC;
362    }
363    if usage.contains(vk::ImageUsageFlags::TRANSFER_DST) {
364        bits |= wgt::TextureUses::COPY_DST;
365    }
366    if usage.contains(vk::ImageUsageFlags::SAMPLED) {
367        bits |= wgt::TextureUses::RESOURCE;
368    }
369    if usage.contains(vk::ImageUsageFlags::COLOR_ATTACHMENT) {
370        bits |= wgt::TextureUses::COLOR_TARGET;
371    }
372    if usage.contains(vk::ImageUsageFlags::DEPTH_STENCIL_ATTACHMENT) {
373        bits |= wgt::TextureUses::DEPTH_STENCIL_READ | wgt::TextureUses::DEPTH_STENCIL_WRITE;
374    }
375    if usage.contains(vk::ImageUsageFlags::STORAGE) {
376        bits |= wgt::TextureUses::STORAGE_READ_ONLY
377            | wgt::TextureUses::STORAGE_WRITE_ONLY
378            | wgt::TextureUses::STORAGE_READ_WRITE
379            | wgt::TextureUses::STORAGE_ATOMIC;
380    }
381    if usage.contains(vk::ImageUsageFlags::TRANSIENT_ATTACHMENT) {
382        bits |= wgt::TextureUses::TRANSIENT;
383    }
384    bits
385}
386
387pub fn map_texture_dimension(dim: wgt::TextureDimension) -> vk::ImageType {
388    match dim {
389        wgt::TextureDimension::D1 => vk::ImageType::TYPE_1D,
390        wgt::TextureDimension::D2 => vk::ImageType::TYPE_2D,
391        wgt::TextureDimension::D3 => vk::ImageType::TYPE_3D,
392    }
393}
394
395pub fn map_index_format(index_format: wgt::IndexFormat) -> vk::IndexType {
396    match index_format {
397        wgt::IndexFormat::Uint16 => vk::IndexType::UINT16,
398        wgt::IndexFormat::Uint32 => vk::IndexType::UINT32,
399    }
400}
401
402pub fn map_vertex_format(vertex_format: wgt::VertexFormat) -> vk::Format {
403    use wgt::VertexFormat as Vf;
404    match vertex_format {
405        Vf::Uint8 => vk::Format::R8_UINT,
406        Vf::Uint8x2 => vk::Format::R8G8_UINT,
407        Vf::Uint8x4 => vk::Format::R8G8B8A8_UINT,
408        Vf::Sint8 => vk::Format::R8_SINT,
409        Vf::Sint8x2 => vk::Format::R8G8_SINT,
410        Vf::Sint8x4 => vk::Format::R8G8B8A8_SINT,
411        Vf::Unorm8 => vk::Format::R8_UNORM,
412        Vf::Unorm8x2 => vk::Format::R8G8_UNORM,
413        Vf::Unorm8x4 => vk::Format::R8G8B8A8_UNORM,
414        Vf::Snorm8 => vk::Format::R8_SNORM,
415        Vf::Snorm8x2 => vk::Format::R8G8_SNORM,
416        Vf::Snorm8x4 => vk::Format::R8G8B8A8_SNORM,
417        Vf::Uint16 => vk::Format::R16_UINT,
418        Vf::Uint16x2 => vk::Format::R16G16_UINT,
419        Vf::Uint16x4 => vk::Format::R16G16B16A16_UINT,
420        Vf::Sint16 => vk::Format::R16_SINT,
421        Vf::Sint16x2 => vk::Format::R16G16_SINT,
422        Vf::Sint16x4 => vk::Format::R16G16B16A16_SINT,
423        Vf::Unorm16 => vk::Format::R16_UNORM,
424        Vf::Unorm16x2 => vk::Format::R16G16_UNORM,
425        Vf::Unorm16x4 => vk::Format::R16G16B16A16_UNORM,
426        Vf::Snorm16 => vk::Format::R16_SNORM,
427        Vf::Snorm16x2 => vk::Format::R16G16_SNORM,
428        Vf::Snorm16x4 => vk::Format::R16G16B16A16_SNORM,
429        Vf::Float16 => vk::Format::R16_SFLOAT,
430        Vf::Float16x2 => vk::Format::R16G16_SFLOAT,
431        Vf::Float16x4 => vk::Format::R16G16B16A16_SFLOAT,
432        Vf::Float32 => vk::Format::R32_SFLOAT,
433        Vf::Float32x2 => vk::Format::R32G32_SFLOAT,
434        Vf::Float32x3 => vk::Format::R32G32B32_SFLOAT,
435        Vf::Float32x4 => vk::Format::R32G32B32A32_SFLOAT,
436        Vf::Uint32 => vk::Format::R32_UINT,
437        Vf::Uint32x2 => vk::Format::R32G32_UINT,
438        Vf::Uint32x3 => vk::Format::R32G32B32_UINT,
439        Vf::Uint32x4 => vk::Format::R32G32B32A32_UINT,
440        Vf::Sint32 => vk::Format::R32_SINT,
441        Vf::Sint32x2 => vk::Format::R32G32_SINT,
442        Vf::Sint32x3 => vk::Format::R32G32B32_SINT,
443        Vf::Sint32x4 => vk::Format::R32G32B32A32_SINT,
444        Vf::Float64 => vk::Format::R64_SFLOAT,
445        Vf::Float64x2 => vk::Format::R64G64_SFLOAT,
446        Vf::Float64x3 => vk::Format::R64G64B64_SFLOAT,
447        Vf::Float64x4 => vk::Format::R64G64B64A64_SFLOAT,
448        Vf::Unorm10_10_10_2 => vk::Format::A2B10G10R10_UNORM_PACK32,
449        Vf::Unorm8x4Bgra => vk::Format::B8G8R8A8_UNORM,
450    }
451}
452
453pub fn map_aspects(aspects: crate::FormatAspects) -> vk::ImageAspectFlags {
454    let mut flags = vk::ImageAspectFlags::empty();
455    if aspects.contains(crate::FormatAspects::COLOR) {
456        flags |= vk::ImageAspectFlags::COLOR;
457    }
458    if aspects.contains(crate::FormatAspects::DEPTH) {
459        flags |= vk::ImageAspectFlags::DEPTH;
460    }
461    if aspects.contains(crate::FormatAspects::STENCIL) {
462        flags |= vk::ImageAspectFlags::STENCIL;
463    }
464    if aspects.contains(crate::FormatAspects::PLANE_0) {
465        flags |= vk::ImageAspectFlags::PLANE_0;
466    }
467    if aspects.contains(crate::FormatAspects::PLANE_1) {
468        flags |= vk::ImageAspectFlags::PLANE_1;
469    }
470    if aspects.contains(crate::FormatAspects::PLANE_2) {
471        flags |= vk::ImageAspectFlags::PLANE_2;
472    }
473    flags
474}
475
476pub fn map_attachment_ops(
477    op: crate::AttachmentOps,
478) -> (vk::AttachmentLoadOp, vk::AttachmentStoreOp) {
479    let load_op = if op.contains(crate::AttachmentOps::LOAD) {
480        vk::AttachmentLoadOp::LOAD
481    } else if op.contains(crate::AttachmentOps::LOAD_DONT_CARE) {
482        vk::AttachmentLoadOp::DONT_CARE
483    } else if op.contains(crate::AttachmentOps::LOAD_CLEAR) {
484        vk::AttachmentLoadOp::CLEAR
485    } else {
486        unreachable!()
487    };
488    let store_op = if op.contains(crate::AttachmentOps::STORE) {
489        vk::AttachmentStoreOp::STORE
490    } else if op.contains(crate::AttachmentOps::STORE_DISCARD) {
491        vk::AttachmentStoreOp::DONT_CARE
492    } else {
493        unreachable!()
494    };
495    (load_op, store_op)
496}
497
498pub fn map_present_mode(mode: wgt::PresentMode) -> vk::PresentModeKHR {
499    match mode {
500        wgt::PresentMode::Immediate => vk::PresentModeKHR::IMMEDIATE,
501        wgt::PresentMode::Mailbox => vk::PresentModeKHR::MAILBOX,
502        wgt::PresentMode::Fifo => vk::PresentModeKHR::FIFO,
503        wgt::PresentMode::FifoRelaxed => vk::PresentModeKHR::FIFO_RELAXED,
504        wgt::PresentMode::AutoNoVsync | wgt::PresentMode::AutoVsync => {
505            unreachable!("Cannot create swapchain with Auto PresentationMode")
506        }
507    }
508}
509
510pub fn map_vk_present_mode(mode: vk::PresentModeKHR) -> Option<wgt::PresentMode> {
511    // Not exposed in Ash yet.
512    const FIFO_LATEST_READY: vk::PresentModeKHR = vk::PresentModeKHR::from_raw(1_000_361_000);
513
514    // See https://registry.khronos.org/vulkan/specs/latest/man/html/VkPresentModeKHR.html
515    match mode {
516        vk::PresentModeKHR::IMMEDIATE => Some(wgt::PresentMode::Immediate),
517        vk::PresentModeKHR::MAILBOX => Some(wgt::PresentMode::Mailbox),
518        vk::PresentModeKHR::FIFO => Some(wgt::PresentMode::Fifo),
519        vk::PresentModeKHR::FIFO_RELAXED => Some(wgt::PresentMode::FifoRelaxed),
520
521        // Modes that aren't exposed yet.
522        vk::PresentModeKHR::SHARED_DEMAND_REFRESH => None,
523        vk::PresentModeKHR::SHARED_CONTINUOUS_REFRESH => None,
524        FIFO_LATEST_READY => None,
525
526        _ => {
527            log::debug!("Unrecognized present mode {mode:?}");
528            None
529        }
530    }
531}
532
533pub fn map_composite_alpha_mode(mode: wgt::CompositeAlphaMode) -> vk::CompositeAlphaFlagsKHR {
534    match mode {
535        wgt::CompositeAlphaMode::Opaque => vk::CompositeAlphaFlagsKHR::OPAQUE,
536        wgt::CompositeAlphaMode::PreMultiplied => vk::CompositeAlphaFlagsKHR::PRE_MULTIPLIED,
537        wgt::CompositeAlphaMode::PostMultiplied => vk::CompositeAlphaFlagsKHR::POST_MULTIPLIED,
538        wgt::CompositeAlphaMode::Inherit => vk::CompositeAlphaFlagsKHR::INHERIT,
539        wgt::CompositeAlphaMode::Auto => unreachable!(),
540    }
541}
542
543pub fn map_vk_composite_alpha(flags: vk::CompositeAlphaFlagsKHR) -> Vec<wgt::CompositeAlphaMode> {
544    let mut modes = Vec::new();
545    if flags.contains(vk::CompositeAlphaFlagsKHR::OPAQUE) {
546        modes.push(wgt::CompositeAlphaMode::Opaque);
547    }
548    if flags.contains(vk::CompositeAlphaFlagsKHR::PRE_MULTIPLIED) {
549        modes.push(wgt::CompositeAlphaMode::PreMultiplied);
550    }
551    if flags.contains(vk::CompositeAlphaFlagsKHR::POST_MULTIPLIED) {
552        modes.push(wgt::CompositeAlphaMode::PostMultiplied);
553    }
554    if flags.contains(vk::CompositeAlphaFlagsKHR::INHERIT) {
555        modes.push(wgt::CompositeAlphaMode::Inherit);
556    }
557    modes
558}
559
560pub fn map_buffer_usage(usage: wgt::BufferUses) -> vk::BufferUsageFlags {
561    let mut flags = vk::BufferUsageFlags::empty();
562    if usage.contains(wgt::BufferUses::COPY_SRC) {
563        flags |= vk::BufferUsageFlags::TRANSFER_SRC;
564    }
565    if usage.contains(wgt::BufferUses::COPY_DST) {
566        flags |= vk::BufferUsageFlags::TRANSFER_DST;
567    }
568    if usage.contains(wgt::BufferUses::UNIFORM) {
569        flags |= vk::BufferUsageFlags::UNIFORM_BUFFER;
570    }
571    if usage.intersects(wgt::BufferUses::STORAGE_READ_ONLY | wgt::BufferUses::STORAGE_READ_WRITE) {
572        flags |= vk::BufferUsageFlags::STORAGE_BUFFER;
573    }
574    if usage.contains(wgt::BufferUses::INDEX) {
575        flags |= vk::BufferUsageFlags::INDEX_BUFFER;
576    }
577    if usage.contains(wgt::BufferUses::VERTEX) {
578        flags |= vk::BufferUsageFlags::VERTEX_BUFFER;
579    }
580    if usage.contains(wgt::BufferUses::INDIRECT) {
581        flags |= vk::BufferUsageFlags::INDIRECT_BUFFER;
582    }
583    if usage.contains(wgt::BufferUses::ACCELERATION_STRUCTURE_SCRATCH) {
584        flags |= vk::BufferUsageFlags::STORAGE_BUFFER | vk::BufferUsageFlags::SHADER_DEVICE_ADDRESS;
585    }
586    if usage.intersects(
587        wgt::BufferUses::BOTTOM_LEVEL_ACCELERATION_STRUCTURE_INPUT
588            | wgt::BufferUses::TOP_LEVEL_ACCELERATION_STRUCTURE_INPUT,
589    ) {
590        flags |= vk::BufferUsageFlags::ACCELERATION_STRUCTURE_BUILD_INPUT_READ_ONLY_KHR
591            | vk::BufferUsageFlags::SHADER_DEVICE_ADDRESS;
592    }
593    if usage.intersects(wgt::BufferUses::ACCELERATION_STRUCTURE_QUERY) {
594        flags |= vk::BufferUsageFlags::TRANSFER_DST;
595    }
596    if usage.intersects(wgt::BufferUses::RAY_TRACING_PIPELINE_SHADER_DATA) {
597        flags |= vk::BufferUsageFlags::SHADER_BINDING_TABLE_KHR
598            | vk::BufferUsageFlags::SHADER_DEVICE_ADDRESS;
599    }
600    flags
601}
602
603pub fn map_buffer_usage_to_barrier(
604    usage: wgt::BufferUses,
605) -> (vk::PipelineStageFlags, vk::AccessFlags) {
606    let mut stages = vk::PipelineStageFlags::empty();
607    let mut access = vk::AccessFlags::empty();
608    let shader_stages = vk::PipelineStageFlags::VERTEX_SHADER
609        | vk::PipelineStageFlags::FRAGMENT_SHADER
610        | vk::PipelineStageFlags::COMPUTE_SHADER;
611
612    if usage.contains(wgt::BufferUses::MAP_READ) {
613        stages |= vk::PipelineStageFlags::HOST;
614        access |= vk::AccessFlags::HOST_READ;
615    }
616    if usage.contains(wgt::BufferUses::MAP_WRITE) {
617        stages |= vk::PipelineStageFlags::HOST;
618        access |= vk::AccessFlags::HOST_WRITE;
619    }
620    if usage.contains(wgt::BufferUses::COPY_SRC) {
621        stages |= vk::PipelineStageFlags::TRANSFER;
622        access |= vk::AccessFlags::TRANSFER_READ;
623    }
624    if usage.contains(wgt::BufferUses::COPY_DST) {
625        stages |= vk::PipelineStageFlags::TRANSFER;
626        access |= vk::AccessFlags::TRANSFER_WRITE;
627    }
628    if usage.contains(wgt::BufferUses::UNIFORM) {
629        stages |= shader_stages;
630        access |= vk::AccessFlags::UNIFORM_READ;
631    }
632    if usage.intersects(wgt::BufferUses::STORAGE_READ_ONLY) {
633        stages |= shader_stages;
634        access |= vk::AccessFlags::SHADER_READ;
635    }
636    if usage.intersects(wgt::BufferUses::STORAGE_READ_WRITE) {
637        stages |= shader_stages;
638        access |= vk::AccessFlags::SHADER_READ | vk::AccessFlags::SHADER_WRITE;
639    }
640    if usage.contains(wgt::BufferUses::INDEX) {
641        stages |= vk::PipelineStageFlags::VERTEX_INPUT;
642        access |= vk::AccessFlags::INDEX_READ;
643    }
644    if usage.contains(wgt::BufferUses::VERTEX) {
645        stages |= vk::PipelineStageFlags::VERTEX_INPUT;
646        access |= vk::AccessFlags::VERTEX_ATTRIBUTE_READ;
647    }
648    if usage.contains(wgt::BufferUses::INDIRECT) {
649        stages |= vk::PipelineStageFlags::DRAW_INDIRECT;
650        access |= vk::AccessFlags::INDIRECT_COMMAND_READ;
651    }
652    if usage.intersects(
653        wgt::BufferUses::BOTTOM_LEVEL_ACCELERATION_STRUCTURE_INPUT
654            | wgt::BufferUses::TOP_LEVEL_ACCELERATION_STRUCTURE_INPUT
655            | wgt::BufferUses::ACCELERATION_STRUCTURE_SCRATCH,
656    ) {
657        stages |= vk::PipelineStageFlags::ACCELERATION_STRUCTURE_BUILD_KHR;
658        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_READ_KHR
659            | vk::AccessFlags::ACCELERATION_STRUCTURE_WRITE_KHR
660            | vk::AccessFlags::SHADER_READ;
661    }
662    if usage.contains(wgt::BufferUses::ACCELERATION_STRUCTURE_QUERY) {
663        stages |= vk::PipelineStageFlags::TRANSFER;
664        access |= vk::AccessFlags::TRANSFER_WRITE;
665    }
666
667    (stages, access)
668}
669
670pub fn map_view_dimension(dim: wgt::TextureViewDimension) -> vk::ImageViewType {
671    match dim {
672        wgt::TextureViewDimension::D1 => vk::ImageViewType::TYPE_1D,
673        wgt::TextureViewDimension::D2 => vk::ImageViewType::TYPE_2D,
674        wgt::TextureViewDimension::D2Array => vk::ImageViewType::TYPE_2D_ARRAY,
675        wgt::TextureViewDimension::Cube => vk::ImageViewType::CUBE,
676        wgt::TextureViewDimension::CubeArray => vk::ImageViewType::CUBE_ARRAY,
677        wgt::TextureViewDimension::D3 => vk::ImageViewType::TYPE_3D,
678    }
679}
680
681pub fn map_copy_extent(extent: &crate::CopyExtent) -> vk::Extent3D {
682    vk::Extent3D {
683        width: extent.width,
684        height: extent.height,
685        depth: extent.depth,
686    }
687}
688
689pub fn map_subresource_range(
690    range: &wgt::ImageSubresourceRange,
691    format: wgt::TextureFormat,
692) -> vk::ImageSubresourceRange {
693    vk::ImageSubresourceRange {
694        aspect_mask: map_aspects(crate::FormatAspects::new(format, range.aspect)),
695        base_mip_level: range.base_mip_level,
696        level_count: range.mip_level_count.unwrap_or(vk::REMAINING_MIP_LEVELS),
697        base_array_layer: range.base_array_layer,
698        layer_count: range
699            .array_layer_count
700            .unwrap_or(vk::REMAINING_ARRAY_LAYERS),
701    }
702}
703
704// Special subresource range mapping for dealing with barriers
705// so that we account for the "hidden" depth aspect in emulated Stencil8.
706pub(super) fn map_subresource_range_combined_aspect(
707    range: &wgt::ImageSubresourceRange,
708    format: wgt::TextureFormat,
709    private_caps: &super::PrivateCapabilities,
710) -> vk::ImageSubresourceRange {
711    let mut range = map_subresource_range(range, format);
712    if !private_caps.texture_s8 && format == wgt::TextureFormat::Stencil8 {
713        range.aspect_mask |= vk::ImageAspectFlags::DEPTH;
714    }
715    range
716}
717
718pub fn map_subresource_layers(
719    base: &crate::TextureCopyBase,
720) -> (vk::ImageSubresourceLayers, vk::Offset3D) {
721    let offset = vk::Offset3D {
722        x: base.origin.x as i32,
723        y: base.origin.y as i32,
724        z: base.origin.z as i32,
725    };
726    let subresource = vk::ImageSubresourceLayers {
727        aspect_mask: map_aspects(base.aspect),
728        mip_level: base.mip_level,
729        base_array_layer: base.array_layer,
730        layer_count: 1,
731    };
732    (subresource, offset)
733}
734
735pub fn map_filter_mode(mode: wgt::FilterMode) -> vk::Filter {
736    match mode {
737        wgt::FilterMode::Nearest => vk::Filter::NEAREST,
738        wgt::FilterMode::Linear => vk::Filter::LINEAR,
739    }
740}
741
742pub fn map_mip_filter_mode(mode: wgt::MipmapFilterMode) -> vk::SamplerMipmapMode {
743    match mode {
744        wgt::MipmapFilterMode::Nearest => vk::SamplerMipmapMode::NEAREST,
745        wgt::MipmapFilterMode::Linear => vk::SamplerMipmapMode::LINEAR,
746    }
747}
748
749pub fn map_address_mode(mode: wgt::AddressMode) -> vk::SamplerAddressMode {
750    match mode {
751        wgt::AddressMode::ClampToEdge => vk::SamplerAddressMode::CLAMP_TO_EDGE,
752        wgt::AddressMode::Repeat => vk::SamplerAddressMode::REPEAT,
753        wgt::AddressMode::MirrorRepeat => vk::SamplerAddressMode::MIRRORED_REPEAT,
754        wgt::AddressMode::ClampToBorder => vk::SamplerAddressMode::CLAMP_TO_BORDER,
755        // wgt::AddressMode::MirrorClamp => vk::SamplerAddressMode::MIRROR_CLAMP_TO_EDGE,
756    }
757}
758
759pub fn map_border_color(border_color: wgt::SamplerBorderColor) -> vk::BorderColor {
760    match border_color {
761        wgt::SamplerBorderColor::TransparentBlack | wgt::SamplerBorderColor::Zero => {
762            vk::BorderColor::FLOAT_TRANSPARENT_BLACK
763        }
764        wgt::SamplerBorderColor::OpaqueBlack => vk::BorderColor::FLOAT_OPAQUE_BLACK,
765        wgt::SamplerBorderColor::OpaqueWhite => vk::BorderColor::FLOAT_OPAQUE_WHITE,
766    }
767}
768
769pub fn map_comparison(fun: wgt::CompareFunction) -> vk::CompareOp {
770    use wgt::CompareFunction as Cf;
771    match fun {
772        Cf::Never => vk::CompareOp::NEVER,
773        Cf::Less => vk::CompareOp::LESS,
774        Cf::LessEqual => vk::CompareOp::LESS_OR_EQUAL,
775        Cf::Equal => vk::CompareOp::EQUAL,
776        Cf::GreaterEqual => vk::CompareOp::GREATER_OR_EQUAL,
777        Cf::Greater => vk::CompareOp::GREATER,
778        Cf::NotEqual => vk::CompareOp::NOT_EQUAL,
779        Cf::Always => vk::CompareOp::ALWAYS,
780    }
781}
782
783pub fn map_shader_stage(stage: wgt::ShaderStages) -> vk::ShaderStageFlags {
784    let mut flags = vk::ShaderStageFlags::empty();
785    if stage.contains(wgt::ShaderStages::VERTEX) {
786        flags |= vk::ShaderStageFlags::VERTEX;
787    }
788    if stage.contains(wgt::ShaderStages::FRAGMENT) {
789        flags |= vk::ShaderStageFlags::FRAGMENT;
790    }
791    if stage.contains(wgt::ShaderStages::COMPUTE) {
792        flags |= vk::ShaderStageFlags::COMPUTE;
793    }
794    if stage.contains(wgt::ShaderStages::TASK) {
795        flags |= vk::ShaderStageFlags::TASK_EXT;
796    }
797    if stage.contains(wgt::ShaderStages::MESH) {
798        flags |= vk::ShaderStageFlags::MESH_EXT;
799    }
800    if stage.contains(wgt::ShaderStages::RAY_GENERATION) {
801        flags |= vk::ShaderStageFlags::RAYGEN_KHR;
802    }
803    if stage.contains(wgt::ShaderStages::MISS) {
804        flags |= vk::ShaderStageFlags::MISS_KHR;
805    }
806    if stage.contains(wgt::ShaderStages::ANY_HIT) {
807        flags |= vk::ShaderStageFlags::ANY_HIT_KHR;
808    }
809    if stage.contains(wgt::ShaderStages::CLOSEST_HIT) {
810        flags |= vk::ShaderStageFlags::CLOSEST_HIT_KHR;
811    }
812    flags
813}
814
815pub fn map_binding_type(ty: wgt::BindingType) -> vk::DescriptorType {
816    match ty {
817        wgt::BindingType::Buffer {
818            ty,
819            has_dynamic_offset,
820            ..
821        } => match ty {
822            wgt::BufferBindingType::Storage { .. } => match has_dynamic_offset {
823                true => vk::DescriptorType::STORAGE_BUFFER_DYNAMIC,
824                false => vk::DescriptorType::STORAGE_BUFFER,
825            },
826            wgt::BufferBindingType::Uniform => match has_dynamic_offset {
827                true => vk::DescriptorType::UNIFORM_BUFFER_DYNAMIC,
828                false => vk::DescriptorType::UNIFORM_BUFFER,
829            },
830        },
831        wgt::BindingType::Sampler { .. } => vk::DescriptorType::SAMPLER,
832        wgt::BindingType::Texture { .. } => vk::DescriptorType::SAMPLED_IMAGE,
833        wgt::BindingType::StorageTexture { .. } => vk::DescriptorType::STORAGE_IMAGE,
834        wgt::BindingType::AccelerationStructure { .. } => {
835            vk::DescriptorType::ACCELERATION_STRUCTURE_KHR
836        }
837        wgt::BindingType::ExternalTexture => unimplemented!(),
838    }
839}
840
841pub fn map_topology(topology: wgt::PrimitiveTopology) -> vk::PrimitiveTopology {
842    use wgt::PrimitiveTopology as Pt;
843    match topology {
844        Pt::PointList => vk::PrimitiveTopology::POINT_LIST,
845        Pt::LineList => vk::PrimitiveTopology::LINE_LIST,
846        Pt::LineStrip => vk::PrimitiveTopology::LINE_STRIP,
847        Pt::TriangleList => vk::PrimitiveTopology::TRIANGLE_LIST,
848        Pt::TriangleStrip => vk::PrimitiveTopology::TRIANGLE_STRIP,
849    }
850}
851
852pub fn map_polygon_mode(mode: wgt::PolygonMode) -> vk::PolygonMode {
853    match mode {
854        wgt::PolygonMode::Fill => vk::PolygonMode::FILL,
855        wgt::PolygonMode::Line => vk::PolygonMode::LINE,
856        wgt::PolygonMode::Point => vk::PolygonMode::POINT,
857    }
858}
859
860pub fn map_front_face(front_face: wgt::FrontFace) -> vk::FrontFace {
861    match front_face {
862        wgt::FrontFace::Cw => vk::FrontFace::CLOCKWISE,
863        wgt::FrontFace::Ccw => vk::FrontFace::COUNTER_CLOCKWISE,
864    }
865}
866
867pub fn map_cull_face(face: wgt::Face) -> vk::CullModeFlags {
868    match face {
869        wgt::Face::Front => vk::CullModeFlags::FRONT,
870        wgt::Face::Back => vk::CullModeFlags::BACK,
871    }
872}
873
874pub fn map_stencil_op(op: wgt::StencilOperation) -> vk::StencilOp {
875    use wgt::StencilOperation as So;
876    match op {
877        So::Keep => vk::StencilOp::KEEP,
878        So::Zero => vk::StencilOp::ZERO,
879        So::Replace => vk::StencilOp::REPLACE,
880        So::Invert => vk::StencilOp::INVERT,
881        So::IncrementClamp => vk::StencilOp::INCREMENT_AND_CLAMP,
882        So::IncrementWrap => vk::StencilOp::INCREMENT_AND_WRAP,
883        So::DecrementClamp => vk::StencilOp::DECREMENT_AND_CLAMP,
884        So::DecrementWrap => vk::StencilOp::DECREMENT_AND_WRAP,
885    }
886}
887
888pub fn map_stencil_face(
889    face: &wgt::StencilFaceState,
890    compare_mask: u32,
891    write_mask: u32,
892) -> vk::StencilOpState {
893    vk::StencilOpState {
894        fail_op: map_stencil_op(face.fail_op),
895        pass_op: map_stencil_op(face.pass_op),
896        depth_fail_op: map_stencil_op(face.depth_fail_op),
897        compare_op: map_comparison(face.compare),
898        compare_mask,
899        write_mask,
900        reference: 0,
901    }
902}
903
904fn map_blend_factor(factor: wgt::BlendFactor) -> vk::BlendFactor {
905    use wgt::BlendFactor as Bf;
906    match factor {
907        Bf::Zero => vk::BlendFactor::ZERO,
908        Bf::One => vk::BlendFactor::ONE,
909        Bf::Src => vk::BlendFactor::SRC_COLOR,
910        Bf::OneMinusSrc => vk::BlendFactor::ONE_MINUS_SRC_COLOR,
911        Bf::SrcAlpha => vk::BlendFactor::SRC_ALPHA,
912        Bf::OneMinusSrcAlpha => vk::BlendFactor::ONE_MINUS_SRC_ALPHA,
913        Bf::Dst => vk::BlendFactor::DST_COLOR,
914        Bf::OneMinusDst => vk::BlendFactor::ONE_MINUS_DST_COLOR,
915        Bf::DstAlpha => vk::BlendFactor::DST_ALPHA,
916        Bf::OneMinusDstAlpha => vk::BlendFactor::ONE_MINUS_DST_ALPHA,
917        Bf::SrcAlphaSaturated => vk::BlendFactor::SRC_ALPHA_SATURATE,
918        Bf::Constant => vk::BlendFactor::CONSTANT_COLOR,
919        Bf::OneMinusConstant => vk::BlendFactor::ONE_MINUS_CONSTANT_COLOR,
920        Bf::Src1 => vk::BlendFactor::SRC1_COLOR,
921        Bf::OneMinusSrc1 => vk::BlendFactor::ONE_MINUS_SRC1_COLOR,
922        Bf::Src1Alpha => vk::BlendFactor::SRC1_ALPHA,
923        Bf::OneMinusSrc1Alpha => vk::BlendFactor::ONE_MINUS_SRC1_ALPHA,
924    }
925}
926
927fn map_blend_op(operation: wgt::BlendOperation) -> vk::BlendOp {
928    use wgt::BlendOperation as Bo;
929    match operation {
930        Bo::Add => vk::BlendOp::ADD,
931        Bo::Subtract => vk::BlendOp::SUBTRACT,
932        Bo::ReverseSubtract => vk::BlendOp::REVERSE_SUBTRACT,
933        Bo::Min => vk::BlendOp::MIN,
934        Bo::Max => vk::BlendOp::MAX,
935    }
936}
937
938pub fn map_blend_component(
939    component: &wgt::BlendComponent,
940) -> (vk::BlendOp, vk::BlendFactor, vk::BlendFactor) {
941    let op = map_blend_op(component.operation);
942    let src = map_blend_factor(component.src_factor);
943    let dst = map_blend_factor(component.dst_factor);
944    (op, src, dst)
945}
946
947pub fn map_pipeline_statistics(
948    types: wgt::PipelineStatisticsTypes,
949) -> vk::QueryPipelineStatisticFlags {
950    use wgt::PipelineStatisticsTypes as Pst;
951    let mut flags = vk::QueryPipelineStatisticFlags::empty();
952    if types.contains(Pst::VERTEX_SHADER_INVOCATIONS) {
953        flags |= vk::QueryPipelineStatisticFlags::VERTEX_SHADER_INVOCATIONS;
954    }
955    if types.contains(Pst::CLIPPER_INVOCATIONS) {
956        flags |= vk::QueryPipelineStatisticFlags::CLIPPING_INVOCATIONS;
957    }
958    if types.contains(Pst::CLIPPER_PRIMITIVES_OUT) {
959        flags |= vk::QueryPipelineStatisticFlags::CLIPPING_PRIMITIVES;
960    }
961    if types.contains(Pst::FRAGMENT_SHADER_INVOCATIONS) {
962        flags |= vk::QueryPipelineStatisticFlags::FRAGMENT_SHADER_INVOCATIONS;
963    }
964    if types.contains(Pst::COMPUTE_SHADER_INVOCATIONS) {
965        flags |= vk::QueryPipelineStatisticFlags::COMPUTE_SHADER_INVOCATIONS;
966    }
967    flags
968}
969
970pub fn map_acceleration_structure_format(
971    format: crate::AccelerationStructureFormat,
972) -> vk::AccelerationStructureTypeKHR {
973    match format {
974        crate::AccelerationStructureFormat::TopLevel => vk::AccelerationStructureTypeKHR::TOP_LEVEL,
975        crate::AccelerationStructureFormat::BottomLevel => {
976            vk::AccelerationStructureTypeKHR::BOTTOM_LEVEL
977        }
978    }
979}
980
981pub fn map_acceleration_structure_build_mode(
982    format: crate::AccelerationStructureBuildMode,
983) -> vk::BuildAccelerationStructureModeKHR {
984    match format {
985        crate::AccelerationStructureBuildMode::Build => {
986            vk::BuildAccelerationStructureModeKHR::BUILD
987        }
988        crate::AccelerationStructureBuildMode::Update => {
989            vk::BuildAccelerationStructureModeKHR::UPDATE
990        }
991    }
992}
993
994pub fn map_acceleration_structure_flags(
995    flags: crate::AccelerationStructureBuildFlags,
996) -> vk::BuildAccelerationStructureFlagsKHR {
997    let mut vk_flags = vk::BuildAccelerationStructureFlagsKHR::empty();
998
999    if flags.contains(crate::AccelerationStructureBuildFlags::PREFER_FAST_TRACE) {
1000        vk_flags |= vk::BuildAccelerationStructureFlagsKHR::PREFER_FAST_TRACE;
1001    }
1002
1003    if flags.contains(crate::AccelerationStructureBuildFlags::PREFER_FAST_BUILD) {
1004        vk_flags |= vk::BuildAccelerationStructureFlagsKHR::PREFER_FAST_BUILD;
1005    }
1006
1007    if flags.contains(crate::AccelerationStructureBuildFlags::ALLOW_UPDATE) {
1008        vk_flags |= vk::BuildAccelerationStructureFlagsKHR::ALLOW_UPDATE;
1009    }
1010
1011    if flags.contains(crate::AccelerationStructureBuildFlags::LOW_MEMORY) {
1012        vk_flags |= vk::BuildAccelerationStructureFlagsKHR::LOW_MEMORY;
1013    }
1014
1015    if flags.contains(crate::AccelerationStructureBuildFlags::ALLOW_COMPACTION) {
1016        vk_flags |= vk::BuildAccelerationStructureFlagsKHR::ALLOW_COMPACTION
1017    }
1018
1019    if flags.contains(crate::AccelerationStructureBuildFlags::ALLOW_RAY_HIT_VERTEX_RETURN) {
1020        vk_flags |= vk::BuildAccelerationStructureFlagsKHR::ALLOW_DATA_ACCESS
1021    }
1022
1023    vk_flags
1024}
1025
1026pub fn map_acceleration_structure_geometry_flags(
1027    flags: crate::AccelerationStructureGeometryFlags,
1028) -> vk::GeometryFlagsKHR {
1029    let mut vk_flags = vk::GeometryFlagsKHR::empty();
1030
1031    if flags.contains(crate::AccelerationStructureGeometryFlags::OPAQUE) {
1032        vk_flags |= vk::GeometryFlagsKHR::OPAQUE;
1033    }
1034
1035    if flags.contains(crate::AccelerationStructureGeometryFlags::NO_DUPLICATE_ANY_HIT_INVOCATION) {
1036        vk_flags |= vk::GeometryFlagsKHR::NO_DUPLICATE_ANY_HIT_INVOCATION;
1037    }
1038
1039    vk_flags
1040}
1041
1042pub fn map_acceleration_structure_usage_to_barrier(
1043    usage: crate::AccelerationStructureUses,
1044    features: wgt::Features,
1045) -> (vk::PipelineStageFlags, vk::AccessFlags) {
1046    let mut stages = vk::PipelineStageFlags::empty();
1047    let mut access = vk::AccessFlags::empty();
1048
1049    if usage.contains(crate::AccelerationStructureUses::BUILD_INPUT) {
1050        stages |= vk::PipelineStageFlags::ACCELERATION_STRUCTURE_BUILD_KHR;
1051        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_READ_KHR;
1052    }
1053    if usage.contains(crate::AccelerationStructureUses::QUERY_INPUT) {
1054        stages |= vk::PipelineStageFlags::ACCELERATION_STRUCTURE_BUILD_KHR;
1055        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_READ_KHR;
1056    }
1057    if usage.contains(crate::AccelerationStructureUses::BUILD_OUTPUT) {
1058        stages |= vk::PipelineStageFlags::ACCELERATION_STRUCTURE_BUILD_KHR;
1059        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_WRITE_KHR;
1060    }
1061    if usage.contains(crate::AccelerationStructureUses::SHADER_INPUT)
1062        && features.contains(wgt::Features::EXPERIMENTAL_RAY_QUERY)
1063    {
1064        stages |= vk::PipelineStageFlags::VERTEX_SHADER
1065            | vk::PipelineStageFlags::FRAGMENT_SHADER
1066            | vk::PipelineStageFlags::COMPUTE_SHADER;
1067        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_READ_KHR;
1068    }
1069    if usage.contains(crate::AccelerationStructureUses::SHADER_INPUT)
1070        && features.contains(wgt::Features::EXPERIMENTAL_RAY_TRACING_PIPELINES)
1071    {
1072        stages |= vk::PipelineStageFlags::RAY_TRACING_SHADER_KHR;
1073        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_READ_KHR;
1074    }
1075    if usage.contains(crate::AccelerationStructureUses::COPY_SRC) {
1076        stages |= vk::PipelineStageFlags::ACCELERATION_STRUCTURE_BUILD_KHR;
1077        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_READ_KHR;
1078    }
1079    if usage.contains(crate::AccelerationStructureUses::COPY_DST) {
1080        stages |= vk::PipelineStageFlags::ACCELERATION_STRUCTURE_BUILD_KHR;
1081        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_WRITE_KHR;
1082    }
1083
1084    (stages, access)
1085}
1086
1087#[cfg(test)]
1088mod tests {
1089    use super::*;
1090
1091    /// `map_vk_color_space` and `map_surface_color_space` must stay mutually
1092    /// inverse so that a color space reported in the surface capabilities is
1093    /// exactly what the swapchain is created with.
1094    #[test]
1095    fn color_space_round_trip() {
1096        for vk_color_space in [
1097            vk::ColorSpaceKHR::SRGB_NONLINEAR,
1098            vk::ColorSpaceKHR::EXTENDED_SRGB_LINEAR_EXT,
1099            vk::ColorSpaceKHR::EXTENDED_SRGB_NONLINEAR_EXT,
1100            vk::ColorSpaceKHR::DISPLAY_P3_NONLINEAR_EXT,
1101            vk::ColorSpaceKHR::HDR10_ST2084_EXT,
1102            vk::ColorSpaceKHR::HDR10_HLG_EXT,
1103        ] {
1104            let mapped = map_vk_color_space(vk_color_space).unwrap();
1105            assert_eq!(map_surface_color_space(mapped), vk_color_space);
1106        }
1107    }
1108
1109    #[test]
1110    fn unknown_color_spaces_are_dropped() {
1111        for vk_color_space in [
1112            vk::ColorSpaceKHR::BT2020_LINEAR_EXT,
1113            vk::ColorSpaceKHR::DOLBYVISION_EXT,
1114            vk::ColorSpaceKHR::PASS_THROUGH_EXT,
1115            vk::ColorSpaceKHR::ADOBERGB_NONLINEAR_EXT,
1116        ] {
1117            assert!(map_vk_color_space(vk_color_space).is_none());
1118            assert!(map_vk_surface_formats(vk::SurfaceFormatKHR {
1119                format: vk::Format::B8G8R8A8_UNORM,
1120                color_space: vk_color_space,
1121            })
1122            .is_none());
1123        }
1124    }
1125
1126    #[test]
1127    fn hdr10_surface_format_maps() {
1128        let (format, color_space) = map_vk_surface_formats(vk::SurfaceFormatKHR {
1129            format: vk::Format::A2B10G10R10_UNORM_PACK32,
1130            color_space: vk::ColorSpaceKHR::HDR10_ST2084_EXT,
1131        })
1132        .unwrap();
1133        assert_eq!(format, wgt::TextureFormat::Rgb10a2Unorm);
1134        assert_eq!(color_space, wgt::SurfaceColorSpace::Bt2100Pq);
1135    }
1136}
wgpu_hal/vulkan/conv.rs

wgpu_hal/vulkan/
conv.rs
