main.py

from tkinter import *
from tkinter import ttk
import random
from colours import *

# Importing algorithms 
from algorithms.selectionSort import selectionSort  
from algorithms.insertSort import insertion_sort

# Main window 
window = Tk()
window.title("Sorting Algorithms Visualization")
window.maxsize(10000, 700)
window.config(bg = WHITE)

algorithm_name = StringVar()
algo_list = ['Selection Sort', 'Insert Sort']

data = []

# This function will draw randomly generated list data[] on the canvas as vertical bars
def drawData(data, colorArray):
    canvas.delete("all")
    canvas_width = 1000
    canvas_height = 400
    x_width = canvas_width / (len(data) + 1)
    offset = 4
    spacing = 2
    normalizedData = [i / max(data) for i in data]

    for i, height in enumerate(normalizedData):
        x0 = i * x_width + offset + spacing
        y0 = canvas_height - height * 390
        x1 = (i + 1) * x_width + offset
        y1 = canvas_height
        canvas.create_rectangle(x0, y0, x1, y1, fill=colorArray[i])

    window.update_idletasks()

# This function will generate array with random values every time we hit the generate button
def generate():
    global data

    data = []
    for i in range(0, 100):
        random_value = random.randint(1, 150)
        data.append(random_value)

    drawData(data, [BLUE for x in range(len(data))])


# This function will trigger a selected algorithm and start sorting
def sort():
    global data
    timeTick = 0.001
    
    if algo_menu.get() == 'Selection Sort':
        selectionSort(data, drawData, timeTick)
    elif algo_menu.get() == 'Insert Sort':
        insertion_sort(data, drawData, timeTick)


### User interface here ###
UI_frame = Frame(window, width= 600, height=500, bg=WHITE)
UI_frame.grid(row=0, column=0, padx=10, pady=5)

# dropdown to select sorting algorithm 
l1 = Label(UI_frame, text="Algorithm: ", bg=WHITE)
l1.grid(row=0, column=0, padx=10, pady=5, sticky=W)
algo_menu = ttk.Combobox(UI_frame, textvariable=algorithm_name, values=algo_list)
algo_menu.grid(row=0, column=1, padx=5, pady=5)
algo_menu.current(0)

# sort button 
b1 = Button(UI_frame, text="Sort", command=sort, bg=LIGHT_GRAY)
b1.grid(row=2, column=1, padx=5, pady=5)

# button for generating array 
b3 = Button(UI_frame, text="Generate Array", command=generate, bg=LIGHT_GRAY)
b3.grid(row=2, column=0, padx=5, pady=5)

# canvas to draw our array 
canvas = Canvas(window, width=1000, height=400, bg=WHITE)
canvas.grid(row=1, column=0, padx=10, pady=5)


window.mainloop()