from simulation.simulation import Simulation import pygame #from experiments.flocking import parameters as p # from experiments.aggregation import parameters as p from experiments.covid import parameters as p """ Code for multi-agent simulation in PyGame with/without physical objects in the environment """ if __name__ == "__main__": pygame.init() sim = Simulation(num_agents=p.N_AGENTS, screen_size=p.SCREEN, swarm_type=p.SWARM, iterations=p.FRAMES) sim.run()
from simulation.simulation import Simulation from display.screen import Screen from brain.space import Volume import numpy as np screen_size = np.array([600, 600]) BACKGROUND_COLOR = (154, 154, 158) RECORD_SETTINGS = ("gallery/imgs", "gallery/videos", "result", 24) space = Volume(shape='rect', location=np.array([0, 0, 0]), size=screen_size) sim = Simulation( space, Screen(screen_size, BACKGROUND_COLOR, "GENERIC SIMULATION NAME", space, RECORD_SETTINGS)) for i in range(1): sim.create_random_agent(14) for i in range(300): sim.environment.generate_new_food() sim.run(-1, 1, 0.5)