@graphty/algorithms 2.1.2 → 2.2.0

package/dist/src/indexed/leiden.js

@@ -0,0 +1,384 @@
+import { renumberPartition } from "@graphty/graph-format";
+import { withGroups } from "./components.js";
+import { exactEdgeWeights, mulberry32 } from "./label-propagation.js";
+import { modularity } from "./modularity.js";
+/** The randomness of refinement: 0.01 as in the paper and leidenalg. */
+const THETA = 0.01;
+/**
+ * Passes in a row that may fail to improve modularity by more than the threshold before the run
+ * stops. A failed pass is discarded, but the next one draws fresh visit orders and can still find a
+ * move the last one missed: on the shared random fixtures, stopping at the first failure leaves the
+ * mean modularity over 40 seeds below the legacy function's, and three restores it.
+ */
+const PATIENCE = 3;
+/**
+ * Per-arc weights and weighted degrees of a level.
+ * @param s - An undirected snapshot
+ * @param edgeW - Exact weight per logical edge
+ * @returns The level
+ */
+function levelOf(s, edgeW) {
+    const { nodeCount: n, rowPtr, colIdx, arcToEdge } = s;
+    const arcW = new Float64Array(s.arcCount);
+    const deg = new Float64Array(n);
+    for (let u = 0; u < n; u++) {
+        for (let a = rowPtr[u], end = rowPtr[u + 1]; a < end; a++) {
+            const w = edgeW[arcToEdge[a]];
+            arcW[a] = w;
+            // An undirected self-loop is one arc; it adds to the degree twice.
+            deg[u] += colIdx[a] === u ? 2 * w : w;
+        }
+    }
+    return { s, edgeW, arcW, deg };
+}
+/**
+ * Shuffle `0..n-1` with the generator.
+ * @param n - Length
+ * @param rand - Generator
+ * @returns The permutation
+ */
+function shuffled(n, rand) {
+    const order = new Uint32Array(n);
+    for (let i = 0; i < n; i++) {
+        order[i] = i;
+    }
+    for (let i = n - 1; i >= 1; i--) {
+        const j = Math.floor(rand() * (i + 1));
+        const t = order[i];
+        order[i] = order[j];
+        order[j] = t;
+    }
+    return order;
+}
+/**
+ * Leiden's fast local moving: nodes wait in a queue started as a seeded shuffle; a visited node
+ * moves to the neighbouring community (or an empty one) with the largest modularity gain, and when
+ * it moves, its neighbours outside its new community are queued again. Ends when the queue empties.
+ * @param level - The level
+ * @param comm - Community of every node, below the node count; moved in place
+ * @param rand - Generator
+ * @param resolution - Resolution gamma
+ * @param m2 - Twice the total edge weight
+ */
+function moveNodesFast(level, comm, rand, resolution, m2) {
+    const { s, arcW, deg } = level;
+    const { nodeCount: n, rowPtr, colIdx } = s;
+    const tot = new Float64Array(n);
+    const size = new Uint32Array(n);
+    for (let u = 0; u < n; u++) {
+        tot[comm[u]] += deg[u];
+        size[comm[u]]++;
+    }
+    const empty = [];
+    for (let c = n - 1; c >= 0; c--) {
+        if (size[c] === 0) {
+            empty.push(c);
+        }
+    }
+    const acc = new Float64Array(n);
+    const stamp = new Int32Array(n).fill(-1);
+    const touched = new Uint32Array(n);
+    const capacity = n + 1;
+    const queue = new Uint32Array(capacity);
+    queue.set(shuffled(n, rand));
+    const queued = new Uint8Array(n).fill(1);
+    let head = 0;
+    let tail = n;
+    let visit = 0;
+    while (head !== tail) {
+        const u = queue[head];
+        head = head + 1 === capacity ? 0 : head + 1;
+        queued[u] = 0;
+        visit++;
+        const cu = comm[u];
+        const ku = deg[u];
+        tot[cu] -= ku;
+        size[cu]--;
+        let cnt = 0;
+        for (let a = rowPtr[u], end = rowPtr[u + 1]; a < end; a++) {
+            const v = colIdx[a];
+            if (v === u) {
+                continue;
+            }
+            const c = comm[v];
+            if (stamp[c] !== visit) {
+                stamp[c] = visit;
+                acc[c] = 0;
+                touched[cnt++] = c;
+            }
+            acc[c] += arcW[a];
+        }
+        let best = cu;
+        let bestGain = (stamp[cu] === visit ? acc[cu] : 0) - (resolution * ku * tot[cu]) / m2;
+        for (let i = 0; i < cnt; i++) {
+            const c = touched[i];
+            const gain = acc[c] - (resolution * ku * tot[c]) / m2;
+            if (c !== cu && gain > bestGain) {
+                bestGain = gain;
+                best = c;
+            }
+        }
+        // An empty community gains 0; when u's own is empty now, staying already is that choice.
+        if (size[cu] > 0 && bestGain < 0) {
+            best = empty[empty.length - 1];
+        }
+        if (best !== cu && size[best] === 0) {
+            empty.pop();
+        }
+        tot[best] += ku;
+        size[best]++;
+        if (best === cu) {
+            continue;
+        }
+        comm[u] = best;
+        if (size[cu] === 0) {
+            empty.push(cu);
+        }
+        for (let a = rowPtr[u], end = rowPtr[u + 1]; a < end; a++) {
+            const v = colIdx[a];
+            if (queued[v] === 0 && comm[v] !== best) {
+                queued[v] = 1;
+                queue[tail] = v;
+                tail = tail + 1 === capacity ? 0 : tail + 1;
+            }
+        }
+    }
+}
+/**
+ * Leiden's refinement: inside every community, nodes start alone and, visited in a seeded order,
+ * a node still alone and well connected to its community merges into the well-connected
+ * sub-community of that community, drawn at random weighted by its modularity gain. Every
+ * sub-community is connected, which is the guarantee Leiden adds over Louvain.
+ *
+ * The target is drawn at random among the candidates with a non-negative gain, with probability
+ * proportional to exp(gain / theta) where the gain is in modularity units, as the paper does
+ * (theta 0.01, leidenalg's default).
+ * @param level - The level
+ * @param comm - Community of every node
+ * @param rand - Generator
+ * @param resolution - Resolution gamma
+ * @param m2 - Twice the total edge weight
+ * @returns The sub-community of every node, labelled by a member's index (not dense)
+ */
+function refine(level, comm, rand, resolution, m2) {
+    const { s, arcW, deg } = level;
+    const { nodeCount: n, rowPtr, colIdx } = s;
+    const commTot = new Float64Array(n);
+    for (let u = 0; u < n; u++) {
+        commTot[comm[u]] += deg[u];
+    }
+    const ref = new Uint32Array(n);
+    const refTot = new Float64Array(n);
+    const refSize = new Uint32Array(n);
+    // Weight from each sub-community to the rest of its community.
+    const ext = new Float64Array(n);
+    for (let u = 0; u < n; u++) {
+        ref[u] = u;
+        refTot[u] = deg[u];
+        refSize[u] = 1;
+        for (let a = rowPtr[u], end = rowPtr[u + 1]; a < end; a++) {
+            const v = colIdx[a];
+            if (v !== u && comm[v] === comm[u]) {
+                ext[u] += arcW[a];
+            }
+        }
+    }
+    const acc = new Float64Array(n);
+    const stamp = new Int32Array(n).fill(-1);
+    const touched = new Uint32Array(n);
+    const order = shuffled(n, rand);
+    for (let i = 0; i < n; i++) {
+        const v = order[i];
+        const c = comm[v];
+        const kv = deg[v];
+        if (refSize[v] !== 1 || ext[v] < (resolution * kv * (commTot[c] - kv)) / m2) {
+            continue;
+        }
+        let cnt = 0;
+        for (let a = rowPtr[v], end = rowPtr[v + 1]; a < end; a++) {
+            const x = colIdx[a];
+            if (x === v || comm[x] !== c) {
+                continue;
+            }
+            const r = ref[x];
+            if (stamp[r] !== v) {
+                stamp[r] = v;
+                acc[r] = 0;
+                touched[cnt++] = r;
+            }
+            acc[r] += arcW[a];
+        }
+        // Candidates: well-connected sub-communities with a non-negative gain, drawn with probability
+        // exp(gain in modularity units / theta), shifted by the best gain so exp never overflows.
+        const gainOf = (r) => acc[r] - (resolution * kv * refTot[r]) / m2;
+        let maxGain = -Infinity;
+        let kept = 0;
+        for (let j = 0; j < cnt; j++) {
+            const r = touched[j];
+            const gain = gainOf(r);
+            if (gain >= 0 && ext[r] >= (resolution * refTot[r] * (commTot[c] - refTot[r])) / m2) {
+                touched[kept++] = r;
+                maxGain = Math.max(maxGain, gain);
+            }
+        }
+        if (kept === 0) {
+            continue;
+        }
+        const weight = (r) => Math.exp(((gainOf(r) - maxGain) * 2) / m2 / THETA);
+        let total = 0;
+        for (let j = 0; j < kept; j++) {
+            total += weight(touched[j]);
+        }
+        let draw = rand() * total;
+        let best = touched[kept - 1];
+        for (let j = 0; j < kept - 1; j++) {
+            draw -= weight(touched[j]);
+            if (draw < 0) {
+                best = touched[j];
+                break;
+            }
+        }
+        ref[v] = best;
+        refSize[v] = 0;
+        refSize[best]++;
+        refTot[best] += kv;
+        // The edges between v and its new sub-community are now inside it.
+        ext[best] += ext[v] - 2 * acc[best];
+    }
+    return ref;
+}
+/**
+ * One pass of Leiden from a starting partition of the original graph: local moving, refinement,
+ * aggregation by the refined partition with each aggregate node starting in its community, repeated
+ * until local moving leaves every community a single node of its level.
+ * @param base - Level 0, the original graph
+ * @param start - Dense community of every original node
+ * @param rand - Generator
+ * @param resolution - Resolution gamma
+ * @param m2 - Twice the total edge weight
+ * @returns The partition of the original nodes (not dense) and the levels processed
+ */
+function pass(base, start, rand, resolution, m2) {
+    const n0 = base.s.nodeCount;
+    // Which node of the current level holds each original node.
+    const top = new Uint32Array(n0);
+    for (let u = 0; u < n0; u++) {
+        top[u] = u;
+    }
+    let level = base;
+    let comm = start.slice();
+    let levels = 0;
+    for (;;) {
+        moveNodesFast(level, comm, rand, resolution, m2);
+        levels++;
+        const dense = renumberPartition(comm);
+        if (dense.count === level.s.nodeCount) {
+            break;
+        }
+        let blocks = refine(level, dense.labels, rand, resolution, m2);
+        if (renumberPartition(blocks).count === level.s.nodeCount) {
+            // No sub-community merged: aggregate by the communities themselves, as Louvain does, so
+            // the next level is smaller.
+            blocks = dense.labels;
+        }
+        const d = level.s.contract(blocks);
+        const { nodeRemap, edgeRemap } = d;
+        const next = d.snapshot;
+        const edgeW = new Float64Array(next.edgeCount);
+        for (let e = 0; e < level.edgeW.length; e++) {
+            edgeW[edgeRemap === null ? e : edgeRemap[e]] += level.edgeW[e];
+        }
+        const nextComm = new Uint32Array(next.nodeCount);
+        for (let u = 0; u < level.s.nodeCount; u++) {
+            nextComm[nodeRemap === null ? u : nodeRemap[u]] = dense.labels[u];
+        }
+        if (nodeRemap !== null) {
+            for (let u = 0; u < n0; u++) {
+                top[u] = nodeRemap[top[u]];
+            }
+        }
+        comm = renumberPartition(nextComm).labels;
+        level = levelOf(next, edgeW);
+    }
+    const out = new Uint32Array(n0);
+    for (let u = 0; u < n0; u++) {
+        out[u] = comm[top[u]];
+    }
+    return { comm: out, levels };
+}
+/**
+ * Leiden community detection (Traag, Waltman and van Eck, Sci. Rep. 9:5233, 2019) over an
+ * undirected snapshot: fast local moving, refinement into well-connected sub-communities, and
+ * aggregation by `contract()` once per level. Whole passes repeat from the previous answer, keeping
+ * a pass only when it raises modularity, until three passes in a row improve it by no more than
+ * `threshold` or `maxIterations` passes have run; every pass starts by moving single original
+ * nodes, so the answer is also a local optimum of those moves.
+ *
+ * Modularity follows graph-format's `weightedDegree()` convention (a self-loop counts twice), which
+ * on a graph without self-loops equals the legacy `leiden`'s. Weights are the exact f64 ones when
+ * the snapshot keeps them. The partition is not the legacy function's move for move: the legacy
+ * `leiden` splits disconnected communities rather than refining them, and visits nodes in another
+ * order; what this guarantees is connected communities. Both are randomised heuristics, so for one
+ * seed either can come out ahead; on the shared fixtures this one's modularity is at least the
+ * legacy's at the default seed and on average over seeds.
+ * @param s - An undirected snapshot
+ * @param options - Resolution, seed, pass cap and stopping threshold
+ * @returns The partition, its modularity, and the levels processed
+ * @throws Error on a directed snapshot; RangeError for a negative, NaN or infinite weight or a
+ *   fractional seed
+ * @public
+ */
+export function leiden(s, options = {}) {
+    if (s.directed) {
+        throw new Error("Leiden requires an undirected graph. Pass s.toUndirected().snapshot.");
+    }
+    const resolution = options.resolution ?? 1;
+    const randomSeed = options.randomSeed ?? 42;
+    const maxIterations = options.maxIterations ?? 100;
+    const threshold = options.threshold ?? 1e-7;
+    if (!Number.isInteger(randomSeed)) {
+        throw new RangeError(`randomSeed must be a finite integer, got ${randomSeed}`);
+    }
+    const n = s.nodeCount;
+    const given = exactEdgeWeights(s) ?? s.edgeList().weights;
+    const edgeW = given === null ? new Float64Array(s.edgeCount).fill(1) : Float64Array.from(given);
+    for (let e = 0; e < edgeW.length; e++) {
+        const w = edgeW[e];
+        if (!(w >= 0) || w === Infinity) {
+            throw new RangeError(`edge ${e} has weight ${w}; Leiden needs finite, non-negative weights`);
+        }
+    }
+    const base = levelOf(s, edgeW);
+    let m2 = 0;
+    for (let u = 0; u < n; u++) {
+        m2 += base.deg[u];
+    }
+    let labels = new Uint32Array(n);
+    for (let u = 0; u < n; u++) {
+        labels[u] = u;
+    }
+    if (m2 === 0) {
+        return { ...withGroups(labels, n), modularity: 0, iterations: 0 };
+    }
+    const quality = (partition) => modularity(s, partition, { resolution, weights: base.arcW });
+    const rand = mulberry32(randomSeed);
+    let q = quality(labels);
+    let iterations = 0;
+    let idle = 0;
+    for (let i = 0; i < maxIterations && idle < PATIENCE; i++) {
+        const { comm, levels } = pass(base, labels, rand, resolution, m2);
+        iterations += levels;
+        const candidate = renumberPartition(comm).labels;
+        const qNext = quality(candidate);
+        if (qNext > q) {
+            idle = qNext - q <= threshold ? idle + 1 : 0;
+            labels = candidate;
+            q = qNext;
+        }
+        else {
+            idle++;
+        }
+    }
+    return { ...withGroups(labels, renumberPartition(labels).count), modularity: q, iterations };
+}
+//# sourceMappingURL=leiden.js.map

package/dist/src/indexed/leiden.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"leiden.js","sourceRoot":"","sources":["../../../src/indexed/leiden.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgC,iBAAiB,EAAY,MAAM,uBAAuB,CAAC;AAElG,OAAO,EAAoB,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACtE,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AA8B7C,wEAAwE;AACxE,MAAM,KAAK,GAAG,IAAI,CAAC;AAEnB;;;;;GAKG;AACH,MAAM,QAAQ,GAAG,CAAC,CAAC;AAanB;;;;;GAKG;AACH,SAAS,OAAO,CAAC,CAAgB,EAAE,KAAU;IACzC,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,CAAC,CAAC;IACtD,MAAM,IAAI,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;IAC1C,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACxD,MAAM,CAAC,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9B,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACZ,mEAAmE;YACnE,GAAG,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1C,CAAC;IACL,CAAC;IACD,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;AACnC,CAAC;AAED;;;;;GAKG;AACH,SAAS,QAAQ,CAAC,CAAS,EAAE,IAAkB;IAC3C,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;IACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IACjB,CAAC;IACD,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACnB,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACpB,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IACjB,CAAC;IACD,OAAO,KAAK,CAAC;AACjB,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,aAAa,CAAC,KAAY,EAAE,IAAS,EAAE,IAAkB,EAAE,UAAkB,EAAE,EAAU;IAC9F,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC;IAC/B,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;IAC3C,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IAChC,MAAM,IAAI,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;IAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;QACvB,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACpB,CAAC;IACD,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;YAChB,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACL,CAAC;IACD,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IAChC,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACzC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;IACnC,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,CAAC;IACvB,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,QAAQ,CAAC,CAAC;IACxC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACzC,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,OAAO,IAAI,KAAK,IAAI,EAAE,CAAC;QACnB,MAAM,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;QACtB,IAAI,GAAG,IAAI,GAAG,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC;QAC5C,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACd,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACnB,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QAClB,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;QACd,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;QACX,IAAI,GAAG,GAAG,CAAC,CAAC;QACZ,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACxD,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YACpB,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACV,SAAS;YACb,CAAC;YACD,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YAClB,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,KAAK,EAAE,CAAC;gBACrB,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;gBACjB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBACX,OAAO,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC;YACvB,CAAC;YACD,GAAG,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC;QACtB,CAAC;QACD,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,IAAI,QAAQ,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,UAAU,GAAG,EAAE,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC;QACtF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACrB,MAAM,IAAI,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,UAAU,GAAG,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;YACtD,IAAI,CAAC,KAAK,EAAE,IAAI,IAAI,GAAG,QAAQ,EAAE,CAAC;gBAC9B,QAAQ,GAAG,IAAI,CAAC;gBAChB,IAAI,GAAG,CAAC,CAAC;YACb,CAAC;QACL,CAAC;QACD,yFAAyF;QACzF,IAAI,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;YAC/B,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACnC,CAAC;QACD,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YAClC,KAAK,CAAC,GAAG,EAAE,CAAC;QAChB,CAAC;QACD,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAChB,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACb,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;YACd,SAAS;QACb,CAAC;QACD,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;QACf,IAAI,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;YACjB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnB,CAAC;QACD,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACxD,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YACpB,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;gBACtC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBACd,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAChB,IAAI,GAAG,IAAI,GAAG,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC;YAChD,CAAC;QACL,CAAC;IACL,CAAC;AACL,CAAC;AAED;;;;;;;;;;;;;;;GAeG;AACH,SAAS,MAAM,CAAC,KAAY,EAAE,IAAS,EAAE,IAAkB,EAAE,UAAkB,EAAE,EAAU;IACvF,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC;IAC/B,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;IAC3C,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IACpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;IAC/B,CAAC;IACD,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;IAC/B,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IACnC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;IACnC,+DAA+D;IAC/D,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACX,MAAM,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QACnB,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACf,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACxD,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YACpB,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;gBACjC,GAAG,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC;YACtB,CAAC;QACL,CAAC;IACL,CAAC;IACD,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IAChC,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACzC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;IACnC,MAAM,KAAK,GAAG,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACnB,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QAClB,IAAI,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,UAAU,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;YAC1E,SAAS;QACb,CAAC;QACD,IAAI,GAAG,GAAG,CAAC,CAAC;QACZ,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACxD,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YACpB,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC3B,SAAS;YACb,CAAC;YACD,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;YACjB,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;gBACjB,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBACb,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBACX,OAAO,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC;YACvB,CAAC;YACD,GAAG,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC;QACtB,CAAC;QACD,8FAA8F;QAC9F,0FAA0F;QAC1F,MAAM,MAAM,GAAG,CAAC,CAAS,EAAU,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,UAAU,GAAG,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;QAClF,IAAI,OAAO,GAAG,CAAC,QAAQ,CAAC;QACxB,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACrB,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YACvB,IAAI,IAAI,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;gBAClF,OAAO,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC;gBACpB,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YACtC,CAAC;QACL,CAAC;QACD,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;YACb,SAAS;QACb,CAAC;QACD,MAAM,MAAM,GAAG,CAAC,CAAS,EAAU,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;QACzF,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,KAAK,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,IAAI,GAAG,IAAI,EAAE,GAAG,KAAK,CAAC;QAC1B,IAAI,IAAI,GAAG,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;QAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3B,IAAI,IAAI,GAAG,CAAC,EAAE,CAAC;gBACX,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;gBAClB,MAAM;YACV,CAAC;QACL,CAAC;QACD,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;QACd,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACf,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAChB,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACnB,mEAAmE;QACnE,GAAG,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC;IACD,OAAO,GAAG,CAAC;AACf,CAAC;AAED;;;;;;;;;;GAUG;AACH,SAAS,IAAI,CACT,IAAW,EACX,KAAU,EACV,IAAkB,EAClB,UAAkB,EAClB,EAAU;IAEV,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IAC5B,4DAA4D;IAC5D,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,EAAE,CAAC,CAAC;IAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IACf,CAAC;IACD,IAAI,KAAK,GAAG,IAAI,CAAC;IACjB,IAAI,IAAI,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;IACzB,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,SAAS,CAAC;QACN,aAAa,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;QACjD,MAAM,EAAE,CAAC;QACT,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,KAAK,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YACpC,MAAM;QACV,CAAC;QACD,IAAI,MAAM,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;QAC/D,IAAI,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YACxD,wFAAwF;YACxF,6BAA6B;YAC7B,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;QAC1B,CAAC;QACD,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACnC,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,CAAC,CAAC;QACnC,MAAM,IAAI,GAAG,CAAC,CAAC,QAAQ,CAAC;QACxB,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,KAAK,CAAC,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACnE,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,QAAQ,CAAC,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;YACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC1B,GAAG,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/B,CAAC;QACL,CAAC;QACD,IAAI,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;QAC1C,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACjC,CAAC;IACD,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,EAAE,CAAC,CAAC;IAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1B,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC;AACjC,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,UAAU,MAAM,CAAC,CAAgB,EAAE,UAAyB,EAAE;IAChE,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,sEAAsE,CAAC,CAAC;IAC5F,CAAC;IACD,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,CAAC,CAAC;IAC3C,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,EAAE,CAAC;IAC5C,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,GAAG,CAAC;IACnD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC;IAC5C,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC;QAChC,MAAM,IAAI,UAAU,CAAC,4CAA4C,UAAU,EAAE,CAAC,CAAC;IACnF,CAAC;IACD,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC;IACtB,MAAM,KAAK,GAAG,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC;IAC1D,MAAM,KAAK,GAAG,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChG,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACnB,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,QAAQ,EAAE,CAAC;YAC9B,MAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,eAAe,CAAC,6CAA6C,CAAC,CAAC;QACjG,CAAC;IACL,CAAC;IACD,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAC/B,IAAI,EAAE,GAAG,CAAC,CAAC;IACX,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC;IACD,IAAI,MAAM,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;IAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IAClB,CAAC;IACD,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC;QACX,OAAO,EAAE,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;IACtE,CAAC;IACD,MAAM,OAAO,GAAG,CAAC,SAAc,EAAU,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACzG,MAAM,IAAI,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC;IACpC,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IACxB,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,IAAI,IAAI,GAAG,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;QACxD,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;QAClE,UAAU,IAAI,MAAM,CAAC;QACrB,MAAM,SAAS,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;QACjD,MAAM,KAAK,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACZ,IAAI,GAAG,KAAK,GAAG,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,GAAG,SAAS,CAAC;YACnB,CAAC,GAAG,KAAK,CAAC;QACd,CAAC;aAAM,CAAC;YACJ,IAAI,EAAE,CAAC;QACX,CAAC;IACL,CAAC;IACD,OAAO,EAAE,GAAG,UAAU,CAAC,MAAM,EAAE,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC;AACjG,CAAC"}

package/dist/src/indexed/link-prediction.d.ts

@@ -0,0 +1,147 @@
+/**
+ * Link prediction over a snapshot: common-neighbour and Adamic-Adar scores for one pair, a list of
+ * pairs, every pair, or one node's candidates, and the ranking metrics of a held-out evaluation.
+ *
+ * Every score is a merge of two sorted rows with simple-graph semantics: a neighbour reached by
+ * parallel arcs counts once, and a self-loop makes a node its own neighbour. With
+ * `{ directed: true }` a pair (u, v) intersects out(u) with in(v) -- the nodes on a path u -> z -> v
+ * -- and otherwise the two out rows (the two rows of an undirected snapshot).
+ *
+ * A node index at or beyond `nodeCount`, `INVALID_INDEX` included, is an absent node: its pairs
+ * score 0 and it has no candidates.
+ * @module
+ */
+import type { F64, GraphSnapshot, U32 } from "@graphty/graph-format";
+import { type CommonNeighborsOptions } from "./common-neighbors.js";
+/** Options of the link prediction functions. @public */
+export interface LinkPredictionOptions extends CommonNeighborsOptions {
+    /** Also score pairs already joined by an arc u -> v. Default false. */
+    readonly includeExisting?: boolean | undefined;
+    /**
+     * Keep only the first `topK` ranked pairs. For the whole-graph predictions a value that is not
+     * positive keeps them all; for a node's candidates the default is 10 and the value is a slice
+     * end, so 0 keeps none and -2 drops the last two.
+     */
+    readonly topK?: number | undefined;
+}
+/** Options of the per-node candidate functions. @public */
+export interface CandidateOptions extends LinkPredictionOptions {
+    /** The node indices to consider, in this order; default every node in index order. */
+    readonly candidates?: ArrayLike<number> | undefined;
+}
+/** Node pairs as two parallel index lists: pair k is (sources[k], targets[k]). @public */
+export interface NodePairs {
+    readonly sources: ArrayLike<number>;
+    readonly targets: ArrayLike<number>;
+}
+/** Ranked pairs, highest score first, ties in enumeration order. @public */
+export interface LinkPredictionResult extends NodePairs {
+    readonly sources: U32;
+    readonly targets: U32;
+    readonly scores: F64;
+}
+/** The best-F1 threshold's precision and recall, that F1, and the ranking AUC. @public */
+export interface LinkPredictionMetrics {
+    readonly precision: number;
+    readonly recall: number;
+    readonly f1Score: number;
+    readonly auc: number;
+}
+/**
+ * Common-neighbour scores of every pair not already joined, ranked.
+ * @param s - The snapshot
+ * @param o - Options
+ * @returns The pairs with a positive score, highest first
+ * @public
+ */
+export declare function commonNeighborsPrediction(s: GraphSnapshot, o?: LinkPredictionOptions): LinkPredictionResult;
+/**
+ * Common-neighbour scores of the given pairs, in order.
+ * @param s - The snapshot
+ * @param pairs - The pairs
+ * @param o - Options
+ * @returns One score per pair
+ * @public
+ */
+export declare function commonNeighborsForPairs(s: GraphSnapshot, pairs: NodePairs, o?: CommonNeighborsOptions): F64;
+/**
+ * One node's best common-neighbour candidates, ranked.
+ * @param s - The snapshot
+ * @param u - The node index
+ * @param o - Options
+ * @returns (u, v) pairs with a positive score, highest first
+ * @public
+ */
+export declare function getTopCandidatesForNode(s: GraphSnapshot, u: number, o?: CandidateOptions): LinkPredictionResult;
+/**
+ * Ranking metrics of the common-neighbour score on held-out edges against non-edges.
+ * @param s - The training snapshot
+ * @param edges - The held-out edges
+ * @param nonEdges - The non-edges
+ * @param o - Options
+ * @returns The metrics
+ * @public
+ */
+export declare function evaluateCommonNeighbors(s: GraphSnapshot, edges: NodePairs, nonEdges: NodePairs, o?: CommonNeighborsOptions): LinkPredictionMetrics;
+/**
+ * The Adamic-Adar index of two nodes: the sum over distinct common neighbours z of 1 / ln(degree(z)).
+ * @param s - The snapshot
+ * @param u - A node index
+ * @param v - A node index
+ * @param o - Options
+ * @returns The score
+ * @public
+ */
+export declare function adamicAdarScore(s: GraphSnapshot, u: number, v: number, o?: CommonNeighborsOptions): number;
+/**
+ * Adamic-Adar scores of every pair not already joined, ranked.
+ * @param s - The snapshot
+ * @param o - Options
+ * @returns The pairs with a positive score, highest first
+ * @public
+ */
+export declare function adamicAdarPrediction(s: GraphSnapshot, o?: LinkPredictionOptions): LinkPredictionResult;
+/**
+ * Adamic-Adar scores of the given pairs, in order.
+ * @param s - The snapshot
+ * @param pairs - The pairs
+ * @param o - Options
+ * @returns One score per pair
+ * @public
+ */
+export declare function adamicAdarForPairs(s: GraphSnapshot, pairs: NodePairs, o?: CommonNeighborsOptions): F64;
+/**
+ * One node's best Adamic-Adar candidates, ranked.
+ * @param s - The snapshot
+ * @param u - The node index
+ * @param o - Options
+ * @returns (u, v) pairs with a positive score, highest first
+ * @public
+ */
+export declare function getTopAdamicAdarCandidatesForNode(s: GraphSnapshot, u: number, o?: CandidateOptions): LinkPredictionResult;
+/**
+ * Ranking metrics of the Adamic-Adar score on held-out edges against non-edges.
+ * @param s - The training snapshot
+ * @param edges - The held-out edges
+ * @param nonEdges - The non-edges
+ * @param o - Options
+ * @returns The metrics
+ * @public
+ */
+export declare function evaluateAdamicAdar(s: GraphSnapshot, edges: NodePairs, nonEdges: NodePairs, o?: CommonNeighborsOptions): LinkPredictionMetrics;
+/**
+ * Both evaluations on the same held-out sets. The common-neighbour side always intersects the two
+ * OUT rows, whatever `directed` says, as the legacy function of this name does; the Adamic-Adar side
+ * honours `directed`.
+ * @param s - The training snapshot
+ * @param edges - The held-out edges
+ * @param nonEdges - The non-edges
+ * @param o - Options
+ * @returns The two sets of metrics
+ * @public
+ */
+export declare function compareAdamicAdarWithCommonNeighbors(s: GraphSnapshot, edges: NodePairs, nonEdges: NodePairs, o?: CommonNeighborsOptions): {
+    adamicAdar: LinkPredictionMetrics;
+    commonNeighbors: LinkPredictionMetrics;
+};
+//# sourceMappingURL=link-prediction.d.ts.map

package/dist/src/indexed/link-prediction.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"link-prediction.d.ts","sourceRoot":"","sources":["../../../src/indexed/link-prediction.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,KAAK,EAAiB,GAAG,EAAE,aAAa,EAAE,GAAG,EAAE,MAAM,uBAAuB,CAAC;AAEpF,OAAO,EAAE,KAAK,sBAAsB,EAAkB,MAAM,uBAAuB,CAAC;AAEpF,wDAAwD;AACxD,MAAM,WAAW,qBAAsB,SAAQ,sBAAsB;IACjE,uEAAuE;IACvE,QAAQ,CAAC,eAAe,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC/C;;;;OAIG;IACH,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACtC;AAED,2DAA2D;AAC3D,MAAM,WAAW,gBAAiB,SAAQ,qBAAqB;IAC3D,sFAAsF;IACtF,QAAQ,CAAC,UAAU,CAAC,EAAE,SAAS,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC;CACvD;AAED,0FAA0F;AAC1F,MAAM,WAAW,SAAS;IACtB,QAAQ,CAAC,OAAO,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IACpC,QAAQ,CAAC,OAAO,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;CACvC;AAED,4EAA4E;AAC5E,MAAM,WAAW,oBAAqB,SAAQ,SAAS;IACnD,QAAQ,CAAC,OAAO,EAAE,GAAG,CAAC;IACtB,QAAQ,CAAC,OAAO,EAAE,GAAG,CAAC;IACtB,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC;CACxB;AAED,0FAA0F;AAC1F,MAAM,WAAW,qBAAqB;IAClC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;CACxB;AAqMD;;;;;;GAMG;AACH,wBAAgB,yBAAyB,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC,GAAE,qBAA0B,GAAG,oBAAoB,CAE/G;AAED;;;;;;;GAOG;AACH,wBAAgB,uBAAuB,CAAC,CAAC,EAAE,aAAa,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,GAAE,sBAA2B,GAAG,GAAG,CAE/G;AAED;;;;;;;GAOG;AACH,wBAAgB,uBAAuB,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,GAAE,gBAAqB,GAAG,oBAAoB,CAEnH;AAED;;;;;;;;GAQG;AACH,wBAAgB,uBAAuB,CACnC,CAAC,EAAE,aAAa,EAChB,KAAK,EAAE,SAAS,EAChB,QAAQ,EAAE,SAAS,EACnB,CAAC,GAAE,sBAA2B,GAC/B,qBAAqB,CAGvB;AAED;;;;;;;;GAQG;AACH,wBAAgB,eAAe,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,GAAE,sBAA2B,GAAG,MAAM,CAE9G;AAED;;;;;;GAMG;AACH,wBAAgB,oBAAoB,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC,GAAE,qBAA0B,GAAG,oBAAoB,CAE1G;AAED;;;;;;;GAOG;AACH,wBAAgB,kBAAkB,CAAC,CAAC,EAAE,aAAa,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,GAAE,sBAA2B,GAAG,GAAG,CAE1G;AAED;;;;;;;GAOG;AACH,wBAAgB,iCAAiC,CAC7C,CAAC,EAAE,aAAa,EAChB,CAAC,EAAE,MAAM,EACT,CAAC,GAAE,gBAAqB,GACzB,oBAAoB,CAEtB;AAED;;;;;;;;GAQG;AACH,wBAAgB,kBAAkB,CAC9B,CAAC,EAAE,aAAa,EAChB,KAAK,EAAE,SAAS,EAChB,QAAQ,EAAE,SAAS,EACnB,CAAC,GAAE,sBAA2B,GAC/B,qBAAqB,CAGvB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,oCAAoC,CAChD,CAAC,EAAE,aAAa,EAChB,KAAK,EAAE,SAAS,EAChB,QAAQ,EAAE,SAAS,EACnB,CAAC,GAAE,sBAA2B,GAC/B;IAAE,UAAU,EAAE,qBAAqB,CAAC;IAAC,eAAe,EAAE,qBAAqB,CAAA;CAAE,CAK/E"}