batframework 1.0.8a2__py3-none-any.whl → 1.0.8a3__py3-none-any.whl

batFramework/animatedSprite.py

@@ -2,7 +2,6 @@ import batFramework as bf
 import pygame
 
 
-
 def search_index(target, lst):
     cumulative_sum = 0
     for index, value in enumerate(lst):
@@ -13,105 +12,156 @@ def search_index(target, lst):
 
 
 class AnimState:
-    def __init__(self, file, width, height, frame_length_list:list|int) -> None:
-        self.frames: list[pygame.Surface] = bf.utils.img_slice(file, width, height)
-        self.frames_flipX: list[pygame.Surface] = bf.utils.img_slice(
-            file, width, height, True
+    def __init__(
+        self,
+        name: str,
+        surface: pygame.Surface,
+        width,
+        height,
+        duration_list: list | int,
+    ) -> None:
+        self.frames: list[pygame.Surface] = list(
+            bf.utils.split_surface(
+                surface, width, height, False, convert_alpha
+            ).values()
+        )
+        self.frames_flipX: list[pygame.Surface] = list(
+            bf.utils.split_surface(surface, width, height, True, convert_alpha).values()
         )
 
-        self.frame_length_list = []
-        self.ffl_length = 0
-        self.set_frame_length_list(frame_length_list)
+        self.name = name
+        self.duration_list: list[int] = []
+        self.duration_list_length = 0
+        self.set_duration_list(duration_list)
 
-    def get_frame_index(self, counter:float|int):
-        return search_index(int(counter % self.ffl_length), self.frame_length_list)
+    def __repr__(self):
+        return f"AnimState({self.name})"
+
+    def counter_to_frame(self, counter: float | int) -> int:
+        return search_index(
+            int(counter % self.duration_list_length), self.duration_list
+        )
 
     def get_frame(self, counter, flip):
-        i = self.get_frame_index(counter)
+        i = self.counter_to_frame(counter)
         return self.frames_flipX[i] if flip else self.frames[i]
 
-    def set_frame_length_list(self,frame_length_list:list[int]|int):
-        if isinstance(frame_length_list,int):
-            frame_length_list = [frame_length_list] * len(self.frames)
-        if len(frame_length_list) != len(self.frames) : 
-            raise ValueError("frame_length_list should have values for all frames")
-        self.frame_length_list = frame_length_list
-        self.ffl_length = sum(self.frame_length_list)
-        
+    def set_duration_list(self, duration_list: list[int] | int):
+        if isinstance(duration_list, int):
+            duration_list = [duration_list] * len(self.frames)
+        if len(duration_list) != len(self.frames):
+            raise ValueError("duration_list should have values for all frames")
+        self.duration_list = duration_list
+        self.duration_list_length = sum(self.duration_list)
+
 
 class AnimatedSprite(bf.DynamicEntity):
     def __init__(self, size=None) -> None:
         super().__init__(size, no_surface=True)
-        self.float_counter = 0
+        self.float_counter: float = 0
         self.animStates: dict[str, AnimState] = {}
-        self.current_animState :str = ""
+        self.current_state: AnimState | None = None
         self.flipX = False
         self._locked = False
+        self.paused: bool = False
+
+    def pause(self) -> None:
+        self.paused = True
+
+    def resume(self) -> None:
+        self.paused = False
 
-    def set_counter(self,value:float):
+    def toggle_pause(self) -> None:
+        self.paused = not self.paused
+
+    def set_counter(self, value: float) -> None:
         self.float_counter = value
-        
 
-    def lock_animState(self):
+    def set_frame(self, frame_index: int) -> None:
+        if not self.current_state:
+            return
+        total = sum(self.current_state.duration_list)
+        frame_index = max(0, min(total, frame_index))
+        new_counter = 0
+        i = 0
+        while frame_index < total:
+            if self.current_state.counter_to_frame(new_counter) >= frame_index:
+                break
+            new_counter += self.current_state.duration_list[i]
+            i += 1
+        self.set_counter(new_counter)
+
+    def lock(self) -> None:
         self._locked = True
 
-    def unlock_animState(self):
+    def unlock(self) -> None:
         self._locked = False
 
-    def set_flipX(self, value):
+    def set_flipX(self, value) -> None:
         self.flipX = value
 
-    def remove_animState(self, name:str):
-        if not name in self.animStates :return
-        self.animStates.pop(name) 
-        if self.current_animState == name : self.current_animState = list(self.animStates.keys())[0] if self.animStates else "" 
+    def remove_animState(self, name: str) -> bool:
+        if not name in self.animStates:
+            return False
+        self.animStates.pop(name)
+        if self.current_state and self.current_state.name == name:
+            self.current_animState = (
+                list(self.animStates.keys())[0] if self.animStates else ""
+            )
+        return True
 
     def add_animState(
-        self, name: str, file: str, size: tuple[int, int], frame_length_list: list[int]
-    ):
+        self,
+        name: str,
+        surface: pygame.Surface,
+        size: tuple[int, int],
+        duration_list: list[int],
+        convert_alpha: bool = True,
+    ) -> bool:
         if name in self.animStates:
-            return
-        self.animStates[name] = AnimState(file, *size, frame_length_list)
-        if len(self.animStates) == 1 : self.set_animState(name)
+            return False
+        self.animStates[name] = AnimState(name, surface, *size, duration_list)
+        if len(self.animStates) == 1:
+            self.set_animState(name)
+        return True
 
-    def set_animState(self, state:str, reset_counter=True, lock=False):
+    def set_animState(self, state: str, reset_counter=True, lock=False) -> bool:
         if state not in self.animStates or self._locked:
             return False
-        self.current_animState = state
-        self.rect = (
-            self.animStates[self.current_animState]
-            .frames[0]
-            .get_frect(center=self.rect.center)
-        )
-        if reset_counter or self.float_counter > sum(
-            self.get_state().frame_length_list
-        ):
+
+        animState = self.animStates[state]
+        self.current_state = animState
+
+        self.rect = self.current_state.frames[0].get_frect(center=self.rect.center)
+
+        if reset_counter or self.float_counter > sum(animState.duration_list):
             self.float_counter = 0
         if lock:
-            self.lock_animState()
+            self.lock()
         return True
 
-    def get_state(self):
-        return self.animStates.get(self.current_animState,None)
+    def get_animState(self) -> AnimState | None:
+        return self.current_state
 
-    def get_frame_index(self):
-        return self.animStates[self.current_animState].get_frame_index(
-            self.float_counter
-        )
-
-    def update(self, dt: float):
-        if not self.animStates : return
-        self.float_counter += 60 * dt
-        if self.float_counter > self.get_state().ffl_length:
-            self.float_counter = 0
+    def update(self, dt: float) -> None:
+        s = self.get_animState()
+        if not self.animStates or s is None:
+            return
+        if not self.paused:
+            self.float_counter += 60 * dt
+            if self.float_counter > s.duration_list_length:
+                self.float_counter = 0
         self.do_update(dt)
 
-    def draw(self, camera: bf.Camera) -> bool:
-        if not self.visible or not camera.intersects(self.rect) or not self.animStates:
-            return False
-        # pygame.draw.rect(camera.surface,"purple",camera.transpose(self.rect).move(2,2))
+    def draw(self, camera: bf.Camera) -> int:
+        if (
+            not self.visible
+            or not camera.rect.colliderect(self.rect)
+            or not self.current_state
+        ):
+            return 0
         camera.surface.blit(
-            self.get_state().get_frame(self.float_counter, self.flipX),
-            camera.transpose(self.rect),
+            self.current_state.get_frame(self.float_counter, self.flipX),
+            camera.world_to_screen(self.rect),
         )
-        return True
+        return 1

batFramework/audioManager.py

@@ -5,24 +5,25 @@ pygame.mixer.init()
 
 
 class AudioManager(metaclass=bf.Singleton):
-    def __init__(self):
-        self.sounds: dict[str : dict[str, pygame.mixer.Sound, bool]] = {}
-        self.musics: dict[str:str] = {}
-        self.current_music = None
-        self.music_volume = 1
-        self.sound_volume = 1
+    def __init__(self) -> None:
+        self.sounds: dict = {}
+        self.musics: dict = {}
+        self.current_music: str | None = None
+        self.music_volume: float = 1
+        self.sound_volume: float = 1
         pygame.mixer_music.set_endevent(bf.const.MUSIC_END_EVENT)
 
-    def free_sounds(self, force=False):
+    def free_sounds(self, force: bool = False):
         if force:
             self.sounds = {}
             return
-        to_remove = []
-        for name, data in self.sounds.items():
-            if not data["persistent"]:
-                to_remove.append(name)
+        self.sounds: dict = {
+            key: value for key, value in self.sounds.items() if value["persistent"]
+        }
 
-        _ = [self.sounds.pop(i) for i in to_remove]
+    def free_music(self):
+        if self.current_music:
+            pygame.mixer.music.unload(self.current_music)
 
     def set_sound_volume(self, volume: float):
         self.sound_volume = volume
@@ -31,13 +32,19 @@ class AudioManager(metaclass=bf.Singleton):
         self.music_volume = volume
         pygame.mixer_music.set_volume(volume)
 
-    def has_sound(self, name):
+    def get_music_volume(self) -> float:
+        return self.music_volume
+
+    def get_sound_volume(self) -> float:
+        return self.sound_volume
+
+    def has_sound(self, name: str):
         return name in self.sounds
 
     def load_sound(self, name, path, persistent=False) -> pygame.mixer.Sound:
         if name in self.sounds:
             return self.sounds[name]["sound"]
-        path = bf.utils.get_path(path)
+        path = bf.ResourceManager().get_path(path)
         self.sounds[name] = {
             "path": path,
             "sound": pygame.mixer.Sound(path),
@@ -45,24 +52,57 @@ class AudioManager(metaclass=bf.Singleton):
         }
         return self.sounds[name]["sound"]
 
-    def play_sound(self, name, volume=1):
-        self.sounds[name]["sound"].set_volume(volume * self.sound_volume)
-        self.sounds[name]["sound"].play()
-
-    def stop_sound(self, name):
-        if name in self.sounds:
+    def load_sounds(self, sound_data_list: list[tuple[str, str, bool]]) -> None:
+        for data in sound_data_list:
+            self.load_sound(*data)
+        return
+
+    def play_sound(self, name, volume=1) -> bool:
+        """
+        Play the sound file with the given name.
+        returns True if the sound was played
+
+        """
+        try:
+            self.sounds[name]["sound"].set_volume(volume * self.sound_volume)
+            self.sounds[name]["sound"].play()
+            return True
+        except KeyError:
+            # print(f"Sound '{name}' not loaded in AudioManager.")
+            return False
+
+    def stop_sound(self, name) -> bool:
+        try:
             self.sounds[name]["sound"].stop()
+            return True
+        except KeyError:
+            return False
+            # print(f"Sound '{name}' not loaded in AudioManager.")
 
     def load_music(self, name, path):
-        self.musics[name] = bf.utils.get_path(path)
-
-    def play_music(self, name, loop=0, fade=500):
-        if name in self.musics:
+        self.musics[name] = bf.ResourceManager().get_path(path)
+        return
+
+    def load_musics(self, music_data_list: list[tuple[str, str]]):
+        for data in music_data_list:
+            self.load_music(*data)
+        return
+
+    def play_music(self, name, loop=0, fade=500) -> bool:
+        """
+        Play the sound file with the given 'name'.
+        Fades with the given 'fade' time in ms.
+        Music will loop 'loop' times (indefinitely if -1).
+        returns True if the sound was played
+        """
+        try:
             pygame.mixer_music.load(self.musics[name])
             pygame.mixer_music.play(loop, fade_ms=fade)
             self.current_music = name
-        else:
-            print(f"Music '{name}' not found in AudioManager.")
+            return True
+        except KeyError:
+            return False
+            # print(f"Music '{name}' not loaded in AudioManager.")
 
     def stop_music(self):
         if not self.current_music:
@@ -83,3 +123,6 @@ class AudioManager(metaclass=bf.Singleton):
         if not self.current_music:
             return
         pygame.mixer_music.unpause()
+
+    def get_current_music(self) -> str | None:
+        return self.current_music