pptx-shapes/pptx_shapes/shapes/arrow.py at master · dronperminov/pptx-shapes · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
from dataclasses import dataclass, field
from typing import Optional

from lxml import etree

from pptx_shapes.entities.bbox import BBox
from pptx_shapes.entities.namespace_helper import NamespaceHelper
from pptx_shapes.enums import ArrowType
from pptx_shapes.shapes.shape import Shape
from pptx_shapes.style.arrowhead import ArrowHead
from pptx_shapes.style.stroke_style import StrokeStyle


@dataclass
class Arrow(Shape):
    x1: float
    y1: float
    x2: float
    y2: float
    start_head: Optional[ArrowHead] = None
    end_head: Optional[ArrowHead] = field(default_factory=lambda: ArrowHead(head=ArrowType.TRIANGLE))
    stroke: StrokeStyle = field(default_factory=lambda: StrokeStyle(color="black"))

    def to_xml(self, shape_id: int, ns_helper: NamespaceHelper) -> etree.Element:
        node = ns_helper.element("p:cxnSp")

        nvcxnsppr = ns_helper.element("p:nvCxnSpPr", parent=node)
        ns_helper.element("p:cNvPr", {"id": str(shape_id), "name": f"Arrow {shape_id}"}, parent=nvcxnsppr)
        ns_helper.element("p:cNvCxnSpPr", parent=nvcxnsppr)
        ns_helper.element("p:nvPr", parent=nvcxnsppr)

        flips = {"flipH": "0" if self.x1 <= self.x2 else "1", "flipV": "0" if self.y1 <= self.y2 else "1"}
        bbox = self.bbox()

        sppr = ns_helper.element("p:spPr", parent=node)
        sppr.append(self.make_xfrm(ns_helper, flips, x=bbox.x, y=bbox.y, width=bbox.width, height=bbox.height))
        ns_helper.element("a:avLst", parent=ns_helper.element("a:prstGeom", {"prst": "straightConnector1"}, parent=sppr))

        ln = self.stroke.to_xml(ns_helper)
        if self.start_head is not None:
            ns_helper.element("a:headEnd", self.start_head.to_pptx(), ln)

        if self.end_head:
            ns_helper.element("a:tailEnd", self.end_head.to_pptx(), ln)

        sppr.append(ln)

        return node

    def bbox(self) -> BBox:
        x = min(self.x1, self.x2)
        y = min(self.y1, self.y2)
        width = abs(self.x2 - self.x1)
        height = abs(self.y2 - self.y1)

        return BBox(x=x, y=y, width=width, height=height)