fork/Graphite
1
0
Fork 0
mirror of https://github.com/GraphiteEditor/Graphite.git synced 2026-09-16 23:08:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Implement node registry (#822)

Browse Source
This commit is contained in:
TrueDoctor
2022-10-26 00:32:50 +02:00
committed by Keavon Chambers
parent c3fbc4eac9
commit 11c6413251
10 changed files with 213 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
node-graph/borrow_stack/src/lib.rs
View File

@@ -5,25 +5,24 @@ use std::{
sync::atomic::{AtomicUsize, Ordering},
};
pub trait BorrowStack<'n> {
pub trait BorrowStack {
type Item;
/// # Safety
unsafe fn push(&self, value: Self::Item);
/// # Safety
unsafe fn pop(&self);
/// # Safety
unsafe fn get(&self) -> &'n [Self::Item];
unsafe fn get(&self) -> &'static [Self::Item];
}
#[derive(Debug)]
pub struct FixedSizeStack<'n, T> {
pub struct FixedSizeStack<T: dyn_any::StaticTypeSized> {
data: Pin<Box<[MaybeUninit<T>]>>,
capacity: usize,
len: AtomicUsize,
_phantom: PhantomData<&'n ()>,
}
impl<'n, T: Unpin + 'n> FixedSizeStack<'n, T> {
impl<'n, T: Unpin + 'n + dyn_any::StaticTypeSized> FixedSizeStack<T> {
pub fn new(capacity: usize) -> Self {
let layout = std::alloc::Layout::array::<MaybeUninit<T>>(capacity).unwrap();
let array = unsafe { std::alloc::alloc(layout) };
@@ -33,7 +32,6 @@ impl<'n, T: Unpin + 'n> FixedSizeStack<'n, T> {
data: array,
capacity,
len: AtomicUsize::new(0),
_phantom: PhantomData,
}
}
@@ -44,19 +42,19 @@ impl<'n, T: Unpin + 'n> FixedSizeStack<'n, T> {
pub fn is_empty(&self) -> bool {
self.len.load(Ordering::SeqCst) == 0
}
pub fn push_fn(&self, f: impl FnOnce(&'n [T]) -> T) {
unsafe { self.push(f(self.get())) }
pub fn push_fn(&self, f: impl FnOnce(&'static [T::Static]) -> T) {
unsafe { self.push(std::mem::transmute_copy(&f(self.get()))) }
}
}
impl<'n, T> BorrowStack<'n> for FixedSizeStack<'n, T> {
type Item = T;
impl<'n, T: 'n + dyn_any::StaticTypeSized> BorrowStack for FixedSizeStack<T> {
type Item = T::Static;
unsafe fn push(&self, value: Self::Item) {
let len = self.len.load(Ordering::SeqCst);
assert!(len < self.capacity);
let ptr = self.data[len].as_ptr();
(ptr as *mut T).write(value);
(ptr as *mut T::Static).write(value);
self.len.fetch_add(1, Ordering::SeqCst);
}
@@ -66,8 +64,8 @@ impl<'n, T> BorrowStack<'n> for FixedSizeStack<'n, T> {
self.len.fetch_sub(1, Ordering::SeqCst);
}
unsafe fn get(&self) -> &'n [Self::Item] {
std::slice::from_raw_parts(self.data.as_ptr() as *const T, self.len.load(Ordering::SeqCst))
unsafe fn get(&self) -> &'static [Self::Item] {
std::slice::from_raw_parts(self.data.as_ptr() as *const T::Static, self.len.load(Ordering::SeqCst))
}
}