blob: 650837d7aaf241b4475a1aa95c9c3358e459fb71 [file] [log] [blame]
/*
* Copyright 2021 Google LLC.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "src/sksl/SkSLMangler.h"
#include "include/core/SkString.h"
#include "include/core/SkTypes.h"
#include "src/base/SkStringView.h"
#include "src/sksl/ir/SkSLSymbolTable.h"
#include <algorithm>
#include <cstring>
#include <ctype.h>
namespace SkSL {
std::string Mangler::uniqueName(std::string_view baseName, SymbolTable* symbolTable) {
SkASSERT(symbolTable);
// Private names might begin with a $. Strip that off.
if (skstd::starts_with(baseName, '$')) {
baseName.remove_prefix(1);
}
// The inliner runs more than once, so the base name might already have been mangled and have a
// prefix like "_123_x". Let's strip that prefix off to make the generated code easier to read.
if (skstd::starts_with(baseName, '_')) {
// Determine if we have a string of digits.
int offset = 1;
while (isdigit(baseName[offset])) {
++offset;
}
// If we found digits, another underscore, and anything else, that's the mangler prefix.
// Strip it off.
if (offset > 1 && baseName[offset] == '_' && baseName[offset + 1] != '\0') {
baseName.remove_prefix(offset + 1);
} else {
// This name doesn't contain a mangler prefix, but it does start with an underscore.
// OpenGL disallows two consecutive underscores anywhere in the string, and we'll be
// adding one as part of the mangler prefix, so strip the leading underscore.
baseName.remove_prefix(1);
}
}
// Append a unique numeric prefix to avoid name overlap. Check the symbol table to make sure
// we're not reusing an existing name. (Note that within a single compilation pass, this check
// isn't fully comprehensive, as code isn't always generated in top-to-bottom order.)
// This code is a performance hotspot. Assemble the string manually to save a few cycles.
char uniqueName[256];
uniqueName[0] = '_';
char* uniqueNameEnd = uniqueName + std::size(uniqueName);
for (;;) {
// _123
char* endPtr = SkStrAppendS32(uniqueName + 1, fCounter++);
// _123_
*endPtr++ = '_';
// _123_baseNameTruncatedToFit (no null terminator, because string_view doesn't require one)
int baseNameCopyLength = std::min<int>(baseName.size(), uniqueNameEnd - endPtr);
memcpy(endPtr, baseName.data(), baseNameCopyLength);
endPtr += baseNameCopyLength;
std::string_view uniqueNameView(uniqueName, endPtr - uniqueName);
if (symbolTable->find(uniqueNameView) == nullptr) {
return std::string(uniqueNameView);
}
}
}
} // namespace SkSL