soil/examples/programmatic/programmatic_sim.py

"""
Example of a fully programmatic simulation, without definition files.
"""
from soil import Simulation, Environment, agents
from networkx import Graph
import logging


def mygenerator():
    # Add only a node
    G = Graph()
    G.add_node(1)
    G.add_node(2)
    return G


class MyAgent(agents.NetworkAgent, agents.FSM):
    times_run = 0
    @agents.default_state
    @agents.state
    def neutral(self):
        self.debug("I am running")
        if self.prob(0.2):
            self.times_run += 1
            self.info("This runs 2/10 times on average")


class ProgrammaticEnv(Environment):

    def init(self):
        self.create_network(generator=mygenerator)
        assert len(self.G)
        self.populate_network(agent_class=MyAgent)
        self.add_agent_reporter('times_run')


simulation = Simulation(
    name="Programmatic",
    model=ProgrammaticEnv,
    seed='Program',
    num_trials=1,
    max_time=100,
    dump=False,
)

if __name__ == "__main__":
    # By default, logging will only print WARNING logs (and above).
    # You need to choose a lower logging level to get INFO/DEBUG traces
    logging.basicConfig(level=logging.INFO)
    envs = simulation.run()

    for agent in envs[0].agents:
        print(agent.times_run)