Helper for visiting statements.

1 files changed, 8 insertions, 3 deletions

diff --git a/libjulia/backends/evm/EVMCodeTransform.cpp b/libjulia/backends/evm/EVMCodeTransform.cpp
index a50af9af..c543a507 100644
--- a/libjulia/backends/evm/EVMCodeTransform.cpp
+++ b/libjulia/backends/evm/EVMCodeTransform.cpp
@@ -345,8 +345,7 @@ void CodeTransform::operator()(ForLoop const& _forLoop)
 	m_scope = m_info.scopes.at(&_forLoop.pre).get();
 	int stackStartHeight = m_assembly.stackHeight();
 
-	for (auto const& statement: _forLoop.pre.statements)
-		boost::apply_visitor(*this, statement);
+	visitStatements(_forLoop.pre.statements);
 
 	// TODO: When we implement break and continue, the labels and the stack heights at that point
 	// have to be stored in a stack.
@@ -382,7 +381,7 @@ void CodeTransform::operator()(Block const& _block)
 	m_scope = m_info.scopes.at(&_block).get();
 
 	int blockStartStackHeight = m_assembly.stackHeight();
-	std::for_each(_block.statements.begin(), _block.statements.end(), boost::apply_visitor(*this));
+	visitStatements(_block.statements);
 
 	finalizeBlock(_block, blockStartStackHeight);
 	m_scope = originalScope;
@@ -426,6 +425,12 @@ void CodeTransform::visitExpression(Statement const& _expression)
 	expectDeposit(1, height);
 }
 
+void CodeTransform::visitStatements(vector<Statement> const& _statements)
+{
+	for (auto const& statement: _statements)
+		boost::apply_visitor(*this, statement);
+}
+
 void CodeTransform::finalizeBlock(Block const& _block, int blockStartStackHeight)
 {
 	m_assembly.setSourceLocation(_block.location);