use bevy::{prelude::*, tasks::prelude::*};
use rand::random;
struct Velocity(Vec2);
fn spawn_system(
commands: &mut Commands,
asset_server: Res<AssetServer>,
mut materials: ResMut<Assets<ColorMaterial>>,
) {
commands.spawn(Camera2dBundle::default());
let texture_handle = asset_server.load("branding/icon.png");
let material = materials.add(texture_handle.into());
for _ in 0..128 {
commands
.spawn(SpriteBundle {
material: material.clone(),
transform: Transform::from_scale(Vec3::splat(0.1)),
..Default::default()
})
.with(Velocity(
20.0 * Vec2::new(random::<f32>() - 0.5, random::<f32>() - 0.5),
));
}
}
fn move_system(pool: Res<ComputeTaskPool>, mut sprites: Query<(&mut Transform, &Velocity)>) {
sprites
.par_iter_mut(32)
.for_each(&pool, |(mut transform, velocity)| {
transform.translation += velocity.0.extend(0.0);
});
}
fn bounce_system(
pool: Res<ComputeTaskPool>,
windows: Res<Windows>,
mut sprites: Query<(&Transform, &mut Velocity)>,
) {
let window = windows.get_primary().expect("No primary window.");
let width = window.width();
let height = window.height();
let left = width / -2.0;
let right = width / 2.0;
let bottom = height / -2.0;
let top = height / 2.0;
sprites
.par_iter_mut(32)
.filter(|(transform, _)| {
!(left < transform.translation.x
&& transform.translation.x < right
&& bottom < transform.translation.y
&& transform.translation.y < top)
})
.for_each(&pool, |(_, mut v)| {
v.0 = -v.0;
});
}
fn main() {
App::build()
.add_plugins(DefaultPlugins)
.add_startup_system(spawn_system.system())
.add_system(move_system.system())
.add_system(bounce_system.system())
.run();
}