Cborn_Algorithm_Visualizer_v2/data.py at main · DgNShN/Cborn_Algorithm_Visualizer_v2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from utils import generate_random_data, generate_sorted_random_data


class DataManager:
    def __init__(self, size: int):
        self.original_data = generate_random_data(size)
        self.current_data = self.original_data[:]

    def regenerate(self, size: int, sorted_mode: bool = False):
        if sorted_mode:
            self.original_data = generate_sorted_random_data(size)
        else:
            self.original_data = generate_random_data(size)
        self.current_data = self.original_data[:]

    def reset(self):
        self.current_data = self.original_data[:]

    def get_data(self):
        return self.current_data[:]

    def set_data(self, new_data):
        self.current_data = new_data[:]