drawlib 0.2.3.dev4__py3-none-any.whl → 0.2.3.dev5__py3-none-any.whl

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.dev4"
+LIB_VERSION = "0.2.3.dev5"
 
 # please list active main committers (1+ commits per month)
 AUTHORS: Final[List[str]] = [

drawlib/v0_2/private/core_canvas/original_arrow.py

@@ -11,7 +11,7 @@
 """Canvas's original arrow feature implementation module."""
 
 import math
-from typing import List, Literal, Optional, Tuple, Union
+from typing import Any, List, Literal, Optional, Tuple, Union
 
 from matplotlib.patches import PathPatch
 from matplotlib.path import Path
@@ -219,8 +219,8 @@ class CanvasOriginalArrowFeature(CanvasBase):
         new_xys.insert(0, xys_start)
         new_xys.append(xys_end)
         aph = ArrowPolylineHelper(new_xys, r, 100)
-        parallel_xys1 = aph.get_parallel_curve_points(tail_width / 2)
-        parallel_xys2 = aph.get_parallel_curve_points(tail_width / -2)
+        parallel_xys1 = aph.get_parallel_points(tail_width / 2)
+        parallel_xys2 = aph.get_parallel_points(tail_width / -2)
         parallel_xys2.reverse()
 
         if head in {"<-", "<->"}:
@@ -568,6 +568,14 @@ 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[:]
+            return
+
         points = []
         bezier_start: Tuple[float, float] = (0, 0)
         last_i = len(xys) - 2
@@ -592,24 +600,122 @@ class ArrowPolylineHelper:
             points.extend(get_points(bezier_points))
             bezier_start = p2
 
-        self._original_curve_points = points
+        self._original_points = remove_consecutive_duplicates(points)
 
-    def get_parallel_curve_points(self, distance: float) -> List[Tuple[float, float]]:
+    def get_parallel_points(self, distance: float) -> List[Tuple[float, float]]:
+        """Internal function"""
+        if self._r == 0:
+            return self._get_parallel_straight_points(distance)
+        return self._get_parallel_curved_points(distance)
+
+    def _get_parallel_curved_points(self, distance: float) -> List[Tuple[float, float]]:
         """Internal function"""
 
         def get_distance(p1: Tuple[float, float], p2: Tuple[float, float]) -> float:
             return math.sqrt((p1[0] - p2[0]) ** 2 + (p1[1] - p2[1]) ** 2)
 
         parallel_curve_points = []
-        for i in range(1, len(self._original_curve_points)):
-            p0, p1 = self._original_curve_points[i - 1], self._original_curve_points[i]
+        for i in range(1, len(self._original_points)):
+            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
             parallel_curve_points.append((px0, py0))
-            if i == len(self._original_curve_points) - 1:
+            if i == len(self._original_points) - 1:
                 parallel_curve_points.append((px1, py1))
 
         return parallel_curve_points
+
+    def _get_parallel_straight_points(self, distance: float) -> List[Tuple[float, float]]:
+        """Internal function"""
+
+        def get_parallel_line_xys(
+            xy1: Tuple[float, float],
+            xy2: Tuple[float, float],
+        ) -> Tuple[Tuple[float, float], Tuple[float, float]]:
+            """Internal function"""
+            x1, y1 = xy1
+            x2, y2 = xy2
+
+            # calc vector
+            vx = x2 - x1
+            vy = y2 - y1
+            length = math.sqrt(vx**2 + vy**2)
+            dx = -vy / length * distance
+            dy = vx / length * distance
+
+            # add vector
+            x1_prime = x1 + dx
+            y1_prime = y1 + dy
+            x2_prime = x2 + dx
+            y2_prime = y2 + dy
+
+            return (x1_prime, y1_prime), (x2_prime, y2_prime)
+
+        def find_lines_intersection(
+            xy1: Tuple[float, float],
+            xy2: Tuple[float, float],
+            xy3: Tuple[float, float],
+            xy4: Tuple[float, float],
+        ) -> Tuple[float, float]:
+            """Internal function"""
+            x1, y1 = xy1
+            x2, y2 = xy2
+            x3, y3 = xy3
+            x4, y4 = xy4
+
+            # Check if lines are vertical
+            if x1 == x2 and x3 == x4:
+                raise ValueError(f'line "{xy1}" -> "{xy2}" and "{xy3}" => "{xy4}" are parallel.')
+
+            # Handle vertical lines
+            if x1 == x2:
+                m2 = (y4 - y3) / (x4 - x3)
+                b2 = y3 - m2 * x3
+                x = x1
+                y = m2 * x + b2
+                return (x, y)
+
+            if x3 == x4:
+                m1 = (y2 - y1) / (x2 - x1)
+                b1 = y1 - m1 * x1
+                x = x3
+                y = m1 * x + b1
+                return (x, y)
+
+            m1 = (y2 - y1) / (x2 - x1)
+            b1 = y1 - m1 * x1
+            m2 = (y4 - y3) / (x4 - x3)
+            b2 = y3 - m2 * x3
+
+            if m1 == m2:
+                raise ValueError(f'line "{xy1}" -> "{xy2}" and "{xy3}" => "{xy4}" are parallel.')
+
+            x = (b2 - b1) / (m1 - m2)
+            y = m1 * x + b1
+
+            return (x, y)
+
+        xys = self._original_points
+        parallel_lines: List[Tuple[Tuple[float, float], Tuple[float, float]]] = []
+        for i in range(len(xys) - 1):
+            xy1, xy2 = get_parallel_line_xys(xys[i], xys[i + 1])
+            parallel_lines.append((xy1, xy2))
+
+        points: List[Tuple[float, float]] = []
+        for i in range(len(parallel_lines) - 1):
+            xy1, xy2 = parallel_lines[i]
+            xy3, xy4 = parallel_lines[i + 1]
+            xy = find_lines_intersection(xy1, xy2, xy3, xy4)
+            points.append(xy)
+
+        first_point = parallel_lines[0][0]
+        last_point = parallel_lines[len(parallel_lines) - 1][1]
+
+        points.insert(0, first_point)
+        points.append(last_point)
+
+        return points

{drawlib-0.2.3.dev4.dist-info → drawlib-0.2.3.dev5.dist-info}/METADATA

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

{drawlib-0.2.3.dev4.dist-info → drawlib-0.2.3.dev5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-drawlib/__init__.py,sha256=jXsDFzb9eWZYUjfkP43y1J-umimTxB5iXTEx201e0Ow,2678
+drawlib/__init__.py,sha256=yuaqeWXsVZ_zJ5lJedaZVNNb5A7sKIjkkQeZm_vLhak,2678
 drawlib/__main__.py,sha256=UDsH6DiEIWBEwlClbYI10aWYBw_gOmaNCB0TSImWZ-c,604
 drawlib/apis.py,sha256=Q1Uz4U6IK63ZtjaV_UTWPZq7xDpxTz99qHfpGsL8SnI,531
 drawlib/assets/__init__.py,sha256=bYT4EWquLr0oyEziAuixlD9M4EPnMY6PyhknOxJjxqY,437
@@ -26,7 +26,7 @@ drawlib/v0_2/private/core_canvas/base.py,sha256=FipdPyX9M-JWL9FQpAT1r8hoo9o1wyCS
 drawlib/v0_2/private/core_canvas/canvas.py,sha256=haDITKf-Wz2zA2zDC_NmrdH-QsSagukQ7LT7xEB2crY,9594
 drawlib/v0_2/private/core_canvas/image.py,sha256=1KoLmCTbr4VoTZi0rI_e1_HIYOmpGklygeq5ACgwVus,6958
 drawlib/v0_2/private/core_canvas/line.py,sha256=qp5IcQUTfLyH1hi4JOKa-h_8TVWRWCEthfSSEEErMgQ,17862
-drawlib/v0_2/private/core_canvas/original_arrow.py,sha256=vTnL7krRxEtplF4CGF5pYnwI0T_YtuNaQ_5bI3zmvCU,21231
+drawlib/v0_2/private/core_canvas/original_arrow.py,sha256=rJJhoBf1Z2nWt1Ny6WTRYhoBLQHU5xzuXPVxLjyyuy8,24632
 drawlib/v0_2/private/core_canvas/original_polygon.py,sha256=uvBbfr72J3QEw9efxHdd5YXl1IKZhEA82qzEXdEK7Kc,14546
 drawlib/v0_2/private/core_canvas/patches.py,sha256=uDRZMYqMnkh37lXVLGwTUmsb7XuNjNMNZBjP3Cgi5Eg,14211
 drawlib/v0_2/private/core_canvas/text.py,sha256=bqdiqptdXeV8HVjp0Fx2pl2eA0q29XHYtWcK4WSyCw8,3520
@@ -75,8 +75,8 @@ drawlib/v0_2/private/validators/smartarts.py,sha256=fOAlC0syvZVk-GCLQgaPhhfWr3gc
 drawlib/v0_2/private/validators/style.py,sha256=R7aLlaVfzdo33L7GLQ3WVfzzD_EsgV2-Y7ERcbFq7QI,3308
 drawlib/v0_2/private/validators/text.py,sha256=ELT87nekr9hMx7m-wpVTku0QPxL3qCIX4cxI5RbFoIc,1126
 drawlib/v0_2/private/validators/types.py,sha256=KtiIgivuaFyFaxIGeqr5N8xpzqIk9soiacSzcFDdjSA,4155
-drawlib-0.2.3.dev4.dist-info/LICENSE.txt,sha256=h_ZiliCgEDqIYpxmUPmqGGP87VBcUHil1S4wUI48ZKg,11344
-drawlib-0.2.3.dev4.dist-info/METADATA,sha256=keuYahbx9v68956lsqVpKMBmY3l0G6rHq28VVvphFkc,4007
-drawlib-0.2.3.dev4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-drawlib-0.2.3.dev4.dist-info/entry_points.txt,sha256=Lp6mbrRMP-g6c7P6Iz21CEjceLITCWrma3b0ncP30oQ,49
-drawlib-0.2.3.dev4.dist-info/RECORD,,
+drawlib-0.2.3.dev5.dist-info/LICENSE.txt,sha256=h_ZiliCgEDqIYpxmUPmqGGP87VBcUHil1S4wUI48ZKg,11344
+drawlib-0.2.3.dev5.dist-info/METADATA,sha256=ICI9-Seg7xPDVr4pIG0xdrSjd9iQea3YEfNOjDe-9do,4007
+drawlib-0.2.3.dev5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+drawlib-0.2.3.dev5.dist-info/entry_points.txt,sha256=Lp6mbrRMP-g6c7P6Iz21CEjceLITCWrma3b0ncP30oQ,49
+drawlib-0.2.3.dev5.dist-info/RECORD,,