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

batFramework/gui/textInput.py

@@ -0,0 +1,331 @@
+import batFramework as bf
+from typing import Self, Callable
+from .label import Label
+from .interactiveWidget import InteractiveWidget
+import pygame
+
+def find_next_word(s: str, start_index: int) -> int:
+    length = len(s)
+    # Ensure the starting index is within the bounds of the string
+    if start_index < 0 or start_index >= length:
+        raise ValueError("Starting index is out of bounds")
+    index = start_index
+    # If the start_index is at a space, skip leading spaces
+    if s[index] in [' ','\n']:
+        while index < length and s[index]  in [' ','\n']:
+            index += 1
+        # If we've reached the end of the string
+        if index >= length:
+            return -1
+    else:
+        # If the start_index is within a word, move to the end of that word
+        while index < length and s[index] not in [' ','\n']:
+            index += 1
+        if index == length:
+            return index    
+    # Return the index of the start of the next word or -1 if no more words are found
+    return index if index < length else -1
+
+def find_prev_word(s: str, start_index: int) -> int:
+    if start_index <= 0 : return 0
+    length = len(s)
+    
+    # Ensure the starting index is within the bounds of the string
+    if start_index < 0 or start_index >= length:
+        raise ValueError("Starting index is out of bounds")
+    
+    index = start_index
+    
+    # If the start_index is at a space, skip trailing spaces
+    if s[index] in [' ', '\n']:
+        while index > 0 and s[index-1] in [' ', '\n']:
+            index -= 1
+        # If we've reached the beginning of the string
+        if index <= 0:
+            return 0 if s[0] not in [' ', '\n'] else -1
+    else:
+        # If the start_index is within a word, move to the start of that word
+        while index > 0 and s[index-1] not in [' ', '\n']:
+            index -= 1
+        if index == 0 and s[index] not in [' ', '\n']:
+            return 0
+    
+    # Return the index of the start of the previous word or -1 if no more words are found
+    return index if index > 0 or (index == 0 and s[0] not in [' ', '\n']) else -1
+
+
+class TextInput(Label, InteractiveWidget):
+    def __init__(self) -> None:
+        self.cursor_position = (0, 0)
+        self.old_key_repeat = (0, 0)
+        self.placeholder_text = ""
+        self.cursor_timer = bf.Timer(0.2, self._cursor_toggle, loop=-1).start()
+        self.cursor_timer.pause()
+        self.show_cursor = False
+        self._cursor_blink_show : bool = True 
+        self.on_modify: Callable[[str], str]| None = None
+        self.set_click_pass_through(False)
+        super().__init__("")
+        self.set_outline_color("black")
+        self.alignment = bf.alignment.TOPLEFT
+
+    def set_placeholder_text(self, text: str) -> Self:
+        self.placeholder_text = text
+        self.dirty_surface = True
+        return self
+
+
+    def set_modify_callback(self, callback: Callable[[str], str]) -> Self:
+        self.on_modify = callback
+        return self
+
+    def __str__(self) -> str:
+        return f"TextInput"
+
+    def _cursor_toggle(self, value: bool | None = None):
+        if value is None:
+            value = not self._cursor_blink_show
+        self._cursor_blink_show = value
+        self.dirty_surface = True
+
+    def on_click_down(self, button,event):
+        if button == 1:
+            self.get_focus()
+            event.consumed = True
+        super().on_click_down(button,event)
+
+    def do_on_enter(self):
+        pygame.mouse.set_cursor(pygame.SYSTEM_CURSOR_IBEAM)
+
+    def do_on_exit(self):
+        pygame.mouse.set_cursor(bf.const.DEFAULT_CURSOR)
+
+    def do_on_get_focus(self):
+        self.cursor_timer.resume()
+        # self._cursor_toggle(True)
+        self.show_cursor = True
+        self.old_key_repeat = pygame.key.get_repeat()
+        pygame.key.set_repeat(200, 50)
+
+    def do_on_lose_focus(self):
+        self.cursor_timer.pause()
+        self.show_cursor = False
+        pygame.key.set_repeat(*self.old_key_repeat)
+
+    def get_line(self, line: int) -> str | None:
+        if line < 0:
+            return None
+        lines = self.text_widget.text.split('\n')
+        if line >= len(lines):
+            return None
+        return lines[line]
+
+    def get_debug_outlines(self):
+        yield from super().get_debug_outlines()
+        if self.visible:
+            # offset = self._get_outline_offset() if self.show_text_outline else (0,0)
+            # yield (self.text_widget.rect.move(self.rect.x - offset[0] - self.scroll.x,self.rect.y - offset[1] - self.scroll.y), "purple")
+            yield (self.get_cursor_rect().move(*self.text_widget.rect.topleft),"green")
+
+
+    def get_min_required_size(self) -> tuple[float, float]:
+        size = self.text_widget.get_min_required_size()
+        return self.expand_rect_with_padding(
+            (0, 0,size[0]+self.get_cursor_rect().w,size[1])
+        ).size
+
+
+    def set_cursor_position(self, position: tuple[int, int]) -> Self:
+        x, y = position
+
+        lines = self.text_widget.text.split('\n')
+        y = max(0, min(y, len(lines) - 1))
+        line_length = len(lines[y])
+        x = max(0, min(x, line_length))
+        self.cursor_position = (x,y)
+        return self
+
+    def get_cursor_rect(self) -> pygame.FRect:
+        if not self.text_widget.font_object:
+            return pygame.FRect(0, 0, 0, 0)
+        font = self.text_widget.font_object
+
+        lines = self.text_widget.text.split('\n')
+        line_x, line_y = self.cursor_position
+        line = lines[line_y]
+
+        # # Clamp line_y and line_x to valid ranges
+        # line_y = max(0, min(line_y, len(lines) - 1))
+        # line_x = max(0, min(line_x, len(line)))
+
+        line_height = font.get_linesize()
+
+        # Calculate the pixel x position of the cursor in the current line
+        x = font.size(line[:line_x])[0]
+        y = line_height * line_y
+
+        if self.text_widget.show_text_outline:
+            offset = self.text_widget._get_outline_offset()
+            x+=offset[0]
+            y+=offset[1]
+        
+        res = pygame.FRect(x,y,1,line_height)
+        return  res
+
+    def ensure_cursor_visible(self):
+        pass
+
+    def cursor_to_absolute(self, position: tuple[int, int]) -> int:
+        x, y = position
+
+        y = max(0, min(y, len(self.text_widget.text.split('\n')) - 1))
+        lines = self.text_widget.text.split('\n')
+        x = max(0, min(x, len(lines[y])))
+
+        absolute_position = sum(len(line) + 1 for line in lines[:y]) + x
+        return absolute_position
+
+    def absolute_to_cursor(self, absolute: int) -> tuple[int, int]:
+        text = self.text_widget.text
+        lines = text.split('\n')
+        current_pos = 0
+
+        for line_no, line in enumerate(lines):
+            if absolute <= current_pos + len(line):
+                return (absolute - current_pos, line_no)
+            current_pos += len(line) + 1
+
+        return (len(lines[-1]), len(lines) - 1)
+
+    def handle_event(self, event):
+        # TODO fix tab_focus not working when textInput in focus
+        super().handle_event(event)
+        if event.consumed or(not self.is_focused or event.type not in [pygame.TEXTINPUT, pygame.KEYDOWN]):
+            return
+
+        text = self.get_text()
+        current_pos = self.cursor_to_absolute(self.cursor_position)
+        pressed = pygame.key.get_pressed()
+
+        if event.type == pygame.TEXTINPUT:
+            # Insert text at the current cursor position
+            self.set_text(f"{text[:current_pos]}{event.text}{text[current_pos:]}")
+            self.set_cursor_position(self.absolute_to_cursor(current_pos + len(event.text)))
+        elif event.type == pygame.KEYDOWN:
+            match event.key:
+                case pygame.K_ESCAPE:
+                    self.lose_focus()
+
+                case pygame.K_BACKSPACE if current_pos > 0:
+                    # Remove the character before the cursor
+                    delta = current_pos-1
+                    if pressed[pygame.K_LCTRL] or pressed[pygame.K_RCTRL]:
+                        delta = find_prev_word(self.text_widget.text,current_pos-1)
+                        if delta <0: delta = 0
+                        
+                    self.set_text(f"{text[:delta]}{text[current_pos:]}")
+                    self.set_cursor_position(self.absolute_to_cursor(delta))
+                    self._cursor_toggle(True)
+
+                case pygame.K_DELETE if current_pos < len(text):
+                    # Remove the character at the cursor
+                    self.set_text(f"{text[:current_pos]}{text[current_pos + 1:]}")
+                    self._cursor_toggle(True)
+
+                case pygame.K_RIGHT:
+                    if current_pos < len(text):
+                        self.handle_cursor_movement(pressed, current_pos, direction="right")
+                    self._cursor_toggle(True)
+                    
+                case pygame.K_LEFT:
+                    if current_pos > 0:
+                        self.handle_cursor_movement(pressed, current_pos, direction="left")
+                    self._cursor_toggle(True)
+
+                case pygame.K_UP:
+                    # Move cursor up one line
+                    self.set_cursor_position((self.cursor_position[0], self.cursor_position[1] - 1))
+                    self._cursor_toggle(True)
+
+                case pygame.K_DOWN:
+                    # Move cursor down one line
+                    self.set_cursor_position((self.cursor_position[0], self.cursor_position[1] + 1))
+                    self._cursor_toggle(True)
+
+                case pygame.K_RETURN:
+                    # Insert a newline at the current cursor position
+                    self.set_text(f"{text[:current_pos]}\n{text[current_pos:]}")
+                    self.set_cursor_position(self.absolute_to_cursor(current_pos + 1))
+                    self._cursor_toggle(True)
+                case _ :
+                    if event.unicode:
+                        event.consumed = True
+                    return
+
+            event.consumed = True
+
+    def handle_cursor_movement(self, pressed, current_pos, direction):
+        if direction == "right":
+            if pressed[pygame.K_LCTRL] or pressed[pygame.K_RCTRL]:
+                next_word_pos = find_next_word(self.text_widget.text, current_pos)
+                self.set_cursor_position(self.absolute_to_cursor(next_word_pos if next_word_pos != -1 else current_pos + 1))
+            else:
+                self.set_cursor_position(self.absolute_to_cursor(current_pos + 1))
+        elif direction == "left":
+            if pressed[pygame.K_LCTRL] or pressed[pygame.K_RCTRL]:
+                prev_word_pos = find_prev_word(self.text_widget.text, current_pos - 1)
+                self.set_cursor_position(self.absolute_to_cursor(prev_word_pos if prev_word_pos != -1 else current_pos - 1))
+            else:
+                self.set_cursor_position(self.absolute_to_cursor(current_pos - 1))
+
+
+    def set_text(self, text: str) -> Self:
+        if self.on_modify:
+            text = self.on_modify(text)
+        if text == "" and self.placeholder_text:
+            text = self.placeholder_text
+        if text != "" and text == self.placeholder_text:
+            self.text_widget.set_text("")
+        self.text_widget.set_text(text)
+
+        return self
+    def _draw_cursor(self,camera:bf.Camera) -> None:
+        if not self.show_cursor or not self._cursor_blink_show:
+            return
+        
+        cursor_rect = self.get_cursor_rect()
+        cursor_rect.move_ip(*self.text_widget.rect.topleft)
+        cursor_rect = camera.world_to_screen(cursor_rect)
+        if self.text_widget.show_text_outline:
+            pygame.draw.rect(camera.surface, self.text_widget.text_outline_color, cursor_rect.inflate(2,2))
+        pygame.draw.rect(camera.surface, self.text_widget.text_color, cursor_rect)
+
+    def draw(self,camera:bf.Camera) -> None:
+        super().draw(camera)
+        self._draw_cursor(camera)
+
+    def align_text(
+        self, text_rect: pygame.FRect, area: pygame.FRect, alignment: bf.alignment
+    ):
+        cursor_rect = self.get_cursor_rect()
+        
+        if alignment == bf.alignment.LEFT:
+            alignment = bf.alignment.MIDLEFT
+        elif alignment == bf.alignment.MIDRIGHT:
+            alignment = bf.alignment.MIDRIGHT
+        pos = area.__getattribute__(alignment.value)
+        text_rect.__setattr__(alignment.value, pos)
+        scroll = self.text_widget.scroll
+        
+        if cursor_rect.right > area.right+scroll.x:
+            scroll.x=cursor_rect.right - area.right
+        elif cursor_rect.x < scroll.x+area.left:
+            scroll.x= cursor_rect.left - area.left
+        # self.scroll.x = 0
+        scroll.x = max(scroll.x,0)
+
+        if cursor_rect.bottom > scroll.y + area.bottom:
+            scroll.y = cursor_rect.bottom - area.bottom
+        elif cursor_rect.y < scroll.y + area.top:
+            scroll.y = cursor_rect.top - area.top
+        scroll.y = max(scroll.y, 0)

batFramework/gui/textWidget.py

@@ -0,0 +1,308 @@
+from math import ceil
+import pygame
+from .widget import Widget
+import batFramework as bf
+from typing import Literal, Self,Union
+
+class TextWidget(Widget):
+    def __init__(self, text:str):
+        super().__init__()
+        self.text = text
+
+        # Allows scrolling the text
+        self.allow_scroll : bool = True
+
+        # Scroll variable
+        # TODO make scroll work
+        self.scroll :pygame.Vector2 = pygame.Vector2(0,0)
+
+        # Enable/Disable antialiasing
+        self.antialias: bool = bf.FontManager().DEFAULT_ANTIALIAS
+
+        self.text_size = bf.FontManager().DEFAULT_FONT_SIZE
+
+        self.auto_wraplength: bool = False
+
+        self.text_color: tuple[int, int, int] | str = "black"
+
+        self.text_bg_color : tuple[int,int,int]| str |  None = None
+
+
+        self.text_outline_color: tuple[int, int, int] | str = "gray50"
+
+        self._text_outline_mask = pygame.Mask((3, 3), fill=True)
+
+        self.line_alignment = pygame.FONT_LEFT
+        # font name (given when loaded by utils) to use for the text
+
+        self.font_name = None
+        # reference to the font object
+        self.font_object = None
+        # Rect containing the text of the label
+        self.show_text_outline: bool = False
+
+        self.is_italic: bool = False
+
+        self.is_bold: bool = False
+
+        self.is_underlined: bool = False
+
+        super().__init__()
+        self.set_debug_color("purple")
+        self.set_autoresize(True)
+        self.set_font(force=True)
+        self.set_convert_alpha(True)
+
+
+    def set_padding(self, value): # can't set padding
+        return self
+
+    def __str__(self) -> str:
+        return f"TextWidget({repr(self.text)})"
+    
+    def set_allow_scroll(self, value:bool)->Self:
+        if self.allow_scroll == value: return self
+        self.allow_scroll = value
+        self.dirty_surface = True
+        return self
+
+    def set_scroll(self,x=None,y=None)->Self:
+        x = x if x is not None else self.scroll.x
+        y = y if y is not None else self.scroll.y
+
+        self.scroll.update(x,y)
+        self.dirty_surface = True
+        return self
+
+    def scroll_by(self,x=0,y=0)->Self:
+        self.scroll += x,y
+        self.dirty_surface = True
+        return self
+
+    def set_text_color(self, color) -> Self:
+        self.text_color = color
+        self.dirty_surface = True
+        return self
+
+    def set_text_bg_color(self, color) -> Self:
+        self.text_bg_color = color
+        self.set_convert_alpha(color is None)
+        self.dirty_surface = True
+        return self
+
+
+    def top_at(self, x, y):
+        return None
+
+    def set_line_alignment(self, alignment: Union[Literal["left"], Literal["right"], Literal["center"]]) -> Self:
+        self.line_alignment = alignment
+        self.dirty_surface = True
+        return self
+
+    def set_italic(self, value: bool) -> Self:
+        if value == self.is_italic:
+            return self
+        self.is_italic = value
+        if self.autoresize_h or self.autoresize_w:
+            self.dirty_shape = True
+        else:
+            self.dirty_surface = True
+        return self
+
+    def set_bold(self, value: bool) -> Self:
+        if value == self.is_bold:
+            return self
+        self.is_bold = value
+        if self.autoresize_h or self.autoresize_w:
+            self.dirty_shape = True
+        else:
+            self.dirty_surface = True
+        return self
+
+    def set_underlined(self, value: bool) -> Self:
+        if value == self.is_underlined:
+            return self
+        self.is_underlined = value
+        self.dirty_surface = True
+        return self
+    
+    def set_text_outline_mask_size(self,size:tuple[int,int])->Self:
+        old_size = self._text_outline_mask.get_size()
+        min_w, min_h = min(old_size[0], size[0]), min(old_size[1], size[1])
+        m = [
+            [self._text_outline_mask.get_at((x, y)) for x in range(min_w)]
+            for y in range(min_h)
+        ]
+        self._text_outline_mask = pygame.Mask(size, fill=True)
+        self.set_text_outline_matrix(m)
+        return self
+
+    def set_text_outline_matrix(self, matrix: list[list[0 | 1]]) -> Self:
+        if matrix is None:
+            matrix = [[0 for _ in range(3)] for _ in range(3)]
+        for y in range(3):
+            for x in range(3):
+                self._text_outline_mask.set_at((x, y), matrix[2 - y][2 - x])
+        self.dirty_shape = True
+        return self
+
+    def set_text_outline_color(self, color) -> Self:
+        self.text_outline_color = color
+        self.dirty_surface = True
+        return self
+
+    def set_show_text_outline(self,value:bool) -> Self:
+        self.show_text_outline = value
+        self.dirty_shape = True
+        return self
+
+    def set_auto_wraplength(self, val: bool) -> Self:
+        self.auto_wraplength = val
+        if self.autoresize_h or self.autoresize_w:
+            self.dirty_shape = True
+        else:
+            self.dirty_surface = True
+        return self
+
+    def get_debug_outlines(self):
+        if self.visible:
+            yield from super().get_debug_outlines()
+
+    def set_font(self, font_name: str = None, force: bool = False) -> Self:
+        if font_name == self.font_name and not force:
+            return self
+        self.font_name = font_name
+        self.font_object = bf.FontManager().get_font(self.font_name, self.text_size)
+        if self.autoresize_h or self.autoresize_w:
+            self.dirty_shape = True
+        else:
+            self.dirty_surface = True
+        return self
+
+    def set_text_size(self, text_size: int) -> Self:
+        text_size = (text_size // 2) * 2
+        if text_size == self.text_size:
+            return self
+        self.text_size = text_size
+        self.font_object = bf.FontManager().get_font(self.font_name, self.text_size)
+        self.dirty_shape = True
+        return self
+
+    def get_text_size(self) -> int:
+        return self.text_size
+
+
+    def is_antialias(self) -> bool:
+        return self.antialias
+
+    def set_antialias(self, value: bool) -> Self:
+        self.antialias = value
+        self.dirty_surface = True
+        return self
+
+    def set_text(self, text: str) -> Self:
+        if text == self.text:
+            return self
+        self.text = text
+        self.dirty_shape = True
+        return self
+
+    def get_min_required_size(self) -> tuple[float, float]:
+        if not self.font_object : return 0,0
+
+        tmp_text = self.text
+        if self.text.endswith('\n'):
+            tmp_text+=" " # hack to have correct size if ends with newline
+        params = {
+            "font_name": self.font_object.name,
+            "text": tmp_text,
+            "antialias": self.antialias,
+            "color": self.text_color,
+            "bgcolor": self.text_bg_color,
+            "wraplength": int(self.get_inner_width()) if self.auto_wraplength and not self.autoresize_w else 0,
+        }
+
+        size = list(self._render_font(params).get_size())
+        size[1]= max(size[1],self.font_object.get_ascent() - self.font_object.get_descent())
+        if not self.show_text_outline:
+            return size
+        s = self._get_outline_offset()
+        return size[0] + s[0]*2, size[1] + s[1]*2
+    
+
+    def get_text(self) -> str:
+        return self.text
+
+    def _render_font(self, params: dict) -> pygame.Surface:
+        params.pop("font_name")
+        # save old settings
+        old_italic = self.font_object.get_italic()
+        old_bold = self.font_object.get_bold()
+        old_underline = self.font_object.get_underline()
+        old_align = self.font_object.align
+        # setup font
+        self.font_object.set_italic(self.is_italic)
+        self.font_object.set_bold(self.is_bold)
+        self.font_object.set_underline(self.is_underlined)
+        self.font_object.align = self.line_alignment
+        surf = self.font_object.render(**params)
+        # reset font
+        self.font_object.set_italic(old_italic)
+        self.font_object.set_bold(old_bold)
+        self.font_object.set_underline(old_underline)
+        self.font_object.align = old_align
+        return surf
+
+    def _get_outline_offset(self)->tuple[int,int]:
+        mask_size = self._text_outline_mask.get_size()
+        return  mask_size[0]//2,mask_size[1]//2
+
+
+    def build(self) -> bool:
+        """
+        return True if size changed
+        """
+        target_size = self.resolve_size(self.get_min_required_size())
+        if self.rect.size != target_size:
+            self.set_size(target_size)
+            return True
+        return False
+        
+
+
+    def paint(self) -> None:
+        self._resize_surface()
+        if self.font_object is None:
+            print(f"No font for widget with text : '{self}' :(")
+            return
+        
+            
+        wrap = int(self.get_inner_width()) if self.auto_wraplength and not self.autoresize_w else 0
+        params = {
+            "font_name": self.font_object.name,
+            "text": self.text,
+            "antialias": self.antialias,
+            "color": self.text_color,
+            "bgcolor": self.text_bg_color if not self.show_text_outline else None,
+            "wraplength": wrap,
+        }
+
+        if self.text_bg_color is None : 
+            self.surface = self.surface.convert_alpha()
+
+        bg_fill_color = (0, 0, 0, 0) if self.text_bg_color is None else  self.text_bg_color 
+        self.surface.fill(bg_fill_color)
+
+        text_surf = self._render_font(params)
+
+        if self.show_text_outline:
+            mask = pygame.mask.from_surface(text_surf).convolve(self._text_outline_mask)
+            outline_surf = mask.to_surface(
+                setcolor=self.text_outline_color,
+                unsetcolor=bg_fill_color
+            )
+
+            outline_surf.blit(text_surf,self._get_outline_offset())
+            text_surf = outline_surf
+
+        self.surface.blit(text_surf, -self.scroll)