lyra-engine/lyra-ecs/src/component_info.rs

use std::{any::{TypeId, type_name}, alloc::{Layout, LayoutError}};

#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub struct MemoryLayout {
    pub size: usize,
    pub alignment: usize
}

impl TryInto<Layout> for MemoryLayout {
    type Error = LayoutError;

    fn try_into(self) -> Result<Layout, Self::Error> {
        Layout::from_size_align(self.size, self.alignment)
    }
}

impl From<Layout> for MemoryLayout {
    fn from(value: Layout) -> Self {
        Self {
            size: value.size(),
            alignment: value.align(),
        }
    }
}

impl MemoryLayout {
    pub fn new(size: usize, alignment: usize) -> Self {
        MemoryLayout {
            size,
            alignment
        }
    }

    pub unsafe fn into_layout_unchecked(self) -> Layout {
        Layout::from_size_align_unchecked(self.size, self.alignment)
    }
}

/// A dynamic type id. Supports types that are not known to Rust.
#[derive(Clone, Copy, Hash, Debug, PartialEq, Eq)]
pub enum DynTypeId {
    Rust(TypeId),
    Unknown(u128),
}

impl Into<DynTypeId> for u128 {
    fn into(self) -> DynTypeId {
        DynTypeId::Unknown(self)
    }
}

impl Into<DynTypeId> for TypeId {
    fn into(self) -> DynTypeId {
        DynTypeId::Rust(self)
    }
}

/* impl From<TypeId> for DynTypeId {
    fn from(value: TypeId) -> Self {
        DynTypeId::Rust(value)
    }
} */

impl DynTypeId {
    pub fn of<T: 'static>() -> Self {
        Self::Rust(TypeId::of::<T>())
    }

    pub fn is<T: 'static>(&self) -> bool {
        match self {
            DynTypeId::Rust(tyid) => tyid.clone() == TypeId::of::<T>(),
            DynTypeId::Unknown(_) => false,
        }
    }

    pub fn is_id<I: Into<DynTypeId>>(&self, id: I) -> bool {
        let id = id.into();
        *self == id
    }

        /// Force self into a rust TypeId, will panic if this type is not a Rust type.
    pub fn as_rust(&self) -> TypeId {
        match self {
            DynTypeId::Rust(t) => t.clone(),
            DynTypeId::Unknown(_) => panic!("This type is unknown to rust, cannot construct a TypeId from it!"),
        }
    }
}

#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub struct ComponentInfo {
    pub type_id: DynTypeId,
    //pub name: String,
    pub layout: MemoryLayout,
}

impl ComponentInfo {
    pub fn new<T: 'static>() -> Self {
        Self {
            type_id: TypeId::of::<T>().into(),
            //name: type_name::<T>().to_string(),
            layout: MemoryLayout::from(Layout::new::<T>()),
        }
    }

    /// Create ComponentInfo from a type that is not known to rust
    pub fn new_unknown<D>(type_id: D, name: &str, layout: MemoryLayout) -> Self
    where
        D: Into<DynTypeId>,
    {
        Self {
            type_id: type_id.into(),
            //name: name.to_string(),
            layout,
        }
    }
}