Data-Sorting-Visualized/sortingManager.py at main · PyricRL/Data-Sorting-Visualized · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
import sortingUtils
from sortingUtils import createTestData
from bubbleSortAlgorithm import bubbleSort, bubbleSortNoVisible
from bogoSortAlgorithm import bogoSort, bogoSortNoVisible
from miracleSortAlgorithm import miracleSort, miracleSortNoVisible
from stalinSortAlgorithm import stalinSort, stalinSortNoVisible
from selectionSortAlgorithm import selectionSort, selectionSortNoVisible
from mergeSortAlgorithm import mergeSort, mergeSortNoVisible
from insertionSortAlgorithm import insertionSort, insertionSortNoVisible
from quickSortAlgorithm import quickSort, quickSortNoVisible
from cocktailSortAlgorithm import cocktailSort, cocktailSortNoVisible
from heapSortAlgorithm import heapSort, heapSortNoVisible
from radixSortAlgorithm import radixSort, radixSortNoVisible
from countSortAlgorithm import countSort, countSortNoVisible

class SortingManager:
    def __init__(self):
        self.sortingAlgorithms = {
            "Bubble Sort": (bubbleSort, bubbleSortNoVisible),
            "Bogo Sort": (bogoSort, bogoSortNoVisible),
            "Miracle Sort": (miracleSort, miracleSortNoVisible),
            "Stalin Sort": (stalinSort, stalinSortNoVisible),
            "Selection Sort": (selectionSort, selectionSortNoVisible),
            "Merge Sort": (mergeSort, mergeSortNoVisible),
            "Insertion Sort": (insertionSort, insertionSortNoVisible),
            "Quick Sort": (quickSort, quickSortNoVisible),
            "Cocktail Sort": (cocktailSort, cocktailSortNoVisible),
            "Heap Sort": (heapSort, heapSortNoVisible),
            "Radix Sort": (radixSort, radixSortNoVisible),
            "Count Sort": (countSort, countSortNoVisible),
        }

        self.sortNames = list(self.sortingAlgorithms.keys())
        self.currentSortIndex = 0
        self.algorithmName = self.sortNames[self.currentSortIndex]
        self.generatorFunc = None
        self.resetSort()

    def resetSort(self):
        self.algorithmName = self.sortNames[self.currentSortIndex]
        sortingUtils.algorithmName = self.algorithmName
        sortingUtils.data.clear()
        createTestData(2048)
        sortingUtils.sorting = False
        sortingUtils.comparedIndices.clear()
        sortingUtils.selectedIndices.clear()
        sortingUtils.swaps = 0
        sortingUtils.comparisons = 0
        sortingUtils.colorAnim = None

        sortFunc, noVisualSortFunc = self.sortingAlgorithms[self.algorithmName]

        if self.algorithmName == "Merge Sort" or self.algorithmName == "Quick Sort":
            self.generatorFunc = sortFunc(sortingUtils.data, 0, len(sortingUtils.data) - 1, sortingUtils.delay)
            noVisualSortFunc(sortingUtils.data, 0, len(sortingUtils.data) - 1)
        else:
            self.generatorFunc = sortFunc(sortingUtils.data, sortingUtils.delay)
            noVisualSortFunc(sortingUtils.data)

    def switchSort(self):
        self.currentSortIndex = (self.currentSortIndex + 1) % len(self.sortNames)
        self.resetSort()
        sortingUtils.sorting = False
        sortingUtils.sortTimeVisual = 0