docs/bevy/examples/stress_tests/many_materials.rs 3.2 K · 106 lines · raw · history

1 //! Benchmark to test rendering many animated materials
2 use argh::FromArgs;
3 use bevy::{
4 diagnostic::{FrameTimeDiagnosticsPlugin, LogDiagnosticsPlugin},
5 prelude::*,
6 window::{PresentMode, WindowResolution},
7 winit::WinitSettings,
8 };
9 use std::f32::consts::PI;
10
11 #[derive(FromArgs, Resource)]
12 /// Command-line arguments for the `many_materials` stress test.
13 struct Args {
14 /// the size of the grid of materials to render (n x n)
15 #[argh(option, short = 'n', default = "10")]
16 grid_size: usize,
17 }
18
19 fn main() {
20 // `from_env` panics on the web
21 #[cfg(not(target_arch = "wasm32"))]
22 let args: Args = argh::from_env();
23 #[cfg(target_arch = "wasm32")]
24 let args = Args::from_args(&[], &[]).unwrap();
25
26 App::new()
27 .add_plugins((
28 DefaultPlugins.set(WindowPlugin {
29 primary_window: Some(Window {
30 resolution: WindowResolution::new(1920, 1080).with_scale_factor_override(1.0),
31 title: "many_materials".into(),
32 present_mode: PresentMode::AutoNoVsync,
33 ..default()
34 }),
35 ..default()
36 }),
37 FrameTimeDiagnosticsPlugin::default(),
38 LogDiagnosticsPlugin::default(),
39 ))
40 .insert_resource(WinitSettings::continuous())
41 .insert_resource(args)
42 .add_systems(Startup, setup)
43 .add_systems(Update, animate_materials)
44 .run();
45 }
46
47 fn setup(
48 mut commands: Commands,
49 args: Res<Args>,
50 mesh_assets: ResMut<Assets<Mesh>>,
51 material_assets: ResMut<Assets<StandardMaterial>>,
52 ) {
53 warn!(include_str!("warning_string.txt"));
54
55 let args = args.into_inner();
56 let material_assets = material_assets.into_inner();
57 let mesh_assets = mesh_assets.into_inner();
58 let n = args.grid_size;
59
60 // Camera
61 let w = n as f32;
62 commands.spawn((
63 Camera3d::default(),
64 Transform::from_xyz(w * 1.25, w + 1.0, w * 1.25)
65 .looking_at(Vec3::new(0.0, (w * -1.1) + 1.0, 0.0), Vec3::Y),
66 ));
67
68 // Light
69 commands.spawn((
70 Transform::from_rotation(Quat::from_euler(EulerRot::ZYX, 0.0, 1.0, -PI / 4.)),
71 DirectionalLight {
72 illuminance: 3000.0,
73 shadow_maps_enabled: true,
74 ..default()
75 },
76 ));
77
78 // Cubes
79 let mesh_handle = mesh_assets.add(Cuboid::from_size(Vec3::ONE));
80 for x in 0..n {
81 for z in 0..n {
82 commands.spawn((
83 Mesh3d(mesh_handle.clone()),
84 MeshMaterial3d(material_assets.add(Color::WHITE)),
85 Transform::from_translation(Vec3::new(x as f32, 0.0, z as f32)),
86 ));
87 }
88 }
89 }
90
91 fn animate_materials(
92 material_handles: Query<&MeshMaterial3d<StandardMaterial>>,
93 time: Res<Time>,
94 mut materials: ResMut<Assets<StandardMaterial>>,
95 ) {
96 for (i, material_handle) in material_handles.iter().enumerate() {
97 if let Some(mut material) = materials.get_mut(material_handle) {
98 let color = Color::hsl(
99 ((i as f32 * 2.345 + time.elapsed_secs()) * 100.0) % 360.0,
100 1.0,
101 0.5,
102 );
103 material.base_color = color;
104 }
105 }
106 }