github/powdr
1
1
Fork 0
mirror of https://github.com/powdr-labs/powdr.git synced 2026-04-20 03:03:25 -04:00
Code Issues Packages Projects Releases Wiki Activity

Visitor for FunctionValueDefinition

Browse Source
This commit is contained in:
chriseth
2023-10-10 15:50:50 +02:00
parent 7940c8887f
commit 48d6de72bd
1 changed files with 32 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
ast/src/analyzed/visitor.rs
View File

@@ -79,3 +79,35 @@ impl<T> ExpressionVisitable<parsed::Expression<T, Reference>> for Identity<T> {
.try_for_each(move |item| item.visit_expressions(f, o))
}
}
impl<T> ExpressionVisitable<Expression<T>> for FunctionValueDefinition<T> {
fn visit_expressions_mut<F, B>(&mut self, f: &mut F, o: VisitOrder) -> ControlFlow<B>
where
F: FnMut(&mut Expression<T>) -> ControlFlow<B>,
{
match self {
FunctionValueDefinition::Mapping(e)
| FunctionValueDefinition::Query(e)
| FunctionValueDefinition::Expression(e) => e.visit_expressions_mut(f, o),
FunctionValueDefinition::Array(array) => array
.iter_mut()
.flat_map(|a| a.pattern.iter_mut())
.try_for_each(move |item| item.visit_expressions_mut(f, o)),
}
}
fn visit_expressions<F, B>(&self, f: &mut F, o: VisitOrder) -> ControlFlow<B>
where
F: FnMut(&Expression<T>) -> ControlFlow<B>,
{
match self {
FunctionValueDefinition::Mapping(e)
| FunctionValueDefinition::Query(e)
| FunctionValueDefinition::Expression(e) => e.visit_expressions(f, o),
FunctionValueDefinition::Array(array) => array
.iter()
.flat_map(|a| a.pattern().iter())
.try_for_each(move |item| item.visit_expressions(f, o)),
}
}
}