def seperation(self,fish, minDistance):
if len(fish) < 1:
return Vector()
distance = 0
numClose = 0
distsum = Vector()
for boid in fish:
distance = (self.pos-boid.pos).length()
if distance < minDistance and distance != 0:
numClose += 1
distsum += (boid.pos-self.pos).divide(distance**2)
if numClose == 0:
return Vector()
return (-distsum.divide(numClose)).normalize()
def allign(self,fish):
if len(fish) < 1:
return
# calculate the average velocities of the other boids
avg = Vector()
count = 0
for boid in fish:
if(boid.pos - self.pos).length()<self.per:
count+=1
avg += boid.vel
if count>0:
avg = avg.divide(count)
# set our velocity towards the others
return (avg).normalize()
else:
return Vector()
def cohesion(self,fish):
if len(fish) < 1:
return
# calculate the average distances from the other boids
com = Vector()
count = 0
for boid in fish:
if boid.pos == self.pos:
continue
elif (boid.pos - self.pos).length()<self.per:
com += (self.pos - boid.pos)
count+=1
if count>0:
com = com.divide(count)
return -com.normalize()
else:
return Vector()