import numpy as np
from random import shuffle
from algorithm_visualizer import alvito
from utils import random_array
import matplotlib.pyplot as plt
from IPython.display import HTML, display
from IPython.utils import io
avo = alvito()
avo.fps = 1
avo.show_xlable = False
avo.title_fontsize = 9
avo.rectangle_linewidth = 2
avo.dpi = 80
avo.save_dir = 'cmap_gifs/'
avo.custom_save_name = True
shape = (10,1)
array = random_array(shape)
with io.capture_output() as captured: # supressing output when looping over all the cmaps
for cmap_name in plt.colormaps():
avo.colormap = cmap_name
avo.save_name = cmap_name
avo.insertionSort(array, cmap_name)
for i in range(0,len(plt.colormaps()),11):
display_sting = "<table><tr>"
for cmap in plt.colormaps()[i:i+11]:
display_sting += f'<td><img src=\'cmap_gifs/{cmap}.gif\'></td>'
display_sting += "</tr></table>"
display(HTML(display_sting))