def populate(self): for i in range(self.population_size): if random.uniform(0, 1) > self.predator_chance: organism = Organism() organism.randomize() self.population.append(organism) else: predator = Predator() predator.randomize() self.population.append(predator)
def click(self, keys): pos = pygame.mouse.get_pos() result = self.get_organism(pos) # print the contents of the location if result is not None: print(result) else: if keys[pygame.K_o]: new_organism = Organism(pos[0], pos[1]) new_organism.randomize() self.population.append(new_organism) elif keys[pygame.K_f]: new_plant = Plant(pygame.Rect(pos[0], pos[1], 6, 6)) self.vegetation.append(new_plant) elif keys[pygame.K_p]: new_predator = Predator(pos[0], pos[1]) new_predator.randomize() self.population.append(new_predator)