Refactory the IR/builder relationship (#2342)

I'm thinking about how to handle multiple files, and I think the current IRFactory is useful as a file-focused thing. So shifting/renaming accordingly. (doing this in its own PR to make the history a little cleaner for git's move detection)
2026-10-05 22:02:55 +01:00 · 2022-10-25 12:55:26 -07:00
parent 5c3f48b0fb
commit d51ca271af
10 changed files with 67 additions and 80 deletions
@@ -0,0 +1,91 @@
+// 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_SEMANTICS_SEMANTICS_PARSE_TREE_HANDLER_H_
+#define CARBON_TOOLCHAIN_SEMANTICS_SEMANTICS_PARSE_TREE_HANDLER_H_
+
+#include "common/check.h"
+#include "toolchain/parser/parse_tree.h"
+#include "toolchain/semantics/semantics_ir.h"
+
+namespace Carbon {
+
+// Handles processing of a ParseTree for semantics.
+class SemanticsParseTreeHandler {
+ public:
+  // Stores references for work.
+  explicit SemanticsParseTreeHandler(const TokenizedBuffer& tokens,
+                                     const ParseTree& parse_tree,
+                                     SemanticsIR& semantics)
+      : tokens_(&tokens), parse_tree_(&parse_tree), semantics_(&semantics) {}
+
+  // Outputs the ParseTree information into SemanticsIR.
+  auto Build() -> void;
+
+ private:
+  struct TraversalStackEntry {
+    ParseTree::Node parse_node;
+    llvm::Optional<SemanticsNodeId> result_id;
+  };
+
+  auto Push(ParseTree::Node parse_node) -> void {
+    node_stack_.push_back({parse_node, llvm::None});
+  }
+
+  auto Push(ParseTree::Node parse_node, SemanticsNode node) -> void {
+    auto node_id = semantics_->AddNode(node);
+    node_stack_.push_back({parse_node, node_id});
+  }
+
+  auto Pop(ParseNodeKind pop_parse_kind) -> void {
+    auto back = node_stack_.back();
+    auto parse_kind = parse_tree_->node_kind(back.parse_node);
+    CARBON_CHECK(parse_kind == pop_parse_kind)
+        << "Expected " << pop_parse_kind << ", found " << parse_kind;
+    CARBON_CHECK(!back.result_id) << "Expected no result ID on " << parse_kind;
+    node_stack_.pop_back();
+  }
+
+  auto PopWithResult() -> SemanticsNodeId {
+    auto back = node_stack_.back();
+    auto node_id = *back.result_id;
+    node_stack_.pop_back();
+    return node_id;
+  }
+
+  auto PopWithResult(ParseNodeKind pop_parse_kind) -> SemanticsNodeId {
+    auto back = node_stack_.back();
+    auto parse_kind = parse_tree_->node_kind(back.parse_node);
+    auto node_id = *back.result_id;
+    CARBON_CHECK(parse_kind == pop_parse_kind)
+        << "Expected " << pop_parse_kind << ", found " << parse_kind;
+    node_stack_.pop_back();
+    return node_id;
+  }
+
+  // Parse node handlers.
+  auto HandleDeclaredName(ParseTree::Node parse_node) -> void;
+  auto HandleFunctionDefinition(ParseTree::Node parse_node) -> void;
+  auto HandleFunctionDefinitionStart(ParseTree::Node parse_node) -> void;
+  auto HandleInfixOperator(ParseTree::Node parse_node) -> void;
+  auto HandleLiteral(ParseTree::Node parse_node) -> void;
+  auto HandleParameterList(ParseTree::Node parse_node) -> void;
+  auto HandleReturnStatement(ParseTree::Node parse_node) -> void;
+
+  // Tokens for getting data on literals.
+  const TokenizedBuffer* tokens_;
+
+  // The file's parse tree.
+  const ParseTree* parse_tree_;
+
+  // The SemanticsIR being added to.
+  SemanticsIR* semantics_;
+
+  // The stack during Build. Will contain file-level parse nodes on return.
+  llvm::SmallVector<TraversalStackEntry> node_stack_;
+};
+
+}  // namespace Carbon
+
+#endif  // CARBON_TOOLCHAIN_SEMANTICS_SEMANTICS_PARSE_TREE_HANDLER_H_