63 lines
2.2 KiB
Rust
63 lines
2.2 KiB
Rust
use std::{any::{Any, TypeId}, cell::{Ref, RefMut}, ptr::NonNull};
|
|
|
|
use lyra_ecs::{World, ResourceObject};
|
|
|
|
use crate::{Reflect, FromType};
|
|
|
|
#[derive(Clone)]
|
|
pub struct ReflectedResource {
|
|
pub type_id: TypeId,
|
|
|
|
fn_reflect: for<'a> fn (world: &'a World) -> Option<Ref<'a, dyn Reflect>>,
|
|
fn_reflect_mut: for<'a> fn (world: &'a mut World) -> Option<RefMut<'a, dyn Reflect>>,
|
|
fn_reflect_ptr: fn (world: &mut World) -> Option<NonNull<u8>>,
|
|
fn_refl_insert: fn (world: &mut World, this: Box<dyn Reflect>),
|
|
}
|
|
|
|
impl ReflectedResource {
|
|
/// Retrieves the reflected resource from the world.
|
|
pub fn reflect<'a>(&self, world: &'a World) -> Option<Ref<'a, dyn Reflect>> {
|
|
(self.fn_reflect)(world)
|
|
}
|
|
|
|
/// Retrieves a mutable reflected resource from the world.
|
|
pub fn reflect_mut<'a>(&self, world: &'a mut World) -> Option<RefMut<'a, dyn Reflect>> {
|
|
(self.fn_reflect_mut)(world)
|
|
}
|
|
|
|
pub fn reflect_ptr(&self, world: &mut World) -> Option<NonNull<u8>> {
|
|
(self.fn_reflect_ptr)(world)
|
|
}
|
|
|
|
/// Insert the resource into the world.
|
|
pub fn insert(&self, world: &mut World, this: Box<dyn Reflect>) {
|
|
(self.fn_refl_insert)(world, this)
|
|
}
|
|
}
|
|
|
|
impl<T: ResourceObject + Reflect> FromType<T> for ReflectedResource {
|
|
fn from_type() -> Self {
|
|
Self {
|
|
type_id: TypeId::of::<T>(),
|
|
fn_reflect: |world: &World| {
|
|
world.try_get_resource::<T>()
|
|
.map(|r| r as Ref<dyn Reflect>)
|
|
},
|
|
fn_reflect_mut: |world: &mut World| {
|
|
world.try_get_resource_mut::<T>()
|
|
.map(|r| r as RefMut<dyn Reflect>)
|
|
},
|
|
fn_reflect_ptr: |world: &mut World| unsafe {
|
|
world.try_get_resource_ptr::<T>()
|
|
.map(|ptr| ptr.cast::<u8>())
|
|
},
|
|
fn_refl_insert: |world: &mut World, this: Box<dyn Reflect>| {
|
|
let res = this as Box<dyn Any>;
|
|
let res = res.downcast::<T>()
|
|
.expect("Provided a non-matching type to ReflectedResource insert method!");
|
|
let res = *res;
|
|
world.add_resource(res);
|
|
}
|
|
}
|
|
}
|
|
} |