Mathesis Wiki

Diese Seite ist nicht editierbar. Sie können den Quelltext sehen, jedoch nicht verändern. Kontaktieren Sie den Administrator, wenn Sie glauben, dass hier ein Fehler vorliegt.
<code Python>
import plotly.graph_objects as go
import random
from plotly.subplots import make_subplots

from Agent import Agent
from World import World

# create a room for Agents
world = World()
world.populate()
world.savePositions()
world.lessMovement()

frames = []


for step in range(world.steps):
# movement for Agents
    for agent1 in range(world.number_of_agents):
        world.agents_list[agent1].move()
        world.updatePostitions(agent1)

# check for collision; infection; quarantine
        for agent2 in range(agent1,world.number_of_agents):
            world.changeCondition1(world.agents_list[agent1], world.agents_list[agent2], probability = 0.3)
        world.changeCondition2(world.agents_list[agent1], quarantine = 0)

    world.updateLists()

# save current attributes of Agents (positions, size, color, shape)
    frames.append(go.Frame(data=[go.Scatter(x=world.x, y=world.y, \
        mode = "markers", \
        marker=dict(size = 10, color = [world.agents_list[i].color for i in range(world.number_of_agents)], \
        )),
        go.Scatter(y = world.infected_series),go.Scatter(y = world.susceptible_series),go.Scatter(y = world.recovered_series)] ))





# ======= VISUALIZATION =======

# properties of the motion field and graphs
fig = make_subplots(
    rows=2, cols=1, row_heights=[4, 2], \
    subplot_titles=('Agents Simulation', 'Data Visualization'))

# add main square for agents
fig.add_shape(type="rect",
    x0=-Agent.radius, y0=-Agent.radius, x1=world.width+Agent.radius, y1=world.height+Agent.radius,
    line=dict(color="black")
)

# add square for quarantine
fig.add_shape(type="rect",
    x0=world.width+Agent.radius +10, y0=-Agent.radius, x1=world.width+Agent.radius+30, y1=(world.height+Agent.radius)/2,
    line=dict(color="black")
)

# add Agents at the right positions in the main square
fig.add_trace(go.Scatter(x = [world.agents_list[i].xstartpos for i in range(world.number_of_agents)], y = [world.agents_list[i].ystartpos for i in range(world.number_of_agents)], name = 'Agents', mode = "markers", \
marker=dict(size = 10, color='cornflowerblue', \
)), row=1, col=1)

# add traces for infected, susceptible, recovered
fig.add_trace(go.Scatter(y=[0], marker_color = "red", marker_size=1, name = 'Infected'), row=2, col=1)
fig.add_trace(go.Scatter(y=[0], marker_color = "cornflowerblue", marker_size=1, name = 'Susceptible'), row=2, col=1)
fig.add_trace(go.Scatter(y=[0], marker_color = "gray", marker_size=1, name = 'Recovered'), row=2, col=1)

fig.frames=frames

# create start button
button = dict(label='Start',
             method='animate',
             args=[None, dict(frame=dict(duration=50, redraw=False),
                              transition=dict(duration=0),
                              fromcurrent=True,
                              mode='immediate')])
fig.update_layout(updatemenus=[dict(type='buttons',
                              showactive=False,
                              y=0,
                              x=1.05,
                              xanchor='left',
                              yanchor='bottom',
                              buttons=[button] )],
                  width=800, height=500)

# set axes properties
fig.update_layout(xaxis1_showgrid = False, xaxis1_showticklabels = False, yaxis1_showgrid = False, yaxis1_showticklabels = False, \
                  xaxis2_range=[0, world.steps], xaxis2_autorange=False,
                  xaxis2_gridcolor = 'lightgrey', xaxis2_linecolor = 'black',
                  yaxis2_range=[0, world.number_of_agents + 1], yaxis2_autorange=False,
                  yaxis2_gridcolor = 'lightgrey', yaxis2_linecolor = 'black')

# adjust background
fig.update_layout(autosize=False, width=1000, height=1000,
                  paper_bgcolor="white", plot_bgcolor = 'white')

# add titel for the quarantine square
fig.add_annotation(text ="Quarantine", xref = "paper", yref = "paper", x=0.965, y=0.8, showarrow = False, font = {'size' : 15})

# show the simulation
fig.show()

</code>
Mathesis Wiki

Benutzer-Werkzeuge

Webseiten-Werkzeuge

Seiten-Werkzeuge
