main.py

import argparse
from agents.rule_based_agent import RBAgent
from agents.ompas import OMPASAgent
from craftbots.simulation import Simulation
from gui.main_window import CraftBotsGUI
from platform_interface_pb2_grpc import *
from platform_interface_pb2 import *

if __name__ == '__main__':

    # parse command line arguments
    arg_parser = argparse.ArgumentParser()
    arg_parser.add_argument('-f', help="configuration file", type=str, default='craftbots/config/simple_configuration.yaml')
    args = arg_parser.parse_args()

    # Simulation
    sim = Simulation(configuration_file=args.f)

    # agent
    #agent = RBAgent()
    agent = OMPASAgent()
    sim.agents.append(agent)

    # GUI
    gui = CraftBotsGUI(sim)
    gui.simulation.reset_simulation()
    gui.simulation.start_simulation()

    gui.start_window()