Files
carbon-lang/toolchain/semantics/semantics_parse_tree_handler.h
T
Jon Ross-Perkins d51ca271af 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)
2022-10-25 12:55:26 -07:00

92 lines
3.1 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_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_