Spaces:
Build error
Build error
File size: 1,942 Bytes
84d2a97 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
use std::sync::Arc;
use ash::vk;
use crate::*;
/// Struct that represents a compiled shader.
/// If you want to create a new shader, you have to compile it first using `shaderc`.
pub struct Shader {
/// Device that owns the shader.
device: Arc<Device>,
/// Vulkan shader module handle.
vk_shader_module: vk::ShaderModule,
}
// Mark `Shader` as a GPU resource that should be kept alive while it's in use by the GPU context.
impl Resource for Shader {}
impl Shader {
/// Create a new shader from the given compiled shader code.
/// `shader_code` is a compiled shader code in the binary SPIR-V format.
pub fn new(device: Arc<Device>, shader_code: &[u8]) -> GpuResult<Arc<Self>> {
// Decode SPIR-V from bytes with correct alignment.
let mut spv_file = std::io::Cursor::new(shader_code);
let shader_code = ash::util::read_spv(&mut spv_file)
.map_err(|_| GpuError::Other("Failed to read SPIR-V shader code".to_string()))?;
// Create shader.
let shader_module_create_info = vk::ShaderModuleCreateInfo::default().code(&shader_code);
let shader_module = unsafe {
device.vk_device().create_shader_module(
&shader_module_create_info,
device.cpu_allocation_callbacks(),
)?
};
Ok(Arc::new(Self {
device,
vk_shader_module: shader_module,
}))
}
pub fn vk_shader_module(&self) -> vk::ShaderModule {
self.vk_shader_module
}
}
impl Drop for Shader {
fn drop(&mut self) {
if self.vk_shader_module != vk::ShaderModule::null() {
unsafe {
self.device.vk_device().destroy_shader_module(
self.vk_shader_module,
self.device.cpu_allocation_callbacks(),
);
}
self.vk_shader_module = vk::ShaderModule::null();
}
}
}
|