mirror of
https://github.com/bevyengine/bevy
synced 2024-12-23 19:43:07 +00:00
3a2a68852c
# Bevy ECS V2
This is a rewrite of Bevy ECS (basically everything but the new executor/schedule, which are already awesome). The overall goal was to improve the performance and versatility of Bevy ECS. Here is a quick bulleted list of changes before we dive into the details:
* Complete World rewrite
* Multiple component storage types:
* Tables: fast cache friendly iteration, slower add/removes (previously called Archetypes)
* Sparse Sets: fast add/remove, slower iteration
* Stateful Queries (caches query results for faster iteration. fragmented iteration is _fast_ now)
* Stateful System Params (caches expensive operations. inspired by @DJMcNab's work in #1364)
* Configurable System Params (users can set configuration when they construct their systems. once again inspired by @DJMcNab's work)
* Archetypes are now "just metadata", component storage is separate
* Archetype Graph (for faster archetype changes)
* Component Metadata
* Configure component storage type
* Retrieve information about component size/type/name/layout/send-ness/etc
* Components are uniquely identified by a densely packed ComponentId
* TypeIds are now totally optional (which should make implementing scripting easier)
* Super fast "for_each" query iterators
* Merged Resources into World. Resources are now just a special type of component
* EntityRef/EntityMut builder apis (more efficient and more ergonomic)
* Fast bitset-backed `Access<T>` replaces old hashmap-based approach everywhere
* Query conflicts are determined by component access instead of archetype component access (to avoid random failures at runtime)
* With/Without are still taken into account for conflicts, so this should still be comfy to use
* Much simpler `IntoSystem` impl
* Significantly reduced the amount of hashing throughout the ecs in favor of Sparse Sets (indexed by densely packed ArchetypeId, ComponentId, BundleId, and TableId)
* Safety Improvements
* Entity reservation uses a normal world reference instead of unsafe transmute
* QuerySets no longer transmute lifetimes
* Made traits "unsafe" where relevant
* More thorough safety docs
* WorldCell
* Exposes safe mutable access to multiple resources at a time in a World
* Replaced "catch all" `System::update_archetypes(world: &World)` with `System::new_archetype(archetype: &Archetype)`
* Simpler Bundle implementation
* Replaced slow "remove_bundle_one_by_one" used as fallback for Commands::remove_bundle with fast "remove_bundle_intersection"
* Removed `Mut<T>` query impl. it is better to only support one way: `&mut T`
* Removed with() from `Flags<T>` in favor of `Option<Flags<T>>`, which allows querying for flags to be "filtered" by default
* Components now have is_send property (currently only resources support non-send)
* More granular module organization
* New `RemovedComponents<T>` SystemParam that replaces `query.removed::<T>()`
* `world.resource_scope()` for mutable access to resources and world at the same time
* WorldQuery and QueryFilter traits unified. FilterFetch trait added to enable "short circuit" filtering. Auto impled for cases that don't need it
* Significantly slimmed down SystemState in favor of individual SystemParam state
* System Commands changed from `commands: &mut Commands` back to `mut commands: Commands` (to allow Commands to have a World reference)
Fixes #1320
## `World` Rewrite
This is a from-scratch rewrite of `World` that fills the niche that `hecs` used to. Yes, this means Bevy ECS is no longer a "fork" of hecs. We're going out our own!
(the only shared code between the projects is the entity id allocator, which is already basically ideal)
A huge shout out to @SanderMertens (author of [flecs](https://github.com/SanderMertens/flecs)) for sharing some great ideas with me (specifically hybrid ecs storage and archetype graphs). He also helped advise on a number of implementation details.
## Component Storage (The Problem)
Two ECS storage paradigms have gained a lot of traction over the years:
* **Archetypal ECS**:
* Stores components in "tables" with static schemas. Each "column" stores components of a given type. Each "row" is an entity.
* Each "archetype" has its own table. Adding/removing an entity's component changes the archetype.
* Enables super-fast Query iteration due to its cache-friendly data layout
* Comes at the cost of more expensive add/remove operations for an Entity's components, because all components need to be copied to the new archetype's "table"
* **Sparse Set ECS**:
* Stores components of the same type in densely packed arrays, which are sparsely indexed by densely packed unsigned integers (Entity ids)
* Query iteration is slower than Archetypal ECS because each entity's component could be at any position in the sparse set. This "random access" pattern isn't cache friendly. Additionally, there is an extra layer of indirection because you must first map the entity id to an index in the component array.
* Adding/removing components is a cheap, constant time operation
Bevy ECS V1, hecs, legion, flec, and Unity DOTS are all "archetypal ecs-es". I personally think "archetypal" storage is a good default for game engines. An entity's archetype doesn't need to change frequently in general, and it creates "fast by default" query iteration (which is a much more common operation). It is also "self optimizing". Users don't need to think about optimizing component layouts for iteration performance. It "just works" without any extra boilerplate.
Shipyard and EnTT are "sparse set ecs-es". They employ "packing" as a way to work around the "suboptimal by default" iteration performance for specific sets of components. This helps, but I didn't think this was a good choice for a general purpose engine like Bevy because:
1. "packs" conflict with each other. If bevy decides to internally pack the Transform and GlobalTransform components, users are then blocked if they want to pack some custom component with Transform.
2. users need to take manual action to optimize
Developers selecting an ECS framework are stuck with a hard choice. Select an "archetypal" framework with "fast iteration everywhere" but without the ability to cheaply add/remove components, or select a "sparse set" framework to cheaply add/remove components but with slower iteration performance.
## Hybrid Component Storage (The Solution)
In Bevy ECS V2, we get to have our cake and eat it too. It now has _both_ of the component storage types above (and more can be added later if needed):
* **Tables** (aka "archetypal" storage)
* The default storage. If you don't configure anything, this is what you get
* Fast iteration by default
* Slower add/remove operations
* **Sparse Sets**
* Opt-in
* Slower iteration
* Faster add/remove operations
These storage types complement each other perfectly. By default Query iteration is fast. If developers know that they want to add/remove a component at high frequencies, they can set the storage to "sparse set":
```rust
world.register_component(
ComponentDescriptor:🆕:<MyComponent>(StorageType::SparseSet)
).unwrap();
```
## Archetypes
Archetypes are now "just metadata" ... they no longer store components directly. They do store:
* The `ComponentId`s of each of the Archetype's components (and that component's storage type)
* Archetypes are uniquely defined by their component layouts
* For example: entities with "table" components `[A, B, C]` _and_ "sparse set" components `[D, E]` will always be in the same archetype.
* The `TableId` associated with the archetype
* For now each archetype has exactly one table (which can have no components),
* There is a 1->Many relationship from Tables->Archetypes. A given table could have any number of archetype components stored in it:
* Ex: an entity with "table storage" components `[A, B, C]` and "sparse set" components `[D, E]` will share the same `[A, B, C]` table as an entity with `[A, B, C]` table component and `[F]` sparse set components.
* This 1->Many relationship is how we preserve fast "cache friendly" iteration performance when possible (more on this later)
* A list of entities that are in the archetype and the row id of the table they are in
* ArchetypeComponentIds
* unique densely packed identifiers for (ArchetypeId, ComponentId) pairs
* used by the schedule executor for cheap system access control
* "Archetype Graph Edges" (see the next section)
## The "Archetype Graph"
Archetype changes in Bevy (and a number of other archetypal ecs-es) have historically been expensive to compute. First, you need to allocate a new vector of the entity's current component ids, add or remove components based on the operation performed, sort it (to ensure it is order-independent), then hash it to find the archetype (if it exists). And thats all before we get to the _already_ expensive full copy of all components to the new table storage.
The solution is to build a "graph" of archetypes to cache these results. @SanderMertens first exposed me to the idea (and he got it from @gjroelofs, who came up with it). They propose adding directed edges between archetypes for add/remove component operations. If `ComponentId`s are densely packed, you can use sparse sets to cheaply jump between archetypes.
Bevy takes this one step further by using add/remove `Bundle` edges instead of `Component` edges. Bevy encourages the use of `Bundles` to group add/remove operations. This is largely for "clearer game logic" reasons, but it also helps cut down on the number of archetype changes required. `Bundles` now also have densely-packed `BundleId`s. This allows us to use a _single_ edge for each bundle operation (rather than needing to traverse N edges ... one for each component). Single component operations are also bundles, so this is strictly an improvement over a "component only" graph.
As a result, an operation that used to be _heavy_ (both for allocations and compute) is now two dirt-cheap array lookups and zero allocations.
## Stateful Queries
World queries are now stateful. This allows us to:
1. Cache archetype (and table) matches
* This resolves another issue with (naive) archetypal ECS: query performance getting worse as the number of archetypes goes up (and fragmentation occurs).
2. Cache Fetch and Filter state
* The expensive parts of fetch/filter operations (such as hashing the TypeId to find the ComponentId) now only happen once when the Query is first constructed
3. Incrementally build up state
* When new archetypes are added, we only process the new archetypes (no need to rebuild state for old archetypes)
As a result, the direct `World` query api now looks like this:
```rust
let mut query = world.query::<(&A, &mut B)>();
for (a, mut b) in query.iter_mut(&mut world) {
}
```
Requiring `World` to generate stateful queries (rather than letting the `QueryState` type be constructed separately) allows us to ensure that _all_ queries are properly initialized (and the relevant world state, such as ComponentIds). This enables QueryState to remove branches from its operations that check for initialization status (and also enables query.iter() to take an immutable world reference because it doesn't need to initialize anything in world).
However in systems, this is a non-breaking change. State management is done internally by the relevant SystemParam.
## Stateful SystemParams
Like Queries, `SystemParams` now also cache state. For example, `Query` system params store the "stateful query" state mentioned above. Commands store their internal `CommandQueue`. This means you can now safely use as many separate `Commands` parameters in your system as you want. `Local<T>` system params store their `T` value in their state (instead of in Resources).
SystemParam state also enabled a significant slim-down of SystemState. It is much nicer to look at now.
Per-SystemParam state naturally insulates us from an "aliased mut" class of errors we have hit in the past (ex: using multiple `Commands` system params).
(credit goes to @DJMcNab for the initial idea and draft pr here #1364)
## Configurable SystemParams
@DJMcNab also had the great idea to make SystemParams configurable. This allows users to provide some initial configuration / values for system parameters (when possible). Most SystemParams have no config (the config type is `()`), but the `Local<T>` param now supports user-provided parameters:
```rust
fn foo(value: Local<usize>) {
}
app.add_system(foo.system().config(|c| c.0 = Some(10)));
```
## Uber Fast "for_each" Query Iterators
Developers now have the choice to use a fast "for_each" iterator, which yields ~1.5-3x iteration speed improvements for "fragmented iteration", and minor ~1.2x iteration speed improvements for unfragmented iteration.
```rust
fn system(query: Query<(&A, &mut B)>) {
// you now have the option to do this for a speed boost
query.for_each_mut(|(a, mut b)| {
});
// however normal iterators are still available
for (a, mut b) in query.iter_mut() {
}
}
```
I think in most cases we should continue to encourage "normal" iterators as they are more flexible and more "rust idiomatic". But when that extra "oomf" is needed, it makes sense to use `for_each`.
We should also consider using `for_each` for internal bevy systems to give our users a nice speed boost (but that should be a separate pr).
## Component Metadata
`World` now has a `Components` collection, which is accessible via `world.components()`. This stores mappings from `ComponentId` to `ComponentInfo`, as well as `TypeId` to `ComponentId` mappings (where relevant). `ComponentInfo` stores information about the component, such as ComponentId, TypeId, memory layout, send-ness (currently limited to resources), and storage type.
## Significantly Cheaper `Access<T>`
We used to use `TypeAccess<TypeId>` to manage read/write component/archetype-component access. This was expensive because TypeIds must be hashed and compared individually. The parallel executor got around this by "condensing" type ids into bitset-backed access types. This worked, but it had to be re-generated from the `TypeAccess<TypeId>`sources every time archetypes changed.
This pr removes TypeAccess in favor of faster bitset access everywhere. We can do this thanks to the move to densely packed `ComponentId`s and `ArchetypeComponentId`s.
## Merged Resources into World
Resources had a lot of redundant functionality with Components. They stored typed data, they had access control, they had unique ids, they were queryable via SystemParams, etc. In fact the _only_ major difference between them was that they were unique (and didn't correlate to an entity).
Separate resources also had the downside of requiring a separate set of access controls, which meant the parallel executor needed to compare more bitsets per system and manage more state.
I initially got the "separate resources" idea from `legion`. I think that design was motivated by the fact that it made the direct world query/resource lifetime interactions more manageable. It certainly made our lives easier when using Resources alongside hecs/bevy_ecs. However we already have a construct for safely and ergonomically managing in-world lifetimes: systems (which use `Access<T>` internally).
This pr merges Resources into World:
```rust
world.insert_resource(1);
world.insert_resource(2.0);
let a = world.get_resource::<i32>().unwrap();
let mut b = world.get_resource_mut::<f64>().unwrap();
*b = 3.0;
```
Resources are now just a special kind of component. They have their own ComponentIds (and their own resource TypeId->ComponentId scope, so they don't conflict wit components of the same type). They are stored in a special "resource archetype", which stores components inside the archetype using a new `unique_components` sparse set (note that this sparse set could later be used to implement Tags). This allows us to keep the code size small by reusing existing datastructures (namely Column, Archetype, ComponentFlags, and ComponentInfo). This allows us the executor to use a single `Access<ArchetypeComponentId>` per system. It should also make scripting language integration easier.
_But_ this merge did create problems for people directly interacting with `World`. What if you need mutable access to multiple resources at the same time? `world.get_resource_mut()` borrows World mutably!
## WorldCell
WorldCell applies the `Access<ArchetypeComponentId>` concept to direct world access:
```rust
let world_cell = world.cell();
let a = world_cell.get_resource_mut::<i32>().unwrap();
let b = world_cell.get_resource_mut::<f64>().unwrap();
```
This adds cheap runtime checks (a sparse set lookup of `ArchetypeComponentId` and a counter) to ensure that world accesses do not conflict with each other. Each operation returns a `WorldBorrow<'w, T>` or `WorldBorrowMut<'w, T>` wrapper type, which will release the relevant ArchetypeComponentId resources when dropped.
World caches the access sparse set (and only one cell can exist at a time), so `world.cell()` is a cheap operation.
WorldCell does _not_ use atomic operations. It is non-send, does a mutable borrow of world to prevent other accesses, and uses a simple `Rc<RefCell<ArchetypeComponentAccess>>` wrapper in each WorldBorrow pointer.
The api is currently limited to resource access, but it can and should be extended to queries / entity component access.
## Resource Scopes
WorldCell does not yet support component queries, and even when it does there are sometimes legitimate reasons to want a mutable world ref _and_ a mutable resource ref (ex: bevy_render and bevy_scene both need this). In these cases we could always drop down to the unsafe `world.get_resource_unchecked_mut()`, but that is not ideal!
Instead developers can use a "resource scope"
```rust
world.resource_scope(|world: &mut World, a: &mut A| {
})
```
This temporarily removes the `A` resource from `World`, provides mutable pointers to both, and re-adds A to World when finished. Thanks to the move to ComponentIds/sparse sets, this is a cheap operation.
If multiple resources are required, scopes can be nested. We could also consider adding a "resource tuple" to the api if this pattern becomes common and the boilerplate gets nasty.
## Query Conflicts Use ComponentId Instead of ArchetypeComponentId
For safety reasons, systems cannot contain queries that conflict with each other without wrapping them in a QuerySet. On bevy `main`, we use ArchetypeComponentIds to determine conflicts. This is nice because it can take into account filters:
```rust
// these queries will never conflict due to their filters
fn filter_system(a: Query<&mut A, With<B>>, b: Query<&mut B, Without<B>>) {
}
```
But it also has a significant downside:
```rust
// these queries will not conflict _until_ an entity with A, B, and C is spawned
fn maybe_conflicts_system(a: Query<(&mut A, &C)>, b: Query<(&mut A, &B)>) {
}
```
The system above will panic at runtime if an entity with A, B, and C is spawned. This makes it hard to trust that your game logic will run without crashing.
In this pr, I switched to using `ComponentId` instead. This _is_ more constraining. `maybe_conflicts_system` will now always fail, but it will do it consistently at startup. Naively, it would also _disallow_ `filter_system`, which would be a significant downgrade in usability. Bevy has a number of internal systems that rely on disjoint queries and I expect it to be a common pattern in userspace.
To resolve this, I added a new `FilteredAccess<T>` type, which wraps `Access<T>` and adds with/without filters. If two `FilteredAccess` have with/without values that prove they are disjoint, they will no longer conflict.
## EntityRef / EntityMut
World entity operations on `main` require that the user passes in an `entity` id to each operation:
```rust
let entity = world.spawn((A, )); // create a new entity with A
world.get::<A>(entity);
world.insert(entity, (B, C));
world.insert_one(entity, D);
```
This means that each operation needs to look up the entity location / verify its validity. The initial spawn operation also requires a Bundle as input. This can be awkward when no components are required (or one component is required).
These operations have been replaced by `EntityRef` and `EntityMut`, which are "builder-style" wrappers around world that provide read and read/write operations on a single, pre-validated entity:
```rust
// spawn now takes no inputs and returns an EntityMut
let entity = world.spawn()
.insert(A) // insert a single component into the entity
.insert_bundle((B, C)) // insert a bundle of components into the entity
.id() // id returns the Entity id
// Returns EntityMut (or panics if the entity does not exist)
world.entity_mut(entity)
.insert(D)
.insert_bundle(SomeBundle::default());
{
// returns EntityRef (or panics if the entity does not exist)
let d = world.entity(entity)
.get::<D>() // gets the D component
.unwrap();
// world.get still exists for ergonomics
let d = world.get::<D>(entity).unwrap();
}
// These variants return Options if you want to check existence instead of panicing
world.get_entity_mut(entity)
.unwrap()
.insert(E);
if let Some(entity_ref) = world.get_entity(entity) {
let d = entity_ref.get::<D>().unwrap();
}
```
This _does not_ affect the current Commands api or terminology. I think that should be a separate conversation as that is a much larger breaking change.
## Safety Improvements
* Entity reservation in Commands uses a normal world borrow instead of an unsafe transmute
* QuerySets no longer transmutes lifetimes
* Made traits "unsafe" when implementing a trait incorrectly could cause unsafety
* More thorough safety docs
## RemovedComponents SystemParam
The old approach to querying removed components: `query.removed:<T>()` was confusing because it had no connection to the query itself. I replaced it with the following, which is both clearer and allows us to cache the ComponentId mapping in the SystemParamState:
```rust
fn system(removed: RemovedComponents<T>) {
for entity in removed.iter() {
}
}
```
## Simpler Bundle implementation
Bundles are no longer responsible for sorting (or deduping) TypeInfo. They are just a simple ordered list of component types / data. This makes the implementation smaller and opens the door to an easy "nested bundle" implementation in the future (which i might even add in this pr). Duplicate detection is now done once per bundle type by World the first time a bundle is used.
## Unified WorldQuery and QueryFilter types
(don't worry they are still separate type _parameters_ in Queries .. this is a non-breaking change)
WorldQuery and QueryFilter were already basically identical apis. With the addition of `FetchState` and more storage-specific fetch methods, the overlap was even clearer (and the redundancy more painful).
QueryFilters are now just `F: WorldQuery where F::Fetch: FilterFetch`. FilterFetch requires `Fetch<Item = bool>` and adds new "short circuit" variants of fetch methods. This enables a filter tuple like `(With<A>, Without<B>, Changed<C>)` to stop evaluating the filter after the first mismatch is encountered. FilterFetch is automatically implemented for `Fetch` implementations that return bool.
This forces fetch implementations that return things like `(bool, bool, bool)` (such as the filter above) to manually implement FilterFetch and decide whether or not to short-circuit.
## More Granular Modules
World no longer globs all of the internal modules together. It now exports `core`, `system`, and `schedule` separately. I'm also considering exporting `core` submodules directly as that is still pretty "glob-ey" and unorganized (feedback welcome here).
## Remaining Draft Work (to be done in this pr)
* ~~panic on conflicting WorldQuery fetches (&A, &mut A)~~
* ~~bevy `main` and hecs both currently allow this, but we should protect against it if possible~~
* ~~batch_iter / par_iter (currently stubbed out)~~
* ~~ChangedRes~~
* ~~I skipped this while we sort out #1313. This pr should be adapted to account for whatever we land on there~~.
* ~~The `Archetypes` and `Tables` collections use hashes of sorted lists of component ids to uniquely identify each archetype/table. This hash is then used as the key in a HashMap to look up the relevant ArchetypeId or TableId. (which doesn't handle hash collisions properly)~~
* ~~It is currently unsafe to generate a Query from "World A", then use it on "World B" (despite the api claiming it is safe). We should probably close this gap. This could be done by adding a randomly generated WorldId to each world, then storing that id in each Query. They could then be compared to each other on each `query.do_thing(&world)` operation. This _does_ add an extra branch to each query operation, so I'm open to other suggestions if people have them.~~
* ~~Nested Bundles (if i find time)~~
## Potential Future Work
* Expand WorldCell to support queries.
* Consider not allocating in the empty archetype on `world.spawn()`
* ex: return something like EntityMutUninit, which turns into EntityMut after an `insert` or `insert_bundle` op
* this actually regressed performance last time i tried it, but in theory it should be faster
* Optimize SparseSet::insert (see `PERF` comment on insert)
* Replace SparseArray `Option<T>` with T::MAX to cut down on branching
* would enable cheaper get_unchecked() operations
* upstream fixedbitset optimizations
* fixedbitset could be allocation free for small block counts (store blocks in a SmallVec)
* fixedbitset could have a const constructor
* Consider implementing Tags (archetype-specific by-value data that affects archetype identity)
* ex: ArchetypeA could have `[A, B, C]` table components and `[D(1)]` "tag" component. ArchetypeB could have `[A, B, C]` table components and a `[D(2)]` tag component. The archetypes are different, despite both having D tags because the value inside D is different.
* this could potentially build on top of the `archetype.unique_components` added in this pr for resource storage.
* Consider reverting `all_tuples` proc macro in favor of the old `macro_rules` implementation
* all_tuples is more flexible and produces cleaner documentation (the macro_rules version produces weird type parameter orders due to parser constraints)
* but unfortunately all_tuples also appears to make Rust Analyzer sad/slow when working inside of `bevy_ecs` (does not affect user code)
* Consider "resource queries" and/or "mixed resource and entity component queries" as an alternative to WorldCell
* this is basically just "systems" so maybe it's not worth it
* Add more world ops
* `world.clear()`
* `world.reserve<T: Bundle>(count: usize)`
* Try using the old archetype allocation strategy (allocate new memory on resize and copy everything over). I expect this to improve batch insertion performance at the cost of unbatched performance. But thats just a guess. I'm not an allocation perf pro :)
* Adapt Commands apis for consistency with new World apis
## Benchmarks
key:
* `bevy_old`: bevy `main` branch
* `bevy`: this branch
* `_foreach`: uses an optimized for_each iterator
* ` _sparse`: uses sparse set storage (if unspecified assume table storage)
* `_system`: runs inside a system (if unspecified assume test happens via direct world ops)
### Simple Insert (from ecs_bench_suite)
### Simpler Iter (from ecs_bench_suite)
### Fragment Iter (from ecs_bench_suite)
### Sparse Fragmented Iter
Iterate a query that matches 5 entities from a single matching archetype, but there are 100 unmatching archetypes
### Schedule (from ecs_bench_suite)
### Add Remove Component (from ecs_bench_suite)
### Add Remove Component Big
Same as the test above, but each entity has 5 "large" matrix components and 1 "large" matrix component is added and removed
### Get Component
Looks up a single component value a large number of times
900 lines
34 KiB
Rust
900 lines
34 KiB
Rust
use crate::{
|
|
archetype::{Archetype, ArchetypeId, Archetypes},
|
|
bundle::{Bundle, BundleInfo},
|
|
component::{Component, ComponentFlags, ComponentId, Components, StorageType},
|
|
entity::{Entity, EntityLocation},
|
|
storage::{SparseSet, Storages},
|
|
world::{Mut, World},
|
|
};
|
|
use std::any::TypeId;
|
|
|
|
pub struct EntityRef<'w> {
|
|
world: &'w World,
|
|
entity: Entity,
|
|
location: EntityLocation,
|
|
}
|
|
|
|
impl<'w> EntityRef<'w> {
|
|
#[inline]
|
|
pub(crate) fn new(world: &'w World, entity: Entity, location: EntityLocation) -> Self {
|
|
Self {
|
|
world,
|
|
entity,
|
|
location,
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub fn id(&self) -> Entity {
|
|
self.entity
|
|
}
|
|
|
|
#[inline]
|
|
pub fn location(&self) -> EntityLocation {
|
|
self.location
|
|
}
|
|
|
|
#[inline]
|
|
pub fn archetype(&self) -> &Archetype {
|
|
// SAFE: EntityRefs always point to valid entities. Valid entities always have valid archetypes
|
|
unsafe {
|
|
self.world
|
|
.archetypes
|
|
.get_unchecked(self.location.archetype_id)
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub fn world(&mut self) -> &World {
|
|
self.world
|
|
}
|
|
|
|
#[inline]
|
|
pub fn contains<T: Component>(&self) -> bool {
|
|
self.contains_type_id(TypeId::of::<T>())
|
|
}
|
|
|
|
#[inline]
|
|
pub fn contains_id(&self, component_id: ComponentId) -> bool {
|
|
// SAFE: entity location is valid
|
|
unsafe { contains_component_with_id(self.world, component_id, self.location) }
|
|
}
|
|
|
|
#[inline]
|
|
pub fn contains_type_id(&self, type_id: TypeId) -> bool {
|
|
// SAFE: entity location is valid
|
|
unsafe { contains_component_with_type(self.world, type_id, self.location) }
|
|
}
|
|
|
|
#[inline]
|
|
pub fn get<T: Component>(&self) -> Option<&'w T> {
|
|
// SAFE: entity location is valid and returned component is of type T
|
|
unsafe {
|
|
get_component_with_type(self.world, TypeId::of::<T>(), self.entity, self.location)
|
|
.map(|value| &*value.cast::<T>())
|
|
}
|
|
}
|
|
|
|
/// # Safety
|
|
/// This allows aliased mutability. You must make sure this call does not result in multiple mutable references to the same component
|
|
#[inline]
|
|
pub unsafe fn get_unchecked_mut<T: Component>(&self) -> Option<Mut<'w, T>> {
|
|
get_component_and_flags_with_type(self.world, TypeId::of::<T>(), self.entity, self.location)
|
|
.map(|(value, flags)| Mut {
|
|
value: &mut *value.cast::<T>(),
|
|
flags: &mut *flags,
|
|
})
|
|
}
|
|
}
|
|
|
|
pub struct EntityMut<'w> {
|
|
world: &'w mut World,
|
|
entity: Entity,
|
|
location: EntityLocation,
|
|
}
|
|
|
|
impl<'w> EntityMut<'w> {
|
|
/// # Safety
|
|
/// entity and location _must_ be valid
|
|
#[inline]
|
|
pub(crate) unsafe fn new(
|
|
world: &'w mut World,
|
|
entity: Entity,
|
|
location: EntityLocation,
|
|
) -> Self {
|
|
EntityMut {
|
|
world,
|
|
entity,
|
|
location,
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub fn id(&self) -> Entity {
|
|
self.entity
|
|
}
|
|
|
|
#[inline]
|
|
pub fn location(&self) -> EntityLocation {
|
|
self.location
|
|
}
|
|
|
|
#[inline]
|
|
pub fn archetype(&self) -> &Archetype {
|
|
// SAFE: EntityRefs always point to valid entities. Valid entities always have valid archetypes
|
|
unsafe {
|
|
self.world
|
|
.archetypes
|
|
.get_unchecked(self.location.archetype_id)
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub fn contains<T: Component>(&self) -> bool {
|
|
self.contains_type_id(TypeId::of::<T>())
|
|
}
|
|
|
|
#[inline]
|
|
pub fn contains_id(&self, component_id: ComponentId) -> bool {
|
|
// SAFE: entity location is valid
|
|
unsafe { contains_component_with_id(self.world, component_id, self.location) }
|
|
}
|
|
|
|
#[inline]
|
|
pub fn contains_type_id(&self, type_id: TypeId) -> bool {
|
|
// SAFE: entity location is valid
|
|
unsafe { contains_component_with_type(self.world, type_id, self.location) }
|
|
}
|
|
|
|
#[inline]
|
|
pub fn get<T: Component>(&self) -> Option<&'w T> {
|
|
// SAFE: entity location is valid and returned component is of type T
|
|
unsafe {
|
|
get_component_with_type(self.world, TypeId::of::<T>(), self.entity, self.location)
|
|
.map(|value| &*value.cast::<T>())
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub fn get_mut<T: Component>(&mut self) -> Option<Mut<'w, T>> {
|
|
// SAFE: world access is unique, entity location is valid, and returned component is of type T
|
|
unsafe {
|
|
get_component_and_flags_with_type(
|
|
self.world,
|
|
TypeId::of::<T>(),
|
|
self.entity,
|
|
self.location,
|
|
)
|
|
.map(|(value, flags)| Mut {
|
|
value: &mut *value.cast::<T>(),
|
|
flags: &mut *flags,
|
|
})
|
|
}
|
|
}
|
|
|
|
/// # Safety
|
|
/// This allows aliased mutability. You must make sure this call does not result in multiple mutable references to the same component
|
|
#[inline]
|
|
pub unsafe fn get_unchecked_mut<T: Component>(&self) -> Option<Mut<'w, T>> {
|
|
get_component_and_flags_with_type(self.world, TypeId::of::<T>(), self.entity, self.location)
|
|
.map(|(value, flags)| Mut {
|
|
value: &mut *value.cast::<T>(),
|
|
flags: &mut *flags,
|
|
})
|
|
}
|
|
|
|
// TODO: factor out non-generic part to cut down on monomorphization (just check perf)
|
|
// TODO: move relevant methods to World (add/remove bundle)
|
|
pub fn insert_bundle<T: Bundle>(&mut self, bundle: T) -> &mut Self {
|
|
let entity = self.entity;
|
|
let entities = &mut self.world.entities;
|
|
let archetypes = &mut self.world.archetypes;
|
|
let components = &mut self.world.components;
|
|
let storages = &mut self.world.storages;
|
|
|
|
let bundle_info = self.world.bundles.init_info::<T>(components);
|
|
let current_location = self.location;
|
|
|
|
let new_location = unsafe {
|
|
// SAFE: component ids in `bundle_info` and self.location are valid
|
|
let new_archetype_id = add_bundle_to_archetype(
|
|
archetypes,
|
|
storages,
|
|
components,
|
|
self.location.archetype_id,
|
|
bundle_info,
|
|
);
|
|
if new_archetype_id == current_location.archetype_id {
|
|
current_location
|
|
} else {
|
|
let old_table_row;
|
|
let old_table_id;
|
|
{
|
|
let old_archetype = archetypes.get_unchecked_mut(current_location.archetype_id);
|
|
let result = old_archetype.swap_remove(current_location.index);
|
|
if let Some(swapped_entity) = result.swapped_entity {
|
|
// SAFE: entity is live and is contained in an archetype that exists
|
|
entities.meta[swapped_entity.id as usize].location = current_location;
|
|
}
|
|
old_table_row = result.table_row;
|
|
old_table_id = old_archetype.table_id()
|
|
}
|
|
let new_archetype = archetypes.get_unchecked_mut(new_archetype_id);
|
|
|
|
if old_table_id == new_archetype.table_id() {
|
|
new_archetype.allocate(entity, old_table_row)
|
|
} else {
|
|
let (old_table, new_table) = storages
|
|
.tables
|
|
.get_2_mut(old_table_id, new_archetype.table_id());
|
|
// PERF: store "non bundle" components in edge, then just move those to avoid redundant copies
|
|
let move_result =
|
|
old_table.move_to_superset_unchecked(old_table_row, new_table);
|
|
|
|
let new_location = new_archetype.allocate(entity, move_result.new_row);
|
|
// if an entity was moved into this entity's table spot, update its table row
|
|
if let Some(swapped_entity) = move_result.swapped_entity {
|
|
let swapped_location = entities.get(swapped_entity).unwrap();
|
|
// SAFE: entity is live and is therefore contained in an archetype that exists
|
|
archetypes
|
|
.get_unchecked_mut(swapped_location.archetype_id)
|
|
.set_entity_table_row(swapped_location.index, old_table_row);
|
|
}
|
|
new_location
|
|
}
|
|
|
|
// Sparse set components are intentionally ignored here. They don't need to move
|
|
}
|
|
};
|
|
self.location = new_location;
|
|
entities.meta[self.entity.id as usize].location = new_location;
|
|
|
|
// SAFE: archetype was created if it didn't already exist
|
|
let archetype = unsafe { archetypes.get_unchecked_mut(new_location.archetype_id) };
|
|
// SAFE: archetype tables always exists
|
|
let table = unsafe { storages.tables.get_unchecked_mut(archetype.table_id()) };
|
|
let table_row = archetype.entity_table_row(new_location.index);
|
|
let from_bundle = archetype.edges().get_from_bundle(bundle_info.id).unwrap();
|
|
// SAFE: table row is valid
|
|
unsafe {
|
|
bundle_info.write_components(
|
|
&mut storages.sparse_sets,
|
|
entity,
|
|
table,
|
|
table_row,
|
|
&from_bundle.bundle_flags,
|
|
bundle,
|
|
)
|
|
};
|
|
self
|
|
}
|
|
|
|
pub fn remove_bundle<T: Bundle>(&mut self) -> Option<T> {
|
|
let archetypes = &mut self.world.archetypes;
|
|
let storages = &mut self.world.storages;
|
|
let components = &mut self.world.components;
|
|
let entities = &mut self.world.entities;
|
|
let removed_components = &mut self.world.removed_components;
|
|
|
|
let bundle_info = self.world.bundles.init_info::<T>(components);
|
|
let old_location = self.location;
|
|
let new_archetype_id = unsafe {
|
|
remove_bundle_from_archetype(
|
|
archetypes,
|
|
storages,
|
|
components,
|
|
old_location.archetype_id,
|
|
bundle_info,
|
|
false,
|
|
)?
|
|
};
|
|
|
|
if new_archetype_id == old_location.archetype_id {
|
|
return None;
|
|
}
|
|
|
|
// SAFE: current entity archetype is valid
|
|
let old_archetype = unsafe { archetypes.get_unchecked_mut(old_location.archetype_id) };
|
|
let mut bundle_components = bundle_info.component_ids.iter().cloned();
|
|
let entity = self.entity;
|
|
// SAFE: bundle components are iterated in order, which guarantees that the component type matches
|
|
let result = unsafe {
|
|
T::from_components(|| {
|
|
let component_id = bundle_components.next().unwrap();
|
|
// SAFE: entity location is valid and table row is removed below
|
|
remove_component(
|
|
components,
|
|
storages,
|
|
old_archetype,
|
|
removed_components,
|
|
component_id,
|
|
entity,
|
|
old_location,
|
|
)
|
|
})
|
|
};
|
|
|
|
let remove_result = old_archetype.swap_remove(old_location.index);
|
|
if let Some(swapped_entity) = remove_result.swapped_entity {
|
|
entities.meta[swapped_entity.id as usize].location = old_location;
|
|
}
|
|
let old_table_row = remove_result.table_row;
|
|
let old_table_id = old_archetype.table_id();
|
|
// SAFE: new archetype exists thanks to remove_bundle_from_archetype
|
|
let new_archetype = unsafe { archetypes.get_unchecked_mut(new_archetype_id) };
|
|
|
|
let new_location = if old_table_id == new_archetype.table_id() {
|
|
unsafe { new_archetype.allocate(entity, old_table_row) }
|
|
} else {
|
|
let (old_table, new_table) = storages
|
|
.tables
|
|
.get_2_mut(old_table_id, new_archetype.table_id());
|
|
|
|
// SAFE: table_row exists. All "missing" components have been extracted into the bundle above and the caller takes ownership
|
|
let move_result =
|
|
unsafe { old_table.move_to_and_forget_missing_unchecked(old_table_row, new_table) };
|
|
|
|
// SAFE: new_table_row is a valid position in new_archetype's table
|
|
let new_location = unsafe { new_archetype.allocate(entity, move_result.new_row) };
|
|
|
|
// if an entity was moved into this entity's table spot, update its table row
|
|
if let Some(swapped_entity) = move_result.swapped_entity {
|
|
let swapped_location = entities.get(swapped_entity).unwrap();
|
|
// SAFE: entity is live and is contained in an archetype that exists
|
|
let archetype =
|
|
unsafe { archetypes.get_unchecked_mut(swapped_location.archetype_id) };
|
|
archetype.set_entity_table_row(swapped_location.index, old_table_row);
|
|
}
|
|
|
|
new_location
|
|
};
|
|
|
|
self.location = new_location;
|
|
entities.meta[self.entity.id as usize].location = new_location;
|
|
|
|
Some(result)
|
|
}
|
|
|
|
/// Remove any components in the bundle that the entity has.
|
|
pub fn remove_bundle_intersection<T: Bundle>(&mut self) {
|
|
let archetypes = &mut self.world.archetypes;
|
|
let storages = &mut self.world.storages;
|
|
let components = &mut self.world.components;
|
|
let entities = &mut self.world.entities;
|
|
let removed_components = &mut self.world.removed_components;
|
|
|
|
let bundle_info = self.world.bundles.init_info::<T>(components);
|
|
let old_location = self.location;
|
|
let new_archetype_id = unsafe {
|
|
remove_bundle_from_archetype(
|
|
archetypes,
|
|
storages,
|
|
components,
|
|
old_location.archetype_id,
|
|
bundle_info,
|
|
true,
|
|
)
|
|
.expect("intersections should always return a result")
|
|
};
|
|
|
|
if new_archetype_id == old_location.archetype_id {
|
|
return;
|
|
}
|
|
|
|
// SAFE: current entity archetype is valid
|
|
let old_archetype = unsafe { archetypes.get_unchecked_mut(old_location.archetype_id) };
|
|
let entity = self.entity;
|
|
for component_id in bundle_info.component_ids.iter().cloned() {
|
|
if old_archetype.contains(component_id) {
|
|
// SAFE: entity location is valid and table row is removed below
|
|
unsafe {
|
|
remove_component(
|
|
components,
|
|
storages,
|
|
old_archetype,
|
|
removed_components,
|
|
component_id,
|
|
entity,
|
|
old_location,
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
let remove_result = old_archetype.swap_remove(old_location.index);
|
|
if let Some(swapped_entity) = remove_result.swapped_entity {
|
|
entities.meta[swapped_entity.id as usize].location = old_location;
|
|
}
|
|
let old_table_row = remove_result.table_row;
|
|
let old_table_id = old_archetype.table_id();
|
|
// SAFE: new archetype exists thanks to remove_bundle_from_archetype
|
|
let new_archetype = unsafe { archetypes.get_unchecked_mut(new_archetype_id) };
|
|
|
|
let new_location = if old_table_id == new_archetype.table_id() {
|
|
unsafe { new_archetype.allocate(entity, old_table_row) }
|
|
} else {
|
|
let (old_table, new_table) = storages
|
|
.tables
|
|
.get_2_mut(old_table_id, new_archetype.table_id());
|
|
|
|
// SAFE: table_row exists
|
|
let move_result =
|
|
unsafe { old_table.move_to_and_drop_missing_unchecked(old_table_row, new_table) };
|
|
|
|
// SAFE: new_table_row is a valid position in new_archetype's table
|
|
let new_location = unsafe { new_archetype.allocate(entity, move_result.new_row) };
|
|
|
|
// if an entity was moved into this entity's table spot, update its table row
|
|
if let Some(swapped_entity) = move_result.swapped_entity {
|
|
let swapped_location = entities.get(swapped_entity).unwrap();
|
|
// SAFE: entity is live and is contained in an archetype that exists
|
|
let archetype =
|
|
unsafe { archetypes.get_unchecked_mut(swapped_location.archetype_id) };
|
|
archetype.set_entity_table_row(swapped_location.index, old_table_row);
|
|
}
|
|
|
|
new_location
|
|
};
|
|
|
|
self.location = new_location;
|
|
entities.meta[self.entity.id as usize].location = new_location;
|
|
}
|
|
|
|
pub fn insert<T: Component>(&mut self, value: T) -> &mut Self {
|
|
self.insert_bundle((value,))
|
|
}
|
|
|
|
pub fn remove<T: Component>(&mut self) -> Option<T> {
|
|
self.remove_bundle::<(T,)>().map(|v| v.0)
|
|
}
|
|
|
|
pub fn despawn(self) {
|
|
let world = self.world;
|
|
world.flush();
|
|
let location = world
|
|
.entities
|
|
.free(self.entity)
|
|
.expect("entity should exist at this point.");
|
|
let table_row;
|
|
let moved_entity;
|
|
{
|
|
// SAFE: entity is live and is contained in an archetype that exists
|
|
let archetype = unsafe { world.archetypes.get_unchecked_mut(location.archetype_id) };
|
|
for component_id in archetype.components() {
|
|
let removed_components = world
|
|
.removed_components
|
|
.get_or_insert_with(component_id, Vec::new);
|
|
removed_components.push(self.entity);
|
|
}
|
|
let remove_result = archetype.swap_remove(location.index);
|
|
if let Some(swapped_entity) = remove_result.swapped_entity {
|
|
world.entities.meta[swapped_entity.id as usize].location = location;
|
|
}
|
|
table_row = remove_result.table_row;
|
|
|
|
for component_id in archetype.sparse_set_components() {
|
|
let sparse_set = world.storages.sparse_sets.get_mut(*component_id).unwrap();
|
|
sparse_set.remove(self.entity);
|
|
}
|
|
// SAFE: tables and table rows stored in archetypes always exist
|
|
moved_entity = unsafe {
|
|
world
|
|
.storages
|
|
.tables
|
|
.get_unchecked_mut(archetype.table_id())
|
|
.swap_remove_unchecked(table_row)
|
|
};
|
|
};
|
|
|
|
if let Some(moved_entity) = moved_entity {
|
|
let moved_location = world.entities.get(moved_entity).unwrap();
|
|
// SAFE: entity is live and is contained in an archetype that exists
|
|
let archetype = unsafe {
|
|
world
|
|
.archetypes
|
|
.get_unchecked_mut(moved_location.archetype_id)
|
|
};
|
|
archetype.set_entity_table_row(moved_location.index, table_row);
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub fn world(&mut self) -> &World {
|
|
self.world
|
|
}
|
|
|
|
/// # Safety
|
|
/// Caller must not modify the world in a way that changes the current entity's location
|
|
/// If the caller _does_ do something that could change the location, self.update_location() must be
|
|
/// called before using any other methods in EntityMut
|
|
#[inline]
|
|
pub unsafe fn world_mut(&mut self) -> &mut World {
|
|
self.world
|
|
}
|
|
|
|
/// Updates the internal entity location to match the current location in the internal [World].
|
|
/// This is only needed if the user called [EntityMut::world], which enables the location to change.
|
|
pub fn update_location(&mut self) {
|
|
self.location = self.world.entities().get(self.entity).unwrap();
|
|
}
|
|
}
|
|
|
|
/// # Safety
|
|
/// `entity_location` must be within bounds of the given archetype and `entity` must exist inside the archetype
|
|
#[inline]
|
|
unsafe fn get_component(
|
|
world: &World,
|
|
component_id: ComponentId,
|
|
entity: Entity,
|
|
location: EntityLocation,
|
|
) -> Option<*mut u8> {
|
|
let archetype = world.archetypes.get_unchecked(location.archetype_id);
|
|
// SAFE: component_id exists and is therefore valid
|
|
let component_info = world.components.get_info_unchecked(component_id);
|
|
match component_info.storage_type() {
|
|
StorageType::Table => {
|
|
// SAFE: tables stored in archetype always exist
|
|
let table = world.storages.tables.get_unchecked(archetype.table_id());
|
|
let components = table.get_column(component_id)?;
|
|
let table_row = archetype.entity_table_row(location.index);
|
|
// SAFE: archetypes only store valid table_rows and the stored component type is T
|
|
Some(components.get_unchecked(table_row))
|
|
}
|
|
StorageType::SparseSet => world
|
|
.storages
|
|
.sparse_sets
|
|
.get(component_id)
|
|
.and_then(|sparse_set| sparse_set.get(entity)),
|
|
}
|
|
}
|
|
|
|
/// # Safety
|
|
/// Caller must ensure that `component_id` is valid
|
|
#[inline]
|
|
unsafe fn get_component_and_flags(
|
|
world: &World,
|
|
component_id: ComponentId,
|
|
entity: Entity,
|
|
location: EntityLocation,
|
|
) -> Option<(*mut u8, *mut ComponentFlags)> {
|
|
let archetype = world.archetypes.get_unchecked(location.archetype_id);
|
|
let component_info = world.components.get_info_unchecked(component_id);
|
|
match component_info.storage_type() {
|
|
StorageType::Table => {
|
|
// SAFE: tables stored in archetype always exist
|
|
let table = world.storages.tables.get_unchecked(archetype.table_id());
|
|
let components = table.get_column(component_id)?;
|
|
let table_row = archetype.entity_table_row(location.index);
|
|
// SAFE: archetypes only store valid table_rows and the stored component type is T
|
|
Some((
|
|
components.get_unchecked(table_row),
|
|
components.get_flags_unchecked(table_row),
|
|
))
|
|
}
|
|
StorageType::SparseSet => world
|
|
.storages
|
|
.sparse_sets
|
|
.get(component_id)
|
|
.and_then(|sparse_set| sparse_set.get_with_flags(entity)),
|
|
}
|
|
}
|
|
|
|
/// # Safety
|
|
// `entity_location` must be within bounds of the given archetype and `entity` must exist inside the archetype
|
|
/// The relevant table row must be removed separately
|
|
/// `component_id` must be valid
|
|
#[inline]
|
|
unsafe fn remove_component(
|
|
components: &Components,
|
|
storages: &mut Storages,
|
|
archetype: &Archetype,
|
|
removed_components: &mut SparseSet<ComponentId, Vec<Entity>>,
|
|
component_id: ComponentId,
|
|
entity: Entity,
|
|
location: EntityLocation,
|
|
) -> *mut u8 {
|
|
let component_info = components.get_info_unchecked(component_id);
|
|
let removed_components = removed_components.get_or_insert_with(component_id, Vec::new);
|
|
removed_components.push(entity);
|
|
match component_info.storage_type() {
|
|
StorageType::Table => {
|
|
// SAFE: tables stored in archetype always exist
|
|
let table = storages.tables.get_unchecked(archetype.table_id());
|
|
// SAFE: archetypes will always point to valid columns
|
|
let components = table.get_column(component_id).unwrap();
|
|
let table_row = archetype.entity_table_row(location.index);
|
|
// SAFE: archetypes only store valid table_rows and the stored component type is T
|
|
components.get_unchecked(table_row)
|
|
}
|
|
StorageType::SparseSet => storages
|
|
.sparse_sets
|
|
.get_mut(component_id)
|
|
.unwrap()
|
|
.remove_and_forget(entity)
|
|
.unwrap(),
|
|
}
|
|
}
|
|
|
|
/// # Safety
|
|
/// `entity_location` must be within bounds of an archetype that exists.
|
|
unsafe fn get_component_with_type(
|
|
world: &World,
|
|
type_id: TypeId,
|
|
entity: Entity,
|
|
location: EntityLocation,
|
|
) -> Option<*mut u8> {
|
|
let component_id = world.components.get_id(type_id)?;
|
|
get_component(world, component_id, entity, location)
|
|
}
|
|
|
|
/// # Safety
|
|
/// `entity_location` must be within bounds of an archetype that exists.
|
|
pub(crate) unsafe fn get_component_and_flags_with_type(
|
|
world: &World,
|
|
type_id: TypeId,
|
|
entity: Entity,
|
|
location: EntityLocation,
|
|
) -> Option<(*mut u8, *mut ComponentFlags)> {
|
|
let component_id = world.components.get_id(type_id)?;
|
|
get_component_and_flags(world, component_id, entity, location)
|
|
}
|
|
|
|
/// # Safety
|
|
/// `entity_location` must be within bounds of an archetype that exists.
|
|
unsafe fn contains_component_with_type(
|
|
world: &World,
|
|
type_id: TypeId,
|
|
location: EntityLocation,
|
|
) -> bool {
|
|
if let Some(component_id) = world.components.get_id(type_id) {
|
|
contains_component_with_id(world, component_id, location)
|
|
} else {
|
|
false
|
|
}
|
|
}
|
|
|
|
/// # Safety
|
|
/// `entity_location` must be within bounds of an archetype that exists.
|
|
unsafe fn contains_component_with_id(
|
|
world: &World,
|
|
component_id: ComponentId,
|
|
location: EntityLocation,
|
|
) -> bool {
|
|
world
|
|
.archetypes
|
|
.get_unchecked(location.archetype_id)
|
|
.contains(component_id)
|
|
}
|
|
|
|
/// Adds a bundle to the given archetype and returns the resulting archetype. This could be the same [ArchetypeId],
|
|
/// in the event that adding the given bundle does not result in an Archetype change. Results are cached in the
|
|
/// Archetype Graph to avoid redundant work.
|
|
/// # Safety
|
|
/// `archetype_id` must exist and components in `bundle_info` must exist
|
|
pub(crate) unsafe fn add_bundle_to_archetype(
|
|
archetypes: &mut Archetypes,
|
|
storages: &mut Storages,
|
|
components: &mut Components,
|
|
archetype_id: ArchetypeId,
|
|
bundle_info: &BundleInfo,
|
|
) -> ArchetypeId {
|
|
if let Some(archetype_id) = archetypes
|
|
.get_unchecked(archetype_id)
|
|
.edges()
|
|
.get_add_bundle(bundle_info.id)
|
|
{
|
|
return archetype_id;
|
|
}
|
|
let mut new_table_components = Vec::new();
|
|
let mut new_sparse_set_components = Vec::new();
|
|
let mut tracking_flags = Vec::with_capacity(bundle_info.component_ids.len());
|
|
|
|
let current_archetype = archetypes.get_unchecked_mut(archetype_id);
|
|
for component_id in bundle_info.component_ids.iter().cloned() {
|
|
if current_archetype.contains(component_id) {
|
|
tracking_flags.push(ComponentFlags::MUTATED);
|
|
} else {
|
|
tracking_flags.push(ComponentFlags::ADDED);
|
|
let component_info = components.get_info_unchecked(component_id);
|
|
match component_info.storage_type() {
|
|
StorageType::Table => new_table_components.push(component_id),
|
|
StorageType::SparseSet => {
|
|
storages.sparse_sets.get_or_insert(component_info);
|
|
new_sparse_set_components.push(component_id)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if new_table_components.is_empty() && new_sparse_set_components.is_empty() {
|
|
let edges = current_archetype.edges_mut();
|
|
// the archetype does not change when we add this bundle
|
|
edges.set_add_bundle(bundle_info.id, archetype_id);
|
|
edges.set_from_bundle(bundle_info.id, archetype_id, tracking_flags);
|
|
archetype_id
|
|
} else {
|
|
let table_id;
|
|
let table_components;
|
|
let sparse_set_components;
|
|
// the archetype changes when we add this bundle. prepare the new archetype and storages
|
|
{
|
|
let current_archetype = archetypes.get_unchecked_mut(archetype_id);
|
|
table_components = if new_table_components.is_empty() {
|
|
// if there are no new table components, we can keep using this table
|
|
table_id = current_archetype.table_id();
|
|
current_archetype.table_components().to_vec()
|
|
} else {
|
|
new_table_components.extend(current_archetype.table_components());
|
|
// sort to ignore order while hashing
|
|
new_table_components.sort();
|
|
// SAFE: all component ids in `new_table_components` exist
|
|
table_id = storages
|
|
.tables
|
|
.get_id_or_insert(&new_table_components, components);
|
|
|
|
new_table_components
|
|
};
|
|
|
|
sparse_set_components = if new_sparse_set_components.is_empty() {
|
|
current_archetype.sparse_set_components().to_vec()
|
|
} else {
|
|
new_sparse_set_components.extend(current_archetype.sparse_set_components());
|
|
// sort to ignore order while hashing
|
|
new_sparse_set_components.sort();
|
|
new_sparse_set_components
|
|
};
|
|
};
|
|
let new_archetype_id =
|
|
archetypes.get_id_or_insert(table_id, table_components, sparse_set_components);
|
|
// add an edge from the old archetype to the new archetype
|
|
archetypes
|
|
.get_unchecked_mut(archetype_id)
|
|
.edges_mut()
|
|
.set_add_bundle(bundle_info.id, new_archetype_id);
|
|
// add a "from bundle" edge from the new archetype to the old archetype
|
|
archetypes
|
|
.get_unchecked_mut(new_archetype_id)
|
|
.edges_mut()
|
|
.set_from_bundle(bundle_info.id, new_archetype_id, tracking_flags);
|
|
new_archetype_id
|
|
}
|
|
}
|
|
|
|
/// Removes a bundle from the given archetype and returns the resulting archetype (or None if the removal was invalid).
|
|
/// in the event that adding the given bundle does not result in an Archetype change. Results are cached in the
|
|
/// Archetype Graph to avoid redundant work.
|
|
/// if `intersection` is false, attempting to remove a bundle with components _not_ contained in the current archetype will fail,
|
|
/// returning None. if `intersection` is true, components in the bundle but not in the current archetype will be ignored
|
|
/// # Safety
|
|
/// `archetype_id` must exist and components in `bundle_info` must exist
|
|
unsafe fn remove_bundle_from_archetype(
|
|
archetypes: &mut Archetypes,
|
|
storages: &mut Storages,
|
|
components: &mut Components,
|
|
archetype_id: ArchetypeId,
|
|
bundle_info: &BundleInfo,
|
|
intersection: bool,
|
|
) -> Option<ArchetypeId> {
|
|
// check the archetype graph to see if the Bundle has been removed from this archetype in the past
|
|
let remove_bundle_result = {
|
|
// SAFE: entity location is valid and therefore the archetype exists
|
|
let current_archetype = archetypes.get_unchecked_mut(archetype_id);
|
|
if intersection {
|
|
current_archetype
|
|
.edges()
|
|
.get_remove_bundle_intersection(bundle_info.id)
|
|
} else {
|
|
current_archetype.edges().get_remove_bundle(bundle_info.id)
|
|
}
|
|
};
|
|
let result = if let Some(result) = remove_bundle_result {
|
|
// this Bundle removal result is cached. just return that!
|
|
result
|
|
} else {
|
|
let mut next_table_components;
|
|
let mut next_sparse_set_components;
|
|
let next_table_id;
|
|
{
|
|
// SAFE: entity location is valid and therefore the archetype exists
|
|
let current_archetype = archetypes.get_unchecked_mut(archetype_id);
|
|
let mut removed_table_components = Vec::new();
|
|
let mut removed_sparse_set_components = Vec::new();
|
|
for component_id in bundle_info.component_ids.iter().cloned() {
|
|
if current_archetype.contains(component_id) {
|
|
// SAFE: bundle components were already initialized by bundles.get_info
|
|
let component_info = components.get_info_unchecked(component_id);
|
|
match component_info.storage_type() {
|
|
StorageType::Table => removed_table_components.push(component_id),
|
|
StorageType::SparseSet => removed_sparse_set_components.push(component_id),
|
|
}
|
|
} else if !intersection {
|
|
// a component in the bundle was not present in the entity's archetype, so this removal is invalid
|
|
// cache the result in the archetype graph
|
|
current_archetype
|
|
.edges_mut()
|
|
.set_remove_bundle(bundle_info.id, None);
|
|
return None;
|
|
}
|
|
}
|
|
|
|
// sort removed components so we can do an efficient "sorted remove". archetype components are already sorted
|
|
removed_table_components.sort();
|
|
removed_sparse_set_components.sort();
|
|
next_table_components = current_archetype.table_components().to_vec();
|
|
next_sparse_set_components = current_archetype.sparse_set_components().to_vec();
|
|
sorted_remove(&mut next_table_components, &removed_table_components);
|
|
sorted_remove(
|
|
&mut next_sparse_set_components,
|
|
&removed_sparse_set_components,
|
|
);
|
|
|
|
next_table_id = if removed_table_components.is_empty() {
|
|
current_archetype.table_id()
|
|
} else {
|
|
// SAFE: all components in next_table_components exist
|
|
storages
|
|
.tables
|
|
.get_id_or_insert(&next_table_components, components)
|
|
};
|
|
}
|
|
|
|
let new_archetype_id = archetypes.get_id_or_insert(
|
|
next_table_id,
|
|
next_table_components,
|
|
next_sparse_set_components,
|
|
);
|
|
Some(new_archetype_id)
|
|
};
|
|
// SAFE: entity location is valid and therefore the archetype exists
|
|
let current_archetype = archetypes.get_unchecked_mut(archetype_id);
|
|
// cache the result in an edge
|
|
if intersection {
|
|
current_archetype
|
|
.edges_mut()
|
|
.set_remove_bundle_intersection(bundle_info.id, result);
|
|
} else {
|
|
current_archetype
|
|
.edges_mut()
|
|
.set_remove_bundle(bundle_info.id, result);
|
|
}
|
|
result
|
|
}
|
|
|
|
fn sorted_remove<T: Eq + Ord + Copy>(source: &mut Vec<T>, remove: &[T]) {
|
|
let mut remove_index = 0;
|
|
source.retain(|value| {
|
|
while remove_index < remove.len() && *value > remove[remove_index] {
|
|
remove_index += 1;
|
|
}
|
|
|
|
if remove_index < remove.len() {
|
|
*value != remove[remove_index]
|
|
} else {
|
|
true
|
|
}
|
|
})
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
#[test]
|
|
fn sorted_remove() {
|
|
let mut a = vec![1, 2, 3, 4, 5, 6, 7];
|
|
let b = vec![1, 2, 3, 5, 7];
|
|
super::sorted_remove(&mut a, &b);
|
|
|
|
assert_eq!(a, vec![4, 6]);
|
|
|
|
let mut a = vec![1];
|
|
let b = vec![1];
|
|
super::sorted_remove(&mut a, &b);
|
|
|
|
assert_eq!(a, vec![]);
|
|
|
|
let mut a = vec![1];
|
|
let b = vec![2];
|
|
super::sorted_remove(&mut a, &b);
|
|
|
|
assert_eq!(a, vec![1]);
|
|
}
|
|
}
|