cluster.rb

module Kmeans
  class Cluster
    attr_accessor :centroid, :points

    def initialize(centroid)
      @centroid = centroid
      @points = []
    end

    def centroid!
      xa = ya = 0
      old_centroid = @centroid

      @points.each do |point|
        xa += point.x
        ya += point.y
      end

      xa /= points.length
      ya /= points.length

      @centroid = Point.new(xa, ya)
      #puts old_centroid.to_a
      #puts @centroid.to_a
      return GpsWaypoints.distance(old_centroid.to_a,centroid.to_a)
    end
  end
end