oximo_expr/
visit.rs

1use crate::arena::{ExprArena, ExprId, ExprNode};
2
3/// Pre-order visitor over an arena. Backends implement this to translate the
4/// expression tree into solver-specific representations without copying.
5pub trait Visitor {
6    fn visit(&mut self, arena: &ExprArena, id: ExprId, node: &ExprNode);
7}
8
9/// Walk the subtree rooted at `id` in pre-order.
10pub fn walk<V: Visitor>(arena: &ExprArena, id: ExprId, visitor: &mut V) {
11    let node = arena.get(id);
12    visitor.visit(arena, id, node);
13    match node {
14        ExprNode::Add(children) | ExprNode::Mul(children) => {
15            let kids: Vec<ExprId> = children.iter().copied().collect();
16            for child in kids {
17                walk(arena, child, visitor);
18            }
19        }
20        ExprNode::Neg(inner)
21        | ExprNode::Sin(inner)
22        | ExprNode::Cos(inner)
23        | ExprNode::Exp(inner)
24        | ExprNode::Log(inner) => {
25            let inner = *inner;
26            walk(arena, inner, visitor);
27        }
28        ExprNode::Pow(base, exp) => {
29            let base = *base;
30            let exp = *exp;
31            walk(arena, base, visitor);
32            walk(arena, exp, visitor);
33        }
34        ExprNode::Const(_) | ExprNode::Var(_) | ExprNode::Param(_) | ExprNode::Linear { .. } => {}
35    }
36}
oximo_expr/visit.rs

oximo_expr/
visit.rs
