mirror of
https://github.com/gsi-upm/soil
synced 2024-11-14 15:32:29 +00:00
feab0ba79e
The examples weren't being properly tested in the last commit. When we fixed that a lot of bugs in the new implementation of environment and agent were found, which accounts for most of these changes. The main difference is the mechanism to load simulations from a configuration file. For that to work, we had to rework our module loading code in `serialization` and add a `source_file` attribute to configurations (and simulations, for that matter).
41 lines
1.0 KiB
Python
41 lines
1.0 KiB
Python
from soil.agents import FSM, state, default_state
|
|
from soil.time import Delta
|
|
|
|
|
|
class Fibonacci(FSM):
|
|
"""Agent that only executes in t_steps that are Fibonacci numbers"""
|
|
prev = 1
|
|
|
|
@default_state
|
|
@state
|
|
def counting(self):
|
|
self.log("Stopping at {}".format(self.now))
|
|
prev, self["prev"] = self["prev"], max([self.now, self["prev"]])
|
|
return None, Delta(prev)
|
|
|
|
|
|
class Odds(FSM):
|
|
"""Agent that only executes in odd t_steps"""
|
|
|
|
@default_state
|
|
@state
|
|
def odds(self):
|
|
self.log("Stopping at {}".format(self.now))
|
|
return None, Delta(1 + self.now % 2)
|
|
|
|
|
|
from soil import Environment, Simulation
|
|
from networkx import complete_graph
|
|
|
|
|
|
class TimeoutsEnv(Environment):
|
|
def init(self):
|
|
self.create_network(generator=complete_graph, n=2)
|
|
self.add_agent(agent_class=Fibonacci, node_id=0)
|
|
self.add_agent(agent_class=Odds, node_id=1)
|
|
|
|
|
|
sim = Simulation(model=TimeoutsEnv, max_steps=10, interval=1)
|
|
|
|
if __name__ == "__main__":
|
|
sim.run(dump=False) |