BoidsSwarm.lua
fffea8140876master

BoidsSwarm.lua
View Options

			local assertArgs = include "assertArgs"

			local util = include "util"

			local BoidsSwarm = lib "class" ("BoidsSwarm")

			BoidsSwarm:include(mixin "_has" ("Boids"))
			BoidsSwarm:include(mixin "_has" ("Bounds"))

			BoidsSwarm:include(mixin "_has" ("TargetPosition"))
			BoidsSwarm:include(mixin "_has" ("AvoidPosition"))

			function BoidsSwarm:initialize(args)
			assertArgs(args, "bounds")
			self:setBoids({})
			self:setBounds(args.bounds)
			self:setTargetPosition(args.targetPosition)
			self:setAvoidPosition(args.avoidPosition or false)
			return self
			end

			function BoidsSwarm:update(dt)
			local boids = self:getBoids()
			for i = 1, #boids do
			boids[i]:update(dt)
			end
			end

			function BoidsSwarm:draw()
			local boids = self:getBoids()
			for i = 1, #boids do
			boids[i]:draw()
			end
			end

			function BoidsSwarm:addBoid(boid)
			boid:setSwarm(self)
			table.insert(self:getBoids(), boid)
			return self
			end

			function BoidsSwarm:removeBoid(boid)
			local boids = self:getBoids()
			for i, _boid in pairs(boids) do
			if boid == _boid then
			boid:setSwarm(nil)
			table.remove(boids, i)
			end
			end
			return self
			end

			function BoidsSwarm:isNeighbour(boid, other)
			return boid ~= other
			end

			function BoidsSwarm:forAllNeighbours(boid, f)
			local boids = self:getBoids()

			local n = 0
			for i = 1, #boids do
			local _boid = boids[i]
			if self:isNeighbour(boid, _boid) then
			f(_boid)
			n = n + 1
			end
			end

			return n
			end

			return BoidsSwarm