kaxe 1.4.4.dev3__tar.gz → 1.4.4.dev4__tar.gz

{kaxe-1.4.4.dev3/src/kaxe.egg-info → kaxe-1.4.4.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kaxe
-Version: 1.4.4.dev3
+Version: 1.4.4.dev4
 Summary: A small graphing tool for functions, points, equations and more
 Author-email: Valter Yde Daugberg <valteryde@hotmail.com>
 Project-URL: Homepage, https://github.com/valteryde/kaxe
@@ -25,6 +25,8 @@ Requires-Dist: pysdl2-dll
 Requires-Dist: numpy-stl
 Requires-Dist: psutil
 Requires-Dist: opencv-python-headless
+Provides-Extra: pdf
+Requires-Dist: reportlab>=4.0; extra == "pdf"
 Dynamic: license-file
 
 

{kaxe-1.4.4.dev3 → kaxe-1.4.4.dev4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kaxe"
-version = "1.4.4.dev3"
+version = "1.4.4.dev4"
 authors = [
   { name="Valter Yde Daugberg", email="valteryde@hotmail.com" },
 ]
@@ -29,6 +29,9 @@ dependencies = [
   "opencv-python-headless",
 ]
 
+[project.optional-dependencies]
+pdf = ["reportlab>=4.0"]
+
 [project.urls]
 Homepage = "https://github.com/valteryde/kaxe"
 Issues = "https://github.com/valteryde/kaxe/issues"

{kaxe-1.4.4.dev3 → kaxe-1.4.4.dev4}/src/kaxe/core/svg.py

@@ -8,6 +8,7 @@ import io
 import math
 import os
 import xml.etree.ElementTree as ET
+from io import BytesIO
 from typing import Any, Optional, Union
 
 from PIL import Image
@@ -369,6 +370,12 @@ class SvgDocument:
         body = _ascii_safe_svg_markup(body)
         return '<?xml version="1.0" encoding="UTF-8" standalone="no"?>\n' + body
 
+    def to_pdf(self, fname: Optional[Union[str, BytesIO]] = None) -> bytes:
+        """Render this document to PDF bytes via ReportLab."""
+        from .svg_pdf import write_pdf
+
+        return write_pdf(self, fname)
+
 
 def parse_svg_root(xml: str) -> ET.Element:
     """Parse SVG XML and return the root element."""
@@ -433,7 +440,7 @@ def infer_format(fname: Union[str, os.PathLike, Any], format: Optional[str] = No
     """Infer save format from explicit format, file extension, or default to png."""
     if format is not None:
         fmt = format.lower().lstrip(".")
-        if fmt in ("svg", "png"):
+        if fmt in ("svg", "png", "pdf"):
             return fmt
         raise ValueError(f"Unsupported format: {format}")
 
@@ -443,5 +450,7 @@ def infer_format(fname: Union[str, os.PathLike, Any], format: Optional[str] = No
             return "svg"
         if lower.endswith(".png"):
             return "png"
+        if lower.endswith(".pdf"):
+            return "pdf"
 
     return "png"

kaxe-1.4.4.dev4/src/kaxe/core/svg_pdf.py

@@ -0,0 +1,541 @@
+"""Convert Kaxe SvgDocument trees to vector PDF via ReportLab."""
+
+from __future__ import annotations
+
+import base64
+import io
+import logging
+import math
+import os
+import re
+import xml.etree.ElementTree as ET
+from dataclasses import dataclass
+from typing import TYPE_CHECKING, Any, Optional, Union
+
+from PIL import Image
+
+from .svg import SVG_NS, XLINK_NS, is_file_path
+
+if TYPE_CHECKING:
+    from .svg import SvgDocument
+
+_PDF_INSTALL_HINT = "PDF export requires reportlab. Install with: pip install kaxe[pdf]"
+_FONDI_FONTS_REGISTERED = False
+
+_TRANSFORM_RE = re.compile(
+    r"(translate|rotate|matrix)\s*\(([^)]*)\)"
+)
+_PATH_CMD_RE = re.compile(r"([MLAZ])\s*([^MLAZ]*)", re.IGNORECASE)
+
+
+def _require_reportlab():
+    try:
+        import reportlab  # noqa: F401
+    except ImportError as exc:
+        raise ImportError(_PDF_INSTALL_HINT) from exc
+
+
+def _local_tag(tag: str) -> str:
+    return tag.split("}")[-1] if "}" in tag else tag
+
+
+def _attr(el: ET.Element, name: str) -> Optional[str]:
+    if name in el.attrib:
+        return el.attrib[name]
+    for key, value in el.attrib.items():
+        if key.split("}")[-1] == name:
+            return value
+    return None
+
+
+def _parse_float(value: Optional[str], default: float = 0.0) -> float:
+    if value is None or value == "":
+        return default
+    return float(value)
+
+
+def _parse_rgb(value: Optional[str]) -> tuple[float, float, float]:
+    if not value or value == "none":
+        return 0.0, 0.0, 0.0
+    match = re.match(r"rgb\(\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*\)", value)
+    if match:
+        return float(match.group(1)), float(match.group(2)), float(match.group(3))
+    return 0.0, 0.0, 0.0
+
+
+def _rl_color(value: Optional[str], opacity: Optional[str] = None):
+    from reportlab.lib.colors import Color
+
+    r, g, b = _parse_rgb(value)
+    alpha = float(opacity) if opacity is not None else 1.0
+    return Color(r / 255.0, g / 255.0, b / 255.0, alpha=alpha)
+
+
+def _register_fondi_fonts() -> None:
+    global _FONDI_FONTS_REGISTERED
+    if _FONDI_FONTS_REGISTERED:
+        return
+
+    _require_reportlab()
+    import fondi
+    from pathlib import Path
+    from reportlab.pdfbase import pdfmetrics
+    from reportlab.pdfbase.ttfonts import TTFont
+
+    resources = Path(fondi.__file__).parent / "resources"
+    regular_ttf = resources / "cmu.serif-roman.ttf"
+    if not regular_ttf.is_file():
+        logging.warning("Fondi regular TTF not found; PDF math labels will use Helvetica")
+        _FONDI_FONTS_REGISTERED = True
+        return
+
+    pdfmetrics.registerFont(TTFont("FondiNewCM", str(regular_ttf)))
+    pdfmetrics.registerFontFamily(
+        "FondiNewCM",
+        normal="FondiNewCM",
+        italic="FondiNewCM",
+    )
+    _FONDI_FONTS_REGISTERED = True
+
+
+def _resolve_font_name(family: Optional[str], style: Optional[str]) -> str:
+    if family == "FondiNewCM":
+        return "FondiNewCM"
+    if family:
+        logging.warning("Unknown PDF font family %r; using Helvetica", family)
+    return "Helvetica"
+
+
+@dataclass
+class _Matrix:
+    a: float = 1.0
+    b: float = 0.0
+    c: float = 0.0
+    d: float = 1.0
+    e: float = 0.0
+    f: float = 0.0
+
+    def multiply(self, other: "_Matrix") -> "_Matrix":
+        return _Matrix(
+            a=self.a * other.a + self.c * other.b,
+            b=self.b * other.a + self.d * other.b,
+            c=self.a * other.c + self.c * other.d,
+            d=self.b * other.c + self.d * other.d,
+            e=self.a * other.e + self.c * other.f + self.e,
+            f=self.b * other.e + self.d * other.f + self.f,
+        )
+
+    def map_point(self, x: float, y: float) -> tuple[float, float]:
+        return (
+            self.a * x + self.c * y + self.e,
+            self.b * x + self.d * y + self.f,
+        )
+
+
+def _parse_transform(transform: Optional[str]) -> _Matrix:
+    matrix = _Matrix()
+    if not transform:
+        return matrix
+
+    for match in _TRANSFORM_RE.finditer(transform.strip()):
+        name = match.group(1).lower()
+        parts = [p.strip() for p in re.split(r"[\s,]+", match.group(2).strip()) if p.strip()]
+
+        if name == "translate":
+            tx = float(parts[0]) if parts else 0.0
+            ty = float(parts[1]) if len(parts) > 1 else 0.0
+            matrix = matrix.multiply(_Matrix(e=tx, f=ty))
+        elif name == "rotate":
+            angle = math.radians(float(parts[0]))
+            cx = float(parts[1]) if len(parts) > 1 else 0.0
+            cy = float(parts[2]) if len(parts) > 2 else 0.0
+            cos_a = math.cos(angle)
+            sin_a = math.sin(angle)
+            rot = _Matrix(a=cos_a, b=sin_a, c=-sin_a, d=cos_a)
+            to_origin = _Matrix(e=-cx, f=-cy)
+            back = _Matrix(e=cx, f=cy)
+            matrix = matrix.multiply(back.multiply(rot.multiply(to_origin)))
+        elif name == "matrix" and len(parts) >= 6:
+            matrix = matrix.multiply(
+                _Matrix(
+                    a=float(parts[0]),
+                    b=float(parts[1]),
+                    c=float(parts[2]),
+                    d=float(parts[3]),
+                    e=float(parts[4]),
+                    f=float(parts[5]),
+                )
+            )
+    return matrix
+
+
+class _PdfBuilder:
+    def __init__(self, width: int, height: int):
+        _require_reportlab()
+        from reportlab.graphics.shapes import Drawing
+
+        self.height = height
+        self.drawing = Drawing(width, height)
+        self._shapes: list[Any] = []
+
+    def _to_rl(self, matrix: _Matrix, x: float, y: float) -> tuple[float, float]:
+        sx, sy = matrix.map_point(x, y)
+        return sx, self.height - sy
+
+    def _add(self, shape) -> None:
+        self._shapes.append(shape)
+
+    def _render_element(self, el: ET.Element, matrix: _Matrix) -> None:
+        from reportlab.graphics.shapes import (
+            Circle,
+            Group,
+            Image as RlImage,
+            Line,
+            Path,
+            Polygon,
+            PolyLine,
+            Rect,
+            String,
+        )
+        from reportlab.lib.utils import ImageReader
+
+        tag = _local_tag(el.tag)
+        opacity = _attr(el, "opacity")
+
+        if tag == "g":
+            child_matrix = matrix.multiply(_parse_transform(_attr(el, "transform")))
+            group = Group()
+            child_shapes: list[Any] = []
+            saved = self._shapes
+            self._shapes = child_shapes
+            for child in el:
+                self._render_element(child, child_matrix)
+            self._shapes = saved
+            for shape in child_shapes:
+                group.add(shape)
+            if child_shapes:
+                self._add(group)
+            return
+
+        if tag == "rect":
+            x = _parse_float(_attr(el, "x"))
+            y = _parse_float(_attr(el, "y"))
+            w = _parse_float(_attr(el, "width"))
+            h = _parse_float(_attr(el, "height"))
+            x0, y0 = self._to_rl(matrix, x, y)
+            x1, y1 = self._to_rl(matrix, x + w, y + h)
+            rect = Rect(
+                min(x0, x1),
+                min(y0, y1),
+                abs(x1 - x0),
+                abs(y1 - y0),
+                fillColor=_rl_color(_attr(el, "fill"), opacity),
+                strokeColor=None,
+            )
+            self._add(rect)
+            return
+
+        if tag == "line":
+            x1, y1 = self._to_rl(matrix, _parse_float(_attr(el, "x1")), _parse_float(_attr(el, "y1")))
+            x2, y2 = self._to_rl(matrix, _parse_float(_attr(el, "x2")), _parse_float(_attr(el, "y2")))
+            stroke = _rl_color(_attr(el, "stroke"), opacity)
+            width = _parse_float(_attr(el, "stroke-width"), 1.0)
+            self._add(Line(x1, y1, x2, y2, strokeColor=stroke, strokeWidth=width))
+            return
+
+        if tag == "circle":
+            cx, cy = self._to_rl(matrix, _parse_float(_attr(el, "cx")), _parse_float(_attr(el, "cy")))
+            radius = _parse_float(_attr(el, "r"))
+            fill = _attr(el, "fill")
+            if fill and fill != "none":
+                self._add(
+                    Circle(
+                        cx,
+                        cy,
+                        radius,
+                        fillColor=_rl_color(fill, opacity),
+                        strokeColor=None,
+                    )
+                )
+            else:
+                self._add(
+                    Circle(
+                        cx,
+                        cy,
+                        radius,
+                        fillColor=None,
+                        strokeColor=_rl_color(_attr(el, "stroke"), opacity),
+                        strokeWidth=_parse_float(_attr(el, "stroke-width"), 1.0),
+                    )
+                )
+            return
+
+        if tag == "polygon":
+            points = self._parse_points(_attr(el, "points"), matrix)
+            self._add(
+                Polygon(
+                    points,
+                    fillColor=_rl_color(_attr(el, "fill"), opacity),
+                    strokeColor=None,
+                )
+            )
+            return
+
+        if tag == "polyline":
+            points = self._parse_points(_attr(el, "points"), matrix)
+            self._add(
+                PolyLine(
+                    points,
+                    fillColor=None,
+                    strokeColor=_rl_color(_attr(el, "stroke"), opacity),
+                    strokeWidth=_parse_float(_attr(el, "stroke-width"), 1.0),
+                )
+            )
+            return
+
+        if tag == "path":
+            path = self._parse_path(_attr(el, "d"), matrix)
+            if path is not None:
+                path.fillColor = _rl_color(_attr(el, "fill"), opacity)
+                path.strokeColor = None
+                self._add(path)
+            return
+
+        if tag == "image":
+            href = _attr(el, "href") or _attr(el, f"{{{XLINK_NS}}}href")
+            if not href or not href.startswith("data:image"):
+                return
+            header, encoded = href.split(",", 1)
+            raw = base64.b64decode(encoded)
+            img = Image.open(io.BytesIO(raw))
+            x = _parse_float(_attr(el, "x"))
+            y = _parse_float(_attr(el, "y"))
+            w = _parse_float(_attr(el, "width"), img.width)
+            h = _parse_float(_attr(el, "height"), img.height)
+
+            image_matrix = matrix.multiply(_parse_transform(_attr(el, "transform")))
+            x0, y0 = self._to_rl(image_matrix, x, y)
+            x1, y1 = self._to_rl(image_matrix, x + w, y + h)
+            rl_img = RlImage(
+                min(x0, x1),
+                min(y0, y1),
+                abs(x1 - x0),
+                abs(y1 - y0),
+                path=ImageReader(io.BytesIO(raw)),
+            )
+            self._add(rl_img)
+            return
+
+        if tag == "text":
+            x = _parse_float(_attr(el, "x"))
+            y = _parse_float(_attr(el, "y"))
+            rl_x, rl_y = self._to_rl(matrix, x, y)
+            text = (el.text or "").strip()
+            if not text:
+                text = "".join(el.itertext())
+            font_name = _resolve_font_name(_attr(el, "font-family"), _attr(el, "font-style"))
+            font_size = _parse_float(_attr(el, "font-size"), 12.0)
+            anchor = _attr(el, "text-anchor") or "start"
+            text_anchor = {"middle": "middle", "end": "end"}.get(anchor, "start")
+            self._add(
+                String(
+                    rl_x,
+                    rl_y,
+                    text,
+                    fontName=font_name,
+                    fontSize=font_size,
+                    fillColor=_rl_color(_attr(el, "fill"), opacity),
+                    textAnchor=text_anchor,
+                )
+            )
+            return
+
+    def _parse_points(self, value: Optional[str], matrix: _Matrix) -> list[float]:
+        if not value:
+            return []
+        nums = [float(n) for n in re.split(r"[\s,]+", value.strip()) if n]
+        points: list[float] = []
+        for i in range(0, len(nums) - 1, 2):
+            x, y = self._to_rl(matrix, nums[i], nums[i + 1])
+            points.extend([x, y])
+        return points
+
+    def _parse_path(self, d: Optional[str], matrix: _Matrix):
+        from reportlab.graphics.shapes import Path
+
+        if not d:
+            return None
+
+        path = Path()
+        current_x = 0.0
+        current_y = 0.0
+        start_x = 0.0
+        start_y = 0.0
+        subpath_start_x = 0.0
+        subpath_start_y = 0.0
+
+        for match in _PATH_CMD_RE.finditer(d):
+            cmd = match.group(1).upper()
+            nums = [float(n) for n in re.split(r"[\s,]+", match.group(2).strip()) if n]
+
+            if cmd == "M" and len(nums) >= 2:
+                current_x, current_y = nums[0], nums[1]
+                start_x, start_y = current_x, current_y
+                subpath_start_x, subpath_start_y = current_x, current_y
+                x, y = self._to_rl(matrix, current_x, current_y)
+                path.moveTo(x, y)
+            elif cmd == "L" and len(nums) >= 2:
+                current_x, current_y = nums[0], nums[1]
+                x, y = self._to_rl(matrix, current_x, current_y)
+                path.lineTo(x, y)
+            elif cmd == "A" and len(nums) >= 7:
+                rx, ry, _, large_arc, sweep, end_x, end_y = nums[:7]
+                start_x, start_y = current_x, current_y
+                self._append_circular_arc(
+                    path,
+                    matrix,
+                    start_x,
+                    start_y,
+                    end_x,
+                    end_y,
+                    rx,
+                    bool(int(large_arc)),
+                    bool(int(sweep)),
+                )
+                current_x, current_y = end_x, end_y
+            elif cmd == "Z":
+                current_x, current_y = subpath_start_x, subpath_start_y
+                x, y = self._to_rl(matrix, current_x, current_y)
+                path.lineTo(x, y)
+                path.closePath()
+
+        return path
+
+    def _append_circular_arc(
+        self,
+        path,
+        matrix: _Matrix,
+        x1: float,
+        y1: float,
+        x2: float,
+        y2: float,
+        radius: float,
+        large_arc: bool,
+        sweep: bool,
+    ) -> None:
+        dx = x2 - x1
+        dy = y2 - y1
+        dist = math.hypot(dx, dy)
+        if dist == 0 or radius == 0:
+            x, y = self._to_rl(matrix, x2, y2)
+            path.lineTo(x, y)
+            return
+
+        dist = min(dist, 2 * radius)
+        mid_x = (x1 + x2) / 2
+        mid_y = (y1 + y2) / 2
+        chord = dist / 2
+        sagitta_sq = max(radius * radius - chord * chord, 0.0)
+        sagitta = math.sqrt(sagitta_sq)
+
+        perp_x = -(y2 - y1) / dist
+        perp_y = (x2 - x1) / dist
+
+        cx1 = mid_x + perp_x * sagitta
+        cy1 = mid_y + perp_y * sagitta
+        cx2 = mid_x - perp_x * sagitta
+        cy2 = mid_y - perp_y * sagitta
+
+        start1 = math.atan2(y1 - cy1, x1 - cx1)
+        end1 = math.atan2(y2 - cy1, x2 - cx1)
+        start2 = math.atan2(y1 - cy2, x1 - cx2)
+        end2 = math.atan2(y2 - cy2, x2 - cx2)
+
+        def arc_span(start: float, end: float) -> float:
+            span = end - start
+            if sweep and span < 0:
+                span += 2 * math.pi
+            if not sweep and span > 0:
+                span -= 2 * math.pi
+            return abs(span)
+
+        use_first = arc_span(start1, end1) > math.pi
+        if large_arc:
+            cx, cy, start, end = (
+                (cx1, cy1, start1, end1) if use_first else (cx2, cy2, start2, end2)
+            )
+        else:
+            cx, cy, start, end = (
+                (cx1, cy1, start1, end1) if not use_first else (cx2, cy2, start2, end2)
+            )
+
+        if sweep and end < start:
+            end += 2 * math.pi
+        if not sweep and end > start:
+            end -= 2 * math.pi
+
+        steps = max(8, int(abs(end - start) / (math.pi / 16)))
+        for i in range(1, steps + 1):
+            t = start + (end - start) * (i / steps)
+            px = cx + radius * math.cos(t)
+            py = cy + radius * math.sin(t)
+            x, y = self._to_rl(matrix, px, py)
+            path.lineTo(x, y)
+
+    def build(self, elements: list[ET.Element]):
+        identity = _Matrix()
+        for el in elements:
+            self._render_element(el, identity)
+        for shape in self._shapes:
+            self.drawing.add(shape)
+        return self.drawing
+
+
+def document_to_pdf(doc: "SvgDocument") -> bytes:
+    _register_fondi_fonts()
+    from reportlab.graphics import renderPDF
+
+    builder = _PdfBuilder(doc.width, doc.height)
+    drawing = builder.build(doc._elements)
+    buf = io.BytesIO()
+    renderPDF.drawToFile(drawing, buf)
+    return buf.getvalue()
+
+
+def write_pdf(doc: "SvgDocument", fname: Optional[Union[str, io.BytesIO]] = None) -> bytes:
+    pdf_bytes = document_to_pdf(doc)
+    if fname is not None:
+        if is_file_path(fname):
+            with open(os.fspath(fname), "wb") as f:
+                f.write(pdf_bytes)
+        else:
+            fname.write(pdf_bytes)
+    return pdf_bytes
+
+
+def image_to_pdf_page(img: Image.Image, fname: Optional[Union[str, io.BytesIO]] = None) -> bytes:
+    """Wrap a raster image in a single-page PDF."""
+    _require_reportlab()
+    from reportlab.lib.utils import ImageReader
+    from reportlab.pdfgen import canvas
+
+    width, height = img.size
+    buf = io.BytesIO()
+    if img.mode != "RGB":
+        img = img.convert("RGB")
+    png_buf = io.BytesIO()
+    img.save(png_buf, format="PNG")
+    png_buf.seek(0)
+
+    pdf = canvas.Canvas(buf, pagesize=(width, height))
+    pdf.drawImage(ImageReader(png_buf), 0, 0, width=width, height=height)
+    pdf.save()
+    pdf_bytes = buf.getvalue()
+
+    if fname is not None:
+        if is_file_path(fname):
+            with open(os.fspath(fname), "wb") as f:
+                f.write(pdf_bytes)
+        else:
+            fname.write(pdf_bytes)
+    return pdf_bytes

{kaxe-1.4.4.dev3 → kaxe-1.4.4.dev4}/src/kaxe/core/window.py

@@ -513,9 +513,34 @@ class Window(AttrObject):
         return xml
 
 
+    def __pdfPaint__(self, fname=None) -> bytes:
+        startTime = time.time()
+        if self.showProgressBar: pbar = tqdm.tqdm(total=len(self.shapes), desc='Decorating PDF')
+
+        winSize = self.width+self.padding[0]+self.padding[2], self.height+self.padding[1]+self.padding[3]
+        doc = SvgDocument(winSize)
+
+        if self.getAttr('backgroundColor')[3] != 0:
+            background = shapes.Rectangle(0, 0, winSize[0], winSize[1], color=self.getAttr('backgroundColor'))
+            background.draw(doc)
+
+        for shape in self.shapes:
+            shape.draw(doc)
+            if self.showProgressBar: pbar.update()
+
+        pdf_bytes = doc.to_pdf(fname)
+
+        if self.showProgressBar: pbar.close()
+        if self.printDebugInfo: logging.info('Painted PDF in {}s'.format(str(round(time.time() - startTime, 4))))
+
+        return pdf_bytes
+
+
     def __paint__(self, fname=None, format: str = "png"):
         if format == "svg":
             return self.__svgPaint__(fname)
+        if format == "pdf":
+            return self.__pdfPaint__(fname)
         return self.__pillowPaint__(fname)
 
 
@@ -532,12 +557,13 @@ class Window(AttrObject):
         fname : str | BytesIO
             The filename where the image will be saved or a BytesIO object to save the image in memory.
         format : str, optional
-            Output format: ``"png"`` or ``"svg"``. Inferred from ``fname`` extension when omitted.
+            Output format: ``"png"``, ``"svg"``, or ``"pdf"``. Inferred from ``fname`` extension when omitted.
         
         Examples
         --------
         >>> plt.save( path/where/image/saved.png )
         >>> plt.save( path/where/image/saved.svg )
+        >>> plt.save( path/where/image/saved.pdf )
 
         """
 
@@ -554,7 +580,7 @@ class Window(AttrObject):
         totStartTime = time.time()
 
         self.__bake__()
-        result = self.__paint__(fname if fmt == "svg" else None, format=fmt)
+        result = self.__paint__(fname if fmt in ("svg", "pdf") else None, format=fmt)
 
         if fmt == "png":
             if fname is not None: