Debug draw CoM is working!... CoM has error

The debug tool can draw a center-of-mass dot, but it doesn't look like
a sensible location. Time to check on the math, I guess.

src/birdoids_plugin.rs

@@ -37,7 +37,7 @@ impl Plugin for BoidsPlugin {
 }
 
 #[derive(Component)]
-struct Boid;
+pub(crate) struct Boid;
 
 // It's a Boid, but with an extra component so the player
 // can control it from the keyboard
@@ -45,13 +45,13 @@ struct Boid;
 struct PlayerBoid;
 
 #[derive(Component, Deref, DerefMut)]
-struct Velocity(Vec3);
+pub(crate) struct Velocity(Vec3);
 
 #[derive(Component, Deref, DerefMut)]
-struct Acceleration(Vec3);
+pub(crate) struct Acceleration(Vec3);
 
 #[derive(Component)]
-struct TrackedByKdTree;
+pub(crate) struct TrackedByKdTree;
 
 #[derive(Bundle)]
 struct BoidBundle {
@@ -207,6 +207,30 @@ fn cohesion(
     }
 }
 
+pub(crate) fn center_of_boids(points: impl Iterator<Item = Vec2>) -> Option<Vec2> {
+    // Average the points by summing them all together, and dividing by
+    // the total count.
+    // Passing the points as an iterator means we lose the length of the
+    // list. The `.enumerate()` iterator reintroduces that count.
+    let mut points = points.enumerate();
+    
+    // Empty iterators have no points and so no center of mass.
+    // Try to get the first one, but exit with None if it doesn't yield.
+    let init = points.next()?;
+    
+    // if we get one, fold all the remaining values into it.
+    let (len, sum) = points.fold(
+        init,
+        |(len, sum), (idx, point)| {
+            // replace length with most recent index
+            // add running sum & new point for new running sum
+            (idx, sum + point)
+        });
+    
+    let avg = sum / (len as f32);
+    Some(avg)
+}
+
 fn separation(
     spatial_tree: Res<KDTree2<TrackedByKdTree>>,
     mut boids: Query<(&Transform, &mut Acceleration), With<Boid>>,

src/debug_plugin.rs

@@ -1,5 +1,14 @@
 use bevy::{prelude::*, sprite::MaterialMesh2dBundle, window::PrimaryWindow};
+use bevy_spatial::{
+    kdtree::KDTree2,
+    SpatialAccess,
+};
 
+use crate::birdoids_plugin::{
+    center_of_boids, Acceleration, Boid, TrackedByKdTree, Velocity
+};
+
+const SCANRADIUS: f32 = 50.0;
 pub struct BoidsDebugPlugin;
 
 impl Plugin for BoidsDebugPlugin {
@@ -8,7 +17,7 @@ impl Plugin for BoidsDebugPlugin {
             .add_systems(FixedUpdate, (
                 update_cursor,
                 update_scanner_mode,
-                print_gizmo_config,
+                do_scan,
             ));
     }
 }
@@ -21,7 +30,7 @@ fn setup(
     commands.spawn((
         ScannerWidget::default(),
         MaterialMesh2dBundle {
-            mesh: meshes.add(Annulus::new(9.5, 10.0)).into(),
+            mesh: meshes.add(Annulus::new(SCANRADIUS - 1.0, SCANRADIUS)).into(),
             material: materials.add(Color::srgb(0.0, 0.0, 0.0)),
             ..default()
         },
@@ -113,3 +122,40 @@ fn print_gizmo_config(
     let (select, scan) = query.get_single().unwrap();
     println!("Selection: {select:?}, Scanning: {scan:?}");
 }
+
+fn do_scan(
+    boids_query: Query<(&Transform, &Velocity, &Acceleration), With<Boid>>,
+    scanner_query: Query<(&Transform, &SelectionMode, &ScannerMode), With<Cursor>>,
+    spatial_tree: Res<KDTree2<TrackedByKdTree>>,
+    /* Push info to summary somewhere */
+    mut gizmos: Gizmos,
+) {
+    let (cursor_pos, select_mode, scan_mode) = scanner_query.get_single().unwrap();
+    match select_mode {
+        SelectionMode::NearestSingle => todo!(),
+        SelectionMode::CircularArea => {
+            let boids = spatial_tree.within_distance(
+                cursor_pos.translation.xy(),
+                SCANRADIUS,
+            );
+            match scan_mode {
+                ScannerMode::CenterOfMass => {
+                    if let Some(center_mass) = center_of_boids(
+                        // boids returns too many things.
+                        // Map over it and extract only the Vec3's
+                        boids.iter().map(|item| {
+                            item.0
+                        })
+                    ) {
+                        gizmos.circle_2d(
+                            center_mass,
+                            1.0,
+                            bevy::color::palettes::css::RED
+                        );
+                    }
+                },
+                ScannerMode::Velocity => todo!(),
+            }
+        },
+    }
+}