roland/xo-numeric
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

xo-expression: unify variables within each lambda layer

Browse source
This commit is contained in:
Roland Conybeare 2024-07-03 14:31:47 -04:00
commit fdfe2e7270
1 changed files with 22 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/expression/Lambda.cpp
View file

@ -110,6 +110,28 @@ namespace xo {
var_map[arg->name()] = arg;
}
body_ = body_->xform_layer
([&var_map](ref::brw<Expression> x) -> ref::rp<Expression>
{
if (x->extype() == exprtype::variable) {
ref::brw<Variable> var = Variable::from(x);
auto ix = var_map.find(var->name());
if (ix == var_map.end()) {
/* add to var_map */
var_map[var->name()] = var.get();
return var.get();
} else {
/* substitute already-encountered var_map[] member */
return ix->second;
}
} else {
return x.get();
}
});
}
this->type_str_ = ss.str();