structviz 0.2__py3-none-any.whl

structviz/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1"

structviz/cli.py

@@ -0,0 +1,92 @@
+import click
+import yaml
+import os
+import json
+from .models import EquityChart, OrgChart
+from .generators import generate_equity_dot, generate_org_dot, generate_org_excalidraw, generate_org_d2
+from .render import render_dot
+from .html_wrapper import generate_html
+from .excalidraw_render import render_excalidraw_to_png
+from .d2_render import render_d2
+
+@click.command()
+@click.argument('yaml_file', type=click.Path(exists=True))
+@click.option('-t', '--type', 'chart_type', required=True,
+              type=click.Choice(['equity', 'org']), help='Chart type')
+@click.option('-f', '--formats', default='png',
+              help='Comma-separated output formats (png,pdf,svg,html,excalidraw,d2)')
+@click.option('-o', '--output', default='output/', help='Output directory')
+@click.option('-p', '--papersize', default=None,
+              help='Paper preset (a4/a3/letter) or custom "W,H[mm|in]" '
+                   'e.g. "210,297" or "8.5,11in" (overrides YAML)')
+@click.option('-d', '--dpi', default=None, type=int,
+              help='Output DPI (overrides YAML meta.dpi)')
+def main(yaml_file, chart_type, formats, output, papersize, dpi):
+    """Generate charts from YAML definition."""
+    with open(yaml_file, 'r', encoding='utf-8') as f:
+        data = yaml.safe_load(f)
+
+    os.makedirs(output, exist_ok=True)
+    base_name = os.path.splitext(os.path.basename(yaml_file))[0]
+    output_base = os.path.join(output, base_name)
+
+    if chart_type == 'equity':
+        chart = EquityChart(**data)
+    elif chart_type == 'org':
+        chart = OrgChart(**data)
+    else:
+        raise click.BadParameter("type must be equity or org")
+
+    fmt_list = [f.strip() for f in formats.split(',')]
+
+    # Excalidraw path
+    if 'excalidraw' in fmt_list:
+        if chart_type == 'org':
+            json_str = generate_org_excalidraw(chart, paper_size=papersize or chart.meta.papersize,
+                                                dpi=dpi or chart.meta.dpi)
+            path = output_base + '.excalidraw'
+            with open(path, 'w', encoding='utf-8') as f:
+                f.write(json_str)
+            png_path = output_base + '_ex.png'
+            render_excalidraw_to_png(json_str, png_path)
+            click.echo(f"Excalidraw saved to {path} + {png_path}")
+        else:
+            click.echo("Excalidraw only supports org charts currently.")
+
+    # D2 path
+    if 'd2' in fmt_list and chart_type == 'org':
+        d2_source = generate_org_d2(chart)
+        d2_path = output_base + '.d2'
+        with open(d2_path, 'w', encoding='utf-8') as f:
+            f.write(d2_source)
+        svg_path = render_d2(d2_source, output_base + '_d2.svg')
+        click.echo(f"D2 saved to {d2_path} + {svg_path}")
+
+    # DOT path (png, pdf, svg, html)
+    dot_formats = [f for f in fmt_list if f not in ('excalidraw', 'html', 'd2')]
+    if dot_formats:
+        if chart_type == 'equity':
+            dot_source = generate_equity_dot(chart, paper_size=papersize or chart.meta.papersize,
+                                              dpi=dpi or chart.meta.dpi)
+        else:
+            dot_source = generate_org_dot(chart, paper_size=papersize or chart.meta.papersize,
+                                           dpi=dpi or chart.meta.dpi)
+        render_dot(dot_source, output_base, dot_formats)
+
+    if 'html' in fmt_list:
+        svg_path = output_base + '.svg'
+        if not os.path.exists(svg_path):
+            if chart_type == 'equity':
+                ds = generate_equity_dot(chart, paper_size=papersize or chart.meta.papersize,
+                                          dpi=dpi or chart.meta.dpi)
+            else:
+                ds = generate_org_dot(chart, paper_size=papersize or chart.meta.papersize,
+                                       dpi=dpi or chart.meta.dpi)
+            render_dot(ds, output_base, ['svg'])
+        generate_html(svg_path, chart.meta.title, output_base + '.html')
+
+    if dot_formats or 'html' in fmt_list:
+        click.echo(f"Chart generated in {output}")
+
+if __name__ == '__main__':
+    main()

structviz/d2_render.py

@@ -0,0 +1,169 @@
+"""D2 渲染器：编译 D2 源码 → SVG，含直角线后处理"""
+
+import subprocess
+import tempfile
+import os
+import re
+import xml.etree.ElementTree as ET
+
+
+def render_d2(d2_source: str, output_path: str, cleanup: bool = True) -> str:
+    """编译 D2 源码到 SVG，返回输出文件路径"""
+    # 写临时文件
+    with tempfile.NamedTemporaryFile(mode='w', suffix='.d2', delete=False) as f:
+        tmp_d2 = f.name
+        f.write(d2_source)
+
+    svg_path = output_path
+    if not svg_path.endswith('.svg'):
+        svg_path = output_path + '.svg'
+
+    try:
+        result = subprocess.run(
+            ['d2', '--pad=0', '--layout=elk', tmp_d2, svg_path],
+            capture_output=True, text=True, timeout=60
+        )
+        if result.returncode != 0:
+            raise RuntimeError(f"d2 compile failed: {result.stderr}")
+
+        # SVG 后处理
+        _post_process_svg(svg_path)
+
+    finally:
+        if cleanup:
+            os.unlink(tmp_d2)
+
+    return svg_path
+
+
+def _flatten_tspan(content: str) -> str:
+    """把 <text> 内的 <tspan> 多行拆成独立 <text>，兼容 resvg"""
+    def replace_tspan(match):
+        full = match.group(0)
+        attrs = match.group(1)
+        # 提取属性
+        x_val = re.search(r'x="([^"]*)"', attrs).group(1)
+        y_val = re.search(r'y="([^"]*)"', attrs).group(1)
+        fill = re.search(r'fill="([^"]*)"', attrs)
+        fill = fill.group(1) if fill else '#0A0F25'
+        base_y = float(y_val)
+
+        # 保留 class 和 style
+        cls = re.search(r'class="([^"]*)"', attrs)
+        style = re.search(r'style="([^"]*)"', attrs)
+        extra = f'class="{cls.group(1)}"' if cls else ''
+        if style:
+            extra += f' style="{style.group(1)}"'
+            ta = re.search(r'text-anchor:(\w+)', style.group(1))
+            anchor = ta.group(1) if ta else 'middle'
+        else:
+            anchor = 'middle'
+
+        # 逐行生成独立 <text>
+        result = []
+        current_y = base_y
+        for ts in re.finditer(r'<tspan[^>]*dy="([^"]*)"[^>]*>([^<]*)</tspan>', full):
+            current_y += float(ts.group(1))
+            result.append(
+                f'<text x="{x_val}" y="{current_y:.1f}" fill="{fill}" '
+                f'{extra} text-anchor="{anchor}">{ts.group(2)}</text>'
+            )
+        return '\n'.join(result) if result else full
+
+    # 逐个 <text>...</text> 处理，只替换含 <tspan> 的
+    def patch(s):
+        if '<tspan' not in s:
+            return s
+        m = re.match(r'<text(\s[^>]*)>(.*?)</text>', s, re.DOTALL)
+        if m:
+            return replace_tspan(m)
+        return s
+
+    return re.sub(
+        r'<text\s[^>]*>.*?</text>',
+        lambda m: patch(m.group(0)),
+        content,
+        flags=re.DOTALL
+    )
+
+
+def _post_process_svg(svg_path: str):
+    """SVG 后处理：去嵌套、去 legend、斜线转直角"""
+    with open(svg_path, 'r', encoding='utf-8') as f:
+        content = f.read()
+
+    # 1. 剥离 XML 声明
+    xml_decl = ''
+    if content.startswith('<?xml'):
+        xml_end = content.index('?>') + 2
+        xml_decl = content[:xml_end]
+        content = content[xml_end:]
+
+    # 2. 去外层 SVG，把 xmlns 挪到内层
+    outer = re.match(r'<svg[^>]*>', content).group(0)
+    # 提取所有 xmlns 属性
+    xmlns_attrs = ' '.join(re.findall(r'xmlns(?::\w+)?="[^"]*"', outer))
+    content = re.sub(r'^<svg[^>]*>', '', content, count=1)
+    # 给内层 SVG 补上 xmlns
+    content = re.sub(r'^(<svg)\b', r'\1 ' + xmlns_attrs + ' ', content, count=1)
+    content = re.sub(r'</svg>$', '', content)
+
+    # 3. 删除 layout/routing 图例
+    for legend_class in ['bGF5b3V0', 'cm91dGluZw==']:
+        while True:
+            pattern = f'<g class="{legend_class}">'
+            idx = content.find(pattern)
+            if idx == -1:
+                break
+            depth = 1
+            pos = idx + len(pattern)
+            while depth > 0:
+                ot = content.find('<g ', pos)
+                ct = content.find('</g>', pos)
+                if ot >= 0 and ot < ct:
+                    depth += 1
+                    pos = ot + 3
+                else:
+                    depth -= 1
+                    pos = ct + 4
+            content = content[:idx] + content[pos:]
+
+    # 4. tspan 多行拆成独立 <text>（兼容 resvg）
+    content = _flatten_tspan(content)
+
+    # 5. 查找板块中心 x（font-size:24px 的容器标题）
+    plate_centers = {}
+    for m in re.finditer(r'<text x="([^"]*)" y="([^"]*)"[^>]*font-size:24px[^>]*>([^<]*)</text>', content):
+        plate_centers[m.group(3)] = float(m.group(1))
+
+    # 5. 连接线转直角 + 终点对齐板块中心
+    def fix_path(match):
+        d = match.group(1)
+        parts = d.split()
+        if len(parts) != 6 or parts[0] != 'M' or parts[3] != 'L':
+            return match.group(0)
+
+        x1, y1 = float(parts[1]), float(parts[2])
+        x2, y2 = float(parts[4]), float(parts[5])
+        y_mid = (y1 + y2) / 2
+
+        # 只对到达板块区域的连线（y2 > 0）做中心对齐
+        if y2 > 0 and plate_centers:
+            best_dist = 500
+            target_x = x2
+            for cx in plate_centers.values():
+                dist = abs(x2 - cx)
+                if dist < best_dist:
+                    best_dist = dist
+                    target_x = cx
+        else:
+            target_x = x2
+
+        return f'd="M {x1} {y1} L {x1} {y_mid} L {target_x} {y_mid} L {target_x} {y2}"'
+
+    content = re.sub(r'd="(M [^"]*)"', fix_path, content)
+
+    # 6. 写回
+    content = xml_decl + content
+    with open(svg_path, 'w', encoding='utf-8') as f:
+        f.write(content)

structviz/excalidraw_render.py

@@ -0,0 +1,213 @@
+"""Render Excalidraw JSON to PNG using Pillow."""
+import json
+from PIL import Image, ImageDraw, ImageFont
+
+NAMED_COLORS = {"white": (255,255,255), "black": (0,0,0), "transparent": None,
+               "red": (255,0,0), "green": (0,255,0), "blue": (0,0,255),
+               "grey": (128,128,128), "gray": (128,128,128)}
+
+def _parse_color(c: str):
+    """Return (r,g,b) or None for transparent."""
+    if not c or c == "transparent":
+        return None
+    if c in NAMED_COLORS:
+        return NAMED_COLORS[c]
+    h = c.lstrip('#')
+    return tuple(int(h[i:i+2], 16) for i in (0, 2, 4))
+
+def _get_font(size: int) -> ImageFont.FreeTypeFont:
+    """Find a CJK-capable font, falling back to default."""
+    import os, glob
+    candidates = []
+    # User fonts
+    for pattern in [
+        os.path.expanduser("~/.fonts/**/*.otf"),
+        os.path.expanduser("~/.fonts/**/*.ttf"),
+        os.path.expanduser("~/.fonts/**/*.ttc"),
+        "/usr/share/fonts/**/*.ttc",
+        "/usr/share/fonts/**/*.otf",
+        "/usr/share/fonts/**/*.ttf",
+    ]:
+        candidates.extend(glob.glob(pattern, recursive=True))
+
+    # Prefer CJK fonts
+    cjk = [p for p in candidates if any(k in p.lower() for k in
+            ['cjk', 'noto', 'wqy', 'chinese', 'hans', 'simsun', 'songti', 'heiti', 'hanyishufang'])]
+    for path in cjk + candidates:
+        try:
+            return ImageFont.truetype(path, size)
+        except Exception:
+            continue
+    return ImageFont.load_default()
+
+
+def _split_lines(text: str, font, max_w: int, draw) -> list:
+    """Split text into lines fitting max_w."""
+    lines = text.split('\n')
+    result = []
+    for line in lines:
+        if not line:
+            result.append('')
+            continue
+        bbox = draw.textbbox((0, 0), line, font=font)
+        if bbox[2] - bbox[0] <= max_w:
+            result.append(line)
+        else:
+            # crude line break: split into chunks
+            current = ''
+            for ch in line:
+                test = current + ch
+                bbox = draw.textbbox((0, 0), test, font=font)
+                if bbox[2] - bbox[0] > max_w and current:
+                    result.append(current)
+                    current = ch
+                else:
+                    current = test
+            if current:
+                result.append(current)
+    return result
+
+
+def render_excalidraw_to_png(excalidraw_json: str, output_path: str, scale: float = 1.0):
+    """Render an Excalidraw JSON string to a PNG file."""
+    data = json.loads(excalidraw_json) if isinstance(excalidraw_json, str) else excalidraw_json
+    elements = data.get("elements", [])
+
+    # Find canvas bounds
+    min_x = min_y = float('inf')
+    max_x = max_y = float('-inf')
+    for el in elements:
+        if el.get("isDeleted"):
+            continue
+        x, y = el.get("x", 0), el.get("y", 0)
+        w, h = el.get("width", 0), el.get("height", 0)
+        if el["type"] == "arrow" and "points" in el:
+            for px, py in el["points"]:
+                min_x = min(min_x, x + px)
+                min_y = min(min_y, y + py)
+                max_x = max(max_x, x + px)
+                max_y = max(max_y, y + py)
+        else:
+            min_x = min(min_x, x)
+            min_y = min(min_y, y)
+            max_x = max(max_x, x + w)
+            max_y = max(max_y, y + h)
+
+    pad = 40
+    min_x = max(0, min_x - pad)
+    min_y = max(0, min_y - pad)
+    max_x += pad
+    max_y += pad
+
+    w_img = int((max_x - min_x) * scale)
+    h_img = int((max_y - min_y) * scale)
+    img = Image.new('RGB', (max(w_img, 100), max(h_img, 100)), 'white')
+    draw = ImageDraw.Draw(img)
+
+    # Group elements by type for correct z-order: rectangles first, then text, then arrows
+    rects = [e for e in elements if e["type"] == "rectangle"]
+    texts = [e for e in elements if e["type"] == "text"]
+    arrows = [e for e in elements if e["type"] == "arrow"]
+
+    # Draw rectangles
+    for el in rects:
+        if el.get("isDeleted"):
+            continue
+        x = (el["x"] - min_x) * scale
+        y = (el["y"] - min_y) * scale
+        w = el["width"] * scale
+        h = el["height"] * scale
+        bg = el.get("backgroundColor", "#ffffff")
+        stroke = el.get("strokeColor", "#1e1e1e")
+        sw = max(1, int(el.get("strokeWidth", 1) * scale))
+
+        fill_color = _parse_color(bg) if bg and bg != "transparent" else None
+        if el.get("roundness"):
+            r = int(8 * scale)
+            draw.rounded_rectangle([x, y, x + w, y + h], radius=r,
+                                   fill=fill_color, outline=_parse_color(stroke), width=sw)
+        else:
+            draw.rectangle([x, y, x + w, y + h],
+                          fill=fill_color, outline=_parse_color(stroke), width=sw)
+
+        # Dashed style
+        if el.get("strokeStyle") == "dashed":
+            # Draw dashed outline over solid one
+            import math
+            dash_len = 6 * scale
+            gap_len = 4 * scale
+            total = dash_len + gap_len
+            # Simplified: draw dashed segments along the perimeter
+            perimeter = 2 * (w + h)
+            n_dashes = int(perimeter / total)
+            # Just redraw with PIL dashed (Pillow 10.2+)
+            pass  # Pillow rounded_rectangle doesn't support dashes natively
+
+    # Draw text
+    font = _get_font(14)
+    for el in texts:
+        if el.get("isDeleted"):
+            continue
+        text = el.get("text", "")
+        if not text:
+            continue
+        x = (el["x"] - min_x) * scale
+        y = (el["y"] - min_y) * scale
+        w = el["width"] * scale
+        h = el["height"] * scale
+        color = _parse_color(el.get("strokeColor", "#1e1e1e"))
+        fs = int(el.get("fontSize", 14) * scale)
+        try:
+            font = _get_font(max(8, fs))
+        except Exception:
+            font = ImageFont.load_default()
+
+        lines = text.split('\n')
+        line_h = fs + 4
+        total_h = len(lines) * line_h
+        text_y = y + (h - total_h) / 2
+        for li, line in enumerate(lines):
+            if not line:
+                continue
+            bbox = draw.textbbox((0, 0), line, font=font)
+            tw = bbox[2] - bbox[0]
+            align = el.get("textAlign", "center")
+            if align == "center":
+                tx = x + (w - tw) / 2
+            elif align == "right":
+                tx = x + w - tw - 2
+            else:
+                tx = x + 2
+            draw.text((tx, text_y + li * line_h), line, fill=color, font=font)
+
+    # Draw arrows
+    for el in arrows:
+        if el.get("isDeleted"):
+            continue
+        points = el.get("points", [])
+        if len(points) < 2:
+            continue
+        x0 = (el["x"] - min_x) * scale
+        y0 = (el["y"] - min_y) * scale
+        stroke = _parse_color(el.get("strokeColor", "#7F8C8D"))
+        sw = max(1, int(el.get("strokeWidth", 2) * scale))
+
+        pts = [(x0 + p[0] * scale, y0 + p[1] * scale) for p in points]  # points are relative
+        draw.line(pts, fill=stroke, width=sw)
+
+        # Arrowhead
+        import math
+        x2, y2 = pts[-1]
+        x1, y1 = pts[-2]
+        angle = math.atan2(y2 - y1, x2 - x1)
+        arrow_len = 10 * scale
+        a1 = angle + math.pi * 0.85
+        a2 = angle - math.pi * 0.85
+        draw.polygon([
+            (x2, y2),
+            (x2 - arrow_len * math.cos(a1), y2 - arrow_len * math.sin(a1)),
+            (x2 - arrow_len * math.cos(a2), y2 - arrow_len * math.sin(a2)),
+        ], fill=stroke)
+
+    img.save(output_path, "PNG")
+    print(f"Saved {output_path} ({img.size[0]}x{img.size[1]}px)")

structviz/generators/__init__.py

@@ -0,0 +1,4 @@
+from .equity import generate_equity_dot
+from .org import generate_org_dot
+from .excalidraw import generate_org_excalidraw
+from .d2 import generate_org_d2

structviz/generators/d2.py

@@ -0,0 +1,196 @@
+"""D2 源码生成器：OrgChart 模型 → D2 声明式布局"""
+
+from typing import List, Optional
+from ..models import OrgChart, OrgNode, OrgBlock, OrgGroup
+from ..themes import THEMES
+
+
+def _theme_colors(theme_name: str) -> dict:
+    """从 StructViz theme 提取颜色映射"""
+    theme = THEMES.get(theme_name, THEMES["corporate"])
+    return {
+        "leader_bg": theme.get("card_header_bg", "#2C3E50"),
+        "leader_font": theme.get("card_header_font", "#FFFFFF"),
+        "deputy_bg": theme.get("card_body_bg", "#BDC3C7"),
+        "deputy_font": theme.get("dept_fontcolor", "#2C3E50"),
+        "dept_bg": theme.get("dept_fillcolor", "#EBF5FB"),
+        "dept_border": theme.get("dept_border", "#2980B9"),
+        "dept_font": theme.get("dept_fontcolor", "#2C3E50"),
+        "plate_stroke": theme.get("card_border_color", "#34495E"),
+        "plate_bg": theme.get("cluster_bg", "#F4F6F7"),
+        "edge_color": theme.get("edge_color", "#0D32B2"),
+        "bg": theme.get("bgcolor", "#F4F6F7"),
+    }
+
+
+def _escape(s: str) -> str:
+    """转义 D2 字符串中的特殊字符"""
+    return s.replace('"', '\\"').replace('\n', '\\n')
+
+
+def _wrap_text(s: str, width: int) -> str:
+    """超长文字自动插入 \\n 换行（已有手动换行则跳过自动切分）"""
+    if not width or width <= 0:
+        return s
+    # 已含手动换行 → 保留原样
+    if '\n' in s:
+        return s
+    # 纯单行超长 → 按宽度切分
+    if len(s) <= width:
+        return s
+    wrapped = []
+    rest = s
+    while len(rest) > width:
+        wrapped.append(rest[:width])
+        rest = rest[width:]
+    if rest:
+        wrapped.append(rest)
+    return '\n'.join(wrapped)
+
+
+def _dept_height(title: str) -> int:
+    """根据内容行数估算节点高度"""
+    lines = title.count('\\n') + 1
+    return 40 if lines <= 1 else max(40, lines * 20 + 8)
+
+
+def generate_org_d2(chart: OrgChart) -> str:
+    """将 OrgChart 模型转为 D2 源码"""
+    c = _theme_colors(chart.meta.theme)
+    layout = chart.meta.layout
+    max_cols = layout.max_columns if layout and layout.max_columns else 3
+    wrap_w = layout.wrap_width if layout and layout.wrap_width else 0
+    dept_gap = layout.dept_gap if layout and layout.dept_gap else 6
+    dept_w = layout.dept_width if layout and layout.dept_width else 150
+
+    lines: List[str] = []
+
+    # ── 全局设置 ──
+    lines.append('layout: elk')
+    lines.append('direction: down')
+    lines.append(f'style.fill: "{c["bg"]}"')
+    lines.append('')
+
+    # ── 节点 class 定义 ──
+    lines.append('classes: {')
+    lines.append('  top-node: {')
+    lines.append('    shape: rectangle')
+    lines.append(f'    style: {{ fill: "{c["leader_bg"]}"; font-color: "{c["leader_font"]}"; font-size: 16; bold: true }}')
+    lines.append('    width: 120; height: 50')
+    lines.append('  }')
+    lines.append('  top-sub: {')
+    lines.append('    shape: rectangle')
+    lines.append(f'    style: {{ fill: "{c["deputy_bg"]}"; font-color: "{c["deputy_font"]}"; font-size: 12 }}')
+    lines.append('    width: 90; height: 36')
+    lines.append('  }')
+    lines.append('  dept: {')
+    lines.append('    shape: rectangle')
+    lines.append(f'    style: {{ fill: "{c["dept_bg"]}"; stroke: "{c["dept_border"]}"; font-color: "{c["dept_font"]}"; font-size: 12; border-radius: 4 }}')
+    lines.append(f'    width: {dept_w}; height: 40')
+    lines.append('  }')
+    lines.append('}')
+    lines.append('')
+
+    # ── 收集分组信息 ──
+    leadership_nodes: List[OrgNode] = []
+    deputy_nodes: List[OrgNode] = []
+    plate_blocks: List[OrgBlock] = []
+    leader_id: Optional[str] = None
+
+    for group in chart.groups:
+        if group.level == 1:
+            leadership_nodes = group.nodes
+            deputy_nodes = group.right_nodes
+            if leadership_nodes:
+                leader_id = leadership_nodes[0].id
+        else:
+            for b in group.blocks:
+                if b.position == "bottom":
+                    plate_blocks.append(b)
+
+    # ── 领导层 ──
+    lines.append('leadership: {')
+    lines.append('  near: top-center')
+    lines.append('  grid-rows: 2')
+    lines.append('  grid-gap: 16')
+    lines.append('  label: ""')
+    lines.append('  style: { stroke-width: 0; fill: transparent }')
+    lines.append('')
+
+    # 司库中心
+    if leadership_nodes:
+        leader = leadership_nodes[0]
+        lines.append(f'  {leader.id}: "{_escape(leader.title)}" {{ class: top-node }}')
+    lines.append('')
+
+    # 中心领导虚线框（主任/副主任）
+    if deputy_nodes:
+        dgap = layout.deputy_gap if layout and layout.deputy_gap else 40
+        lines.append('  deputies: {')
+        lines.append('    label: "中心领导"')
+        lines.append('    grid-columns: 2')
+        lines.append(f'    grid-gap: {dgap}')
+        lines.append(f'    style: {{ stroke: "{c["plate_stroke"]}"; stroke-dash: 5; border-radius: 8; fill: "{c["plate_bg"]}" }}')
+        lines.append('')
+        for d in deputy_nodes:
+            lines.append(f'    {d.id}: "{_escape(d.title)}" {{ class: top-sub }}')
+        lines.append('  }')
+        lines.append('')
+
+    # 司库中心 → 中心领导
+    if leader_id and deputy_nodes:
+        lines.append(f'  {leader_id} -> deputies')
+        lines.append('')
+
+    lines.append('}')
+    lines.append('')
+
+    # ── 三板块横排 ──
+    if plate_blocks:
+        lines.append('plates: {')
+        lines.append(f'  grid-columns: {len(plate_blocks)}')
+        lines.append('  grid-gap: 36')
+        lines.append('  style: { stroke-width: 0; fill: transparent; font-color: transparent }')
+        lines.append('')
+
+        for block in plate_blocks:
+            # 分离 cluster_head 和普通部门节点
+            heads = [n for n in block.nodes if n.is_cluster_head]
+            depts = [n for n in block.nodes if not n.is_cluster_head]
+            ncols = min(block.max_columns or max_cols, len(depts)) if depts else 1
+
+            lines.append(f'  {block.id}: {{')
+            lines.append(f'    label: "{_escape(block.label or block.id)}"')
+            lines.append(f'    grid-columns: {ncols}')
+            lines.append(f'    grid-gap: {dept_gap}')
+            lines.append(f'    style: {{ fill: "{c["plate_bg"]}"; stroke: "{c["plate_stroke"]}"; stroke-dash: 5; border-radius: 8 }}')
+            lines.append('')
+
+            # 部门节点（自动换行）
+            for d in depts:
+                title = _wrap_text(d.title, wrap_w) if wrap_w > 0 else d.title
+                escaped = _escape(title)
+                h = _dept_height(escaped)
+                if h > 40:
+                    lines.append(f'    {d.id}: "{escaped}" {{ class: dept; height: {h} }}')
+                else:
+                    lines.append(f'    {d.id}: "{escaped}" {{ class: dept }}')
+
+            # 如果有 cluster_head，也作为 dept 节点（或跳过，用 label 代替）
+            # 这里跳过 cluster_head 节点，因为 block.label 已经充当标题
+            lines.append('  }')
+            lines.append('')
+
+        lines.append('}')
+        lines.append('')
+
+    # ── 连线：中心领导 → 各板块 ──
+    if deputy_nodes and plate_blocks:
+        for block in plate_blocks:
+            lines.append(f'leadership.deputies -> plates.{block.id}')
+    elif leader_id and plate_blocks:
+        for block in plate_blocks:
+            lines.append(f'leadership.{leader_id} -> plates.{block.id}')
+    lines.append('')
+
+    return '\n'.join(lines)