Janady13/thatdeveloperguy-website / tools/svg2rml.py

1#!/usr/bin/env python3 2"""SVG -> RML shapes (native Rive paths; the CLI cannot import SVG). 3 4Handles the Adobe image-trace dialect and simple hand-authored SVG: M/L/H/V/C/Q/Z (absolute or relative), no transforms/clips, fills as rgb()/#hex/url(#gradient), solid strokes (width/cap/join/opacity), 5linear/radial gradients in userSpaceOnUse. Output: a list of <Shape> strings, FRONT-MOST FIRST (RML paint order is the 6reverse of SVG document order), with vertices in artboard space after (scale, dx, dy). Verify with rive . --verify / inspect. 7 8 from tools.svg2rml import svg_to_shapes 9 shapes, stats = svg_to_shapes(svg_text, scale=0.25, dx=100, dy=50, nid=idgen, name_prefix="door") 10""" 11import math, re 12 13_num = r"[-+]?(?:\d+\.?\d*|\.\d+)(?:[eE][-+]?\d+)?" 14 15 16def _rgb(text): 17 m = re.search(r"rgb\(\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*\)", text) 18 if m: 19 return tuple(int(v) for v in m.groups()) 20 m = re.search(r"#([0-9a-fA-F]{6})", text) 21 if m: 22 h = m.group(1) 23 return int(h[0:2], 16), int(h[2:4], 16), int(h[4:6], 16) 24 return None 25 26 27def _argb(rgb, alpha=1.0): 28 return "%02X%02X%02X%02X" % (round(alpha * 255), rgb[0], rgb[1], rgb[2]) 29 30 31def parse_gradients(svg): 32 grads = {} 33 for kind, attrs, body in re.findall(r"<(linear|radial)Gradient([^>]*)>(.*?)</\1Gradient>", svg, re.S): 34 a = dict(re.findall(r'([\w:-]+)="([^"]*)"', attrs)) 35 stops = [] 36 for s_attrs in re.findall(r"<stop([^>]*)/?>", body): 37 sa = dict(re.findall(r'([\w:-]+)="([^"]*)"', s_attrs)) 38 off = sa.get("offset", "0").strip() 39 pos = float(off[:-1]) / 100 if off.endswith("%") else float(off) 40 style = sa.get("style", "") 41 col = _rgb(style) or _rgb(sa.get("stop-color", "")) or (0, 0, 0) 42 om = re.search(r"stop-opacity:\s*([\d.]+)", style) 43 op = float(om.group(1)) if om else float(sa.get("stop-opacity", 1)) 44 stops.append((pos, col, op)) 45 grads[a["id"]] = dict(kind=kind, attrs=a, stops=stops) 46 return grads 47 48 49def _subpaths(d): 50 """Split a path (M/L/H/V/C/Q/Z, absolute or relative; Q is elevated to a cubic) into contours: 51 (points, controls, closed) where controls[i] = (c1, c2) for the segment points[i] -> points[i+1].""" 52 tokens = re.findall(r"[MLHVCQZmlhvcqz]|" + _num, d) 53 contours = [] 54 i = 0 55 cur = (0.0, 0.0); start = (0.0, 0.0) 56 pts, ctrls = [], [] 57 cmd = None 58 def flush(closed): 59 nonlocal pts, ctrls 60 if pts: 61 if closed and len(pts) > 1 and abs(pts[-1][0] - pts[0][0]) < 1e-6 and abs(pts[-1][1] - pts[0][1]) < 1e-6: 62 pts.pop() 63 elif closed: 64 ctrls.append((pts[-1], pts[0])) # straight close 65 contours.append((pts, ctrls, closed)) 66 pts, ctrls = [], [] 67 while i < len(tokens): 68 t = tokens[i] 69 if re.match(r"[A-Za-z]$", t): 70 cmd = t; i += 1 71 if cmd in "Zz": 72 flush(True); cur = start 73 continue 74 rel = cmd.islower(); c = cmd.upper() 75 def P(k): # k-th number pair from i 76 x, y = float(tokens[i + k]), float(tokens[i + k + 1]) 77 return (cur[0] + x, cur[1] + y) if rel else (x, y) 78 if c == "M": 79 flush(False); cur = P(0); start = cur; pts, ctrls = [cur], []; i += 2; cmd = "l" if rel else "L" 80 elif c == "L": 81 p = P(0); ctrls.append((cur, p)); pts.append(p); cur = p; i += 2 82 elif c == "H": 83 x = float(tokens[i]); p = ((cur[0] + x) if rel else x, cur[1]); ctrls.append((cur, p)); pts.append(p); cur = p; i += 1 84 elif c == "V": 85 y = float(tokens[i]); p = (cur[0], (cur[1] + y) if rel else y); ctrls.append((cur, p)); pts.append(p); cur = p; i += 1 86 elif c == "C": 87 c1, c2, p = P(0), P(2), P(4); ctrls.append((c1, c2)); pts.append(p); cur = p; i += 6 88 elif c == "Q": 89 q, p = P(0), P(2) 90 c1 = (cur[0] + 2 / 3 * (q[0] - cur[0]), cur[1] + 2 / 3 * (q[1] - cur[1])); c2 = (p[0] + 2 / 3 * (q[0] - p[0]), p[1] + 2 / 3 * (q[1] - p[1])) 91 ctrls.append((c1, c2)); pts.append(p); cur = p; i += 4 92 else: 93 i += 1 94 flush(False) 95 return contours 96 97 98def _vertex(prev_c2, p, next_c1, T): 99 x, y = T(*p) 100 ins = T(*prev_c2) if prev_c2 else (x, y) 101 outs = T(*next_c1) if next_c1 else (x, y) 102 din = math.hypot(ins[0] - x, ins[1] - y); dout = math.hypot(outs[0] - x, outs[1] - y) 103 if din < 0.05 and dout < 0.05: 104 return f'<StraightVertex x="{x:.2f}" y="{y:.2f}"/>' 105 rin = math.atan2(ins[1] - y, ins[0] - x); rout = math.atan2(outs[1] - y, outs[0] - x) 106 return f'<CubicDetachedVertex x="{x:.2f}" y="{y:.2f}" inRotation="{rin:.4f}" inDistance="{din:.2f}" outRotation="{rout:.4f}" outDistance="{dout:.2f}"/>' 107 108 109def svg_to_shapes(svg, scale=1.0, dx=0.0, dy=0.0, nid=None, name_prefix="p", drop_fill=None, drop_tolerance=90, min_area=0.0): 110 """Returns (shapes_front_to_back, stats). drop_fill=(r,g,b) removes chroma-key paths (Manhattan distance < drop_tolerance).""" 111 grads = parse_gradients(svg) 112 T = lambda x, y: (x * scale + dx, y * scale + dy) 113 shapes = [] 114 stats = dict(paths=0, dropped=0, vertices=0, gradients=0) 115 for attrs in re.findall(r"<path([^>]*)/?>", svg): 116 a = dict(re.findall(r'([\w:-]+)="([^"]*)"', attrs)) 117 d = a.get("d"); fill = a.get("fill", "#000000") 118 if not d or (fill == "none" and a.get("stroke", "none") == "none"): 119 continue 120 op = float(a.get("opacity", a.get("fill-opacity", 1))) 121 solid = _rgb(fill) if not fill.startswith("url(") else None 122 if drop_fill and solid and sum(abs(solid[i] - drop_fill[i]) for i in range(3)) < drop_tolerance: 123 stats["dropped"] += 1; continue 124 paths_xml = [] 125 for pts, ctrls, closed in _subpaths(d): 126 if len(pts) < 2: 127 continue 128 n = len(pts) 129 vs = [] 130 for i, p in enumerate(pts): 131 prev_c2 = ctrls[i - 1][1] if (i > 0 and i - 1 < len(ctrls)) else (ctrls[-1][1] if (closed and len(ctrls) == n) else None) 132 next_c1 = ctrls[i][0] if i < len(ctrls) else None 133 vs.append(_vertex(prev_c2, p, next_c1, T)) 134 stats["vertices"] += len(vs) 135 paths_xml.append(f'<PointsPath isClosed="{"true" if closed else "false"}" name="P">{"".join(vs)}</PointsPath>') 136 if not paths_xml: 137 continue 138 if min_area > 0: # drop trace shards: bbox (artboard px) below the floor 139 xs = [float(v) for v in re.findall(r' x="([-\d.]+)"', "".join(paths_xml))]; ys = [float(v) for v in re.findall(r' y="([-\d.]+)"', "".join(paths_xml))] 140 if xs and (max(xs) - min(xs)) * (max(ys) - min(ys)) < min_area: 141 stats["dropped"] += 1; continue 142 stroke = "" 143 sc = a.get("stroke", "none") 144 if sc != "none" and _rgb(sc): 145 cap = {"round": "round", "square": "square"}.get(a.get("stroke-linecap", ""), "butt"); join = {"round": "round", "bevel": "bevel"}.get(a.get("stroke-linejoin", ""), "miter") 146 sop = float(a.get("stroke-opacity", 1)) * op 147 stroke = f'<Stroke thickness="{float(a.get("stroke-width", 1)) * scale:.2f}" cap="{cap}" join="{join}" name="S"><SolidColor colorValue="{_argb(_rgb(sc), sop)}" name="C"/></Stroke>' 148 if fill == "none": 149 paint = "" 150 elif solid: 151 paint = f'<Fill name="F"><SolidColor colorValue="{_argb(solid, op)}" name="C"/></Fill>' 152 else: 153 gid = re.search(r"url\(#([^)]+)\)", fill).group(1); g = grads.get(gid) 154 if not g: 155 continue 156 stats["gradients"] += 1 157 ga = g["attrs"] 158 stops = "".join(f'<GradientStop colorValue="{_argb(c, o * op)}" position="{pos:.3f}"/>' for pos, c, o in g["stops"]) 159 if g["kind"] == "linear": 160 x1, y1 = T(float(ga.get("x1", 0)), float(ga.get("y1", 0))); x2, y2 = T(float(ga.get("x2", 0)), float(ga.get("y2", 0))) 161 paint = f'<Fill name="F"><LinearGradient startX="{x1:.2f}" startY="{y1:.2f}" endX="{x2:.2f}" endY="{y2:.2f}" name="G">{stops}</LinearGradient></Fill>' 162 else: 163 cx, cy = T(float(ga.get("cx", 0)), float(ga.get("cy", 0))); r = float(ga.get("r", 1)) * scale 164 paint = f'<Fill name="F"><RadialGradient startX="{cx:.2f}" startY="{cy:.2f}" endX="{cx + r:.2f}" endY="{cy:.2f}" name="G">{stops}</RadialGradient></Fill>' 165 stats["paths"] += 1 166 sid = nid() if nid else "" 167 idattr = f' id="{sid}"' if sid else "" 168 shapes.append(f'<Shape x="0" y="0" name="{name_prefix}_{stats["paths"]}"{idattr}>{"".join(paths_xml)}{paint}{stroke}</Shape>') 169 shapes.reverse() # SVG paints last-on-top; RML paints first-on-top 170 return shapes, stats 171 172 173if __name__ == "__main__": 174 import sys, json 175 svg = open(sys.argv[1]).read() 176 shapes, stats = svg_to_shapes(svg) 177 print(json.dumps(stats)) 178