mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 22:02:23 +01:00
josh11b
a602be89e2
Use this new category to replace the unconstrained `NodeId` child of `MemberAccessExpr ` and `PointerMemberAccessExpr`. For now this new category matches `IdentifierName` and `BaseName`, but later this will be expanded to support `a.(b.c)` and `p->(b.c)` syntactic forms. QUESTION: Is it time to make a `node_category.def` x-macro file? ANSWER: Not yet.
117 lines
3.8 KiB
C++
117 lines
3.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
|
|
|
|
#include "toolchain/parse/node_kind.h"
|
|
|
|
#include "common/check.h"
|
|
#include "llvm/ADT/StringExtras.h"
|
|
#include "toolchain/parse/typed_nodes.h"
|
|
|
|
namespace Carbon::Parse {
|
|
|
|
auto operator<<(llvm::raw_ostream& output, NodeCategory category)
|
|
-> llvm::raw_ostream& {
|
|
if (!category) {
|
|
output << "<none>";
|
|
} else {
|
|
llvm::ListSeparator sep("|");
|
|
|
|
#define CARBON_NODE_CATEGORY(Name) \
|
|
if (!!(category & NodeCategory::Name)) { \
|
|
output << sep << #Name; \
|
|
}
|
|
CARBON_NODE_CATEGORY(Decl);
|
|
CARBON_NODE_CATEGORY(Expr);
|
|
CARBON_NODE_CATEGORY(MemberName);
|
|
CARBON_NODE_CATEGORY(Modifier);
|
|
CARBON_NODE_CATEGORY(NameComponent);
|
|
CARBON_NODE_CATEGORY(Pattern);
|
|
CARBON_NODE_CATEGORY(Statement);
|
|
#undef CARBON_NODE_CATEGORY
|
|
}
|
|
return output;
|
|
}
|
|
|
|
CARBON_DEFINE_ENUM_CLASS_NAMES(NodeKind) = {
|
|
#define CARBON_PARSE_NODE_KIND(Name) CARBON_ENUM_CLASS_NAME_STRING(Name)
|
|
#include "toolchain/parse/node_kind.def"
|
|
};
|
|
|
|
auto NodeKind::has_bracket() const -> bool {
|
|
static constexpr bool HasBracket[] = {
|
|
#define CARBON_PARSE_NODE_KIND_BRACKET(...) true,
|
|
#define CARBON_PARSE_NODE_KIND_CHILD_COUNT(...) false,
|
|
#include "toolchain/parse/node_kind.def"
|
|
};
|
|
return HasBracket[AsInt()];
|
|
}
|
|
|
|
auto NodeKind::bracket() const -> NodeKind {
|
|
// Nodes are never self-bracketed, so we use that for nodes that instead set
|
|
// child_count.
|
|
static constexpr NodeKind Bracket[] = {
|
|
#define CARBON_PARSE_NODE_KIND_BRACKET(Name, BracketName, ...) \
|
|
NodeKind::BracketName,
|
|
#define CARBON_PARSE_NODE_KIND_CHILD_COUNT(Name, ...) NodeKind::Name,
|
|
#include "toolchain/parse/node_kind.def"
|
|
};
|
|
auto bracket = Bracket[AsInt()];
|
|
CARBON_CHECK(bracket != *this) << *this;
|
|
return bracket;
|
|
}
|
|
|
|
auto NodeKind::child_count() const -> int32_t {
|
|
static constexpr int32_t ChildCount[] = {
|
|
#define CARBON_PARSE_NODE_KIND_BRACKET(...) -1,
|
|
#define CARBON_PARSE_NODE_KIND_CHILD_COUNT(Name, Size, ...) Size,
|
|
#include "toolchain/parse/node_kind.def"
|
|
};
|
|
auto child_count = ChildCount[AsInt()];
|
|
CARBON_CHECK(child_count >= 0) << *this;
|
|
return child_count;
|
|
}
|
|
|
|
auto NodeKind::CheckMatchesTokenKind(Lex::TokenKind token_kind, bool has_error)
|
|
-> void {
|
|
static constexpr Lex::TokenKind TokenIfValid[] = {
|
|
#define CARBON_IF_VALID(LexTokenKind) LexTokenKind
|
|
#define CARBON_PARSE_NODE_KIND_BRACKET(Name, BracketName, LexTokenKind) \
|
|
Lex::TokenKind::LexTokenKind,
|
|
#define CARBON_PARSE_NODE_KIND_CHILD_COUNT(Name, Size, LexTokenKind) \
|
|
Lex::TokenKind::LexTokenKind,
|
|
#include "toolchain/parse/node_kind.def"
|
|
};
|
|
static constexpr Lex::TokenKind TokenIfError[] = {
|
|
#define CARBON_IF_VALID(LexTokenKind) Error
|
|
#define CARBON_PARSE_NODE_KIND_BRACKET(Name, BracketName, LexTokenKind) \
|
|
Lex::TokenKind::LexTokenKind,
|
|
#define CARBON_PARSE_NODE_KIND_CHILD_COUNT(Name, Size, LexTokenKind) \
|
|
Lex::TokenKind::LexTokenKind,
|
|
#include "toolchain/parse/node_kind.def"
|
|
};
|
|
|
|
Lex::TokenKind expected_token_kind =
|
|
has_error ? TokenIfError[AsInt()] : TokenIfValid[AsInt()];
|
|
// Error indicates that the kind shouldn't be enforced.
|
|
CARBON_CHECK(Lex::TokenKind::Error == expected_token_kind ||
|
|
token_kind == expected_token_kind)
|
|
<< "Created parse node with NodeKind " << *this << " and has_error "
|
|
<< has_error << " for lexical token kind " << token_kind
|
|
<< ", but expected token kind " << expected_token_kind;
|
|
}
|
|
|
|
auto NodeKind::category() const -> NodeCategory {
|
|
return definition().category();
|
|
}
|
|
|
|
auto NodeKind::definition() const -> const Definition& {
|
|
static constexpr const Definition* Table[] = {
|
|
#define CARBON_PARSE_NODE_KIND(Name) &Parse::Name::Kind,
|
|
#include "toolchain/parse/node_kind.def"
|
|
};
|
|
return *Table[AsInt()];
|
|
}
|
|
|
|
} // namespace Carbon::Parse
|