rumtk_arena/
dune.rs

1/*
2 *     rumtk attempts to implement HL7 and medical protocols for interoperability in medicine.
3 *     This toolkit aims to be reliable, simple, performant, and standards compliant.
4 *     Copyright (C) 2026  Luis M. Santos, M.D. <lsantos@medicalmasses.com>
5 *     Copyright (C) 2026  MedicalMasses L.L.C. <contact@medicalmasses.com>
6 *
7 *     This program is free software: you can redistribute it and/or modify
8 *     it under the terms of the GNU General Public License as published by
9 *     the Free Software Foundation, either version 3 of the License, or
10 *     (at your option) any later version.
11 *
12 *     This program is distributed in the hope that it will be useful,
13 *     but WITHOUT ANY WARRANTY; without even the implied warranty of
14 *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 *     GNU General Public License for more details.
16 *
17 *     You should have received a copy of the GNU General Public License
18 *     along with this program.  If not, see <https://www.gnu.org/licenses/>.
19 */
20use crate::{direct_alloc, direct_dealloc, MemoryPool};
21use std::alloc::GlobalAlloc;
22use std::alloc::Layout;
23use std::sync::Mutex;
24
25pub struct Arrakis {
26    dunes: Mutex<MemoryPool>,
27}
28
29impl Arrakis {
30    pub const fn with_capacity(allocation_size: usize) -> Self {
31        Self { dunes: Mutex::new(MemoryPool::with_chunk_size(allocation_size)) }
32    }
33
34    #[inline]
35    unsafe fn allocate(&self, layout: Layout) -> *mut u8 {
36        let mut dunes = self.dunes.lock().unwrap();
37        dunes.allocate(layout)
38    }
39
40    #[inline]
41    unsafe fn deallocate(&self,ptr: *mut u8, layout: Layout) {
42        let mut dunes = self.dunes.lock().unwrap();
43        dunes.deallocate(ptr, layout);
44    }
45}
46
47unsafe impl GlobalAlloc for Arrakis {
48    #[cfg(feature = "fast_global_allocator")]
49    #[inline]
50    unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
51        direct_alloc(layout.size())
52    }
53
54    #[cfg(not(feature = "fast_global_allocator"))]
55    #[inline]
56    unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
57        self.allocate(layout)
58    }
59    #[cfg(feature = "fast_global_allocator")]
60    #[inline]
61    unsafe fn dealloc(&self, ptr: *mut u8, _layout: Layout) {
62        direct_dealloc(ptr, _layout.size())
63    }
64
65    #[cfg(not(feature = "fast_global_allocator"))]
66    #[inline]
67    unsafe fn dealloc(&self, ptr: *mut u8, _layout: Layout) {
68        self.deallocate(ptr, _layout)
69    }
70}
71
72#[macro_export]
73macro_rules! rumtk_dune_new {
74    (  ) => {{
75        use $crate::constants::DEFAULT_GLOBAL_MB_ALLOCATION;
76        rumtk_dune_new!(DEFAULT_GLOBAL_MB_ALLOCATION)
77    }};
78    ( $size:expr ) => {{
79        use std::sync::LazyLock;
80        use $crate::dune::{Arrakis};
81        use $crate::constants::DEFAULT_GLOBAL_MB_ALLOCATION;
82        Arrakis::with_capacity(DEFAULT_GLOBAL_MB_ALLOCATION)
83    }}
84}
85
86
rumtk_arena/dune.rs

rumtk_arena/
dune.rs
