mirror of
https://github.com/ethereum/solidity
synced 2023-10-03 13:03:40 +00:00
cf6fe5a777
The private function `illegalName` was converted into a public function. A new public getter function `usedNames()` returns a reference to `m_usedNames`. A public function reset that would change all `m_usedNames` to *only* the names that are used in the AST. Also resets the counter. This will be used before the final NameSimplifier step in the optimization phase. The first two functions were added so that `NameSimplifier` could use it instead of traversing the AST using `NameCollector` to find used names.
71 lines
2.1 KiB
C++
71 lines
2.1 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
|
|
/**
|
|
* Optimiser component that can create new unique names.
|
|
*/
|
|
#pragma once
|
|
|
|
#include <libyul/AsmDataForward.h>
|
|
|
|
#include <libyul/YulString.h>
|
|
|
|
#include <set>
|
|
|
|
namespace solidity::yul
|
|
{
|
|
struct Dialect;
|
|
|
|
/**
|
|
* Optimizer component that can be used to generate new names that
|
|
* do not conflict with existing names.
|
|
*
|
|
* Tries to keep names short and appends decimals to disambiguate.
|
|
*/
|
|
class NameDispenser
|
|
{
|
|
public:
|
|
/// Initialize the name dispenser with all the names used in the given AST.
|
|
explicit NameDispenser(Dialect const& _dialect, Block const& _ast, std::set<YulString> _reservedNames = {});
|
|
/// Initialize the name dispenser with the given used names.
|
|
explicit NameDispenser(Dialect const& _dialect, std::set<YulString> _usedNames);
|
|
|
|
/// @returns a currently unused name that should be similar to _nameHint.
|
|
YulString newName(YulString _nameHint);
|
|
|
|
/// Mark @a _name as used, i.e. the dispenser's newName function will not
|
|
/// return it.
|
|
void markUsed(YulString _name) { m_usedNames.insert(_name); }
|
|
|
|
std::set<YulString> const& usedNames() { return m_usedNames; }
|
|
|
|
/// Returns true if `_name` is either used or is a restricted identifier.
|
|
bool illegalName(YulString _name);
|
|
|
|
/// Resets `m_usedNames` with *only* the names that are used in the AST. Also resets value of
|
|
/// `m_counter` to zero.
|
|
void reset(Block const& _ast);
|
|
|
|
private:
|
|
Dialect const& m_dialect;
|
|
std::set<YulString> m_usedNames;
|
|
std::set<YulString> m_reservedNames;
|
|
size_t m_counter = 0;
|
|
};
|
|
|
|
}
|