mirror of
https://github.com/ethereum/solidity
synced 2023-10-03 13:03:40 +00:00
104 lines
2.9 KiB
C++
104 lines
2.9 KiB
C++
/*
|
|
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/>.
|
|
*/
|
|
// SPDX-License-Identifier: GPL-3.0
|
|
#pragma once
|
|
|
|
#include <libyul/optimiser/Metrics.h>
|
|
#include <libyul/optimiser/NameDispenser.h>
|
|
#include <libyul/AsmData.h>
|
|
#include <libyul/Dialect.h>
|
|
#include <libyul/Exceptions.h>
|
|
|
|
#include <liblangutil/SourceLocation.h>
|
|
|
|
#include <libsolutil/CommonData.h>
|
|
|
|
#include <variant>
|
|
|
|
namespace solidity::yul::unusedFunctionsCommon
|
|
{
|
|
|
|
template<typename T>
|
|
std::vector<T> filter(std::vector<T> const& _vec, std::vector<bool> const& _mask)
|
|
{
|
|
yulAssert(_vec.size() == _mask.size(), "");
|
|
|
|
std::vector<T> ret;
|
|
|
|
for (size_t i = 0; i < _mask.size(); ++i)
|
|
if (_mask[i])
|
|
ret.push_back(_vec[i]);
|
|
|
|
return ret;
|
|
}
|
|
|
|
/// Returns true if applying UnusedFunctionParameterPruner is not helpful or redundant because the
|
|
/// inliner will be able to handle it anyway.
|
|
bool tooSimpleToBePruned(FunctionDefinition const& _f)
|
|
{
|
|
return _f.body.statements.size() <= 1 && CodeSize::codeSize(_f.body) <= 1;
|
|
}
|
|
|
|
FunctionDefinition createLinkingFunction(
|
|
FunctionDefinition const& _original,
|
|
std::pair<std::vector<bool>, std::vector<bool>> const& _usedParametersAndReturns,
|
|
YulString const& _originalFunctionName,
|
|
YulString const& _linkingFunctionName,
|
|
NameDispenser& _nameDispenser
|
|
)
|
|
{
|
|
auto generateTypedName = [&](TypedName t)
|
|
{
|
|
return TypedName{
|
|
t.location,
|
|
_nameDispenser.newName(t.name),
|
|
t.type
|
|
};
|
|
};
|
|
|
|
langutil::SourceLocation loc = _original.location;
|
|
|
|
FunctionDefinition linkingFunction{
|
|
loc,
|
|
_linkingFunctionName,
|
|
util::applyMap(_original.parameters, generateTypedName),
|
|
util::applyMap(_original.returnVariables, generateTypedName),
|
|
{loc, {}} // body
|
|
};
|
|
|
|
FunctionCall call{loc, Identifier{loc, _originalFunctionName}, {}};
|
|
for (auto const& p: filter(linkingFunction.parameters, _usedParametersAndReturns.first))
|
|
call.arguments.emplace_back(Identifier{loc, p.name});
|
|
|
|
Assignment assignment{loc, {}, nullptr};
|
|
|
|
for (auto const& r: filter(linkingFunction.returnVariables, _usedParametersAndReturns.second))
|
|
assignment.variableNames.emplace_back(Identifier{loc, r.name});
|
|
|
|
if (assignment.variableNames.empty())
|
|
linkingFunction.body.statements.emplace_back(ExpressionStatement{loc, std::move(call)});
|
|
else
|
|
{
|
|
assignment.value = std::make_unique<Expression>(std::move(call));
|
|
linkingFunction.body.statements.emplace_back(std::move(assignment));
|
|
}
|
|
|
|
return linkingFunction;
|
|
}
|
|
|
|
}
|