Add benchmarks for raw entity allocation

benches/benches/bevy_ecs/world/entity_allocator.rs

@@ -0,0 +1,131 @@
+use core::hint::black_box;
+
+use bevy_ecs::prelude::*;
+use criterion::{BatchSize, Criterion};
+
+pub fn entity_allocator_benches(criterion: &mut Criterion) {
+    const ENTITY_COUNTS: [u32; 3] = [1, 100, 10_000];
+
+    let mut group = criterion.benchmark_group("entity_allocator_allocate_fresh");
+    group.warm_up_time(core::time::Duration::from_millis(500));
+    group.measurement_time(core::time::Duration::from_secs(4));
+
+    for entity_count in ENTITY_COUNTS {
+        group.bench_function(format!("{entity_count}_entities"), |bencher| {
+            bencher.iter_batched_ref(
+                World::default,
+                |world| {
+                    for _ in 0..entity_count {
+                        let entity = world.entity_allocator().alloc();
+                        black_box(entity);
+                    }
+                },
+                BatchSize::SmallInput,
+            );
+        });
+    }
+
+    group.finish();
+
+    let mut group = criterion.benchmark_group("entity_allocator_allocate_fresh_bulk");
+    group.warm_up_time(core::time::Duration::from_millis(500));
+    group.measurement_time(core::time::Duration::from_secs(4));
+
+    for entity_count in ENTITY_COUNTS {
+        group.bench_function(format!("{entity_count}_entities"), |bencher| {
+            bencher.iter_batched_ref(
+                World::default,
+                |world| {
+                    for entity in world.entity_allocator().alloc_many(entity_count) {
+                        black_box(entity);
+                    }
+                },
+                BatchSize::SmallInput,
+            );
+        });
+    }
+
+    group.finish();
+
+    let mut group = criterion.benchmark_group("entity_allocator_free");
+    group.warm_up_time(core::time::Duration::from_millis(500));
+    group.measurement_time(core::time::Duration::from_secs(4));
+
+    for entity_count in ENTITY_COUNTS {
+        group.bench_function(format!("{entity_count}_entities"), |bencher| {
+            bencher.iter_batched_ref(
+                || {
+                    let world = World::new();
+                    let entities =
+                        Vec::from_iter(world.entity_allocator().alloc_many(entity_count));
+                    (world, entities)
+                },
+                |(world, entities)| {
+                    entities
+                        .drain(..)
+                        .for_each(|e| world.entity_allocator_mut().free(e));
+                },
+                BatchSize::SmallInput,
+            );
+        });
+    }
+
+    group.finish();
+
+    let mut group = criterion.benchmark_group("entity_allocator_allocate_reused");
+    group.warm_up_time(core::time::Duration::from_millis(500));
+    group.measurement_time(core::time::Duration::from_secs(4));
+
+    for entity_count in ENTITY_COUNTS {
+        group.bench_function(format!("{entity_count}_entities"), |bencher| {
+            bencher.iter_batched_ref(
+                || {
+                    let mut world = World::new();
+                    let mut entities =
+                        Vec::from_iter(world.entity_allocator().alloc_many(entity_count));
+                    entities
+                        .drain(..)
+                        .for_each(|e| world.entity_allocator_mut().free(e));
+                    world
+                },
+                |world| {
+                    for _ in 0..entity_count {
+                        let entity = world.entity_allocator().alloc();
+                        black_box(entity);
+                    }
+                },
+                BatchSize::SmallInput,
+            );
+        });
+    }
+
+    group.finish();
+
+    let mut group = criterion.benchmark_group("entity_allocator_allocate_reused_bulk");
+    group.warm_up_time(core::time::Duration::from_millis(500));
+    group.measurement_time(core::time::Duration::from_secs(4));
+
+    for entity_count in ENTITY_COUNTS {
+        group.bench_function(format!("{entity_count}_entities"), |bencher| {
+            bencher.iter_batched_ref(
+                || {
+                    let mut world = World::new();
+                    let mut entities =
+                        Vec::from_iter(world.entity_allocator().alloc_many(entity_count));
+                    entities
+                        .drain(..)
+                        .for_each(|e| world.entity_allocator_mut().free(e));
+                    world
+                },
+                |world| {
+                    for entity in world.entity_allocator().alloc_many(entity_count) {
+                        black_box(entity);
+                    }
+                },
+                BatchSize::SmallInput,
+            );
+        });
+    }
+
+    group.finish();
+}

benches/benches/bevy_ecs/world/mod.rs

@@ -1,6 +1,7 @@
 mod commands;
 mod despawn;
 mod despawn_recursive;
+mod entity_allocator;
 mod entity_hash;
 mod spawn;
 mod world_get;
@@ -9,6 +10,7 @@ use commands::*;
 use criterion::criterion_group;
 use despawn::*;
 use despawn_recursive::*;
+use entity_allocator::*;
 use entity_hash::*;
 use spawn::*;
 use world_get::*;
@@ -40,4 +42,5 @@ criterion_group!(
     query_get_components_mut_5,
     query_get_components_mut_10,
     entity_set_build_and_lookup,
+    entity_allocator_benches,
 );