mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-11-15 21:28:10 +00:00
Richard Smith
85d5b40429
- `Peek()` no longer returns a useful value, because we often don't allocate a `NodeBlockId` until we finish building the block. Remove it and update both its callers. - Rename `PeekForAdd()` to `PeekOrAdd()` since it's no longer used to get a block ID to add elements into. - `PushForAdd()` was unused. Remove it. - Add `NodeBlockStack::AddNode` to combine the operations of adding a node and inserting it into a block.
70 lines
1.9 KiB
C++
70 lines
1.9 KiB
C++
// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
|
|
// Exceptions. See /LICENSE for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
#include "toolchain/check/node_block_stack.h"
|
|
|
|
#include "common/vlog.h"
|
|
#include "llvm/ADT/STLExtras.h"
|
|
#include "llvm/ADT/StringExtras.h"
|
|
#include "toolchain/sem_ir/node.h"
|
|
|
|
namespace Carbon::Check {
|
|
|
|
auto NodeBlockStack::Push(SemIR::NodeBlockId id) -> void {
|
|
CARBON_VLOG() << name_ << " Push " << size_ << "\n";
|
|
CARBON_CHECK(size_ < (1 << 20))
|
|
<< "Excessive stack size: likely infinite loop";
|
|
if (size_ == static_cast<int>(stack_.size())) {
|
|
stack_.emplace_back();
|
|
}
|
|
stack_[size_].Reset(id);
|
|
++size_;
|
|
}
|
|
|
|
auto NodeBlockStack::PeekOrAdd(int depth) -> SemIR::NodeBlockId {
|
|
CARBON_CHECK(size() > depth) << "no such block";
|
|
int index = size() - depth - 1;
|
|
auto& slot = stack_[index];
|
|
if (!slot.id.is_valid()) {
|
|
slot.id = semantics_ir_->AddNodeBlockId();
|
|
}
|
|
return slot.id;
|
|
}
|
|
|
|
auto NodeBlockStack::Pop() -> SemIR::NodeBlockId {
|
|
CARBON_CHECK(!empty()) << "no current block";
|
|
--size_;
|
|
auto& back = stack_[size_];
|
|
|
|
// Finalize the block.
|
|
if (!back.content.empty() && back.id != SemIR::NodeBlockId::Unreachable) {
|
|
if (back.id.is_valid()) {
|
|
semantics_ir_->SetNodeBlock(back.id, back.content);
|
|
} else {
|
|
back.id = semantics_ir_->AddNodeBlock(back.content);
|
|
}
|
|
}
|
|
|
|
CARBON_VLOG() << name_ << " Pop " << size_ << ": " << back.id << "\n";
|
|
if (!back.id.is_valid()) {
|
|
return SemIR::NodeBlockId::Empty;
|
|
}
|
|
return back.id;
|
|
}
|
|
|
|
auto NodeBlockStack::PrintForStackDump(llvm::raw_ostream& output) const
|
|
-> void {
|
|
output << name_ << ":\n";
|
|
for (const auto& [i, entry] : llvm::enumerate(stack_)) {
|
|
output << "\t" << i << ".\t" << entry.id << "\t{";
|
|
llvm::ListSeparator sep;
|
|
for (auto id : entry.content) {
|
|
output << sep << id;
|
|
}
|
|
output << "}\n";
|
|
}
|
|
}
|
|
|
|
} // namespace Carbon::Check
|