forestia/bevy
2
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Make builder types take and return Self (#10001)

Browse Source 
# Objective

Closes #9955.

Use the same interface for all "pure" builder types: taking and
returning `Self` (and not `&mut Self`).

## Solution

Changed `DynamicSceneBuilder`, `SceneFilter` and `TableBuilder` to take
and return `Self`.

## Changelog

### Changed

- `DynamicSceneBuilder` and `SceneBuilder` methods in `bevy_ecs` now
take and return `Self`.

## Migration guide

When using `bevy_ecs::DynamicSceneBuilder` and `bevy_ecs::SceneBuilder`,
instead of binding the builder to a variable, directly use it. Methods
on those types now consume `Self`, so you will need to re-bind the
builder if you don't `build` it immediately.

Before:
```rust
let mut scene_builder = DynamicSceneBuilder::from_world(&world);
let scene = scene_builder.extract_entity(a).extract_entity(b).build();
```

After:
 ```rust
let scene = DynamicSceneBuilder::from_world(&world)
    .extract_entity(a)
    .extract_entity(b)
    .build();
```
This commit is contained in:
Kanabenki 2023-10-09 21:46:17 +02:00 committed by GitHub
parent 39c68e3f92
commit 569e2ac80f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 124 additions and 114 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
crates/bevy_ecs/src/storage/table.rs
View File

@ -526,13 +526,16 @@ impl TableBuilder {
} }
} }
pub fn add_column(&mut self, component_info: &ComponentInfo) { #[must_use]
pub fn add_column(mut self, component_info: &ComponentInfo) -> Self {
self.columns.insert( self.columns.insert(
component_info.id(), component_info.id(),
Column::with_capacity(component_info, self.capacity), Column::with_capacity(component_info, self.capacity),
); );
self
} }
#[must_use]
pub fn build(self) -> Table { pub fn build(self) -> Table {
Table { Table {
columns: self.columns.into_immutable(), columns: self.columns.into_immutable(),
@ -865,7 +868,7 @@ impl Tables {
.or_insert_with(|| { .or_insert_with(|| {
let mut table = TableBuilder::with_capacity(0, component_ids.len()); let mut table = TableBuilder::with_capacity(0, component_ids.len());
for component_id in component_ids { for component_id in component_ids {
table.add_column(components.get_info_unchecked(*component_id)); table = table.add_column(components.get_info_unchecked(*component_id));
} }
tables.push(table.build()); tables.push(table.build());
(component_ids.to_vec(), TableId::new(tables.len() - 1)) (component_ids.to_vec(), TableId::new(tables.len() - 1))
@ -929,9 +932,9 @@ mod tests {
let mut storages = Storages::default(); let mut storages = Storages::default();
let component_id = components.init_component::<W<TableRow>>(&mut storages); let component_id = components.init_component::<W<TableRow>>(&mut storages);
let columns = &[component_id]; let columns = &[component_id];
let mut builder = TableBuilder::with_capacity(0, columns.len()); let mut table = TableBuilder::with_capacity(0, columns.len())
builder.add_column(components.get_info(component_id).unwrap()); .add_column(components.get_info(component_id).unwrap())
let mut table = builder.build(); .build();
let entities = (0..200).map(Entity::from_raw).collect::<Vec<_>>(); let entities = (0..200).map(Entity::from_raw).collect::<Vec<_>>();
for entity in &entities { for entity in &entities {
// SAFETY: we allocate and immediately set data afterwards // SAFETY: we allocate and immediately set data afterwards

18
crates/bevy_scene/src/dynamic_scene.rs
View File

@ -51,12 +51,10 @@ impl DynamicScene {
/// Create a new dynamic scene from a given world. /// Create a new dynamic scene from a given world.
pub fn from_world(world: &World) -> Self { pub fn from_world(world: &World) -> Self {
let mut builder = DynamicSceneBuilder::from_world(world); DynamicSceneBuilder::from_world(world)
.extract_entities(world.iter_entities().map(|entity| entity.id()))
builder.extract_entities(world.iter_entities().map(|entity| entity.id())); .extract_resources()
builder.extract_resources(); .build()
builder.build()
} }
/// Write the resources, the dynamic entities, and their corresponding components to the given world. /// Write the resources, the dynamic entities, and their corresponding components to the given world.
@ -220,10 +218,10 @@ mod tests {
// We then write this relationship to a new scene, and then write that scene back to the // We then write this relationship to a new scene, and then write that scene back to the
// world to create another parent and child relationship // world to create another parent and child relationship
let mut scene_builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
scene_builder.extract_entity(original_parent_entity); .extract_entity(original_parent_entity)
scene_builder.extract_entity(original_child_entity); .extract_entity(original_child_entity)
let scene = scene_builder.build(); .build();
let mut entity_map = HashMap::default(); let mut entity_map = HashMap::default();
scene.write_to_world(&mut world, &mut entity_map).unwrap(); scene.write_to_world(&mut world, &mut entity_map).unwrap();

149
crates/bevy_scene/src/dynamic_scene_builder.rs
View File

@ -50,9 +50,7 @@ use std::collections::BTreeMap;
/// # let mut world = World::default(); /// # let mut world = World::default();
/// # world.init_resource::<AppTypeRegistry>(); /// # world.init_resource::<AppTypeRegistry>();
/// # let entity = world.spawn(ComponentA).id(); /// # let entity = world.spawn(ComponentA).id();
/// let mut builder = DynamicSceneBuilder::from_world(&world); /// let dynamic_scene = DynamicSceneBuilder::from_world(&world).extract_entity(entity).build();
/// builder.extract_entity(entity);
/// let dynamic_scene = builder.build();
/// ``` /// ```
pub struct DynamicSceneBuilder<'w> { pub struct DynamicSceneBuilder<'w> {
extracted_resources: BTreeMap<ComponentId, Box<dyn Reflect>>, extracted_resources: BTreeMap<ComponentId, Box<dyn Reflect>>,
@ -75,13 +73,15 @@ impl<'w> DynamicSceneBuilder<'w> {
} }
/// Specify a custom component [`SceneFilter`] to be used with this builder. /// Specify a custom component [`SceneFilter`] to be used with this builder.
pub fn with_filter(&mut self, filter: SceneFilter) -> &mut Self { #[must_use]
pub fn with_filter(mut self, filter: SceneFilter) -> Self {
self.component_filter = filter; self.component_filter = filter;
self self
} }
/// Specify a custom resource [`SceneFilter`] to be used with this builder. /// Specify a custom resource [`SceneFilter`] to be used with this builder.
pub fn with_resource_filter(&mut self, filter: SceneFilter) -> &mut Self { #[must_use]
pub fn with_resource_filter(mut self, filter: SceneFilter) -> Self {
self.resource_filter = filter; self.resource_filter = filter;
self self
} }
@ -92,8 +92,9 @@ impl<'w> DynamicSceneBuilder<'w> {
/// ///
/// This is the inverse of [`deny`](Self::deny). /// This is the inverse of [`deny`](Self::deny).
/// If `T` has already been denied, then it will be removed from the denylist. /// If `T` has already been denied, then it will be removed from the denylist.
pub fn allow<T: Component>(&mut self) -> &mut Self { #[must_use]
self.component_filter.allow::<T>(); pub fn allow<T: Component>(mut self) -> Self {
self.component_filter = self.component_filter.allow::<T>();
self self
} }
@ -103,8 +104,9 @@ impl<'w> DynamicSceneBuilder<'w> {
/// ///
/// This is the inverse of [`allow`](Self::allow). /// This is the inverse of [`allow`](Self::allow).
/// If `T` has already been allowed, then it will be removed from the allowlist. /// If `T` has already been allowed, then it will be removed from the allowlist.
pub fn deny<T: Component>(&mut self) -> &mut Self { #[must_use]
self.component_filter.deny::<T>(); pub fn deny<T: Component>(mut self) -> Self {
self.component_filter = self.component_filter.deny::<T>();
self self
} }
@ -113,7 +115,8 @@ impl<'w> DynamicSceneBuilder<'w> {
/// This is useful for resetting the filter so that types may be selectively [denied]. /// This is useful for resetting the filter so that types may be selectively [denied].
/// ///
/// [denied]: Self::deny /// [denied]: Self::deny
pub fn allow_all(&mut self) -> &mut Self { #[must_use]
pub fn allow_all(mut self) -> Self {
self.component_filter = SceneFilter::allow_all(); self.component_filter = SceneFilter::allow_all();
self self
} }
@ -123,7 +126,8 @@ impl<'w> DynamicSceneBuilder<'w> {
/// This is useful for resetting the filter so that types may be selectively [allowed]. /// This is useful for resetting the filter so that types may be selectively [allowed].
/// ///
/// [allowed]: Self::allow /// [allowed]: Self::allow
pub fn deny_all(&mut self) -> &mut Self { #[must_use]
pub fn deny_all(mut self) -> Self {
self.component_filter = SceneFilter::deny_all(); self.component_filter = SceneFilter::deny_all();
self self
} }
@ -134,8 +138,9 @@ impl<'w> DynamicSceneBuilder<'w> {
/// ///
/// This is the inverse of [`deny_resource`](Self::deny_resource). /// This is the inverse of [`deny_resource`](Self::deny_resource).
/// If `T` has already been denied, then it will be removed from the denylist. /// If `T` has already been denied, then it will be removed from the denylist.
pub fn allow_resource<T: Resource>(&mut self) -> &mut Self { #[must_use]
self.resource_filter.allow::<T>(); pub fn allow_resource<T: Resource>(mut self) -> Self {
self.resource_filter = self.resource_filter.allow::<T>();
self self
} }
@ -145,8 +150,9 @@ impl<'w> DynamicSceneBuilder<'w> {
/// ///
/// This is the inverse of [`allow_resource`](Self::allow_resource). /// This is the inverse of [`allow_resource`](Self::allow_resource).
/// If `T` has already been allowed, then it will be removed from the allowlist. /// If `T` has already been allowed, then it will be removed from the allowlist.
pub fn deny_resource<T: Resource>(&mut self) -> &mut Self { #[must_use]
self.resource_filter.deny::<T>(); pub fn deny_resource<T: Resource>(mut self) -> Self {
self.resource_filter = self.resource_filter.deny::<T>();
self self
} }
@ -155,7 +161,8 @@ impl<'w> DynamicSceneBuilder<'w> {
/// This is useful for resetting the filter so that types may be selectively [denied]. /// This is useful for resetting the filter so that types may be selectively [denied].
/// ///
/// [denied]: Self::deny_resource /// [denied]: Self::deny_resource
pub fn allow_all_resources(&mut self) -> &mut Self { #[must_use]
pub fn allow_all_resources(mut self) -> Self {
self.resource_filter = SceneFilter::allow_all(); self.resource_filter = SceneFilter::allow_all();
self self
} }
@ -165,7 +172,8 @@ impl<'w> DynamicSceneBuilder<'w> {
/// This is useful for resetting the filter so that types may be selectively [allowed]. /// This is useful for resetting the filter so that types may be selectively [allowed].
/// ///
/// [allowed]: Self::allow_resource /// [allowed]: Self::allow_resource
pub fn deny_all_resources(&mut self) -> &mut Self { #[must_use]
pub fn deny_all_resources(mut self) -> Self {
self.resource_filter = SceneFilter::deny_all(); self.resource_filter = SceneFilter::deny_all();
self self
} }
@ -174,6 +182,7 @@ impl<'w> DynamicSceneBuilder<'w> {
/// ///
/// To make sure the dynamic scene doesn't contain entities without any components, call /// To make sure the dynamic scene doesn't contain entities without any components, call
/// [`Self::remove_empty_entities`] before building the scene. /// [`Self::remove_empty_entities`] before building the scene.
#[must_use]
pub fn build(self) -> DynamicScene { pub fn build(self) -> DynamicScene {
DynamicScene { DynamicScene {
resources: self.extracted_resources.into_values().collect(), resources: self.extracted_resources.into_values().collect(),
@ -184,14 +193,16 @@ impl<'w> DynamicSceneBuilder<'w> {
/// Extract one entity from the builder's [`World`]. /// Extract one entity from the builder's [`World`].
/// ///
/// Re-extracting an entity that was already extracted will have no effect. /// Re-extracting an entity that was already extracted will have no effect.
pub fn extract_entity(&mut self, entity: Entity) -> &mut Self { #[must_use]
pub fn extract_entity(self, entity: Entity) -> Self {
self.extract_entities(std::iter::once(entity)) self.extract_entities(std::iter::once(entity))
} }
/// Despawns all entities with no components. /// Despawns all entities with no components.
/// ///
/// These were likely created because none of their components were present in the provided type registry upon extraction. /// These were likely created because none of their components were present in the provided type registry upon extraction.
pub fn remove_empty_entities(&mut self) -> &mut Self { #[must_use]
pub fn remove_empty_entities(mut self) -> Self {
self.extracted_scene self.extracted_scene
.retain(|_, entity| !entity.components.is_empty()); .retain(|_, entity| !entity.components.is_empty());
@ -222,16 +233,17 @@ impl<'w> DynamicSceneBuilder<'w> {
/// # let _entity = world.spawn(MyComponent).id(); /// # let _entity = world.spawn(MyComponent).id();
/// let mut query = world.query_filtered::<Entity, With<MyComponent>>(); /// let mut query = world.query_filtered::<Entity, With<MyComponent>>();
/// ///
/// let mut builder = DynamicSceneBuilder::from_world(&world); /// let scene = DynamicSceneBuilder::from_world(&world)
/// builder.extract_entities(query.iter(&world)); /// .extract_entities(query.iter(&world))
/// let scene = builder.build(); /// .build();
/// ``` /// ```
/// ///
/// Note that components extracted from queried entities must still pass through the filter if one is set. /// Note that components extracted from queried entities must still pass through the filter if one is set.
/// ///
/// [`allow`]: Self::allow /// [`allow`]: Self::allow
/// [`deny`]: Self::deny /// [`deny`]: Self::deny
pub fn extract_entities(&mut self, entities: impl Iterator<Item = Entity>) -> &mut Self { #[must_use]
pub fn extract_entities(mut self, entities: impl Iterator<Item = Entity>) -> Self {
let type_registry = self.original_world.resource::<AppTypeRegistry>().read(); let type_registry = self.original_world.resource::<AppTypeRegistry>().read();
for entity in entities { for entity in entities {
@ -295,14 +307,14 @@ impl<'w> DynamicSceneBuilder<'w> {
/// # world.init_resource::<AppTypeRegistry>(); /// # world.init_resource::<AppTypeRegistry>();
/// world.insert_resource(MyResource); /// world.insert_resource(MyResource);
/// ///
/// let mut builder = DynamicSceneBuilder::from_world(&world); /// let mut builder = DynamicSceneBuilder::from_world(&world).extract_resources();
/// builder.extract_resources();
/// let scene = builder.build(); /// let scene = builder.build();
/// ``` /// ```
/// ///
/// [`allow_resource`]: Self::allow_resource /// [`allow_resource`]: Self::allow_resource
/// [`deny_resource`]: Self::deny_resource /// [`deny_resource`]: Self::deny_resource
pub fn extract_resources(&mut self) -> &mut Self { #[must_use]
pub fn extract_resources(mut self) -> Self {
let type_registry = self.original_world.resource::<AppTypeRegistry>().read(); let type_registry = self.original_world.resource::<AppTypeRegistry>().read();
for (component_id, _) in self.original_world.storages().resources.iter() { for (component_id, _) in self.original_world.storages().resources.iter() {
@ -374,9 +386,9 @@ mod tests {
let entity = world.spawn((ComponentA, ComponentB)).id(); let entity = world.spawn((ComponentA, ComponentB)).id();
let mut builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
builder.extract_entity(entity); .extract_entity(entity)
let scene = builder.build(); .build();
assert_eq!(scene.entities.len(), 1); assert_eq!(scene.entities.len(), 1);
assert_eq!(scene.entities[0].entity, entity); assert_eq!(scene.entities[0].entity, entity);
@ -394,10 +406,10 @@ mod tests {
let entity = world.spawn((ComponentA, ComponentB)).id(); let entity = world.spawn((ComponentA, ComponentB)).id();
let mut builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
builder.extract_entity(entity); .extract_entity(entity)
builder.extract_entity(entity); .extract_entity(entity)
let scene = builder.build(); .build();
assert_eq!(scene.entities.len(), 1); assert_eq!(scene.entities.len(), 1);
assert_eq!(scene.entities[0].entity, entity); assert_eq!(scene.entities[0].entity, entity);
@ -419,9 +431,9 @@ mod tests {
let entity = world.spawn((ComponentA, ComponentB)).id(); let entity = world.spawn((ComponentA, ComponentB)).id();
let mut builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
builder.extract_entity(entity); .extract_entity(entity)
let scene = builder.build(); .build();
assert_eq!(scene.entities.len(), 1); assert_eq!(scene.entities.len(), 1);
assert_eq!(scene.entities[0].entity, entity); assert_eq!(scene.entities[0].entity, entity);
@ -441,12 +453,11 @@ mod tests {
let entity_c = world.spawn_empty().id(); let entity_c = world.spawn_empty().id();
let entity_d = world.spawn_empty().id(); let entity_d = world.spawn_empty().id();
let mut builder = DynamicSceneBuilder::from_world(&world);
// Insert entities out of order // Insert entities out of order
builder.extract_entity(entity_b); let builder = DynamicSceneBuilder::from_world(&world)
builder.extract_entities([entity_d, entity_a].into_iter()); .extract_entity(entity_b)
builder.extract_entity(entity_c); .extract_entities([entity_d, entity_a].into_iter())
.extract_entity(entity_c);
let mut entities = builder.build().entities.into_iter(); let mut entities = builder.build().entities.into_iter();
@ -474,9 +485,9 @@ mod tests {
let _entity_b = world.spawn(ComponentB).id(); let _entity_b = world.spawn(ComponentB).id();
let mut query = world.query_filtered::<Entity, With<ComponentA>>(); let mut query = world.query_filtered::<Entity, With<ComponentA>>();
let mut builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
builder.extract_entities(query.iter(&world)); .extract_entities(query.iter(&world))
let scene = builder.build(); .build();
assert_eq!(scene.entities.len(), 2); assert_eq!(scene.entities.len(), 2);
let mut scene_entities = vec![scene.entities[0].entity, scene.entities[1].entity]; let mut scene_entities = vec![scene.entities[0].entity, scene.entities[1].entity];
@ -495,10 +506,10 @@ mod tests {
let entity_a = world.spawn(ComponentA).id(); let entity_a = world.spawn(ComponentA).id();
let entity_b = world.spawn(ComponentB).id(); let entity_b = world.spawn(ComponentB).id();
let mut builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
builder.extract_entities([entity_a, entity_b].into_iter()); .extract_entities([entity_a, entity_b].into_iter())
builder.remove_empty_entities(); .remove_empty_entities()
let scene = builder.build(); .build();
assert_eq!(scene.entities.len(), 1); assert_eq!(scene.entities.len(), 1);
assert_eq!(scene.entities[0].entity, entity_a); assert_eq!(scene.entities[0].entity, entity_a);
@ -514,9 +525,9 @@ mod tests {
world.insert_resource(ResourceA); world.insert_resource(ResourceA);
let mut builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
builder.extract_resources(); .extract_resources()
let scene = builder.build(); .build();
assert_eq!(scene.resources.len(), 1); assert_eq!(scene.resources.len(), 1);
assert!(scene.resources[0].represents::<ResourceA>()); assert!(scene.resources[0].represents::<ResourceA>());
@ -532,10 +543,10 @@ mod tests {
world.insert_resource(ResourceA); world.insert_resource(ResourceA);
let mut builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
builder.extract_resources(); .extract_resources()
builder.extract_resources(); .extract_resources()
let scene = builder.build(); .build();
assert_eq!(scene.resources.len(), 1); assert_eq!(scene.resources.len(), 1);
assert!(scene.resources[0].represents::<ResourceA>()); assert!(scene.resources[0].represents::<ResourceA>());
@ -557,11 +568,10 @@ mod tests {
let entity_a = world.spawn(ComponentA).id(); let entity_a = world.spawn(ComponentA).id();
let entity_b = world.spawn(ComponentB).id(); let entity_b = world.spawn(ComponentB).id();
let mut builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
builder
.allow::<ComponentA>() .allow::<ComponentA>()
.extract_entities([entity_a_b, entity_a, entity_b].into_iter()); .extract_entities([entity_a_b, entity_a, entity_b].into_iter())
let scene = builder.build(); .build();
assert_eq!(scene.entities.len(), 3); assert_eq!(scene.entities.len(), 3);
assert!(scene.entities[0].components[0].represents::<ComponentA>()); assert!(scene.entities[0].components[0].represents::<ComponentA>());
@ -585,11 +595,10 @@ mod tests {
let entity_a = world.spawn(ComponentA).id(); let entity_a = world.spawn(ComponentA).id();
let entity_b = world.spawn(ComponentB).id(); let entity_b = world.spawn(ComponentB).id();
let mut builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
builder
.deny::<ComponentA>() .deny::<ComponentA>()
.extract_entities([entity_a_b, entity_a, entity_b].into_iter()); .extract_entities([entity_a_b, entity_a, entity_b].into_iter())
let scene = builder.build(); .build();
assert_eq!(scene.entities.len(), 3); assert_eq!(scene.entities.len(), 3);
assert!(scene.entities[0].components[0].represents::<ComponentB>()); assert!(scene.entities[0].components[0].represents::<ComponentB>());
@ -612,9 +621,10 @@ mod tests {
world.insert_resource(ResourceA); world.insert_resource(ResourceA);
world.insert_resource(ResourceB); world.insert_resource(ResourceB);
let mut builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
builder.allow_resource::<ResourceA>().extract_resources(); .allow_resource::<ResourceA>()
let scene = builder.build(); .extract_resources()
.build();
assert_eq!(scene.resources.len(), 1); assert_eq!(scene.resources.len(), 1);
assert!(scene.resources[0].represents::<ResourceA>()); assert!(scene.resources[0].represents::<ResourceA>());
@ -635,9 +645,10 @@ mod tests {
world.insert_resource(ResourceA); world.insert_resource(ResourceA);
world.insert_resource(ResourceB); world.insert_resource(ResourceB);
let mut builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
builder.deny_resource::<ResourceA>().extract_resources(); .deny_resource::<ResourceA>()
let scene = builder.build(); .extract_resources()
.build();
assert_eq!(scene.resources.len(), 1); assert_eq!(scene.resources.len(), 1);
assert!(scene.resources[0].represents::<ResourceB>()); assert!(scene.resources[0].represents::<ResourceB>());

50
crates/bevy_scene/src/scene_filter.rs
View File

@ -70,7 +70,8 @@ impl SceneFilter {
/// [`Denylist`]: SceneFilter::Denylist /// [`Denylist`]: SceneFilter::Denylist
/// [`Unset`]: SceneFilter::Unset /// [`Unset`]: SceneFilter::Unset
/// [`Allowlist`]: SceneFilter::Allowlist /// [`Allowlist`]: SceneFilter::Allowlist
pub fn allow<T: Any>(&mut self) -> &mut Self { #[must_use]
pub fn allow<T: Any>(self) -> Self {
self.allow_by_id(TypeId::of::<T>()) self.allow_by_id(TypeId::of::<T>())
} }
@ -84,10 +85,11 @@ impl SceneFilter {
/// [`Denylist`]: SceneFilter::Denylist /// [`Denylist`]: SceneFilter::Denylist
/// [`Unset`]: SceneFilter::Unset /// [`Unset`]: SceneFilter::Unset
/// [`Allowlist`]: SceneFilter::Allowlist /// [`Allowlist`]: SceneFilter::Allowlist
pub fn allow_by_id(&mut self, type_id: TypeId) -> &mut Self { #[must_use]
match self { pub fn allow_by_id(mut self, type_id: TypeId) -> Self {
match &mut self {
Self::Unset => { Self::Unset => {
*self = Self::Allowlist(HashSet::from([type_id])); self = Self::Allowlist(HashSet::from([type_id]));
} }
Self::Allowlist(list) => { Self::Allowlist(list) => {
list.insert(type_id); list.insert(type_id);
@ -109,7 +111,8 @@ impl SceneFilter {
/// [`Allowlist`]: SceneFilter::Allowlist /// [`Allowlist`]: SceneFilter::Allowlist
/// [`Unset`]: SceneFilter::Unset /// [`Unset`]: SceneFilter::Unset
/// [`Denylist`]: SceneFilter::Denylist /// [`Denylist`]: SceneFilter::Denylist
pub fn deny<T: Any>(&mut self) -> &mut Self { #[must_use]
pub fn deny<T: Any>(self) -> Self {
self.deny_by_id(TypeId::of::<T>()) self.deny_by_id(TypeId::of::<T>())
} }
@ -123,9 +126,10 @@ impl SceneFilter {
/// [`Allowlist`]: SceneFilter::Allowlist /// [`Allowlist`]: SceneFilter::Allowlist
/// [`Unset`]: SceneFilter::Unset /// [`Unset`]: SceneFilter::Unset
/// [`Denylist`]: SceneFilter::Denylist /// [`Denylist`]: SceneFilter::Denylist
pub fn deny_by_id(&mut self, type_id: TypeId) -> &mut Self { #[must_use]
match self { pub fn deny_by_id(mut self, type_id: TypeId) -> Self {
Self::Unset => *self = Self::Denylist(HashSet::from([type_id])), match &mut self {
Self::Unset => self = Self::Denylist(HashSet::from([type_id])),
Self::Allowlist(list) => { Self::Allowlist(list) => {
list.remove(&type_id); list.remove(&type_id);
} }
@ -231,27 +235,21 @@ mod tests {
#[test] #[test]
fn should_set_list_type_if_none() { fn should_set_list_type_if_none() {
let mut filter = SceneFilter::Unset; let filter = SceneFilter::Unset.allow::<i32>();
filter.allow::<i32>();
assert!(matches!(filter, SceneFilter::Allowlist(_))); assert!(matches!(filter, SceneFilter::Allowlist(_)));
let mut filter = SceneFilter::Unset; let filter = SceneFilter::Unset.deny::<i32>();
filter.deny::<i32>();
assert!(matches!(filter, SceneFilter::Denylist(_))); assert!(matches!(filter, SceneFilter::Denylist(_)));
} }
#[test] #[test]
fn should_add_to_list() { fn should_add_to_list() {
let mut filter = SceneFilter::default(); let filter = SceneFilter::default().allow::<i16>().allow::<i32>();
filter.allow::<i16>();
filter.allow::<i32>();
assert_eq!(2, filter.len()); assert_eq!(2, filter.len());
assert!(filter.is_allowed::<i16>()); assert!(filter.is_allowed::<i16>());
assert!(filter.is_allowed::<i32>()); assert!(filter.is_allowed::<i32>());
let mut filter = SceneFilter::default(); let filter = SceneFilter::default().deny::<i16>().deny::<i32>();
filter.deny::<i16>();
filter.deny::<i32>();
assert_eq!(2, filter.len()); assert_eq!(2, filter.len());
assert!(filter.is_denied::<i16>()); assert!(filter.is_denied::<i16>());
assert!(filter.is_denied::<i32>()); assert!(filter.is_denied::<i32>());
@ -259,18 +257,18 @@ mod tests {
#[test] #[test]
fn should_remove_from_list() { fn should_remove_from_list() {
let mut filter = SceneFilter::default(); let filter = SceneFilter::default()
filter.allow::<i16>(); .allow::<i16>()
filter.allow::<i32>(); .allow::<i32>()
filter.deny::<i32>(); .deny::<i32>();
assert_eq!(1, filter.len()); assert_eq!(1, filter.len());
assert!(filter.is_allowed::<i16>()); assert!(filter.is_allowed::<i16>());
assert!(!filter.is_allowed::<i32>()); assert!(!filter.is_allowed::<i32>());
let mut filter = SceneFilter::default(); let filter = SceneFilter::default()
filter.deny::<i16>(); .deny::<i16>()
filter.deny::<i32>(); .deny::<i32>()
filter.allow::<i32>(); .allow::<i32>();
assert_eq!(1, filter.len()); assert_eq!(1, filter.len());
assert!(filter.is_denied::<i16>()); assert!(filter.is_denied::<i16>());
assert!(!filter.is_denied::<i32>()); assert!(!filter.is_denied::<i32>());

8
crates/bevy_scene/src/serde.rs
View File

@ -592,10 +592,10 @@ mod tests {
world.insert_resource(MyResource { foo: 123 }); world.insert_resource(MyResource { foo: 123 });
let mut builder = DynamicSceneBuilder::from_world(&world); let scene = DynamicSceneBuilder::from_world(&world)
builder.extract_entities([a, b, c].into_iter()); .extract_entities([a, b, c].into_iter())
builder.extract_resources(); .extract_resources()
let scene = builder.build(); .build();
let expected = r#"( let expected = r#"(
resources: { resources: {