diff options
Diffstat (limited to 'lib/rules/cohesion.rb')
| -rw-r--r-- | lib/rules/cohesion.rb | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/lib/rules/cohesion.rb b/lib/rules/cohesion.rb index efbd71d..5abefd7 100644 --- a/lib/rules/cohesion.rb +++ b/lib/rules/cohesion.rb @@ -1,21 +1,23 @@ FF::Sys.new('Cohesion', priority: 50) do center_mass = [0.0,0.0] - boids_count = FF::Cmp::Boids.each.to_a.count + boids_count = FF::Cmp::Fish[0].entities.count - FF::Cmp::Boids.each do |boid| + FF::Cmp::Fish[0].entities.each do |ent| + boid = ent.components[FF::Cmp::Boids].first center_mass[0] += boid.x center_mass[1] += boid.y end - FF::Cmp::Boids.each do |boid_update| + FF::Cmp::Fish[0].entities.each do |ent| + boid_update = ent.components[FF::Cmp::Boids].first 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 + boid_update.cx += (move_boid[0] - boid_update.x) / $config.cohesion + boid_update.cy += (move_boid[1] - boid_update.y) / $config.cohesion end end |