mirror of
https://github.com/gbdev/rgbds.git
synced 2025-11-21 18:52:07 +00:00
Use std::get_if instead of std::visit (#1367)
`std::visit` is (arguably) cleaner code, but older versions of gcc and clang (not very old; the ones packaged with Ubuntu 22.04 LTS) compile them as tables of function pointers, instead of efficient jump tables.
This commit is contained in:
@@ -62,28 +62,27 @@ std::string const &FileStackNode::name() const {
|
||||
}
|
||||
|
||||
std::string const &FileStackNode::dump(uint32_t curLineNo) const {
|
||||
Visitor visitor{
|
||||
[this](std::vector<uint32_t> const &iters) -> std::string const & {
|
||||
assert(this->parent); // REPT nodes use their parent's name
|
||||
std::string const &lastName = this->parent->dump(this->lineNo);
|
||||
fprintf(stderr, " -> %s", lastName.c_str());
|
||||
for (uint32_t i = iters.size(); i--;)
|
||||
fprintf(stderr, "::REPT~%" PRIu32, iters[i]);
|
||||
return lastName;
|
||||
},
|
||||
[this](std::string const &name) -> std::string const & {
|
||||
if (this->parent) {
|
||||
this->parent->dump(this->lineNo);
|
||||
fprintf(stderr, " -> %s", name.c_str());
|
||||
} else {
|
||||
fputs(name.c_str(), stderr);
|
||||
}
|
||||
return name;
|
||||
},
|
||||
};
|
||||
std::string const &topName = std::visit(visitor, data);
|
||||
fprintf(stderr, "(%" PRIu32 ")", curLineNo);
|
||||
return topName;
|
||||
if (std::holds_alternative<std::vector<uint32_t>>(data)) {
|
||||
assert(parent); // REPT nodes use their parent's name
|
||||
std::string const &lastName = parent->dump(lineNo);
|
||||
fputs(" -> ", stderr);
|
||||
fputs(lastName.c_str(), stderr);
|
||||
std::vector<uint32_t> const &nodeIters = iters();
|
||||
for (uint32_t i = nodeIters.size(); i--;) {
|
||||
fprintf(stderr, "::REPT~%" PRIu32, nodeIters[i]);
|
||||
}
|
||||
fprintf(stderr, "(%" PRIu32 ")", curLineNo);
|
||||
return lastName;
|
||||
} else {
|
||||
if (parent) {
|
||||
parent->dump(lineNo);
|
||||
fputs(" -> ", stderr);
|
||||
}
|
||||
std::string const &nodeName = name();
|
||||
fputs(nodeName.c_str(), stderr);
|
||||
fprintf(stderr, "(%" PRIu32 ")", curLineNo);
|
||||
return nodeName;
|
||||
}
|
||||
}
|
||||
|
||||
void fstk_DumpCurrent() {
|
||||
|
||||
@@ -449,14 +449,11 @@ void lexer_OpenFileView(
|
||||
}
|
||||
|
||||
void lexer_RestartRept(uint32_t lineNo) {
|
||||
std::visit(
|
||||
Visitor{
|
||||
[](MmappedLexerState &mmap) { mmap.offset = 0; },
|
||||
[](ViewedLexerState &view) { view.offset = 0; },
|
||||
[](auto &) {},
|
||||
},
|
||||
lexerState->content
|
||||
);
|
||||
if (auto *mmap = std::get_if<MmappedLexerState>(&lexerState->content); mmap) {
|
||||
mmap->offset = 0;
|
||||
} else if (auto *view = std::get_if<ViewedLexerState>(&lexerState->content); view) {
|
||||
view->offset = 0;
|
||||
}
|
||||
initState(*lexerState);
|
||||
lexerState->lineNo = lineNo;
|
||||
}
|
||||
@@ -475,17 +472,12 @@ void lexer_CleanupState(LexerState &state) {
|
||||
// `lexerStateEOL`, but there's currently no situation in which this should happen.
|
||||
assert(&state != lexerStateEOL);
|
||||
|
||||
std::visit(
|
||||
Visitor{
|
||||
[](MmappedLexerState &mmap) {
|
||||
if (!mmap.isReferenced)
|
||||
munmap(mmap.ptr, mmap.size);
|
||||
},
|
||||
[](BufferedLexerState &cbuf) { close(cbuf.fd); },
|
||||
[](auto &) {},
|
||||
},
|
||||
state.content
|
||||
);
|
||||
if (auto *mmap = std::get_if<MmappedLexerState>(&state.content); mmap) {
|
||||
if (!mmap->isReferenced)
|
||||
munmap(mmap->ptr, mmap->size);
|
||||
} else if (auto *cbuf = std::get_if<BufferedLexerState>(&state.content); cbuf) {
|
||||
close(cbuf->fd);
|
||||
}
|
||||
}
|
||||
|
||||
void lexer_SetMode(LexerMode mode) {
|
||||
@@ -654,57 +646,52 @@ static int peekInternal(uint8_t distance) {
|
||||
LEXER_BUF_SIZE
|
||||
);
|
||||
|
||||
return std::visit(
|
||||
Visitor{
|
||||
[&distance](MmappedLexerState &mmap) -> int {
|
||||
if (size_t idx = mmap.offset + distance; idx < mmap.size)
|
||||
return (uint8_t)mmap.ptr[idx];
|
||||
return EOF;
|
||||
},
|
||||
[&distance](ViewedLexerState &view) -> int {
|
||||
if (size_t idx = view.offset + distance; idx < view.size)
|
||||
return (uint8_t)view.ptr[idx];
|
||||
return EOF;
|
||||
},
|
||||
[&distance](BufferedLexerState &cbuf) -> int {
|
||||
if (cbuf.nbChars > distance)
|
||||
return (uint8_t)cbuf.buf[(cbuf.index + distance) % LEXER_BUF_SIZE];
|
||||
if (auto *mmap = std::get_if<MmappedLexerState>(&lexerState->content); mmap) {
|
||||
if (size_t idx = mmap->offset + distance; idx < mmap->size)
|
||||
return (uint8_t)mmap->ptr[idx];
|
||||
return EOF;
|
||||
} else if (auto *view = std::get_if<ViewedLexerState>(&lexerState->content); view) {
|
||||
if (size_t idx = view->offset + distance; idx < view->size)
|
||||
return (uint8_t)view->ptr[idx];
|
||||
return EOF;
|
||||
} else {
|
||||
assert(std::holds_alternative<BufferedLexerState>(lexerState->content));
|
||||
auto &cbuf = std::get<BufferedLexerState>(lexerState->content);
|
||||
|
||||
// Buffer isn't full enough, read some chars in
|
||||
size_t target = LEXER_BUF_SIZE - cbuf.nbChars; // Aim: making the buf full
|
||||
if (cbuf.nbChars > distance)
|
||||
return (uint8_t)cbuf.buf[(cbuf.index + distance) % LEXER_BUF_SIZE];
|
||||
|
||||
// Compute the index we'll start writing to
|
||||
size_t writeIndex = (cbuf.index + cbuf.nbChars) % LEXER_BUF_SIZE;
|
||||
// Buffer isn't full enough, read some chars in
|
||||
size_t target = LEXER_BUF_SIZE - cbuf.nbChars; // Aim: making the buf full
|
||||
|
||||
// If the range to fill passes over the buffer wrapping point, we need two reads
|
||||
if (writeIndex + target > LEXER_BUF_SIZE) {
|
||||
size_t nbExpectedChars = LEXER_BUF_SIZE - writeIndex;
|
||||
size_t nbReadChars = readInternal(cbuf, writeIndex, nbExpectedChars);
|
||||
// Compute the index we'll start writing to
|
||||
size_t writeIndex = (cbuf.index + cbuf.nbChars) % LEXER_BUF_SIZE;
|
||||
|
||||
cbuf.nbChars += nbReadChars;
|
||||
// If the range to fill passes over the buffer wrapping point, we need two reads
|
||||
if (writeIndex + target > LEXER_BUF_SIZE) {
|
||||
size_t nbExpectedChars = LEXER_BUF_SIZE - writeIndex;
|
||||
size_t nbReadChars = readInternal(cbuf, writeIndex, nbExpectedChars);
|
||||
|
||||
writeIndex += nbReadChars;
|
||||
if (writeIndex == LEXER_BUF_SIZE)
|
||||
writeIndex = 0;
|
||||
cbuf.nbChars += nbReadChars;
|
||||
|
||||
// If the read was incomplete, don't perform a second read
|
||||
target -= nbReadChars;
|
||||
if (nbReadChars < nbExpectedChars)
|
||||
target = 0;
|
||||
}
|
||||
if (target != 0)
|
||||
cbuf.nbChars += readInternal(cbuf, writeIndex, target);
|
||||
writeIndex += nbReadChars;
|
||||
if (writeIndex == LEXER_BUF_SIZE)
|
||||
writeIndex = 0;
|
||||
|
||||
if (cbuf.nbChars > distance)
|
||||
return (uint8_t)cbuf.buf[(cbuf.index + distance) % LEXER_BUF_SIZE];
|
||||
// If the read was incomplete, don't perform a second read
|
||||
target -= nbReadChars;
|
||||
if (nbReadChars < nbExpectedChars)
|
||||
target = 0;
|
||||
}
|
||||
if (target != 0)
|
||||
cbuf.nbChars += readInternal(cbuf, writeIndex, target);
|
||||
|
||||
// If there aren't enough chars even after refilling, give up
|
||||
return EOF;
|
||||
},
|
||||
[](std::monostate) -> int { return EOF; },
|
||||
},
|
||||
lexerState->content
|
||||
);
|
||||
if (cbuf.nbChars > distance)
|
||||
return (uint8_t)cbuf.buf[(cbuf.index + distance) % LEXER_BUF_SIZE];
|
||||
|
||||
// If there aren't enough chars even after refilling, give up
|
||||
return EOF;
|
||||
}
|
||||
}
|
||||
|
||||
// forward declarations for peek
|
||||
@@ -782,22 +769,20 @@ restart:
|
||||
} else {
|
||||
// Advance within the file contents
|
||||
lexerState->colNo++;
|
||||
std::visit(
|
||||
Visitor{
|
||||
[](MmappedLexerState &mmap) { mmap.offset++; },
|
||||
[](ViewedLexerState &view) { view.offset++; },
|
||||
[](BufferedLexerState &cbuf) {
|
||||
assert(cbuf.index < LEXER_BUF_SIZE);
|
||||
cbuf.index++;
|
||||
if (cbuf.index == LEXER_BUF_SIZE)
|
||||
cbuf.index = 0; // Wrap around if necessary
|
||||
assert(cbuf.nbChars > 0);
|
||||
cbuf.nbChars--;
|
||||
},
|
||||
[](std::monostate) {},
|
||||
},
|
||||
lexerState->content
|
||||
);
|
||||
if (auto *mmap = std::get_if<MmappedLexerState>(&lexerState->content); mmap) {
|
||||
mmap->offset++;
|
||||
} else if (auto *view = std::get_if<ViewedLexerState>(&lexerState->content); view) {
|
||||
view->offset++;
|
||||
} else {
|
||||
assert(std::holds_alternative<BufferedLexerState>(lexerState->content));
|
||||
auto &cbuf = std::get<BufferedLexerState>(lexerState->content);
|
||||
assert(cbuf.index < LEXER_BUF_SIZE);
|
||||
cbuf.index++;
|
||||
if (cbuf.index == LEXER_BUF_SIZE)
|
||||
cbuf.index = 0; // Wrap around if necessary
|
||||
assert(cbuf.nbChars > 0);
|
||||
cbuf.nbChars--;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -2225,13 +2210,16 @@ yy::parser::symbol_type yylex() {
|
||||
lexerState->lastToken = token.type;
|
||||
lexerState->atLineStart = token.type == T_(NEWLINE) || token.type == T_(EOB);
|
||||
|
||||
return std::visit(
|
||||
Visitor{
|
||||
[&token](std::monostate) { return yy::parser::symbol_type(token.type); },
|
||||
[&token](auto &&value) { return yy::parser::symbol_type(token.type, value); },
|
||||
},
|
||||
token.value
|
||||
);
|
||||
if (auto *numValue = std::get_if<uint32_t>(&token.value); numValue) {
|
||||
return yy::parser::symbol_type(token.type, *numValue);
|
||||
} else if (auto *stringValue = std::get_if<String>(&token.value); stringValue) {
|
||||
return yy::parser::symbol_type(token.type, *stringValue);
|
||||
} else if (auto *strValue = std::get_if<std::string>(&token.value); strValue) {
|
||||
return yy::parser::symbol_type(token.type, *strValue);
|
||||
} else {
|
||||
assert(std::holds_alternative<std::monostate>(token.value));
|
||||
return yy::parser::symbol_type(token.type);
|
||||
}
|
||||
}
|
||||
|
||||
static void startCapture(CaptureBody &capture) {
|
||||
@@ -2242,21 +2230,14 @@ static void startCapture(CaptureBody &capture) {
|
||||
lexerState->disableInterpolation = true;
|
||||
|
||||
capture.lineNo = lexer_GetLineNo();
|
||||
capture.body = std::visit(
|
||||
Visitor{
|
||||
[](MmappedLexerState &mmap) -> char const * {
|
||||
return lexerState->expansions.empty() ? &mmap.ptr[mmap.offset] : nullptr;
|
||||
},
|
||||
[](ViewedLexerState &view) -> char const * {
|
||||
return lexerState->expansions.empty() ? &view.ptr[view.offset] : nullptr;
|
||||
},
|
||||
[](auto &) -> char const * { return nullptr; },
|
||||
},
|
||||
lexerState->content
|
||||
);
|
||||
|
||||
if (capture.body == nullptr) {
|
||||
// Indicates to retrieve the capture buffer when done capturing
|
||||
if (auto *mmap = std::get_if<MmappedLexerState>(&lexerState->content);
|
||||
mmap && lexerState->expansions.empty()) {
|
||||
capture.body = &mmap->ptr[mmap->offset];
|
||||
} else if (auto *view = std::get_if<ViewedLexerState>(&lexerState->content);
|
||||
view && lexerState->expansions.empty()) {
|
||||
capture.body = &view->ptr[view->offset];
|
||||
} else {
|
||||
capture.body = nullptr; // Indicates to retrieve the capture buffer when done capturing
|
||||
assert(lexerState->captureBuf == nullptr);
|
||||
lexerState->captureBuf = new (std::nothrow) std::vector<char>();
|
||||
if (!lexerState->captureBuf)
|
||||
@@ -2342,7 +2323,7 @@ bool lexer_CaptureMacroBody(CaptureBody &capture) {
|
||||
startCapture(capture);
|
||||
|
||||
// If the file is `mmap`ed, we need not to unmap it to keep access to the macro
|
||||
if (MmappedLexerState *mmap = std::get_if<MmappedLexerState>(&lexerState->content); mmap)
|
||||
if (auto *mmap = std::get_if<MmappedLexerState>(&lexerState->content); mmap)
|
||||
mmap->isReferenced = true;
|
||||
|
||||
int c = EOF;
|
||||
|
||||
@@ -2689,14 +2689,12 @@ static std::string strfmt(
|
||||
} else if (argIndex >= args.size()) {
|
||||
// Will warn after formatting is done.
|
||||
str += '%';
|
||||
} else if (auto *n = std::get_if<uint32_t>(&args[argIndex]); n) {
|
||||
str.append(fmt.formatNumber(*n));
|
||||
} else {
|
||||
str.append(std::visit(
|
||||
Visitor{
|
||||
[&fmt](uint32_t n) { return fmt.formatNumber(n); },
|
||||
[&fmt](std::string const &s) { return fmt.formatString(s); },
|
||||
},
|
||||
args[argIndex]
|
||||
));
|
||||
assert(std::holds_alternative<std::string>(args[argIndex]));
|
||||
auto &s = std::get<std::string>(args[argIndex]);
|
||||
str.append(fmt.formatString(s));
|
||||
}
|
||||
|
||||
argIndex++;
|
||||
|
||||
@@ -61,21 +61,20 @@ static int32_t CallbackPC() {
|
||||
|
||||
int32_t Symbol::getValue() const {
|
||||
assert(std::holds_alternative<int32_t>(data) || std::holds_alternative<int32_t (*)()>(data));
|
||||
if (int32_t const *value = std::get_if<int32_t>(&data); value) {
|
||||
if (auto *value = std::get_if<int32_t>(&data); value) {
|
||||
return type == SYM_LABEL ? *value + getSection()->org : *value;
|
||||
}
|
||||
return getOutputValue();
|
||||
}
|
||||
|
||||
int32_t Symbol::getOutputValue() const {
|
||||
return std::visit(
|
||||
Visitor{
|
||||
[](int32_t value) -> int32_t { return value; },
|
||||
[](int32_t (*callback)()) -> int32_t { return callback(); },
|
||||
[](auto &) -> int32_t { return 0; },
|
||||
},
|
||||
data
|
||||
);
|
||||
if (auto *value = std::get_if<int32_t>(&data); value) {
|
||||
return *value;
|
||||
} else if (auto *callback = std::get_if<int32_t (*)()>(&data); callback) {
|
||||
return (*callback)();
|
||||
} else {
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
std::string_view *Symbol::getMacro() const {
|
||||
|
||||
Reference in New Issue
Block a user