109 lines
4.1 KiB
Python
109 lines
4.1 KiB
Python
import os
|
|
|
|
from pypdevs.simulator import Simulator
|
|
from plot_template import make_plot_ships_script, make_plot_box_script
|
|
|
|
# from system_solution import * # Teacher's solution
|
|
from system import *
|
|
|
|
## Parameters ##
|
|
|
|
gen_num = 500 # how many ships to generate
|
|
|
|
# How often to generate a ship (on average)
|
|
gen_rate = 1/60/4 # once every 4 minutes
|
|
|
|
# Ship size will be sampled uniformly from the following list.
|
|
gen_types = [1,1,2] # ship size '1' twice as likely to be generated as ship size '2'
|
|
|
|
# Load balancer...
|
|
priorities = {
|
|
# you can outcomment one of these lines to reduce the number of experiments (useful for debugging):
|
|
PRIORITIZE_BIGGER_SHIPS: "bigger",
|
|
PRIORITIZE_SMALLER_SHIPS: "smaller",
|
|
}
|
|
strategies = {
|
|
# you can outcomment one of these lines to reduce the number of experiments (useful for debugging):
|
|
STRATEGY_ROUND_ROBIN: "roundrobin",
|
|
STRATEGY_FILL_ER_UP: "fillerup",
|
|
}
|
|
|
|
# The number of locks and their capacities
|
|
lock_capacities=[3,2] # two locks, of capacity 3 and 2
|
|
|
|
# The different parameters to try for lock_max_wait
|
|
lock_max_waits = [ 0.0+i*120.0 for i in range(5) ] # all these values will be attempted
|
|
# lock_max_waits = [ 15.0 ] # <-- uncomment if you only want to run an experiment with this value (useful for debugging)
|
|
|
|
# How long does it take for a ship to pass through a lock
|
|
passthrough_duration = 60.0*15 # 15 minutes
|
|
|
|
outdir = "assignment_output"
|
|
|
|
plots_ships = []
|
|
plots_box = []
|
|
|
|
os.makedirs(outdir, exist_ok=True)
|
|
|
|
# try all combinations of priorities and strategies (4 total)
|
|
for priority in priorities:
|
|
for strategy in strategies:
|
|
values = []
|
|
# and in each experiment, try a bunch of different values for the 'lock_max_wait' parameter:
|
|
for lock_max_wait in lock_max_waits:
|
|
print("Run simulation:", priorities[priority], strategies[strategy], "max_wait =",lock_max_wait)
|
|
sys = LockQueueingSystem(
|
|
# See system.py for explanation of these values:
|
|
seed=0,
|
|
gen_num=gen_num,
|
|
gen_rate=gen_rate,
|
|
gen_types=gen_types,
|
|
load_balancer_strategy=strategy,
|
|
lock_capacities=lock_capacities,
|
|
priority=priority,
|
|
lock_max_wait=lock_max_wait,
|
|
passthrough_duration=passthrough_duration,
|
|
)
|
|
sim = Simulator(sys)
|
|
sim.setClassicDEVS()
|
|
# sim.setVerbose() # <-- uncomment to see what's going on
|
|
sim.simulate()
|
|
|
|
# all the ships that made it through
|
|
ships = sys.sink.state.ships
|
|
values.append([ship.queueing_duration for ship in ships])
|
|
|
|
# Write out all the ship queueuing durations for every 'lock_max_wait' parameter
|
|
# for every ship, we write a line:
|
|
# <ship_num>, time_max_wait0, time_max_wait1, time_max_wait2, ... time_max_wait10
|
|
filename = f'{outdir}/output_{strategies[strategy]}_{priorities[priority]}.csv'
|
|
with open(filename, 'w') as f:
|
|
try:
|
|
for i in range(gen_num):
|
|
f.write("%s" % i)
|
|
for j in range(len(values)):
|
|
f.write(", %5f" % (values[j][i]))
|
|
f.write("\n")
|
|
except IndexError as e:
|
|
raise Exception("There was an IndexError, meaning that fewer ships have made it to the sink than expected.\nYour model is not (yet) correct.") from e
|
|
|
|
# Generate gnuplot code:
|
|
plots_ships.append(make_plot_ships_script(
|
|
priority=priorities[priority],
|
|
strategy=strategies[strategy],
|
|
max_waits=lock_max_waits,
|
|
gen_num=gen_num,
|
|
))
|
|
plots_box.append(make_plot_box_script(
|
|
priority=priorities[priority],
|
|
strategy=strategies[strategy],
|
|
max_waits=lock_max_waits,
|
|
gen_num=gen_num,
|
|
))
|
|
|
|
# Finally, write out a single gnuplot script that plots everything
|
|
with open(f'{outdir}/plot.gnuplot', 'w') as f:
|
|
# first plot the ships
|
|
f.write('\n\n'.join(plots_ships))
|
|
# then do the box plots
|
|
f.write('\n\n'.join(plots_box))
|