From 501ec02e894865836f2960b0bbc16b3448e3707f Mon Sep 17 00:00:00 2001
From: realtradam <mr.malczewski@gmail.com>
Date: Mon, 9 Aug 2021 09:15:38 -0400
Subject: ruboids!

---
 lib/rules/cohesion.rb | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 lib/rules/cohesion.rb

(limited to 'lib/rules/cohesion.rb')

diff --git a/lib/rules/cohesion.rb b/lib/rules/cohesion.rb
new file mode 100644
index 0000000..efbd71d
--- /dev/null
+++ b/lib/rules/cohesion.rb
@@ -0,0 +1,21 @@
+FF::Sys.new('Cohesion', priority: 50) do
+
+  center_mass = [0.0,0.0]
+  boids_count = FF::Cmp::Boids.each.to_a.count
+
+  FF::Cmp::Boids.each do |boid|
+    center_mass[0] += boid.x
+    center_mass[1] += boid.y
+  end
+
+  FF::Cmp::Boids.each do |boid_update|
+    move_boid = center_mass.dup
+    move_boid[0] -= boid_update.x
+    move_boid[1] -= boid_update.y
+    move_boid[0] /= boids_count - 1.0
+    move_boid[1] /= boids_count - 1.0
+
+    boid_update.cx += (move_boid[0] - boid_update.x) / $cohesion
+    boid_update.cy += (move_boid[1] - boid_update.y) / $cohesion
+  end
+end
-- 
cgit v1.2.3