mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-08-25 13:12:00 +00:00
dec0c62997
Avoid linearly searching for prefix_i. This becomes very slow when adding the same prefix many times. Instead, cache each prefix and the last index that was assigned, and start searching from there. Note that this locally speeds up MaxLimitTests.MaxBufferBindingSize on GL backends from about 10 minutes to a few seconds. This test creates a very large struct, and Tint's GLES backend runs the PadStructs transform, which produces a struct with 16384 members. Creating these members triggers the behaviour described above when creating names for each member. Also replaced std::unordered_maps with utils::Hashmap. Change-Id: I86678a049ac229ce617d500971e9c077ff5c10a1 Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/121440 Reviewed-by: Ben Clayton <bclayton@google.com> Commit-Queue: Antonio Maiorano <amaiorano@google.com> Kokoro: Kokoro <noreply+kokoro@google.com>
103 lines
3.5 KiB
C++
103 lines
3.5 KiB
C++
// Copyright 2020 The Tint Authors.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#ifndef SRC_TINT_SYMBOL_TABLE_H_
|
|
#define SRC_TINT_SYMBOL_TABLE_H_
|
|
|
|
#include <string>
|
|
#include "utils/hashmap.h"
|
|
|
|
#include "src/tint/symbol.h"
|
|
|
|
namespace tint {
|
|
|
|
/// Holds mappings from symbols to their associated string names
|
|
class SymbolTable {
|
|
public:
|
|
/// Constructor
|
|
/// @param program_id the identifier of the program that owns this symbol
|
|
/// table
|
|
explicit SymbolTable(tint::ProgramID program_id);
|
|
/// Copy constructor
|
|
SymbolTable(const SymbolTable&);
|
|
/// Move Constructor
|
|
SymbolTable(SymbolTable&&);
|
|
/// Destructor
|
|
~SymbolTable();
|
|
|
|
/// Copy assignment
|
|
/// @param other the symbol table to copy
|
|
/// @returns the new symbol table
|
|
SymbolTable& operator=(const SymbolTable& other);
|
|
/// Move assignment
|
|
/// @param other the symbol table to move
|
|
/// @returns the symbol table
|
|
SymbolTable& operator=(SymbolTable&& other);
|
|
|
|
/// Registers a name into the symbol table, returning the Symbol.
|
|
/// @param name the name to register
|
|
/// @returns the symbol representing the given name
|
|
Symbol Register(const std::string& name);
|
|
|
|
/// Returns the symbol for the given `name`
|
|
/// @param name the name to lookup
|
|
/// @returns the symbol for the name or Symbol() if not found.
|
|
Symbol Get(const std::string& name) const;
|
|
|
|
/// Returns the name for the given symbol
|
|
/// @param symbol the symbol to retrieve the name for
|
|
/// @returns the symbol name or "" if not found
|
|
std::string NameFor(const Symbol symbol) const;
|
|
|
|
/// Returns a new unique symbol with the given name, possibly suffixed with a
|
|
/// unique number.
|
|
/// @param name the symbol name
|
|
/// @returns a new, unnamed symbol with the given name. If the name is already
|
|
/// taken then this will be suffixed with an underscore and a unique numerical
|
|
/// value
|
|
Symbol New(std::string name = "");
|
|
|
|
/// Foreach calls the callback function `F` for each symbol in the table.
|
|
/// @param callback must be a function or function-like object with the
|
|
/// signature: `void(Symbol, const std::string&)`
|
|
template <typename F>
|
|
void Foreach(F&& callback) const {
|
|
for (auto it : symbol_to_name_) {
|
|
callback(it.key, it.value);
|
|
}
|
|
}
|
|
|
|
/// @returns the identifier of the Program that owns this symbol table.
|
|
tint::ProgramID ProgramID() const { return program_id_; }
|
|
|
|
private:
|
|
// The value to be associated to the next registered symbol table entry.
|
|
uint32_t next_symbol_ = 1;
|
|
|
|
utils::Hashmap<Symbol, std::string, 0> symbol_to_name_;
|
|
utils::Hashmap<std::string, Symbol, 0> name_to_symbol_;
|
|
utils::Hashmap<std::string, size_t, 0> last_prefix_to_index_;
|
|
tint::ProgramID program_id_;
|
|
};
|
|
|
|
/// @param symbol_table the SymbolTable
|
|
/// @returns the ProgramID that owns the given SymbolTable
|
|
inline ProgramID ProgramIDOf(const SymbolTable& symbol_table) {
|
|
return symbol_table.ProgramID();
|
|
}
|
|
|
|
} // namespace tint
|
|
|
|
#endif // SRC_TINT_SYMBOL_TABLE_H_
|