@tracegraph/graph-engine 0.1.4 → 0.2.0

package/dist/index.d.mts

@@ -31,6 +31,12 @@ type GraphNode = {
     size: number;
     /** The original TraceEvent that produced this node. */
     data: TraceEvent;
+    /**
+     * Set when this node represents N collapsed siblings of the same type.
+     * The label will already include the count; this field allows renderers
+     * to display a badge or different visual treatment.
+     */
+    collapsedCount?: number;
 };
 type GraphEdgeType = 'parent' | 'parallel_branch';
 type GraphEdge = {

package/dist/index.d.ts

@@ -31,6 +31,12 @@ type GraphNode = {
     size: number;
     /** The original TraceEvent that produced this node. */
     data: TraceEvent;
+    /**
+     * Set when this node represents N collapsed siblings of the same type.
+     * The label will already include the count; this field allows renderers
+     * to display a badge or different visual treatment.
+     */
+    collapsedCount?: number;
 };
 type GraphEdgeType = 'parent' | 'parallel_branch';
 type GraphEdge = {

package/dist/index.js

@@ -170,7 +170,74 @@ function traceSessionToGraph(session) {
       edges.push({ id: edgeId, source: sourceNodeId, target: targetNodeId, type: edgeType });
     }
   }
-  return { nodes, edges, captureLevel: session.captureLevel };
+  return collapseSiblings({ nodes, edges, captureLevel: session.captureLevel });
+}
+var SIBLING_COLLAPSE_THRESHOLD = 5;
+function collapseSiblings(graph) {
+  const { nodes, edges, captureLevel } = graph;
+  if (nodes.length <= SIBLING_COLLAPSE_THRESHOLD * 2) {
+    return graph;
+  }
+  const nodeById = new Map(nodes.map((n) => [n.id, n]));
+  const childrenByParent = /* @__PURE__ */ new Map();
+  for (const edge of edges) {
+    if (!childrenByParent.has(edge.source)) {
+      childrenByParent.set(edge.source, /* @__PURE__ */ new Map());
+    }
+    const target = nodeById.get(edge.target);
+    if (!target) continue;
+    const typeMap = childrenByParent.get(edge.source);
+    if (!typeMap.has(target.type)) typeMap.set(target.type, []);
+    typeMap.get(target.type).push(edge.target);
+  }
+  const willCollapse = /* @__PURE__ */ new Set();
+  for (const typeMap of childrenByParent.values()) {
+    for (const nodeIds of typeMap.values()) {
+      if (nodeIds.length >= SIBLING_COLLAPSE_THRESHOLD) {
+        for (const id of nodeIds) willCollapse.add(id);
+      }
+    }
+  }
+  if (willCollapse.size === 0) return graph;
+  const summaryNodes = [];
+  const summaryEdges = [];
+  for (const [parentId, typeMap] of childrenByParent) {
+    if (willCollapse.has(parentId)) continue;
+    for (const [type, nodeIds] of typeMap) {
+      if (nodeIds.length < SIBLING_COLLAPSE_THRESHOLD) continue;
+      const first = nodeById.get(nodeIds[0]);
+      const typeName = type.replace(/_/g, " ");
+      const summaryId = `summary__${parentId}__${type}`;
+      summaryNodes.push({
+        id: summaryId,
+        label: `${typeName} \xD7${nodeIds.length}`,
+        type,
+        language: first.language,
+        framework: first.framework,
+        color: first.color,
+        size: Math.min(10, Math.ceil(Math.log10(nodeIds.length + 1)) + 3),
+        collapsedCount: nodeIds.length,
+        data: first.data
+      });
+      summaryEdges.push({
+        id: `${parentId}\u2192${summaryId}`,
+        source: parentId,
+        target: summaryId,
+        type: "parent"
+      });
+    }
+  }
+  return {
+    nodes: [
+      ...nodes.filter((n) => !willCollapse.has(n.id)),
+      ...summaryNodes
+    ],
+    edges: [
+      ...edges.filter((e) => !willCollapse.has(e.source) && !willCollapse.has(e.target)),
+      ...summaryEdges
+    ],
+    captureLevel
+  };
 }
 
 // src/signature.ts

package/dist/index.mjs

@@ -132,7 +132,74 @@ function traceSessionToGraph(session) {
       edges.push({ id: edgeId, source: sourceNodeId, target: targetNodeId, type: edgeType });
     }
   }
-  return { nodes, edges, captureLevel: session.captureLevel };
+  return collapseSiblings({ nodes, edges, captureLevel: session.captureLevel });
+}
+var SIBLING_COLLAPSE_THRESHOLD = 5;
+function collapseSiblings(graph) {
+  const { nodes, edges, captureLevel } = graph;
+  if (nodes.length <= SIBLING_COLLAPSE_THRESHOLD * 2) {
+    return graph;
+  }
+  const nodeById = new Map(nodes.map((n) => [n.id, n]));
+  const childrenByParent = /* @__PURE__ */ new Map();
+  for (const edge of edges) {
+    if (!childrenByParent.has(edge.source)) {
+      childrenByParent.set(edge.source, /* @__PURE__ */ new Map());
+    }
+    const target = nodeById.get(edge.target);
+    if (!target) continue;
+    const typeMap = childrenByParent.get(edge.source);
+    if (!typeMap.has(target.type)) typeMap.set(target.type, []);
+    typeMap.get(target.type).push(edge.target);
+  }
+  const willCollapse = /* @__PURE__ */ new Set();
+  for (const typeMap of childrenByParent.values()) {
+    for (const nodeIds of typeMap.values()) {
+      if (nodeIds.length >= SIBLING_COLLAPSE_THRESHOLD) {
+        for (const id of nodeIds) willCollapse.add(id);
+      }
+    }
+  }
+  if (willCollapse.size === 0) return graph;
+  const summaryNodes = [];
+  const summaryEdges = [];
+  for (const [parentId, typeMap] of childrenByParent) {
+    if (willCollapse.has(parentId)) continue;
+    for (const [type, nodeIds] of typeMap) {
+      if (nodeIds.length < SIBLING_COLLAPSE_THRESHOLD) continue;
+      const first = nodeById.get(nodeIds[0]);
+      const typeName = type.replace(/_/g, " ");
+      const summaryId = `summary__${parentId}__${type}`;
+      summaryNodes.push({
+        id: summaryId,
+        label: `${typeName} \xD7${nodeIds.length}`,
+        type,
+        language: first.language,
+        framework: first.framework,
+        color: first.color,
+        size: Math.min(10, Math.ceil(Math.log10(nodeIds.length + 1)) + 3),
+        collapsedCount: nodeIds.length,
+        data: first.data
+      });
+      summaryEdges.push({
+        id: `${parentId}\u2192${summaryId}`,
+        source: parentId,
+        target: summaryId,
+        type: "parent"
+      });
+    }
+  }
+  return {
+    nodes: [
+      ...nodes.filter((n) => !willCollapse.has(n.id)),
+      ...summaryNodes
+    ],
+    edges: [
+      ...edges.filter((e) => !willCollapse.has(e.source) && !willCollapse.has(e.target)),
+      ...summaryEdges
+    ],
+    captureLevel
+  };
 }
 
 // src/signature.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tracegraph/graph-engine",
-  "version": "0.1.4",
+  "version": "0.2.0",
   "description": "Converts TraceSession to a graph structure; produces baselines, diffs, and findings",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -19,8 +19,8 @@
     "access": "public"
   },
   "dependencies": {
-    "@tracegraph/trace-sanitizer": "0.1.4",
-    "@tracegraph/shared-types": "0.1.4"
+    "@tracegraph/shared-types": "0.2.0",
+    "@tracegraph/trace-sanitizer": "0.2.0"
   },
   "scripts": {
     "build": "tsup src/index.ts --format cjs,esm --dts --external '@tracegraph/*'",