diff options
| -rw-r--r-- | stockton-skeleton/src/context.rs | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/stockton-skeleton/src/context.rs b/stockton-skeleton/src/context.rs index ea9810f..72b15c7 100644 --- a/stockton-skeleton/src/context.rs +++ b/stockton-skeleton/src/context.rs @@ -274,6 +274,38 @@ impl StatefulRenderingContext<Normal> { Ok(self.existing_memory_pool::<P>().unwrap()) } + /// Allocate memory from the given pool. + /// See [`crate::mem::MemoryPool::alloc`] + pub fn alloc<P: MemoryPool>(&mut self, size: u64, align: u64) -> Result<P::Block> { + self.ensure_memory_pool::<P>()?; + + let device = self.lock_device()?; + let mut pool = self + .existing_memory_pool::<P>() + .unwrap() + .write() + .map_err(|_| LockPoisoned::MemoryPool)?; + + Ok(pool.alloc(&device, size, align)?.0) + } + + /// Free memory from the given pool. + /// See [`crate::mem::MemoryPool::free`] + pub fn free<P: MemoryPool>(&mut self, block: P::Block) -> Result<()> { + self.ensure_memory_pool::<P>()?; + + let device = self.lock_device()?; + let mut pool = self + .existing_memory_pool::<P>() + .unwrap() + .write() + .map_err(|_| LockPoisoned::MemoryPool)?; + + pool.free(&device, block); + + Ok(()) + } + /// Ensure the specified memory pool is initialised. #[allow(clippy::map_entry)] // We can't follow the suggestion because of a borrowing issue pub fn ensure_memory_pool<P: MemoryPool>(&mut self) -> Result<()> { |