solidity/libyul/optimiser/InlinableExpressionFunctionFinder.cpp

70 lines
2.4 KiB
C++
Raw Normal View History

2017-11-23 16:42:06 +00:00
/*
This file is part of solidity.
solidity is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
solidity is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with solidity. If not, see <http://www.gnu.org/licenses/>.
*/
/**
* Optimiser component that identifies functions to be inlined.
*/
2018-10-15 09:52:35 +00:00
#include <libyul/optimiser/InlinableExpressionFunctionFinder.h>
2017-11-23 16:42:06 +00:00
2018-10-15 09:52:35 +00:00
#include <libyul/optimiser/Utilities.h>
#include <libyul/AsmData.h>
2017-11-23 16:42:06 +00:00
using namespace std;
using namespace dev;
using namespace yul;
2017-11-23 16:42:06 +00:00
2018-02-06 14:30:36 +00:00
void InlinableExpressionFunctionFinder::operator()(Identifier const& _identifier)
2017-11-23 16:42:06 +00:00
{
checkAllowed(_identifier.name);
ASTWalker::operator()(_identifier);
}
2018-02-06 14:30:36 +00:00
void InlinableExpressionFunctionFinder::operator()(FunctionCall const& _funCall)
2017-11-23 16:42:06 +00:00
{
checkAllowed(_funCall.functionName.name);
ASTWalker::operator()(_funCall);
}
2018-02-06 14:30:36 +00:00
void InlinableExpressionFunctionFinder::operator()(FunctionDefinition const& _function)
2017-11-23 16:42:06 +00:00
{
if (_function.returnVariables.size() == 1 && _function.body.statements.size() == 1)
{
YulString retVariable = _function.returnVariables.front().name;
2017-11-23 16:42:06 +00:00
Statement const& bodyStatement = _function.body.statements.front();
if (bodyStatement.type() == typeid(Assignment))
{
Assignment const& assignment = boost::get<Assignment>(bodyStatement);
if (assignment.variableNames.size() == 1 && assignment.variableNames.front().name == retVariable)
{
2017-11-22 10:11:48 +00:00
// FIXME: use code size metric here
2017-11-23 16:42:06 +00:00
// We cannot overwrite previous settings, because this function definition
// would not be valid here if we were searching inside a functionally inlinable
// function body.
2018-05-09 09:43:14 +00:00
assertThrow(m_disallowedIdentifiers.empty() && !m_foundDisallowedIdentifier, OptimizerException, "");
m_disallowedIdentifiers = set<YulString>{retVariable, _function.name};
2017-11-23 16:42:06 +00:00
boost::apply_visitor(*this, *assignment.value);
if (!m_foundDisallowedIdentifier)
m_inlinableFunctions[_function.name] = &_function;
m_disallowedIdentifiers.clear();
m_foundDisallowedIdentifier = false;
}
}
}
ASTWalker::operator()(_function.body);
}