BubbleSortVisualization.py

import random
import time
import pygame
from pygame.locals import *
 
pygame.init()
 
width, height = 500, 500
screen = pygame.display.set_mode((width, height))

GREEN = (0,255,0)
BLACK = (0,0,0)
WHITE = (255,255,255)

def create_list(start_num,end_num,length):
    new_list = []

    for i in range(length):
        new_list.append(random.randint(start_num,end_num))
    return new_list

test_list = create_list(0,400,100)
sorted = False

def draw_time(screen,x,y,time):
    font = pygame.font.Font(None, 36)
    text = font.render("Time: " + str(time) + "s", 1, WHITE)
    screen.blit(text,(x,y))


def display_list():
    screen.fill(BLACK)
    rect_x = 0

    for i in range(len(test_list)):
        pygame.draw.rect(screen,(0,255,0),(rect_x,500-test_list[i],4,test_list[i]))
        rect_x += 5

    pygame.display.flip()

def sort():
    
    #Iterate through the whole list
    for i in range (len(test_list)):

        for j in range(0,len(test_list)-i-1):

            if test_list[j] > test_list[j+1]:
                
                test_list[j],test_list[j+1] = test_list[j+1],test_list[j]
                display_list()
                #pygame.time.wait(5)


# Game loop.
run = True
start = time.time()
sort()
end = time.time()
time_elapsed = round(end - start,2)

while run:
  
    for event in pygame.event.get():
        if event.type == QUIT:
            run = False
    
    draw_time(screen,75,75,time_elapsed)

    pygame.display.flip()

pygame.quit()