mirror of
https://github.com/bevyengine/bevy
synced 2024-11-29 08:00:20 +00:00
efda7f3f9c
Takes the first two commits from #15375 and adds suggestions from this comment: https://github.com/bevyengine/bevy/pull/15375#issuecomment-2366968300 See #15375 for more reasoning/motivation. ## Rebasing (rerunning) ```rust git switch simpler-lint-fixes git reset --hard main cargo fmt --all -- --unstable-features --config normalize_comments=true,imports_granularity=Crate cargo fmt --all git add --update git commit --message "rustfmt" cargo clippy --workspace --all-targets --all-features --fix cargo fmt --all -- --unstable-features --config normalize_comments=true,imports_granularity=Crate cargo fmt --all git add --update git commit --message "clippy" git cherry-pick e6c0b94f6795222310fb812fa5c4512661fc7887 ```
101 lines
2.6 KiB
Rust
101 lines
2.6 KiB
Rust
//! An example showing how to save screenshots to disk
|
|
|
|
use bevy::{
|
|
prelude::*,
|
|
render::view::{
|
|
cursor::CursorIcon,
|
|
screenshot::{save_to_disk, Capturing, Screenshot},
|
|
},
|
|
window::SystemCursorIcon,
|
|
};
|
|
|
|
fn main() {
|
|
App::new()
|
|
.add_plugins(DefaultPlugins)
|
|
.add_systems(Startup, setup)
|
|
.add_systems(Update, (screenshot_on_spacebar, screenshot_saving))
|
|
.run();
|
|
}
|
|
|
|
fn screenshot_on_spacebar(
|
|
mut commands: Commands,
|
|
input: Res<ButtonInput<KeyCode>>,
|
|
mut counter: Local<u32>,
|
|
) {
|
|
if input.just_pressed(KeyCode::Space) {
|
|
let path = format!("./screenshot-{}.png", *counter);
|
|
*counter += 1;
|
|
commands
|
|
.spawn(Screenshot::primary_window())
|
|
.observe(save_to_disk(path));
|
|
}
|
|
}
|
|
|
|
fn screenshot_saving(
|
|
mut commands: Commands,
|
|
screenshot_saving: Query<Entity, With<Capturing>>,
|
|
windows: Query<Entity, With<Window>>,
|
|
) {
|
|
let Ok(window) = windows.get_single() else {
|
|
return;
|
|
};
|
|
match screenshot_saving.iter().count() {
|
|
0 => {
|
|
commands.entity(window).remove::<CursorIcon>();
|
|
}
|
|
x if x > 0 => {
|
|
commands
|
|
.entity(window)
|
|
.insert(CursorIcon::from(SystemCursorIcon::Progress));
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
/// set up a simple 3D scene
|
|
fn setup(
|
|
mut commands: Commands,
|
|
mut meshes: ResMut<Assets<Mesh>>,
|
|
mut materials: ResMut<Assets<StandardMaterial>>,
|
|
) {
|
|
// plane
|
|
commands.spawn(PbrBundle {
|
|
mesh: meshes.add(Plane3d::default().mesh().size(5.0, 5.0)),
|
|
material: materials.add(Color::srgb(0.3, 0.5, 0.3)),
|
|
..default()
|
|
});
|
|
// cube
|
|
commands.spawn(PbrBundle {
|
|
mesh: meshes.add(Cuboid::default()),
|
|
material: materials.add(Color::srgb(0.8, 0.7, 0.6)),
|
|
transform: Transform::from_xyz(0.0, 0.5, 0.0),
|
|
..default()
|
|
});
|
|
// light
|
|
commands.spawn(PointLightBundle {
|
|
point_light: PointLight {
|
|
shadows_enabled: true,
|
|
..default()
|
|
},
|
|
transform: Transform::from_xyz(4.0, 8.0, 4.0),
|
|
..default()
|
|
});
|
|
// camera
|
|
commands.spawn(Camera3dBundle {
|
|
transform: Transform::from_xyz(-2.0, 2.5, 5.0).looking_at(Vec3::ZERO, Vec3::Y),
|
|
..default()
|
|
});
|
|
|
|
commands.spawn(
|
|
TextBundle::from_section(
|
|
"Press <spacebar> to save a screenshot to disk",
|
|
TextStyle::default(),
|
|
)
|
|
.with_style(Style {
|
|
position_type: PositionType::Absolute,
|
|
top: Val::Px(12.0),
|
|
left: Val::Px(12.0),
|
|
..default()
|
|
}),
|
|
);
|
|
}
|