drawlib 0.2.3.dev5__tar.gz → 0.2.3.dev6__tar.gz

{drawlib-0.2.3.dev5 → drawlib-0.2.3.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: drawlib
-Version: 0.2.3.dev5
+Version: 0.2.3.dev6
 Summary: Python drawing library. Illustration as Code.
 Home-page: https://www.drawlib.com
 Author: Yuichi Ito

{drawlib-0.2.3.dev5 → drawlib-0.2.3.dev6}/drawlib/__init__.py

@@ -14,7 +14,7 @@ import sys
 from typing import Final, List
 
 # please update here when you release new version
-LIB_VERSION = "0.2.3.dev5"
+LIB_VERSION = "0.2.3.dev6"
 
 # please list active main committers (1+ commits per month)
 AUTHORS: Final[List[str]] = [

{drawlib-0.2.3.dev5 → drawlib-0.2.3.dev6}/drawlib/v0_2/private/core/util.py

@@ -11,7 +11,7 @@
 """Utility module for converting drawlib data to matplotlib data."""
 
 import math
-from typing import Any, Callable, Dict, Literal, Optional, Tuple, Union
+from typing import Any, Callable, Dict, List, Literal, Optional, Tuple, Union
 
 from matplotlib.font_manager import FontProperties
 from matplotlib.text import Text
@@ -238,6 +238,58 @@ class ImageUtil:
 class LineUtil:
     """A utility class for handling line styles and options."""
 
+    @staticmethod
+    def sanitize_xys(xys: List[Tuple[float, float]]) -> List[Tuple[float, float]]:
+        def remove_consecutive_duplicates(xys: List[Tuple[float, float]]) -> List[Tuple[float, float]]:
+            return [v for i, v in enumerate(xys) if i == 0 or v != xys[i - 1]]
+
+        def merge_straight_lines(xys: List[Tuple[float, float]]) -> List[Tuple[float, float]]:  # noqa: C901
+            points: List[Tuple[float, float]] = []
+
+            skip_next = False
+            for i in range(len(xys) - 1):
+                if skip_next:
+                    skip_next = False
+                    continue
+
+                if i == 0:
+                    points.append(xys[i])
+                    continue
+
+                x0, y0 = points[-1]
+                x1, y1 = xys[i]
+                x2, y2 = xys[i + 1]
+                if x1 - x0 == 0:
+                    if x2 - x1 == 0:
+                        points.append(xys[i + 1])
+                        skip_next = True
+                        continue
+                    else:
+                        points.append(xys[i])
+                        if i == len(xys) - 2:
+                            points.append(xys[i + 1])
+                        continue
+
+                if x2 - x1 == 0:
+                    points.append(xys[i])
+                    if i == len(xys) - 2:
+                        points.append(xys[i + 1])
+                    continue
+
+                m1 = (y1 - y0) / (x1 - x0)
+                m2 = (y2 - y1) / (x2 - x1)
+                if m1 == m2:
+                    skip_next = True
+                    points.append(xys[i + 1])
+                else:
+                    points.append(xys[i])
+                    if i == len(xys) - 2:
+                        points.append(xys[i + 1])
+
+            return points
+
+        return merge_straight_lines(remove_consecutive_duplicates(xys))
+
     @staticmethod
     def format_style(
         style: Union[LineStyle, str, None],

{drawlib-0.2.3.dev5 → drawlib-0.2.3.dev6}/drawlib/v0_2/private/core_canvas/line.py

@@ -268,7 +268,7 @@ class CanvasLineFeature(CanvasBase):
         """
         style = LineUtil.format_style(style)
         validator.validate_line_args(locals())
-
+        xys = LineUtil.sanitize_xys(xys)
         self.lines_bezier(
             xy=xys[0],
             path_points=xys[1:],  # type: ignore
@@ -300,10 +300,12 @@ class CanvasLineFeature(CanvasBase):
         """
         style = LineUtil.format_style(style)
         validator.validate_line_args(locals())
+
         if len(xys) == 2:
-            self.lines(xys, width=width, arrowhead=arrowhead, style=style)
+            self.line(xys[0], xys[1], width=width, arrowhead=arrowhead, style=style)
             return
 
+        xys = LineUtil.sanitize_xys(xys)
         path_points = []
         last_i = len(xys) - 2
         # last_xy = (0, 0)

{drawlib-0.2.3.dev5 → drawlib-0.2.3.dev6}/drawlib/v0_2/private/core_canvas/original_arrow.py

@@ -19,7 +19,7 @@ from matplotlib.path import Path
 import drawlib.v0_2.private.validators.args as validator
 from drawlib.v0_2.private.core.model import ShapeStyle, ShapeTextStyle
 from drawlib.v0_2.private.core.theme import dtheme
-from drawlib.v0_2.private.core.util import ShapeUtil
+from drawlib.v0_2.private.core.util import LineUtil, ShapeUtil
 from drawlib.v0_2.private.core_canvas.base import CanvasBase
 from drawlib.v0_2.private.core_canvas.line import LineArcHelper
 from drawlib.v0_2.private.util import (
@@ -205,6 +205,7 @@ class CanvasOriginalArrowFeature(CanvasBase):
             new_point = (a[0] + scaled_vector[0], a[1] + scaled_vector[1])
             return new_point
 
+        xys = LineUtil.sanitize_xys(xys)
         if head in {"<-", "<->"}:
             xys_start = point_on_line(xys[0], xys[1], head_length)
         else:
@@ -568,9 +569,6 @@ class ArrowPolylineHelper:
                 curve.append((x, y))
             return curve
 
-        def remove_consecutive_duplicates(lst: List[Any]) -> List[Any]:
-            return [v for i, v in enumerate(lst) if i == 0 or v != lst[i - 1]]
-
         self._r = r
         if r == 0:
             self._original_points = xys[:]
@@ -600,7 +598,7 @@ class ArrowPolylineHelper:
             points.extend(get_points(bezier_points))
             bezier_start = p2
 
-        self._original_points = remove_consecutive_duplicates(points)
+        self._original_points = points
 
     def get_parallel_points(self, distance: float) -> List[Tuple[float, float]]:
         """Internal function"""
@@ -619,7 +617,6 @@ class ArrowPolylineHelper:
             p0, p1 = self._original_points[i - 1], self._original_points[i]
             dx, dy = p1[0] - p0[0], p1[1] - p0[1]
             length = get_distance(p0, p1)
-            print(p0, p1, length)
             nx, ny = -dy / length, dx / length
             px0, py0 = p0[0] + distance * nx, p0[1] + distance * ny
             px1, py1 = p1[0] + distance * nx, p1[1] + distance * ny

drawlib-0.2.3.dev6/drawlib/v0_2/private/smartarts/boxlist.py

@@ -0,0 +1,257 @@
+# Copyright (c) 2024 Yuichi Ito (yuichi@yuichi.com)
+#
+# This software is licensed under the Apache License, Version 2.0.
+# For more information, please visit: https://github.com/yuichi110/drawlib
+#
+# This software is provided "as is", without warranty of any kind,
+# express or implied, including but not limited to the warranties of
+# merchantability, fitness for a particular purpose and noninfringement.
+
+
+"""BoxList implementation module."""
+
+import dataclasses
+from typing import Any, List, Literal, Optional, Tuple, Union
+
+from drawlib.v0_2.private.core.colors import Colors
+from drawlib.v0_2.private.core.model import ShapeStyle, ShapeTextStyle
+from drawlib.v0_2.private.core.theme import dtheme
+from drawlib.v0_2.private.core_canvas.canvas import rectangle
+from drawlib.v0_2.private.util import error_handler
+
+
+class BoxList:
+    """A class to draw a list of boxes with text, supporting highlighting of certain boxes.
+
+    Args:
+        box_style (Union[str, ShapeStyle, None]): The style for the boxes.
+        text_style (Union[str, ShapeTextStyle, None]): The style for the text inside the boxes.
+        box_highlight_style (Union[str, ShapeStyle, None]): The style for the highlighted boxes.
+        text_highlight_style (Union[str, ShapeTextStyle, None]): The style for the text inside the highlighted boxes.
+    """
+
+    @error_handler
+    def __init__(
+        self,
+        default_box_style: Union[str, ShapeStyle, None] = None,
+        default_text_style: Union[str, ShapeTextStyle, None] = None,
+    ) -> None:
+        """Initialize BoxList.
+
+        Args:
+            default_box_style (Union[str, ShapeStyle, None]): The style for the boxes.
+            default_text_style (Union[str, ShapeTextStyle, None]): The style for the text inside the boxes.
+            box_highlight_style (Union[str, ShapeStyle, None]): The style for the highlighted boxes.
+            text_highlight_style (Union[str, ShapeTextStyle, None]):
+                    The style for the text inside the highlighted boxes.
+        """
+        if isinstance(default_box_style, str):
+            default_box_style = dtheme.rectanglestyles.get(default_box_style)
+        if default_box_style is None:
+            default_box_style = dtheme.rectanglestyles.get()
+        default_box_style.halign = "center"
+        default_box_style.valign = "center"
+        self._default_box_style = default_box_style
+
+        if isinstance(default_text_style, str):
+            default_text_style = dtheme.rectangletextstyles.get(default_text_style)
+        if default_text_style is None:
+            default_text_style = dtheme.rectangletextstyles.get()
+        self._default_text_style = default_text_style
+
+        self._list: List[_Item] = []
+
+    @error_handler
+    def append(
+        self,
+        text: str,
+        box_style: Union[str, ShapeStyle, None] = None,
+        text_style: Union[str, ShapeTextStyle, None] = None,
+    ) -> None:
+        """
+        Appends a new box with text to the BoxList.
+
+        Args:
+            text (str): The text to be displayed inside the box.
+            box_style (Union[str, ShapeStyle, None], optional):
+                The style for the box. Can be a style name, a ShapeStyle object, or None.
+                If None, the default box style is used.
+            text_style (Union[str, ShapeTextStyle, None], optional):
+                The style for the text inside the box. Can be a style name, a ShapeTextStyle object, or None.
+                If None, the default text style is used.
+        """
+        self.extend([text], box_style=box_style, text_style=text_style)
+
+    @error_handler
+    def insert(
+        self,
+        index: int,
+        text: str,
+        box_style: Union[str, ShapeStyle, None] = None,
+        text_style: Union[str, ShapeTextStyle, None] = None,
+    ) -> None:
+        """
+        Inserts a new box with text at a specified position in the BoxList.
+
+        Args:
+            index (int): The position at which to insert the new box.
+            text (str): The text to be displayed inside the box.
+            box_style (Union[str, ShapeStyle, None], optional):
+                The style for the box. Can be a style name, a ShapeStyle object, or None.
+                If None, the default box style is used.
+            text_style (Union[str, ShapeTextStyle, None], optional):
+                The style for the text inside the box. Can be a style name, a ShapeTextStyle object, or None.
+                If None, the default text style is used.
+        """
+        is_custom_style = box_style is not None or text_style is not None
+
+        if isinstance(box_style, str):
+            box_style = dtheme.rectanglestyles.get(box_style)
+        elif box_style is None:
+            box_style = self._default_box_style
+
+        if isinstance(text_style, str):
+            text_style = dtheme.rectangletextstyles.get(text_style)
+        elif text_style is None:
+            text_style = self._default_text_style
+
+        item = _Item(
+            text=text,
+            box_style=box_style,
+            text_style=text_style,
+            is_custom_style=is_custom_style,
+        )
+        self._list.insert(index, item)
+
+    @error_handler
+    def extend(
+        self,
+        texts: List[str],
+        box_style: Union[str, ShapeStyle, None] = None,
+        text_style: Union[str, ShapeTextStyle, None] = None,
+    ) -> None:
+        """
+        Extends the BoxList by appending multiple boxes with text.
+
+        Args:
+            texts (List[str]): A list of texts to be displayed inside the boxes.
+            box_style (Union[str, ShapeStyle, None], optional):
+                The style for the boxes. Can be a style name, a ShapeStyle object, or None.
+                If None, the default box style is used.
+            text_style (Union[str, ShapeTextStyle, None], optional):
+                The style for the text inside the boxes. Can be a style name, a ShapeTextStyle object, or None.
+                If None, the default text style is used.
+        """
+        is_custom_style = box_style is not None or text_style is not None
+
+        if isinstance(box_style, str):
+            box_style = dtheme.rectanglestyles.get(box_style)
+        elif box_style is None:
+            box_style = self._default_box_style
+
+        if isinstance(text_style, str):
+            text_style = dtheme.rectangletextstyles.get(text_style)
+        elif text_style is None:
+            text_style = self._default_text_style
+
+        for text in texts:
+            item = _Item(
+                text=text,
+                box_style=box_style,
+                text_style=text_style,
+                is_custom_style=is_custom_style,
+            )
+            self._list.append(item)
+
+    @error_handler
+    def draw(
+        self,
+        xy: Tuple[float, float],
+        box_width: float,
+        box_height: float,
+        align: Literal["left", "right", "bottom", "top"] = "left",
+    ) -> None:
+        """Draws a list of boxes at the specified location.
+
+        Args:
+            xy (Tuple[float, float]): The starting point (x, y) to draw the list of boxes.
+            box_width (float): The width of each box.
+            box_height (float): The height of each box.
+            items (List[Any]): The list of items to be displayed inside the boxes.
+            highlight_indexs (Optional[List[int]]):
+                    The list of indices of the boxes to be highlighted.
+            length (Optional[int]): The total number of boxes to be drawn.
+            align (Literal["left", "right", "bottom", "top"]):
+                    The alignment of the boxes relative to the starting point.
+
+        """
+        for index, item in enumerate(self._list):
+            if item.is_custom_style:
+                continue
+
+            self._draw_cell(
+                start_xy=xy,
+                index=index,
+                text=item.text,
+                box_width=box_width,
+                box_height=box_height,
+                box_style=item.box_style,
+                text_style=item.text_style,
+                align=align,
+            )
+
+        for index, item in enumerate(self._list):
+            if not item.is_custom_style:
+                continue
+
+            self._draw_cell(
+                start_xy=xy,
+                index=index,
+                text=item.text,
+                box_width=box_width,
+                box_height=box_height,
+                box_style=item.box_style,
+                text_style=item.text_style,
+                align=align,
+            )
+
+    @staticmethod
+    def _draw_cell(
+        start_xy: Tuple[float, float],
+        index: int,
+        text: str,
+        box_width: float,
+        box_height: float,
+        box_style: ShapeStyle,
+        text_style: ShapeTextStyle,
+        align: Literal["left", "right", "bottom", "top"],
+    ) -> None:
+        if align == "left":
+            x = start_xy[0] + box_width * (index + 0.5)
+            y = start_xy[1] + box_height / 2
+        elif align == "right":
+            x = start_xy[0] - box_width * (index + 0.5)
+            y = start_xy[1] + box_height / 2
+        elif align == "bottom":
+            x = start_xy[0] + box_width / 2
+            y = start_xy[1] + box_height * (index + 0.5)
+        else:
+            x = start_xy[0] + box_width / 2
+            y = start_xy[1] - box_height * (index + 0.5)
+
+        rectangle(
+            xy=(x, y),
+            width=box_width,
+            height=box_height,
+            style=box_style,
+            text=text,
+            textstyle=text_style,
+        )
+
+
+@dataclasses.dataclass
+class _Item:
+    text: str
+    box_style: ShapeStyle
+    text_style: ShapeTextStyle
+    is_custom_style: bool

{drawlib-0.2.3.dev5 → drawlib-0.2.3.dev6}/drawlib/v0_2/private/smartarts/gridlayout.py

@@ -49,8 +49,9 @@ class GridLayout:
     @error_handler
     def add(  # noqa: C901
         self,
-        column_range: Tuple[int, int],
-        row_range: Tuple[int, int],
+        position: Tuple[int, int],
+        width: int,
+        height: int,
         r: Optional[int] = None,
         style: Union[str, ShapeStyle, None] = None,
         text: str = "",
@@ -62,8 +63,9 @@ class GridLayout:
         Add an item to the grid layout.
 
         Args:
-            column_range (Tuple[int, int]): The range of columns the item will span.
-            row_range (Tuple[int, int]): The range of rows the item will span.
+            position (Tuple[int, int]): Cell start (column, row) point.
+            width (int): How many column cells.
+            height (int): How many row cells.
             r (Optional[int], optional): The radius of the item. Default is None, which uses the default radius.
             style (Union[str, ShapeStyle, None], optional):
                     The style of the item.
@@ -81,25 +83,24 @@ class GridLayout:
         Raises:
             ValueError: If the column or row ranges are invalid or out of bounds.
         """
-        # validate column
-        c1, c2 = column_range
-        if c1 > c2:
-            column_range = (c2, c1)
-            c1, c2 = c2, c1
-        if c1 < 0:
-            raise ValueError("Column range must be between 0 ~ last-index.")
-        if c2 > self._num_column:
-            raise ValueError("Column range must be between 0 ~ last-index.")
-
-        # validate row
-        r1, r2 = row_range
-        if r1 > r2:
-            row_range = (r2, r1)
-            r1, r2 = r2, r1
-        if r1 < 0:
-            raise ValueError("Row range must be between 0 ~ last-index.")
-        if r2 > self._num_column:
-            raise ValueError("Row range must be between 0 ~ last-index.")
+        # validate
+        if width < 1:
+            raise ValueError("Grid cell must have 1+ columns")
+        if height < 1:
+            raise ValueError("Grid cell must have 1+ rows")
+
+        column_start, row_start = position
+        if not 0 <= column_start < self._num_column:
+            raise ValueError("Grid cell's column start position must be between 0 ~ last-column.")
+        if not 0 <= row_start < self._num_row:
+            raise ValueError("Grid cell's row start position must be between 0 ~ last-column.")
+
+        column_end = column_start + width - 1
+        row_end = row_start + height - 1
+        if column_end >= self._num_column:
+            raise ValueError("Grid cell's column start position must be between 0 ~ last-column.")
+        if row_end >= self._num_row:
+            raise ValueError("Grid cell's row start position must be between 0 ~ last-column.")
 
         # string style to Style class
         if isinstance(style, str):
@@ -137,8 +138,8 @@ class GridLayout:
 
         # push
         item = _GridLayoutItem(
-            column_range=column_range,
-            row_range=row_range,
+            column_range=(column_start, column_end),
+            row_range=(row_start, row_end),
             r=r,
             text=text,
             style=style,

{drawlib-0.2.3.dev5 → drawlib-0.2.3.dev6}/drawlib/v0_2/private/smartarts/pyramid.py

@@ -19,6 +19,13 @@ from drawlib.v0_2.private.core_canvas.canvas import trapezoid, triangle
 from drawlib.v0_2.private.util import error_handler
 
 
+@dataclasses.dataclass
+class _PyramidItem:
+    style: ShapeStyle
+    text: str
+    textstyle: ShapeTextStyle
+
+
 class Pyramid:
     """Class for rendering smart art pyramid."""
 
@@ -116,6 +123,7 @@ class Pyramid:
         height: float,
         margin: float,
         align: Literal["bottom", "top", "left", "right"] = "bottom",
+        order: Literal["vertex_to_base", "base_to_vertex"] = "vertex_to_base",
     ) -> None:
         """Draw smart art pyramid.
 
@@ -125,6 +133,7 @@ class Pyramid:
             height (float): The heifht of the pyramid.
             margin (float): The margin between pyramid items.
             align (str): Alignment of a pyramid.
+            order (str): Item order. "vertex -> base" or "base -> vertex". default is "vertex -> base".
         """
         if len(self._items) == 0:
             raise ValueError("Number of pyramid item is 0.")
@@ -139,6 +148,7 @@ class Pyramid:
             item_heights=item_heights,
             margins=margins,
             align=align,
+            order=order,
         )
 
     @error_handler
@@ -149,6 +159,7 @@ class Pyramid:
         item_heights: List[float],
         margins: List[float],
         align: Literal["bottom", "top", "left", "right"] = "bottom",
+        order: Literal["vertex_to_base", "base_to_vertex"] = "vertex_to_base",
     ) -> None:
         """Draw smart art pyramid with flexible pyramid item heights.
 
@@ -158,6 +169,7 @@ class Pyramid:
             item_heights (float): The height of the each pyramid items.
             margins (float): The margin between pyramid items.
             align (str): Alignment of a pyramid.
+            order (str): Item order. "vertex -> base" or "base -> vertex". default is "vertex -> base".
 
         Raises:
             ValueError: If the lengths of column_widths, column_margins, row_heights, or row_margins are incorrect.
@@ -172,35 +184,38 @@ class Pyramid:
         if len(self._items) != len(margins) + 1:
             raise ValueError('Number of pyramid item and arg "margins" length does not match.')
 
+        items = self._items[::-1] if order == "vertex_to_base" else self._items[::]
+
         if align == "bottom":
-            self._draw_flexible_bottom(xy, width, item_heights, margins)
+            self._draw_flexible_bottom(xy, width, item_heights, margins, items)
         elif align == "top":
-            self._draw_flexible_top(xy, width, item_heights, margins)
+            self._draw_flexible_top(xy, width, item_heights, margins, items)
         elif align == "left":
-            self._draw_flexible_left(xy, width, item_heights, margins)
+            self._draw_flexible_left(xy, width, item_heights, margins, items)
         elif align == "right":
-            self._draw_flexible_right(xy, width, item_heights, margins)
+            self._draw_flexible_right(xy, width, item_heights, margins, items)
         else:
             raise ValueError("Drawlib internal error.")
 
+    @staticmethod
     def _draw_flexible_bottom(
-        self,
         xy: Tuple[float, float],
         width: float,
         item_heights: List[float],
         margins: List[float],
+        items: List[_PyramidItem],
     ) -> None:
         x = xy[0] + width / 2
         height = sum(item_heights) + sum(margins)
         current_height = 0
-        for i, item in enumerate(self._items):
+        for i, item in enumerate(items):
             text = item.text
             style = item.style
             style.halign = "center"
             style.valign = "bottom"
             textstyle = item.textstyle
 
-            is_last = i == len(self._items) - 1
+            is_last = i == len(items) - 1
             if is_last:
                 ratio = (height - current_height) / height
                 item_width = ratio * width
@@ -231,24 +246,25 @@ class Pyramid:
             )
             current_height += item_height + margins[i]
 
+    @staticmethod
     def _draw_flexible_top(
-        self,
         xy: Tuple[float, float],
         width: float,
         item_heights: List[float],
         margins: List[float],
+        items: List[_PyramidItem],
     ) -> None:
         x = xy[0] + width / 2
         height = sum(item_heights) + sum(margins)
         current_height = 0
-        for i, item in enumerate(self._items):
+        for i, item in enumerate(items):
             text = item.text
             style = item.style
             style.halign = "center"
             style.valign = "bottom"
             textstyle = item.textstyle
 
-            is_last = i == len(self._items) - 1
+            is_last = i == len(items) - 1
             if is_last:
                 ratio = (height - current_height) / height
                 item_width = ratio * width
@@ -284,17 +300,18 @@ class Pyramid:
             )
             current_height += item_height + margins[i]
 
+    @staticmethod
     def _draw_flexible_left(
-        self,
         xy: Tuple[float, float],
         width: float,
         item_heights: List[float],
         margins: List[float],
+        items: List[_PyramidItem],
     ) -> None:
         y = xy[1] + width / 2
         height = sum(item_heights) + sum(margins)
         current_height = 0
-        for i, item in enumerate(self._items):
+        for i, item in enumerate(items):
             text = item.text
             style = item.style
             style.halign = "center"
@@ -304,7 +321,7 @@ class Pyramid:
             if textstyle.angle is None:
                 textstyle.angle = 0
 
-            is_last = i == len(self._items) - 1
+            is_last = i == len(items) - 1
             if is_last:
                 ratio = (height - current_height) / height
                 item_width = ratio * width
@@ -339,17 +356,18 @@ class Pyramid:
             )
             current_height += item_height + margins[i]
 
+    @staticmethod
     def _draw_flexible_right(
-        self,
         xy: Tuple[float, float],
         width: float,
         item_heights: List[float],
         margins: List[float],
+        items: List[_PyramidItem],
     ) -> None:
         y = xy[1] + width / 2
         height = sum(item_heights) + sum(margins)
         current_height = 0
-        for i, item in enumerate(self._items):
+        for i, item in enumerate(items):
             text = item.text
             style = item.style
             style.halign = "center"
@@ -358,7 +376,7 @@ class Pyramid:
             if textstyle.angle is None:
                 textstyle.angle = 0
 
-            is_last = i == len(self._items) - 1
+            is_last = i == len(items) - 1
             if is_last:
                 ratio = (height - current_height) / height
                 item_width = ratio * width
@@ -392,10 +410,3 @@ class Pyramid:
                 angle=90,
             )
             current_height += item_height + margins[i]
-
-
-@dataclasses.dataclass
-class _PyramidItem:
-    style: ShapeStyle
-    text: str
-    textstyle: ShapeTextStyle

{drawlib-0.2.3.dev5 → drawlib-0.2.3.dev6}/pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "drawlib"
-version = "0.2.3.dev5"
+version = "0.2.3.dev6"
 description = "Python drawing library. Illustration as Code."
 homepage = "https://www.drawlib.com"
 repository = "https://github.com/yuichi110/drawlib"

drawlib-0.2.3.dev5/drawlib/v0_2/private/smartarts/boxlist.py

@@ -1,178 +0,0 @@
-# Copyright (c) 2024 Yuichi Ito (yuichi@yuichi.com)
-#
-# This software is licensed under the Apache License, Version 2.0.
-# For more information, please visit: https://github.com/yuichi110/drawlib
-#
-# This software is provided "as is", without warranty of any kind,
-# express or implied, including but not limited to the warranties of
-# merchantability, fitness for a particular purpose and noninfringement.
-
-
-"""BoxList implementation module."""
-
-from typing import Any, List, Literal, Optional, Tuple, Union
-
-from drawlib.v0_2.private.core.colors import Colors
-from drawlib.v0_2.private.core.model import ShapeStyle, ShapeTextStyle
-from drawlib.v0_2.private.core.theme import dtheme
-from drawlib.v0_2.private.core_canvas.canvas import rectangle
-from drawlib.v0_2.private.util import error_handler
-
-
-class BoxList:
-    """A class to draw a list of boxes with text, supporting highlighting of certain boxes.
-
-    Args:
-        box_style (Union[str, ShapeStyle, None]): The style for the boxes.
-        text_style (Union[str, ShapeTextStyle, None]): The style for the text inside the boxes.
-        box_highlight_style (Union[str, ShapeStyle, None]): The style for the highlighted boxes.
-        text_highlight_style (Union[str, ShapeTextStyle, None]): The style for the text inside the highlighted boxes.
-    """
-
-    @error_handler
-    def __init__(
-        self,
-        box_style: Union[str, ShapeStyle, None] = None,
-        text_style: Union[str, ShapeTextStyle, None] = None,
-        box_highlight_style: Union[str, ShapeStyle, None] = None,
-        text_highlight_style: Union[str, ShapeTextStyle, None] = None,
-    ) -> None:
-        """Initialize BoxList.
-
-        Args:
-            box_style (Union[str, ShapeStyle, None]): The style for the boxes.
-            text_style (Union[str, ShapeTextStyle, None]): The style for the text inside the boxes.
-            box_highlight_style (Union[str, ShapeStyle, None]): The style for the highlighted boxes.
-            text_highlight_style (Union[str, ShapeTextStyle, None]):
-                    The style for the text inside the highlighted boxes.
-        """
-        if isinstance(box_style, str):
-            box_style = dtheme.rectanglestyles.get(box_style)
-        if box_style is None:
-            box_style = dtheme.rectanglestyles.get()
-        box_style.halign = "center"
-        box_style.valign = "center"
-        self._box_style = box_style
-
-        if isinstance(text_style, str):
-            text_style = dtheme.rectangletextstyles.get(text_style)
-        if text_style is None:
-            text_style = dtheme.rectangletextstyles.get()
-        self._text_style = text_style
-
-        if isinstance(box_highlight_style, str):
-            box_highlight_style = dtheme.rectanglestyles.get(box_highlight_style)
-        if box_highlight_style is None:
-            box_highlight_style = box_style
-        box_highlight_style.halign = "center"
-        box_highlight_style.valign = "center"
-        self._box_highlight_style = box_highlight_style
-
-        if isinstance(text_highlight_style, str):
-            text_highlight_style = dtheme.rectangletextstyles.get(text_highlight_style)
-        if text_highlight_style is None:
-            text_highlight_style = text_style
-        self._text_highlight_style = text_highlight_style
-
-    @error_handler
-    def draw(
-        self,
-        xy: Tuple[float, float],
-        box_width: float,
-        box_height: float,
-        items: List[Any],
-        highlight_indexs: Optional[List[int]] = None,
-        length: Optional[int] = None,
-        align: Literal["left", "right", "bottom", "top"] = "left",
-    ) -> None:
-        """Draws a list of boxes at the specified location.
-
-        Args:
-            xy (Tuple[float, float]): The starting point (x, y) to draw the list of boxes.
-            box_width (float): The width of each box.
-            box_height (float): The height of each box.
-            items (List[Any]): The list of items to be displayed inside the boxes.
-            highlight_indexs (Optional[List[int]]):
-                    The list of indices of the boxes to be highlighted.
-            length (Optional[int]): The total number of boxes to be drawn.
-            align (Literal["left", "right", "bottom", "top"]):
-                    The alignment of the boxes relative to the starting point.
-
-        """
-        if highlight_indexs is None:
-            highlight_indexs = []
-
-        if length is not None:
-            indexs = range(length)
-        else:
-            indexs = range(len(items))
-
-        non_highlight_indexs = [item for item in indexs if item not in highlight_indexs]
-        highlight_indexs = [item for item in indexs if item in highlight_indexs]
-
-        for index in non_highlight_indexs:
-            if index <= len(items) - 1:
-                text = str(items[index])
-            else:
-                text = ""
-
-            self._draw_cell(
-                start_xy=xy,
-                index=index,
-                text=text,
-                box_width=box_width,
-                box_height=box_height,
-                box_style=self._box_style,
-                text_style=self._text_style,
-                align=align,
-            )
-
-        for index in highlight_indexs:
-            if index <= len(items) - 1:
-                text = str(items[index])
-            else:
-                text = ""
-
-            self._draw_cell(
-                start_xy=xy,
-                index=index,
-                text=text,
-                box_width=box_width,
-                box_height=box_height,
-                box_style=self._box_highlight_style,
-                text_style=self._text_highlight_style,
-                align=align,
-            )
-
-    @staticmethod
-    def _draw_cell(
-        start_xy: Tuple[float, float],
-        index: int,
-        text: str,
-        box_width: float,
-        box_height: float,
-        box_style: ShapeStyle,
-        text_style: ShapeTextStyle,
-        align: Literal["left", "right", "bottom", "top"],
-    ) -> None:
-        if align == "left":
-            x = start_xy[0] + box_width * (index + 0.5)
-            y = start_xy[1] + box_height / 2
-        elif align == "right":
-            x = start_xy[0] - box_width * (index + 0.5)
-            y = start_xy[1] + box_height / 2
-        elif align == "bottom":
-            x = start_xy[0] + box_width / 2
-            y = start_xy[1] + box_height * (index + 0.5)
-        else:
-            x = start_xy[0] + box_width / 2
-            y = start_xy[1] - box_height * (index + 0.5)
-
-        rectangle(
-            xy=(x, y),
-            width=box_width,
-            height=box_height,
-            style=box_style,
-            text=text,
-            textstyle=text_style,
-        )