cesium-allocator 1.0.1

Allocator pool for managing sets of allocators
Documentation 
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

pub mod allocator;

use std::{
    collections::BTreeMap,
    sync::Arc,
};

use cesium_libmimalloc_sys::heap::mi_heap_new;

use crate::allocator::Allocator;

/// A pool of general allocators.
pub struct AllocatorPool {
    lowest_id: u32,
    heaps: BTreeMap<u32, Arc<Allocator>>,
}

impl AllocatorPool {
    /// Create a new pool for allocators.
    pub fn new() -> Self {
        AllocatorPool {
            lowest_id: 0,
            heaps: BTreeMap::new(),
        }
    }

    /// Create a new allocator
    pub fn new_allocator(&mut self) -> Arc<Allocator> {
        let heap = unsafe { mi_heap_new() };
        let id = self.lowest_id + 1;
        self.lowest_id = id;

        let alloc = Arc::new(Allocator::new(id, heap));
        self.heaps.insert(id, alloc.clone());

        alloc
    }

    /// Gets or creates an allocator
    pub fn get_allocator(&mut self, id: u32, create: Option<bool>) -> Option<Arc<Allocator>> {
        match self.heaps.get(&id) {
            | None => match create {
                | None => None,
                | Some(_) => Some(self.new_allocator()),
            },
            | Some(v) => Some(v.clone()),
        }
    }
}