Document part of bevy_ecs::Commands (#976)

Document part of bevy_ecs::Commands
This commit is contained in:
Michael Tang 2020-12-02 19:11:28 -08:00 committed by GitHub
parent 71e2c7f4e4
commit ff12f69233
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 118 additions and 28 deletions

View File

@ -22,6 +22,8 @@ use std::{
}; };
/// A dynamically typed collection of components /// A dynamically typed collection of components
///
/// See [Bundle]
pub trait DynamicBundle { pub trait DynamicBundle {
/// Invoke a callback on the fields' type IDs, sorted by descending alignment then id /// Invoke a callback on the fields' type IDs, sorted by descending alignment then id
#[doc(hidden)] #[doc(hidden)]
@ -38,6 +40,8 @@ pub trait DynamicBundle {
} }
/// A statically typed collection of components /// A statically typed collection of components
///
/// See [DynamicBundle]
pub trait Bundle: DynamicBundle { pub trait Bundle: DynamicBundle {
#[doc(hidden)] #[doc(hidden)]
fn with_static_ids<T>(f: impl FnOnce(&[TypeId]) -> T) -> T; fn with_static_ids<T>(f: impl FnOnce(&[TypeId]) -> T) -> T;

View File

@ -76,16 +76,16 @@ impl World {
/// let a = world.spawn((123, "abc")); /// let a = world.spawn((123, "abc"));
/// let b = world.spawn((456, true)); /// let b = world.spawn((456, true));
/// ``` /// ```
pub fn spawn(&mut self, components: impl DynamicBundle) -> Entity { pub fn spawn(&mut self, bundle: impl DynamicBundle) -> Entity {
// Ensure all entity allocations are accounted for so `self.entities` can realloc if // Ensure all entity allocations are accounted for so `self.entities` can realloc if
// necessary // necessary
self.flush(); self.flush();
let entity = self.entities.alloc(); let entity = self.entities.alloc();
let archetype_id = components.with_ids(|ids| { let archetype_id = bundle.with_ids(|ids| {
self.index.get(ids).copied().unwrap_or_else(|| { self.index.get(ids).copied().unwrap_or_else(|| {
let x = self.archetypes.len() as u32; let x = self.archetypes.len() as u32;
self.archetypes.push(Archetype::new(components.type_info())); self.archetypes.push(Archetype::new(bundle.type_info()));
self.index.insert(ids.to_vec(), x); self.index.insert(ids.to_vec(), x);
self.archetype_generation += 1; self.archetype_generation += 1;
x x
@ -95,7 +95,7 @@ impl World {
let archetype = &mut self.archetypes[archetype_id as usize]; let archetype = &mut self.archetypes[archetype_id as usize];
unsafe { unsafe {
let index = archetype.allocate(entity); let index = archetype.allocate(entity);
components.put(|ptr, ty, size| { bundle.put(|ptr, ty, size| {
archetype.put_dynamic(ptr, ty, size, index, ComponentFlags::ADDED); archetype.put_dynamic(ptr, ty, size, index, ComponentFlags::ADDED);
true true
}); });
@ -566,7 +566,7 @@ impl World {
pub fn insert( pub fn insert(
&mut self, &mut self,
entity: Entity, entity: Entity,
components: impl DynamicBundle, bundle: impl DynamicBundle,
) -> Result<(), NoSuchEntity> { ) -> Result<(), NoSuchEntity> {
use std::collections::hash_map::Entry; use std::collections::hash_map::Entry;
@ -576,7 +576,7 @@ impl World {
// Assemble Vec<TypeInfo> for the final entity // Assemble Vec<TypeInfo> for the final entity
let arch = &mut self.archetypes[loc.archetype as usize]; let arch = &mut self.archetypes[loc.archetype as usize];
let mut info = arch.types().to_vec(); let mut info = arch.types().to_vec();
for ty in components.type_info() { for ty in bundle.type_info() {
if let Some(ptr) = arch.get_dynamic(ty.id(), ty.layout().size(), loc.index) { if let Some(ptr) = arch.get_dynamic(ty.id(), ty.layout().size(), loc.index) {
ty.drop(ptr.as_ptr()); ty.drop(ptr.as_ptr());
} else { } else {
@ -601,7 +601,7 @@ impl World {
if target == loc.archetype { if target == loc.archetype {
// Update components in the current archetype // Update components in the current archetype
let arch = &mut self.archetypes[loc.archetype as usize]; let arch = &mut self.archetypes[loc.archetype as usize];
components.put(|ptr, ty, size| { bundle.put(|ptr, ty, size| {
arch.put_dynamic(ptr, ty, size, loc.index, ComponentFlags::MUTATED); arch.put_dynamic(ptr, ty, size, loc.index, ComponentFlags::MUTATED);
true true
}); });
@ -625,7 +625,7 @@ impl World {
self.entities.get_mut(moved).unwrap().index = old_index; self.entities.get_mut(moved).unwrap().index = old_index;
} }
components.put(|ptr, ty, size| { bundle.put(|ptr, ty, size| {
let had_component = source_arch.has_dynamic(ty); let had_component = source_arch.has_dynamic(ty);
let flags = if had_component { let flags = if had_component {
ComponentFlags::MUTATED ComponentFlags::MUTATED

View File

@ -16,7 +16,7 @@ pub(crate) struct Spawn<T>
where where
T: DynamicBundle + Send + Sync + 'static, T: DynamicBundle + Send + Sync + 'static,
{ {
components: T, bundle: T,
} }
impl<T> Command for Spawn<T> impl<T> Command for Spawn<T>
@ -24,7 +24,7 @@ where
T: DynamicBundle + Send + Sync + 'static, T: DynamicBundle + Send + Sync + 'static,
{ {
fn write(self: Box<Self>, world: &mut World, _resources: &mut Resources) { fn write(self: Box<Self>, world: &mut World, _resources: &mut Resources) {
world.spawn(self.components); world.spawn(self.bundle);
} }
} }
@ -33,7 +33,7 @@ where
I: IntoIterator, I: IntoIterator,
I::Item: Bundle, I::Item: Bundle,
{ {
components_iter: I, bundles_iter: I,
} }
impl<I> Command for SpawnBatch<I> impl<I> Command for SpawnBatch<I>
@ -42,7 +42,7 @@ where
I::Item: Bundle, I::Item: Bundle,
{ {
fn write(self: Box<Self>, world: &mut World, _resources: &mut Resources) { fn write(self: Box<Self>, world: &mut World, _resources: &mut Resources) {
world.spawn_batch(self.components_iter); world.spawn_batch(self.bundles_iter);
} }
} }
@ -64,7 +64,7 @@ where
T: DynamicBundle + Send + Sync + 'static, T: DynamicBundle + Send + Sync + 'static,
{ {
entity: Entity, entity: Entity,
components: T, bundle: T,
} }
impl<T> Command for Insert<T> impl<T> Command for Insert<T>
@ -72,7 +72,7 @@ where
T: DynamicBundle + Send + Sync + 'static, T: DynamicBundle + Send + Sync + 'static,
{ {
fn write(self: Box<Self>, world: &mut World, _resources: &mut Resources) { fn write(self: Box<Self>, world: &mut World, _resources: &mut Resources) {
world.insert(self.entity, self.components).unwrap(); world.insert(self.entity, self.bundle).unwrap();
} }
} }
@ -181,6 +181,7 @@ impl<T: Resource> Command for InsertLocalResource<T> {
} }
} }
/// A list of commands that will be run to populate a `World` and `Resources`.
#[derive(Default)] #[derive(Default)]
pub struct Commands { pub struct Commands {
commands: Vec<Box<dyn Command>>, commands: Vec<Box<dyn Command>>,
@ -189,38 +190,78 @@ pub struct Commands {
} }
impl Commands { impl Commands {
pub fn spawn(&mut self, components: impl DynamicBundle + Send + Sync + 'static) -> &mut Self { /// Creates a new entity with the components contained in `bundle`.
///
/// Note that `bundle` is a [DynamicBundle], which is a collection of components. [DynamicBundle] is automatically implemented for tuples of components. You can also create your own bundle types by deriving [`derive@Bundle`]. If you would like to spawn an entity with a single component, consider wrapping the component in a tuple (which [DynamicBundle] is implemented for).
///
/// See [`Self::set_current_entity`], [`Self::insert`].
///
/// # Example
///
/// ```
/// use bevy_ecs::prelude::*;
///
/// struct Component1;
/// struct Component2;
///
/// #[derive(Bundle)]
/// struct ExampleBundle {
/// a: Component1,
/// b: Component2,
/// }
///
/// fn example_system(mut commands: Commands) {
/// // Create a new entity with a component bundle.
/// commands.spawn(ExampleBundle {
/// a: Component1,
/// b: Component2,
/// });
///
/// // Create a new entity with a single component.
/// commands.spawn((Component1,));
/// // Create a new entity with two components.
/// commands.spawn((Component1, Component2));
/// }
/// ```
pub fn spawn(&mut self, bundle: impl DynamicBundle + Send + Sync + 'static) -> &mut Self {
let entity = self let entity = self
.entity_reserver .entity_reserver
.as_ref() .as_ref()
.expect("Entity reserver has not been set.") .expect("Entity reserver has not been set.")
.reserve_entity(); .reserve_entity();
self.current_entity = Some(entity); self.set_current_entity(entity);
self.commands.push(Box::new(Insert { entity, components })); self.insert(entity, bundle);
self self
} }
pub fn spawn_batch<I>(&mut self, components_iter: I) -> &mut Self /// Equivalent to iterating `bundles_iter` and calling [`Self::spawn`] on each bundle, but slightly more performant.
pub fn spawn_batch<I>(&mut self, bundles_iter: I) -> &mut Self
where where
I: IntoIterator + Send + Sync + 'static, I: IntoIterator + Send + Sync + 'static,
I::Item: Bundle, I::Item: Bundle,
{ {
self.add_command(SpawnBatch { components_iter }) self.add_command(SpawnBatch { bundles_iter })
} }
/// Despawns only the specified entity, ignoring any other consideration. /// Despawns only the specified entity, not including its children.
pub fn despawn(&mut self, entity: Entity) -> &mut Self { pub fn despawn(&mut self, entity: Entity) -> &mut Self {
self.add_command(Despawn { entity }) self.add_command(Despawn { entity })
} }
/// Inserts a bundle of components into `entity`.
///
/// See [`World::insert`].
pub fn insert( pub fn insert(
&mut self, &mut self,
entity: Entity, entity: Entity,
components: impl DynamicBundle + Send + Sync + 'static, bundle: impl DynamicBundle + Send + Sync + 'static,
) -> &mut Self { ) -> &mut Self {
self.add_command(Insert { entity, components }) self.add_command(Insert { entity, bundle })
} }
/// Inserts a single component into `entity`.
///
/// See [`World::insert_one`].
pub fn insert_one(&mut self, entity: Entity, component: impl Component) -> &mut Self { pub fn insert_one(&mut self, entity: Entity, component: impl Component) -> &mut Self {
self.add_command(InsertOne { entity, component }) self.add_command(InsertOne { entity, component })
} }
@ -229,6 +270,9 @@ impl Commands {
self.add_command(InsertResource { resource }) self.add_command(InsertResource { resource })
} }
/// Insert a resource that is local to a specific system.
///
/// See [`crate::System::id`].
pub fn insert_local_resource<T: Resource>( pub fn insert_local_resource<T: Resource>(
&mut self, &mut self,
system_id: SystemId, system_id: SystemId,
@ -240,6 +284,7 @@ impl Commands {
}) })
} }
/// See [`World::remove_one`].
pub fn remove_one<T>(&mut self, entity: Entity) -> &mut Self pub fn remove_one<T>(&mut self, entity: Entity) -> &mut Self
where where
T: Component, T: Component,
@ -250,6 +295,7 @@ impl Commands {
}) })
} }
/// See [`World::remove`].
pub fn remove<T>(&mut self, entity: Entity) -> &mut Self pub fn remove<T>(&mut self, entity: Entity) -> &mut Self
where where
T: Bundle + Send + Sync + 'static, T: Bundle + Send + Sync + 'static,
@ -260,18 +306,54 @@ impl Commands {
}) })
} }
pub fn with_bundle( /// Adds a bundle of components to the current entity.
&mut self, ///
components: impl DynamicBundle + Send + Sync + 'static, /// See [`Self::with`], [`Self::current_entity`].
) -> &mut Self { pub fn with_bundle(&mut self, bundle: impl DynamicBundle + Send + Sync + 'static) -> &mut Self {
let current_entity = self.current_entity.expect("Cannot add components because the 'current entity' is not set. You should spawn an entity first."); let current_entity = self.current_entity.expect("Cannot add bundle because the 'current entity' is not set. You should spawn an entity first.");
self.commands.push(Box::new(Insert { self.commands.push(Box::new(Insert {
entity: current_entity, entity: current_entity,
components, bundle,
})); }));
self self
} }
/// Adds a single component to the current entity.
///
/// See [`Self::with_bundle`], [`Self::current_entity`].
///
/// # Warning
///
/// It's possible to call this with a bundle, but this is likely not intended and [`Self::with_bundle`] should be used instead. If `with` is called with a bundle, the bundle itself will be added as a component instead of the bundles' inner components each being added.
///
/// # Example
///
/// `with` can be chained with [`Self::spawn`].
///
/// ```
/// use bevy_ecs::prelude::*;
///
/// struct Component1;
/// struct Component2;
///
/// fn example_system(mut commands: Commands) {
/// // Create a new entity with a `Component1` and `Component2`.
/// commands.spawn((Component1,)).with(Component2);
///
/// // Psst! These are also equivalent to the line above!
/// commands.spawn((Component1, Component2));
/// commands.spawn(()).with(Component1).with(Component2);
/// #[derive(Bundle)]
/// struct ExampleBundle {
/// a: Component1,
/// b: Component2,
/// }
/// commands.spawn(()).with_bundle(ExampleBundle {
/// a: Component1,
/// b: Component2,
/// });
/// }
/// ```
pub fn with(&mut self, component: impl Component) -> &mut Self { pub fn with(&mut self, component: impl Component) -> &mut Self {
let current_entity = self.current_entity.expect("Cannot add component because the 'current entity' is not set. You should spawn an entity first."); let current_entity = self.current_entity.expect("Cannot add component because the 'current entity' is not set. You should spawn an entity first.");
self.commands.push(Box::new(InsertOne { self.commands.push(Box::new(InsertOne {
@ -281,22 +363,26 @@ impl Commands {
self self
} }
/// Adds a command directly to the command list. Prefer this to [`Self::add_command_boxed`] if the type of `command` is statically known.
pub fn add_command<C: Command + 'static>(&mut self, command: C) -> &mut Self { pub fn add_command<C: Command + 'static>(&mut self, command: C) -> &mut Self {
self.commands.push(Box::new(command)); self.commands.push(Box::new(command));
self self
} }
/// See [`Self::add_command`].
pub fn add_command_boxed(&mut self, command: Box<dyn Command>) -> &mut Self { pub fn add_command_boxed(&mut self, command: Box<dyn Command>) -> &mut Self {
self.commands.push(command); self.commands.push(command);
self self
} }
/// Runs all the stored commands on `world` and `resources`. The command buffer is emptied as a part of this call.
pub fn apply(&mut self, world: &mut World, resources: &mut Resources) { pub fn apply(&mut self, world: &mut World, resources: &mut Resources) {
for command in self.commands.drain(..) { for command in self.commands.drain(..) {
command.write(world, resources); command.write(world, resources);
} }
} }
/// Returns the current entity, set by [`Self::spawn`] or with [`Self::set_current_entity`].
pub fn current_entity(&self) -> Option<Entity> { pub fn current_entity(&self) -> Option<Entity> {
self.current_entity self.current_entity
} }