Make the modifier more flexible.

This commit is contained in:
chriseth 2017-12-05 19:46:16 +01:00
parent a08d853bbb
commit 6769a9a503
2 changed files with 10 additions and 6 deletions

View File

@ -104,24 +104,24 @@ void ASTModifier::operator()(Assignment& _assignment)
{
for (auto& name: _assignment.variableNames)
(*this)(name);
boost::apply_visitor(*this, *_assignment.value);
visit(*_assignment.value);
}
void ASTModifier::operator()(VariableDeclaration& _varDecl)
{
if (_varDecl.value)
boost::apply_visitor(*this, *_varDecl.value);
visit(*_varDecl.value);
}
void ASTModifier::operator()(If& _if)
{
boost::apply_visitor(*this, *_if.condition);
visit(*_if.condition);
(*this)(_if.body);
}
void ASTModifier::operator()(Switch& _switch)
{
boost::apply_visitor(*this, *_switch.expression);
visit(*_switch.expression);
for (auto& _case: _switch.cases)
{
if (_case.value)
@ -138,7 +138,7 @@ void ASTModifier::operator()(FunctionDefinition& _fun)
void ASTModifier::operator()(ForLoop& _for)
{
(*this)(_for.pre);
boost::apply_visitor(*this, *_for.condition);
visit(*_for.condition);
(*this)(_for.post);
(*this)(_for.body);
}

View File

@ -92,7 +92,11 @@ protected:
void walkVector(T&& _statements)
{
for (auto& st: _statements)
boost::apply_visitor(*this, st);
visit(st);
}
virtual void visit(Statement& _st)
{
boost::apply_visitor(*this, _st);
}
};