52 lines
1.5 KiB
Rust
52 lines
1.5 KiB
Rust
use ash::vk;
|
|
use gpu_allocator::vulkan::{Allocation, AllocationCreateDesc, AllocationScheme, Allocator};
|
|
use gpu_allocator::MemoryLocation;
|
|
use parking_lot::RwLock;
|
|
|
|
use ash::prelude::VkResult;
|
|
use std::sync::Arc;
|
|
|
|
pub struct VulkanImageMemory {
|
|
allocation: Option<Allocation>,
|
|
allocator: Arc<RwLock<Allocator>>,
|
|
}
|
|
|
|
impl VulkanImageMemory {
|
|
pub fn new(
|
|
device: &Arc<ash::Device>,
|
|
allocator: &Arc<RwLock<Allocator>>,
|
|
requirements: vk::MemoryRequirements,
|
|
image: &vk::Image,
|
|
) -> VkResult<VulkanImageMemory> {
|
|
let allocation = allocator
|
|
.write()
|
|
.allocate(&AllocationCreateDesc {
|
|
name: "imagemem",
|
|
requirements,
|
|
location: MemoryLocation::GpuOnly,
|
|
linear: false,
|
|
allocation_scheme: AllocationScheme::DedicatedImage(*image),
|
|
})
|
|
.unwrap();
|
|
|
|
unsafe {
|
|
device.bind_image_memory(*image, allocation.memory(), 0)?;
|
|
Ok(VulkanImageMemory {
|
|
allocation: Some(allocation),
|
|
allocator: Arc::clone(allocator),
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Drop for VulkanImageMemory {
|
|
fn drop(&mut self) {
|
|
let allocation = self.allocation.take();
|
|
if let Some(allocation) = allocation {
|
|
if let Err(e) = self.allocator.write().free(allocation) {
|
|
println!("librashader-runtime-vk: [warn] failed to deallocate image buffer {e}")
|
|
}
|
|
}
|
|
}
|
|
}
|