batframework 1.0.10__py3-none-any.whl → 2.0.0a1__py3-none-any.whl

batFramework/camera.py

@@ -1,123 +1,245 @@
-import batFramework as bf
-import pygame
-from pygame.math import Vector2
-
-
-class Camera:
-    def __init__(self, flags=0) -> None:
-        self.rect = pygame.FRect(0, 0, *bf.const.RESOLUTION)
-        self.flags: int = flags
-        # self.blit_special_flags : int =pygame.BLEND_PREMULTIPLIED if (flags & pygame.SRCALPHA) else 0
-        self.blit_special_flags: int = pygame.BLEND_ALPHA_SDL2
-        self._clear_color : pygame.Color = pygame.Color(0, 0, 0, 0)
-        self.zoom_factor = 1
-        self.cached_surfaces: dict[float, pygame.Surface] = {}
-        self.surface: pygame.Surface = pygame.Surface((0,0))
-        self.follow_point_func = None
-        self.max_zoom = 2
-        self.min_zoom = 0.1
-        self.zoom(1)
-
-    def set_clear_color(self, color: pygame.Color|tuple):
-        if not isinstance(color,pygame.Color):
-            color = pygame.Color(color)
-        self._clear_color = color
-
-    def set_max_zoom(self,value:float):
-        self.max_zoom = value
-
-    def set_min_zoom(self,value:float):
-        self.min_zoom = value
-
-    def clear(self):
-        """
-        Clear the camera surface with the set clear color
-        (default is tranparent)
-        """
-        self.surface.fill(self._clear_color)
-
-    def get_center(self):
-        return self.rect.center
-
-    
-
-    def move(self, x, y):
-        """
-        Moves the camera rect by the given coordinates (+=)
-        """
-        self.rect.topleft += Vector2(x, y)
-
-    def set_position(self, x, y):
-        """
-        Set the camera rect topleft position
-        """
-        self.rect.topleft = (x, y)
-
-    def set_center(self, x, y):
-        """
-        Set the camera rect center position
-        """
-        self.rect.center = (x, y)
-
-    def set_follow_dynamic_point(self, func):
-        """
-        Set the following function (returns tuple x y)
-        (camera will center its position to the center of the given coordinates)
-        """
-        self.follow_point_func = func
-
-    def zoom_by(self, amount:float):
-        self.zoom(self.zoom_factor + amount)
-
-    def zoom(self, factor):
-        if factor <self.min_zoom:
-            return
-        if factor > self.max_zoom:
-            return
-        factor = round(factor,2)
-        self.zoom_factor = factor
-        if factor not in self.cached_surfaces:
-            # if factor != 1 : print("creating new surface in cache : ",tuple(i * factor for i in const.RESOLUTION), self.flags)
-            self.cached_surfaces[factor] = pygame.Surface(
-                tuple(i / factor for i in bf.const.RESOLUTION), flags=self.flags
-            ).convert_alpha()
-            # c = self.surface.get_colorkey() if self.surface else None
-            # if c : self.cached_surfaces[factor].set_colorkey(c)
-            self.cached_surfaces[factor].fill((0, 0, 0, 0))
-
-        self.surface = self.cached_surfaces[self.zoom_factor]
-        self.rect = self.surface.get_frect(center=self.rect.center)
-        # if factor != 1 : print(self.cached_surfaces)
-
-    def intersects(self, rect: pygame.Rect | pygame.FRect) -> bool:
-        return (
-            self.rect.x < rect.right
-            and self.rect.right > rect.x
-            and self.rect.y < rect.bottom
-            and self.rect.bottom > rect.y
-        )
-
-
-    def transpose(self, rect: pygame.Rect | pygame.FRect) -> pygame.Rect | pygame.FRect:
-        x, y = round(rect.x - self.rect.left), round(rect.y - self.rect.top)
-        return pygame.Rect(x, y, *rect.size)
-
-    def convert_screen_to_world(self, x, y):
-        return x / self.zoom_factor + self.rect.x, y / self.zoom_factor + self.rect.y
-
-    def update(self, dt):
-        if self.follow_point_func:
-            target = self.follow_point_func()
-            self.rect.center = Vector2(self.rect.center).lerp(target, ((dt * 60) * 0.1))
-
-    def draw(self, surface: pygame.Surface):
-        # pygame.draw.circle(surface,bf.color.ORANGE,self.surface.get_rect().center,4)
-        if self.zoom_factor == 1:
-            surface.blit(self.surface, (0, 0), special_flags=self.blit_special_flags)
-            return
-        # print("From",self.surface,"Target RESOLUTION",const.RESOLUTION,"Destination",surface)
-        surface.blit(
-            pygame.transform.scale(self.surface, bf.const.RESOLUTION),
-            (0, 0),
-            special_flags=self.blit_special_flags,
-        )
+import pygame
+from pygame.math import Vector2
+import batFramework as bf
+from typing import Self
+import math
+
+
+class Camera:
+    def __init__(
+        self, flags=0, size: tuple[int, int] | None = None, convert_alpha: bool = False,fullscreen:bool=True
+    ) -> None:
+        self.cached_surfaces: dict[tuple[int, int], pygame.Surface] = {}
+        self.fullscreen : bool = fullscreen # auto fill the screen (i.e react to VIDEORESIZE event)
+        self.flags: int = flags | (pygame.SRCALPHA if convert_alpha else 0)
+        self.blit_special_flags: int = pygame.BLEND_ALPHA_SDL2
+
+        size = size if size else bf.const.RESOLUTION
+        self.rect = pygame.FRect(0, 0, *size)
+        self.transform_target_surface : pygame.Surface = pygame.Surface(self.rect.size,self.flags)
+        self.should_convert_alpha: bool = convert_alpha
+        self.world_rect = pygame.FRect(0, 0, *self.rect.size)
+
+        self.vector_center = Vector2(0, 0)
+        self.rotation = 0.0  # Rotation in degrees
+
+        self.surface: pygame.Surface = pygame.Surface((0, 0))  # dynamic : create new at each new zoom value
+
+        self._clear_color: pygame.typing.ColorLike = pygame.Color(0, 0, 0, 0) if convert_alpha else pygame.Color(0, 0, 0)
+
+        self.follow_point_func = None
+        self.damping = float("inf")
+        self.dead_zone_radius = 10
+
+        self.zoom_factor = 1
+        self.max_zoom = 2
+        self.min_zoom = 0.1
+        self.zoom(1,force=True)
+
+    def get_mouse_pos(self) -> tuple[float, float]:
+        return self.screen_to_world(pygame.mouse.get_pos())
+
+    def set_clear_color(self, color: pygame.Color | tuple | str) -> Self:
+        self._clear_color = color
+        return self
+
+    def set_max_zoom(self, value: float) -> Self:
+        self.max_zoom = value
+        return self
+
+    def set_min_zoom(self, value: float) -> Self:
+        self.min_zoom = value
+        return self
+
+    def set_rotation(self, angle: float) -> Self:
+        """
+        Set the camera rotation in degrees.
+        """
+        self.rotation = angle % 360
+        return self
+    
+    def rotate_by(self,angle:float)->Self:
+        """
+        Increment rotation by given angle in degrees.
+        """
+        self.rotation+=angle
+        self.rotation%=360
+        return self
+
+    def clear(self) -> None:
+        if self._clear_color is None:
+            return
+        self.surface.fill(self._clear_color)
+
+    def get_center(self) -> tuple[float, float]:
+        return self.world_rect.center
+
+    def get_position(self) -> tuple[float, float]:
+        return self.world_rect.topleft
+
+    def move_by(self, x: float | int, y: float | int) -> Self:
+        # self.world_rect.move_ip(x, y)
+        self.world_rect.x += x
+        self.world_rect.y += y
+        return self
+
+    def set_position(self, x, y) -> Self:
+        self.world_rect.topleft = (x, y)
+        return self
+
+    def set_center(self, x, y) -> Self:
+        self.world_rect.center = (x, y)
+        return self
+
+    def set_follow_point_func(self, func) -> Self:
+        self.follow_point_func = func
+        return self
+
+    def set_follow_speed(self, speed: float) -> Self:
+        self.follow_speed = speed
+        return self
+
+    def set_follow_damping(self, damping: float) -> Self:
+        self.damping = damping
+        return self
+
+    def set_dead_zone_radius(self, radius: float) -> Self:
+        self.dead_zone_radius = radius
+        return self
+
+    def zoom_by(self, amount: float) -> Self:
+        return self.zoom(self.zoom_factor + amount)
+
+    def zoom(self, factor: float,force:bool=False) -> Self:
+        clamped = max(self.min_zoom, min(self.max_zoom, round(factor, 2)))
+        if clamped == self.zoom_factor and not force:
+            return self
+
+        self.zoom_factor = clamped
+        new_res = tuple([round((i / clamped) / 2) * 2 for i in self.rect.size])
+
+        if self.surface.get_size() != new_res:
+            self.surface = self._get_cached_surface((new_res[0],new_res[1]))
+
+        self.world_rect = self.surface.get_frect(center=self.world_rect.center)
+        self.clear()
+        return self
+
+    def _free_cache(self):
+        self.cached_surfaces.clear()
+
+    def _get_cached_surface(self, new_size: tuple[int, int]):
+        surface = self.cached_surfaces.get(new_size)
+        if surface is None:
+            surface = pygame.Surface(new_size, flags=self.flags)
+            # if self.flags & pygame.SRCALPHA:
+            #     surface = surface.convert_alpha()
+            self.cached_surfaces[new_size] = surface
+        return surface
+
+    def set_size(self, size: tuple[int, int] | None = None) -> Self:
+        if size is None:
+            size = bf.const.RESOLUTION
+        center = self.rect.center
+        self.rect.size = size
+        self.rect.center = center
+        self.transform_target_surface = pygame.Surface(self.rect.size,self.flags)
+        self.world_rect.center = (size[0] / 2, size[1] / 2)
+        self.zoom(self.zoom_factor)
+        return self
+
+    def intersects(self, rect: pygame.Rect | pygame.FRect) -> bool:
+        return self.world_rect.colliderect(rect)
+
+    def world_to_screen(self, rect: pygame.Rect | pygame.FRect) -> pygame.FRect:
+        return pygame.FRect(
+            (rect[0] - self.world_rect.left), (rect[1] - self.world_rect.top), rect[2], rect[3]
+        )
+
+    def world_to_screen_scaled(self, rect: pygame.Rect | pygame.FRect) -> pygame.FRect:
+        screen_rect = self.world_to_screen(rect)
+        return pygame.FRect(
+            screen_rect.x * self.zoom_factor,
+            screen_rect.y * self.zoom_factor,
+            screen_rect.w * self.zoom_factor,
+            screen_rect.h * self.zoom_factor,
+        )
+
+    def world_to_screen_point(self, point: tuple[float, float] | tuple[int, int]) -> tuple[float, float]:
+        return (
+            (point[0] - self.world_rect.x),
+            (point[1] - self.world_rect.y),
+        )
+
+    def world_to_screen_point_scaled(self, point: tuple[float, float] | tuple[int, int]) -> tuple[float, float]:
+        return (
+            (point[0] - self.world_rect.x) * self.zoom_factor,
+            (point[1] - self.world_rect.y) * self.zoom_factor,
+        )
+
+
+    def screen_to_world(self, point: tuple[float, float] | tuple[int, int]) -> tuple[float, float]:
+        """
+        roates, scales and translates point to world coordinates
+
+        """
+        # Center of the screen (zoomed+rotated surface)
+        cx, cy = self.rect.w / 2, self.rect.h / 2
+
+        # Offset from center
+        dx, dy = point[0] - cx, point[1] - cy
+
+        # rotate that offset
+        if self.rotation != 0:
+            angle_rad = math.radians(self.rotation)
+            cos_a = math.cos(angle_rad)
+            sin_a = math.sin(angle_rad)
+            dx, dy = cos_a * dx - sin_a * dy, sin_a * dx + cos_a * dy
+
+        # Un-zoom and add camera position
+        wx = (dx + cx) / self.zoom_factor + self.world_rect.x
+        wy = (dy + cy) / self.zoom_factor + self.world_rect.y
+        return wx, wy
+
+
+    def update(self, dt: float):
+        if not self.follow_point_func or not (math.isfinite(dt) and dt > 0):
+            return
+
+        target = Vector2(self.follow_point_func())
+        self.vector_center.xy.update(self.world_rect.center)
+
+        if self.damping == float("inf"):
+            self.vector_center.xy = target.xy
+        elif math.isfinite(self.damping) and self.damping > 0:
+            damping_factor = 1 - math.exp(-self.damping * dt)
+            if not math.isnan(damping_factor):
+                diff = target - self.vector_center
+                self.vector_center += diff * damping_factor
+
+        self.world_rect.center = self.vector_center
+
+
+    def draw(self, surface: pygame.Surface):
+        """
+        Draw the camera view onto the provided surface with proper scaling and rotation.
+
+        Args:
+            surface (pygame.Surface): Surface to draw the camera view onto.
+        """
+        # Scale the camera surface to the target size
+        if self.zoom_factor == 1 and self.rotation == 0:
+            surface.blit(self.surface, (0, 0), special_flags=self.blit_special_flags)
+            return
+
+        pygame.transform.scale(self.surface, self.rect.size, self.transform_target_surface)
+
+        result_surface = self.transform_target_surface
+
+        if self.rotation != 0:
+            # Rotate around the center of the target surface
+            rotated_surface = pygame.transform.rotate(result_surface, self.rotation)
+            rect = rotated_surface.get_rect(center=(self.rect.w // 2, self.rect.h // 2))
+            surface.blit(rotated_surface, rect.topleft, special_flags=self.blit_special_flags)
+        else:
+            surface.blit(result_surface, (0, 0), special_flags=self.blit_special_flags)

batFramework/constants.py

@@ -1,75 +1,57 @@
-import pygame
-from enum import Enum
-
-
-class Constants:
-    SCREEN = None
-    RESOLUTION: tuple[int, int] = (1280,720)
-    VSYNC = 0
-    FLAGS: int = pygame.SCALED | pygame.RESIZABLE
-    FPS: int = 60
-    RESOURCE_PATH = "."
-    
-    @staticmethod
-    def init_screen(resolution:tuple[int,int],flags:int= 0, vsync:int= 0):
-        Constants.RESOLUTION = resolution
-        Constants.FLAGS = flags
-        Constants.VSYNC = vsync
-        Constants.SCREEN = pygame.display.set_mode(Constants.RESOLUTION, Constants.FLAGS,vsync=Constants.VSYNC)
-        print(f"Window : {resolution[0]}x{resolution[1]}px | flags:{flags.bit_count()}, vsync:{pygame.display.is_vsync()}")
-    MUSIC_END_EVENT = pygame.event.custom_type()
-
-    # ------------GUI SPECIFIC
-    DEFAULT_TEXT_SIZE: int = 8
-
-    GUI_SCALE: int = 1
-    # ---------------------
-
-    @staticmethod
-    def set_resolution(resolution : tuple[int,int]):
-        Constants.RESOLUTION = resolution
-
-    @staticmethod
-    def set_resource_path(path: str):
-        print("set resource path :", path)
-        Constants.RESOURCE_PATH = path
-
-    @staticmethod
-    def set_fps_limit(value: int):
-        Constants.FPS = value
-        print("FPS limit to : ", value)
-
-    @staticmethod
-    def set_gui_scale(value: int):
-        Constants.GUI_SCALE = value
-        print("GUI_SCALE to : ", value)
-
-    @staticmethod
-    def set_default_text_size(size:int):
-        Constants.DEFAULT_TEXT_SIZE = size
-
-class Colors:
-    LIGHT_CYAN = (179, 229, 252)
-    WASHED_BLUE = (52, 73, 94)
-    RIVER_BLUE = (52, 152, 219)
-    DARK_INDIGO = (40, 53, 147)
-    LIGHT_BLUE = (3, 169, 244)
-    DEEP_BLUE = (41, 121, 255)
-    DARK_BLUE = (44, 62, 80)
-    GREEN = (67, 160, 71)
-    DARK_GREEN = (39, 174, 96)
-    BROWN = (109, 76, 65)
-    DARK_RED = (192, 57, 43)
-    ORANGE = (251, 140, 0)
-    CLOUD_WHITE = (236, 240, 241)
-    SILVER = (189, 195, 199)
-    DARK_GRAY = (66, 66, 66)
-
-    DARK_GB = (27, 42, 9)
-    SHADE_GB = (14, 69, 11)
-    BASE_GB = (73, 107, 34)
-    LIGHT_GB = (154, 158, 63)
-
-class Axis(Enum):
-    HORIZONTAL  = 1
-    VERTICAL    = 2
+import pygame
+
+
+class Constants:
+    SCREEN: pygame.Surface = None
+    RESOLUTION: tuple[int, int] = (1280, 720)
+    WIDTH = 1280
+    HEIGHT = 720
+    VSYNC = 0
+    FLAGS: int = pygame.SCALED | pygame.RESIZABLE
+    FPS: int = 60
+    MUSIC_END_EVENT = pygame.event.custom_type()
+
+    DEFAULT_CURSOR = pygame.cursors.Cursor(pygame.SYSTEM_CURSOR_ARROW)
+    DEFAULT_HOVER_CURSOR = pygame.cursors.Cursor(pygame.SYSTEM_CURSOR_ARROW)
+    DEFAULT_CLICK_CURSOR = pygame.cursors.Cursor(pygame.SYSTEM_CURSOR_ARROW)
+
+    BF_INITIALIZED: bool = False
+    ALLOW_DEBUG : bool = True
+
+    WIDGET_KEY_REPEAT_DELAY = 200
+    WIDGET_KEY_REPEAT_INTERVAL = 50
+    GLOBAL_KEY_REPEAT_DELAY = 200
+    GLOBAL_KEY_REPEAT_INTERVAL = 50
+
+
+    @staticmethod
+    def set_allow_debug(allow_debug:bool):
+        Constants.ALLOW_DEBUG = allow_debug
+
+    @staticmethod
+    def set_resolution(resolution: tuple[int, int]):
+        Constants.RESOLUTION = resolution
+        Constants.WIDTH = resolution[0]
+        Constants.HEIGHT = resolution[1]
+        Constants.update_screen()
+
+    @staticmethod
+    def update_screen():
+        if Constants.BF_INITIALIZED:
+            Constants.SCREEN = pygame.display.set_mode(Constants.RESOLUTION,Constants.FLAGS,vsync=Constants.VSYNC)
+
+    @staticmethod
+    def set_default_cursor(cursor: pygame.Cursor):
+        Constants.DEFAULT_CURSOR = cursor
+
+    @staticmethod
+    def set_default_hover_cursor(cursor: pygame.Cursor):
+        Constants.DEFAULT_HOVER_CURSOR = cursor
+
+    @staticmethod
+    def set_default_click_cursor(cursor: pygame.Cursor):
+        Constants.DEFAULT_CLICK_CURSOR = cursor
+
+    @staticmethod
+    def set_fps_limit(value: int):
+        Constants.FPS = value