use std::any::TypeId;
use std::marker::PhantomData;
use std::prelude::v1::*;
use downcast_rs::Downcast;
use fnv::FnvHashMap as HashMap;
use crate::trust_cell::{Ref, RefMut, TrustCell};
#[derive(Clone, Debug, Eq, Hash, Ord, PartialEq, PartialOrd)]
pub struct ResourceId {
type_id: TypeId,
}
impl ResourceId {
#[inline]
pub fn new<T: 'static>() -> Self {
ResourceId {
type_id: std::any::TypeId::of::<T>(),
}
}
}
pub trait Resource: Downcast + Send + Sync + 'static {}
impl<T> Resource for T where T: Downcast + Send + Sync {}
mod __resource_mopafy_scope {
#![allow(clippy::all)]
use super::Resource;
downcast_rs::impl_downcast!(Resource);
}
pub struct ResourceMapBuilder {
resource_map: ResourceMap,
}
impl ResourceMapBuilder {
pub fn new() -> Self {
ResourceMapBuilder {
resource_map: ResourceMap::new(),
}
}
pub fn with_resource<R>(
mut self,
r: R,
) -> Self
where
R: Resource,
{
self.resource_map.insert(r);
self
}
pub fn insert<R>(
&mut self,
r: R,
) where
R: Resource,
{
self.resource_map.insert(r);
}
pub fn build(self) -> ResourceMap {
self.resource_map
}
}
#[derive(Default)]
pub struct ResourceMap {
resources: HashMap<ResourceId, TrustCell<Box<dyn Resource>>>,
}
impl ResourceMap {
pub fn new() -> Self {
ResourceMap {
resources: HashMap::default(),
}
}
pub fn try_insert_default<R>(&mut self)
where
R: Resource + Default,
{
if !self.has_value::<R>() {
self.insert(R::default());
}
}
pub fn insert<R>(
&mut self,
r: R,
) where
R: Resource,
{
self.insert_by_id(ResourceId::new::<R>(), r);
}
pub fn remove<R>(&mut self) -> Option<R>
where
R: Resource,
{
self.remove_by_id(ResourceId::new::<R>())
}
fn insert_by_id<R>(
&mut self,
id: ResourceId,
r: R,
) where
R: Resource,
{
let _old = self.resources.insert(id, TrustCell::new(Box::new(r)));
}
fn remove_by_id<R>(
&mut self,
id: ResourceId,
) -> Option<R>
where
R: Resource,
{
self.resources
.remove(&id)
.map(TrustCell::into_inner)
.map(|x: Box<dyn Resource>| x.downcast())
.map(|x: Result<Box<R>, _>| x.ok().unwrap())
.map(|x| *x)
}
fn unwrap_resource<R>(resource: Option<R>) -> R {
if resource.is_none() {
let name = core::any::type_name::<R>();
panic!("Resource not found: {}", name);
}
resource.unwrap()
}
pub fn fetch<R: Resource>(&self) -> ReadBorrow<R> {
let result = self.try_fetch();
Self::unwrap_resource(result)
}
pub fn try_fetch<R: Resource>(&self) -> Option<ReadBorrow<R>> {
let res_id = ResourceId::new::<R>();
self.resources.get(&res_id).map(|r| ReadBorrow {
inner: Ref::map(r.borrow(), Box::as_ref),
phantom: PhantomData,
})
}
pub fn fetch_mut<R: Resource>(&self) -> WriteBorrow<R> {
let result = self.try_fetch_mut();
Self::unwrap_resource(result)
}
pub fn try_fetch_mut<R: Resource>(&self) -> Option<WriteBorrow<R>> {
let res_id = ResourceId::new::<R>();
self.resources.get(&res_id).map(|r| WriteBorrow::<R> {
inner: RefMut::map(r.borrow_mut(), Box::as_mut),
phantom: PhantomData,
})
}
pub fn has_value<R>(&self) -> bool
where
R: Resource,
{
self.has_value_raw(ResourceId::new::<R>())
}
fn has_value_raw(
&self,
id: ResourceId,
) -> bool {
self.resources.contains_key(&id)
}
pub fn keys(&self) -> impl Iterator<Item = &ResourceId> {
self.resources.iter().map(|x| x.0)
}
}
pub trait DataRequirement<'a> {
type Borrow: DataBorrow;
fn fetch(resource_map: &'a ResourceMap) -> Self::Borrow;
}
impl<'a> DataRequirement<'a> for () {
type Borrow = ();
fn fetch(_: &'a ResourceMap) -> Self::Borrow {}
}
pub struct Read<T: Resource> {
phantom_data: PhantomData<T>,
}
pub type ReadOption<T> = Option<Read<T>>;
impl<'a, T: Resource> DataRequirement<'a> for Read<T> {
type Borrow = ReadBorrow<'a, T>;
fn fetch(resource_map: &'a ResourceMap) -> Self::Borrow {
resource_map.fetch::<T>()
}
}
impl<'a, T: Resource> DataRequirement<'a> for Option<Read<T>> {
type Borrow = Option<ReadBorrow<'a, T>>;
fn fetch(resource_map: &'a ResourceMap) -> Self::Borrow {
resource_map.try_fetch::<T>()
}
}
pub struct Write<T: Resource> {
phantom_data: PhantomData<T>,
}
pub type WriteOption<T> = Option<Write<T>>;
impl<'a, T: Resource> DataRequirement<'a> for Write<T> {
type Borrow = WriteBorrow<'a, T>;
fn fetch(resource_map: &'a ResourceMap) -> Self::Borrow {
resource_map.fetch_mut::<T>()
}
}
impl<'a, T: Resource> DataRequirement<'a> for Option<Write<T>> {
type Borrow = Option<WriteBorrow<'a, T>>;
fn fetch(resource_map: &'a ResourceMap) -> Self::Borrow {
resource_map.try_fetch_mut::<T>()
}
}
pub trait DataBorrow {}
impl DataBorrow for () {}
pub struct ReadBorrow<'a, T> {
inner: Ref<'a, dyn Resource>,
phantom: PhantomData<&'a T>,
}
impl<'a, T> DataBorrow for ReadBorrow<'a, T> {}
impl<'a, T> DataBorrow for Option<ReadBorrow<'a, T>> {}
impl<'a, T> std::ops::Deref for ReadBorrow<'a, T>
where
T: Resource,
{
type Target = T;
fn deref(&self) -> &T {
self.inner.downcast_ref().unwrap()
}
}
impl<'a, T> Clone for ReadBorrow<'a, T> {
fn clone(&self) -> Self {
ReadBorrow {
inner: self.inner.clone(),
phantom: PhantomData,
}
}
}
pub struct WriteBorrow<'a, T> {
inner: RefMut<'a, dyn Resource>,
phantom: PhantomData<&'a mut T>,
}
impl<'a, T> DataBorrow for WriteBorrow<'a, T> {}
impl<'a, T> DataBorrow for Option<WriteBorrow<'a, T>> {}
impl<'a, T> std::ops::Deref for WriteBorrow<'a, T>
where
T: Resource,
{
type Target = T;
fn deref(&self) -> &T {
self.inner.downcast_ref().unwrap()
}
}
impl<'a, T> std::ops::DerefMut for WriteBorrow<'a, T>
where
T: Resource,
{
fn deref_mut(&mut self) -> &mut T {
self.inner.downcast_mut().unwrap()
}
}
macro_rules! impl_data {
( $($ty:ident),* ) => {
impl<$($ty),*> DataBorrow for ( $( $ty , )* )
where $( $ty : DataBorrow ),*
{
}
impl<'a, $($ty),*> DataRequirement<'a> for ( $( $ty , )* )
where $( $ty : DataRequirement<'a> ),*
{
type Borrow = ( $( <$ty as DataRequirement<'a>>::Borrow, )* );
fn fetch(resource_map: &'a ResourceMap) -> Self::Borrow {
#![allow(unused_variables)]
( $( <$ty as DataRequirement<'a>>::fetch(resource_map), )* )
}
}
};
}
mod impl_data {
#![cfg_attr(rustfmt, rustfmt_skip)]
use super::*;
impl_data!(A);
impl_data!(A, B);
impl_data!(A, B, C);
impl_data!(A, B, C, D);
impl_data!(A, B, C, D, E);
impl_data!(A, B, C, D, E, F);
impl_data!(A, B, C, D, E, F, G);
impl_data!(A, B, C, D, E, F, G, H);
impl_data!(A, B, C, D, E, F, G, H, I);
impl_data!(A, B, C, D, E, F, G, H, I, J);
}