vulkan: move vulkan instance creation to separate module
This commit is contained in:
parent
8989dd7987
commit
de0e1ad58e
5 changed files with 271 additions and 231 deletions
243
src/vulkan_core.rs
Normal file
243
src/vulkan_core.rs
Normal file
|
|
@ -0,0 +1,243 @@
|
|||
use {
|
||||
ahash::{AHashMap, AHashSet},
|
||||
ash::{
|
||||
Entry, Instance, LoadingError,
|
||||
ext::{debug_utils, validation_features},
|
||||
vk::{
|
||||
self, API_VERSION_1_3, ApplicationInfo, Bool32, DebugUtilsMessageSeverityFlagsEXT,
|
||||
DebugUtilsMessageTypeFlagsEXT, DebugUtilsMessengerCallbackDataEXT,
|
||||
DebugUtilsMessengerCreateInfoEXT, DebugUtilsMessengerEXT, ExtensionProperties, FALSE,
|
||||
InstanceCreateInfo, LayerProperties, ValidationFeaturesEXT, api_version_major,
|
||||
api_version_minor, api_version_patch, api_version_variant,
|
||||
},
|
||||
},
|
||||
isnt::std_1::collections::IsntHashMapExt,
|
||||
log::Level,
|
||||
once_cell::sync::Lazy,
|
||||
run_on_drop::on_drop,
|
||||
std::{
|
||||
ffi::{CStr, CString, c_void},
|
||||
fmt::{Display, Formatter},
|
||||
slice,
|
||||
sync::Arc,
|
||||
},
|
||||
thiserror::Error,
|
||||
uapi::{Ustr, ustr},
|
||||
};
|
||||
|
||||
static VULKAN_ENTRY: Lazy<Result<Entry, Arc<LoadingError>>> =
|
||||
Lazy::new(|| unsafe { Entry::load() }.map_err(Arc::new));
|
||||
|
||||
static VULKAN_VALIDATION: Lazy<bool> =
|
||||
Lazy::new(|| std::env::var("JAY_VULKAN_VALIDATION").ok().as_deref() == Some("1"));
|
||||
|
||||
#[derive(Debug, Error)]
|
||||
pub enum VulkanCoreError {
|
||||
#[error("Could not load libvulkan.so")]
|
||||
Load(#[source] Arc<LoadingError>),
|
||||
#[error("Could not list instance extensions")]
|
||||
InstanceExtensions(#[source] vk::Result),
|
||||
#[error("Could not list instance layers")]
|
||||
InstanceLayers(#[source] vk::Result),
|
||||
#[error("Missing required instance extension {0:?}")]
|
||||
MissingInstanceExtension(&'static CStr),
|
||||
#[error("Could not create an instance")]
|
||||
CreateInstance(#[source] vk::Result),
|
||||
#[error("Could not create a debug-utils messenger")]
|
||||
Messenger(#[source] vk::Result),
|
||||
}
|
||||
|
||||
pub struct VulkanCoreInstance {
|
||||
_entry: &'static Entry,
|
||||
pub instance: Instance,
|
||||
debug_utils: debug_utils::Instance,
|
||||
messenger: DebugUtilsMessengerEXT,
|
||||
pub log_level: Level,
|
||||
}
|
||||
|
||||
impl VulkanCoreInstance {
|
||||
pub fn new(log_level: Level) -> Result<Self, VulkanCoreError> {
|
||||
let entry = match &*VULKAN_ENTRY {
|
||||
Ok(e) => e,
|
||||
Err(e) => return Err(VulkanCoreError::Load(e.clone())),
|
||||
};
|
||||
let extensions = get_instance_extensions(entry, None)?;
|
||||
for &ext in REQUIRED_INSTANCE_EXTENSIONS {
|
||||
if extensions.not_contains_key(ext) {
|
||||
return Err(VulkanCoreError::MissingInstanceExtension(ext));
|
||||
}
|
||||
}
|
||||
let mut enabled_extensions: Vec<_> = REQUIRED_INSTANCE_EXTENSIONS
|
||||
.iter()
|
||||
.map(|c| c.as_ptr())
|
||||
.collect();
|
||||
let app_info = ApplicationInfo::default()
|
||||
.api_version(API_VERSION)
|
||||
.application_name(c"jay")
|
||||
.application_version(1);
|
||||
let mut severity = DebugUtilsMessageSeverityFlagsEXT::empty()
|
||||
| DebugUtilsMessageSeverityFlagsEXT::ERROR
|
||||
| DebugUtilsMessageSeverityFlagsEXT::WARNING;
|
||||
if *VULKAN_VALIDATION {
|
||||
severity |= DebugUtilsMessageSeverityFlagsEXT::INFO
|
||||
| DebugUtilsMessageSeverityFlagsEXT::VERBOSE;
|
||||
}
|
||||
let types = DebugUtilsMessageTypeFlagsEXT::empty()
|
||||
| DebugUtilsMessageTypeFlagsEXT::VALIDATION
|
||||
| DebugUtilsMessageTypeFlagsEXT::PERFORMANCE
|
||||
| DebugUtilsMessageTypeFlagsEXT::GENERAL;
|
||||
let mut debug_info = DebugUtilsMessengerCreateInfoEXT::default()
|
||||
.message_severity(severity)
|
||||
.message_type(types)
|
||||
.pfn_user_callback(Some(debug_callback));
|
||||
let validation_features = [
|
||||
// ash::vk::ValidationFeatureEnableEXT::DEBUG_PRINTF,
|
||||
// ash::vk::ValidationFeatureEnableEXT::BEST_PRACTICES,
|
||||
// ash::vk::ValidationFeatureEnableEXT::SYNCHRONIZATION_VALIDATION,
|
||||
// ash::vk::ValidationFeatureEnableEXT::GPU_ASSISTED,
|
||||
];
|
||||
let mut validation_info =
|
||||
ValidationFeaturesEXT::default().enabled_validation_features(&validation_features);
|
||||
let mut create_info = InstanceCreateInfo::default()
|
||||
.application_info(&app_info)
|
||||
.push_next(&mut debug_info);
|
||||
let validation_layer_name = VALIDATION_LAYER.as_ptr();
|
||||
if *VULKAN_VALIDATION {
|
||||
if get_available_layers(entry)?.contains(VALIDATION_LAYER) {
|
||||
create_info =
|
||||
create_info.enabled_layer_names(slice::from_ref(&validation_layer_name));
|
||||
let extensions = get_instance_extensions(entry, Some(VALIDATION_LAYER))?;
|
||||
if extensions.contains_key(validation_features::NAME) {
|
||||
enabled_extensions.push(validation_features::NAME.as_ptr());
|
||||
create_info = create_info.push_next(&mut validation_info);
|
||||
} else {
|
||||
log::warn!("{:?} is not available", validation_features::NAME);
|
||||
}
|
||||
} else {
|
||||
log::warn!(
|
||||
"Vulkan validation was requested but validation layers are not available"
|
||||
);
|
||||
}
|
||||
}
|
||||
create_info = create_info.enabled_extension_names(&enabled_extensions);
|
||||
let instance = match unsafe { entry.create_instance(&create_info, None) } {
|
||||
Ok(i) => i,
|
||||
Err(e) => return Err(VulkanCoreError::CreateInstance(e)),
|
||||
};
|
||||
let destroy_instance = on_drop(|| unsafe { instance.destroy_instance(None) });
|
||||
let debug_utils = debug_utils::Instance::new(entry, &instance);
|
||||
let messenger = unsafe { debug_utils.create_debug_utils_messenger(&debug_info, None) };
|
||||
let messenger = match messenger {
|
||||
Ok(m) => m,
|
||||
Err(e) => return Err(VulkanCoreError::Messenger(e)),
|
||||
};
|
||||
destroy_instance.forget();
|
||||
Ok(Self {
|
||||
_entry: entry,
|
||||
instance,
|
||||
debug_utils,
|
||||
messenger,
|
||||
log_level,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
impl Drop for VulkanCoreInstance {
|
||||
fn drop(&mut self) {
|
||||
unsafe {
|
||||
self.debug_utils
|
||||
.destroy_debug_utils_messenger(self.messenger, None);
|
||||
self.instance.destroy_instance(None);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
const REQUIRED_INSTANCE_EXTENSIONS: &[&CStr] = &[debug_utils::NAME];
|
||||
|
||||
const VALIDATION_LAYER: &CStr = c"VK_LAYER_KHRONOS_validation";
|
||||
|
||||
pub type Extensions = AHashMap<CString, u32>;
|
||||
|
||||
fn get_instance_extensions(
|
||||
entry: &Entry,
|
||||
layer: Option<&CStr>,
|
||||
) -> Result<Extensions, VulkanCoreError> {
|
||||
unsafe {
|
||||
entry
|
||||
.enumerate_instance_extension_properties(layer)
|
||||
.map_err(VulkanCoreError::InstanceExtensions)
|
||||
.map(map_extension_properties)
|
||||
}
|
||||
}
|
||||
|
||||
fn get_available_layers(entry: &Entry) -> Result<AHashSet<CString>, VulkanCoreError> {
|
||||
unsafe {
|
||||
entry
|
||||
.enumerate_instance_layer_properties()
|
||||
.map_err(VulkanCoreError::InstanceLayers)
|
||||
.map(map_layer_properties)
|
||||
}
|
||||
}
|
||||
|
||||
fn map_layer_properties(props: Vec<LayerProperties>) -> AHashSet<CString> {
|
||||
props
|
||||
.into_iter()
|
||||
.map(|e| unsafe { CStr::from_ptr(e.layer_name.as_ptr()).to_owned() })
|
||||
.collect()
|
||||
}
|
||||
|
||||
pub fn map_extension_properties(props: Vec<ExtensionProperties>) -> Extensions {
|
||||
props
|
||||
.into_iter()
|
||||
.map(|e| {
|
||||
let s = unsafe { CStr::from_ptr(e.extension_name.as_ptr()) };
|
||||
(s.to_owned(), e.spec_version)
|
||||
})
|
||||
.collect()
|
||||
}
|
||||
|
||||
unsafe extern "system" fn debug_callback(
|
||||
message_severity: DebugUtilsMessageSeverityFlagsEXT,
|
||||
_message_types: DebugUtilsMessageTypeFlagsEXT,
|
||||
p_callback_data: *const DebugUtilsMessengerCallbackDataEXT,
|
||||
_p_user_data: *mut c_void,
|
||||
) -> Bool32 {
|
||||
let _level = match message_severity {
|
||||
DebugUtilsMessageSeverityFlagsEXT::ERROR => Level::Error,
|
||||
DebugUtilsMessageSeverityFlagsEXT::WARNING => Level::Warn,
|
||||
DebugUtilsMessageSeverityFlagsEXT::INFO => Level::Info,
|
||||
DebugUtilsMessageSeverityFlagsEXT::VERBOSE => Level::Trace,
|
||||
_ => Level::Warn,
|
||||
};
|
||||
let data = unsafe { &*p_callback_data };
|
||||
let message = unsafe { Ustr::from_ptr(data.p_message) };
|
||||
let message_id_name = if data.p_message_id_name.is_null() {
|
||||
ustr!("<null>")
|
||||
} else {
|
||||
unsafe { Ustr::from_ptr(data.p_message_id_name) }
|
||||
};
|
||||
log::log!(
|
||||
Level::Info,
|
||||
"VULKAN: {} ({})",
|
||||
message.display(),
|
||||
message_id_name.display()
|
||||
);
|
||||
FALSE
|
||||
}
|
||||
|
||||
pub struct ApiVersionDisplay(pub u32);
|
||||
|
||||
impl Display for ApiVersionDisplay {
|
||||
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
||||
write!(
|
||||
f,
|
||||
"{}.{}.{}.{}",
|
||||
api_version_variant(self.0),
|
||||
api_version_major(self.0),
|
||||
api_version_minor(self.0),
|
||||
api_version_patch(self.0),
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
pub const API_VERSION: u32 = API_VERSION_1_3;
|
||||
Loading…
Add table
Add a link
Reference in a new issue