mirror of
https://github.com/gsi-upm/soil
synced 2024-09-20 15:31:43 +00:00
7d1c800490
* Graphs are not saved by default (not backwards compatible) * Modified newsspread examples * More granular options to save results (exporting to CSV and GEXF are now optional) * Updated tutorial to include exporting options * Removed references from environment to simulation * Added parallelism to simulations (can be turned off with a flag or argument).
19 lines
489 B
Python
19 lines
489 B
Python
from . import BaseAgent
|
|
|
|
import os.path
|
|
import matplotlib
|
|
import matplotlib.pyplot as plt
|
|
import networkx as nx
|
|
|
|
|
|
class DrawingAgent(BaseAgent):
|
|
"""
|
|
Agent that draws the state of the network.
|
|
"""
|
|
|
|
def step(self):
|
|
# Outside effects
|
|
f = plt.figure()
|
|
nx.draw(self.env.G, node_size=10, width=0.2, pos=nx.spring_layout(self.env.G, scale=100), ax=f.add_subplot(111))
|
|
f.savefig(os.path.join(self.env.get_path(), "graph-"+str(self.env.now)+".png"))
|