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>
335 lines
12 KiB
Python
335 lines
12 KiB
Python
"""Function Plot → 静态 SVG 渲染 + 共享几何计算。
|
||
|
||
只输出纯几何与 <text> 的 SVG（无 script/foreignObject/内联事件），可安全内嵌 HTML。
|
||
所有文本与颜色都经过转义/校验，不把用户输入直接拼进标记。
|
||
|
||
几何计算（范围解析、采样、刻度、非有限点分段）统一收敛到 ``compute_geometry``，
|
||
返回像素坐标的 ``PlotGeometry``；``render_svg`` 只做 SVG 序列化，reportlab 后端
|
||
（``render_reportlab.py``）消费同一份几何，保证 PDF 与 SVG 视觉一致。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import html
|
||
import math
|
||
import re
|
||
from dataclasses import dataclass
|
||
|
||
from app.plot.model import FunctionPlot, StaticRenderResult
|
||
from app.plot.parser import PlotParseError, evaluate, parse_expression
|
||
|
||
_WIDTH = 640
|
||
_HEIGHT = 480
|
||
_MARGIN = 52 # 四周留白，放轴刻度与标签
|
||
_SAMPLES = 400
|
||
_PALETTE = ["#0969da", "#d1242f", "#1a7f37", "#8250df", "#bf8700", "#e36209"]
|
||
_COLOR_RE = re.compile(r"^#[0-9a-fA-F]{3,8}$")
|
||
|
||
|
||
def _safe_color(color: str | None, fallback: str) -> str:
|
||
return color.strip() if color and _COLOR_RE.match(color.strip()) else fallback
|
||
|
||
|
||
def _valid_span(lo: float, hi: float) -> bool:
|
||
"""范围跨度有效：端点有限、跨度有限且大于零。
|
||
|
||
端点相减可能溢出为 ``inf``（如 ``-1e308`` 到 ``1e308``），需单独校验跨度，
|
||
否则后续坐标换算会生成含 ``nan`` 的 SVG。
|
||
"""
|
||
span = hi - lo
|
||
return math.isfinite(lo) and math.isfinite(hi) and math.isfinite(span) and span > 0
|
||
|
||
|
||
def _fmt_num(v: float) -> str:
|
||
if v == 0:
|
||
return "0"
|
||
if abs(v) >= 1e6 or abs(v) < 1e-6:
|
||
return f"{v:.2e}"
|
||
return f"{v:.6g}"
|
||
|
||
|
||
def _nice_step(span: float, target_ticks: int = 6) -> float:
|
||
raw = abs(span) / target_ticks
|
||
if not math.isfinite(raw) or raw <= 0:
|
||
return 1.0 # 兜底步长，避免 span 为 0/inf 时产生非法刻度
|
||
mag = 10 ** math.floor(math.log10(raw))
|
||
for m in (1, 2, 5, 10):
|
||
if raw <= m * mag:
|
||
return m * mag
|
||
return 10 * mag
|
||
|
||
|
||
def _ticks(lo: float, hi: float, step: float) -> list[float]:
|
||
# 防御：非法步长直接返回空，避免除零
|
||
if not math.isfinite(step) or step <= 0:
|
||
return []
|
||
first = math.ceil(lo / step) * step
|
||
values: list[float] = []
|
||
v = first
|
||
# 有上限的整数索引推进 + 步长推进校验，防止浮点精度导致 v+step==v 的死循环
|
||
for _ in range(1000):
|
||
if v > hi + step * 1e-9:
|
||
break
|
||
values.append(v)
|
||
nxt = v + step
|
||
if nxt <= v:
|
||
break # 步长小于当前数值的浮点精度，已无法推进
|
||
v = nxt
|
||
return values
|
||
|
||
|
||
def _compute_range(
|
||
fns: list[tuple[object, object]],
|
||
xmin: float,
|
||
xmax: float,
|
||
) -> tuple[float, float]:
|
||
"""采样确定 y 范围；取有限样本的 min/max 加 5% 余量。"""
|
||
ys: list[float] = []
|
||
for _expr, tree in fns:
|
||
for i in range(_SAMPLES + 1):
|
||
x = xmin + (xmax - xmin) * i / _SAMPLES
|
||
try:
|
||
y = evaluate(tree, x) # type: ignore[arg-type]
|
||
except (ValueError, ZeroDivisionError, OverflowError, TypeError):
|
||
continue
|
||
# 复数等非实数结果直接跳过，不参与范围统计
|
||
if isinstance(y, (int, float)) and math.isfinite(y):
|
||
ys.append(y)
|
||
|
||
if not ys:
|
||
return -10.0, 10.0
|
||
lo, hi = min(ys), max(ys)
|
||
if lo == hi:
|
||
lo -= 1.0
|
||
hi += 1.0
|
||
pad = (hi - lo) * 0.05
|
||
return lo - pad, hi + pad
|
||
|
||
|
||
def _sx(x: float, xmin: float, xmax: float) -> float:
|
||
"""数据 x → 像素 x（SVG y-down 约定，原点左上）。"""
|
||
return _MARGIN + (x - xmin) / (xmax - xmin) * (_WIDTH - 2 * _MARGIN)
|
||
|
||
|
||
def _sy(y: float, ymin: float, ymax: float) -> float:
|
||
"""数据 y → 像素 y（SVG y-down 约定，原点左上）。"""
|
||
return _HEIGHT - _MARGIN - (y - ymin) / (ymax - ymin) * (_HEIGHT - 2 * _MARGIN)
|
||
|
||
|
||
@dataclass
|
||
class PlotGeometry:
|
||
"""已解析的几何：范围、轴位置、刻度、曲线像素点段、标签与 warnings。
|
||
|
||
像素坐标统一为 SVG y-down 约定；reportlab 后端（y-up）自行翻转 y。
|
||
"""
|
||
|
||
width: int
|
||
height: int
|
||
xmin: float
|
||
xmax: float
|
||
ymin: float
|
||
ymax: float
|
||
x_axis_y: float # 数据空间里 x 轴所在 y（过原点则 0，否则贴边）
|
||
y_axis_x: float # 数据空间里 y 轴所在 x（过原点则 0，否则贴边）
|
||
xticks: list[float]
|
||
yticks: list[float]
|
||
polylines: list[list[list[tuple[float, float]]]] # 按表达式分组：段 → 像素点
|
||
colors: list[str] # 与 polylines 对齐
|
||
xlabel: str | None
|
||
ylabel: str | None
|
||
grid: bool
|
||
warnings: list[str]
|
||
|
||
|
||
def _sample_segments(
|
||
tree: object,
|
||
xmin: float,
|
||
xmax: float,
|
||
ymin: float,
|
||
ymax: float,
|
||
) -> list[list[tuple[float, float]]]:
|
||
"""采样并映射为像素点段；非有限点处断段，避免画穿渐近线。"""
|
||
segments: list[list[tuple[float, float]]] = []
|
||
points: list[tuple[float, float]] = []
|
||
for i in range(_SAMPLES + 1):
|
||
x = xmin + (xmax - xmin) * i / _SAMPLES
|
||
try:
|
||
y = evaluate(tree, x) # type: ignore[arg-type]
|
||
except (ValueError, ZeroDivisionError, OverflowError, TypeError):
|
||
y = math.nan
|
||
if not isinstance(y, (int, float)) or not math.isfinite(y):
|
||
if points:
|
||
segments.append(points)
|
||
points = []
|
||
continue
|
||
px = _sx(x, xmin, xmax)
|
||
py = _sy(y, ymin, ymax)
|
||
# 映射后的坐标必须有限：显式 range 下极端 y 值可能让像素坐标溢出为 inf
|
||
if not (math.isfinite(px) and math.isfinite(py)):
|
||
if points:
|
||
segments.append(points)
|
||
points = []
|
||
continue
|
||
points.append((px, py))
|
||
if points:
|
||
segments.append(points)
|
||
return segments
|
||
|
||
|
||
def compute_geometry(plot: FunctionPlot) -> PlotGeometry:
|
||
"""解析并计算几何，供 SVG 与 reportlab 后端复用。"""
|
||
warnings: list[str] = []
|
||
xmin, xmax = plot.domain
|
||
if not _valid_span(xmin, xmax):
|
||
warnings.append("domain 无效，回退到 [-10, 10]")
|
||
xmin, xmax = -10.0, 10.0
|
||
|
||
# 重新解析并编译表达式（parse_source 已校验，这里异常只在模型被绕过时触发）
|
||
fns: list[tuple[object, object]] = []
|
||
for expr in plot.expressions:
|
||
try:
|
||
tree = parse_expression(expr.expression)
|
||
except PlotParseError as exc:
|
||
warnings.append(f"表达式无法渲染，已跳过：{expr.expression}（{exc.diagnostic.message}）")
|
||
continue
|
||
fns.append((expr, tree))
|
||
|
||
# 纵轴范围：显式 range 有效则用之；无效（退化/非有限/跨度溢出）丢弃并自动采样重算
|
||
if plot.range is not None:
|
||
lo, hi = float(plot.range[0]), float(plot.range[1])
|
||
if _valid_span(lo, hi):
|
||
ymin, ymax = lo, hi
|
||
else:
|
||
warnings.append("range 无效，改用自动范围")
|
||
ymin, ymax = _compute_range(fns, xmin, xmax)
|
||
else:
|
||
ymin, ymax = _compute_range(fns, xmin, xmax)
|
||
|
||
# 最终防线：自动范围在极端样本下也可能溢出，坐标映射前必须保证跨度有限且大于零
|
||
if not _valid_span(ymin, ymax):
|
||
warnings.append("y 范围跨度无法表示，回退到 [-10, 10]")
|
||
ymin, ymax = -10.0, 10.0
|
||
|
||
x_axis_y = 0.0 if ymin <= 0 <= ymax else ymin
|
||
y_axis_x = 0.0 if xmin <= 0 <= xmax else xmin
|
||
xticks = _ticks(xmin, xmax, _nice_step(xmax - xmin))
|
||
yticks = _ticks(ymin, ymax, _nice_step(ymax - ymin))
|
||
|
||
polylines: list[list[list[tuple[float, float]]]] = []
|
||
colors: list[str] = []
|
||
for i, (expr, tree) in enumerate(fns):
|
||
color = _safe_color(expr.color, _PALETTE[i % len(_PALETTE)])
|
||
colors.append(color)
|
||
polylines.append(_sample_segments(tree, xmin, xmax, ymin, ymax))
|
||
|
||
return PlotGeometry(
|
||
width=_WIDTH,
|
||
height=_HEIGHT,
|
||
xmin=xmin,
|
||
xmax=xmax,
|
||
ymin=ymin,
|
||
ymax=ymax,
|
||
x_axis_y=x_axis_y,
|
||
y_axis_x=y_axis_x,
|
||
xticks=xticks,
|
||
yticks=yticks,
|
||
polylines=polylines,
|
||
colors=colors,
|
||
xlabel=plot.axes.xlabel,
|
||
ylabel=plot.axes.ylabel,
|
||
grid=plot.axes.grid,
|
||
warnings=warnings,
|
||
)
|
||
|
||
|
||
# --- SVG 序列化（与 compute_geometry 共用，保证字节级稳定） ---
|
||
def _grid_svg(geo: PlotGeometry) -> str:
|
||
sx = lambda x: _sx(x, geo.xmin, geo.xmax)
|
||
sy = lambda y: _sy(y, geo.ymin, geo.ymax)
|
||
parts: list[str] = []
|
||
for x in geo.xticks:
|
||
parts.append(
|
||
f'<line x1="{sx(x):.2f}" y1="{sy(geo.ymin):.2f}" x2="{sx(x):.2f}" '
|
||
f'y2="{sy(geo.ymax):.2f}" stroke="#eaeef2"/>'
|
||
)
|
||
for y in geo.yticks:
|
||
parts.append(
|
||
f'<line x1="{sx(geo.xmin):.2f}" y1="{sy(y):.2f}" x2="{sx(geo.xmax):.2f}" '
|
||
f'y2="{sy(y):.2f}" stroke="#eaeef2"/>'
|
||
)
|
||
return "".join(parts)
|
||
|
||
|
||
def _axes_svg(geo: PlotGeometry) -> str:
|
||
sx = lambda x: _sx(x, geo.xmin, geo.xmax)
|
||
sy = lambda y: _sy(y, geo.ymin, geo.ymax)
|
||
parts: list[str] = []
|
||
# 坐标轴：过原点则画在原点，否则贴边，保证始终有参照系
|
||
parts.append(
|
||
f'<line x1="{sx(geo.xmin):.2f}" y1="{sy(geo.x_axis_y):.2f}" x2="{sx(geo.xmax):.2f}" '
|
||
f'y2="{sy(geo.x_axis_y):.2f}" stroke="#57606a"/>'
|
||
)
|
||
parts.append(
|
||
f'<line x1="{sx(geo.y_axis_x):.2f}" y1="{sy(geo.ymin):.2f}" x2="{sx(geo.y_axis_x):.2f}" '
|
||
f'y2="{sy(geo.ymax):.2f}" stroke="#57606a"/>'
|
||
)
|
||
# x 轴刻度数字（画在轴下方）
|
||
for x in geo.xticks:
|
||
parts.append(
|
||
f'<text x="{sx(x):.2f}" y="{sy(geo.x_axis_y) + 14:.2f}" text-anchor="middle" '
|
||
f'font-size="10" fill="#57606a">{html.escape(_fmt_num(x))}</text>'
|
||
)
|
||
# y 轴刻度数字（画在轴左侧）
|
||
for y in geo.yticks:
|
||
parts.append(
|
||
f'<text x="{sx(geo.y_axis_x) - 6:.2f}" y="{sy(y) + 3:.2f}" text-anchor="end" '
|
||
f'font-size="10" fill="#57606a">{html.escape(_fmt_num(y))}</text>'
|
||
)
|
||
return "".join(parts)
|
||
|
||
|
||
def _polylines_svg(geo: PlotGeometry) -> str:
|
||
parts: list[str] = []
|
||
for segments, color in zip(geo.polylines, geo.colors):
|
||
for seg in segments:
|
||
points = " ".join(f"{px:.2f},{py:.2f}" for px, py in seg)
|
||
parts.append(f'<polyline points="{points}" fill="none" stroke="{color}"/>')
|
||
return "".join(parts)
|
||
|
||
|
||
def _labels_svg(geo: PlotGeometry) -> str:
|
||
parts: list[str] = []
|
||
if geo.xlabel:
|
||
parts.append(
|
||
f'<text x="{geo.width / 2:.2f}" y="{geo.height - 10:.2f}" text-anchor="middle" '
|
||
f'font-size="12" fill="#1f2328">{html.escape(geo.xlabel)}</text>'
|
||
)
|
||
if geo.ylabel:
|
||
parts.append(
|
||
f'<text x="16" y="{geo.height / 2:.2f}" text-anchor="middle" font-size="12" '
|
||
f'fill="#1f2328" transform="rotate(-90 16 {geo.height / 2:.2f})">'
|
||
f'{html.escape(geo.ylabel)}</text>'
|
||
)
|
||
return "".join(parts)
|
||
|
||
|
||
def render_svg(plot: FunctionPlot) -> StaticRenderResult:
|
||
"""把已解析的 FunctionPlot 渲染为内嵌 SVG。"""
|
||
geo = compute_geometry(plot)
|
||
parts: list[str] = [
|
||
f'<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 {geo.width} {geo.height}" role="img">'
|
||
]
|
||
if geo.grid:
|
||
parts.append(_grid_svg(geo))
|
||
parts.append(_axes_svg(geo))
|
||
parts.append(_polylines_svg(geo))
|
||
parts.append(_labels_svg(geo))
|
||
parts.append("</svg>")
|
||
|
||
return StaticRenderResult(
|
||
content="".join(parts),
|
||
width=geo.width,
|
||
height=geo.height,
|
||
warnings=geo.warnings,
|
||
)
|