graphwise 1.4.0 → 1.4.1

package/dist/index/index.js

@@ -25,6 +25,8 @@ function base(graph, seeds, config) {
 	const { maxNodes = 0, maxIterations = 0, maxPaths = 0, priority = degreePriority, debug = false } = config ?? {};
 	if (seeds.length === 0) return emptyResult("base", startTime);
 	const numFrontiers = seeds.length;
+	const allVisited = /* @__PURE__ */ new Set();
+	const combinedVisited = /* @__PURE__ */ new Map();
 	const visitedByFrontier = [];
 	const predecessors = [];
 	const queues = [];
@@ -36,15 +38,16 @@ function base(graph, seeds, config) {
 		if (seed === void 0) continue;
 		const seedNode = seed.id;
 		predecessors[i]?.set(seedNode, null);
-		const seedPriority = priority(seedNode, createPriorityContext(graph, seedNode, i, visitedByFrontier, [], 0));
+		combinedVisited.set(seedNode, i);
+		allVisited.add(seedNode);
+		const seedPriority = priority(seedNode, createPriorityContext(graph, seedNode, i, combinedVisited, allVisited, [], 0));
 		queues[i]?.push({
 			nodeId: seedNode,
 			frontierIndex: i,
 			predecessor: null
 		}, seedPriority);
 	}
-	const allVisited = /* @__PURE__ */ new Set();
-	const sampledEdges = /* @__PURE__ */ new Set();
+	const sampledEdgeMap = /* @__PURE__ */ new Map();
 	const discoveredPaths = [];
 	let iterations = 0;
 	let edgesTraversed = 0;
@@ -89,6 +92,7 @@ function base(graph, seeds, config) {
 		const frontierVisited = visitedByFrontier[activeFrontier];
 		if (frontierVisited === void 0 || frontierVisited.has(nodeId)) continue;
 		frontierVisited.set(nodeId, activeFrontier);
+		combinedVisited.set(nodeId, activeFrontier);
 		if (predecessor !== null) {
 			const predMap = predecessors[activeFrontier];
 			if (predMap !== void 0) predMap.set(nodeId, predecessor);
@@ -110,11 +114,16 @@ function base(graph, seeds, config) {
 		const neighbours = graph.neighbours(nodeId);
 		for (const neighbour of neighbours) {
 			edgesTraversed++;
-			const edgeKey = nodeId < neighbour ? `${nodeId}::${neighbour}` : `${neighbour}::${nodeId}`;
-			sampledEdges.add(edgeKey);
+			const [s, t] = nodeId < neighbour ? [nodeId, neighbour] : [neighbour, nodeId];
+			let targets = sampledEdgeMap.get(s);
+			if (targets === void 0) {
+				targets = /* @__PURE__ */ new Set();
+				sampledEdgeMap.set(s, targets);
+			}
+			targets.add(t);
 			const frontierVisited = visitedByFrontier[activeFrontier];
 			if (frontierVisited === void 0 || frontierVisited.has(neighbour)) continue;
-			const neighbourPriority = priority(neighbour, createPriorityContext(graph, neighbour, activeFrontier, visitedByFrontier, discoveredPaths, iterations + 1));
+			const neighbourPriority = priority(neighbour, createPriorityContext(graph, neighbour, activeFrontier, combinedVisited, allVisited, discoveredPaths, iterations + 1));
 			queue.push({
 				nodeId: neighbour,
 				frontierIndex: activeFrontier,
@@ -126,14 +135,7 @@ function base(graph, seeds, config) {
 	const endTime = performance.now();
 	const visitedPerFrontier = visitedByFrontier.map((m) => new Set(m.keys()));
 	const edgeTuples = /* @__PURE__ */ new Set();
-	for (const edgeKey of sampledEdges) {
-		const parts = edgeKey.split("::");
-		if (parts.length === 2) {
-			const source = parts[0];
-			const target = parts[1];
-			if (source !== void 0 && target !== void 0) edgeTuples.add([source, target]);
-		}
-	}
+	for (const [source, targets] of sampledEdgeMap) for (const target of targets) edgeTuples.add([source, target]);
 	return {
 		paths: discoveredPaths,
 		sampledNodes: allVisited,
@@ -153,10 +155,7 @@ function base(graph, seeds, config) {
 /**
 * Create priority context for a node.
 */
-function createPriorityContext(graph, nodeId, frontierIndex, visitedByFrontier, discoveredPaths, iteration) {
-	const combinedVisited = /* @__PURE__ */ new Map();
-	for (const frontierMap of visitedByFrontier) for (const [id, idx] of frontierMap) combinedVisited.set(id, idx);
-	const allVisited = new Set(combinedVisited.keys());
+function createPriorityContext(graph, nodeId, frontierIndex, combinedVisited, allVisited, discoveredPaths, iteration) {
 	return {
 		graph,
 		degree: graph.degree(nodeId),
@@ -340,10 +339,12 @@ function hae(graph, seeds, config) {
 * visited by OTHER frontiers (not the current frontier).
 */
 function pipePriority(nodeId, context) {
-	const graph = context.graph;
-	const neighbours = new Set(graph.neighbours(nodeId));
+	const neighbours = context.graph.neighbours(nodeId);
 	let pathPotential = 0;
-	for (const [visitedId, frontierIdx] of context.visitedByFrontier) if (frontierIdx !== context.frontierIndex && neighbours.has(visitedId)) pathPotential++;
+	for (const neighbour of neighbours) {
+		const visitedBy = context.visitedByFrontier.get(neighbour);
+		if (visitedBy !== void 0 && visitedBy !== context.frontierIndex) pathPotential++;
+	}
 	return context.degree / (1 + pathPotential);
 }
 /**
@@ -386,10 +387,7 @@ function sage(graph, seeds, config) {
 	*/
 	function sagePriority(nodeId, context) {
 		const pathCount = context.discoveredPaths.length;
-		if (pathCount > 0 && !inPhase2) {
-			inPhase2 = true;
-			for (const path of context.discoveredPaths) for (const node of path.nodes) salienceCounts.set(node, (salienceCounts.get(node) ?? 0) + 1);
-		}
+		if (pathCount > 0 && !inPhase2) inPhase2 = true;
 		if (pathCount > lastPathCount) {
 			for (let i = lastPathCount; i < pathCount; i++) {
 				const path = context.discoveredPaths[i];
@@ -439,6 +437,23 @@ function jaccard(graph, source, target, config) {
 */
 function reach(graph, seeds, config) {
 	let inPhase2 = false;
+	const jaccardCache = /* @__PURE__ */ new Map();
+	/**
+	* Compute Jaccard similarity with caching.
+	*
+	* Exploits symmetry of Jaccard (J(A,B) = J(B,A)) to reduce
+	* duplicate computations when the same pair appears in multiple
+	* discovered paths. Key format ensures consistent ordering.
+	*/
+	function cachedJaccard(source, target) {
+		const key = source < target ? `${source}::${target}` : `${target}::${source}`;
+		let score = jaccardCache.get(key);
+		if (score === void 0) {
+			score = jaccard(graph, source, target);
+			jaccardCache.set(key, score);
+		}
+		return score;
+	}
 	/**
 	* REACH priority function with MI estimation.
 	*/
@@ -450,8 +465,8 @@ function reach(graph, seeds, config) {
 		for (const path of context.discoveredPaths) {
 			const fromNodeId = path.fromSeed.id;
 			const toNodeId = path.toSeed.id;
-			totalMI += jaccard(graph, nodeId, fromNodeId);
-			totalMI += jaccard(graph, nodeId, toNodeId);
+			totalMI += cachedJaccard(nodeId, fromNodeId);
+			totalMI += cachedJaccard(nodeId, toNodeId);
 			endpointCount += 2;
 		}
 		const miHat = endpointCount > 0 ? totalMI / endpointCount : 0;
@@ -499,9 +514,12 @@ function maze(graph, seeds, config) {
 			}
 			lastPathCount = pathCount;
 		}
-		const nodeNeighbours = new Set(graph.neighbours(nodeId));
+		const nodeNeighbours = graph.neighbours(nodeId);
 		let pathPotential = 0;
-		for (const [visitedId, frontierIdx] of context.visitedByFrontier) if (frontierIdx !== context.frontierIndex && nodeNeighbours.has(visitedId)) pathPotential++;
+		for (const neighbour of nodeNeighbours) {
+			const visitedBy = context.visitedByFrontier.get(neighbour);
+			if (visitedBy !== void 0 && visitedBy !== context.frontierIndex) pathPotential++;
+		}
 		if (!inPhase2) return context.degree / (1 + pathPotential);
 		const salience = salienceCounts.get(nodeId) ?? 0;
 		return context.degree / (1 + pathPotential) * (1 / (1 + SALIENCE_WEIGHT * salience));