visualizer.py

import cv2
import numpy as np
from random import randint, shuffle

def isSorted(l):
    previous = l[0]
    for i in range(1, len(l)):
        if l[i] < previous:
            return False
        previous = l[i]
    return True

def insertion_sort(l):
    for index in range(1, len(l)):
        k = l[index]
        j = index-1
        while j >= 0 and k < l[j]:
            l[j+1], l[j] = l[j], l[j+1]
            j-=1
            yield l
        l[j+1] = k
        yield l

def selection_sort(l):
    for i in range(len(l)):  
        min_ind = i 
        for j in range(i+1, len(l)):
            if l[j] < l[min_ind]:
                min_ind = j
        l[i], l[min_ind] = l[min_ind], l[i]
        yield l

def bogo_sort(l):
    while not isSorted(l):
        shuffle(l)
        yield l 
    yield l 

#generate array, window height and width and the empty canvas for drawing with an
#empty copy to reset it every frame
arr = [randint(0, 600) for i in range(50)]
HEIGHT = max(arr) + 1
WIDTH = len(arr) * 10
img = np.zeros((HEIGHT, WIDTH, 3), np.uint8)
reset = img.copy()

#call bogo_sort, selection_sort or insertion_sort here with the array
#as a parameter
steps = insertion_sort(arr)

#prev_step stores the array before the current "step" in the sorting algorithms
#step = whenever the algorithm changes the array
#different is a list of indexes that have changed since the previous step
prev_step, different = arr.copy(), []
for step in steps:
    print(step)
    for i in range(len(arr)):
        if step[i] != prev_step[i]:
            different.append(i)

    for i in range(len(arr)):
        if i in different:
            img[HEIGHT:HEIGHT-step[i]:-1, i*10:i*10+5] = 0, 0, 200
        else:
            img[HEIGHT:HEIGHT-step[i]:-1, i*10:i*10+5] = 255, 255, 255
        cv2.imshow('', img)
    cv2.waitKey(1)
    img = reset.copy()
    #reset different and set prev_step
    prev_step, different = step.copy(), []

#play an ending "animation"
for i in range(len(arr)):
    img[HEIGHT:HEIGHT-prev_step[i]:-1, i*10:i*10+5] = 255, 255, 255
    cv2.imshow('', img)
    cv2.waitKey(50)

cv2.waitKey(0)
cv2.destroyAllWindows()