use crate::component::Component;
use crate::entity_id::EntityId;
use crate::ViewMut;
pub trait AddComponent<T> {
fn add_component_unchecked(&mut self, entity: EntityId, component: T)
where
Self: Sized;
}
impl AddComponent<()> for () {
#[inline]
fn add_component_unchecked(&mut self, _: EntityId, _: ()) {}
}
impl<T: Component, TRACK> AddComponent<T> for ViewMut<'_, T, TRACK> {
#[inline]
#[track_caller]
fn add_component_unchecked(&mut self, entity: EntityId, component: T) {
let _ = self.sparse_set.insert(entity, component, self.current);
}
}
impl<T: Component, TRACK> AddComponent<T> for &mut ViewMut<'_, T, TRACK> {
#[inline]
#[track_caller]
fn add_component_unchecked(&mut self, entity: EntityId, component: T) {
let _ = self.sparse_set.insert(entity, component, self.current);
}
}
impl<T: Component, TRACK> AddComponent<Option<T>> for ViewMut<'_, T, TRACK> {
#[inline]
#[track_caller]
fn add_component_unchecked(&mut self, entity: EntityId, component: Option<T>) {
if let Some(component) = component {
let _ = self.sparse_set.insert(entity, component, self.current);
}
}
}
impl<T: Component, TRACK> AddComponent<Option<T>> for &mut ViewMut<'_, T, TRACK> {
#[inline]
#[track_caller]
fn add_component_unchecked(&mut self, entity: EntityId, component: Option<T>) {
if let Some(component) = component {
let _ = self.sparse_set.insert(entity, component, self.current);
}
}
}
macro_rules! impl_add_component {
($(($storage: ident, $component: ident, $index: tt))+) => {
impl<$($component: Component,)+ $($storage: AddComponent<$component>,)+> AddComponent<($($component,)+)> for ($($storage,)+) {
#[inline]
#[track_caller]
fn add_component_unchecked(&mut self, entity: EntityId, component: ($($component,)+)) {
$(
let _ = self.$index.add_component_unchecked(entity, component.$index);
)+
}
}
}
}
macro_rules! add_component {
($(($storage: ident, $component: ident, $index: tt))+; ($storage1: ident, $component1: ident, $index1: tt) $(($queue_storage: ident, $queue_component: ident, $queue_index: tt))*) => {
impl_add_component![$(($storage, $component, $index))*];
add_component![$(($storage, $component, $index))* ($storage1, $component1, $index1); $(($queue_storage, $queue_component, $queue_index))*];
};
($(($storage: ident, $component: ident, $index: tt))+;) => {
impl_add_component![$(($storage, $component, $index))*];
}
}
#[cfg(not(feature = "extended_tuple"))]
add_component![(ViewA, A, 0); (ViewB, B, 1) (ViewC, C, 2) (ViewD, D, 3) (ViewE, E, 4) (ViewF, F, 5) (ViewG, G, 6) (ViewH, H, 7) (ViewI, I, 8) (ViewJ, J, 9)];
#[cfg(feature = "extended_tuple")]
add_component![
(ViewA, A, 0); (ViewB, B, 1) (ViewC, C, 2) (ViewD, D, 3) (ViewE, E, 4) (ViewF, F, 5) (ViewG, G, 6) (ViewH, H, 7) (ViewI, I, 8) (ViewJ, J, 9)
(ViewK, K, 10) (ViewL, L, 11) (ViewM, M, 12) (ViewN, N, 13) (ViewO, O, 14) (ViewP, P, 15) (ViewQ, Q, 16) (ViewR, R, 17) (ViewS, S, 18) (ViewT, T, 19)
(ViewU, U, 20) (ViewV, V, 21) (ViewW, W, 22) (ViewX, X, 23) (ViewY, Y, 24) (ViewZ, Z, 25) (ViewAA, AA, 26) (ViewBB, BB, 27) (ViewCC, CC, 28) (ViewDD, DD, 29)
(ViewEE, EE, 30) (ViewFF, FF, 31)
];