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

batFramework/gui/widget.py

@@ -1,307 +1,382 @@
-from __future__ import annotations
-from typing import TYPE_CHECKING
-if TYPE_CHECKING:
-    from .constraints import Constraint
-    from .root import Root
-from typing import Self
-
+from typing import TYPE_CHECKING, Self, Callable
+from collections.abc import Iterable
 import batFramework as bf
 import pygame
-from math import ceil
-
-
-
-class Widget(bf.Entity):
-    def __init__(self,convert_alpha=True)->None:
-        super().__init__(convert_alpha=convert_alpha)
-        self.autoresize = False
-        self.parent : None|Self = None 
-        self.is_root :bool = False
-        self.children : list["Widget"] = []
-        self.focusable :bool= False
-        self.constraints : list[Constraint] = []
-        self.gui_depth : int = 0
-        if self.surface : self.surface.fill("white")
-        self.set_debug_color("red")
-        self.padding :tuple[float|int,...]= (0,0,0,0) 
-
-    def set_padding(self,value : float|int|tuple|list)->Self:
-        old_raw_size = (
-            self.rect.w - self.padding[0] - self.padding[2],
-            self.rect.h - self.padding[1] - self.padding[3]
-        )
-        if isinstance(value,list) or isinstance(value,tuple):
-            if len(value) > 4 : return self
-            if any(v<0 for v in value) : return self
-            if len(value) == 2:
-                self.padding = (value[0],value[1],value[0],value[1])
-            else:
-                self.padding = (*value, *self.padding[len(value):])
-        else:
-            self.padding = (value,)*4
 
-        self.set_size(
-            old_raw_size[0] + self.padding[0] +  self.padding[2],
-            old_raw_size[1] + self.padding[1] +  self.padding[3],
-        )
-        return self
-        
-    def inflate_rect_by_padding(self,rect:pygame.FRect)->pygame.FRect:
-        return pygame.FRect(
-            rect[0] - self.padding[0],
-            rect[1]  - self.padding[1],
-            rect[2] + self.padding[0]+self.padding[2],
-            rect[3] + self.padding[1]+self.padding[3]   
+if TYPE_CHECKING:
+    from .constraints.constraints import Constraint
+    from .root import Root
+MAX_CONSTRAINTS = 10
+
+
+class WidgetMeta(type):
+    def __call__(cls, *args, **kwargs):
+        """Called when you call MyNewClass()"""
+        obj = type.__call__(cls, *args, **kwargs)
+        # obj.new_init()
+        bf.StyleManager().register_widget(obj)
+        return obj
+
+
+class Widget(bf.Entity, metaclass=WidgetMeta):
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+        self.children: list["Widget"] = []
+        self.constraints: list[Constraint] = []
+        self.parent: "Widget" = None
+        self.do_sort_children = False
+        self.clip_children: bool = True
+        self.padding = (0, 0, 0, 0)
+        self.dirty_surface: bool = True  #  if true will call paint before drawing
+        self.dirty_shape: bool = (
+            True  #    if true will call (build+paint) before drawing
         )
-    def get_content_left(self)->float:
-        return self.rect.left + self.padding[0]
+        self.dirty_constraints: bool = False
+        self.is_root: bool = False
+        self.autoresize_w, self.autoresize_h = True, True
+        self.__constraint_iteration = 0
 
-    def get_content_top(self)->float:
-        return self.rect.top + self.padding[1]
+    def show(self) -> Self:
+        self.visit(lambda w: w.set_visible(True))
+        return self
 
-    def get_content_right(self)->float:
-        return self.rect.right - self.padding[2]
+    def hide(self) -> Self:
+        self.visit(lambda w: w.set_visible(False))
+        return self
 
-    def get_content_bottom(self)->float:
-        return self.rect.bottom - self.padding[3]
+    def set_clip_children(self, value: bool) -> Self:
+        self.clip_children = value
+        self.dirty_surface = True
+        return self
 
-    def get_content_width(self)->float:
-        return self.rect.w - self.padding[0] - self.padding[2]
+    def __str__(self) -> str:
+        return "Widget"
 
-    def get_content_height(self)->float:
-        return self.rect.h - self.padding[1] - self.padding[3]
+    def set_autoresize(self, value: bool) -> Self:
+        self.autoresize_w = self.autoresize_h = value
+        self.dirty_shape = True
+        return self
 
-    def get_content_rect(self)->pygame.FRect:
-        return pygame.FRect(
-            self.rect.left      + self.padding[0],
-            self.rect.top       + self.padding[1],
-            self.get_content_width(),
-            self.get_content_height()
-        )
+    def set_autoresize_w(self, value: bool) -> Self:
+        self.autoresize_w = value
+        self.dirty_shape = True
+        return self
 
-    def get_content_rect_rel(self)->pygame.FRect:
-        return self.get_content_rect().move(-self.rect.left,-self.rect.top)
+    def set_autoresize_h(self, value: bool) -> Self:
+        self.autoresize_h = value
+        self.dirty_shape = True
+        return self
 
-    def get_content_center(self)->tuple[float,float]:
-        return self.get_content_rect().center
+    def set_render_order(self, render_order: int) -> Self:
+        super().set_render_order(render_order)
+        if self.parent:
+            self.parent.do_sort_children = True
+        return self
 
-    def get_depth(self)->int:
-        if self.is_root or self.parent is None : 
-            self.gui_depth = 0
-        else:
-            self.gui_depth =  self.parent.get_depth() + 1
-        return self.gui_depth
+    def inflate_rect_by_padding(
+        self, rect: pygame.Rect | pygame.FRect
+    ) -> pygame.Rect | pygame.FRect:
+        return pygame.FRect(
+            rect[0] - self.padding[0],
+            rect[1] - self.padding[1],
+            rect[2] + self.padding[0] + self.padding[2],
+            rect[3] + self.padding[1] + self.padding[3],
+        )
 
-    def top_at(self, x: float, y: float) -> "None|Widget":
-        if self.children:
-            for child in reversed(self.children):
-                r = child.top_at(x,y)
-                if r is not None:
-                    return r
-        if self.rect.collidepoint(x,y) and self.visible:
+    def set_position(self, x, y) -> Self:
+        if x is None:
+            x = self.rect.x
+        if y is None:
+            y = self.rect.y
+        if (x, y) == self.rect.topleft:
             return self
-        return None
-            
-    def get_constraint(self,name:str)->Constraint|None:
-        return next((c for c in self.constraints if c.name == name), None)
-
-    def add_constraints(self,*constraints:Constraint)->Self:
-        for c in constraints:
-            self.add_constraint(c,False)
-        self.apply_constraints()
+        dx, dy = x - self.rect.x, y - self.rect.y
+        self.rect.topleft = x, y
+        _ = [c.set_position(c.rect.x + dx, c.rect.y + dy) for c in self.children]
         return self
-        
-    def add_constraint(self,constraint:Constraint,apply:bool=True)->Self:
-        c = self.get_constraint(constraint.name)
-        if c is not None:
-            self.constraints.remove(c)
-        self.constraints.append(constraint)
-        self.apply_constraints()
+
+    def set_center(self, x, y) -> Self:
+        if x is None:
+            x = self.rect.centerx
+        if y is None:
+            y = self.rect.centery
+        if (x, y) == self.rect.center:
+            return self
+        dx, dy = x - self.rect.centerx, y - self.rect.centery
+        self.rect.center = x, y
+        _ = [
+            c.set_center(c.rect.centerx + dx, c.rect.centery + dy)
+            for c in self.children
+        ]
         return self
 
-    def has_constraint(self,name:str)->bool:
-        return any(c.name == name for c in self.constraints)
+    def set_parent_scene(self, parent_scene: bf.Scene) -> Self:
+        super().set_parent_scene(parent_scene)
+        if parent_scene is None:
+            bf.StyleManager().remove_widget(self)
 
-    def apply_all_constraints(self)->None:
-        # print("APPLY ALL CONSTRAINTS IN ",self.to_string())
-        self.apply_constraints()
-        for child in self.children : child.apply_all_constraints()
-    
-    def apply_constraints(self, max_iterations: int = 10) -> None:
-        if not self.parent:
-            # print(f"Warning : can't apply constraints on {self.to_string()} without parent widget")
-            return
-        if not self.constraints:
-            return
-        # Sort constraints based on priority
-        self.constraints.sort(key=lambda c: c.priority)
+        for c in self.children:
+            c.set_parent_scene(parent_scene)
+        return self
 
-        for iteration in range(max_iterations):
-            unsatisfied = []  # Initialize a flag
+    def set_parent(self, parent: "Widget") -> Self:
+        if parent == self.parent:
+            return self
+        if self.parent is not None:
+            self.parent.remove(self)
+        self.parent = parent
+        return self
 
-            for constraint in self.constraints:
-                if not constraint.evaluate(self.parent, self):
-                    unsatisfied.append(constraint)
-                    constraint.apply(self.parent, self)
-            if not unsatisfied:
-            #     data = ''.join(f"\n\t->{c.to_string()}" for c in self.constraints)
-            #     print(self.get_depth()*'\t'+f"Following constraints of {self.to_string()} were all satisfied :{data}")
-                break
-            # print(f"pass {iteration}/{max_iterations} : unsatisfied = {';'.join(c.to_string() for c in unsatisfied)}")
-            if iteration == max_iterations - 1:
-                raise ValueError(f"[WARNING] Following constraints for {self.to_string()} were not satisfied : \n\t{';'.join([c.to_string() for c in unsatisfied])}")
+    def set_padding(self, value: float | int | tuple | list) -> Self:
+        if isinstance(value, Iterable):
+            if len(value) > 4:
+                pass
+            elif any(v < 0 for v in value):
+                pass
+            elif len(value) == 2:
+                self.padding = (value[0], value[1], value[0], value[1])
+            else:
+                self.padding = (*value, *self.padding[len(value) :])
+        else:
+            self.padding = (value,) * 4
 
-    # GETTERS
-    def get_root(self)-> Self|Root|None:
-        if self.is_root: return self
-        if self.parent_scene is not None : return self.parent_scene.root
-        return None if self.parent is None else self.parent.get_root()
+        self.dirty_shape = True
+        return self
 
-    def get_size_int(self)->tuple[int,int]:
-        return (ceil(self.rect.width),ceil(self.rect.height))
+    def get_padded_rect(self) -> pygame.FRect:
+        r = self.rect.inflate(
+            -self.padding[0] - self.padding[2], -self.padding[1] - self.padding[3]
+        )
+        # r.normalize()
+        return r
 
+    def get_min_required_size(self) -> tuple[float, float]:
+        return self.rect.size
 
-    def get_center(self)->tuple[float,float]:
-        return self.rect.center
+    def get_padded_width(self) -> float:
+        return self.rect.w - self.padding[0] - self.padding[2]
 
+    def get_padded_height(self) -> float:
+        return self.rect.h - self.padding[1] - self.padding[3]
 
-    def get_bounding_box(self):
-        yield (self.rect,self._debug_color)
-        yield (self.get_content_rect(),"yellow")
-        for child in self.children:
-            yield from child.get_bounding_box()
+    def get_padded_left(self) -> float:
+        return self.rect.left + self.padding[0]
 
-    def set_autoresize(self,value:bool)-> Self:
-        self.autoresize = value
-        self.build()
-        return self
+    def get_padded_right(self) -> float:
+        return self.rect.right + self.padding[2]
 
-    def set_parent(self,parent:Self|None)->None:
-        if self.parent:
-            self.parent.remove_child(self)
-        self.parent = parent
-        self.apply_all_constraints()
-    # SETTERS
-    
-    def set_root(self) -> Self:
-        self.is_root = True
-        return self
+    def get_padded_center(self) -> tuple[float, float]:
+        return self.get_padded_rect().center
 
-    def set_parent_scene(self,scene)->None:
-        super().set_parent_scene(scene)
-        for child in self.children : 
-            child.set_parent_scene(scene)
+    def get_padded_top(self) -> float:
+        return self.rect.y + self.padding[1]
 
-    def set_x(self,x:float)->Self:
-        delta = x - self.rect.x
-        self.rect.x = x
-        for child in self.children:
-            child.set_x(child.rect.x + delta)
-        self.apply_constraints()
-        return self
+    def get_padded_bottom(self) -> float:
+        return self.rect.bottom - self.padding[3]
 
-    def set_y(self,y:float)->Self:
-        delta = y - self.rect.y
-        self.rect.y = y
+    def get_debug_outlines(self):
+        if not self.visible:
+            return
+        yield (self.rect, self.debug_color)
+        if any(self.padding):
+            yield (self.get_padded_rect(), self.debug_color)
         for child in self.children:
-            child.set_y(child.rect.y + delta)
-        self.apply_constraints()
-
+            yield from child.get_debug_outlines()
+
+    def add_constraints(self, *constraints: "Constraint") -> Self:
+        self.constraints.extend(constraints)
+        seen = set()
+        result = []
+        for c in self.constraints:
+            if c.name not in seen:
+                result.append(c)
+                seen.add(c.name)
+        self.constraints = result
+        self.constraints.sort(key=lambda c: c.priority)
+        self.dirty_constraints = True
         return self
 
-    def set_position(self,x:float,y:float)->Self:
-        delta_x = x - self.rect.x
-        delta_y = y - self.rect.y
-        self.rect.topleft = x,y
-        for child in self.children:
-            child.set_position(child.rect.x + delta_x,child.rect.y+delta_y)
-        self.apply_constraints()
-        return self
-        
-    def set_center(self,x:float,y:float)->Self:
-        delta_x = x - self.rect.centerx
-        delta_y = y - self.rect.centery
-        self.rect.center = x,y
-        for child in self.children:
-            child.set_position(child.rect.x + delta_x,child.rect.y+delta_y)
-        self.apply_constraints()
+    def resolve_constraints(self) -> None:
+        if self.parent is None or not self.constraints:
+            self.dirty_constraints = False
+            return
+        all_good = False
+        constraint_iteration = 0
+        while not all_good:
+            for constraint in self.constraints:
+                if not constraint.evaluate(self.parent, self):
+                    constraint.apply(self.parent, self)
+                    # print(constraint.name,"Applied")
+            constraint_iteration += 1
+            if all(c.evaluate(self.parent, self) for c in self.constraints):
+                all_good = True
+                break
+            elif self.__constraint_iteration > MAX_CONSTRAINTS:
+                print(
+                    self,
+                    "CONSTRAINTS ERROR",
+                    list(
+                        c.name
+                        for c in self.constraints
+                        if not c.evaluate(self.parent, self)
+                    ),
+                )
+                self.dirty_constraints = False
+                return
+        # print("DONE")
+        self.dirty_constraints = False
+
+    def remove_constraints(self, *names: str) -> Self:
+        self.constraints = [c for c in self.constraints if c.name not in names]
         return self
 
+    def has_constraint(self, name: str) -> bool:
+        return any(c.name == name for c in self.constraints)
 
-    def set_size(self, width : float, height: float) -> Self:
-        self.rect.size = (width,height)
-        self.build()
-        return self
-        
-
-    # Other Methods
+    def get_root(self) -> "Root":
+        if self.is_root:
+            return self
+        return self.parent.get_root()
 
-    def print_tree(self,ident:int=0)->None:
-        print('\t'*ident+self.to_string()+(':' if self.children else ''))
-        for child in self.children :
-            child.print_tree(ident+1)
+    def top_at(self, x: float | int, y: float | int) -> "None|Widget":
+        if self.children:
+            for child in reversed(self.children):
+                if child.visible:
+                    r = child.top_at(x, y)
+                    if r is not None:
+                        return r
+        return self if self.visible and self.rect.collidepoint(x, y) else None
+
+    def add(self, *children: "Widget") -> Self:
+        self.children.extend(children)
+        i = len(self.children)
+        for child in children:
+            child.set_render_order(i).set_parent(self).set_parent_scene(
+                self.parent_scene
+            )
+            i += 1
+        if self.parent:
+            self.parent.do_sort_children = True
+        return self
 
-    def to_string(self)->str:
-        
-        return f"{self.to_string_id()}@{*self.rect.topleft,* self.rect.size}"
+    def remove(self, *children: "Widget") -> Self:
+        for child in self.children:
+            if child in children:
+                self.children.remove(child)
+                child.set_parent(None).set_parent_scene(None)
+        if self.parent:
+            self.parent.do_sort_children = True
 
+    def set_size_if_autoresize(self, size: tuple[float, float]) -> Self:
+        size = list(size)
+        size[0] = size[0] if self.autoresize_w else None
+        size[1] = size[1] if self.autoresize_h else None
 
-    def to_string_id(self)->str:
-        return "Widget"
+        self.set_size(size)
 
+        return self
 
-    def do_when_removed(self)->None:
-        if self.parent_scene and self.parent == self.parent_scene.root:
-            self.set_parent(None)
+    def set_size(self, size: tuple) -> Self:
+        size = list(size)
+        if size[0] is None:
+            size[0] = self.rect.w
+        if size[1] is None:
+            size[1] = self.rect.h
+        if size == self.rect.size:
+            return self
+        self.rect.size = size
+        self.dirty_shape = True
+        return self
 
-    # Methods on children
+    def process_event(self, event: pygame.Event) -> bool:
+        # First propagate to children
+        for child in self.children:
+            child.process_event(event)
+        # return True if the method is blocking (no propagation to next children of the scene)
+        super().process_event(event)
+
+    def update(self, dt) -> None:
+        if self.do_sort_children:
+            self.children.sort(key=lambda c: c.render_order)
+            self.do_sort_children = False
+        _ = [c.update(dt) for c in self.children]
+        super().update(dt)
+
+    def build(self) -> None:
+        new_size = tuple(map(int, self.rect.size))
+        if self.surface.get_size() != new_size:
+            new_size = [max(0, i) for i in new_size]
+            self.surface = pygame.Surface(new_size, self.surface_flags)
+            if self.convert_alpha:
+                self.surface = self.surface.convert_alpha()
+
+    def paint(self) -> None:
+        self.surface.fill((0, 0, 0, 0))
+        return self
 
-    def add_child(self,*child:"Widget")->None:
-        for c in child : 
-            self.children.append(c)
-            c.set_parent(self)
-            c.set_parent_scene(self.parent_scene)
-            c.apply_constraints()
-        self.children_modified()
+    def visit(self, func: Callable, top_down: bool = True, *args, **kwargs) -> None:
+        if top_down:
+            func(self, *args, **kwargs)
+        for child in self.children:
+            child.visit(func, top_down)
+        if not top_down:
+            func(self, *args, **kwargs)
 
-    def remove_child(self,child:Self)->None:
-        self.children.remove(child)
-        child.set_parent(None)
-        child.set_parent_scene(None)
-        self.children_modified()
+    def visit_up(self, func, *args, **kwargs) -> None:
+        if func(self, *args, **kwargs):
+            return
+        if self.parent:
+            self.parent.visit_up(func, *args, **kwargs)
+
+    def selective_up(self, widget: "Widget"):
+        # if returns True then stop climbing widget tree
+        if widget.parent and widget.parent.dirty_constraints:
+            return False
+        widget.visit(self.selective_down)
+        return True
+
+    def selective_down(self, widget: "Widget"):
+        if widget.constraints:
+            widget.resolve_constraints()
+        else:
+            widget.dirty_constraints = False
+        if widget.dirty_shape:
+            widget.build()
+            widget.dirty_shape = False
+            self.dirty_surface = True
+
+    def draw(self, camera: bf.Camera) -> None:
+        if self.dirty_shape:
+            self.dirty_constraints = True
+            self.dirty_surface = True
+            self.build()
+            self.dirty_shape = False
+            for child in self.children:
+                child.dirty_constraints = True
+
+        if self.dirty_constraints:
+            if self.parent and self.parent.dirty_constraints:
+                self.parent.visit_up(self.selective_up)
+            else:
+                self.visit(lambda c: c.resolve_constraints())
 
+        if self.dirty_surface:
+            self.paint()
+            self.dirty_surface = False
 
+        super().draw(camera)
 
+        if self.clip_children:
 
-    # if return True -> don't propagate to siblings or parents
-    def process_event(self, event: pygame.Event)->bool:
-        # First propagate to children
-        for child in self.children:
-            if child.process_event(event):
-                return True
-         #return True if the method is blocking (no propagation to next children of the scene)
-        return super().process_event(event)
+            new_clip = camera.world_to_screen(self.get_padded_rect())
+            old_clip = camera.surface.get_clip()
+            new_clip = new_clip.clip(old_clip)
+            camera.surface.set_clip(new_clip)
 
+        _ = [
+            child.draw(camera)
+            for child in sorted(self.children, key=lambda c: c.render_order)
+        ]
 
-    def update(self,dt:float):
-        for child in self.children:
-            child.update(dt)
-
-    def draw(self, camera: bf.Camera) -> int:
-        self.children.sort(key=lambda e: (e.z_depth,e.render_order))
-        return super().draw(camera) + sum([child.draw(camera) for child in self.children])
-
-    def build(self)->None:
-        """
-        This function is called each time the widget's surface has to be updated
-        It usually has to be overriden if inherited to suit the needs of the new class
-        """
-        if not self.surface: return
-        if self.surface.get_size() != self.get_size_int():
-            self.surface = pygame.Surface(self.get_size_int())
-            if self.parent : self.parent.children_modified()
-
-    def children_modified(self)->None:
-        self.apply_constraints()
-        if self.parent and not self.is_root:
-            self.parent.children_modified()
+        if self.clip_children:
+            camera.surface.set_clip(old_clip)