use std::convert::TryInto; #[cfg(feature = "debug")] use std::fmt; use std::mem; use crate::vk; pub type VkResult = Result; impl vk::Result { #[inline] pub fn result(self) -> VkResult<()> { self.result_with_success(()) } #[inline] pub fn result_with_success(self, v: T) -> VkResult { match self { Self::SUCCESS => Ok(v), _ => Err(self), } } #[inline] pub unsafe fn assume_init_on_success(self, v: mem::MaybeUninit) -> VkResult { self.result().map(move |()| v.assume_init()) } } /// Repeatedly calls `f` until it does not return [`vk::Result::INCOMPLETE`] anymore, ensuring all /// available data has been read into the vector. /// /// See for example [`vkEnumerateInstanceExtensionProperties`]: the number of available items may /// change between calls; [`vk::Result::INCOMPLETE`] is returned when the count increased (and the /// vector is not large enough after querying the initial size), requiring Ash to try again. /// /// [`vkEnumerateInstanceExtensionProperties`]: https://www.khronos.org/registry/vulkan/specs/1.3-extensions/man/html/vkEnumerateInstanceExtensionProperties.html pub(crate) unsafe fn read_into_uninitialized_vector, T>( f: impl Fn(&mut N, *mut T) -> vk::Result, ) -> VkResult> where >::Error: std::fmt::Debug, { loop { let mut count = N::default(); f(&mut count, std::ptr::null_mut()).result()?; let mut data = Vec::with_capacity(count.try_into().expect("`N` failed to convert to `usize`")); let err_code = f(&mut count, data.as_mut_ptr()); if err_code != vk::Result::INCOMPLETE { err_code.result()?; data.set_len(count.try_into().expect("`N` failed to convert to `usize`")); break Ok(data); } } } /// Repeatedly calls `f` until it does not return [`vk::Result::INCOMPLETE`] anymore, ensuring all /// available data has been read into the vector. /// /// Items in the target vector are [`default()`][Default::default()]-initialized which is required /// for [`vk::BaseOutStructure`]-like structs where [`vk::BaseOutStructure::s_type`] needs to be a /// valid type and [`vk::BaseOutStructure::p_next`] a valid or [`null`][std::ptr::null_mut()] /// pointer. /// /// See for example [`vkEnumerateInstanceExtensionProperties`]: the number of available items may /// change between calls; [`vk::Result::INCOMPLETE`] is returned when the count increased (and the /// vector is not large enough after querying the initial size), requiring Ash to try again. /// /// [`vkEnumerateInstanceExtensionProperties`]: https://www.khronos.org/registry/vulkan/specs/1.3-extensions/man/html/vkEnumerateInstanceExtensionProperties.html pub(crate) unsafe fn read_into_defaulted_vector< N: Copy + Default + TryInto, T: Default + Clone, >( f: impl Fn(&mut N, *mut T) -> vk::Result, ) -> VkResult> where >::Error: std::fmt::Debug, { loop { let mut count = N::default(); f(&mut count, std::ptr::null_mut()).result()?; let mut data = vec![Default::default(); count.try_into().expect("`N` failed to convert to `usize`")]; let err_code = f(&mut count, data.as_mut_ptr()); if err_code != vk::Result::INCOMPLETE { data.set_len(count.try_into().expect("`N` failed to convert to `usize`")); break err_code.result_with_success(data); } } } #[cfg(feature = "debug")] pub(crate) fn debug_flags + Copy>( f: &mut fmt::Formatter, known: &[(Value, &'static str)], value: Value, ) -> fmt::Result { let mut first = true; let mut accum = value.into(); for &(bit, name) in known { let bit = bit.into(); if bit != 0 && accum & bit == bit { if !first { f.write_str(" | ")?; } f.write_str(name)?; first = false; accum &= !bit; } } if accum != 0 { if !first { f.write_str(" | ")?; } write!(f, "{accum:b}")?; } Ok(()) }