aluminium 0.1.0

vulkan-based graphics engine
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

mod linear_pool;
use std::marker::PhantomData;
use std::sync::atomic::AtomicUsize;
use std::sync::{Arc, Weak};

pub use linear_pool::LinearPool;
use slotmap::SlotMap;

use crate::resources::{Destroy, Res, ResourceKey, Resources};

pub struct Pool<T: Destroy> {
    slots: SlotMap<ResourceKey, T>,
}

impl<T: Destroy> Pool<T> {
    pub fn new() -> Self {
        Self {
            slots: SlotMap::with_key(),
        }
    }

    pub fn insert(&mut self, ctx: Weak<crate::render_context::RenderContext>, resources: Weak<Resources>, value: T) -> Res<T> {
        let key = self.slots.insert(value);
        Res {
            key,
            ref_count: Arc::new(AtomicUsize::new(1)),
            ctx,
            resources,
            _marker: PhantomData,
        }
    }

    pub fn get(&self, res: &Res<T>) -> &T {
        self.slots.get(res.key).expect("Resource not found")
    }

    pub fn get_mut(&mut self, res: &Res<T>) -> &mut T {
        self.slots.get_mut(res.key).expect("Resource not found")
    }

    pub fn remove(&mut self, key: ResourceKey) {
        self.slots.remove(key);
    }
}