lib/rules/cohesion.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

# frozen_string_literal: true

FF::Sys.new('Cohesion', priority: 50) do
  center_mass = [0.0, 0.0]
  boids_count = FF::Cmp::Fish[0].entities.count

  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::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) / $config.cohesion
    boid_update.cy += (move_boid[1] - boid_update.y) / $config.cohesion
  end
end