Merge pull request #87 from FNBBDevs/dev

Dev

.github/workflows/documentation.yml

@@ -6,9 +6,9 @@ permissions:
 
 on:
   push:
-    branches: [ dev, main ]
+    branches: [ main ]
   pull_request:
-    branches: [ dev, main ]
+    branches: [ main ]
   workflow_dispatch:
 
 jobs:

bruhanimate/bruheffect/audio_effect.py

@@ -22,12 +22,25 @@
 
     from bruhcolor import bruhcolored
     from .base_effect import BaseEffect
+    from ..bruhutil import Buffer
 
 
     class AudioEffect(BaseEffect):
+        """
+        A base class for audio effects in a terminal-based visualizer.  
+        """
         def __init__(
-            self, buffer, background: str, num_bands: int = 24, audio_halt: int = 10
+            self, buffer: Buffer, background: str, num_bands: int = 24, audio_halt: int = 10
         ):
+            """
+            Initialize the AudioEffect class.
+
+            Args:
+                buffer (Buffer): The effect buffer to push updates to.
+                background (str): Character or string to use for the background.
+                num_bands (int, optional): Number of EQ bands to show. Defaults to 24.
+                audio_halt (int, optional): How often we should update the bands, (frame_number % halt == 0 then update). Defaults to 10.
+            """
             super(AudioEffect, self).__init__(buffer, background)
             self.FORMAT = pyaudio.paInt16
             self.CHANNELS = 1
@@ -78,14 +91,33 @@ def __init__(
         def set_audio_properties(
             self, num_bands=24, audio_halt=10, use_gradient=True, non_gradient_color=27
         ):
+            """
+            Set the properties for the AudioEffect class.
+
+            Args:
+                num_bands (int, optional): Number of EQ bands to show. Defaults to 24.
+                audio_halt (int, optional): How often we should update the bands, (frame_number % halt == 0 then update). Defaults to 10.
+                use_gradient (bool, optional): Whether or not to use color gradient. Defaults to True.
+                non_gradient_color (int, optional): Color to use if not a gradient. Defaults to 27 (blue).
+            """
             self.BANDS = num_bands
             self.audio_halt = audio_halt
             self.band_ranges = self.generate_even_ranges(self.BANDS, 0, self.buffer.width())
             self.colors = [random.randint(0, 255) for _ in range(self.BANDS)]
             self.use_gradient = use_gradient
             self.non_gradient_color = non_gradient_color
 
-        def evenly_distribute_original_values(self, original_list, desired_width):
+        def evenly_distribute_original_values(self, original_list: list[int], desired_width: int):
+            """
+            Evenly distribute the values in a list to fit within a certain width.
+
+            Args:
+                original_list (list[int]): The list of values to distribute.
+                desired_width (int): The width to which the values should be distributed.
+
+            Returns:
+                list[int]: The evenly distributed list of values.
+            """
             repeat_count = desired_width // len(original_list)
             extra_elements = desired_width % len(original_list)
             expanded_list = []
@@ -96,7 +128,13 @@ def evenly_distribute_original_values(self, original_list, desired_width):
                     extra_elements -= 1
             return expanded_list
 
-        def set_orientation(self, orientation):
+        def set_orientation(self, orientation: str):
+            """
+            Set the orientation of the visualizer.
+
+            Args:
+                orientation (str): The orientation to set ("top" or "bottom").
+            """
             if orientation in ["top", "bottom"]:
                 self.orientation = orientation
                 if self.orientation == "bottom":
@@ -109,6 +147,13 @@ def set_audio_gradient(
             gradient=[232, 233, 235, 237, 239, 241, 243, 245, 247, 249, 251, 253, 255],
             mode="extend",
         ):
+            """
+            Set the audio gradient for visualizing audio data.
+
+            Args:
+                gradient (list, optional): List of colors to use for gradient. Defaults to [232, 233, 235, 237, 239, 241, 243, 245, 247, 249, 251, 253, 255].
+                mode (str, optional): Do we want to repeat the gradient or extend the list evenly. Defaults to "extend".
+            """
             self.base_gradient = gradient
             self.gradient_mode = mode
             if self.gradient_mode == "repeat":
@@ -129,6 +174,18 @@ def set_audio_gradient(
                 ]
 
         def process_audio(self, data, frame_count, time_info, status):
+            """
+            Process the audio data and update the visualizer buffer.
+
+            Args:
+                data (_type_): Audio data from the stream.
+                frame_count (_type_): N/A
+                time_info (_type_): N/A
+                status (_type_): N/A
+
+            Returns:
+                tuple: (data, pyaudio.paContinue)
+            """
             audio_array = np.frombuffer(data, dtype=np.int16)
             fft_result = np.fft.rfft(audio_array)
             magnitudes = np.abs(fft_result)
@@ -139,15 +196,35 @@ def process_audio(self, data, frame_count, time_info, status):
             ]
             return (data, pyaudio.paContinue)
 
-        def map_bands_to_range(self, N):
+        def map_bands_to_range(self, N: int):
+            """
+            Map the bands to a range of values between 0 and N-1.
+
+            Args:
+                N (int): The range of values to map the bands to.
+
+            Returns:
+                list[int]: The mapped bands.
+            """
             min_band = min(self.bands)
             max_band = max(self.bands)
             rand_band = max_band - min_band if max_band != min_band else 1
             normalized_bands = [(band - min_band) / rand_band for band in self.bands]
             scaled_bands = [int(band * N) for band in normalized_bands]
             return scaled_bands
 
-        def generate_even_ranges(self, groups, start, end):
+        def generate_even_ranges(self, groups: list[any], start: int, end: int):
+            """
+            Generate even ranges from a list of groups.
+
+            Args:
+                groups (list[any]): The list of groups to generate ranges from.
+                start (int): The starting index of the ranges.
+                end (int): The ending index of the ranges.
+
+            Returns:
+                _type_: _description_
+            """
             approximate_group_size = round((end - start) / groups)
             intervals = []
             for i in range(groups):
@@ -156,7 +233,13 @@ def generate_even_ranges(self, groups, start, end):
                 intervals.append((group_start, min(group_end, end)))
             return intervals
 
-        def render_frame(self, frame_number):
+        def render_frame(self, frame_number: int):
+            """
+            Render a single frame of the animation.
+
+            Args:
+                frame_number (int): The current frame number of the animation.
+            """
             if frame_number == 0:
                 self.stream.start_stream()
 

bruhanimate/bruheffect/base_effect.py

@@ -15,20 +15,28 @@
 """
 
 from abc import abstractmethod
+from ..bruhutil.bruhffer import Buffer
 
 
 class BaseEffect:
     """
     Class for keeping track of an effect, and updataing it's buffer
     """
 
-    def __init__(self, buffer, background):
+    def __init__(self, buffer: Buffer, background: str):
+        """
+        Base class for all effects.
+
+        Args:
+            buffer (Buffer): Effect buffer to push updates to.
+            background (str): character or string to use for the background.
+        """
         self.buffer = buffer
         self.background = background
         self.background_length = len(background)
 
     @abstractmethod
     def render_frame(self, frame_number):
         """
-        To be defined by each effect
+        To be defined by each effect.
         """