carbon-lang/toolchain/semantics/semantics_node_block_stack.cpp

// 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/semantics/semantics_node_block_stack.h"

#include "common/vlog.h"
#include "toolchain/semantics/semantics_node.h"

namespace Carbon {

auto SemanticsNodeBlockStack::Push() -> void {
  CARBON_VLOG() << "NodeBlock Push " << stack_.size() << "\n";
  CARBON_CHECK(stack_.size() < (1 << 20))
      << "Excessive stack size: likely infinite loop";
  stack_.push_back(SemanticsNodeBlockId::Invalid);
}

auto SemanticsNodeBlockStack::PushWithUnconditionalAlloc()
    -> SemanticsNodeBlockId {
  SemanticsNodeBlockId block_id(node_blocks_->size());
  CARBON_VLOG() << "NodeBlock Push " << stack_.size() << ": " << block_id
                << "\n";
  CARBON_CHECK(stack_.size() < (1 << 20))
      << "Excessive stack size: likely infinite loop";
  node_blocks_->resize(block_id.index + 1);
  stack_.push_back(block_id);
  return block_id;
}

auto SemanticsNodeBlockStack::PeekForAdd() -> SemanticsNodeBlockId {
  auto& back = stack_.back();
  if (!back.is_valid()) {
    SemanticsNodeBlockId block_id(node_blocks_->size());
    node_blocks_->resize(block_id.index + 1);
    back = block_id;
    CARBON_VLOG() << "NodeBlock Add " << stack_.size() - 1 << ": " << back
                  << "\n";
  }
  return back;
}

auto SemanticsNodeBlockStack::Pop() -> SemanticsNodeBlockId {
  auto back = stack_.pop_back_val();
  CARBON_VLOG() << "NodeBlock Pop " << stack_.size() << ": " << back << "\n";
  if (!back.is_valid()) {
    return SemanticsNodeBlockId::Empty;
  }
  return back;
}

auto SemanticsNodeBlockStack::PrintForStackDump(llvm::raw_ostream& output) const
    -> void {
  output << "SemanticsNodeBlockStack:\n";
  for (int i = 0; i < static_cast<int>(stack_.size()); ++i) {
    output << "\t" << i << ".\t" << stack_[i] << "\n";
  }
}

}  // namespace Carbon