simulation.py

from creature import Creature, distance
import random
import pygame as pg
import time
from display import Displayer
import numpy as np
from logs import Log



class Simulation:

    def __init__(self):

        # constants
        self.starting_food = 100
        self.starting_creatures = 4
        self.food_spawn_probability = 0.08
        self.simulation_start_time = time.time()
        self.field_size = 1000
        self.reproduction_cost = 1.2

        # objects
        self.foods = [(random.uniform(0, self.field_size), random.uniform(0, self.field_size)) for i in range(self.starting_food)]
        self.creatures = [Creature() for i in range(self.starting_creatures)]

    def run(self):
        displayer = Displayer(self)
        log = Log(self)
        dt = 0.001
        while not displayer.quit(self):

            start_time = time.time()
            self.move_frame(dt)
            displayer.display_frame(self)
            log.save_frame()

            dt = time.time() - start_time
        log.save_log("logs3.obj")
    
    def cost(self, genes, dt):
        return(6*genes["velocity"]**2 + genes["vision"]**2 + genes["grab range"]**2 + 100)*dt*2e-5

    def move_frame(self, dt):
        for creature in self.creatures[:]:
            creature.move(self, dt)
            for food in self.foods[:]:
                if distance(food, creature.position) < creature.genes["grab range"]:
                    creature.hunger = creature.hunger + 1
                    self.foods.remove(food)
                    
            creature.hunger -= self.cost(creature.genes, dt)
            if creature.hunger < 0:
                self.creatures.remove(creature)
                print("creature died")
                
        if random.uniform(0, 1) < self.food_spawn_probability:
            self.foods.append((random.uniform(0, self.field_size), random.uniform(0, self.field_size)))



if __name__ == "__main__":
    simulation1 = Simulation()
    simulation1.run()