nordic-dev.net/vulkano
nordic-dev.net/vulkano
2
0
Fork 0
mirror of https://github.com/vulkano-rs/vulkano.git synced 2024-11-22 06:45:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improvements for the StandardCommandPool (#800)

Browse Source 
* Fix the Arc<StandardCommandPool> being destroyed every time

* Simplify code for alloc

* Add test
This commit is contained in:
tomaka 2017-09-06 09:40:19 +02:00 committed by GitHub
parent 883d0cb94e
commit 6d50ef2891
1 changed files with 26 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
vulkano/src/command_buffer/pool/standard.rs
View File

@ -31,6 +31,9 @@ use device::DeviceOwned;
/// Standard implementation of a command pool.
///
/// It is guaranteed that the allocated command buffers keep the `Arc<StandardCommandPool>` alive.
/// This is desirable so that we can store a `Weak<StandardCommandPool>`.
///
/// Will use one Vulkan pool per thread in order to avoid locking. Will try to reuse command
/// buffers. Command buffers can't be moved between threads during the building process, but
/// finished command buffers can.
@ -92,20 +95,8 @@ unsafe impl CommandPool for Arc<StandardCommandPool> {
// Get an appropriate `Arc<StandardCommandPoolPerThread>`.
let per_thread = match hashmap.entry(thread::current().id()) {
Entry::Occupied(mut entry) => {
if let Some(entry) = entry.get().upgrade() {
entry
} else {
let new_pool =
UnsafeCommandPool::new(self.device.clone(), self.queue_family(), false, true)?;
let pt = Arc::new(StandardCommandPoolPerThread {
pool: Mutex::new(new_pool),
available_primary_command_buffers: MsQueue::new(),
available_secondary_command_buffers: MsQueue::new(),
});
entry.insert(Arc::downgrade(&pt));
pt
}
// The `unwrap()` can't fail, since we retained only valid members earlier.
entry.get().upgrade().unwrap()
},
Entry::Vacant(entry) => {
let new_pool =
@ -138,6 +129,7 @@ unsafe impl CommandPool for Arc<StandardCommandPool> {
inner: StandardCommandPoolAlloc {
cmd: Some(cmd),
pool: per_thread.clone(),
pool_parent: self.clone(),
secondary: secondary,
device: self.device.clone(),
},
@ -159,6 +151,7 @@ unsafe impl CommandPool for Arc<StandardCommandPool> {
inner: StandardCommandPoolAlloc {
cmd: Some(cmd),
pool: per_thread.clone(),
pool_parent: self.clone(),
secondary: secondary,
device: self.device.clone(),
},
@ -228,6 +221,8 @@ pub struct StandardCommandPoolAlloc {
cmd: Option<UnsafeCommandPoolAlloc>,
// We hold a reference to the command pool for our destructor.
pool: Arc<StandardCommandPoolPerThread>,
// Keep alive the `StandardCommandPool`, otherwise it would be destroyed.
pool_parent: Arc<StandardCommandPool>,
// True if secondary command buffer.
secondary: bool,
// The device we belong to. Necessary because of the `DeviceOwned` trait implementation.
@ -281,7 +276,9 @@ impl Drop for StandardCommandPoolAlloc {
mod tests {
use command_buffer::pool::CommandPool;
use command_buffer::pool::CommandPoolBuilderAlloc;
use command_buffer::pool::StandardCommandPool;
use device::Device;
use std::sync::Arc;
use VulkanObject;
#[test]
@ -298,4 +295,19 @@ mod tests {
let cb2 = pool.alloc(false, 1).unwrap().next().unwrap();
assert_eq!(raw, cb2.inner().internal_object());
}
#[test]
fn pool_kept_alive_by_allocs() {
let (device, queue) = gfx_dev_and_queue!();
let pool = Arc::new(StandardCommandPool::new(device, queue.family()));
let pool_weak = Arc::downgrade(&pool);
let cb = pool.alloc(false, 1).unwrap().next().unwrap();
drop(pool);
assert!(pool_weak.upgrade().is_some());
drop(cb);
assert!(pool_weak.upgrade().is_none());
}
}