Files
carbon-lang/toolchain/check/node_block_stack.h
T
Jon Ross-Perkins 1c748c0f14 Split semantics into check and sem_ir directories (#3176)
Continuing along with #3070. Note this is just a file rename, with BUILD
edits; every file previously in semantics/ should show as moved (except
maybe BUILDs, which split).
2023-08-31 19:54:32 +00:00

90 lines
2.8 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
#ifndef CARBON_TOOLCHAIN_CHECK_NODE_BLOCK_STACK_H_
#define CARBON_TOOLCHAIN_CHECK_NODE_BLOCK_STACK_H_
#include "llvm/ADT/SmallVector.h"
#include "toolchain/sem_ir/file.h"
#include "toolchain/sem_ir/node.h"
namespace Carbon::Check {
// Wraps the stack of node blocks for Context.
//
// All pushes and pops will be vlogged.
class NodeBlockStack {
public:
explicit NodeBlockStack(llvm::StringLiteral name, SemIR::File& semantics_ir,
llvm::raw_ostream* vlog_stream)
: name_(name), semantics_ir_(&semantics_ir), vlog_stream_(vlog_stream) {}
// Pushes an existing node block.
auto Push(SemIR::NodeBlockId id) -> void;
// Pushes a new node block. It will be invalid unless PeekForAdd is called in
// order to support lazy allocation.
auto Push() -> void { Push(SemIR::NodeBlockId::Invalid); }
// Pushes a new unreachable code block.
auto PushUnreachable() -> void { Push(SemIR::NodeBlockId::Unreachable); }
// Allocates and pushes a new node block.
auto PushForAdd() -> SemIR::NodeBlockId {
Push();
return PeekForAdd();
}
// Peeks at the top node block. This does not trigger lazy allocation, so the
// returned node block may be invalid.
auto Peek() -> SemIR::NodeBlockId {
CARBON_CHECK(!stack_.empty()) << "no current block";
return stack_.back();
}
// Returns the top node block, allocating one if it's still invalid.
auto PeekForAdd() -> SemIR::NodeBlockId;
// Pops the top node block. This will always return a valid node block;
// SemIR::NodeBlockId::Empty is returned if one wasn't allocated.
auto Pop() -> SemIR::NodeBlockId;
// Pops the top node block, ensuring that it is lazily allocated if it's
// empty. For use when more nodes will be added to the block later.
auto PopForAdd() -> SemIR::NodeBlockId {
PeekForAdd();
return Pop();
}
// Returns whether the current block is statically reachable.
auto is_current_block_reachable() -> bool {
return Peek() != SemIR::NodeBlockId::Unreachable;
}
// Prints the stack for a stack dump.
auto PrintForStackDump(llvm::raw_ostream& output) const -> void;
auto empty() const -> bool { return stack_.empty(); }
auto size() const -> size_t { return stack_.size(); }
private:
// A name for debugging.
llvm::StringLiteral name_;
// The underlying SemIR::File instance. Always non-null.
SemIR::File* semantics_ir_;
// Whether to print verbose output.
llvm::raw_ostream* vlog_stream_;
// The actual stack.
// PushEntry and PopEntry control modification in order to centralize
// vlogging.
llvm::SmallVector<SemIR::NodeBlockId> stack_;
};
} // namespace Carbon::Check
#endif // CARBON_TOOLCHAIN_CHECK_NODE_BLOCK_STACK_H_