# sim.addCar(1)
# sim.addCar(2)
# taskNodes1 = OrderedDict()
# taskNodes1["11"] = 2
# taskNodes1["22"] = 3
# taskNodes1["12"] = 2
# sim.addTask(taskNodes1)
# taskNodes2 = OrderedDict()
# taskNodes2["22"] = 1
# taskNodes2["12"] = 1
# taskNodes2["11"] = 1
# sim.addTask(taskNodes2)
sim = Simulator()
sim.addNode(1,1)
sim.addNode(3,1)
sim.addCharger(2,2)
sim.addCar(1)
sim.addCar(2)
taskNodes1 = OrderedDict()
taskNodes1["31"] = 4
taskNodes1["11"] = 2
taskNodes1["31"] = 3
sim.addTask(taskNodes1)
print("=======================")
print("\n")
def setup():
""" Establishes a Simulator, Nodes, Cars, and Tasks.
Used to quickly alter how the simulation is set up.
"""
print("=======================")
sim = Simulator()
sim.addNode(0, 0)
sim.addNode(0, 1)
sim.addCharger(1, 0)
sim.addCar(1)
sim.addCar(2)
sim.addCar(3)
sim.addCar(4)
sim.addCar(5)
taskNodes1 = OrderedDict() #3 steps = 1
taskNodes1["00"] = 1
taskNodes1["01"] = 1
taskNodes2 = OrderedDict() #6 steps = 2
taskNodes2["00"] = 2
taskNodes2["01"] = 3
#1 hr = 3 steps
#Start (Step 1, 7 jobs)
addTasks(sim, 3, 1, taskNodes1)
addTasks(sim, 2, 1, taskNodes2)
#Hour 1 (Step 4, 0 jobs)
addTasks(sim, 0, 4, taskNodes1)
addTasks(sim, 0, 4, taskNodes2)
#Hour 2 (Step 7, 4 jobs)
addTasks(sim, 2, 7, taskNodes1)
addTasks(sim, 1, 7, taskNodes2)
#Hour 3 (Step 10, 6 jobs)
addTasks(sim, 2, 10, taskNodes1)
addTasks(sim, 2, 10, taskNodes2)
#Hour 4 (Step 13, 10 jobs)
addTasks(sim, 1, 13, taskNodes1)
addTasks(sim, 4, 13, taskNodes2)
#Hour 5 (Step 16, 6 jobs)
addTasks(sim, 4, 16, taskNodes1)
addTasks(sim, 1, 16, taskNodes2)
#Hour 6 (Step 19, 4 jobs)
addTasks(sim, 2, 19, taskNodes1)
addTasks(sim, 1, 19, taskNodes2)
#Hour 7 (Step 22, 2 jobs)
addTasks(sim, 0, 22, taskNodes1)
addTasks(sim, 1, 22, taskNodes2)
#Hour 8 (Step 25, 0 jobs)
addTasks(sim, 0, 25, taskNodes1)
addTasks(sim, 0, 25, taskNodes2)
#Hour 9 (Step 28, 0 jobs)
addTasks(sim, 0, 28, taskNodes1)
addTasks(sim, 0, 28, taskNodes2)
#Hour 10 (Step 31, 0 jobs)
addTasks(sim, 0, 31, taskNodes1)
addTasks(sim, 0, 31, taskNodes2)
#Hour 11 (Step 34, 0 jobs)
addTasks(sim, 0, 34, taskNodes1)
addTasks(sim, 0, 34, taskNodes2)
#Hour 12 (Step 37, 0 jobs)
addTasks(sim, 0, 37, taskNodes1)
addTasks(sim, 0, 37, taskNodes2)
#End (Step 40)
print("=======================")
print("\n")
return sim
