bevy/benches/benches/bevy_reflect/struct.rs
2025-06-26 19:38:19 +00:00

874 lines
20 KiB
Rust

use core::{hint::black_box, time::Duration};
use benches::bench;
use bevy_reflect::{DynamicStruct, GetField, PartialReflect, Reflect, Struct};
use criterion::{
criterion_group, measurement::Measurement, AxisScale, BatchSize, BenchmarkGroup, BenchmarkId,
Criterion, PlotConfiguration, Throughput,
};
criterion_group!(
benches,
concrete_struct_apply,
concrete_struct_field,
concrete_struct_type_info,
concrete_struct_to_dynamic_struct,
dynamic_struct_to_dynamic_struct,
dynamic_struct_apply,
dynamic_struct_get_field,
dynamic_struct_insert,
);
const WARM_UP_TIME: Duration = Duration::from_millis(500);
const MEASUREMENT_TIME: Duration = Duration::from_secs(4);
const SIZES: [usize; 4] = [16, 32, 64, 128];
/// Creates a [`BenchmarkGroup`] with common configuration shared by all benchmarks within this
/// module.
fn create_group<'a, M: Measurement>(c: &'a mut Criterion<M>, name: &str) -> BenchmarkGroup<'a, M> {
let mut group = c.benchmark_group(name);
group
.warm_up_time(WARM_UP_TIME)
.measurement_time(MEASUREMENT_TIME)
// Make the plots logarithmic, matching `SIZES`' scale.
.plot_config(PlotConfiguration::default().summary_scale(AxisScale::Logarithmic));
group
}
fn concrete_struct_field(criterion: &mut Criterion) {
let mut group = create_group(criterion, bench!("concrete_struct_field"));
let structs: [Box<dyn Struct>; 4] = [
Box::new(Struct16::default()),
Box::new(Struct32::default()),
Box::new(Struct64::default()),
Box::new(Struct128::default()),
];
for s in structs {
let field_count = s.field_len();
group.bench_with_input(
BenchmarkId::from_parameter(field_count),
&s,
|bencher, s| {
let field_names = (0..field_count)
.map(|i| format!("field_{i}"))
.collect::<Vec<_>>();
bencher.iter(|| {
for name in &field_names {
black_box(s.field(black_box(name)));
}
});
},
);
}
}
fn concrete_struct_apply(criterion: &mut Criterion) {
let mut group = create_group(criterion, bench!("concrete_struct_apply"));
// Use functions that produce trait objects of varying concrete types as the
// input to the benchmark.
let inputs: &[fn() -> (Box<dyn Struct>, Box<dyn PartialReflect>)] = &[
|| (Box::new(Struct16::default()), Box::new(Struct16::default())),
|| (Box::new(Struct32::default()), Box::new(Struct32::default())),
|| (Box::new(Struct64::default()), Box::new(Struct64::default())),
|| {
(
Box::new(Struct128::default()),
Box::new(Struct128::default()),
)
},
];
for input in inputs {
let field_count = input().0.field_len();
group.throughput(Throughput::Elements(field_count as u64));
group.bench_with_input(
BenchmarkId::new("apply_concrete", field_count),
input,
|bencher, input| {
bencher.iter_batched(
input,
|(mut obj, patch)| obj.apply(black_box(patch.as_ref())),
BatchSize::SmallInput,
);
},
);
}
for input in inputs {
let field_count = input().0.field_len();
group.throughput(Throughput::Elements(field_count as u64));
group.bench_with_input(
BenchmarkId::new("apply_dynamic", field_count),
input,
|bencher, input| {
bencher.iter_batched(
|| {
let (obj, _) = input();
let patch = obj.to_dynamic_struct();
(obj, patch)
},
|(mut obj, patch)| obj.apply(black_box(&patch)),
BatchSize::SmallInput,
);
},
);
}
}
fn concrete_struct_type_info(criterion: &mut Criterion) {
let mut group = create_group(criterion, bench!("concrete_struct_type_info"));
let structs: [(Box<dyn Struct>, Box<dyn Struct>); 5] = [
(
Box::new(Struct1::default()),
Box::new(GenericStruct1::<u32>::default()),
),
(
Box::new(Struct16::default()),
Box::new(GenericStruct16::<u32>::default()),
),
(
Box::new(Struct32::default()),
Box::new(GenericStruct32::<u32>::default()),
),
(
Box::new(Struct64::default()),
Box::new(GenericStruct64::<u32>::default()),
),
(
Box::new(Struct128::default()),
Box::new(GenericStruct128::<u32>::default()),
),
];
for (standard, generic) in structs {
let field_count = standard.field_len();
group.bench_with_input(
BenchmarkId::new("NonGeneric", field_count),
&standard,
|bencher, s| {
bencher.iter(|| s.get_represented_type_info());
},
);
group.bench_with_input(
BenchmarkId::new("Generic", field_count),
&generic,
|bencher, s| {
bencher.iter(|| s.get_represented_type_info());
},
);
}
}
fn concrete_struct_to_dynamic_struct(criterion: &mut Criterion) {
let mut group = create_group(criterion, bench!("concrete_struct_to_dynamic_struct"));
let structs: [(Box<dyn Struct>, Box<dyn Struct>); 5] = [
(
Box::new(Struct1::default()),
Box::new(GenericStruct1::<u32>::default()),
),
(
Box::new(Struct16::default()),
Box::new(GenericStruct16::<u32>::default()),
),
(
Box::new(Struct32::default()),
Box::new(GenericStruct32::<u32>::default()),
),
(
Box::new(Struct64::default()),
Box::new(GenericStruct64::<u32>::default()),
),
(
Box::new(Struct128::default()),
Box::new(GenericStruct128::<u32>::default()),
),
];
for (standard, generic) in structs {
let field_count = standard.field_len();
group.bench_with_input(
BenchmarkId::new("NonGeneric", field_count),
&standard,
|bencher, s| {
bencher.iter(|| s.to_dynamic_struct());
},
);
group.bench_with_input(
BenchmarkId::new("Generic", field_count),
&generic,
|bencher, s| {
bencher.iter(|| s.to_dynamic_struct());
},
);
}
}
fn dynamic_struct_to_dynamic_struct(criterion: &mut Criterion) {
let mut group = create_group(criterion, bench!("dynamic_struct_to_dynamic_struct"));
let structs: [Box<dyn Struct>; 5] = [
Box::new(Struct1::default().to_dynamic_struct()),
Box::new(Struct16::default().to_dynamic_struct()),
Box::new(Struct32::default().to_dynamic_struct()),
Box::new(Struct64::default().to_dynamic_struct()),
Box::new(Struct128::default().to_dynamic_struct()),
];
for s in structs {
let field_count = s.field_len();
group.bench_with_input(
BenchmarkId::from_parameter(field_count),
&s,
|bencher, s| {
bencher.iter(|| s.to_dynamic_struct());
},
);
}
}
fn dynamic_struct_apply(criterion: &mut Criterion) {
let mut group = create_group(criterion, bench!("dynamic_struct_apply"));
let patches: &[(fn() -> Box<dyn PartialReflect>, usize)] = &[
(|| Box::new(Struct16::default()), 16),
(|| Box::new(Struct32::default()), 32),
(|| Box::new(Struct64::default()), 64),
(|| Box::new(Struct128::default()), 128),
];
for (patch, field_count) in patches {
let field_count = *field_count;
group.throughput(Throughput::Elements(field_count as u64));
let mut base = DynamicStruct::default();
for i in 0..field_count {
let field_name = format!("field_{i}");
base.insert(&field_name, 1u32);
}
group.bench_with_input(
BenchmarkId::new("apply_concrete", field_count),
&patch,
|bencher, patch| {
bencher.iter_batched(
|| (base.to_dynamic_struct(), patch()),
|(mut base, patch)| base.apply(black_box(&*patch)),
BatchSize::SmallInput,
);
},
);
}
for field_count in SIZES {
group.throughput(Throughput::Elements(field_count as u64));
group.bench_with_input(
BenchmarkId::new("apply_dynamic", field_count),
&field_count,
|bencher, &field_count| {
let mut base = DynamicStruct::default();
let mut patch = DynamicStruct::default();
for i in 0..field_count {
let field_name = format!("field_{i}");
base.insert(&field_name, 0u32);
patch.insert(&field_name, 1u32);
}
bencher.iter_batched(
|| base.to_dynamic_struct(),
|mut base| base.apply(black_box(&patch)),
BatchSize::SmallInput,
);
},
);
}
}
fn dynamic_struct_insert(criterion: &mut Criterion) {
let mut group = create_group(criterion, bench!("dynamic_struct_insert"));
for field_count in SIZES {
group.throughput(Throughput::Elements(field_count as u64));
group.bench_with_input(
BenchmarkId::from_parameter(field_count),
&field_count,
|bencher, field_count| {
let mut s = DynamicStruct::default();
for i in 0..*field_count {
let field_name = format!("field_{i}");
s.insert(&field_name, ());
}
let field = format!("field_{field_count}");
bencher.iter_batched(
|| s.to_dynamic_struct(),
|mut s| {
s.insert(black_box(&field), ());
},
BatchSize::SmallInput,
);
},
);
}
group.finish();
}
fn dynamic_struct_get_field(criterion: &mut Criterion) {
let mut group = create_group(criterion, bench!("dynamic_struct_get_field"));
for field_count in SIZES {
group.throughput(Throughput::Elements(field_count as u64));
group.bench_with_input(
BenchmarkId::from_parameter(field_count),
&field_count,
|bencher, field_count| {
let mut s = DynamicStruct::default();
for i in 0..*field_count {
let field_name = format!("field_{i}");
s.insert(&field_name, ());
}
let field = black_box("field_63");
bencher.iter(|| s.get_field::<()>(field));
},
);
}
}
#[derive(Clone, Default, Reflect)]
struct Struct1 {
field_0: u32,
}
#[derive(Clone, Default, Reflect)]
struct Struct16 {
field_0: u32,
field_1: u32,
field_2: u32,
field_3: u32,
field_4: u32,
field_5: u32,
field_6: u32,
field_7: u32,
field_8: u32,
field_9: u32,
field_10: u32,
field_11: u32,
field_12: u32,
field_13: u32,
field_14: u32,
field_15: u32,
}
#[derive(Clone, Default, Reflect)]
struct Struct32 {
field_0: u32,
field_1: u32,
field_2: u32,
field_3: u32,
field_4: u32,
field_5: u32,
field_6: u32,
field_7: u32,
field_8: u32,
field_9: u32,
field_10: u32,
field_11: u32,
field_12: u32,
field_13: u32,
field_14: u32,
field_15: u32,
field_16: u32,
field_17: u32,
field_18: u32,
field_19: u32,
field_20: u32,
field_21: u32,
field_22: u32,
field_23: u32,
field_24: u32,
field_25: u32,
field_26: u32,
field_27: u32,
field_28: u32,
field_29: u32,
field_30: u32,
field_31: u32,
}
#[derive(Clone, Default, Reflect)]
struct Struct64 {
field_0: u32,
field_1: u32,
field_2: u32,
field_3: u32,
field_4: u32,
field_5: u32,
field_6: u32,
field_7: u32,
field_8: u32,
field_9: u32,
field_10: u32,
field_11: u32,
field_12: u32,
field_13: u32,
field_14: u32,
field_15: u32,
field_16: u32,
field_17: u32,
field_18: u32,
field_19: u32,
field_20: u32,
field_21: u32,
field_22: u32,
field_23: u32,
field_24: u32,
field_25: u32,
field_26: u32,
field_27: u32,
field_28: u32,
field_29: u32,
field_30: u32,
field_31: u32,
field_32: u32,
field_33: u32,
field_34: u32,
field_35: u32,
field_36: u32,
field_37: u32,
field_38: u32,
field_39: u32,
field_40: u32,
field_41: u32,
field_42: u32,
field_43: u32,
field_44: u32,
field_45: u32,
field_46: u32,
field_47: u32,
field_48: u32,
field_49: u32,
field_50: u32,
field_51: u32,
field_52: u32,
field_53: u32,
field_54: u32,
field_55: u32,
field_56: u32,
field_57: u32,
field_58: u32,
field_59: u32,
field_60: u32,
field_61: u32,
field_62: u32,
field_63: u32,
}
#[derive(Clone, Default, Reflect)]
struct Struct128 {
field_0: u32,
field_1: u32,
field_2: u32,
field_3: u32,
field_4: u32,
field_5: u32,
field_6: u32,
field_7: u32,
field_8: u32,
field_9: u32,
field_10: u32,
field_11: u32,
field_12: u32,
field_13: u32,
field_14: u32,
field_15: u32,
field_16: u32,
field_17: u32,
field_18: u32,
field_19: u32,
field_20: u32,
field_21: u32,
field_22: u32,
field_23: u32,
field_24: u32,
field_25: u32,
field_26: u32,
field_27: u32,
field_28: u32,
field_29: u32,
field_30: u32,
field_31: u32,
field_32: u32,
field_33: u32,
field_34: u32,
field_35: u32,
field_36: u32,
field_37: u32,
field_38: u32,
field_39: u32,
field_40: u32,
field_41: u32,
field_42: u32,
field_43: u32,
field_44: u32,
field_45: u32,
field_46: u32,
field_47: u32,
field_48: u32,
field_49: u32,
field_50: u32,
field_51: u32,
field_52: u32,
field_53: u32,
field_54: u32,
field_55: u32,
field_56: u32,
field_57: u32,
field_58: u32,
field_59: u32,
field_60: u32,
field_61: u32,
field_62: u32,
field_63: u32,
field_64: u32,
field_65: u32,
field_66: u32,
field_67: u32,
field_68: u32,
field_69: u32,
field_70: u32,
field_71: u32,
field_72: u32,
field_73: u32,
field_74: u32,
field_75: u32,
field_76: u32,
field_77: u32,
field_78: u32,
field_79: u32,
field_80: u32,
field_81: u32,
field_82: u32,
field_83: u32,
field_84: u32,
field_85: u32,
field_86: u32,
field_87: u32,
field_88: u32,
field_89: u32,
field_90: u32,
field_91: u32,
field_92: u32,
field_93: u32,
field_94: u32,
field_95: u32,
field_96: u32,
field_97: u32,
field_98: u32,
field_99: u32,
field_100: u32,
field_101: u32,
field_102: u32,
field_103: u32,
field_104: u32,
field_105: u32,
field_106: u32,
field_107: u32,
field_108: u32,
field_109: u32,
field_110: u32,
field_111: u32,
field_112: u32,
field_113: u32,
field_114: u32,
field_115: u32,
field_116: u32,
field_117: u32,
field_118: u32,
field_119: u32,
field_120: u32,
field_121: u32,
field_122: u32,
field_123: u32,
field_124: u32,
field_125: u32,
field_126: u32,
field_127: u32,
}
#[derive(Clone, Default, Reflect)]
struct GenericStruct1<T: Reflect + Default> {
field_0: T,
}
#[derive(Clone, Default, Reflect)]
struct GenericStruct16<T: Reflect + Default> {
field_0: T,
field_1: T,
field_2: T,
field_3: T,
field_4: T,
field_5: T,
field_6: T,
field_7: T,
field_8: T,
field_9: T,
field_10: T,
field_11: T,
field_12: T,
field_13: T,
field_14: T,
field_15: T,
}
#[derive(Clone, Default, Reflect)]
struct GenericStruct32<T: Reflect + Default> {
field_0: T,
field_1: T,
field_2: T,
field_3: T,
field_4: T,
field_5: T,
field_6: T,
field_7: T,
field_8: T,
field_9: T,
field_10: T,
field_11: T,
field_12: T,
field_13: T,
field_14: T,
field_15: T,
field_16: T,
field_17: T,
field_18: T,
field_19: T,
field_20: T,
field_21: T,
field_22: T,
field_23: T,
field_24: T,
field_25: T,
field_26: T,
field_27: T,
field_28: T,
field_29: T,
field_30: T,
field_31: T,
}
#[derive(Clone, Default, Reflect)]
struct GenericStruct64<T: Reflect + Default> {
field_0: T,
field_1: T,
field_2: T,
field_3: T,
field_4: T,
field_5: T,
field_6: T,
field_7: T,
field_8: T,
field_9: T,
field_10: T,
field_11: T,
field_12: T,
field_13: T,
field_14: T,
field_15: T,
field_16: T,
field_17: T,
field_18: T,
field_19: T,
field_20: T,
field_21: T,
field_22: T,
field_23: T,
field_24: T,
field_25: T,
field_26: T,
field_27: T,
field_28: T,
field_29: T,
field_30: T,
field_31: T,
field_32: T,
field_33: T,
field_34: T,
field_35: T,
field_36: T,
field_37: T,
field_38: T,
field_39: T,
field_40: T,
field_41: T,
field_42: T,
field_43: T,
field_44: T,
field_45: T,
field_46: T,
field_47: T,
field_48: T,
field_49: T,
field_50: T,
field_51: T,
field_52: T,
field_53: T,
field_54: T,
field_55: T,
field_56: T,
field_57: T,
field_58: T,
field_59: T,
field_60: T,
field_61: T,
field_62: T,
field_63: T,
}
#[derive(Clone, Default, Reflect)]
struct GenericStruct128<T: Reflect + Default> {
field_0: T,
field_1: T,
field_2: T,
field_3: T,
field_4: T,
field_5: T,
field_6: T,
field_7: T,
field_8: T,
field_9: T,
field_10: T,
field_11: T,
field_12: T,
field_13: T,
field_14: T,
field_15: T,
field_16: T,
field_17: T,
field_18: T,
field_19: T,
field_20: T,
field_21: T,
field_22: T,
field_23: T,
field_24: T,
field_25: T,
field_26: T,
field_27: T,
field_28: T,
field_29: T,
field_30: T,
field_31: T,
field_32: T,
field_33: T,
field_34: T,
field_35: T,
field_36: T,
field_37: T,
field_38: T,
field_39: T,
field_40: T,
field_41: T,
field_42: T,
field_43: T,
field_44: T,
field_45: T,
field_46: T,
field_47: T,
field_48: T,
field_49: T,
field_50: T,
field_51: T,
field_52: T,
field_53: T,
field_54: T,
field_55: T,
field_56: T,
field_57: T,
field_58: T,
field_59: T,
field_60: T,
field_61: T,
field_62: T,
field_63: T,
field_64: T,
field_65: T,
field_66: T,
field_67: T,
field_68: T,
field_69: T,
field_70: T,
field_71: T,
field_72: T,
field_73: T,
field_74: T,
field_75: T,
field_76: T,
field_77: T,
field_78: T,
field_79: T,
field_80: T,
field_81: T,
field_82: T,
field_83: T,
field_84: T,
field_85: T,
field_86: T,
field_87: T,
field_88: T,
field_89: T,
field_90: T,
field_91: T,
field_92: T,
field_93: T,
field_94: T,
field_95: T,
field_96: T,
field_97: T,
field_98: T,
field_99: T,
field_100: T,
field_101: T,
field_102: T,
field_103: T,
field_104: T,
field_105: T,
field_106: T,
field_107: T,
field_108: T,
field_109: T,
field_110: T,
field_111: T,
field_112: T,
field_113: T,
field_114: T,
field_115: T,
field_116: T,
field_117: T,
field_118: T,
field_119: T,
field_120: T,
field_121: T,
field_122: T,
field_123: T,
field_124: T,
field_125: T,
field_126: T,
field_127: T,
}