59 lines
2.2 KiB
Python
59 lines
2.2 KiB
Python
|
import PySimpleGUI as sg
|
||
|
|
import random
|
||
|
|
# ------- Sort visualizer. Displays bar chart representing list items -------
|
||
|
|
BAR_SPACING, BAR_WIDTH, EDGE_OFFSET = 11, 10, 3
|
||
|
|
DATA_SIZE = GRAPH_SIZE = (700,500) # width, height of the graph portion
|
||
|
|
|
||
|
|
def bubble_sort(arr):
|
||
|
|
def swap(i, j):
|
||
|
|
arr[i], arr[j] = arr[j], arr[i]
|
||
|
|
n = len(arr)
|
||
|
|
swapped = True
|
||
|
|
x = -1
|
||
|
|
while swapped:
|
||
|
|
swapped = False
|
||
|
|
x = x + 1
|
||
|
|
for i in range(1, n - x):
|
||
|
|
if arr[i - 1] > arr[i]:
|
||
|
|
swap(i - 1, i)
|
||
|
|
swapped = True
|
||
|
|
yield arr
|
||
|
|
|
||
|
|
def draw_bars(graph, items): # draws all the bars for all values across screen
|
||
|
|
# type: (sg.Graph, List)->None
|
||
|
|
for i, item in enumerate(items):
|
||
|
|
graph.draw_rectangle(top_left=(i * BAR_SPACING + EDGE_OFFSET, item),
|
||
|
|
bottom_right=(i * BAR_SPACING + EDGE_OFFSET + BAR_WIDTH, 0), fill_color='#76506d')
|
||
|
|
|
||
|
|
def main():
|
||
|
|
sg.change_look_and_feel('LightGreen')
|
||
|
|
# Make list to sort
|
||
|
|
num_bars = DATA_SIZE[0]//(BAR_WIDTH+1)
|
||
|
|
list_to_sort = [DATA_SIZE[1]//num_bars*i for i in range(1,num_bars)]
|
||
|
|
random.shuffle(list_to_sort)
|
||
|
|
|
||
|
|
# define window layout
|
||
|
|
graph = sg.Graph(GRAPH_SIZE, (0,0), DATA_SIZE)
|
||
|
|
layout = [[graph],
|
||
|
|
[sg.T('Speed Faster'), sg.Slider((0,20), orientation='h', default_value=10, key='-SPEED-'), sg.T('Slower')]]
|
||
|
|
|
||
|
|
window = sg.Window('Sort Demonstration', layout, finalize=True)
|
||
|
|
draw_bars(graph, list_to_sort) # draw the initial window's bars
|
||
|
|
|
||
|
|
sg.popup('Click OK to begin Bubblesort') # Wait for user to start it up
|
||
|
|
bsort = bubble_sort(list_to_sort) # get an iterator for the sort
|
||
|
|
timeout=10 # start with 10ms delays between draws
|
||
|
|
while True: # ----- The event loop -----
|
||
|
|
event, values = window.read(timeout=timeout)
|
||
|
|
if event is None:
|
||
|
|
break
|
||
|
|
try:
|
||
|
|
partially_sorted_list = bsort.__next__()
|
||
|
|
except:
|
||
|
|
sg.popup('Sorting done!')
|
||
|
|
break
|
||
|
|
graph.Erase()
|
||
|
|
draw_bars(graph, partially_sorted_list)
|
||
|
|
timeout = int(values['-SPEED-'])
|
||
|
|
window.close()
|
||
|
|
main()
|