yxxandClaude Code
f1ac414866
- 抽取 render.py 共享几何：新增 PlotGeometry + compute_geometry，render_svg 改为薄序列化层，SVG 输出与重构前逐字节一致（8 组用例回归验证） - 新增 app/plot/render_reportlab.py：消费共享几何产出 reportlab 矢量 Drawing （网格/坐标轴 Line、曲线 PolyLine、刻度/标签 String、ylabel Group 旋转）， 复用 STSong-Light 渲染中文，按页面内容宽 renderScale 缩放 - pdf.py _block_function_plot 改为内嵌矢量图（解析/渲染失败或超预算回退占位， 单图失败不阻断整篇）；mermaid 仍占位 - 抽取 FunctionPlotBudget + format_plot_diagnostic 到 _common.py，html/pdf 共用 - 文档同步：PDF 已内嵌函数图像，DOCX 仍占位（栅格化范围外） Co-Authored-By: Claude Code <noreply@anthropic.com>
117 lines
4.3 KiB
Python
117 lines
4.3 KiB
Python
"""Function Plot → reportlab 矢量 Drawing（供 PDF 内嵌）。
|
||
|
||
消费 ``render.compute_geometry`` 的共享几何，产出 ``reportlab.graphics.shapes.Drawing``：
|
||
网格/坐标轴用 ``Line``、曲线用 ``PolyLine``、刻度数字与轴标签用 ``String``。
|
||
reportlab 原点在左下（y-up），与 SVG 的 y-down 相反，故对几何里的像素 y 统一翻转；
|
||
轴标签（ylabel）用 ``Group.rotate`` 旋转为竖向文本。中文字体复用内置 STSong-Light，
|
||
guarded 注册避免与 pdf.py 重复注册。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from reportlab.graphics.shapes import Drawing, Group, Line, PolyLine, String
|
||
from reportlab.lib.colors import HexColor
|
||
from reportlab.pdfbase import pdfmetrics
|
||
from reportlab.pdfbase.cidfonts import UnicodeCIDFont
|
||
|
||
from app.plot.model import FunctionPlot
|
||
from app.plot.render import PlotGeometry, _fmt_num, _sx, _sy, compute_geometry
|
||
|
||
_FONT = "STSong-Light"
|
||
if _FONT not in pdfmetrics.getRegisteredFontNames():
|
||
pdfmetrics.registerFont(UnicodeCIDFont(_FONT))
|
||
|
||
_GRID_COLOR = HexColor("#eaeef2")
|
||
_AXIS_COLOR = HexColor("#57606a")
|
||
_LABEL_COLOR = HexColor("#1f2328")
|
||
_TICK_FONT_SIZE = 10
|
||
_LABEL_FONT_SIZE = 12
|
||
|
||
|
||
def _build_drawing(geo: PlotGeometry) -> Drawing:
|
||
"""由共享几何构建矢量 Drawing（坐标翻转后仍沿用 SVG 的像素布局）。"""
|
||
drawing = Drawing(geo.width, geo.height)
|
||
|
||
# SVG y-down → reportlab y-up：翻转像素 y
|
||
def sx(x: float) -> float:
|
||
return _sx(x, geo.xmin, geo.xmax)
|
||
|
||
def sy(y: float) -> float:
|
||
return geo.height - _sy(y, geo.ymin, geo.ymax)
|
||
|
||
# 网格
|
||
if geo.grid:
|
||
for x in geo.xticks:
|
||
drawing.add(
|
||
Line(sx(x), sy(geo.ymin), sx(x), sy(geo.ymax), strokeColor=_GRID_COLOR, strokeWidth=0.5)
|
||
)
|
||
for y in geo.yticks:
|
||
drawing.add(
|
||
Line(sx(geo.xmin), sy(y), sx(geo.xmax), sy(y), strokeColor=_GRID_COLOR, strokeWidth=0.5)
|
||
)
|
||
|
||
# 坐标轴（过原点画在原点，否则贴边，与 SVG 一致）
|
||
drawing.add(
|
||
Line(sx(geo.xmin), sy(geo.x_axis_y), sx(geo.xmax), sy(geo.x_axis_y), strokeColor=_AXIS_COLOR, strokeWidth=0.7)
|
||
)
|
||
drawing.add(
|
||
Line(sx(geo.y_axis_x), sy(geo.ymin), sx(geo.y_axis_x), sy(geo.ymax), strokeColor=_AXIS_COLOR, strokeWidth=0.7)
|
||
)
|
||
|
||
# 刻度数字（x 轴下方、y 轴左侧）
|
||
for x in geo.xticks:
|
||
drawing.add(
|
||
String(
|
||
sx(x), sy(geo.x_axis_y) - 14, _fmt_num(x),
|
||
fontName=_FONT, fontSize=_TICK_FONT_SIZE, fillColor=_AXIS_COLOR, textAnchor="middle",
|
||
)
|
||
)
|
||
for y in geo.yticks:
|
||
drawing.add(
|
||
String(
|
||
sx(geo.y_axis_x) - 6, sy(y) - 3, _fmt_num(y),
|
||
fontName=_FONT, fontSize=_TICK_FONT_SIZE, fillColor=_AXIS_COLOR, textAnchor="end",
|
||
)
|
||
)
|
||
|
||
# 曲线（非有限点处已由几何断成多段）
|
||
for segments, color in zip(geo.polylines, geo.colors):
|
||
for seg in segments:
|
||
flipped = [(px, geo.height - py) for px, py in seg]
|
||
drawing.add(PolyLine(flipped, strokeColor=HexColor(color), strokeWidth=1.4))
|
||
|
||
# 轴标签
|
||
if geo.xlabel:
|
||
drawing.add(
|
||
String(
|
||
geo.width / 2, 10, geo.xlabel,
|
||
fontName=_FONT, fontSize=_LABEL_FONT_SIZE, fillColor=_LABEL_COLOR, textAnchor="middle",
|
||
)
|
||
)
|
||
if geo.ylabel:
|
||
# 竖向标签：rotate(90) 在 y-up 坐标下等价于 SVG 的 rotate(-90)
|
||
label = Group()
|
||
label.add(
|
||
String(
|
||
16, geo.height / 2, geo.ylabel,
|
||
fontName=_FONT, fontSize=_LABEL_FONT_SIZE, fillColor=_LABEL_COLOR, textAnchor="middle",
|
||
)
|
||
)
|
||
label.rotate(90, 16, geo.height / 2)
|
||
drawing.add(label)
|
||
|
||
return drawing
|
||
|
||
|
||
def render_drawing(plot: FunctionPlot, width: float | None = None) -> Drawing:
|
||
"""把已解析的 FunctionPlot 渲染为 reportlab Drawing（可直接追加到 platypus story）。
|
||
|
||
``width`` 为目标输出宽度（点），用于把 640px 的几何缩放到页面内容宽；省略则按
|
||
原始尺寸输出。缩放只影响 PDF 渲染，不改动共享几何。
|
||
"""
|
||
geo = compute_geometry(plot)
|
||
drawing = _build_drawing(geo)
|
||
if width is not None and width > 0:
|
||
drawing.renderScale = min(1.0, width / geo.width)
|
||
return drawing
|