drawlib 0.2.2__tar.gz → 0.2.3.dev1__tar.gz

{drawlib-0.2.2 → drawlib-0.2.3.dev1}/PKG-INFO

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

{drawlib-0.2.2 → drawlib-0.2.3.dev1}/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.2"
+LIB_VERSION = "0.2.3.dev1"
 
 # please list active main committers (1+ commits per month)
 AUTHORS: Final[List[str]] = [

drawlib-0.2.3.dev1/drawlib/v0_2/private/smartarts/boxtree.py

@@ -0,0 +1,114 @@
+# 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.
+
+
+"""BoxTree implementation module."""
+
+from __future__ import annotations
+
+import dataclasses
+from typing import Callable, Dict, List, Literal, Optional, Tuple, Union
+
+from drawlib.v0_2.private.core.model import LineStyle, ShapeStyle, ShapeTextStyle
+from drawlib.v0_2.private.core.theme import dtheme
+from drawlib.v0_2.private.core_canvas.canvas import line, rectangle
+
+
+class BoxTreeNode:
+    """Class for rendering smart art Tree."""
+
+    def __init__(
+        self,
+        text: str,
+        boxsize: Optional[Tuple[float, float]] = None,
+        boxstyle: Union[str, ShapeStyle, None] = None,
+        box_r: Optional[float] = None,
+        box_horizontal_margin: Optional[float] = None,
+        box_vertical_margin: Optional[float] = None,
+        textstyle: Union[str, ShapeTextStyle, None] = None,
+        linestyle: Union[str, LineStyle, None] = None,
+        line_horizontal_length: Optional[float] = None,
+        line_vertical_length: Optional[float] = None,
+        children: Optional[List[BoxTreeNode]] = None,
+        default_boxsize: Optional[Tuple[float, float]] = None,
+        default_boxstyle: Union[str, ShapeStyle, None] = None,
+        default_box_r: Optional[float] = None,
+        default_box_horizontal_margin: Optional[float] = None,
+        default_box_vertical_margin: Optional[float] = None,
+        default_textstyle: Union[str, ShapeTextStyle, None] = None,
+        default_linestyle: Union[str, LineStyle, None] = None,
+        default_line_horizontal_length: Optional[float] = None,
+        default_line_vertical_length: Optional[float] = None,
+    ) -> None:
+        """Initialize class."""
+        self._text = text
+
+        self._boxsize = boxsize
+        if isinstance(boxstyle, str):
+            boxstyle = dtheme.rectanglestyles.get(boxstyle)
+        self._boxstyle = boxstyle
+        self._box_r = box_r
+        self._box_horizontal_margin = box_horizontal_margin
+        self._box_vertical_margin = box_vertical_margin
+
+        if isinstance(textstyle, str):
+            textstyle = dtheme.rectangletextstyles.get(textstyle)
+        self._textstyle = textstyle
+        if isinstance(linestyle, str):
+            linestyle = dtheme.linestyles.get(linestyle)
+        self._linestyle = linestyle
+
+        self._line_horizontal_length = line_horizontal_length
+        self._line_vertical_length = line_vertical_length
+
+        if children is None:
+            self._children: List[BoxTreeNode] = []
+        else:
+            self._children: List[BoxTreeNode] = children
+
+        self._default_boxsize = default_boxsize
+        if isinstance(default_boxstyle, str):
+            default_boxstyle = dtheme.rectanglestyles.get(default_boxstyle)
+        self._default_boxstyle = default_boxstyle
+        self._default_box_r = default_box_r
+        self._default_box_horizontal_margin = default_box_horizontal_margin
+        self._default_box_vertical_margin = default_box_vertical_margin
+
+        if isinstance(default_textstyle, str):
+            default_textstyle = dtheme.rectangletextstyles.get(default_textstyle)
+        self._default_textstyle: Union[ShapeTextStyle, None] = default_textstyle
+        if isinstance(default_linestyle, str):
+            default_linestyle = dtheme.linestyles.get(default_linestyle)
+        self._default_linestyle: Union[LineStyle, None] = default_linestyle
+
+        self._default_line_horizontal_length = default_line_horizontal_length
+        self._default_line_vertical_length = default_line_vertical_length
+
+        self._drawing_item_name: Optional[str] = None
+
+    def draw(
+        self,
+        xy: Tuple[float, float],
+        orientation: Literal["horizontal", "vertical"],
+        align: Literal["top", "bottom", "center", "left", "right"],
+    ) -> None:
+        """Write docstring later."""
+
+    def _draw_vertical_left(
+        self,
+        xy: Tuple[float, float],
+        orientation: Literal["horizontal", "vertical"],
+        align: Literal["top", "bottom", "center", "left", "right"],
+        default_textstyle: ShapeTextStyle,
+        default_linestyle: LineStyle,
+        default_line_horizontal_margin: float,
+        default_line_horizontal_length: float,
+        default_line_vertical_margin: float,
+    ) -> None:
+        """Write doc string later."""

{drawlib-0.2.2 → drawlib-0.2.3.dev1}/drawlib/v0_2/private/smartarts/dsart.py

@@ -12,4 +12,7 @@
 """Smart art module."""
 
 from drawlib.v0_2.private.smartarts.bubblespeech import bubblespeech
+from drawlib.v0_2.private.smartarts.gridlayout import GridLayout
+from drawlib.v0_2.private.smartarts.pyramid import Pyramid
 from drawlib.v0_2.private.smartarts.sourcecode import SourceCode
+from drawlib.v0_2.private.smartarts.tree import TreeNode

drawlib-0.2.3.dev1/drawlib/v0_2/private/smartarts/gridlayout.py

@@ -0,0 +1,275 @@
+# 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.
+
+
+"""GridLayout implementation module."""
+
+import dataclasses
+from typing import List, Optional, Tuple, Union
+
+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
+
+
+class GridLayout:
+    """Class for rendering multiple rectangles which fit to grid."""
+
+    def __init__(
+        self,
+        num_column: int,
+        num_row: int,
+        default_r: int = 0,
+        default_style: Union[str, ShapeStyle, None] = None,
+        default_textstyle: Union[str, ShapeTextStyle, None] = None,
+    ) -> None:
+        """Initialize class."""
+        self._num_column = num_column
+        self._num_row = num_row
+        self._default_r = default_r
+        if isinstance(default_style, str):
+            default_style = dtheme.rectanglestyles.get(default_style)
+        self._default_style = default_style
+        if isinstance(default_textstyle, str):
+            default_textstyle = dtheme.rectangletextstyles.get(default_textstyle)
+        self._default_textstyle = default_textstyle
+
+        self._items: List[_GridLayoutItem] = []
+
+    def add(  # noqa: C901
+        self,
+        column_range: Tuple[int, int],
+        row_range: Tuple[int, int],
+        r: Optional[int] = None,
+        style: Union[str, ShapeStyle, None] = None,
+        text: str = "",
+        textstyle: Union[str, ShapeTextStyle, None] = None,
+    ) -> None:
+        """
+        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.
+            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.
+                    Can be a string key for a predefined style, a ShapeStyle object,
+                    or None to use the default style.
+            text (str, optional): The text associated with the item. Default is an empty string.
+            textstyle (Union[str, ShapeTextStyle, None], optional):
+                    The text style of the item. Can be a string key for a predefined text style,
+                    a ShapeTextStyle object, or None to use the default text style.
+
+        Raises:
+            ValueError: If the column or row ranges are invalid or out of bounds.
+        """
+        # validate column
+        c1, c2 = column_range
+        if c1 > c2:
+            raise ValueError()
+        if c1 < 0:
+            raise ValueError()
+        if c2 > self._num_column:
+            raise ValueError()
+
+        # validate row
+        r1, r2 = row_range
+        if r1 > r2:
+            raise ValueError()
+        if r1 < 0:
+            raise ValueError()
+        if r2 > self._num_column:
+            raise ValueError()
+
+        # string style to Style class
+        if isinstance(style, str):
+            style = dtheme.rectanglestyles.get(style)
+        if isinstance(textstyle, str):
+            textstyle = dtheme.rectangletextstyles.get(textstyle)
+
+        # apply default if args are None
+        if r is None:
+            r = self._default_r
+        if style is None:
+            style = self._default_style
+        if textstyle is None:
+            textstyle = self._default_textstyle
+
+        # push
+        item = _GridLayoutItem(
+            column_range=column_range,
+            row_range=row_range,
+            r=r,
+            text=text,
+            style=style,
+            textstyle=textstyle,
+        )
+        self._items.append(item)
+
+    def draw(
+        self,
+        xy: Tuple[float, float],
+        width: float,
+        height: float,
+        margin: float,
+        outer_r: int = 0,
+        outer_style: Union[str, ShapeStyle, None] = None,
+    ) -> None:
+        """Draw the grid layout.
+
+        Args:
+            xy (Tuple[float, float]): The x and y coordinates of the top-left corner of the grid.
+            width (float): The total width of the grid.
+            height (float): The total height of the grid.
+            margin (float): The margin between grid items.
+            outer_r (int, optional): The radius for the outer grid border. Default is 0.
+            outer_style (Union[str, ShapeStyle, None], optional):
+                    The style for the outer grid border. Can be a string key for a predefined style,
+                    a ShapeStyle object, or None.
+        """
+        column_widths = [width / self._num_column] * self._num_column
+        row_heights = [height / self._num_row] * self._num_row
+
+        if outer_style is None:
+            column_margins = [margin] + [margin * 2] * (self._num_column - 1) + [margin]
+            row_margins = [margin] + [margin * 2] * (self._num_row - 1) + [margin]
+        else:
+            column_margins = [margin * 2] * (self._num_column + 1)
+            row_margins = [margin * 2] * (self._num_row + 1)
+
+        self.draw_flexible(
+            xy=xy,
+            column_widths=column_widths,
+            column_margins=column_margins,
+            row_heights=row_heights,
+            row_margins=row_margins,
+            outer_r=outer_r,
+            outer_style=outer_style,
+        )
+
+    def draw_flexible(  # noqa: C901
+        self,
+        xy: Tuple[float, float],
+        column_widths: List[float],
+        column_margins: List[float],
+        row_heights: List[float],
+        row_margins: List[float],
+        outer_r: int = 0,
+        outer_style: Union[str, ShapeStyle, None] = None,
+    ) -> None:
+        """Draw the grid layout with flexible column widths and row heights.
+
+        Args:
+            xy (Tuple[float, float]): The x and y coordinates of the top-left corner of the grid.
+            column_widths (List[float]): The widths of each column.
+            column_margins (List[float]): The margins between columns.
+            row_heights (List[float]): The heights of each row.
+            row_margins (List[float]): The margins between rows.
+            outer_r (int, optional): The radius for the outer grid border. Default is 0.
+            outer_style (Union[str, ShapeStyle, None], optional):
+                    The style for the outer grid border.
+                    Can be a string key for a predefined style, a ShapeStyle object, or None.
+
+        Raises:
+            ValueError: If the lengths of column_widths, column_margins, row_heights, or row_margins are incorrect.
+        """
+        # validate
+        if len(column_widths) != self._num_column:
+            raise ValueError()
+        if len(column_margins) != self._num_column + 1:
+            raise ValueError()
+        if len(row_heights) != self._num_row:
+            raise ValueError()
+        if len(row_margins) != self._num_row + 1:
+            raise ValueError()
+
+        # draw outer rectangle
+        if outer_style is not None:
+            if isinstance(outer_style, str):
+                outer_style = dtheme.rectanglestyles.get(outer_style)
+            outer_style.halign = "left"
+            outer_style.valign = "bottom"
+            rectangle(
+                xy=xy,
+                width=sum(column_widths),
+                height=sum(row_heights),
+                r=outer_r,
+                style=outer_style,
+            )
+
+        # utility
+        def sum_till_index(lst: List[float], index: int) -> float:
+            if index == 0:
+                return 0
+            return sum([lst[i] for i in range(index)])
+
+        for item in self._items:
+            column_range = item.column_range
+            row_range = item.row_range
+            r = item.r
+            style = item.style
+            text = item.text
+            textstyle = item.textstyle
+
+            cr0 = column_range[0]
+            cr1 = column_range[1]
+            rr0 = row_range[0]
+            rr1 = row_range[1]
+
+            # coordinate without margin
+            lb_xy = (sum_till_index(column_widths, cr0), sum_till_index(row_heights, rr0))
+            width = sum_till_index(column_widths, cr1 + 1) - lb_xy[0]
+            height = sum_till_index(row_heights, rr1 + 1) - lb_xy[1]
+
+            # get margins
+            margin_left = column_margins[cr0] if cr0 == 0 else column_margins[cr0] / 2
+            margin_right = column_margins[cr1 + 1] if cr1 + 1 == self._num_column else column_margins[cr1 + 1] / 2
+            margin_bottom = row_margins[rr0] if rr0 == 0 else row_margins[rr0] / 2
+            margin_top = row_margins[rr1 + 1] if rr1 + 1 == self._num_row else row_margins[rr1 + 1] / 2
+
+            # update coordinate
+            lb_xy = (lb_xy[0] + margin_left, lb_xy[1] + margin_bottom)
+            width = width - margin_left - margin_right
+            height = height - margin_bottom - margin_top
+
+            # set align left bottom
+            if style is None:
+                style = self._default_style
+            if style is None:
+                style = dtheme.rectanglestyles.get()
+            else:
+                style = style.copy()
+            style.halign = "left"
+            style.valign = "bottom"
+
+            # apply default text style if specified text style is None
+            if textstyle is None:
+                textstyle = self._default_textstyle
+
+            # draw
+            rectangle(
+                xy=(xy[0] + lb_xy[0], xy[1] + lb_xy[1]),
+                width=width,
+                height=height,
+                r=r,
+                style=style,
+                text=text,
+                textstyle=textstyle,
+            )
+
+
+@dataclasses.dataclass
+class _GridLayoutItem:
+    column_range: Tuple[int, int]
+    row_range: Tuple[int, int]
+    r: int
+    style: Optional[ShapeStyle]
+    text: str
+    textstyle: Optional[ShapeTextStyle]

drawlib-0.2.3.dev1/drawlib/v0_2/private/smartarts/pyramid.py

@@ -0,0 +1,189 @@
+# 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.
+
+
+"""Pyramid implementation module."""
+
+import dataclasses
+from typing import List, Optional, Tuple, Union
+
+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 trapezoid, triangle
+
+
+class Pyramid:
+    """Class for rendering smart art pyramid."""
+
+    def __init__(
+        self,
+        default_style: Union[str, ShapeStyle, None] = None,
+        default_textstyle: Union[str, ShapeTextStyle, None] = None,
+    ) -> None:
+        """Initialize class."""
+        if isinstance(default_style, str):
+            default_style = dtheme.rectanglestyles.get(default_style)
+        self._default_style = default_style
+        if isinstance(default_textstyle, str):
+            default_textstyle = dtheme.rectangletextstyles.get(default_textstyle)
+        self._default_textstyle = default_textstyle
+
+        self._items: List[_PyramidItem] = []
+
+    def add(  # noqa: C901
+        self,
+        style: Union[str, ShapeStyle, None] = None,
+        text: str = "",
+        textstyle: Union[str, ShapeTextStyle, None] = None,
+    ) -> None:
+        """
+        Add an item to the pyramid.
+
+        Args:
+            style (Union[str, ShapeStyle, None], optional):
+                    The style of the item.
+                    Can be a string key for a predefined style, a ShapeStyle object,
+                    or None to use the default style.
+            text (str, optional): The text associated with the item. Default is an empty string.
+            textstyle (Union[str, ShapeTextStyle, None], optional):
+                    The text style of the item. Can be a string key for a predefined text style,
+                    a ShapeTextStyle object, or None to use the default text style.
+        """
+        # string style to Style class
+        if isinstance(style, str):
+            style = dtheme.rectanglestyles.get(style)
+        if isinstance(textstyle, str):
+            textstyle = dtheme.rectangletextstyles.get(textstyle)
+
+        # apply default if args are None
+        if style is None:
+            style = self._default_style
+        if textstyle is None:
+            textstyle = self._default_textstyle
+
+        # push
+        item = _PyramidItem(
+            text=text,
+            style=style,
+            textstyle=textstyle,
+        )
+        self._items.append(item)
+
+    def draw(
+        self,
+        xy: Tuple[float, float],
+        width: float,
+        height: float,
+        margin: float,
+    ) -> None:
+        """Draw smart art pyramid.
+
+        Args:
+            xy (Tuple[float, float]): The x and y coordinates of the bottom-left corner of the pyramid.
+            width (float): The width of the pyramid.
+            height (float): The heifht of the pyramid.
+            margin (float): The margin between pyramid items.
+        """
+        if len(self._items) == 0:
+            raise ValueError()
+
+        margins = [margin] * (len(self._items) - 1)
+        item_height = (height - sum(margins)) / len(self._items)
+        item_heights = [item_height] * len(self._items)
+
+        self.draw_flexible(
+            xy=xy,
+            width=width,
+            item_heights=item_heights,
+            margins=margins,
+        )
+
+    def draw_flexible(  # noqa: C901
+        self,
+        xy: Tuple[float, float],
+        width: float,
+        item_heights: List[float],
+        margins: List[float],
+    ) -> None:
+        """Draw smart art pyramid with flexible pyramid item heights.
+
+        Args:
+            xy (Tuple[float, float]): The x and y coordinates of the bottom-left corner of the pyramid.
+            width (float): The width of the pyramid.
+            item_heights (float): The height of the each pyramid items.
+            margins (float): The margin between pyramid items.
+
+        Raises:
+            ValueError: If the lengths of column_widths, column_margins, row_heights, or row_margins are incorrect.
+        """
+        # validate
+        if len(self._items) == 0:
+            raise ValueError()
+
+        if len(self._items) != len(item_heights):
+            raise ValueError()
+
+        if len(self._items) != len(margins) + 1:
+            raise ValueError()
+
+        x = xy[0] + width / 2
+        height = sum(item_heights) + sum(margins)
+        current_height = 0
+        for i, item in enumerate(self._items):
+            text = item.text
+            textstyle = item.textstyle
+            if textstyle is None:
+                textstyle = self._default_textstyle
+            style = item.style
+            if style is None:
+                style = self._default_style
+            if style is None:
+                style = dtheme.rectanglestyles.get()
+            else:
+                style = style.copy()
+            style.halign = "center"
+            style.valign = "bottom"
+
+            is_last = i == len(self._items) - 1
+            if is_last:
+                ratio = (height - current_height) / height
+                item_width = ratio * width
+                item_height = item_heights[i]
+                triangle(
+                    (x, xy[1] + current_height),
+                    width=item_width,
+                    height=item_height,
+                    style=style,
+                    text=text,
+                    textstyle=textstyle,
+                )
+                continue
+
+            bottom_ratio = (height - current_height) / height
+            bottom_width = bottom_ratio * width
+            item_height = item_heights[i]
+            top_ratio = (height - current_height - item_height) / height
+            top_width = top_ratio * width
+            trapezoid(
+                (x, xy[1] + current_height),
+                item_height,
+                bottomedge_width=bottom_width,
+                topedge_width=top_width,
+                style=style,
+                text=text,
+                textstyle=textstyle,
+            )
+            current_height += item_height + margins[i]
+
+
+@dataclasses.dataclass
+class _PyramidItem:
+    style: Optional[ShapeStyle]
+    text: str
+    textstyle: Optional[ShapeTextStyle]

drawlib-0.2.3.dev1/drawlib/v0_2/private/smartarts/tree.py

@@ -0,0 +1,274 @@
+# 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.
+
+
+"""Tree implementation module."""
+
+from __future__ import annotations
+
+import dataclasses
+from typing import Callable, Dict, List, Literal, Optional, Tuple, Union
+
+from drawlib.v0_2.private.core.model import IconStyle, ImageStyle, LineStyle, ShapeStyle, TextStyle
+from drawlib.v0_2.private.core.theme import dtheme
+from drawlib.v0_2.private.core_canvas.canvas import line, text
+
+
+class TreeNode:
+    """Class for rendering smart art Tree."""
+
+    _drawing_item_map: Dict[str, _TreeNodeDrawingItem] = {}
+
+    def __init__(
+        self,
+        text: str,
+        textstyle: Union[str, TextStyle, None] = None,
+        linestyle: Union[str, LineStyle, None] = None,
+        line_horizontal_margin: Optional[float] = None,
+        line_horizontal_length: Optional[float] = None,
+        line_vertical_margin: Optional[float] = None,
+        children: Optional[List[TreeNode]] = None,
+        default_textstyle: Union[str, TextStyle, None] = None,
+        default_linestyle: Union[str, LineStyle, None] = None,
+        default_line_horizontal_margin: Optional[float] = None,
+        default_line_horizontal_length: Optional[float] = None,
+        default_line_vertical_margin: Optional[float] = None,
+    ) -> None:
+        """Initialize class."""
+        self._text = text
+
+        if isinstance(textstyle, str):
+            textstyle = dtheme.textstyles.get(textstyle)
+        self._textstyle = textstyle
+        if isinstance(linestyle, str):
+            linestyle = dtheme.linestyles.get(linestyle)
+        self._linestyle = linestyle
+
+        self._line_horizontal_margin = line_horizontal_margin
+        self._line_horizontal_length = line_horizontal_length
+        self._line_vertical_margin = line_vertical_margin
+
+        if children is None:
+            self._children: List[TreeNode] = []
+        else:
+            self._children: List[TreeNode] = children
+
+        if isinstance(default_textstyle, str):
+            default_textstyle = dtheme.textstyles.get(default_textstyle)
+        self._default_textstyle: Union[TextStyle, None] = default_textstyle
+        if isinstance(default_linestyle, str):
+            default_linestyle = dtheme.linestyles.get(default_linestyle)
+        self._default_linestyle: Union[LineStyle, None] = default_linestyle
+
+        self._default_line_horizontal_margin: Optional[float] = default_line_horizontal_margin
+        self._default_line_horizontal_length: Optional[float] = default_line_horizontal_length
+        self._default_line_vertical_margin: Optional[float] = default_line_vertical_margin
+
+        self._drawing_item_name: Optional[str] = None
+
+    @classmethod
+    def register_drawing_item(
+        cls,
+        name: str,
+        location: Literal["before", "after"],
+        padding_width: float,
+        function: Callable,
+        style: Union[IconStyle, ImageStyle, ShapeStyle, TextStyle],
+        args: dict,
+    ) -> None:
+        """Register a drawing item for the tree node.
+
+        Args:
+            name (str): Name of drawing item.
+            location (Literal["before", "after"]): The location of the drawing item relative to the text.
+            padding_width (float): The padding width for the drawing item.
+            function (Callable): The function to render the drawing item.
+            style (Union[IconStyle, ImageStyle, ShapeStyle, TextStyle]): The style for the drawing item.
+            args (dict): The arguments for the function.
+
+        Returns:
+            TreeNode: The current tree node instance.
+        """
+        style.halign = "left"
+        style.valign = "center"
+
+        item = _TreeNodeDrawingItem(
+            location=location,
+            padding_width=padding_width,
+            function=function,
+            style=style,
+            args=args,
+        )
+
+        cls._drawing_item_map[name] = item
+
+    def set_drawing_item(
+        self,
+        name: str,
+    ) -> TreeNode:
+        """Set a drawing item for the tree node.
+
+        Args:
+            name (str): Name of drawing item.
+
+        Returns:
+            TreeNode: The current tree node instance.
+        """
+        if name not in self._drawing_item_map:
+            raise ValueError(f'Drawing item "{name}" is not registered.')
+        self._drawing_item_name = name
+
+        return self
+
+    def draw(self, xy: Tuple[float, float]) -> None:
+        """Draw the tree node and its children.
+
+        Args:
+            xy (Tuple[float, float]): The coordinates to start drawing.
+
+        Raises:
+            ValueError: If any of the default styles or margins are None.
+        """
+        if self._default_textstyle is None:
+            raise ValueError()
+        if self._default_linestyle is None:
+            raise ValueError()
+        if self._default_line_horizontal_margin is None:
+            raise ValueError()
+        if self._default_line_horizontal_length is None:
+            raise ValueError()
+        if self._default_line_vertical_margin is None:
+            raise ValueError()
+
+        self._draw(
+            xy=xy,
+            default_textstyle=self._default_textstyle,
+            default_linestyle=self._default_linestyle,
+            default_line_horizontal_margin=self._default_line_horizontal_margin,
+            default_line_horizontal_length=self._default_line_horizontal_length,
+            default_line_vertical_margin=self._default_line_vertical_margin,
+        )
+
+    def _draw(  # noqa: C901
+        self,
+        xy: Tuple[float, float],
+        default_textstyle: TextStyle,
+        default_linestyle: LineStyle,
+        default_line_horizontal_margin: float,
+        default_line_horizontal_length: float,
+        default_line_vertical_margin: float,
+    ) -> float:
+        """Draw the tree node and its children (internal method).
+
+        Args:
+            xy (Tuple[float, float]): The coordinates to start drawing.
+            default_textstyle (TextStyle): The default text style.
+            default_linestyle (LineStyle): The default line style.
+            default_line_horizontal_margin (float): The default horizontal line margin.
+            default_line_horizontal_length (float): The default horizontal line length.
+            default_line_vertical_margin (float): The default vertical line margin.
+
+        Returns:
+            float: The y-coordinate after drawing the node and its children.
+        """
+        # update default values
+        if self._default_textstyle is not None:
+            default_textstyle = self._default_textstyle
+        if self._default_linestyle is not None:
+            default_linestyle = self._default_linestyle
+        if self._default_line_horizontal_margin is not None:
+            default_line_horizontal_margin = self._default_line_horizontal_margin
+        if self._default_line_horizontal_length is not None:
+            default_line_horizontal_length = self._default_line_horizontal_length
+        if self._default_line_vertical_margin is not None:
+            default_line_vertical_margin = self._default_line_vertical_margin
+
+        # update values
+        textstyle = self._textstyle if self._textstyle is not None else default_textstyle
+        linestyle = self._linestyle if self._linestyle is not None else default_linestyle
+        if self._line_horizontal_margin is not None:
+            line_horizontal_margin = self._line_horizontal_margin
+        else:
+            line_horizontal_margin = default_line_horizontal_margin
+        if self._line_horizontal_length is not None:
+            line_horizontal_length = self._line_horizontal_length
+        else:
+            line_horizontal_length = default_line_horizontal_length
+        if self._line_vertical_margin is not None:
+            line_vertical_margin = self._line_vertical_margin
+        else:
+            line_vertical_margin = default_line_vertical_margin
+
+        # draw text
+        if self._drawing_item_name is None:
+            textstyle.halign = "left"
+            text(xy=xy, text=self._text, style=textstyle)
+
+        else:
+            drawing_item = self._drawing_item_map[self._drawing_item_name]
+
+            if drawing_item.location == "before":
+                args = drawing_item.args
+                args["xy"] = xy
+                args["style"] = drawing_item.style
+                drawing_item.function(**args)
+
+                textstyle.halign = "left"
+                text(xy=(xy[0] + drawing_item.padding_width, xy[1]), text=self._text, style=textstyle)
+
+            else:
+                textstyle.halign = "left"
+                text(xy=xy, text=self._text, style=textstyle)
+
+                args = drawing_item.args
+                args["xy"] = (xy[0] + drawing_item.padding_width, xy[1])
+                args["style"] = drawing_item.style
+                drawing_item.function(**args)
+
+        # draw children
+        horizontal_line_x1 = xy[0] + line_horizontal_margin
+        horizontal_line_x2 = horizontal_line_x1 + line_horizontal_length * 2 / 3
+        child_x = horizontal_line_x1 + line_horizontal_margin
+        child_y = xy[1]
+        child_y_previous = child_y
+        for child in self._children:
+            child_y -= line_vertical_margin
+            # draw child horizontal line
+            line(
+                xy1=(horizontal_line_x1, child_y),
+                xy2=(horizontal_line_x2, child_y),
+                style=linestyle,
+            )
+            # draw child
+            child_y_previous = child_y
+            child_y = child._draw(
+                xy=(child_x, child_y),
+                default_textstyle=default_textstyle,
+                default_linestyle=default_linestyle,
+                default_line_horizontal_margin=default_line_horizontal_margin,
+                default_line_horizontal_length=default_line_horizontal_length,
+                default_line_vertical_margin=default_line_vertical_margin,
+            )
+        # draw children vertical line
+        if xy[1] != child_y_previous:
+            line(
+                (horizontal_line_x1, xy[1] - line_vertical_margin / 2),
+                (horizontal_line_x1, child_y_previous),
+                style=linestyle,
+            )
+        return child_y
+
+
+@dataclasses.dataclass
+class _TreeNodeDrawingItem:
+    location: Literal["before", "after"]
+    padding_width: float
+    function: Callable
+    style: Union[IconStyle, ImageStyle, ShapeStyle, TextStyle]
+    args: dict

drawlib-0.2.3.dev1/drawlib/v0_2/private/umls/activity.py

@@ -0,0 +1,10 @@
+# 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.
+
+"""Activity diagram implementation module."""

drawlib-0.2.3.dev1/drawlib/v0_2/private/umls/class_.py

@@ -0,0 +1,10 @@
+# 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.
+
+"""Class diagram implementation module."""

drawlib-0.2.3.dev1/drawlib/v0_2/private/umls/communication.py

@@ -0,0 +1,10 @@
+# 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.
+
+"""Communication diagram implementation module."""

drawlib-0.2.3.dev1/drawlib/v0_2/private/umls/component.py

@@ -0,0 +1,10 @@
+# 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.
+
+"""Component diagram implementation module."""

drawlib-0.2.3.dev1/drawlib/v0_2/private/umls/deployment.py

@@ -0,0 +1,10 @@
+# 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.
+
+"""Deployment diagram implementation module."""

drawlib-0.2.3.dev1/drawlib/v0_2/private/umls/duml.py

@@ -0,0 +1,12 @@
+# 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.
+
+# ruff: noqa: F401
+
+"""UML module."""

drawlib-0.2.3.dev1/drawlib/v0_2/private/umls/object.py

@@ -0,0 +1,10 @@
+# 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.
+
+"""Object diagram implementation module."""

drawlib-0.2.3.dev1/drawlib/v0_2/private/umls/package.py

@@ -0,0 +1,10 @@
+# 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.
+
+"""Package diagram implementation module."""

drawlib-0.2.3.dev1/drawlib/v0_2/private/umls/sequence.py

@@ -0,0 +1,10 @@
+# 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.
+
+"""Sequence diagram implementation module."""

drawlib-0.2.3.dev1/drawlib/v0_2/private/umls/state_machine.py

@@ -0,0 +1,10 @@
+# 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.
+
+"""State Machine diagram implementation module."""

drawlib-0.2.3.dev1/drawlib/v0_2/private/umls/usecase.py

@@ -0,0 +1,10 @@
+# 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.
+
+"""Usecase diagram implementation module."""

{drawlib-0.2.2 → drawlib-0.2.3.dev1}/pyproject.toml

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