visualsort

 1from util.module import *
 2from collections.abc import Callable
 3
 4def compare(a: int, b: int, nums: list) -> bool:
 5  """
 6  Compares two numbers and updates the current frame.
 7  
 8  :param a (int): The index of the first number in the array.
 9  :type a: int
10  :param b (int): The index of the second number in the array.
11  :type b: int
12  :param nums (list): The array of numbers.
13  :type nums: list
14  :return: True if the first number is greater than the second number, False otherwise.
15  :rtype: bool
16  :raises IndexError: If either of the indices is out of bounds.
17  """
18  if(a < 0 or a >= len(nums) or b < 0 or b >= len(nums)):
19    raise IndexError("Index out of bounds.")
20  
21  assign_color(a, "green")
22  assign_color(b, "red")
23  return nums[a] > nums[b]
24
25def swap(a: int, b: int, nums: list) -> list:
26  """
27  Swaps two numbers and updates the current frame.
28
29  
30  :param a (int): The index of the first number in the array.
31  :type a: int
32  :param b (int): The index of the second number in the array.
33  :type b: int
34  :param nums (list): The array of numbers.
35  :type nums: list
36  :return: The array of numbers after the swap.
37  :rtype: list
38  :raises IndexError: If either of the indices is out of bounds.
39  """
40  if(a < 0 or a >= len(nums) or b < 0 or b >= len(nums)):
41    raise IndexError("Index out of bounds.")
42  
43  tmp = nums[a]
44  nums[a] = nums[b]
45  nums[b] = tmp
46  add_clip(nums)
47  reset_colors()
48  return nums
49
50def render(algorithm: Callable, video_directory: str = "./", vidoe_name: str = "movie", fps: int = 50) -> None:
51  """
52  Renders the given algorithm to a video.
53
54  :param algorithm (function): The sorting algorithm (writen using the provided visualsort functions) that will be rendered.
55  :type algorithm: function
56  :param video_directory (str): The dirrectory where the video file will be saved.
57  :type video_directory: str
58  :param video_name (str): The diseried name of the video file.
59  :type video_name: str
60  :param fps (int): The frame rate of the video.
61  :type fps: int
62  :raises IndexError: If either of the indices is out of bounds.
63  """
64  nums = generate_numbers()
65  reset_colors()
66  
67  add_clip(nums)
68  
69  algorithm(nums)
70  
71  go_through(nums)
72
73  if not os.path.exists(video_directory):
74    os.mkdir(video_directory)
75
76  final_clip = ImageSequenceClip(clips, fps=fps)
77
78  final_clip.write_videofile(os.path.join(video_directory, vidoe_name + ".mp4"), fps=fps)