Added audio and end sorted pass through, highlighting every bar

Author: pythonioncoder

.idea/Sorting-Alg-Visualizations.iml renamed to .idea/DSA-Visualizations.iml

@@ -1,5 +1,30 @@
 import matplotlib.pyplot as plt
 import numpy as np
+import pygame
+import time
+
+pygame.mixer.init()
+
+
+def play_tone(frequency, duration):
+	sample_rate = 44100  # Hertz
+	n_samples = int(sample_rate * duration)
+
+	# Create a sine wave
+	t = np.linspace(0, duration, n_samples, False)
+	wave = 0.5 * np.sin(2 * np.pi * frequency * t)
+
+	# Convert to 16-bit audio data
+	wave = (wave * 32767).astype(np.int16)
+
+	# Duplicate for stereo if needed
+	wave = np.column_stack((wave, wave))  # Convert to 2 channels (stereo)
+
+	# Make and play sound
+	sound = pygame.sndarray.make_sound(wave)
+	sound.play(-1)
+	time.sleep(duration)
+	sound.stop()
 
 
 class Vis:
@@ -19,14 +44,14 @@ def __init__(self, a=50, barcol='white'):
 		self.lst = np.array(self.lst)
 
 	def display(self):
-		if self.end:
-			plt.close()
-			return
 		plt.clf()
 		plt.gcf().set_facecolor('black')
 		plt.bar(self.x, self.lst, color=self.highlights)
 		plt.axis('off')
 		plt.pause(0.0001)
+		if self.end:
+			plt.close()
+			return
 
 
 def bubble_sort(current: Vis):
@@ -39,8 +64,14 @@ def bubble_sort(current: Vis):
 				unsorted[j] = temp
 				current.highlights[j] = 'green'
 				current.display()
+				play_tone(unsorted[j]*50+100, 0.01)
 				current.highlights[j] = current.barcol
 
+	for i in range(len(unsorted)):
+		current.highlights[i] = 'green'
+		current.display()
+		play_tone(unsorted[i]*50+100, 0.01)
+
 	current.end = True
 	current.display()
 

.idea/modules.xml

@@ -1,5 +1,30 @@
 import matplotlib.pyplot as plt
 import numpy as np
+import pygame
+import time
+
+pygame.mixer.init()
+
+
+def play_tone(frequency, duration):
+	sample_rate = 44100  # Hertz
+	n_samples = int(sample_rate * duration)
+
+	# Create a sine wave
+	t = np.linspace(0, duration, n_samples, False)
+	wave = 0.5 * np.sin(2 * np.pi * frequency * t)
+
+	# Convert to 16-bit audio data
+	wave = (wave * 32767).astype(np.int16)
+
+	# Duplicate for stereo if needed
+	wave = np.column_stack((wave, wave))  # Convert to 2 channels (stereo)
+
+	# Make and play sound
+	sound = pygame.sndarray.make_sound(wave)
+	sound.play(-1)
+	time.sleep(duration)
+	sound.stop()
 
 
 class Vis:
@@ -19,14 +44,14 @@ def __init__(self, a=50, barcol='white'):
 		self.lst = np.array(self.lst)
 
 	def display(self):
-		if self.end:
-			plt.close()
-			return
 		plt.clf()
 		plt.gcf().set_facecolor('black')
 		plt.bar(self.x, self.lst, color=self.highlights)
 		plt.axis('off')
 		plt.pause(0.0001)
+		if self.end:
+			plt.close()
+			return
 
 
 def insertion_sort(current_vis: Vis):
@@ -40,8 +65,14 @@ def insertion_sort(current_vis: Vis):
 			j -= 1
 			current_vis.highlights[j] = 'green'
 			current_vis.display()
+			play_tone(unsorted[j]*50+100, 0.01)
 			current_vis.highlights[j] = current_vis.barcol
 
+	for i in range(len(unsorted)):
+		current_vis.highlights[i] = 'green'
+		current_vis.display()
+		play_tone(unsorted[i]*50+100, 0.01)
+
 	current_vis.end = True
 	current_vis.display()
 

BubbleSortVis.py

@@ -1,5 +1,30 @@
 import matplotlib.pyplot as plt
 import numpy as np
+import pygame
+import time
+
+pygame.mixer.init()
+
+
+def play_tone(frequency, duration):
+	sample_rate = 44100  # Hertz
+	n_samples = int(sample_rate * duration)
+
+	# Create a sine wave
+	t = np.linspace(0, duration, n_samples, False)
+	wave = 0.5 * np.sin(2 * np.pi * frequency * t)
+
+	# Convert to 16-bit audio data
+	wave = (wave * 32767).astype(np.int16)
+
+	# Duplicate for stereo if needed
+	wave = np.column_stack((wave, wave))  # Convert to 2 channels (stereo)
+
+	# Make and play sound
+	sound = pygame.sndarray.make_sound(wave)
+	sound.play(-1)
+	time.sleep(duration)
+	sound.stop()
 
 
 class Vis:
@@ -19,14 +44,14 @@ def __init__(self, a=100, barcol='white'):
 		self.lst = np.array(self.lst)
 
 	def display(self):
-		if self.end:
-			plt.close()
-			return
 		plt.clf()
 		plt.gcf().set_facecolor('black')
 		plt.bar(self.x, self.lst, color=self.highlights)
 		plt.axis('off')
 		plt.pause(0.0001)
+		if self.end:
+			plt.close()
+			return
 
 
 def _merge_sort(current: Vis, lo, hi):
@@ -45,28 +70,36 @@ def _merge_sort(current: Vis, lo, hi):
 			l += 1
 			current.highlights[lo+l+r] = 'green'
 			current.display()
+			play_tone(left[l-1]*50+100, 0.01)
 			current.highlights[lo + l + r] = current.barcol
 		else:
 			current.lst[lo+l+r] = right[r]
 			r += 1
 			current.highlights[lo + l + r] = 'green'
 			current.display()
+			play_tone(right[r-1]*50+100, 0.01)
 			current.highlights[lo + l + r] = current.barcol
 
 	if l < len(left):
 		for i in range(len(left[l:])):
 			current.lst[lo+l+r+i] = left[l+i]
 			current.highlights[lo + l + r] = 'green'
 			current.display()
+			play_tone(left[l+i]*50+100, 0.01)
 			current.highlights[lo + l + r] = current.barcol
 	elif r < len(right):
 		for i in range(len(right[r:])):
 			current.lst[lo+l+r+i] = right[r+i]
 			current.highlights[lo + l + r] = 'green'
 			current.display()
+			play_tone(right[r+i]*50+100, 0.01)
 			current.highlights[lo + l + r] = current.barcol
 
 	if lo == 0 and hi == len(current.lst):
+		for i in range(len(current.lst)):
+			current.highlights[i] = 'green'
+			current.display()
+			play_tone(current.lst[i]*50+100, 0.01)
 		current.end = True
 		current.display()
 

InsertionSortVis.py

@@ -1,5 +1,30 @@
 import matplotlib.pyplot as plt
 import numpy as np
+import pygame
+import time
+
+pygame.mixer.init()
+
+
+def play_tone(frequency, duration):
+	sample_rate = 44100  # Hertz
+	n_samples = int(sample_rate * duration)
+
+	# Create a sine wave
+	t = np.linspace(0, duration, n_samples, False)
+	wave = 0.5 * np.sin(2 * np.pi * frequency * t)
+
+	# Convert to 16-bit audio data
+	wave = (wave * 32767).astype(np.int16)
+
+	# Duplicate for stereo if needed
+	wave = np.column_stack((wave, wave))  # Convert to 2 channels (stereo)
+
+	# Make and play sound
+	sound = pygame.sndarray.make_sound(wave)
+	sound.play(-1)
+	time.sleep(duration)
+	sound.stop()
 
 
 class Vis:
@@ -19,14 +44,14 @@ def __init__(self, a=100, barcol='white'):
 		self.lst = np.array(self.lst)
 
 	def display(self):
-		if self.end:
-			plt.close()
-			return
 		plt.clf()
 		plt.gcf().set_facecolor('black')
 		plt.bar(self.x, self.lst, color=self.highlights)
 		plt.axis('off')
 		plt.pause(0.0001)
+		if self.end:
+			plt.close()
+			return
 
 
 def _quick_sort(current: Vis, lo, hi):
@@ -63,11 +88,16 @@ def _quick_sort(current: Vis, lo, hi):
 		unsorted[j] = temp
 
 		current.display()
+		play_tone((unsorted[i] + unsorted[j]) * 50 / 2 + 100, 0.01)
 		current.highlights[lo+i], current.highlights[lo+j], current.highlights[lo+pivot_index] = current.barcol, current.barcol, current.barcol
 
 	res = _quick_sort(current, lo, lo+i) + _quick_sort(current, lo+i, hi)
 
 	if lo == 0 and hi == len(current.lst):
+		for i in range(len(current.lst)):
+			current.highlights[i] = 'green'
+			current.display()
+			play_tone(current.lst[i]*50+100, 0.01)
 		current.end = True
 		current.display()
 

MergeSortVis.py

@@ -1,5 +1,30 @@
 import matplotlib.pyplot as plt
 import numpy as np
+import pygame
+import time
+
+pygame.mixer.init()
+
+
+def play_tone(frequency, duration):
+	sample_rate = 44100  # Hertz
+	n_samples = int(sample_rate * duration)
+
+	# Create a sine wave
+	t = np.linspace(0, duration, n_samples, False)
+	wave = 0.5 * np.sin(2 * np.pi * frequency * t)
+
+	# Convert to 16-bit audio data
+	wave = (wave * 32767).astype(np.int16)
+
+	# Duplicate for stereo if needed
+	wave = np.column_stack((wave, wave))  # Convert to 2 channels (stereo)
+
+	# Make and play sound
+	sound = pygame.sndarray.make_sound(wave)
+	sound.play(-1)
+	time.sleep(duration)
+	sound.stop()
 
 
 class Vis:
@@ -19,14 +44,14 @@ def __init__(self, a=50, barcol='white'):
 		self.lst = np.array(self.lst)
 
 	def display(self):
-		if self.end:
-			plt.close()
-			return
 		plt.clf()
 		plt.gcf().set_facecolor('black')
 		plt.bar(self.x, self.lst, color=self.highlights)
 		plt.axis('off')
 		plt.pause(0.0001)
+		if self.end:
+			plt.close()
+			return
 
 
 def selection_sort(current: Vis):
@@ -39,13 +64,19 @@ def selection_sort(current: Vis):
 			current.highlights[min_index] = 'red'
 			current.highlights[j] = 'green'
 			current.display()
+			play_tone(unsorted[j]*50+100, 0.01)
 			current.highlights[min_index], current.highlights[j] = current.barcol, current.barcol
 
 		if i != min_index:
 			temp = unsorted[i]
 			unsorted[i] = unsorted[min_index]
 			unsorted[min_index] = temp
 
+	for i in range(len(unsorted)):
+		current.highlights[i] = 'green'
+		current.display()
+		play_tone(unsorted[i]*50+100, 0.01)
+
 	current.end = True
 	current.display()