class Simulation:
drawing = True
lighting_mode = True
food_respawn = False
def __init__(self):
self.screenSize = (800,800)
self.running = False
self.spawn = Spawn([self.screenSize[0]/2, self.screenSize[1]/2])
self.foragers = []
self.food = []
self.frame_num = 0
self.draw_sensors = True
def init(self):
# Pygame init
pygame.init()
self.screen = pygame.display.set_mode(self.screenSize)
self.font = pygame.font.SysFont("monospace", 15)
def copy(self):
new_sim = Simulation()
new_sim.foragers = deepcopy(self.foragers)
new_sim.food = deepcopy(self.food)
return new_sim
def startLoop(self):
self.running = True
while self.running:
self.nextFrame()
def nextFrame(self):
self.__checkEvents()
if Simulation.drawing: self.__draw()
self.__checkCollisions()
self.__move()
self.frame_num += 1
if not Simulation.lighting_mode: time.sleep(1.0/100.0)
def spawnForager(self):
pos = self.spawn.pos[:]
pos[0] += randint(-1*self.spawn.radius, self.spawn.radius)
pos[1] += randint(-1*self.spawn.radius, self.spawn.radius)
theta = randint(0,3)*math.pi/2
self.foragers.append(Forager(pos, theta))
def spawnFood(self):
pos = self.spawn.pos[:]
dx = randint(self.spawn.radius, self.screenSize[0]/2)
dy = randint(self.spawn.radius, self.screenSize[1]/2)
pos[0] += dx * (-1 if randint(0,1) == 0 else 1)
pos[1] += dy * (-1 if randint(0,1) == 0 else 1)
self.food.append(Food(pos))
def __checkEvents(self):
for event in pygame.event.get():
if event.type == pygame.QUIT:
self.running = False
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_SPACE:
Simulation.drawing = (False if Simulation.drawing else True)
print "Toggled drawing to %s" % Simulation.drawing
if event.key == pygame.K_l:
Simulation.lighting_mode = (False if Simulation.lighting_mode else True)
print "Toggled lighting_mode to %s" % Simulation.lighting_mode
def __draw(self):
self.screen.fill((255,255,255))
self.spawn.draw(self.screen)
# Draw foragers
total_food = 0
for forager in self.foragers:
forager.draw(self.screen)
if self.draw_sensors: forager.drawSensor(self.screen)
total_food += forager.food_collected
# Draw score
label = self.font.render("Food Collected: %s" % total_food, 1, (0,0,0))
self.screen.blit(label, (20, 20))
# Draw food
for food in self.food:
food.draw(self.screen)
pygame.display.flip()
def __move(self):
for forager in self.foragers:
forager.move()
def __checkCollisions(self):
for forager in self.foragers:
collissions = forager.checkCollisions(self.food, self.spawn)
for i in range(0,collissions):
if Simulation.food_respawn: self.spawnFood()
def allOffScreen(self):
on_screen = False
for forager in self.foragers:
if (forager.pos[0] > forager.radius) and (forager.pos[0] - forager.radius < self.screenSize[0]) and (forager.pos[1] > forager.radius) and (forager.pos[1] - forager.radius < self.screenSize[1]):
on_screen = True
break
return not on_screen
