BigUpdate

md2gost/renderable/diagram.py

@@ -0,0 +1,68 @@
+from dataclasses import dataclass
+from typing import Generator
+
+from docx.shared import Parented
+
+from .caption import CaptionInfo
+from .image import Image
+from .listing import Listing
+from .renderable import Renderable
+from .requires_numbering import RequiresNumbering
+from ..diagram_renderer import render_diagram
+from ..layout_tracker import LayoutState
+from ..rendered_info import RenderedInfo
+
+
+class DiagramFigure(Renderable, RequiresNumbering):
+    """UML/BPMN/C4 fence → PNG figure (+ optional source listing)."""
+
+    def __init__(
+        self,
+        parent: Parented,
+        lang: str,
+        source: str,
+        caption_info: CaptionInfo | None = None,
+        with_listing: bool = False,
+    ):
+        super().__init__("Рисунок")
+        self._parent = parent
+        self._lang = lang
+        self._source = source
+        self._caption_info = caption_info
+        self._with_listing = with_listing or (
+            caption_info.with_listing if caption_info else False
+        )
+        self._number = None
+        self._listing_number = None
+        if caption_info and caption_info.unique_name:
+            self.unique_name = caption_info.unique_name
+
+        self.listing: Listing | None = None
+        if self._with_listing:
+            listing_caption = CaptionInfo(
+                self.unique_name,  # same id → @Листинг:id
+                caption_info.text if caption_info else None,
+            )
+            self.listing = Listing(parent, lang, listing_caption)
+            self.listing.set_text(source)
+
+    def set_number(self, number: str):
+        self._number = number
+
+    def set_listing_number(self, number: str):
+        self._listing_number = number
+        if self.listing:
+            self.listing.set_number(number)
+
+    def render(self, previous_rendered: RenderedInfo, layout_state: LayoutState) \
+            -> Generator[RenderedInfo, None, None]:
+        png_path = render_diagram(self._lang, self._source)
+        image = Image(self._parent, png_path, self._caption_info)
+        if self._number is not None:
+            image.set_number(self._number)
+        yield from image.render(previous_rendered, layout_state)
+
+        if self.listing is not None:
+            if self._listing_number is not None:
+                self.listing.set_number(self._listing_number)
+            yield from self.listing.render(None, layout_state)