Struct CausalExprArena 

pub struct CausalExprArena { /* private fields */ }

Arena for causal expressions with interned variable sets.

Implementations

impl CausalExprArena

pub fn compile(&self, root: ExprId) -> Result<CompiledEvaluator, EvalError>

Compile root into a topological evaluation plan.

Continuous ExprNode::IntegralOut compiles successfully; evaluation uses DistributionProvider::quadrature or discrete DistributionProvider::support.

impl CausalExprArena

pub fn new() -> Self

Empty arena.

pub fn intern_var_set( &mut self, vars: impl IntoIterator<Item = VariableId>, ) -> VarSetId

Intern a sorted variable set (sorts and dedups input).

pub fn intern_intervention_assignments( &mut self, assignments: impl IntoIterator<Item = InterventionAssignment>, ) -> InterventionSetId

Intern a hard-intervention assignment set (sorted by variable id).

pub fn intern_intervention_set( &mut self, vars: impl IntoIterator<Item = VariableId>, ) -> InterventionSetId

Intern an intervention over variables only (value unspecified / placeholder).

pub fn empty_var_set(&mut self) -> VarSetId

Empty var set.

pub fn empty_intervention_set(&mut self) -> InterventionSetId

Empty intervention set.

pub fn var_set(&self, id: VarSetId) -> &[VariableId]

Look up a var set.

pub fn intervention_assignments( &self, id: InterventionSetId, ) -> &[InterventionAssignment]

Look up intervention assignments.

pub fn intervention_set(&self, id: InterventionSetId) -> Vec<VariableId>

Variables appearing in an intervention set (legacy helper).

pub fn intern_list( &mut self, exprs: impl IntoIterator<Item = ExprId>, ) -> ExprListId

Intern an expression list.

pub fn list(&self, id: ExprListId) -> &[ExprId]

Borrow an interned expression list.

pub fn intern(&mut self, node: ExprNode) -> ExprId

Hash-cons an expression node.

pub fn set_derivation(&mut self, id: ExprId, meta: DerivationMeta)

Attach derivation metadata (does not affect semantic equality).

pub fn set_derivation_if_absent(&mut self, id: ExprId, meta: DerivationMeta)

Attach derivation metadata only when absent (never overwrites ID rules).

pub fn simplify(&mut self, root: ExprId) -> Result<ExprId, SimplifyError>

Simplify root with worklist-style bottom-up rewrite + memoization.

Errors

SimplifyError if a SumOut/IntegralOut binds a variable absent from its body’s free variables — an ill-formed estimand. See SimplifyError docs.

pub fn derivation(&self, id: ExprId) -> Option<&DerivationMeta>

Borrow derivation metadata.

pub fn node(&self, id: ExprId) -> &ExprNode

Borrow a node.

pub fn len(&self) -> usize

Number of nodes.

pub fn is_empty(&self) -> bool

Whether empty.

pub fn var_set_count(&self) -> usize

Number of interned variable sets (for serialization).

pub fn intervention_set_count(&self) -> usize

Number of interned intervention sets (for serialization).

pub fn list_count(&self) -> usize

Number of interned expression lists (for serialization).

pub fn backdoor_ate( &mut self, treatment: VariableId, outcome: VariableId, adjustment: &[VariableId], active: Value, control: Value, ) -> ExprId

Build the backdoor adjustment functional for ATE: E[Y | do(T=active)] − E[Y | do(T=control)] under adjustment by Z.

pub fn frontdoor_ate( &mut self, treatment: VariableId, outcome: VariableId, mediators: &[VariableId], active: Value, control: Value, ) -> ExprId

Build the front-door functional for ATE: E[Y | do(T=active)] − E[Y | do(T=control)], mediated through M via sum_m P(m | t) * sum_t' P(y | m, t') P(t') (FD condition 2 reduces P(m | do(t)) to the observational P(m | t)).

pub fn temporal_mediation_ate( &mut self, treatment: VariableId, outcome: VariableId, mediators: &[VariableId], active: Value, control: Value, ) -> ExprId

Linear temporal-mediation path-product ATE contrast (same product-of-coefficients geometry as front-door under a linear SEM, tagged temporal_mediation — not front-door).

pub fn iv_wald( &mut self, treatment: VariableId, outcome: VariableId, instruments: &[VariableId], active: &Value, control: &Value, ) -> ExprId

Build the Wald IV functional for binary instrument Z: (E[Y|Z=1] − E[Y|Z=0]) / (E[T|Z=1] − E[T|Z=0]).

active / control are recorded in derivation metadata (treatment contrast scaling); the ratio itself conditions on instrument levels 1 and 0.

pub fn pretty(&self, id: ExprId) -> String

Pretty-print an expression (diagnostics only; not an equality key).

pub fn latex(&self, id: ExprId) -> String

Render an expression as LaTeX (diagnostics only; not an equality key).

Trait Implementations

impl Clone for CausalExprArena

fn clone(&self) -> CausalExprArena

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for CausalExprArena

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for CausalExprArena

fn default() -> CausalExprArena

Returns the “default value” for a type.