@cristianormazabal/triton-core 0.1.4 → 0.1.6

package/dist/index.js

@@ -10462,7 +10462,7 @@ function connectorExtents(elements) {
 // src/diagrams/triton/ds/struct/shared.ts
 var ARROW_ID2 = "struct-arrow";
 function arrowDef(color) {
-  return `<marker id="${ARROW_ID2}" markerWidth="8" markerHeight="8" refX="7" refY="4" orient="auto"><path d="M0 0 L8 4 L0 8 z" fill="${color}" /></marker>`;
+  return `<marker id="${ARROW_ID2}" markerWidth="12" markerHeight="12" refX="10" refY="6" orient="auto" markerUnits="userSpaceOnUse"><path d="M0 0 L12 6 L0 12 z" fill="${color}" /></marker>`;
 }
 function lines(input) {
   return input.split(/\r?\n/).map((l) => l.trim()).filter(Boolean);
@@ -10513,6 +10513,8 @@ function parse(input) {
   let indexSide = "before";
   let indexOrder = "normal";
   const ptrs = [];
+  let highlights;
+  let win;
   for (const line of lines(input)) {
     const t = tokenizeDirective(line);
     if (t.length === 0) continue;
@@ -10541,6 +10543,15 @@ function parse(input) {
       }
       continue;
     }
+    if (t[0] === "highlight") {
+      highlights = t.slice(1).map(Number).filter((n) => !isNaN(n) && Number.isInteger(n));
+      continue;
+    }
+    if (t[0] === "window" && t[1]) {
+      const m = t[1].match(/^(\d+)-(\d+)$/);
+      if (m) win = { start: Number(m[1]), end: Number(m[2]) };
+      continue;
+    }
     if (t[0] === "ptr" && t.length >= 4 && t[2] === "->") {
       ptrs.push({ id: t[1], target: parsePointerTarget(t[3]), ...t[4] !== void 0 ? { label: t[4] } : {} });
     }
@@ -10550,7 +10561,9 @@ function parse(input) {
     axis,
     index: { show: indexShow, side: indexSide, order: indexOrder },
     cells: cells2,
-    ptrs
+    ptrs,
+    ...highlights !== void 0 && highlights.length > 0 ? { highlights } : {},
+    ...win !== void 0 ? { window: win } : {}
   };
 }
 function parseCells(tokens) {
@@ -10649,9 +10662,15 @@ function layoutArray(inputDoc, theme) {
       elements.push(p.text("\u2026", slot.x + cellW / 2, slot.y + cellH / 2 + 5, font, palette.textMuted, { anchor: "middle", weight: "bold" }));
       return;
     }
-    elements.push(p.rect(slot, palette.surface, palette.border, 1.5, { rx: 3 }));
-    elements.push(p.text(cell.value, slot.x + cellW / 2, slot.y + cellH / 2 + 5, font, palette.text, { anchor: "middle", weight: "bold" }));
-    const logical = meta.physicalToLogical[physical];
+    const logicalMaybe = meta.physicalToLogical[physical];
+    const logical = logicalMaybe ?? null;
+    const isHighlit = logical !== null && ((doc.highlights?.includes(logical) ?? false) || doc.window !== void 0 && logical >= doc.window.start && logical <= doc.window.end);
+    if (isHighlit) {
+      elements.push(p.rect(slot, palette.primary, palette.primary, 2, { rx: 3, fillOpacity: 0.22 }));
+    } else {
+      elements.push(p.rect(slot, palette.surface, palette.border, 1.5, { rx: 3 }));
+    }
+    elements.push(p.text(cell.value, slot.x + cellW / 2, slot.y + cellH / 2 + 5, font, isHighlit ? palette.primary : palette.text, { anchor: "middle", weight: "bold" }));
     if (doc.index.show && logical !== null) {
       elements.push(renderIndexText(p, theme, String(logical), slot, cellW, cellH, doc.axis, doc.index.side, indexPad));
     }
@@ -10707,7 +10726,9 @@ function normalizeArrayDoc(doc) {
     ptrs: (doc.ptrs ?? []).map((ptr) => {
       if ("idx" in ptr) return { id: ptr.name ?? ptr.id, target: { raw: String(ptr.idx), value: Number(ptr.idx) } };
       return ptr;
-    })
+    }),
+    ...Array.isArray(doc.highlights) && doc.highlights.length > 0 ? { highlights: doc.highlights } : {},
+    ...doc.window !== void 0 ? { window: doc.window } : {}
   };
 }
 function buildSlots(count, origin, cellW, cellH, axis) {
@@ -10974,7 +10995,8 @@ function layoutPoster(ir, theme) {
       const contentW = colW - inset * 2;
       const scale = Math.min(contentW / Math.max(result.scene.viewBox.width, 1), 1);
       const effectiveScale = Math.max(scale, MIN_EMBED_SCALE);
-      const neededH = result.scene.viewBox.height * effectiveScale + cellTitleH + inset * 2;
+      const captionH = cell.caption ? reservedCaptionHeight(cellTheme) : 0;
+      const neededH = result.scene.viewBox.height * effectiveScale + cellTitleH + captionH + inset * 2;
       rowHeights[row] = Math.max(rowHeights[row], neededH);
     }
   }
@@ -11023,8 +11045,23 @@ function layoutPoster(ir, theme) {
       textOccupied.push(t.occupied);
     }
     const inset = unit / 2;
-    const contentRect = { x: cellX + inset, y: cellY + reservedTop + inset, width: cellW - inset * 2, height: cellH - reservedTop - inset * 2 };
+    const captionH = cell.caption ? reservedCaptionHeight(cellTheme) : 0;
+    const contentRect = {
+      x: cellX + inset,
+      y: cellY + reservedTop + inset,
+      width: cellW - inset * 2,
+      height: cellH - reservedTop - captionH - inset * 2
+    };
     cellContent.push(embedScene(result.scene, contentRect));
+    if (cell.caption) {
+      const captionY = cellY + cellH - inset / 2;
+      cellContent.push({ type: "text", content: cell.caption, position: { x: cellX + cellW / 2, y: captionY }, fontSize: cellTheme.typography.smallFontSize, fontFamily: cellTheme.typography.fontFamily, fill: cellTheme.palette.textMuted, anchor: "middle" });
+    }
+    if (cell.notes && cell.notes.length > 0) {
+      for (const note of cell.notes) {
+        cellContent.push(...buildNoteOverlay(note, contentRect, cellTheme));
+      }
+    }
     const scaleX = contentRect.width / Math.max(result.scene.viewBox.width, 1);
     const scaleY = contentRect.height / Math.max(result.scene.viewBox.height, 1);
     const scale = Math.min(scaleX, scaleY, 1);
@@ -11235,6 +11272,47 @@ function titleBoxHeight(theme) {
   const padY = theme.panel.titleChrome === "none" ? 0 : theme.spacing.unit * 0.4;
   return fs + padY * 2;
 }
+function reservedCaptionHeight(theme) {
+  return theme.typography.smallFontSize + theme.spacing.unit;
+}
+function buildNoteOverlay(note, rect, theme) {
+  const { palette, typography, spacing } = theme;
+  const fs = typography.smallFontSize;
+  const pad = spacing.unit * 0.5;
+  const textW = note.text.length * fs * 0.62 + pad * 2;
+  const boxH = fs + pad * 2;
+  const pos = note.position ?? "top-right";
+  let bx, by;
+  switch (pos) {
+    case "top-left":
+      bx = rect.x + spacing.unit * 0.5;
+      by = rect.y + spacing.unit * 0.5;
+      break;
+    case "top-right":
+      bx = rect.x + rect.width - textW - spacing.unit * 0.5;
+      by = rect.y + spacing.unit * 0.5;
+      break;
+    case "bottom-left":
+      bx = rect.x + spacing.unit * 0.5;
+      by = rect.y + rect.height - boxH - spacing.unit * 0.5;
+      break;
+    case "bottom-right":
+      bx = rect.x + rect.width - textW - spacing.unit * 0.5;
+      by = rect.y + rect.height - boxH - spacing.unit * 0.5;
+      break;
+    case "center":
+      bx = rect.x + (rect.width - textW) / 2;
+      by = rect.y + (rect.height - boxH) / 2;
+      break;
+    default:
+      bx = rect.x + rect.width - textW - spacing.unit * 0.5;
+      by = rect.y + spacing.unit * 0.5;
+  }
+  return [
+    { type: "rect", bounds: { x: bx, y: by, width: textW, height: boxH }, fill: palette.surface, stroke: palette.primary, strokeWidth: 1, rx: boxH / 2, fillOpacity: 0.88 },
+    { type: "text", content: note.text, position: { x: bx + textW / 2, y: by + pad + fs * 0.8 }, fontSize: fs, fontFamily: typography.fontFamily, fontWeight: "bold", fill: palette.primary, anchor: "middle" }
+  ];
+}
 function buildCellTitle(title, cellX, cellY, cellW, theme) {
   const { palette, typography, panel, spacing } = theme;
   const unit = spacing.unit;
@@ -13953,8 +14031,18 @@ var poster = {
   parseMermaid(input) {
     const raw = peg$parse3(input);
     const cells2 = raw.cells.map((c) => {
-      const content = parseCell(c);
-      return { id: c.id, title: c.title, content, colSpan: c.span?.colSpan, rowSpan: c.span?.rowSpan, theme: c.theme };
+      const { rawContent, caption, notes } = extractCellAnnotations(c.rawContent);
+      const content = parseCell({ ...c, rawContent });
+      return {
+        id: c.id,
+        title: c.title,
+        content,
+        colSpan: c.span?.colSpan,
+        rowSpan: c.span?.rowSpan,
+        theme: c.theme,
+        ...caption ? { caption } : {},
+        ...notes.length > 0 ? { notes } : {}
+      };
     });
     const explicitLinks = raw.links ?? [];
     const allLinks = [...explicitLinks];
@@ -14009,6 +14097,7 @@ function inferCellKind(rawContent) {
   if (keyword.startsWith("nodegraph")) return "nodegraph";
   if (keyword.startsWith("plan")) return "plan";
   if (keyword.startsWith("btree")) return "btree";
+  if (keyword.startsWith("tree")) return "tree";
   if (keyword.startsWith("page")) return "page";
   if (keyword.startsWith("memory")) return "memory";
   if (keyword.startsWith("linkedlist")) return "linkedlist";
@@ -14033,6 +14122,28 @@ function inferCellKind(rawContent) {
   if (trimmed.includes("|")) return "stat";
   return "text";
 }
+var CAPTION_RE = /^[ \t]*caption[ \t]+"([^"]*)"[ \t]*$/;
+var NOTE_RE = /^[ \t]*note[ \t]+"([^"]*)"(?:[ \t]+at[ \t]+(top-left|top-right|bottom-left|bottom-right|center))?[ \t]*$/;
+function extractCellAnnotations(rawContent) {
+  const inputLines = rawContent.split("\n");
+  const outLines = [];
+  let caption;
+  const notes = [];
+  for (const line of inputLines) {
+    const cm = line.match(CAPTION_RE);
+    if (cm) {
+      caption = cm[1];
+      continue;
+    }
+    const nm = line.match(NOTE_RE);
+    if (nm) {
+      notes.push({ text: nm[1], ...nm[2] ? { position: nm[2] } : {} });
+      continue;
+    }
+    outLines.push(line);
+  }
+  return { rawContent: outLines.join("\n"), caption, notes };
+}
 
 // src/diagrams/mermaid/pie/layout.ts
 function fmtNum(n) {
@@ -39248,7 +39359,7 @@ function nodeStyle(kinds, theme) {
   if (has("join")) return { shape, fill: palette.surface, stroke: palette.secondary, text: palette.text };
   if (has("build") || has("muted")) return { shape, fill: palette.surface, stroke: palette.textMuted, text: palette.text };
   const fill = palette.surface;
-  return { shape, fill, stroke: outlineStroke(fill, theme), text: palette.text };
+  return { shape, fill, stroke: palette.primary, text: palette.text };
 }
 function infoLine(node) {
   return node.info;
@@ -39317,18 +39428,22 @@ function layoutTree(ir, theme) {
     ));
   }
   const byId = new Map(ir.nodes.map((n) => [n.id, n]));
+  const activeEdgeSet = new Set((ir.activePaths ?? []).map(([a, b]) => `${a}:${b}`));
   for (const node of ir.nodes) {
     const pb = box(node.id);
     for (const cid of node.children) {
       const cb = box(cid);
       const { start, end } = connectSlots(pb, cb);
-      elements.push(p.path(`M ${rhu(start.x)} ${rhu(start.y)} L ${rhu(end.x)} ${rhu(end.y)}`, palette.textMuted, 1.5));
+      const isActive = activeEdgeSet.has(`${node.id}:${cid}`);
+      const edgeColor = isActive ? palette.primary : palette.textMuted;
+      const edgeWidth = isActive ? 2.5 : 1.5;
+      elements.push(p.path(`M ${rhu(start.x)} ${rhu(start.y)} L ${rhu(end.x)} ${rhu(end.y)}`, edgeColor, edgeWidth));
       const child = byId.get(cid);
       if (child.edgeLabel) {
         const mx = (start.x + end.x) / 2, my = (start.y + end.y) / 2;
         const w = measureText(child.edgeLabel, smallFont).width + 8;
         elements.push(p.rect({ x: mx - w / 2, y: my - 9, width: w, height: 16 }, palette.background, palette.background, 0, { rx: 3 }));
-        elements.push(p.text(child.edgeLabel, mx, my + 3, smallFont, palette.primary, { anchor: "middle", weight: "bold" }));
+        elements.push(p.text(child.edgeLabel, mx, my + 3, smallFont, isActive ? palette.primary : palette.textMuted, { anchor: "middle", weight: "bold" }));
       }
     }
   }
@@ -40831,14 +40946,46 @@ function buildNodes(lines2) {
 }
 
 // src/diagrams/triton/ds/tree/index.ts
+function extractPathDirectives(input) {
+  const cleanLines = [];
+  const paths = [];
+  const PATH_RE = /^[ \t]*path[ \t]+(.+)$/;
+  for (const line of input.split("\n")) {
+    const m = line.match(PATH_RE);
+    if (m && m[1].includes("->")) {
+      const nodes = m[1].split("->").map((s) => s.trim()).filter(Boolean);
+      if (nodes.length >= 2) {
+        paths.push(nodes);
+        continue;
+      }
+    }
+    cleanLines.push(line);
+  }
+  return { clean: cleanLines.join("\n"), paths };
+}
 var tree = {
   parseMermaid(input) {
-    const raw = parseLines(input);
+    const { clean, paths } = extractPathDirectives(input);
+    const raw = parseLines(clean);
+    const nodes = buildNodes(raw.lines);
+    const labelToId = /* @__PURE__ */ new Map();
+    for (const node of nodes) {
+      if (!labelToId.has(node.label)) labelToId.set(node.label, node.id);
+    }
+    const activePairs = [];
+    for (const seg of paths) {
+      for (let i = 0; i < seg.length - 1; i++) {
+        const fromId = labelToId.get(seg[i]);
+        const toId = labelToId.get(seg[i + 1]);
+        if (fromId && toId) activePairs.push([fromId, toId]);
+      }
+    }
     return {
       version: raw.version,
       metadata: raw.metadata ?? {},
       direction: raw.direction,
-      nodes: buildNodes(raw.lines)
+      nodes,
+      ...activePairs.length > 0 ? { activePaths: activePairs } : {}
     };
   },
   parseYaml(input) {
@@ -41590,7 +41737,7 @@ function buildStrip(pen2, theme, cells2, options) {
     const y = origin.y + (horizontal ? 0 : i * step);
     const slot = { x, y, width: cellWidth, height: cellHeight };
     slots.push(slot);
-    elements.push(pen2.rect(slot, cell.fill ?? palette.surface, palette.border, 1.5, { rx: 3 }));
+    elements.push(pen2.rect(slot, cell.fill ?? palette.surface, cell.stroke ?? palette.border, 1.5, { rx: 3, ...cell.fillOpacity !== void 0 ? { fillOpacity: cell.fillOpacity } : {} }));
     if (cell.label !== void 0) {
       elements.push(pen2.text(
         cell.label,
@@ -42429,6 +42576,7 @@ function parse11(input) {
   let title;
   const rows = [];
   let showIndex = true;
+  const highlights = [];
   for (const line of input.split(/\r?\n/).map((l) => l.trimEnd())) {
     const trimmed = line.trim();
     if (!trimmed) continue;
@@ -42453,8 +42601,15 @@ function parse11(input) {
       rows.push(t.slice(1));
       continue;
     }
+    if (t[0] === "highlight") {
+      for (const tok of t.slice(1)) {
+        const m = tok.match(/^(\d+),(\d+)$/);
+        if (m) highlights.push([Number(m[1]), Number(m[2])]);
+      }
+      continue;
+    }
   }
-  return { ...title !== void 0 ? { title } : {}, rows, showIndex };
+  return { ...title !== void 0 ? { title } : {}, rows, showIndex, ...highlights.length > 0 ? { highlights } : {} };
 }
 function layoutMatrix(doc, theme) {
   const { palette, typography, spacing } = theme;
@@ -42477,7 +42632,10 @@ function layoutMatrix(doc, theme) {
   const anchors = {};
   grid.forEach((row, r) => {
     const rowOrigin = { x: origin.x, y: origin.y + r * cellH };
-    const cellInputs = row.map((v) => ({ label: v }));
+    const cellInputs = row.map((v, c) => {
+      const isHighlit = doc.highlights?.some(([hr, hc]) => hr === r && hc === c) ?? false;
+      return isHighlit ? { label: v, fill: palette.primary, fillOpacity: 0.22, stroke: palette.primary } : { label: v };
+    });
     const strip = buildStrip(p, theme, cellInputs, { origin: rowOrigin, cellWidth: cellW, cellHeight: cellH });
     elements.push(...strip.elements);
     strip.slots.forEach((slot, c) => {
@@ -42595,9 +42753,15 @@ function parse12(input) {
     const m = line.match(EDGE_RE);
     if (m) {
       const from = m[1], to = m[3];
+      const rawLabel = m[4]?.trim();
+      let label;
+      let kind;
+      if (rawLabel === "active") kind = "active";
+      else if (rawLabel === "dashed") kind = "dashed";
+      else label = rawLabel;
       ensure(from);
       ensure(to);
-      edges2.push({ from, to, ...m[4] ? { label: m[4].trim() } : {} });
+      edges2.push({ from, to, ...label ? { label } : {}, ...kind ? { kind } : {} });
     }
   }
   const nodes = order.map((id) => ({ id, label: labels.get(id) }));
@@ -42628,13 +42792,19 @@ function layoutGraph(doc, theme) {
     const b = placed.boxes.has(e.to) ? box(e.to) : void 0;
     if (!a || !b) continue;
     const { start, end } = connectSlots(a, b);
-    const opts2 = doc.directed ? { markerEnd: ARROW_ID2 } : {};
-    elements.push(p.path(`M ${rhu(start.x)} ${rhu(start.y)} L ${rhu(end.x)} ${rhu(end.y)}`, palette.textMuted, 1.5, opts2));
+    const isActive = e.kind === "active";
+    const edgeColor = isActive ? palette.primary : palette.textMuted;
+    const edgeWidth = isActive ? 2.5 : 1.5;
+    const pathOpts = {
+      ...doc.directed ? { markerEnd: ARROW_ID2 } : {},
+      ...e.kind === "dashed" ? { dash: "6 3" } : {}
+    };
+    elements.push(p.path(`M ${rhu(start.x)} ${rhu(start.y)} L ${rhu(end.x)} ${rhu(end.y)}`, edgeColor, edgeWidth, pathOpts));
     if (e.label) {
       const mx = (start.x + end.x) / 2, my = (start.y + end.y) / 2;
       const w = measureText(e.label, small).width + 8;
       elements.push(p.rect({ x: mx - w / 2, y: my - 9, width: w, height: 16 }, palette.background, palette.background, 0, { rx: 3 }));
-      elements.push(p.text(e.label, mx, my + 3, small, palette.primary, { anchor: "middle", weight: "bold" }));
+      elements.push(p.text(e.label, mx, my + 3, small, isActive ? palette.primary : palette.textMuted, { anchor: "middle", weight: "bold" }));
     }
   }
   const anchors = {};