path: root/ecs/src/lib.rs

#![deny(clippy::all, clippy::pedantic)]

use std::any::{Any, TypeId};
use std::collections::{HashMap, HashSet};
use std::fmt::Debug;
use std::marker::PhantomData;
use std::slice::{Iter as SliceIter, IterMut as SliceIterMut};

use crate::component::{Component, Sequence as ComponentSequence};
use crate::event::{Event, Id as EventId};
use crate::system::{
    NoInitParamFlag as NoInitSystemParamFlag,
    Param as SystemParam,
    System,
    TypeErased as TypeErasedSystem,
};
use crate::tuple::FilterExclude as TupleFilterExclude;

pub mod component;
pub mod event;
pub mod system;
pub mod tuple;

pub use ecs_macros::Component;

#[derive(Debug, Default)]
struct Entity
{
    components: Vec<Box<dyn Component>>,
}

#[derive(Debug, Default)]
pub struct World
{
    systems: Vec<TypeErasedSystem>,
    data: WorldData,
}

impl World
{
    #[must_use]
    pub fn new() -> Self
    {
        Self::default()
    }

    pub fn create_entity<Comps>(&mut self, components: Comps)
    where
        Comps: ComponentSequence,
    {
        self.data
            .component_storage
            .entities
            .push(Entity { components: components.into_vec() });
    }

    pub fn register_system<TSystem, SystemImpl>(
        &mut self,
        event: &impl Event,
        system: TSystem,
    ) where
        TSystem: System<SystemImpl>,
    {
        self.systems.push(system.into_type_erased());

        self.data
            .events
            .entry(event.id())
            .or_default()
            .push(self.systems.len() - 1);
    }

    /// Emits a event, running all systems listening to the event for each compatible
    /// entity.
    ///
    /// # Panics
    /// Will panic if a system has dissapeared.
    pub fn emit(&mut self, event: &impl Event)
    {
        let Some(system_indices) = self.data.events.get(&event.id()).cloned() else {
            return;
        };

        for system_index in system_indices {
            let system = self.systems.get_mut(system_index).unwrap();

            system.run(&mut self.data);
        }
    }

    pub fn query<Comps>(&mut self) -> Query<Comps>
    where
        Comps: ComponentSequence,
    {
        Query::new(&mut self.data)
    }
}

#[derive(Debug, Default)]
pub struct WorldData
{
    events: HashMap<EventId, Vec<usize>>,
    component_storage: ComponentStorage,
}

#[derive(Debug)]
pub struct Query<'world, Comps>
where
    Comps: ComponentSequence,
{
    world_data: &'world mut WorldData,
    comps_pd: PhantomData<Comps>,
}

impl<'world, Comps> Query<'world, Comps>
where
    Comps: ComponentSequence,
{
    fn new(world_data: &'world mut WorldData) -> Self
    {
        Self { world_data, comps_pd: PhantomData }
    }
}

impl<'world, Comps> Query<'world, Comps>
where
    Comps: ComponentSequence,
{
    pub fn iter(&self) -> QueryComponentIter<Comps>
    {
        QueryComponentIter {
            entity_iter: self.world_data.component_storage.entities.iter(),
            component_type_ids: Comps::type_ids(),
            comps_pd: PhantomData,
        }
    }

    pub fn iter_mut(&mut self) -> QueryComponentMutIter<Comps>
    {
        QueryComponentMutIter {
            entity_iter: self.world_data.component_storage.entities.iter_mut(),
            component_type_ids: Comps::type_ids(),
            comps_pd: PhantomData,
        }
    }
}

impl<'world, Comps> IntoIterator for &'world Query<'world, Comps>
where
    Comps: ComponentSequence,
{
    type IntoIter = QueryComponentIter<'world, Comps>;
    type Item = Comps::Refs<'world>;

    fn into_iter(self) -> Self::IntoIter
    {
        self.iter()
    }
}

impl<'world, Comps> IntoIterator for &'world mut Query<'world, Comps>
where
    Comps: ComponentSequence,
{
    type IntoIter = QueryComponentMutIter<'world, Comps>;
    type Item = Comps::MutRefs<'world>;

    fn into_iter(self) -> Self::IntoIter
    {
        self.iter_mut()
    }
}

unsafe impl<'world, Comps> SystemParam<'world> for Query<'world, Comps>
where
    Comps: ComponentSequence,
{
    type Flags = NoInitSystemParamFlag;
    type Input = TupleFilterExclude;

    fn initialize<SystemImpl>(_system: &mut impl System<SystemImpl>, _input: Self::Input)
    {
    }

    fn new<SystemImpl>(
        _system: &'world mut impl System<SystemImpl>,
        world_data: &'world mut WorldData,
    ) -> Self
    {
        Self::new(world_data)
    }

    fn is_compatible<Other: SystemParam<'world>>() -> bool
    {
        let other_comparable = Other::get_comparable();

        let Some(other_query_component_ids) =
            other_comparable.downcast_ref::<QueryComponentIds>()
        else {
            return true;
        };

        !other_query_component_ids.contains_component_in::<Comps>()
    }

    fn get_comparable() -> Box<dyn Any>
    {
        Box::new(QueryComponentIds {
            component_type_ids: Comps::type_ids(),
        })
    }
}

#[derive(Debug)]
struct QueryComponentIds
{
    component_type_ids: Vec<TypeId>,
}

impl QueryComponentIds
{
    fn contains_component_in<OtherComps>(&self) -> bool
    where
        OtherComps: ComponentSequence,
    {
        let other_component_type_ids = OtherComps::type_ids()
            .into_iter()
            .collect::<HashSet<TypeId>>();

        // TODO: Make this a bit smarter. Queries with a same component can be compatible
        // if one of the queries have a component the other one does not have
        self.component_type_ids
            .iter()
            .any(|component_type_id| other_component_type_ids.contains(component_type_id))
    }
}

pub struct QueryComponentIter<'world, Comps>
{
    entity_iter: SliceIter<'world, Entity>,
    component_type_ids: Vec<TypeId>,
    comps_pd: PhantomData<Comps>,
}

impl<'world, Comps> Iterator for QueryComponentIter<'world, Comps>
where
    Comps: ComponentSequence + 'world,
{
    type Item = Comps::Refs<'world>;

    fn next(&mut self) -> Option<Self::Item>
    {
        let matching_entity = find_entity_with_components::<&Entity>(
            &mut self.entity_iter,
            &self.component_type_ids,
        )?;

        Some(Comps::from_components(&matching_entity.components))
    }
}

pub struct QueryComponentMutIter<'world, Comps>
{
    entity_iter: SliceIterMut<'world, Entity>,
    component_type_ids: Vec<TypeId>,
    comps_pd: PhantomData<Comps>,
}

impl<'world, Comps> Iterator for QueryComponentMutIter<'world, Comps>
where
    Comps: ComponentSequence + 'world,
{
    type Item = Comps::MutRefs<'world>;

    fn next(&mut self) -> Option<Self::Item>
    {
        let matching_entity = find_entity_with_components::<&mut Entity>(
            &mut self.entity_iter,
            &self.component_type_ids,
        )?;

        Some(Comps::from_components_mut(&mut matching_entity.components))
    }
}

trait EntityRef
{
    fn components(&self) -> &[Box<dyn Component>];
}

impl EntityRef for &Entity
{
    fn components(&self) -> &[Box<dyn Component>]
    {
        &self.components
    }
}

impl EntityRef for &mut Entity
{
    fn components(&self) -> &[Box<dyn Component>]
    {
        &self.components
    }
}

fn find_entity_with_components<EntityRefT>(
    entity_iter: &mut impl Iterator<Item = EntityRefT>,
    component_type_ids: &[TypeId],
) -> Option<EntityRefT>
where
    EntityRefT: EntityRef,
{
    // TODO: This is a really dumb and slow way to do this. Refactor the world
    // to store components in archetypes
    entity_iter.find(|entity| {
        let entity_components: HashSet<_> = entity
            .components()
            .iter()
            .map(|component| component.as_ref().type_id())
            .collect();

        if component_type_ids
            .iter()
            .all(|component_type_id| entity_components.contains(component_type_id))
        {
            return true;
        }

        false
    })
}

#[derive(Debug, Default)]
pub struct ComponentStorage
{
    entities: Vec<Entity>,
}