@cogitator-ai/core 0.3.0 → 0.5.0

package/dist/causal/inference/adjustment.js

@@ -0,0 +1,188 @@
+import { dSeparation } from './d-separation';
+export function findBackdoorAdjustment(graph, treatment, outcome) {
+    if (!graph.hasNode(treatment) || !graph.hasNode(outcome)) {
+        return null;
+    }
+    const treatmentDescendants = new Set(graph.getDescendants(treatment).map((n) => n.id));
+    treatmentDescendants.add(treatment);
+    const allNodes = graph.getNodes().map((n) => n.id);
+    const candidates = allNodes.filter((n) => n !== treatment && n !== outcome && !treatmentDescendants.has(n));
+    const minimalSets = [];
+    for (let size = 0; size <= candidates.length; size++) {
+        for (const subset of combinations(candidates, size)) {
+            if (isValidBackdoorSet(graph, treatment, outcome, subset)) {
+                minimalSets.push(subset);
+                if (size === 0 || minimalSets.length > 0) {
+                    const minSize = minimalSets[0].length;
+                    if (subset.length > minSize)
+                        break;
+                }
+            }
+        }
+        if (minimalSets.length > 0)
+            break;
+    }
+    if (minimalSets.length === 0) {
+        return null;
+    }
+    const bestSet = minimalSets[0];
+    return {
+        variables: bestSet,
+        type: 'backdoor',
+        formula: generateBackdoorFormula(treatment, outcome, bestSet),
+        isMinimal: true,
+        isValid: true,
+    };
+}
+function isValidBackdoorSet(graph, treatment, outcome, adjustmentSet) {
+    const treatmentDescendants = new Set(graph.getDescendants(treatment).map((n) => n.id));
+    for (const node of adjustmentSet) {
+        if (treatmentDescendants.has(node)) {
+            return false;
+        }
+    }
+    const modifiedGraph = createGraphWithoutOutgoingEdges(graph, treatment);
+    const result = dSeparation(modifiedGraph, treatment, outcome, adjustmentSet);
+    return result.separated;
+}
+function createGraphWithoutOutgoingEdges(graph, nodeId) {
+    const cloned = graph.clone();
+    const children = graph.getChildren(nodeId);
+    for (const child of children) {
+        const edge = cloned.getEdgeBetween(nodeId, child.id);
+        if (edge) {
+            cloned.removeEdge(edge.id);
+        }
+    }
+    return cloned;
+}
+export function findFrontdoorAdjustment(graph, treatment, outcome) {
+    if (!graph.hasNode(treatment) || !graph.hasNode(outcome)) {
+        return null;
+    }
+    const paths = graph.findPaths(treatment, outcome);
+    if (paths.length === 0) {
+        return null;
+    }
+    const allNodes = graph.getNodes().map((n) => n.id);
+    const candidates = allNodes.filter((n) => n !== treatment && n !== outcome);
+    for (let size = 1; size <= candidates.length; size++) {
+        for (const subset of combinations(candidates, size)) {
+            if (isValidFrontdoorSet(graph, treatment, outcome, subset)) {
+                return {
+                    variables: subset,
+                    type: 'frontdoor',
+                    formula: generateFrontdoorFormula(treatment, outcome, subset),
+                    isMinimal: true,
+                    isValid: true,
+                };
+            }
+        }
+    }
+    return null;
+}
+function isValidFrontdoorSet(graph, treatment, outcome, mediators) {
+    const mediatorSet = new Set(mediators);
+    const paths = graph.findPaths(treatment, outcome);
+    for (const path of paths) {
+        let intercepted = false;
+        for (let i = 1; i < path.nodes.length - 1; i++) {
+            if (mediatorSet.has(path.nodes[i])) {
+                intercepted = true;
+                break;
+            }
+        }
+        if (!intercepted)
+            return false;
+    }
+    for (const m of mediators) {
+        const treatmentParents = graph.getParents(treatment);
+        for (const parent of treatmentParents) {
+            if (graph.getEdgeBetween(parent.id, m) || isAncestor(graph, parent.id, m)) {
+                return false;
+            }
+        }
+    }
+    for (const m of mediators) {
+        const confounders = findConfounders(graph, m, outcome);
+        const backdoorSet = [treatment, ...confounders.filter((c) => c !== treatment)];
+        const separated = dSeparation(createGraphWithoutOutgoingEdges(graph, m), m, outcome, backdoorSet);
+        if (!separated.separated) {
+            return false;
+        }
+    }
+    return true;
+}
+function isAncestor(graph, potential, of) {
+    const ancestors = graph.getAncestors(of);
+    return ancestors.some((a) => a.id === potential);
+}
+function findConfounders(graph, node1, node2) {
+    const ancestors1 = new Set(graph.getAncestors(node1).map((n) => n.id));
+    const ancestors2 = new Set(graph.getAncestors(node2).map((n) => n.id));
+    const confounders = [];
+    for (const a of ancestors1) {
+        if (ancestors2.has(a)) {
+            confounders.push(a);
+        }
+    }
+    return confounders;
+}
+function generateBackdoorFormula(treatment, outcome, adjustmentSet) {
+    if (adjustmentSet.length === 0) {
+        return `P(${outcome}|do(${treatment})) = Σ P(${outcome}|${treatment})`;
+    }
+    const adjustVars = adjustmentSet.join(', ');
+    return `P(${outcome}|do(${treatment})) = Σ_{${adjustVars}} P(${outcome}|${treatment}, ${adjustVars}) P(${adjustVars})`;
+}
+function generateFrontdoorFormula(treatment, outcome, mediators) {
+    const m = mediators.join(', ');
+    return `P(${outcome}|do(${treatment})) = Σ_{${m}} P(${m}|${treatment}) Σ_{${treatment}'} P(${outcome}|${m}, ${treatment}') P(${treatment}')`;
+}
+export function findAllAdjustmentSets(graph, treatment, outcome, maxSize = 5) {
+    const results = [];
+    const backdoor = findBackdoorAdjustment(graph, treatment, outcome);
+    if (backdoor) {
+        results.push(backdoor);
+    }
+    const frontdoor = findFrontdoorAdjustment(graph, treatment, outcome);
+    if (frontdoor) {
+        results.push(frontdoor);
+    }
+    const treatmentDescendants = new Set(graph.getDescendants(treatment).map((n) => n.id));
+    const allNodes = graph.getNodes().map((n) => n.id);
+    const candidates = allNodes.filter((n) => n !== treatment && n !== outcome && !treatmentDescendants.has(n));
+    for (let size = 0; size <= Math.min(maxSize, candidates.length); size++) {
+        for (const subset of combinations(candidates, size)) {
+            if (isValidBackdoorSet(graph, treatment, outcome, subset)) {
+                const exists = results.some((r) => r.type === 'backdoor' &&
+                    r.variables.length === subset.length &&
+                    r.variables.every((v) => subset.includes(v)));
+                if (!exists) {
+                    results.push({
+                        variables: subset,
+                        type: 'backdoor',
+                        formula: generateBackdoorFormula(treatment, outcome, subset),
+                        isMinimal: size === (backdoor?.variables.length ?? 0),
+                        isValid: true,
+                    });
+                }
+            }
+        }
+    }
+    return results;
+}
+function* combinations(arr, size) {
+    if (size === 0) {
+        yield [];
+        return;
+    }
+    if (arr.length < size)
+        return;
+    for (let i = 0; i <= arr.length - size; i++) {
+        for (const rest of combinations(arr.slice(i + 1), size - 1)) {
+            yield [arr[i], ...rest];
+        }
+    }
+}
+//# sourceMappingURL=adjustment.js.map

package/dist/causal/inference/adjustment.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"adjustment.js","sourceRoot":"","sources":["../../../src/causal/inference/adjustment.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAE7C,MAAM,UAAU,sBAAsB,CACpC,KAAkB,EAClB,SAAiB,EACjB,OAAe;IAEf,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACzD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACvF,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAEpC,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACnD,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAChC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,IAAI,CAAC,KAAK,OAAO,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,CACxE,CAAC;IAEF,MAAM,WAAW,GAAe,EAAE,CAAC;IAEnC,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,IAAI,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC;QACrD,KAAK,MAAM,MAAM,IAAI,YAAY,CAAC,UAAU,EAAE,IAAI,CAAC,EAAE,CAAC;YACpD,IAAI,kBAAkB,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,CAAC;gBAC1D,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACzB,IAAI,IAAI,KAAK,CAAC,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzC,MAAM,OAAO,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;oBACtC,IAAI,MAAM,CAAC,MAAM,GAAG,OAAO;wBAAE,MAAM;gBACrC,CAAC;YACH,CAAC;QACH,CAAC;QACD,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC;YAAE,MAAM;IACpC,CAAC;IAED,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,OAAO,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;IAE/B,OAAO;QACL,SAAS,EAAE,OAAO;QAClB,IAAI,EAAE,UAAU;QAChB,OAAO,EAAE,uBAAuB,CAAC,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC;QAC7D,SAAS,EAAE,IAAI;QACf,OAAO,EAAE,IAAI;KACd,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CACzB,KAAkB,EAClB,SAAiB,EACjB,OAAe,EACf,aAAuB;IAEvB,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACvF,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;QACjC,IAAI,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACnC,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,MAAM,aAAa,GAAG,+BAA+B,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IACxE,MAAM,MAAM,GAAG,WAAW,CAAC,aAAa,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;IAE7E,OAAO,MAAM,CAAC,SAAS,CAAC;AAC1B,CAAC;AAED,SAAS,+BAA+B,CAAC,KAAkB,EAAE,MAAc;IACzE,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;IAC7B,MAAM,QAAQ,GAAG,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAE3C,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;QAC7B,MAAM,IAAI,GAAG,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;QACrD,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,KAAkB,EAClB,SAAiB,EACjB,OAAe;IAEf,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACzD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAClD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACnD,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,IAAI,CAAC,KAAK,OAAO,CAAC,CAAC;IAE5E,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,IAAI,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC;QACrD,KAAK,MAAM,MAAM,IAAI,YAAY,CAAC,UAAU,EAAE,IAAI,CAAC,EAAE,CAAC;YACpD,IAAI,mBAAmB,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,CAAC;gBAC3D,OAAO;oBACL,SAAS,EAAE,MAAM;oBACjB,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,wBAAwB,CAAC,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC;oBAC7D,SAAS,EAAE,IAAI;oBACf,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,mBAAmB,CAC1B,KAAkB,EAClB,SAAiB,EACjB,OAAe,EACf,SAAmB;IAEnB,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC;IAEvC,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAClD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/C,IAAI,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnC,WAAW,GAAG,IAAI,CAAC;gBACnB,MAAM;YACR,CAAC;QACH,CAAC;QACD,IAAI,CAAC,WAAW;YAAE,OAAO,KAAK,CAAC;IACjC,CAAC;IAED,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;QAC1B,MAAM,gBAAgB,GAAG,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACrD,KAAK,MAAM,MAAM,IAAI,gBAAgB,EAAE,CAAC;YACtC,IAAI,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC;gBAC1E,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;QAC1B,MAAM,WAAW,GAAG,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;QACvD,MAAM,WAAW,GAAG,CAAC,SAAS,EAAE,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC;QAE/E,MAAM,SAAS,GAAG,WAAW,CAC3B,+BAA+B,CAAC,KAAK,EAAE,CAAC,CAAC,EACzC,CAAC,EACD,OAAO,EACP,WAAW,CACZ,CAAC;QACF,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC;YACzB,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,UAAU,CAAC,KAAkB,EAAE,SAAiB,EAAE,EAAU;IACnE,MAAM,SAAS,GAAG,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;IACzC,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,SAAS,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,eAAe,CAAC,KAAkB,EAAE,KAAa,EAAE,KAAa;IACvE,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACvE,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAEvE,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,KAAK,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC;QAC3B,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;YACtB,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,SAAS,uBAAuB,CAC9B,SAAiB,EACjB,OAAe,EACf,aAAuB;IAEvB,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,KAAK,OAAO,OAAO,SAAS,YAAY,OAAO,IAAI,SAAS,GAAG,CAAC;IACzE,CAAC;IAED,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5C,OAAO,KAAK,OAAO,OAAO,SAAS,WAAW,UAAU,OAAO,OAAO,IAAI,SAAS,KAAK,UAAU,OAAO,UAAU,GAAG,CAAC;AACzH,CAAC;AAED,SAAS,wBAAwB,CAAC,SAAiB,EAAE,OAAe,EAAE,SAAmB;IACvF,MAAM,CAAC,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/B,OAAO,KAAK,OAAO,OAAO,SAAS,WAAW,CAAC,OAAO,CAAC,IAAI,SAAS,QAAQ,SAAS,QAAQ,OAAO,IAAI,CAAC,KAAK,SAAS,QAAQ,SAAS,IAAI,CAAC;AAC/I,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,KAAkB,EAClB,SAAiB,EACjB,OAAe,EACf,OAAO,GAAG,CAAC;IAEX,MAAM,OAAO,GAAoB,EAAE,CAAC;IAEpC,MAAM,QAAQ,GAAG,sBAAsB,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IACnE,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACzB,CAAC;IAED,MAAM,SAAS,GAAG,uBAAuB,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IACrE,IAAI,SAAS,EAAE,CAAC;QACd,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC1B,CAAC;IAED,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAEvF,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACnD,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAChC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,IAAI,CAAC,KAAK,OAAO,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,CACxE,CAAC;IAEF,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC;QACxE,KAAK,MAAM,MAAM,IAAI,YAAY,CAAC,UAAU,EAAE,IAAI,CAAC,EAAE,CAAC;YACpD,IAAI,kBAAkB,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,CAAC;gBAC1D,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CACzB,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,IAAI,KAAK,UAAU;oBACrB,CAAC,CAAC,SAAS,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM;oBACpC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAC/C,CAAC;gBACF,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,OAAO,CAAC,IAAI,CAAC;wBACX,SAAS,EAAE,MAAM;wBACjB,IAAI,EAAE,UAAU;wBAChB,OAAO,EAAE,uBAAuB,CAAC,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC;wBAC5D,SAAS,EAAE,IAAI,KAAK,CAAC,QAAQ,EAAE,SAAS,CAAC,MAAM,IAAI,CAAC,CAAC;wBACrD,OAAO,EAAE,IAAI;qBACd,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,QAAQ,CAAC,CAAC,YAAY,CAAI,GAAQ,EAAE,IAAY;IAC9C,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;QACf,MAAM,EAAE,CAAC;QACT,OAAO;IACT,CAAC;IACD,IAAI,GAAG,CAAC,MAAM,GAAG,IAAI;QAAE,OAAO;IAE9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,CAAC,MAAM,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,KAAK,MAAM,IAAI,IAAI,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,EAAE,CAAC;YAC5D,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/causal/inference/counterfactual.d.ts

@@ -0,0 +1,24 @@
+import type { CausalGraph, CounterfactualQuery, CounterfactualResult } from '@cogitator-ai/types';
+export interface CounterfactualReasonerOptions {
+    defaultNoiseMean?: number;
+    defaultNoiseStd?: number;
+    maxIterations?: number;
+    convergenceThreshold?: number;
+}
+export declare class CounterfactualReasoner {
+    private options;
+    constructor(options?: CounterfactualReasonerOptions);
+    evaluate(graph: CausalGraph, query: CounterfactualQuery): CounterfactualResult;
+    private abduction;
+    private action;
+    private prediction;
+    private evaluateEquation;
+    private evaluateEquationWithNoise;
+    private computeValue;
+    private sampleNoise;
+    private sampleGaussian;
+    private toNumber;
+    private generateExplanation;
+}
+export declare function evaluateCounterfactual(graph: CausalGraph, query: CounterfactualQuery, options?: CounterfactualReasonerOptions): CounterfactualResult;
+//# sourceMappingURL=counterfactual.d.ts.map

package/dist/causal/inference/counterfactual.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"counterfactual.d.ts","sourceRoot":"","sources":["../../../src/causal/inference/counterfactual.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,WAAW,EAEX,mBAAmB,EACnB,oBAAoB,EAErB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,WAAW,6BAA6B;IAC5C,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B;AAED,qBAAa,sBAAsB;IACjC,OAAO,CAAC,OAAO,CAA0C;gBAE7C,OAAO,GAAE,6BAAkC;IASvD,QAAQ,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,mBAAmB,GAAG,oBAAoB;IAyB9E,OAAO,CAAC,SAAS;IAuBjB,OAAO,CAAC,MAAM;IAkBd,OAAO,CAAC,UAAU;IA8ClB,OAAO,CAAC,gBAAgB;IAmBxB,OAAO,CAAC,yBAAyB;IAoDjC,OAAO,CAAC,YAAY;IAmBpB,OAAO,CAAC,WAAW;IAwBnB,OAAO,CAAC,cAAc;IAOtB,OAAO,CAAC,QAAQ;IAOhB,OAAO,CAAC,mBAAmB;CAe5B;AAED,wBAAgB,sBAAsB,CACpC,KAAK,EAAE,WAAW,EAClB,KAAK,EAAE,mBAAmB,EAC1B,OAAO,CAAC,EAAE,6BAA6B,GACtC,oBAAoB,CAGtB"}

package/dist/causal/inference/counterfactual.js

@@ -0,0 +1,198 @@
+export class CounterfactualReasoner {
+    options;
+    constructor(options = {}) {
+        this.options = {
+            defaultNoiseMean: options.defaultNoiseMean ?? 0,
+            defaultNoiseStd: options.defaultNoiseStd ?? 1,
+            maxIterations: options.maxIterations ?? 100,
+            convergenceThreshold: options.convergenceThreshold ?? 0.001,
+        };
+    }
+    evaluate(graph, query) {
+        const abductionResult = this.abduction(graph, query.factual);
+        const actionResult = this.action(graph, query.intervention, abductionResult);
+        const predictionResult = this.prediction(graph, query.target, actionResult);
+        const factualValue = query.factual[query.target] ??
+            this.computeValue(graph, query.target, query.factual, abductionResult);
+        return {
+            query,
+            factualValue,
+            counterfactualValue: predictionResult.value,
+            probability: predictionResult.probability,
+            explanation: this.generateExplanation(query, factualValue, predictionResult.value),
+            reasoning: {
+                abduction: abductionResult,
+                action: actionResult,
+                prediction: predictionResult.values,
+            },
+        };
+    }
+    abduction(graph, factual) {
+        const noiseTerms = {};
+        const sortedNodes = graph.topologicalSort();
+        for (const nodeId of sortedNodes) {
+            const node = graph.getNode(nodeId);
+            if (!node)
+                continue;
+            if (factual[nodeId] !== undefined && node.equation) {
+                const observedValue = this.toNumber(factual[nodeId]);
+                const predictedValue = this.evaluateEquation(node.equation, graph, nodeId, factual, {});
+                noiseTerms[nodeId] = observedValue - predictedValue;
+            }
+            else {
+                noiseTerms[nodeId] = this.sampleNoise(node);
+            }
+        }
+        return noiseTerms;
+    }
+    action(_graph, intervention, noiseTerms) {
+        const result = {};
+        for (const [key, value] of Object.entries(noiseTerms)) {
+            result[key] = value;
+        }
+        for (const [variable, value] of Object.entries(intervention)) {
+            result[variable] = value;
+        }
+        return result;
+    }
+    prediction(graph, target, state) {
+        const sortedNodes = graph.topologicalSort();
+        const computedValues = { ...state };
+        for (const nodeId of sortedNodes) {
+            if (computedValues[nodeId] !== undefined && typeof computedValues[nodeId] !== 'number') {
+                continue;
+            }
+            const node = graph.getNode(nodeId);
+            if (!node)
+                continue;
+            if (node.equation) {
+                const noiseValue = typeof computedValues[nodeId] === 'number' ? computedValues[nodeId] : 0;
+                const parentValues = {};
+                for (const parent of graph.getParents(nodeId)) {
+                    if (computedValues[parent.id] !== undefined) {
+                        parentValues[parent.id] = computedValues[parent.id];
+                    }
+                }
+                computedValues[nodeId] = this.evaluateEquationWithNoise(node.equation, parentValues, noiseValue);
+            }
+        }
+        return {
+            value: computedValues[target] ?? 0,
+            probability: 1.0,
+            values: computedValues,
+        };
+    }
+    evaluateEquation(equation, graph, nodeId, values, _noiseTerms) {
+        const parents = graph.getParents(nodeId);
+        const parentValues = {};
+        for (const parent of parents) {
+            if (values[parent.id] !== undefined) {
+                parentValues[parent.id] = values[parent.id];
+            }
+        }
+        return this.evaluateEquationWithNoise(equation, parentValues, 0);
+    }
+    evaluateEquationWithNoise(equation, parentValues, noise) {
+        switch (equation.type) {
+            case 'linear': {
+                let result = equation.intercept ?? 0;
+                if (equation.coefficients) {
+                    for (const [variable, coefficient] of Object.entries(equation.coefficients)) {
+                        const value = this.toNumber(parentValues[variable] ?? 0);
+                        result += coefficient * value;
+                    }
+                }
+                return result + noise;
+            }
+            case 'logistic': {
+                let linearPart = equation.intercept ?? 0;
+                if (equation.coefficients) {
+                    for (const [variable, coefficient] of Object.entries(equation.coefficients)) {
+                        const value = this.toNumber(parentValues[variable] ?? 0);
+                        linearPart += coefficient * value;
+                    }
+                }
+                linearPart += noise;
+                return 1 / (1 + Math.exp(-linearPart));
+            }
+            case 'polynomial': {
+                let result = equation.intercept ?? 0;
+                if (equation.coefficients) {
+                    for (const [variable, coefficient] of Object.entries(equation.coefficients)) {
+                        const value = this.toNumber(parentValues[variable] ?? 0);
+                        const [varName, powerStr] = variable.split('^');
+                        const power = powerStr ? parseInt(powerStr, 10) : 1;
+                        const actualValue = this.toNumber(parentValues[varName] ?? value);
+                        result += coefficient * Math.pow(actualValue, power);
+                    }
+                }
+                return result + noise;
+            }
+            case 'custom': {
+                return noise;
+            }
+            default:
+                return noise;
+        }
+    }
+    computeValue(graph, target, observed, noiseTerms) {
+        const node = graph.getNode(target);
+        if (!node?.equation) {
+            return observed[target] ?? 0;
+        }
+        const parentValues = {};
+        for (const parent of graph.getParents(target)) {
+            parentValues[parent.id] = observed[parent.id] ?? 0;
+        }
+        return this.evaluateEquationWithNoise(node.equation, parentValues, noiseTerms[target] ?? 0);
+    }
+    sampleNoise(node) {
+        const params = node.equation?.noiseParams ?? {};
+        const distribution = node.equation?.noiseDistribution ?? 'gaussian';
+        switch (distribution) {
+            case 'gaussian': {
+                const mean = params.mean ?? this.options.defaultNoiseMean;
+                const std = params.std ?? this.options.defaultNoiseStd;
+                return this.sampleGaussian(mean, std);
+            }
+            case 'uniform': {
+                const min = params.min ?? -1;
+                const max = params.max ?? 1;
+                return min + Math.random() * (max - min);
+            }
+            case 'bernoulli': {
+                const p = params.p ?? 0.5;
+                return Math.random() < p ? 1 : 0;
+            }
+            default:
+                return 0;
+        }
+    }
+    sampleGaussian(mean, std) {
+        const u1 = Math.random();
+        const u2 = Math.random();
+        const z = Math.sqrt(-2 * Math.log(u1)) * Math.cos(2 * Math.PI * u2);
+        return mean + std * z;
+    }
+    toNumber(value) {
+        if (typeof value === 'number')
+            return value;
+        if (typeof value === 'boolean')
+            return value ? 1 : 0;
+        const parsed = parseFloat(value);
+        return isNaN(parsed) ? 0 : parsed;
+    }
+    generateExplanation(query, factualValue, counterfactualValue) {
+        const interventionDesc = Object.entries(query.intervention)
+            .map(([k, v]) => `${k}=${v}`)
+            .join(', ');
+        return (`Given the factual observation where ${query.target}=${factualValue}, ` +
+            `if we had intervened to set ${interventionDesc}, ` +
+            `then ${query.target} would have been ${counterfactualValue}.`);
+    }
+}
+export function evaluateCounterfactual(graph, query, options) {
+    const reasoner = new CounterfactualReasoner(options);
+    return reasoner.evaluate(graph, query);
+}
+//# sourceMappingURL=counterfactual.js.map

package/dist/causal/inference/counterfactual.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"counterfactual.js","sourceRoot":"","sources":["../../../src/causal/inference/counterfactual.ts"],"names":[],"mappings":"AAeA,MAAM,OAAO,sBAAsB;IACzB,OAAO,CAA0C;IAEzD,YAAY,UAAyC,EAAE;QACrD,IAAI,CAAC,OAAO,GAAG;YACb,gBAAgB,EAAE,OAAO,CAAC,gBAAgB,IAAI,CAAC;YAC/C,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,CAAC;YAC7C,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,GAAG;YAC3C,oBAAoB,EAAE,OAAO,CAAC,oBAAoB,IAAI,KAAK;SAC5D,CAAC;IACJ,CAAC;IAED,QAAQ,CAAC,KAAkB,EAAE,KAA0B;QACrD,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAE7D,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;QAE7E,MAAM,gBAAgB,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAE5E,MAAM,YAAY,GAChB,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC;YAC3B,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;QAEzE,OAAO;YACL,KAAK;YACL,YAAY;YACZ,mBAAmB,EAAE,gBAAgB,CAAC,KAAK;YAC3C,WAAW,EAAE,gBAAgB,CAAC,WAAW;YACzC,WAAW,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,YAAY,EAAE,gBAAgB,CAAC,KAAK,CAAC;YAClF,SAAS,EAAE;gBACT,SAAS,EAAE,eAAe;gBAC1B,MAAM,EAAE,YAAY;gBACpB,UAAU,EAAE,gBAAgB,CAAC,MAAM;aACpC;SACF,CAAC;IACJ,CAAC;IAEO,SAAS,CACf,KAAkB,EAClB,OAAkD;QAElD,MAAM,UAAU,GAA2B,EAAE,CAAC;QAC9C,MAAM,WAAW,GAAG,KAAK,CAAC,eAAe,EAAE,CAAC;QAE5C,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;YACjC,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACnC,IAAI,CAAC,IAAI;gBAAE,SAAS;YAEpB,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,SAAS,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACnD,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;gBACrD,MAAM,cAAc,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;gBACxF,UAAU,CAAC,MAAM,CAAC,GAAG,aAAa,GAAG,cAAc,CAAC;YACtD,CAAC;iBAAM,CAAC;gBACN,UAAU,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,MAAM,CACZ,MAAmB,EACnB,YAAuD,EACvD,UAAkC;QAElC,MAAM,MAAM,GAA8C,EAAE,CAAC;QAE7D,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;YACtD,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACtB,CAAC;QAED,KAAK,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC;QAC3B,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,UAAU,CAChB,KAAkB,EAClB,MAAc,EACd,KAAgD;QAMhD,MAAM,WAAW,GAAG,KAAK,CAAC,eAAe,EAAE,CAAC;QAC5C,MAAM,cAAc,GAA8C,EAAE,GAAG,KAAK,EAAE,CAAC;QAE/E,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;YACjC,IAAI,cAAc,CAAC,MAAM,CAAC,KAAK,SAAS,IAAI,OAAO,cAAc,CAAC,MAAM,CAAC,KAAK,QAAQ,EAAE,CAAC;gBACvF,SAAS;YACX,CAAC;YAED,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACnC,IAAI,CAAC,IAAI;gBAAE,SAAS;YAEpB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,MAAM,UAAU,GACd,OAAO,cAAc,CAAC,MAAM,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAE,cAAc,CAAC,MAAM,CAAY,CAAC,CAAC,CAAC,CAAC,CAAC;gBAEtF,MAAM,YAAY,GAA8C,EAAE,CAAC;gBACnE,KAAK,MAAM,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC9C,IAAI,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,SAAS,EAAE,CAAC;wBAC5C,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBACtD,CAAC;gBACH,CAAC;gBAED,cAAc,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,yBAAyB,CACrD,IAAI,CAAC,QAAQ,EACb,YAAY,EACZ,UAAU,CACX,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO;YACL,KAAK,EAAE,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC;YAClC,WAAW,EAAE,GAAG;YAChB,MAAM,EAAE,cAAc;SACvB,CAAC;IACJ,CAAC;IAEO,gBAAgB,CACtB,QAA4B,EAC5B,KAAkB,EAClB,MAAc,EACd,MAAiD,EACjD,WAAmC;QAEnC,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,YAAY,GAA8C,EAAE,CAAC;QAEnE,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,IAAI,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,SAAS,EAAE,CAAC;gBACpC,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC,yBAAyB,CAAC,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;IACnE,CAAC;IAEO,yBAAyB,CAC/B,QAA4B,EAC5B,YAAuD,EACvD,KAAa;QAEb,QAAQ,QAAQ,CAAC,IAAI,EAAE,CAAC;YACtB,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,IAAI,MAAM,GAAG,QAAQ,CAAC,SAAS,IAAI,CAAC,CAAC;gBACrC,IAAI,QAAQ,CAAC,YAAY,EAAE,CAAC;oBAC1B,KAAK,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;wBAC5E,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;wBACzD,MAAM,IAAI,WAAW,GAAG,KAAK,CAAC;oBAChC,CAAC;gBACH,CAAC;gBACD,OAAO,MAAM,GAAG,KAAK,CAAC;YACxB,CAAC;YAED,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,IAAI,UAAU,GAAG,QAAQ,CAAC,SAAS,IAAI,CAAC,CAAC;gBACzC,IAAI,QAAQ,CAAC,YAAY,EAAE,CAAC;oBAC1B,KAAK,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;wBAC5E,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;wBACzD,UAAU,IAAI,WAAW,GAAG,KAAK,CAAC;oBACpC,CAAC;gBACH,CAAC;gBACD,UAAU,IAAI,KAAK,CAAC;gBACpB,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;YACzC,CAAC;YAED,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,IAAI,MAAM,GAAG,QAAQ,CAAC,SAAS,IAAI,CAAC,CAAC;gBACrC,IAAI,QAAQ,CAAC,YAAY,EAAE,CAAC;oBAC1B,KAAK,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;wBAC5E,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;wBACzD,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;wBAChD,MAAM,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;wBACpD,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,CAAC;wBAClE,MAAM,IAAI,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;oBACvD,CAAC;gBACH,CAAC;gBACD,OAAO,MAAM,GAAG,KAAK,CAAC;YACxB,CAAC;YAED,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,OAAO,KAAK,CAAC;YACf,CAAC;YAED;gBACE,OAAO,KAAK,CAAC;QACjB,CAAC;IACH,CAAC;IAEO,YAAY,CAClB,KAAkB,EAClB,MAAc,EACd,QAAmD,EACnD,UAAkC;QAElC,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACnC,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC;YACpB,OAAO,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC;QAED,MAAM,YAAY,GAA8C,EAAE,CAAC;QACnE,KAAK,MAAM,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;QACrD,CAAC;QAED,OAAO,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;IAC9F,CAAC;IAEO,WAAW,CAAC,IAAgB;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC;QAChD,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,EAAE,iBAAiB,IAAI,UAAU,CAAC;QAEpE,QAAQ,YAAY,EAAE,CAAC;YACrB,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC;gBAC1D,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBACvD,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YACxC,CAAC;YACD,KAAK,SAAS,CAAC,CAAC,CAAC;gBACf,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;gBAC7B,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC;gBAC5B,OAAO,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;YAC3C,CAAC;YACD,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,GAAG,CAAC;gBAC1B,OAAO,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACnC,CAAC;YACD;gBACE,OAAO,CAAC,CAAC;QACb,CAAC;IACH,CAAC;IAEO,cAAc,CAAC,IAAY,EAAE,GAAW;QAC9C,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACzB,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACzB,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QACpE,OAAO,IAAI,GAAG,GAAG,GAAG,CAAC,CAAC;IACxB,CAAC;IAEO,QAAQ,CAAC,KAAgC;QAC/C,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC5C,IAAI,OAAO,KAAK,KAAK,SAAS;YAAE,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;QACjC,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IACpC,CAAC;IAEO,mBAAmB,CACzB,KAA0B,EAC1B,YAAuC,EACvC,mBAA8C;QAE9C,MAAM,gBAAgB,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC;aACxD,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;aAC5B,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,OAAO,CACL,uCAAuC,KAAK,CAAC,MAAM,IAAI,YAAY,IAAI;YACvE,+BAA+B,gBAAgB,IAAI;YACnD,QAAQ,KAAK,CAAC,MAAM,oBAAoB,mBAAmB,GAAG,CAC/D,CAAC;IACJ,CAAC;CACF;AAED,MAAM,UAAU,sBAAsB,CACpC,KAAkB,EAClB,KAA0B,EAC1B,OAAuC;IAEvC,MAAM,QAAQ,GAAG,IAAI,sBAAsB,CAAC,OAAO,CAAC,CAAC;IACrD,OAAO,QAAQ,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;AACzC,CAAC"}

package/dist/causal/inference/d-separation.d.ts

@@ -0,0 +1,4 @@
+import type { CausalGraph, DSeparationResult } from '@cogitator-ai/types';
+export declare function dSeparation(graph: CausalGraph, x: string | string[], y: string | string[], z?: string[]): DSeparationResult;
+export declare function findMinimalSeparatingSet(graph: CausalGraph, x: string | string[], y: string | string[], forbidden?: Set<string>): string[] | null;
+//# sourceMappingURL=d-separation.d.ts.map

package/dist/causal/inference/d-separation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"d-separation.d.ts","sourceRoot":"","sources":["../../../src/causal/inference/d-separation.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAc,iBAAiB,EAAc,MAAM,qBAAqB,CAAC;AAElG,wBAAgB,WAAW,CACzB,KAAK,EAAE,WAAW,EAClB,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,EACpB,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,EACpB,CAAC,GAAE,MAAM,EAAO,GACf,iBAAiB,CAkCnB;AAkJD,wBAAgB,wBAAwB,CACtC,KAAK,EAAE,WAAW,EAClB,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,EACpB,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,EACpB,SAAS,cAAoB,GAC5B,MAAM,EAAE,GAAG,IAAI,CAiBjB"}

package/dist/causal/inference/d-separation.js

@@ -0,0 +1,168 @@
+export function dSeparation(graph, x, y, z = []) {
+    const xSet = new Set(Array.isArray(x) ? x : [x]);
+    const ySet = new Set(Array.isArray(y) ? y : [y]);
+    const zSet = new Set(z);
+    const allPaths = [];
+    for (const xNode of xSet) {
+        for (const yNode of ySet) {
+            const paths = findAllUndirectedPaths(graph, xNode, yNode);
+            allPaths.push(...paths);
+        }
+    }
+    const blockedPaths = [];
+    const openPaths = [];
+    for (const path of allPaths) {
+        const blockResult = isPathBlocked(graph, path, zSet);
+        if (blockResult.blocked) {
+            path.isBlocked = true;
+            path.blockingNodes = blockResult.blockingNodes;
+            blockedPaths.push(path);
+        }
+        else {
+            openPaths.push(path);
+        }
+    }
+    return {
+        separated: openPaths.length === 0,
+        paths: allPaths,
+        blockedPaths,
+        openPaths,
+    };
+}
+function findAllUndirectedPaths(graph, from, to, maxLength = 10) {
+    const paths = [];
+    const visited = new Set();
+    const dfs = (current, path, depth) => {
+        if (depth > maxLength)
+            return;
+        if (current === to && path.length > 1) {
+            paths.push({
+                nodes: [...path],
+                edges: [],
+                totalStrength: 1,
+                isBlocked: false,
+                blockingNodes: [],
+            });
+            return;
+        }
+        visited.add(current);
+        const neighbors = new Set();
+        for (const parent of graph.getParents(current)) {
+            neighbors.add(parent.id);
+        }
+        for (const child of graph.getChildren(current)) {
+            neighbors.add(child.id);
+        }
+        for (const neighbor of neighbors) {
+            if (!visited.has(neighbor)) {
+                dfs(neighbor, [...path, neighbor], depth + 1);
+            }
+        }
+        visited.delete(current);
+    };
+    dfs(from, [from], 0);
+    return paths;
+}
+function isPathBlocked(graph, path, conditioningSet) {
+    const blockingNodes = [];
+    for (let i = 1; i < path.nodes.length - 1; i++) {
+        const prev = path.nodes[i - 1];
+        const curr = path.nodes[i];
+        const next = path.nodes[i + 1];
+        const tripleType = getLocalTripleType(graph, prev, curr, next);
+        if (tripleType === 'chain' || tripleType === 'fork') {
+            if (conditioningSet.has(curr)) {
+                blockingNodes.push(curr);
+            }
+        }
+        else if (tripleType === 'collider') {
+            const colliderOrDescendantInZ = isColliderOrDescendantInSet(graph, curr, conditioningSet);
+            if (!colliderOrDescendantInZ) {
+                blockingNodes.push(curr);
+            }
+        }
+    }
+    const blocked = blockingNodes.length > 0 || hasAllBlockedTriples(graph, path, conditioningSet);
+    return { blocked, blockingNodes };
+}
+function getLocalTripleType(graph, a, b, c) {
+    const aToB = graph.getEdgeBetween(a, b) !== undefined;
+    const bToA = graph.getEdgeBetween(b, a) !== undefined;
+    const bToC = graph.getEdgeBetween(b, c) !== undefined;
+    const cToB = graph.getEdgeBetween(c, b) !== undefined;
+    if ((aToB && bToC) || (cToB && bToA)) {
+        return 'chain';
+    }
+    if (bToA && bToC) {
+        return 'fork';
+    }
+    if (aToB && cToB) {
+        return 'collider';
+    }
+    if (aToB || bToA) {
+        if (bToC || cToB) {
+            return 'chain';
+        }
+    }
+    return 'chain';
+}
+function isColliderOrDescendantInSet(graph, nodeId, conditioningSet) {
+    if (conditioningSet.has(nodeId))
+        return true;
+    const descendants = graph.getDescendants(nodeId);
+    for (const desc of descendants) {
+        if (conditioningSet.has(desc.id))
+            return true;
+    }
+    return false;
+}
+function hasAllBlockedTriples(graph, path, conditioningSet) {
+    if (path.nodes.length < 3)
+        return false;
+    for (let i = 1; i < path.nodes.length - 1; i++) {
+        const prev = path.nodes[i - 1];
+        const curr = path.nodes[i];
+        const next = path.nodes[i + 1];
+        const tripleType = getLocalTripleType(graph, prev, curr, next);
+        if (tripleType === 'chain' || tripleType === 'fork') {
+            if (conditioningSet.has(curr)) {
+                return true;
+            }
+        }
+        else if (tripleType === 'collider') {
+            if (!isColliderOrDescendantInSet(graph, curr, conditioningSet)) {
+                return true;
+            }
+        }
+    }
+    return false;
+}
+export function findMinimalSeparatingSet(graph, x, y, forbidden = new Set()) {
+    const xSet = new Set(Array.isArray(x) ? x : [x]);
+    const ySet = new Set(Array.isArray(y) ? y : [y]);
+    const allNodes = graph.getNodes().map((n) => n.id);
+    const candidates = allNodes.filter((n) => !xSet.has(n) && !ySet.has(n) && !forbidden.has(n));
+    for (let size = 0; size <= candidates.length; size++) {
+        for (const subset of combinations(candidates, size)) {
+            const result = dSeparation(graph, x, y, subset);
+            if (result.separated) {
+                return subset;
+            }
+        }
+    }
+    return null;
+}
+function* combinations(arr, size) {
+    if (size === 0) {
+        yield [];
+        return;
+    }
+    if (arr.length < size)
+        return;
+    for (let i = 0; i <= arr.length - size; i++) {
+        for (const rest of combinations(arr.slice(i + 1), size - 1)) {
+            yield [arr[i], ...rest];
+        }
+    }
+}
+//# sourceMappingURL=d-separation.js.map

package/dist/causal/inference/d-separation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"d-separation.js","sourceRoot":"","sources":["../../../src/causal/inference/d-separation.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,WAAW,CACzB,KAAkB,EAClB,CAAoB,EACpB,CAAoB,EACpB,IAAc,EAAE;IAEhB,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACjD,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACjD,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;IAExB,MAAM,QAAQ,GAAiB,EAAE,CAAC;IAElC,KAAK,MAAM,KAAK,IAAI,IAAI,EAAE,CAAC;QACzB,KAAK,MAAM,KAAK,IAAI,IAAI,EAAE,CAAC;YACzB,MAAM,KAAK,GAAG,sBAAsB,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;YAC1D,QAAQ,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,MAAM,YAAY,GAAiB,EAAE,CAAC;IACtC,MAAM,SAAS,GAAiB,EAAE,CAAC;IAEnC,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;QAC5B,MAAM,WAAW,GAAG,aAAa,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QACrD,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,IAAI,CAAC,aAAa,GAAG,WAAW,CAAC,aAAa,CAAC;YAC/C,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;aAAM,CAAC;YACN,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAED,OAAO;QACL,SAAS,EAAE,SAAS,CAAC,MAAM,KAAK,CAAC;QACjC,KAAK,EAAE,QAAQ;QACf,YAAY;QACZ,SAAS;KACV,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAC7B,KAAkB,EAClB,IAAY,EACZ,EAAU,EACV,SAAS,GAAG,EAAE;IAEd,MAAM,KAAK,GAAiB,EAAE,CAAC;IAC/B,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAElC,MAAM,GAAG,GAAG,CAAC,OAAe,EAAE,IAAc,EAAE,KAAa,EAAE,EAAE;QAC7D,IAAI,KAAK,GAAG,SAAS;YAAE,OAAO;QAC9B,IAAI,OAAO,KAAK,EAAE,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,KAAK,CAAC,IAAI,CAAC;gBACT,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC;gBAChB,KAAK,EAAE,EAAE;gBACT,aAAa,EAAE,CAAC;gBAChB,SAAS,EAAE,KAAK;gBAChB,aAAa,EAAE,EAAE;aAClB,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAErB,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;QACpC,KAAK,MAAM,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/C,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC3B,CAAC;QACD,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/C,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC1B,CAAC;QAED,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC3B,GAAG,CAAC,QAAQ,EAAE,CAAC,GAAG,IAAI,EAAE,QAAQ,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;QAED,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC1B,CAAC,CAAC;IAEF,GAAG,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;IACrB,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,aAAa,CACpB,KAAkB,EAClB,IAAgB,EAChB,eAA4B;IAE5B,MAAM,aAAa,GAAa,EAAE,CAAC;IAEnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAE/B,MAAM,UAAU,GAAG,kBAAkB,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAE/D,IAAI,UAAU,KAAK,OAAO,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;YACpD,IAAI,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC9B,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;aAAM,IAAI,UAAU,KAAK,UAAU,EAAE,CAAC;YACrC,MAAM,uBAAuB,GAAG,2BAA2B,CAAC,KAAK,EAAE,IAAI,EAAE,eAAe,CAAC,CAAC;YAC1F,IAAI,CAAC,uBAAuB,EAAE,CAAC;gBAC7B,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,GAAG,CAAC,IAAI,oBAAoB,CAAC,KAAK,EAAE,IAAI,EAAE,eAAe,CAAC,CAAC;IAE/F,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;AACpC,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAkB,EAAE,CAAS,EAAE,CAAS,EAAE,CAAS;IAC7E,MAAM,IAAI,GAAG,KAAK,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC;IACtD,MAAM,IAAI,GAAG,KAAK,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC;IACtD,MAAM,IAAI,GAAG,KAAK,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC;IACtD,MAAM,IAAI,GAAG,KAAK,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC;IAEtD,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC;QACrC,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;QACjB,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;QACjB,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;QACjB,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;YACjB,OAAO,OAAO,CAAC;QACjB,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,2BAA2B,CAClC,KAAkB,EAClB,MAAc,EACd,eAA4B;IAE5B,IAAI,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC;QAAE,OAAO,IAAI,CAAC;IAE7C,MAAM,WAAW,GAAG,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IACjD,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;QAC/B,IAAI,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAAE,OAAO,IAAI,CAAC;IAChD,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,oBAAoB,CAC3B,KAAkB,EAClB,IAAgB,EAChB,eAA4B;IAE5B,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,KAAK,CAAC;IAExC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAE/B,MAAM,UAAU,GAAG,kBAAkB,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAE/D,IAAI,UAAU,KAAK,OAAO,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;YACpD,IAAI,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC9B,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;aAAM,IAAI,UAAU,KAAK,UAAU,EAAE,CAAC;YACrC,IAAI,CAAC,2BAA2B,CAAC,KAAK,EAAE,IAAI,EAAE,eAAe,CAAC,EAAE,CAAC;gBAC/D,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,KAAkB,EAClB,CAAoB,EACpB,CAAoB,EACpB,YAAY,IAAI,GAAG,EAAU;IAE7B,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACjD,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAEjD,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACnD,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAE7F,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,IAAI,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC;QACrD,KAAK,MAAM,MAAM,IAAI,YAAY,CAAC,UAAU,EAAE,IAAI,CAAC,EAAE,CAAC;YACpD,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;YAChD,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACrB,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,QAAQ,CAAC,CAAC,YAAY,CAAI,GAAQ,EAAE,IAAY;IAC9C,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;QACf,MAAM,EAAE,CAAC;QACT,OAAO;IACT,CAAC;IACD,IAAI,GAAG,CAAC,MAAM,GAAG,IAAI;QAAE,OAAO;IAE9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,CAAC,MAAM,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,KAAK,MAAM,IAAI,IAAI,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,EAAE,CAAC;YAC5D,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;AACH,CAAC"}