mode = 'autoplay'
# create a colony
chicken_col = Colony(width=X, height=Y, init_pop=INIT_POP, seed=0)
# plotting object
visualizer = StepVisulizer(chicken_col, multiplier=45)
cycle_counter = -1
single_frame = visualizer.plot_step(cycle=cycle_counter) # returns an np.array
if mode == "interactive":
k = ord('n')
while k==ord('n'):
cycle_counter += 1
chicken_col.progress_a_step()
single_frame = visualizer.plot_step(cycle=cycle_counter)
cv2.imshow(WINDOW_NAME, single_frame)
k = cv2.waitKey(0)
cv2.destroyAllWindows()
elif mode == "dump":
frame = 0
while frame < target_frame:
cycle_counter += 1
frame += 1
chicken_col.progress_a_step()
single_frame = visualizer.plot_step(cycle=cycle_counter)
out_name = os.path.join(target_folder, '%05d.png' % frame)
