Reuse stack slots in Yul to EVM code generation.

This commit is contained in:
liangdzou 2018-09-25 10:47:25 +08:00 committed by chriseth
parent 6240d9e72a
commit 362648a450
11 changed files with 313 additions and 54 deletions

View File

@ -179,13 +179,16 @@ void CodeGenerator::assemble(
AsmAnalysisInfo& _analysisInfo, AsmAnalysisInfo& _analysisInfo,
eth::Assembly& _assembly, eth::Assembly& _assembly,
ExternalIdentifierAccess const& _identifierAccess, ExternalIdentifierAccess const& _identifierAccess,
bool _useNamedLabelsForFunctions bool _useNamedLabelsForFunctions,
bool _optimize
) )
{ {
EthAssemblyAdapter assemblyAdapter(_assembly); EthAssemblyAdapter assemblyAdapter(_assembly);
CodeTransform( CodeTransform(
assemblyAdapter, assemblyAdapter,
_analysisInfo, _analysisInfo,
_parsedData,
_optimize,
false, false,
false, false,
_identifierAccess, _identifierAccess,

View File

@ -85,7 +85,8 @@ public:
yul::AsmAnalysisInfo& _analysisInfo, yul::AsmAnalysisInfo& _analysisInfo,
dev::eth::Assembly& _assembly, dev::eth::Assembly& _assembly,
yul::ExternalIdentifierAccess const& _identifierAccess = yul::ExternalIdentifierAccess(), yul::ExternalIdentifierAccess const& _identifierAccess = yul::ExternalIdentifierAccess(),
bool _useNamedLabelsForFunctions = false bool _useNamedLabelsForFunctions = false,
bool _optimize = false
); );
}; };

View File

@ -122,7 +122,7 @@ void AssemblyStack::optimize(yul::Object& _object)
yul::OptimiserSuite::run(*_object.code, *_object.analysisInfo); yul::OptimiserSuite::run(*_object.code, *_object.analysisInfo);
} }
MachineAssemblyObject AssemblyStack::assemble(Machine _machine) const MachineAssemblyObject AssemblyStack::assemble(Machine _machine, bool _optimize) const
{ {
solAssert(m_analysisSuccessful, ""); solAssert(m_analysisSuccessful, "");
solAssert(m_parserResult, ""); solAssert(m_parserResult, "");
@ -136,7 +136,7 @@ MachineAssemblyObject AssemblyStack::assemble(Machine _machine) const
MachineAssemblyObject object; MachineAssemblyObject object;
eth::Assembly assembly; eth::Assembly assembly;
EthAssemblyAdapter adapter(assembly); EthAssemblyAdapter adapter(assembly);
yul::EVMObjectCompiler::compile(*m_parserResult, adapter, m_language == Language::Yul, false); yul::EVMObjectCompiler::compile(*m_parserResult, adapter, m_language == Language::Yul, false, _optimize);
object.bytecode = make_shared<eth::LinkerObject>(assembly.assemble()); object.bytecode = make_shared<eth::LinkerObject>(assembly.assemble());
object.assembly = assembly.assemblyString(); object.assembly = assembly.assemblyString();
return object; return object;
@ -145,7 +145,7 @@ MachineAssemblyObject AssemblyStack::assemble(Machine _machine) const
{ {
MachineAssemblyObject object; MachineAssemblyObject object;
yul::EVMAssembly assembly(true); yul::EVMAssembly assembly(true);
yul::EVMObjectCompiler::compile(*m_parserResult, assembly, m_language == Language::Yul, true); yul::EVMObjectCompiler::compile(*m_parserResult, assembly, m_language == Language::Yul, true, _optimize);
object.bytecode = make_shared<eth::LinkerObject>(assembly.finalize()); object.bytecode = make_shared<eth::LinkerObject>(assembly.finalize());
/// TODO: fill out text representation /// TODO: fill out text representation
return object; return object;

View File

@ -73,7 +73,8 @@ public:
void optimize(); void optimize();
/// Run the assembly step (should only be called after parseAndAnalyze). /// Run the assembly step (should only be called after parseAndAnalyze).
MachineAssemblyObject assemble(Machine _machine) const; /// @param _optimize does not run the optimizer but performs optimized code generation.
MachineAssemblyObject assemble(Machine _machine, bool _optimize = false) const;
/// @returns the errors generated during parsing, analysis (and potentially assembly). /// @returns the errors generated during parsing, analysis (and potentially assembly).
langutil::ErrorList const& errors() const { return m_errors; } langutil::ErrorList const& errors() const { return m_errors; }

View File

@ -28,6 +28,7 @@ namespace yul
struct YulException: virtual dev::Exception {}; struct YulException: virtual dev::Exception {};
struct OptimizerException: virtual YulException {}; struct OptimizerException: virtual YulException {};
struct CodegenException: virtual YulException {};
struct YulAssertion: virtual YulException {}; struct YulAssertion: virtual YulException {};
/// Assertion that throws an YulAssertion containing the given description if it is not met. /// Assertion that throws an YulAssertion containing the given description if it is not met.

View File

@ -20,6 +20,7 @@
#include <libyul/backends/evm/EVMCodeTransform.h> #include <libyul/backends/evm/EVMCodeTransform.h>
#include <libyul/optimiser/NameCollector.h>
#include <libyul/AsmAnalysisInfo.h> #include <libyul/AsmAnalysisInfo.h>
#include <libyul/AsmData.h> #include <libyul/AsmData.h>
@ -32,6 +33,144 @@ using namespace dev;
using namespace yul; using namespace yul;
using namespace dev::solidity; using namespace dev::solidity;
void VariableReferenceCounter::operator()(Identifier const& _identifier)
{
increaseRefIfFound(_identifier.name);
}
void VariableReferenceCounter::operator()(FunctionDefinition const& _function)
{
Scope* originalScope = m_scope;
solAssert(m_info.virtualBlocks.at(&_function), "");
m_scope = m_info.scopes.at(m_info.virtualBlocks.at(&_function).get()).get();
solAssert(m_scope, "Variable scope does not exist.");
for (auto const& v: _function.returnVariables)
increaseRefIfFound(v.name);
VariableReferenceCounter{m_context, m_info}(_function.body);
m_scope = originalScope;
}
void VariableReferenceCounter::operator()(ForLoop const& _forLoop)
{
Scope* originalScope = m_scope;
// Special scoping rules.
m_scope = m_info.scopes.at(&_forLoop.pre).get();
walkVector(_forLoop.pre.statements);
visit(*_forLoop.condition);
(*this)(_forLoop.body);
(*this)(_forLoop.post);
m_scope = originalScope;
}
void VariableReferenceCounter::operator()(Block const& _block)
{
Scope* originalScope = m_scope;
m_scope = m_info.scopes.at(&_block).get();
ASTWalker::operator()(_block);
m_scope = originalScope;
}
void VariableReferenceCounter::increaseRefIfFound(YulString _variableName)
{
m_scope->lookup(_variableName, Scope::Visitor(
[=](Scope::Variable const& _var)
{
++m_context.variableReferences[&_var];
},
[=](Scope::Label const&) { },
[=](Scope::Function const&) { }
));
}
CodeTransform::CodeTransform(
AbstractAssembly& _assembly,
AsmAnalysisInfo& _analysisInfo,
Block const& _block,
bool _allowStackOpt,
bool _yul,
bool _evm15,
ExternalIdentifierAccess const& _identifierAccess,
bool _useNamedLabelsForFunctions,
int _stackAdjustment,
shared_ptr<Context> _context
):
m_assembly(_assembly),
m_info(_analysisInfo),
m_allowStackOpt(_allowStackOpt),
m_yul(_yul),
m_evm15(_evm15),
m_useNamedLabelsForFunctions(_useNamedLabelsForFunctions),
m_identifierAccess(_identifierAccess),
m_stackAdjustment(_stackAdjustment),
m_context(_context)
{
if (!m_context)
{
// initialize
m_context = make_shared<Context>();
if (m_allowStackOpt)
VariableReferenceCounter{*m_context, m_info}(_block);
}
}
void CodeTransform::decreaseReference(YulString, Scope::Variable const& _var)
{
if (!m_allowStackOpt)
return;
unsigned& ref = m_context->variableReferences.at(&_var);
solAssert(ref >= 1, "");
--ref;
if (ref == 0)
m_variablesScheduledForDeletion.insert(&_var);
}
bool CodeTransform::unreferenced(Scope::Variable const& _var) const
{
return !m_context->variableReferences.count(&_var) || m_context->variableReferences[&_var] == 0;
}
void CodeTransform::freeUnusedVariables()
{
if (!m_allowStackOpt)
return;
for (auto const& identifier: m_scope->identifiers)
if (identifier.second.type() == typeid(Scope::Variable))
{
Scope::Variable const& var = boost::get<Scope::Variable>(identifier.second);
if (m_variablesScheduledForDeletion.count(&var))
deleteVariable(var);
}
while (m_unusedStackSlots.count(m_assembly.stackHeight() - 1))
{
solAssert(m_unusedStackSlots.erase(m_assembly.stackHeight() - 1), "");
m_assembly.appendInstruction(solidity::Instruction::POP);
--m_stackAdjustment;
}
}
void CodeTransform::deleteVariable(Scope::Variable const& _var)
{
solAssert(m_allowStackOpt, "");
solAssert(m_context->variableStackHeights.count(&_var) > 0, "");
m_unusedStackSlots.insert(m_context->variableStackHeights[&_var]);
m_context->variableStackHeights.erase(&_var);
m_context->variableReferences.erase(&_var);
m_variablesScheduledForDeletion.erase(&_var);
}
void CodeTransform::operator()(VariableDeclaration const& _varDecl) void CodeTransform::operator()(VariableDeclaration const& _varDecl)
{ {
solAssert(m_scope, ""); solAssert(m_scope, "");
@ -49,10 +188,40 @@ void CodeTransform::operator()(VariableDeclaration const& _varDecl)
while (variablesLeft--) while (variablesLeft--)
m_assembly.appendConstant(u256(0)); m_assembly.appendConstant(u256(0));
} }
for (auto const& variable: _varDecl.variables)
bool atTopOfStack = true;
for (int varIndex = numVariables - 1; varIndex >= 0; --varIndex)
{ {
auto& var = boost::get<Scope::Variable>(m_scope->identifiers.at(variable.name)); auto& var = boost::get<Scope::Variable>(m_scope->identifiers.at(_varDecl.variables[varIndex].name));
m_context->variableStackHeights[&var] = height++; m_context->variableStackHeights[&var] = height + varIndex;
if (!m_allowStackOpt)
continue;
if (unreferenced(var))
{
if (atTopOfStack)
{
m_context->variableStackHeights.erase(&var);
m_assembly.setSourceLocation(_varDecl.location);
m_assembly.appendInstruction(solidity::Instruction::POP);
--m_stackAdjustment;
}
else
m_variablesScheduledForDeletion.insert(&var);
}
else if (m_unusedStackSlots.empty())
atTopOfStack = false;
else
{
int slot = *m_unusedStackSlots.begin();
m_unusedStackSlots.erase(m_unusedStackSlots.begin());
m_context->variableStackHeights[&var] = slot;
m_assembly.setSourceLocation(_varDecl.location);
if (int heightDiff = variableHeightDiff(var, true))
m_assembly.appendInstruction(solidity::swapInstruction(heightDiff - 1));
m_assembly.appendInstruction(solidity::Instruction::POP);
--m_stackAdjustment;
}
} }
checkStackHeight(&_varDecl); checkStackHeight(&_varDecl);
} }
@ -70,6 +239,7 @@ void CodeTransform::operator()(Assignment const& _assignment)
void CodeTransform::operator()(StackAssignment const& _assignment) void CodeTransform::operator()(StackAssignment const& _assignment)
{ {
solAssert(!m_allowStackOpt, "");
m_assembly.setSourceLocation(_assignment.location); m_assembly.setSourceLocation(_assignment.location);
generateAssignment(_assignment.variableName); generateAssignment(_assignment.variableName);
checkStackHeight(&_assignment); checkStackHeight(&_assignment);
@ -84,6 +254,7 @@ void CodeTransform::operator()(ExpressionStatement const& _statement)
void CodeTransform::operator()(Label const& _label) void CodeTransform::operator()(Label const& _label)
{ {
solAssert(!m_allowStackOpt, "");
m_assembly.setSourceLocation(_label.location); m_assembly.setSourceLocation(_label.location);
solAssert(m_scope, ""); solAssert(m_scope, "");
solAssert(m_scope->identifiers.count(_label.name), ""); solAssert(m_scope->identifiers.count(_label.name), "");
@ -169,11 +340,14 @@ void CodeTransform::operator()(Identifier const& _identifier)
if (m_scope->lookup(_identifier.name, Scope::NonconstVisitor( if (m_scope->lookup(_identifier.name, Scope::NonconstVisitor(
[=](Scope::Variable& _var) [=](Scope::Variable& _var)
{ {
// TODO: opportunity for optimization: Do not DUP if this is the last reference
// to the top most element of the stack
if (int heightDiff = variableHeightDiff(_var, false)) if (int heightDiff = variableHeightDiff(_var, false))
m_assembly.appendInstruction(solidity::dupInstruction(heightDiff)); m_assembly.appendInstruction(solidity::dupInstruction(heightDiff));
else else
// Store something to balance the stack // Store something to balance the stack
m_assembly.appendConstant(u256(0)); m_assembly.appendConstant(u256(0));
decreaseReference(_identifier.name, _var);
}, },
[=](Scope::Label& _label) [=](Scope::Label& _label)
{ {
@ -217,6 +391,7 @@ void CodeTransform::operator()(Literal const& _literal)
void CodeTransform::operator()(yul::Instruction const& _instruction) void CodeTransform::operator()(yul::Instruction const& _instruction)
{ {
solAssert(!m_allowStackOpt, "");
solAssert(!m_evm15 || _instruction.instruction != solidity::Instruction::JUMP, "Bare JUMP instruction used for EVM1.5"); solAssert(!m_evm15 || _instruction.instruction != solidity::Instruction::JUMP, "Bare JUMP instruction used for EVM1.5");
solAssert(!m_evm15 || _instruction.instruction != solidity::Instruction::JUMPI, "Bare JUMPI instruction used for EVM1.5"); solAssert(!m_evm15 || _instruction.instruction != solidity::Instruction::JUMPI, "Bare JUMPI instruction used for EVM1.5");
m_assembly.setSourceLocation(_instruction.location); m_assembly.setSourceLocation(_instruction.location);
@ -329,6 +504,8 @@ void CodeTransform::operator()(FunctionDefinition const& _function)
CodeTransform( CodeTransform(
m_assembly, m_assembly,
m_info, m_info,
_function.body,
m_allowStackOpt,
m_yul, m_yul,
m_evm15, m_evm15,
m_identifierAccess, m_identifierAccess,
@ -350,6 +527,7 @@ void CodeTransform::operator()(FunctionDefinition const& _function)
if (!m_evm15) if (!m_evm15)
stackLayout.push_back(_function.returnVariables.size()); // Move return label to the top stackLayout.push_back(_function.returnVariables.size()); // Move return label to the top
stackLayout += vector<int>(_function.parameters.size(), -1); // discard all arguments stackLayout += vector<int>(_function.parameters.size(), -1); // discard all arguments
for (size_t i = 0; i < _function.returnVariables.size(); ++i) for (size_t i = 0; i < _function.returnVariables.size(); ++i)
stackLayout.push_back(i); // Move return values down, but keep order. stackLayout.push_back(i); // Move return values down, but keep order.
@ -475,20 +653,37 @@ void CodeTransform::visitExpression(Expression const& _expression)
void CodeTransform::visitStatements(vector<Statement> const& _statements) void CodeTransform::visitStatements(vector<Statement> const& _statements)
{ {
for (auto const& statement: _statements) for (auto const& statement: _statements)
{
freeUnusedVariables();
boost::apply_visitor(*this, statement); boost::apply_visitor(*this, statement);
}
freeUnusedVariables();
} }
void CodeTransform::finalizeBlock(Block const& _block, int blockStartStackHeight) void CodeTransform::finalizeBlock(Block const& _block, int blockStartStackHeight)
{ {
m_assembly.setSourceLocation(_block.location); m_assembly.setSourceLocation(_block.location);
freeUnusedVariables();
// pop variables // pop variables
solAssert(m_info.scopes.at(&_block).get() == m_scope, ""); solAssert(m_info.scopes.at(&_block).get() == m_scope, "");
for (size_t i = 0; i < m_scope->numberOfVariables(); ++i) for (auto const& id: m_scope->identifiers)
m_assembly.appendInstruction(solidity::Instruction::POP); if (id.second.type() == typeid(Scope::Variable))
{
Scope::Variable const& var = boost::get<Scope::Variable>(id.second);
if (m_allowStackOpt)
{
solAssert(!m_context->variableStackHeights.count(&var), "");
solAssert(!m_context->variableReferences.count(&var), "");
m_stackAdjustment++;
}
else
m_assembly.appendInstruction(solidity::Instruction::POP);
}
int deposit = m_assembly.stackHeight() - blockStartStackHeight; int deposit = m_assembly.stackHeight() - blockStartStackHeight;
solAssert(deposit == 0, "Invalid stack height at end of block."); solAssert(deposit == 0, "Invalid stack height at end of block: " + to_string(deposit));
checkStackHeight(&_block); checkStackHeight(&_block);
} }
@ -502,13 +697,13 @@ void CodeTransform::generateMultiAssignment(vector<Identifier> const& _variableN
void CodeTransform::generateAssignment(Identifier const& _variableName) void CodeTransform::generateAssignment(Identifier const& _variableName)
{ {
solAssert(m_scope, ""); solAssert(m_scope, "");
auto var = m_scope->lookup(_variableName.name); if (auto var = m_scope->lookup(_variableName.name))
if (var)
{ {
Scope::Variable const& _var = boost::get<Scope::Variable>(*var); Scope::Variable const& _var = boost::get<Scope::Variable>(*var);
if (int heightDiff = variableHeightDiff(_var, true)) if (int heightDiff = variableHeightDiff(_var, true))
m_assembly.appendInstruction(solidity::swapInstruction(heightDiff - 1)); m_assembly.appendInstruction(solidity::swapInstruction(heightDiff - 1));
m_assembly.appendInstruction(solidity::Instruction::POP); m_assembly.appendInstruction(solidity::Instruction::POP);
decreaseReference(_variableName.name, _var);
} }
else else
{ {

View File

@ -20,6 +20,7 @@
#include <libyul/backends/evm/EVMAssembly.h> #include <libyul/backends/evm/EVMAssembly.h>
#include <libyul/optimiser/ASTWalker.h>
#include <libyul/AsmDataForward.h> #include <libyul/AsmDataForward.h>
#include <libyul/AsmScope.h> #include <libyul/AsmScope.h>
@ -37,6 +38,46 @@ namespace yul
struct AsmAnalysisInfo; struct AsmAnalysisInfo;
class EVMAssembly; class EVMAssembly;
struct CodeTransformContext
{
std::map<Scope::Label const*, AbstractAssembly::LabelID> labelIDs;
std::map<Scope::Function const*, AbstractAssembly::LabelID> functionEntryIDs;
std::map<Scope::Variable const*, int> variableStackHeights;
std::map<Scope::Variable const*, unsigned> variableReferences;
};
/**
* Counts the number of references to a variable. This includes actual (read) references
* but also assignments to the variable. It does not include the declaration itself or
* function parameters, but it does include function return parameters.
*
* This component can handle multiple variables of the same name.
*
* Can only be applied to strict assembly.
*/
class VariableReferenceCounter: public yul::ASTWalker
{
public:
explicit VariableReferenceCounter(
CodeTransformContext& _context,
AsmAnalysisInfo const& _assemblyInfo
): m_context(_context), m_info(_assemblyInfo)
{}
public:
void operator()(Identifier const& _identifier);
void operator()(FunctionDefinition const&);
void operator()(ForLoop const&);
void operator()(Block const& _block);
private:
void increaseRefIfFound(YulString _variableName);
CodeTransformContext& m_context;
AsmAnalysisInfo const& m_info;
Scope* m_scope = nullptr;
};
class CodeTransform: public boost::static_visitor<> class CodeTransform: public boost::static_visitor<>
{ {
public: public:
@ -45,6 +86,8 @@ public:
CodeTransform( CodeTransform(
AbstractAssembly& _assembly, AbstractAssembly& _assembly,
AsmAnalysisInfo& _analysisInfo, AsmAnalysisInfo& _analysisInfo,
Block const& _block,
bool _allowStackOpt = false,
bool _yul = false, bool _yul = false,
bool _evm15 = false, bool _evm15 = false,
ExternalIdentifierAccess const& _identifierAccess = ExternalIdentifierAccess(), ExternalIdentifierAccess const& _identifierAccess = ExternalIdentifierAccess(),
@ -52,43 +95,42 @@ public:
): CodeTransform( ): CodeTransform(
_assembly, _assembly,
_analysisInfo, _analysisInfo,
_block,
_allowStackOpt,
_yul, _yul,
_evm15, _evm15,
_identifierAccess, _identifierAccess,
_useNamedLabelsForFunctions, _useNamedLabelsForFunctions,
_assembly.stackHeight(), _assembly.stackHeight(),
std::make_shared<Context>() nullptr
) )
{ {
} }
protected: protected:
struct Context using Context = CodeTransformContext;
{
std::map<Scope::Label const*, AbstractAssembly::LabelID> labelIDs;
std::map<Scope::Function const*, AbstractAssembly::LabelID> functionEntryIDs;
std::map<Scope::Variable const*, int> variableStackHeights;
};
CodeTransform( CodeTransform(
AbstractAssembly& _assembly, AbstractAssembly& _assembly,
AsmAnalysisInfo& _analysisInfo, AsmAnalysisInfo& _analysisInfo,
Block const& _block,
bool _allowStackOpt,
bool _yul, bool _yul,
bool _evm15, bool _evm15,
ExternalIdentifierAccess const& _identifierAccess, ExternalIdentifierAccess const& _identifierAccess,
bool _useNamedLabelsForFunctions, bool _useNamedLabelsForFunctions,
int _stackAdjustment, int _stackAdjustment,
std::shared_ptr<Context> _context std::shared_ptr<Context> _context
): );
m_assembly(_assembly),
m_info(_analysisInfo), void decreaseReference(YulString _name, Scope::Variable const& _var);
m_yul(_yul), bool unreferenced(Scope::Variable const& _var) const;
m_evm15(_evm15), /// Marks slots of variables that are not used anymore
m_useNamedLabelsForFunctions(_useNamedLabelsForFunctions), /// and were defined in the current scope for reuse.
m_identifierAccess(_identifierAccess), /// Also POPs unused topmost stack slots.
m_stackAdjustment(_stackAdjustment), void freeUnusedVariables();
m_context(_context) /// Marks the stack slot of @a _var to be reused.
{} void deleteVariable(Scope::Variable const& _var);
public: public:
void operator()(Instruction const& _instruction); void operator()(Instruction const& _instruction);
@ -137,6 +179,7 @@ private:
AbstractAssembly& m_assembly; AbstractAssembly& m_assembly;
AsmAnalysisInfo& m_info; AsmAnalysisInfo& m_info;
Scope* m_scope = nullptr; Scope* m_scope = nullptr;
bool const m_allowStackOpt = true;
bool m_yul = false; bool m_yul = false;
bool m_evm15 = false; bool m_evm15 = false;
bool m_useNamedLabelsForFunctions = false; bool m_useNamedLabelsForFunctions = false;
@ -147,6 +190,12 @@ private:
/// (EVM 1.0 or 1.5). /// (EVM 1.0 or 1.5).
int m_stackAdjustment = 0; int m_stackAdjustment = 0;
std::shared_ptr<Context> m_context; std::shared_ptr<Context> m_context;
/// Set of variables whose reference counter has reached zero,
/// and whose stack slot will be marked as unused once we reach
/// statement level in the scope where the variable was defined.
std::set<Scope::Variable const*> m_variablesScheduledForDeletion;
std::set<int> m_unusedStackSlots;
}; };
} }

View File

@ -27,13 +27,13 @@
using namespace yul; using namespace yul;
using namespace std; using namespace std;
void EVMObjectCompiler::compile(Object& _object, AbstractAssembly& _assembly, bool _yul, bool _evm15) void EVMObjectCompiler::compile(Object& _object, AbstractAssembly& _assembly, bool _yul, bool _evm15, bool _optimize)
{ {
EVMObjectCompiler compiler(_assembly, _yul, _evm15); EVMObjectCompiler compiler(_assembly, _yul, _evm15);
compiler.run(_object); compiler.run(_object, _optimize);
} }
void EVMObjectCompiler::run(Object& _object) void EVMObjectCompiler::run(Object& _object, bool _optimize)
{ {
map<YulString, AbstractAssembly::SubID> subIDs; map<YulString, AbstractAssembly::SubID> subIDs;
@ -42,7 +42,7 @@ void EVMObjectCompiler::run(Object& _object)
{ {
auto subAssemblyAndID = m_assembly.createSubAssembly(); auto subAssemblyAndID = m_assembly.createSubAssembly();
subIDs[subObject->name] = subAssemblyAndID.second; subIDs[subObject->name] = subAssemblyAndID.second;
compile(*subObject, *subAssemblyAndID.first, m_yul, m_evm15); compile(*subObject, *subAssemblyAndID.first, m_yul, m_evm15, _optimize);
} }
else else
{ {
@ -51,5 +51,6 @@ void EVMObjectCompiler::run(Object& _object)
} }
yulAssert(_object.analysisInfo, "No analysis info."); yulAssert(_object.analysisInfo, "No analysis info.");
CodeTransform{m_assembly, *_object.analysisInfo, m_yul, m_evm15}(*_object.code); yulAssert(_object.code, "No code.");
CodeTransform{m_assembly, *_object.analysisInfo, *_object.code, _optimize, m_yul, m_evm15}(*_object.code);
} }

View File

@ -27,13 +27,13 @@ class AbstractAssembly;
class EVMObjectCompiler class EVMObjectCompiler
{ {
public: public:
static void compile(Object& _object, AbstractAssembly& _assembly, bool _yul, bool _evm15); static void compile(Object& _object, AbstractAssembly& _assembly, bool _yul, bool _evm15, bool _optimize);
private: private:
EVMObjectCompiler(AbstractAssembly& _assembly, bool _yul, bool _evm15): EVMObjectCompiler(AbstractAssembly& _assembly, bool _yul, bool _evm15):
m_assembly(_assembly), m_yul(_yul), m_evm15(_evm15) m_assembly(_assembly), m_yul(_yul), m_evm15(_evm15)
{} {}
void run(Object& _object); void run(Object& _object, bool _optimize);
AbstractAssembly& m_assembly; AbstractAssembly& m_assembly;
bool m_yul = false; bool m_yul = false;

View File

@ -93,6 +93,16 @@ void ASTWalker::operator()(Block const& _block)
walkVector(_block.statements); walkVector(_block.statements);
} }
void ASTWalker::visit(Statement const& _st)
{
boost::apply_visitor(*this, _st);
}
void ASTWalker::visit(Expression const& _e)
{
boost::apply_visitor(*this, _e);
}
void ASTModifier::operator()(FunctionalInstruction& _instr) void ASTModifier::operator()(FunctionalInstruction& _instr)
{ {
walkVector(_instr.arguments | boost::adaptors::reversed); walkVector(_instr.arguments | boost::adaptors::reversed);
@ -155,3 +165,13 @@ void ASTModifier::operator()(Block& _block)
{ {
walkVector(_block.statements); walkVector(_block.statements);
} }
void ASTModifier::visit(Statement& _st)
{
boost::apply_visitor(*this, _st);
}
void ASTModifier::visit(Expression& _e)
{
boost::apply_visitor(*this, _e);
}

View File

@ -58,14 +58,8 @@ public:
virtual void operator()(ForLoop const&); virtual void operator()(ForLoop const&);
virtual void operator()(Block const& _block); virtual void operator()(Block const& _block);
virtual void visit(Statement const& _st) virtual void visit(Statement const& _st);
{ virtual void visit(Expression const& _e);
boost::apply_visitor(*this, _st);
}
virtual void visit(Expression const& _e)
{
boost::apply_visitor(*this, _e);
}
protected: protected:
template <class T> template <class T>
@ -99,14 +93,8 @@ public:
virtual void operator()(ForLoop&); virtual void operator()(ForLoop&);
virtual void operator()(Block& _block); virtual void operator()(Block& _block);
virtual void visit(Statement& _st) virtual void visit(Statement& _st);
{ virtual void visit(Expression& _e);
boost::apply_visitor(*this, _st);
}
virtual void visit(Expression& _e)
{
boost::apply_visitor(*this, _e);
}
protected: protected:
template <class T> template <class T>