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(sf: vk::SurfaceFormatKHR) -> Option<wgt::TextureFormat> {
162    use ash::vk::Format as F;
163    use wgt::TextureFormat as Tf;
164    // List we care about pulled from https://vulkan.gpuinfo.org/listsurfaceformats.php.
165    // Device::create_swapchain() hardcodes linear scRGB for fp16, non-linear sRGB otherwise.
166    Some(match sf.color_space {
167        vk::ColorSpaceKHR::SRGB_NONLINEAR => 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::A2B10G10R10_UNORM_PACK32 => Tf::Rgb10a2Unorm,
176            _ => return None,
177        },
178        vk::ColorSpaceKHR::EXTENDED_SRGB_LINEAR_EXT => match sf.format {
179            F::R16G16B16A16_SFLOAT => Tf::Rgba16Float,
180            _ => return None,
181        },
182        _ => return None,
183    })
184}
185
186impl crate::Attachment<'_, super::TextureView> {
187    pub(super) fn make_attachment_key(&self, ops: crate::AttachmentOps) -> super::AttachmentKey {
188        super::AttachmentKey {
189            format: self.view.raw_format,
190            layout: derive_image_layout(self.usage, self.view.format),
191            ops,
192        }
193    }
194}
195
196impl crate::ColorAttachment<'_, super::TextureView> {
197    pub(super) unsafe fn make_vk_clear_color(&self) -> vk::ClearColorValue {
198        let cv = &self.clear_value;
199        match self.target.view.format.sample_type(None, None).unwrap() {
200            wgt::TextureSampleType::Float { .. } => vk::ClearColorValue {
201                float32: [cv.r as f32, cv.g as f32, cv.b as f32, cv.a as f32],
202            },
203            wgt::TextureSampleType::Sint => vk::ClearColorValue {
204                int32: [cv.r as i32, cv.g as i32, cv.b as i32, cv.a as i32],
205            },
206            wgt::TextureSampleType::Uint => vk::ClearColorValue {
207                uint32: [cv.r as u32, cv.g as u32, cv.b as u32, cv.a as u32],
208            },
209            wgt::TextureSampleType::Depth => unreachable!(),
210        }
211    }
212}
213
214pub fn derive_image_layout(usage: wgt::TextureUses, format: wgt::TextureFormat) -> vk::ImageLayout {
215    // Note: depth textures are always sampled with RODS layout
216    let is_color = !format.is_depth_stencil_format();
217    match usage {
218        wgt::TextureUses::UNINITIALIZED => vk::ImageLayout::UNDEFINED,
219        wgt::TextureUses::COPY_SRC => vk::ImageLayout::TRANSFER_SRC_OPTIMAL,
220        wgt::TextureUses::COPY_DST => vk::ImageLayout::TRANSFER_DST_OPTIMAL,
221        wgt::TextureUses::RESOURCE if is_color => vk::ImageLayout::SHADER_READ_ONLY_OPTIMAL,
222        wgt::TextureUses::COLOR_TARGET => vk::ImageLayout::COLOR_ATTACHMENT_OPTIMAL,
223        wgt::TextureUses::DEPTH_STENCIL_WRITE => vk::ImageLayout::DEPTH_STENCIL_ATTACHMENT_OPTIMAL,
224        _ => {
225            if usage == wgt::TextureUses::PRESENT {
226                vk::ImageLayout::PRESENT_SRC_KHR
227            } else if is_color {
228                vk::ImageLayout::GENERAL
229            } else {
230                vk::ImageLayout::DEPTH_STENCIL_READ_ONLY_OPTIMAL
231            }
232        }
233    }
234}
235
236pub fn map_texture_usage(usage: wgt::TextureUses) -> vk::ImageUsageFlags {
237    let mut flags = vk::ImageUsageFlags::empty();
238    if usage.contains(wgt::TextureUses::COPY_SRC) {
239        flags |= vk::ImageUsageFlags::TRANSFER_SRC;
240    }
241    if usage.contains(wgt::TextureUses::COPY_DST) {
242        flags |= vk::ImageUsageFlags::TRANSFER_DST;
243    }
244    if usage.contains(wgt::TextureUses::RESOURCE) {
245        flags |= vk::ImageUsageFlags::SAMPLED;
246    }
247    if usage.contains(wgt::TextureUses::COLOR_TARGET) {
248        flags |= vk::ImageUsageFlags::COLOR_ATTACHMENT;
249    }
250    if usage
251        .intersects(wgt::TextureUses::DEPTH_STENCIL_READ | wgt::TextureUses::DEPTH_STENCIL_WRITE)
252    {
253        flags |= vk::ImageUsageFlags::DEPTH_STENCIL_ATTACHMENT;
254    }
255    if usage.intersects(
256        wgt::TextureUses::STORAGE_READ_ONLY
257            | wgt::TextureUses::STORAGE_WRITE_ONLY
258            | wgt::TextureUses::STORAGE_READ_WRITE
259            | wgt::TextureUses::STORAGE_ATOMIC,
260    ) {
261        flags |= vk::ImageUsageFlags::STORAGE;
262    }
263    if usage.contains(wgt::TextureUses::TRANSIENT) {
264        flags |= vk::ImageUsageFlags::TRANSIENT_ATTACHMENT;
265    }
266    flags
267}
268
269pub fn map_texture_usage_to_barrier(
270    usage: wgt::TextureUses,
271) -> (vk::PipelineStageFlags, vk::AccessFlags) {
272    let mut stages = vk::PipelineStageFlags::empty();
273    let mut access = vk::AccessFlags::empty();
274    let shader_stages = vk::PipelineStageFlags::VERTEX_SHADER
275        | vk::PipelineStageFlags::FRAGMENT_SHADER
276        | vk::PipelineStageFlags::COMPUTE_SHADER;
277
278    if usage.contains(wgt::TextureUses::COPY_SRC) {
279        stages |= vk::PipelineStageFlags::TRANSFER;
280        access |= vk::AccessFlags::TRANSFER_READ;
281    }
282    if usage.contains(wgt::TextureUses::COPY_DST) {
283        stages |= vk::PipelineStageFlags::TRANSFER;
284        access |= vk::AccessFlags::TRANSFER_WRITE;
285    }
286    if usage.contains(wgt::TextureUses::RESOURCE) {
287        stages |= shader_stages;
288        access |= vk::AccessFlags::SHADER_READ;
289    }
290    if usage.contains(wgt::TextureUses::COLOR_TARGET) {
291        stages |= vk::PipelineStageFlags::COLOR_ATTACHMENT_OUTPUT;
292        access |= vk::AccessFlags::COLOR_ATTACHMENT_READ | vk::AccessFlags::COLOR_ATTACHMENT_WRITE;
293    }
294    if usage.intersects(wgt::TextureUses::DEPTH_STENCIL_READ) {
295        stages |= vk::PipelineStageFlags::EARLY_FRAGMENT_TESTS
296            | vk::PipelineStageFlags::LATE_FRAGMENT_TESTS;
297        access |= vk::AccessFlags::DEPTH_STENCIL_ATTACHMENT_READ;
298    }
299    if usage.intersects(wgt::TextureUses::DEPTH_STENCIL_WRITE) {
300        stages |= vk::PipelineStageFlags::EARLY_FRAGMENT_TESTS
301            | vk::PipelineStageFlags::LATE_FRAGMENT_TESTS;
302        access |= vk::AccessFlags::DEPTH_STENCIL_ATTACHMENT_READ
303            | vk::AccessFlags::DEPTH_STENCIL_ATTACHMENT_WRITE;
304    }
305    if usage.intersects(
306        wgt::TextureUses::STORAGE_READ_ONLY
307            | wgt::TextureUses::STORAGE_READ_WRITE
308            | wgt::TextureUses::STORAGE_ATOMIC,
309    ) {
310        stages |= shader_stages;
311        access |= vk::AccessFlags::SHADER_READ;
312    }
313    if usage.intersects(
314        wgt::TextureUses::STORAGE_WRITE_ONLY
315            | wgt::TextureUses::STORAGE_READ_WRITE
316            | wgt::TextureUses::STORAGE_ATOMIC,
317    ) {
318        stages |= shader_stages;
319        access |= vk::AccessFlags::SHADER_WRITE;
320    }
321
322    if usage == wgt::TextureUses::UNINITIALIZED || usage == wgt::TextureUses::PRESENT {
323        (
324            vk::PipelineStageFlags::TOP_OF_PIPE,
325            vk::AccessFlags::empty(),
326        )
327    } else {
328        (stages, access)
329    }
330}
331
332pub fn map_vk_image_usage(usage: vk::ImageUsageFlags) -> wgt::TextureUses {
333    let mut bits = wgt::TextureUses::empty();
334    if usage.contains(vk::ImageUsageFlags::TRANSFER_SRC) {
335        bits |= wgt::TextureUses::COPY_SRC;
336    }
337    if usage.contains(vk::ImageUsageFlags::TRANSFER_DST) {
338        bits |= wgt::TextureUses::COPY_DST;
339    }
340    if usage.contains(vk::ImageUsageFlags::SAMPLED) {
341        bits |= wgt::TextureUses::RESOURCE;
342    }
343    if usage.contains(vk::ImageUsageFlags::COLOR_ATTACHMENT) {
344        bits |= wgt::TextureUses::COLOR_TARGET;
345    }
346    if usage.contains(vk::ImageUsageFlags::DEPTH_STENCIL_ATTACHMENT) {
347        bits |= wgt::TextureUses::DEPTH_STENCIL_READ | wgt::TextureUses::DEPTH_STENCIL_WRITE;
348    }
349    if usage.contains(vk::ImageUsageFlags::STORAGE) {
350        bits |= wgt::TextureUses::STORAGE_READ_ONLY
351            | wgt::TextureUses::STORAGE_WRITE_ONLY
352            | wgt::TextureUses::STORAGE_READ_WRITE
353            | wgt::TextureUses::STORAGE_ATOMIC;
354    }
355    if usage.contains(vk::ImageUsageFlags::TRANSIENT_ATTACHMENT) {
356        bits |= wgt::TextureUses::TRANSIENT;
357    }
358    bits
359}
360
361pub fn map_texture_dimension(dim: wgt::TextureDimension) -> vk::ImageType {
362    match dim {
363        wgt::TextureDimension::D1 => vk::ImageType::TYPE_1D,
364        wgt::TextureDimension::D2 => vk::ImageType::TYPE_2D,
365        wgt::TextureDimension::D3 => vk::ImageType::TYPE_3D,
366    }
367}
368
369pub fn map_index_format(index_format: wgt::IndexFormat) -> vk::IndexType {
370    match index_format {
371        wgt::IndexFormat::Uint16 => vk::IndexType::UINT16,
372        wgt::IndexFormat::Uint32 => vk::IndexType::UINT32,
373    }
374}
375
376pub fn map_vertex_format(vertex_format: wgt::VertexFormat) -> vk::Format {
377    use wgt::VertexFormat as Vf;
378    match vertex_format {
379        Vf::Uint8 => vk::Format::R8_UINT,
380        Vf::Uint8x2 => vk::Format::R8G8_UINT,
381        Vf::Uint8x4 => vk::Format::R8G8B8A8_UINT,
382        Vf::Sint8 => vk::Format::R8_SINT,
383        Vf::Sint8x2 => vk::Format::R8G8_SINT,
384        Vf::Sint8x4 => vk::Format::R8G8B8A8_SINT,
385        Vf::Unorm8 => vk::Format::R8_UNORM,
386        Vf::Unorm8x2 => vk::Format::R8G8_UNORM,
387        Vf::Unorm8x4 => vk::Format::R8G8B8A8_UNORM,
388        Vf::Snorm8 => vk::Format::R8_SNORM,
389        Vf::Snorm8x2 => vk::Format::R8G8_SNORM,
390        Vf::Snorm8x4 => vk::Format::R8G8B8A8_SNORM,
391        Vf::Uint16 => vk::Format::R16_UINT,
392        Vf::Uint16x2 => vk::Format::R16G16_UINT,
393        Vf::Uint16x4 => vk::Format::R16G16B16A16_UINT,
394        Vf::Sint16 => vk::Format::R16_SINT,
395        Vf::Sint16x2 => vk::Format::R16G16_SINT,
396        Vf::Sint16x4 => vk::Format::R16G16B16A16_SINT,
397        Vf::Unorm16 => vk::Format::R16_UNORM,
398        Vf::Unorm16x2 => vk::Format::R16G16_UNORM,
399        Vf::Unorm16x4 => vk::Format::R16G16B16A16_UNORM,
400        Vf::Snorm16 => vk::Format::R16_SNORM,
401        Vf::Snorm16x2 => vk::Format::R16G16_SNORM,
402        Vf::Snorm16x4 => vk::Format::R16G16B16A16_SNORM,
403        Vf::Float16 => vk::Format::R16_SFLOAT,
404        Vf::Float16x2 => vk::Format::R16G16_SFLOAT,
405        Vf::Float16x4 => vk::Format::R16G16B16A16_SFLOAT,
406        Vf::Float32 => vk::Format::R32_SFLOAT,
407        Vf::Float32x2 => vk::Format::R32G32_SFLOAT,
408        Vf::Float32x3 => vk::Format::R32G32B32_SFLOAT,
409        Vf::Float32x4 => vk::Format::R32G32B32A32_SFLOAT,
410        Vf::Uint32 => vk::Format::R32_UINT,
411        Vf::Uint32x2 => vk::Format::R32G32_UINT,
412        Vf::Uint32x3 => vk::Format::R32G32B32_UINT,
413        Vf::Uint32x4 => vk::Format::R32G32B32A32_UINT,
414        Vf::Sint32 => vk::Format::R32_SINT,
415        Vf::Sint32x2 => vk::Format::R32G32_SINT,
416        Vf::Sint32x3 => vk::Format::R32G32B32_SINT,
417        Vf::Sint32x4 => vk::Format::R32G32B32A32_SINT,
418        Vf::Float64 => vk::Format::R64_SFLOAT,
419        Vf::Float64x2 => vk::Format::R64G64_SFLOAT,
420        Vf::Float64x3 => vk::Format::R64G64B64_SFLOAT,
421        Vf::Float64x4 => vk::Format::R64G64B64A64_SFLOAT,
422        Vf::Unorm10_10_10_2 => vk::Format::A2B10G10R10_UNORM_PACK32,
423        Vf::Unorm8x4Bgra => vk::Format::B8G8R8A8_UNORM,
424    }
425}
426
427pub fn map_aspects(aspects: crate::FormatAspects) -> vk::ImageAspectFlags {
428    let mut flags = vk::ImageAspectFlags::empty();
429    if aspects.contains(crate::FormatAspects::COLOR) {
430        flags |= vk::ImageAspectFlags::COLOR;
431    }
432    if aspects.contains(crate::FormatAspects::DEPTH) {
433        flags |= vk::ImageAspectFlags::DEPTH;
434    }
435    if aspects.contains(crate::FormatAspects::STENCIL) {
436        flags |= vk::ImageAspectFlags::STENCIL;
437    }
438    if aspects.contains(crate::FormatAspects::PLANE_0) {
439        flags |= vk::ImageAspectFlags::PLANE_0;
440    }
441    if aspects.contains(crate::FormatAspects::PLANE_1) {
442        flags |= vk::ImageAspectFlags::PLANE_1;
443    }
444    if aspects.contains(crate::FormatAspects::PLANE_2) {
445        flags |= vk::ImageAspectFlags::PLANE_2;
446    }
447    flags
448}
449
450pub fn map_attachment_ops(
451    op: crate::AttachmentOps,
452) -> (vk::AttachmentLoadOp, vk::AttachmentStoreOp) {
453    let load_op = if op.contains(crate::AttachmentOps::LOAD) {
454        vk::AttachmentLoadOp::LOAD
455    } else {
456        vk::AttachmentLoadOp::CLEAR
457    };
458    let store_op = if op.contains(crate::AttachmentOps::STORE) {
459        vk::AttachmentStoreOp::STORE
460    } else {
461        vk::AttachmentStoreOp::DONT_CARE
462    };
463    (load_op, store_op)
464}
465
466pub fn map_present_mode(mode: wgt::PresentMode) -> vk::PresentModeKHR {
467    match mode {
468        wgt::PresentMode::Immediate => vk::PresentModeKHR::IMMEDIATE,
469        wgt::PresentMode::Mailbox => vk::PresentModeKHR::MAILBOX,
470        wgt::PresentMode::Fifo => vk::PresentModeKHR::FIFO,
471        wgt::PresentMode::FifoRelaxed => vk::PresentModeKHR::FIFO_RELAXED,
472        wgt::PresentMode::AutoNoVsync | wgt::PresentMode::AutoVsync => {
473            unreachable!("Cannot create swapchain with Auto PresentationMode")
474        }
475    }
476}
477
478pub fn map_vk_present_mode(mode: vk::PresentModeKHR) -> Option<wgt::PresentMode> {
479    // Not exposed in Ash yet.
480    const FIFO_LATEST_READY: vk::PresentModeKHR = vk::PresentModeKHR::from_raw(1_000_361_000);
481
482    // See https://registry.khronos.org/vulkan/specs/latest/man/html/VkPresentModeKHR.html
483    match mode {
484        vk::PresentModeKHR::IMMEDIATE => Some(wgt::PresentMode::Immediate),
485        vk::PresentModeKHR::MAILBOX => Some(wgt::PresentMode::Mailbox),
486        vk::PresentModeKHR::FIFO => Some(wgt::PresentMode::Fifo),
487        vk::PresentModeKHR::FIFO_RELAXED => Some(wgt::PresentMode::FifoRelaxed),
488
489        // Modes that aren't exposed yet.
490        vk::PresentModeKHR::SHARED_DEMAND_REFRESH => None,
491        vk::PresentModeKHR::SHARED_CONTINUOUS_REFRESH => None,
492        FIFO_LATEST_READY => None,
493
494        _ => {
495            log::debug!("Unrecognized present mode {mode:?}");
496            None
497        }
498    }
499}
500
501pub fn map_composite_alpha_mode(mode: wgt::CompositeAlphaMode) -> vk::CompositeAlphaFlagsKHR {
502    match mode {
503        wgt::CompositeAlphaMode::Opaque => vk::CompositeAlphaFlagsKHR::OPAQUE,
504        wgt::CompositeAlphaMode::PreMultiplied => vk::CompositeAlphaFlagsKHR::PRE_MULTIPLIED,
505        wgt::CompositeAlphaMode::PostMultiplied => vk::CompositeAlphaFlagsKHR::POST_MULTIPLIED,
506        wgt::CompositeAlphaMode::Inherit => vk::CompositeAlphaFlagsKHR::INHERIT,
507        wgt::CompositeAlphaMode::Auto => unreachable!(),
508    }
509}
510
511pub fn map_vk_composite_alpha(flags: vk::CompositeAlphaFlagsKHR) -> Vec<wgt::CompositeAlphaMode> {
512    let mut modes = Vec::new();
513    if flags.contains(vk::CompositeAlphaFlagsKHR::OPAQUE) {
514        modes.push(wgt::CompositeAlphaMode::Opaque);
515    }
516    if flags.contains(vk::CompositeAlphaFlagsKHR::PRE_MULTIPLIED) {
517        modes.push(wgt::CompositeAlphaMode::PreMultiplied);
518    }
519    if flags.contains(vk::CompositeAlphaFlagsKHR::POST_MULTIPLIED) {
520        modes.push(wgt::CompositeAlphaMode::PostMultiplied);
521    }
522    if flags.contains(vk::CompositeAlphaFlagsKHR::INHERIT) {
523        modes.push(wgt::CompositeAlphaMode::Inherit);
524    }
525    modes
526}
527
528pub fn map_buffer_usage(usage: wgt::BufferUses) -> vk::BufferUsageFlags {
529    let mut flags = vk::BufferUsageFlags::empty();
530    if usage.contains(wgt::BufferUses::COPY_SRC) {
531        flags |= vk::BufferUsageFlags::TRANSFER_SRC;
532    }
533    if usage.contains(wgt::BufferUses::COPY_DST) {
534        flags |= vk::BufferUsageFlags::TRANSFER_DST;
535    }
536    if usage.contains(wgt::BufferUses::UNIFORM) {
537        flags |= vk::BufferUsageFlags::UNIFORM_BUFFER;
538    }
539    if usage.intersects(wgt::BufferUses::STORAGE_READ_ONLY | wgt::BufferUses::STORAGE_READ_WRITE) {
540        flags |= vk::BufferUsageFlags::STORAGE_BUFFER;
541    }
542    if usage.contains(wgt::BufferUses::INDEX) {
543        flags |= vk::BufferUsageFlags::INDEX_BUFFER;
544    }
545    if usage.contains(wgt::BufferUses::VERTEX) {
546        flags |= vk::BufferUsageFlags::VERTEX_BUFFER;
547    }
548    if usage.contains(wgt::BufferUses::INDIRECT) {
549        flags |= vk::BufferUsageFlags::INDIRECT_BUFFER;
550    }
551    if usage.contains(wgt::BufferUses::ACCELERATION_STRUCTURE_SCRATCH) {
552        flags |= vk::BufferUsageFlags::STORAGE_BUFFER | vk::BufferUsageFlags::SHADER_DEVICE_ADDRESS;
553    }
554    if usage.intersects(
555        wgt::BufferUses::BOTTOM_LEVEL_ACCELERATION_STRUCTURE_INPUT
556            | wgt::BufferUses::TOP_LEVEL_ACCELERATION_STRUCTURE_INPUT,
557    ) {
558        flags |= vk::BufferUsageFlags::ACCELERATION_STRUCTURE_BUILD_INPUT_READ_ONLY_KHR
559            | vk::BufferUsageFlags::SHADER_DEVICE_ADDRESS;
560    }
561    if usage.intersects(wgt::BufferUses::ACCELERATION_STRUCTURE_QUERY) {
562        flags |= vk::BufferUsageFlags::TRANSFER_DST;
563    }
564    flags
565}
566
567pub fn map_buffer_usage_to_barrier(
568    usage: wgt::BufferUses,
569) -> (vk::PipelineStageFlags, vk::AccessFlags) {
570    let mut stages = vk::PipelineStageFlags::empty();
571    let mut access = vk::AccessFlags::empty();
572    let shader_stages = vk::PipelineStageFlags::VERTEX_SHADER
573        | vk::PipelineStageFlags::FRAGMENT_SHADER
574        | vk::PipelineStageFlags::COMPUTE_SHADER;
575
576    if usage.contains(wgt::BufferUses::MAP_READ) {
577        stages |= vk::PipelineStageFlags::HOST;
578        access |= vk::AccessFlags::HOST_READ;
579    }
580    if usage.contains(wgt::BufferUses::MAP_WRITE) {
581        stages |= vk::PipelineStageFlags::HOST;
582        access |= vk::AccessFlags::HOST_WRITE;
583    }
584    if usage.contains(wgt::BufferUses::COPY_SRC) {
585        stages |= vk::PipelineStageFlags::TRANSFER;
586        access |= vk::AccessFlags::TRANSFER_READ;
587    }
588    if usage.contains(wgt::BufferUses::COPY_DST) {
589        stages |= vk::PipelineStageFlags::TRANSFER;
590        access |= vk::AccessFlags::TRANSFER_WRITE;
591    }
592    if usage.contains(wgt::BufferUses::UNIFORM) {
593        stages |= shader_stages;
594        access |= vk::AccessFlags::UNIFORM_READ;
595    }
596    if usage.intersects(wgt::BufferUses::STORAGE_READ_ONLY) {
597        stages |= shader_stages;
598        access |= vk::AccessFlags::SHADER_READ;
599    }
600    if usage.intersects(wgt::BufferUses::STORAGE_READ_WRITE) {
601        stages |= shader_stages;
602        access |= vk::AccessFlags::SHADER_READ | vk::AccessFlags::SHADER_WRITE;
603    }
604    if usage.contains(wgt::BufferUses::INDEX) {
605        stages |= vk::PipelineStageFlags::VERTEX_INPUT;
606        access |= vk::AccessFlags::INDEX_READ;
607    }
608    if usage.contains(wgt::BufferUses::VERTEX) {
609        stages |= vk::PipelineStageFlags::VERTEX_INPUT;
610        access |= vk::AccessFlags::VERTEX_ATTRIBUTE_READ;
611    }
612    if usage.contains(wgt::BufferUses::INDIRECT) {
613        stages |= vk::PipelineStageFlags::DRAW_INDIRECT;
614        access |= vk::AccessFlags::INDIRECT_COMMAND_READ;
615    }
616    if usage.intersects(
617        wgt::BufferUses::BOTTOM_LEVEL_ACCELERATION_STRUCTURE_INPUT
618            | wgt::BufferUses::TOP_LEVEL_ACCELERATION_STRUCTURE_INPUT
619            | wgt::BufferUses::ACCELERATION_STRUCTURE_SCRATCH,
620    ) {
621        stages |= vk::PipelineStageFlags::ACCELERATION_STRUCTURE_BUILD_KHR;
622        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_READ_KHR
623            | vk::AccessFlags::ACCELERATION_STRUCTURE_WRITE_KHR
624            | vk::AccessFlags::SHADER_READ;
625    }
626    if usage.contains(wgt::BufferUses::ACCELERATION_STRUCTURE_QUERY) {
627        stages |= vk::PipelineStageFlags::TRANSFER;
628        access |= vk::AccessFlags::TRANSFER_WRITE;
629    }
630
631    (stages, access)
632}
633
634pub fn map_view_dimension(dim: wgt::TextureViewDimension) -> vk::ImageViewType {
635    match dim {
636        wgt::TextureViewDimension::D1 => vk::ImageViewType::TYPE_1D,
637        wgt::TextureViewDimension::D2 => vk::ImageViewType::TYPE_2D,
638        wgt::TextureViewDimension::D2Array => vk::ImageViewType::TYPE_2D_ARRAY,
639        wgt::TextureViewDimension::Cube => vk::ImageViewType::CUBE,
640        wgt::TextureViewDimension::CubeArray => vk::ImageViewType::CUBE_ARRAY,
641        wgt::TextureViewDimension::D3 => vk::ImageViewType::TYPE_3D,
642    }
643}
644
645pub fn map_copy_extent(extent: &crate::CopyExtent) -> vk::Extent3D {
646    vk::Extent3D {
647        width: extent.width,
648        height: extent.height,
649        depth: extent.depth,
650    }
651}
652
653pub fn map_subresource_range(
654    range: &wgt::ImageSubresourceRange,
655    format: wgt::TextureFormat,
656) -> vk::ImageSubresourceRange {
657    vk::ImageSubresourceRange {
658        aspect_mask: map_aspects(crate::FormatAspects::new(format, range.aspect)),
659        base_mip_level: range.base_mip_level,
660        level_count: range.mip_level_count.unwrap_or(vk::REMAINING_MIP_LEVELS),
661        base_array_layer: range.base_array_layer,
662        layer_count: range
663            .array_layer_count
664            .unwrap_or(vk::REMAINING_ARRAY_LAYERS),
665    }
666}
667
668// Special subresource range mapping for dealing with barriers
669// so that we account for the "hidden" depth aspect in emulated Stencil8.
670pub(super) fn map_subresource_range_combined_aspect(
671    range: &wgt::ImageSubresourceRange,
672    format: wgt::TextureFormat,
673    private_caps: &super::PrivateCapabilities,
674) -> vk::ImageSubresourceRange {
675    let mut range = map_subresource_range(range, format);
676    if !private_caps.texture_s8 && format == wgt::TextureFormat::Stencil8 {
677        range.aspect_mask |= vk::ImageAspectFlags::DEPTH;
678    }
679    range
680}
681
682pub fn map_subresource_layers(
683    base: &crate::TextureCopyBase,
684) -> (vk::ImageSubresourceLayers, vk::Offset3D) {
685    let offset = vk::Offset3D {
686        x: base.origin.x as i32,
687        y: base.origin.y as i32,
688        z: base.origin.z as i32,
689    };
690    let subresource = vk::ImageSubresourceLayers {
691        aspect_mask: map_aspects(base.aspect),
692        mip_level: base.mip_level,
693        base_array_layer: base.array_layer,
694        layer_count: 1,
695    };
696    (subresource, offset)
697}
698
699pub fn map_filter_mode(mode: wgt::FilterMode) -> vk::Filter {
700    match mode {
701        wgt::FilterMode::Nearest => vk::Filter::NEAREST,
702        wgt::FilterMode::Linear => vk::Filter::LINEAR,
703    }
704}
705
706pub fn map_mip_filter_mode(mode: wgt::MipmapFilterMode) -> vk::SamplerMipmapMode {
707    match mode {
708        wgt::MipmapFilterMode::Nearest => vk::SamplerMipmapMode::NEAREST,
709        wgt::MipmapFilterMode::Linear => vk::SamplerMipmapMode::LINEAR,
710    }
711}
712
713pub fn map_address_mode(mode: wgt::AddressMode) -> vk::SamplerAddressMode {
714    match mode {
715        wgt::AddressMode::ClampToEdge => vk::SamplerAddressMode::CLAMP_TO_EDGE,
716        wgt::AddressMode::Repeat => vk::SamplerAddressMode::REPEAT,
717        wgt::AddressMode::MirrorRepeat => vk::SamplerAddressMode::MIRRORED_REPEAT,
718        wgt::AddressMode::ClampToBorder => vk::SamplerAddressMode::CLAMP_TO_BORDER,
719        // wgt::AddressMode::MirrorClamp => vk::SamplerAddressMode::MIRROR_CLAMP_TO_EDGE,
720    }
721}
722
723pub fn map_border_color(border_color: wgt::SamplerBorderColor) -> vk::BorderColor {
724    match border_color {
725        wgt::SamplerBorderColor::TransparentBlack | wgt::SamplerBorderColor::Zero => {
726            vk::BorderColor::FLOAT_TRANSPARENT_BLACK
727        }
728        wgt::SamplerBorderColor::OpaqueBlack => vk::BorderColor::FLOAT_OPAQUE_BLACK,
729        wgt::SamplerBorderColor::OpaqueWhite => vk::BorderColor::FLOAT_OPAQUE_WHITE,
730    }
731}
732
733pub fn map_comparison(fun: wgt::CompareFunction) -> vk::CompareOp {
734    use wgt::CompareFunction as Cf;
735    match fun {
736        Cf::Never => vk::CompareOp::NEVER,
737        Cf::Less => vk::CompareOp::LESS,
738        Cf::LessEqual => vk::CompareOp::LESS_OR_EQUAL,
739        Cf::Equal => vk::CompareOp::EQUAL,
740        Cf::GreaterEqual => vk::CompareOp::GREATER_OR_EQUAL,
741        Cf::Greater => vk::CompareOp::GREATER,
742        Cf::NotEqual => vk::CompareOp::NOT_EQUAL,
743        Cf::Always => vk::CompareOp::ALWAYS,
744    }
745}
746
747pub fn map_shader_stage(stage: wgt::ShaderStages) -> vk::ShaderStageFlags {
748    let mut flags = vk::ShaderStageFlags::empty();
749    if stage.contains(wgt::ShaderStages::VERTEX) {
750        flags |= vk::ShaderStageFlags::VERTEX;
751    }
752    if stage.contains(wgt::ShaderStages::FRAGMENT) {
753        flags |= vk::ShaderStageFlags::FRAGMENT;
754    }
755    if stage.contains(wgt::ShaderStages::COMPUTE) {
756        flags |= vk::ShaderStageFlags::COMPUTE;
757    }
758    if stage.contains(wgt::ShaderStages::TASK) {
759        flags |= vk::ShaderStageFlags::TASK_EXT;
760    }
761    if stage.contains(wgt::ShaderStages::MESH) {
762        flags |= vk::ShaderStageFlags::MESH_EXT;
763    }
764    flags
765}
766
767pub fn map_binding_type(ty: wgt::BindingType) -> vk::DescriptorType {
768    match ty {
769        wgt::BindingType::Buffer {
770            ty,
771            has_dynamic_offset,
772            ..
773        } => match ty {
774            wgt::BufferBindingType::Storage { .. } => match has_dynamic_offset {
775                true => vk::DescriptorType::STORAGE_BUFFER_DYNAMIC,
776                false => vk::DescriptorType::STORAGE_BUFFER,
777            },
778            wgt::BufferBindingType::Uniform => match has_dynamic_offset {
779                true => vk::DescriptorType::UNIFORM_BUFFER_DYNAMIC,
780                false => vk::DescriptorType::UNIFORM_BUFFER,
781            },
782        },
783        wgt::BindingType::Sampler { .. } => vk::DescriptorType::SAMPLER,
784        wgt::BindingType::Texture { .. } => vk::DescriptorType::SAMPLED_IMAGE,
785        wgt::BindingType::StorageTexture { .. } => vk::DescriptorType::STORAGE_IMAGE,
786        wgt::BindingType::AccelerationStructure { .. } => {
787            vk::DescriptorType::ACCELERATION_STRUCTURE_KHR
788        }
789        wgt::BindingType::ExternalTexture => unimplemented!(),
790    }
791}
792
793pub fn map_topology(topology: wgt::PrimitiveTopology) -> vk::PrimitiveTopology {
794    use wgt::PrimitiveTopology as Pt;
795    match topology {
796        Pt::PointList => vk::PrimitiveTopology::POINT_LIST,
797        Pt::LineList => vk::PrimitiveTopology::LINE_LIST,
798        Pt::LineStrip => vk::PrimitiveTopology::LINE_STRIP,
799        Pt::TriangleList => vk::PrimitiveTopology::TRIANGLE_LIST,
800        Pt::TriangleStrip => vk::PrimitiveTopology::TRIANGLE_STRIP,
801    }
802}
803
804pub fn map_polygon_mode(mode: wgt::PolygonMode) -> vk::PolygonMode {
805    match mode {
806        wgt::PolygonMode::Fill => vk::PolygonMode::FILL,
807        wgt::PolygonMode::Line => vk::PolygonMode::LINE,
808        wgt::PolygonMode::Point => vk::PolygonMode::POINT,
809    }
810}
811
812pub fn map_front_face(front_face: wgt::FrontFace) -> vk::FrontFace {
813    match front_face {
814        wgt::FrontFace::Cw => vk::FrontFace::CLOCKWISE,
815        wgt::FrontFace::Ccw => vk::FrontFace::COUNTER_CLOCKWISE,
816    }
817}
818
819pub fn map_cull_face(face: wgt::Face) -> vk::CullModeFlags {
820    match face {
821        wgt::Face::Front => vk::CullModeFlags::FRONT,
822        wgt::Face::Back => vk::CullModeFlags::BACK,
823    }
824}
825
826pub fn map_stencil_op(op: wgt::StencilOperation) -> vk::StencilOp {
827    use wgt::StencilOperation as So;
828    match op {
829        So::Keep => vk::StencilOp::KEEP,
830        So::Zero => vk::StencilOp::ZERO,
831        So::Replace => vk::StencilOp::REPLACE,
832        So::Invert => vk::StencilOp::INVERT,
833        So::IncrementClamp => vk::StencilOp::INCREMENT_AND_CLAMP,
834        So::IncrementWrap => vk::StencilOp::INCREMENT_AND_WRAP,
835        So::DecrementClamp => vk::StencilOp::DECREMENT_AND_CLAMP,
836        So::DecrementWrap => vk::StencilOp::DECREMENT_AND_WRAP,
837    }
838}
839
840pub fn map_stencil_face(
841    face: &wgt::StencilFaceState,
842    compare_mask: u32,
843    write_mask: u32,
844) -> vk::StencilOpState {
845    vk::StencilOpState {
846        fail_op: map_stencil_op(face.fail_op),
847        pass_op: map_stencil_op(face.pass_op),
848        depth_fail_op: map_stencil_op(face.depth_fail_op),
849        compare_op: map_comparison(face.compare),
850        compare_mask,
851        write_mask,
852        reference: 0,
853    }
854}
855
856fn map_blend_factor(factor: wgt::BlendFactor) -> vk::BlendFactor {
857    use wgt::BlendFactor as Bf;
858    match factor {
859        Bf::Zero => vk::BlendFactor::ZERO,
860        Bf::One => vk::BlendFactor::ONE,
861        Bf::Src => vk::BlendFactor::SRC_COLOR,
862        Bf::OneMinusSrc => vk::BlendFactor::ONE_MINUS_SRC_COLOR,
863        Bf::SrcAlpha => vk::BlendFactor::SRC_ALPHA,
864        Bf::OneMinusSrcAlpha => vk::BlendFactor::ONE_MINUS_SRC_ALPHA,
865        Bf::Dst => vk::BlendFactor::DST_COLOR,
866        Bf::OneMinusDst => vk::BlendFactor::ONE_MINUS_DST_COLOR,
867        Bf::DstAlpha => vk::BlendFactor::DST_ALPHA,
868        Bf::OneMinusDstAlpha => vk::BlendFactor::ONE_MINUS_DST_ALPHA,
869        Bf::SrcAlphaSaturated => vk::BlendFactor::SRC_ALPHA_SATURATE,
870        Bf::Constant => vk::BlendFactor::CONSTANT_COLOR,
871        Bf::OneMinusConstant => vk::BlendFactor::ONE_MINUS_CONSTANT_COLOR,
872        Bf::Src1 => vk::BlendFactor::SRC1_COLOR,
873        Bf::OneMinusSrc1 => vk::BlendFactor::ONE_MINUS_SRC1_COLOR,
874        Bf::Src1Alpha => vk::BlendFactor::SRC1_ALPHA,
875        Bf::OneMinusSrc1Alpha => vk::BlendFactor::ONE_MINUS_SRC1_ALPHA,
876    }
877}
878
879fn map_blend_op(operation: wgt::BlendOperation) -> vk::BlendOp {
880    use wgt::BlendOperation as Bo;
881    match operation {
882        Bo::Add => vk::BlendOp::ADD,
883        Bo::Subtract => vk::BlendOp::SUBTRACT,
884        Bo::ReverseSubtract => vk::BlendOp::REVERSE_SUBTRACT,
885        Bo::Min => vk::BlendOp::MIN,
886        Bo::Max => vk::BlendOp::MAX,
887    }
888}
889
890pub fn map_blend_component(
891    component: &wgt::BlendComponent,
892) -> (vk::BlendOp, vk::BlendFactor, vk::BlendFactor) {
893    let op = map_blend_op(component.operation);
894    let src = map_blend_factor(component.src_factor);
895    let dst = map_blend_factor(component.dst_factor);
896    (op, src, dst)
897}
898
899pub fn map_pipeline_statistics(
900    types: wgt::PipelineStatisticsTypes,
901) -> vk::QueryPipelineStatisticFlags {
902    use wgt::PipelineStatisticsTypes as Pst;
903    let mut flags = vk::QueryPipelineStatisticFlags::empty();
904    if types.contains(Pst::VERTEX_SHADER_INVOCATIONS) {
905        flags |= vk::QueryPipelineStatisticFlags::VERTEX_SHADER_INVOCATIONS;
906    }
907    if types.contains(Pst::CLIPPER_INVOCATIONS) {
908        flags |= vk::QueryPipelineStatisticFlags::CLIPPING_INVOCATIONS;
909    }
910    if types.contains(Pst::CLIPPER_PRIMITIVES_OUT) {
911        flags |= vk::QueryPipelineStatisticFlags::CLIPPING_PRIMITIVES;
912    }
913    if types.contains(Pst::FRAGMENT_SHADER_INVOCATIONS) {
914        flags |= vk::QueryPipelineStatisticFlags::FRAGMENT_SHADER_INVOCATIONS;
915    }
916    if types.contains(Pst::COMPUTE_SHADER_INVOCATIONS) {
917        flags |= vk::QueryPipelineStatisticFlags::COMPUTE_SHADER_INVOCATIONS;
918    }
919    flags
920}
921
922pub fn map_acceleration_structure_format(
923    format: crate::AccelerationStructureFormat,
924) -> vk::AccelerationStructureTypeKHR {
925    match format {
926        crate::AccelerationStructureFormat::TopLevel => vk::AccelerationStructureTypeKHR::TOP_LEVEL,
927        crate::AccelerationStructureFormat::BottomLevel => {
928            vk::AccelerationStructureTypeKHR::BOTTOM_LEVEL
929        }
930    }
931}
932
933pub fn map_acceleration_structure_build_mode(
934    format: crate::AccelerationStructureBuildMode,
935) -> vk::BuildAccelerationStructureModeKHR {
936    match format {
937        crate::AccelerationStructureBuildMode::Build => {
938            vk::BuildAccelerationStructureModeKHR::BUILD
939        }
940        crate::AccelerationStructureBuildMode::Update => {
941            vk::BuildAccelerationStructureModeKHR::UPDATE
942        }
943    }
944}
945
946pub fn map_acceleration_structure_flags(
947    flags: crate::AccelerationStructureBuildFlags,
948) -> vk::BuildAccelerationStructureFlagsKHR {
949    let mut vk_flags = vk::BuildAccelerationStructureFlagsKHR::empty();
950
951    if flags.contains(crate::AccelerationStructureBuildFlags::PREFER_FAST_TRACE) {
952        vk_flags |= vk::BuildAccelerationStructureFlagsKHR::PREFER_FAST_TRACE;
953    }
954
955    if flags.contains(crate::AccelerationStructureBuildFlags::PREFER_FAST_BUILD) {
956        vk_flags |= vk::BuildAccelerationStructureFlagsKHR::PREFER_FAST_BUILD;
957    }
958
959    if flags.contains(crate::AccelerationStructureBuildFlags::ALLOW_UPDATE) {
960        vk_flags |= vk::BuildAccelerationStructureFlagsKHR::ALLOW_UPDATE;
961    }
962
963    if flags.contains(crate::AccelerationStructureBuildFlags::LOW_MEMORY) {
964        vk_flags |= vk::BuildAccelerationStructureFlagsKHR::LOW_MEMORY;
965    }
966
967    if flags.contains(crate::AccelerationStructureBuildFlags::ALLOW_COMPACTION) {
968        vk_flags |= vk::BuildAccelerationStructureFlagsKHR::ALLOW_COMPACTION
969    }
970
971    if flags.contains(crate::AccelerationStructureBuildFlags::ALLOW_RAY_HIT_VERTEX_RETURN) {
972        vk_flags |= vk::BuildAccelerationStructureFlagsKHR::ALLOW_DATA_ACCESS
973    }
974
975    vk_flags
976}
977
978pub fn map_acceleration_structure_geometry_flags(
979    flags: crate::AccelerationStructureGeometryFlags,
980) -> vk::GeometryFlagsKHR {
981    let mut vk_flags = vk::GeometryFlagsKHR::empty();
982
983    if flags.contains(crate::AccelerationStructureGeometryFlags::OPAQUE) {
984        vk_flags |= vk::GeometryFlagsKHR::OPAQUE;
985    }
986
987    if flags.contains(crate::AccelerationStructureGeometryFlags::NO_DUPLICATE_ANY_HIT_INVOCATION) {
988        vk_flags |= vk::GeometryFlagsKHR::NO_DUPLICATE_ANY_HIT_INVOCATION;
989    }
990
991    vk_flags
992}
993
994pub fn map_acceleration_structure_usage_to_barrier(
995    usage: crate::AccelerationStructureUses,
996    features: wgt::Features,
997) -> (vk::PipelineStageFlags, vk::AccessFlags) {
998    let mut stages = vk::PipelineStageFlags::empty();
999    let mut access = vk::AccessFlags::empty();
1000
1001    if usage.contains(crate::AccelerationStructureUses::BUILD_INPUT) {
1002        stages |= vk::PipelineStageFlags::ACCELERATION_STRUCTURE_BUILD_KHR;
1003        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_READ_KHR;
1004    }
1005    if usage.contains(crate::AccelerationStructureUses::QUERY_INPUT) {
1006        stages |= vk::PipelineStageFlags::ACCELERATION_STRUCTURE_BUILD_KHR;
1007        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_READ_KHR;
1008    }
1009    if usage.contains(crate::AccelerationStructureUses::BUILD_OUTPUT) {
1010        stages |= vk::PipelineStageFlags::ACCELERATION_STRUCTURE_BUILD_KHR;
1011        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_WRITE_KHR;
1012    }
1013    if usage.contains(crate::AccelerationStructureUses::SHADER_INPUT)
1014        && features.contains(wgt::Features::EXPERIMENTAL_RAY_QUERY)
1015    {
1016        stages |= vk::PipelineStageFlags::VERTEX_SHADER
1017            | vk::PipelineStageFlags::FRAGMENT_SHADER
1018            | vk::PipelineStageFlags::COMPUTE_SHADER;
1019        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_READ_KHR;
1020    }
1021    if usage.contains(crate::AccelerationStructureUses::COPY_SRC) {
1022        stages |= vk::PipelineStageFlags::ACCELERATION_STRUCTURE_BUILD_KHR;
1023        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_READ_KHR;
1024    }
1025    if usage.contains(crate::AccelerationStructureUses::COPY_DST) {
1026        stages |= vk::PipelineStageFlags::ACCELERATION_STRUCTURE_BUILD_KHR;
1027        access |= vk::AccessFlags::ACCELERATION_STRUCTURE_WRITE_KHR;
1028    }
1029
1030    (stages, access)
1031}
wgpu_hal/vulkan/conv.rs

wgpu_hal/vulkan/
conv.rs
