tidy

src/lib.rs

@@ -8,7 +8,7 @@ use rstar::{primitives::GeomWithData, RTree};
 mod systems;
 
 pub use systems::{
-    init_naybs, rotate_camera, setup_channels, spawn_background, turkey_time, update_buddies,
+    handle_inputs, init_naybs, setup_channels, spawn_background, turkey_time, update_buddies,
     update_gizmos, update_naybs, update_pos, update_vel,
 };
 
@@ -59,8 +59,8 @@ struct IndexMessage {
 
 struct RawMessage(Vec<EntityPos>);
 
-#[derive(Default, Resource, Deref, DerefMut)]
+#[derive(Resource, Deref, DerefMut)]
-pub struct BackgroundHandle(Option<bevy::tasks::Task<()>>);
+pub struct BackgroundHandle(bevy::tasks::Task<()>);
 
 #[derive(Component)]
 pub struct Toid {
@@ -70,3 +70,6 @@ pub struct Toid {
 
 #[derive(Debug, Default, Clone, Copy, Deref, DerefMut, Resource)]
 pub struct LookAt(Vec3);
+
+#[derive(Resource, Debug, Deref, DerefMut, Clone, Default)]
+pub struct Paused(bool);

src/main.rs

@@ -2,16 +2,18 @@ use bevy::{
     diagnostic::{FrameTimeDiagnosticsPlugin, LogDiagnosticsPlugin},
     input::keyboard::Key,
     prelude::{
-        children, default, px, App, AssetServer, Assets, ButtonInput, Camera3d, ClearColor, Color,
+        children, default, px, warn, App, AssetServer, Assets, ButtonInput, Camera3d, ClearColor,
-        Commands, DefaultPlugins, Entity, FontSize, GlobalAmbientLight, IntoScheduleConfigs, Mesh,
+        Color, Commands, DefaultPlugins, Entity, FontSize, GlobalAmbientLight, IntoScheduleConfigs,
-        Mesh3d, MeshMaterial3d, Node, Plane3d, PositionType, Query, Res, ResMut, StandardMaterial,
+        Mesh, Mesh3d, MeshMaterial3d, Node, Plane3d, PositionType, Query, Res, ResMut,
-        Startup, Text, TextFont, Transform, Update, Vec2, Vec3, Visibility, Window,
+        StandardMaterial, Startup, Text, TextFont, Transform, Update, Vec2, Vec3, Visibility,
+        Window, With,
     },
 };
 use rstar::RTree;
 use toids::{
-    init_naybs, rotate_camera, setup_channels, spawn_background, turkey_time, update_buddies,
+    handle_inputs, init_naybs, setup_channels, spawn_background, turkey_time, update_buddies,
-    update_naybs, update_pos, update_vel, BackgroundHandle, Config, LookAt, Naybs, Positions,
+    update_naybs, update_pos, update_vel, BackgroundHandle, Config, LookAt, Naybs, Paused,
+    Positions,
 };
 
 fn main() {
@@ -23,6 +25,7 @@ fn main() {
             LogDiagnosticsPlugin::default(),
         ))
         .insert_resource(config)
+        .insert_resource(Paused::default())
         .insert_resource(Positions::default())
         .insert_resource(ClearColor(Color::srgb(0.64, 0.745, 0.937))) // a nice light blue
         .insert_resource(GlobalAmbientLight {
@@ -35,16 +38,20 @@ fn main() {
             index: RTree::new(),
             positions: None,
         })
-        .insert_resource(BackgroundHandle::default())
         .insert_resource(LookAt::default())
         .add_systems(
             Startup,
             (setup_channels, setup, init_naybs, spawn_background).chain(),
         )
-        .add_systems(Update, (update_pos, update_buddies, update_vel).chain())
         .add_systems(
             Update,
-            (rotate_camera, close_on_esc, update_naybs.after(update_pos)),
+            (update_pos, update_buddies, update_vel)
+                .chain()
+                .run_if(should_run),
+        )
+        .add_systems(
+            Update,
+            (handle_inputs, close_on_esc, update_naybs.after(update_pos)),
         )
         //.add_systems(Update, update_gizmos)
         .run();
@@ -72,10 +79,8 @@ fn setup(
         Transform::default(),
     ));
 
-    // let toid_model = models.load("models/boid.glb#Scene0");
-
     for _ in 0..config.toids {
-        turkey_time(&mut commands, rand, &models, &mut positions.0);
+        turkey_time(&mut commands, rand, &models, &mut positions);
     }
 
     // instructions
@@ -88,7 +93,7 @@ fn setup(
             ..default()
         },
         children![(
-            Text::new("Up and down for camera forward and back; hold shift to change height.\nLeft or right to move left or right.\nPress 'ESC' to quit."),
+            Text::new("Up and down for camera forward and back; hold shift to change height.\nLeft or right to move left or right.\nPress 'SPACE' to pause, 'ESC' to quit."),
             TextFont {
                 font_size: FontSize::Px(20.0),
                 ..default()
@@ -99,18 +104,27 @@ fn setup(
 
 fn close_on_esc(
     mut commands: Commands,
-    focused_windows: Query<(Entity, &Window)>,
+    windows: Query<Entity, With<Window>>,
     input: Res<ButtonInput<Key>>,
-    mut handle: ResMut<BackgroundHandle>,
+    handle: ResMut<BackgroundHandle>,
 ) {
-    for (window, focus) in focused_windows.iter() {
+    let background_gone = handle.as_ref().is_finished();
-        if !focus.focused {
+    if background_gone {
-            continue;
+        warn!("background is gone, shutting down");
     }
 
-        if input.just_pressed(Key::Escape) {
+    let exited = background_gone || input.just_pressed(Key::Escape);
-            drop((*handle).take());
+    if !exited {
+        return;
+    }
+
+    drop(handle);
+
+    for window in windows.iter() {
         commands.entity(window).despawn();
     }
-    }
+}
+
+fn should_run(paused: Res<Paused>) -> bool {
+    !**paused
 }

src/systems.rs

@@ -5,23 +5,21 @@ use bevy::{input::keyboard::Key, prelude::*, tasks::AsyncComputeTaskPool};
 use rand::RngExt;
 
 use super::{
-    BackgroundHandle, Buddies, GeomWithData, IndexChannel, IndexMessage, LookAt, Naybs, Positions,
+    BackgroundHandle, Buddies, GeomWithData, IndexChannel, IndexMessage, LookAt, Naybs, Paused,
-    RTree, RawMessage, RawPosChannel, Toid, Velocity,
+    Positions, RTree, RawMessage, RawPosChannel, Toid, Velocity,
 };
 
 // toid stuff
-const SPEED: f32 = 6.0;
+const SPEED: f32 = 12.0;
-const SPEED_DIFF_RANGE: f32 = 0.2; // +/- 10%
+const SPEED_DIFF_RANGE: f32 = 0.1;
-const MAX_DELTA_V: f32 = std::f32::consts::PI * 1.5; // basically 360
+const MAX_DELTA_V: f32 = std::f32::consts::PI * 1.2;
-                                                     // degrees/sec
-
 // how far from origin before really wanting to come back
-const RADIUS: f32 = 40.0;
+const RADIUS: f32 = 30.0;
 
 // how close to try to stay to your buddies
-const BUDDY_RADIUS: f32 = 10.0;
+const BUDDY_RADIUS: f32 = SPEED;
 
-const MIN_ALTITUDE: f32 = 3.5;
+const MIN_ALTITUDE: f32 = 13.5;
 
 pub fn setup_channels(mut commands: Commands) {
     let (raw_tx, raw_rx) = bounded(1);
@@ -46,7 +44,6 @@ pub fn update_naybs(
     if let Ok(idx_msg) = idx_chan.rx.try_recv() {
         naybs.index = idx_msg.index;
         let _ = naybs.positions.insert(idx_msg.raw);
-        return;
     }
 
     if let Some(mut raw) = naybs.positions.take() {
@@ -70,9 +67,9 @@ pub fn init_naybs(positions: Res<Positions>, mut naybs: ResMut<Naybs>) {
 }
 
 pub fn spawn_background(
+    mut commands: Commands,
     mut raw_chan: ResMut<RawPosChannel>,
     idx_chan: Res<IndexChannel>,
-    mut handle: ResMut<BackgroundHandle>,
 ) {
     let pool = AsyncComputeTaskPool::get();
 
@@ -82,6 +79,7 @@ pub fn spawn_background(
     let task = pool.spawn(async move {
         let tx = idx_tx;
         let rx = raw_rx;
+
         loop {
             match rx.recv().await {
                 Ok(RawMessage(e_posns)) => {
@@ -104,7 +102,7 @@ pub fn spawn_background(
         }
     });
 
-    let _ = handle.0.insert(task);
+    commands.insert_resource(BackgroundHandle(task));
 }
 
 pub fn turkey_time(
@@ -153,14 +151,16 @@ pub fn update_vel(
 ) {
     let dt = time.delta_secs();
     let max_delta = MAX_DELTA_V * dt;
-    for (xform, mut vel, toid, buddies, entity) in toids.iter_mut() {
+    toids
+        .par_iter_mut()
+        .for_each(|(xform, mut vel, toid, buddies, entity)| {
             let speed = toid.speed;
             let mut dir = vel.normalize();
             let pos = xform.translation;
             let original_dir = dir;
 
             // find buddies and orient; point more towards further-away buddies
-        for buddy in buddies.0.iter() {
+            for buddy in buddies.iter() {
                 let bp = positions.get(buddy).unwrap();
                 let bdir = *bp - pos;
                 let dist = bdir.length();
@@ -174,7 +174,7 @@ pub fn update_vel(
             let min_dist = speed;
             for neighbor in index
                 .index
-            .locate_within_distance(pos.to_array(), min_dist)
+                .locate_within_distance(pos.to_array(), speed.powi(2))
                 .filter(|n| n.data != entity)
             {
                 let bp = Vec3::from_array(*neighbor.geom());
@@ -215,7 +215,7 @@ pub fn update_vel(
             }
 
             **vel = dir * speed;
-    }
+        });
 }
 
 pub fn update_pos(
@@ -230,7 +230,7 @@ pub fn update_pos(
         let look_at = xform.translation + vel.0;
         xform.translation += vel.0 * dt;
         xform.look_at(look_at, Vec3::Y);
-        *positions.entry(entity).or_insert(Vec3::ZERO) = xform.translation;
+        positions.insert(entity, xform.translation);
         new_look += xform.translation;
     }
     **lookat = new_look / positions.len() as f32;
@@ -242,10 +242,12 @@ pub fn update_buddies(
     positions: Res<Positions>,
 ) {
     let d2 = (BUDDY_RADIUS * 1.5).powi(2);
-    for (xform, entity, toid, mut buddies) in toids.iter_mut() {
+    toids
+        .par_iter_mut()
+        .for_each(|(xform, entity, toid, mut buddies)| {
             let pos = xform.translation;
             for buddy in buddies.clone().iter() {
-            let bp = positions[buddy];
+                let bp = positions.get(buddy).unwrap();
                 let bd2 = (bp - pos).length_squared();
                 if bd2 > d2 {
                     buddies.remove(buddy);
@@ -263,7 +265,7 @@ pub fn update_buddies(
                     buddies.insert(neighbor.data);
                 }
             }
-    }
+        });
 }
 
 pub fn update_gizmos(toids: Query<&Transform, With<Toid>>, mut gizmos: Gizmos) {
@@ -278,11 +280,17 @@ pub fn update_gizmos(toids: Query<&Transform, With<Toid>>, mut gizmos: Gizmos) {
     }
 }
 
-pub fn rotate_camera(
+pub fn handle_inputs(
     mut query: Query<&mut Transform, With<Camera>>,
+    mut paused: ResMut<Paused>,
     keys: Res<ButtonInput<Key>>,
     lookat: Res<LookAt>,
 ) {
+    if keys.just_pressed(Key::Space) {
+        let pause = **paused;
+        **paused = !pause;
+    }
+
     let mut xform = query.single_mut().unwrap();
 
     let forward = xform.forward().as_vec3() * 0.7;