mirror of https://github.com/gsi-upm/soil
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
493 B
Python
19 lines
493 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.sim().dir_path, "graph-"+str(self.env.now)+".png"))
|