Given the code below, modify and extend it to add a function monsterGen(numMonst1, numMonst2) to create the Cookie and TwoHeaded monsters with unique names/identifiers (1 mark). Include a function monsterPlot(monsterlist, limits) to set up and plot the data, annotate the identifiers of the monsters, and plot them in the correct colours (2 marks). Modify the stepChange(limits) function to wrap around if the monster goes past the top/bottom/left/right edges. Print the name of the monster and the type of wrap-around it's doing (e.g. Right wrap for C3) (2 marks).
import random
import matplotlib.pyplot as plt
import numpy as np
from codes.monsters import *
limits = (-30, 20)
def simT(ct, X):
monsterlist = []
# timesteps and creature counts
for i in range(ct):
randX = random.randint(limits[0], limits[1])
randY = random.randint(limits[0], limits[1])
monsterlist.append(Cookie("Bruce", [randX, randY])) # blue monster
for i in range(ct):
randX = random.randint(limits[0], limits[1])
randY = random.randint(limits[0], limits[1])
monsterlist.append(TwoHeaded("Bruce", [randX, randY])) # purple Monster
for i in range(ct):
print("###TIMESTEP", i, "###")
for m in monsterlist:
m.stepChange(limits)
plt.xlim(limits[0])
plt.ylim(limits[1])
xvalues = []
yvalues = []
sizes = []
for m in monsterlist:
xvalues.append(m.pos[0])
yvalues.append(m.pos[1])
sizes.append(m.getSize())
plt.scatter(xvalues, yvalues, s=sizes, color="orange")
plt.show()
Activate Windows Go to Settings to activate Windows.
im(6, 10)
