device 0.0.4

A generative 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

use {
  super::*,
  std::alloc::{GlobalAlloc, Layout, System},
};

static ALLOCATED: AtomicUsize = AtomicUsize::new(0);

#[cfg(feature = "allocator")]
#[global_allocator]
static ALLOCATOR: Allocator = Allocator;

pub(crate) struct Allocator;

impl Allocator {
  pub(crate) fn allocated() -> usize {
    ALLOCATED.load(atomic::Ordering::Relaxed)
  }
}

unsafe impl GlobalAlloc for Allocator {
  unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
    let ptr = unsafe { System.alloc(layout) };
    if !ptr.is_null() {
      ALLOCATED.fetch_add(layout.size(), atomic::Ordering::Relaxed);
    }
    ptr
  }

  unsafe fn dealloc(&self, ptr: *mut u8, layout: Layout) {
    unsafe {
      System.dealloc(ptr, layout);
    }
    ALLOCATED.fetch_sub(layout.size(), atomic::Ordering::Relaxed);
  }
}