aboutsummaryrefslogtreecommitdiff
path: root/mingling_core
diff options
context:
space:
mode:
Diffstat (limited to 'mingling_core')
-rw-r--r--mingling_core/src/asset.rs9
-rw-r--r--mingling_core/src/asset/lazy_resource.rs182
-rw-r--r--mingling_core/src/lib.rs1
3 files changed, 189 insertions, 3 deletions
diff --git a/mingling_core/src/asset.rs b/mingling_core/src/asset.rs
index e7945ce..b1fd617 100644
--- a/mingling_core/src/asset.rs
+++ b/mingling_core/src/asset.rs
@@ -11,10 +11,13 @@ pub mod enum_tag;
pub mod global_resource;
#[doc(hidden)]
-pub mod node;
+pub mod help;
#[doc(hidden)]
-pub mod renderer;
+pub mod lazy_resource;
#[doc(hidden)]
-pub mod help;
+pub mod node;
+
+#[doc(hidden)]
+pub mod renderer;
diff --git a/mingling_core/src/asset/lazy_resource.rs b/mingling_core/src/asset/lazy_resource.rs
new file mode 100644
index 0000000..356e348
--- /dev/null
+++ b/mingling_core/src/asset/lazy_resource.rs
@@ -0,0 +1,182 @@
+use crate::{ProgramCollect, ResourceMarker, this};
+
+enum LazyInner<T> {
+ /// Not yet initialized — holds the factory function.
+ /// After init, the factory is **dropped**, no wasted memory.
+ Uninit(Box<dyn FnMut() -> T + Send + Sync>),
+ /// Initialized and ready to go.
+ Init(T),
+}
+
+/// A lazily initialized resource that only creates its value on first access.
+///
+/// Unlike `Option<T>` + persistent `Box<dyn FnMut>`, the factory function is
+/// **consumed and dropped** after initialization. No leftover allocation.
+///
+/// Initialization is triggered by `get_ref()`, `get_mut()`, or `get_clone()`.
+pub struct LazyRes<T: Send + Sync + 'static> {
+ inner: LazyInner<T>,
+}
+
+impl<T: Send + Sync + 'static> LazyRes<T> {
+ /// Creates a new lazily initialized resource with a custom initializer.
+ ///
+ /// The factory `f` is called on first access, then dropped.
+ #[must_use]
+ pub fn new(f: impl FnMut() -> T + Send + Sync + 'static) -> Self {
+ Self {
+ inner: LazyInner::Uninit(Box::new(f)),
+ }
+ }
+
+ fn force_init(&mut self) -> &mut LazyInner<T> {
+ if matches!(&self.inner, LazyInner::Uninit(_)) {
+ // Replace with a temporary poison value so we can move the real factory out.
+ // SAFETY: if the factory panics, the poison value's `unreachable!()` will
+ // catch any subsequent access.
+ let poisoned = LazyInner::Uninit(Box::new(|| {
+ unreachable!("LazyRes poisoned during initialization")
+ }));
+ let old = std::mem::replace(&mut self.inner, poisoned);
+ match old {
+ LazyInner::Uninit(mut f) => {
+ self.inner = LazyInner::Init((f)());
+ }
+ // Already initialized — put it back
+ init @ LazyInner::Init(_) => {
+ self.inner = init;
+ }
+ }
+ }
+ &mut self.inner
+ }
+
+ /// Returns an immutable reference to the inner value,
+ /// calling the initializer on first access.
+ pub fn get_ref(&mut self) -> &T {
+ self.force_init();
+ match &self.inner {
+ LazyInner::Init(t) => t,
+ _ => unreachable!(),
+ }
+ }
+
+ /// Returns a mutable reference to the inner value,
+ /// calling the initializer on first access.
+ pub fn get_mut(&mut self) -> &mut T {
+ self.force_init();
+ match &mut self.inner {
+ LazyInner::Init(t) => t,
+ _ => unreachable!(),
+ }
+ }
+
+ /// Returns a clone of the inner value, calling the initializer if necessary.
+ pub fn get_clone(&mut self) -> T
+ where
+ T: Clone,
+ {
+ self.get_ref().clone()
+ }
+
+ /// Consumes the lazy resource and returns the inner value, if initialized.
+ ///
+ /// Unlike `reset()`, this **drops** the lazy wrapper entirely.
+ /// If you need to re-initialize, just construct a new `LazyRes::new(f)`.
+ pub fn into_inner(self) -> Option<T> {
+ match self.inner {
+ LazyInner::Init(t) => Some(t),
+ LazyInner::Uninit(_) => None,
+ }
+ }
+}
+
+impl<T: Send + Sync + 'static> Default for LazyRes<T>
+where
+ T: Default,
+{
+ /// Creates an uninitialized `LazyRes<T>` whose initializer returns `T::default()`.
+ fn default() -> Self {
+ Self::new(|| T::default())
+ }
+}
+
+impl<T: Send + Sync + 'static> From<T> for LazyRes<T> {
+ /// Creates a `LazyRes<T>` from an already-initialized value.
+ fn from(value: T) -> Self {
+ Self {
+ inner: LazyInner::Init(value),
+ }
+ }
+}
+
+impl<T: Send + Sync + 'static> LazyRes<T> {
+ /// Creates a lazily initialized resource using `T::default()` as the initializer.
+ pub fn lazy_default() -> Self
+ where
+ T: Default,
+ {
+ Self::default()
+ }
+
+ /// Creates a lazily initialized resource with a custom initializer.
+ ///
+ /// Same as `LazyRes::new`.
+ pub fn lazy_init(f: impl FnMut() -> T + Send + Sync + 'static) -> Self {
+ Self::new(f)
+ }
+}
+
+/// Provides convenience methods for types, allowing them to easily
+/// create a corresponding `LazyRes<T>`.
+pub trait LazyInit: Send + Sync + 'static {
+ /// Creates a lazily initialized resource for this type using `Default` as the initializer.
+ fn lazy_default() -> LazyRes<Self>
+ where
+ Self: Default,
+ {
+ LazyRes::default()
+ }
+
+ /// Creates a lazily initialized resource for this type with a custom initializer.
+ fn lazy_init(f: impl FnMut() -> Self + Send + Sync + 'static) -> LazyRes<Self>
+ where
+ Self: Sized,
+ {
+ LazyRes::new(f)
+ }
+}
+
+impl<T: Send + Sync + 'static> LazyInit for T {}
+
+impl<T: Send + Sync + 'static> ResourceMarker for LazyRes<T>
+where
+ T: Default + Clone,
+{
+ /// Clones the lazy resource. The cloned resource retains any initialized value,
+ /// but the initializer is reset to `T::default()`.
+ fn res_clone(&self) -> Self {
+ match &self.inner {
+ LazyInner::Init(t) => Self {
+ inner: LazyInner::Init(t.clone()),
+ },
+ LazyInner::Uninit(_) => Self::default(),
+ }
+ }
+
+ /// Returns a default lazy resource (uninitialized, using `T::default()` as the initializer).
+ fn res_default() -> Self
+ where
+ T: Default,
+ {
+ Self::default()
+ }
+
+ /// Modifies the current lazy resource via the `this` context provided by `C`.
+ fn modify<C>(f: impl FnOnce(&mut Self))
+ where
+ C: ProgramCollect<Enum = C> + 'static,
+ {
+ this::<C>().modify_res(f);
+ }
+}
diff --git a/mingling_core/src/lib.rs b/mingling_core/src/lib.rs
index 6e1e90c..ddb5446 100644
--- a/mingling_core/src/lib.rs
+++ b/mingling_core/src/lib.rs
@@ -30,6 +30,7 @@ pub use crate::asset::dispatcher::*;
pub use crate::asset::enum_tag::*;
pub use crate::asset::global_resource::*;
pub use crate::asset::help::*;
+pub use crate::asset::lazy_resource::*;
pub use crate::asset::node::*;
pub use crate::asset::renderer::*;