mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-25 20:50:12 +01:00
Richard Smith
1be726d3a2
Rearrange `NodeCategory` printing so we get a compile-time error for missing switch cases if it's missing any categories. Add several missing categories. In passing, fix some minor things in the `NodeCategory` class definition, and fix an overly-permissive typed node.
80 lines
2.4 KiB
C++
80 lines
2.4 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 "llvm/ADT/StringExtras.h"
|
|
#include "toolchain/parse/typed_nodes.h"
|
|
|
|
namespace Carbon::Parse {
|
|
|
|
auto NodeCategory::Print(llvm::raw_ostream& out) const -> void {
|
|
llvm::ListSeparator sep("|");
|
|
auto value = value_;
|
|
do {
|
|
// The lowest set bit in the value, or 0 (`None`) if no bits are set.
|
|
auto lowest_bit = static_cast<RawEnumType>(value & -value);
|
|
switch (lowest_bit) {
|
|
#define CARBON_NODE_CATEGORY(Name) \
|
|
case NodeCategory::Name: { \
|
|
out << sep << #Name; \
|
|
break; \
|
|
}
|
|
CARBON_NODE_CATEGORY(Decl);
|
|
CARBON_NODE_CATEGORY(Expr);
|
|
CARBON_NODE_CATEGORY(ImplAs);
|
|
CARBON_NODE_CATEGORY(MemberExpr);
|
|
CARBON_NODE_CATEGORY(MemberName);
|
|
CARBON_NODE_CATEGORY(Modifier);
|
|
CARBON_NODE_CATEGORY(Pattern);
|
|
CARBON_NODE_CATEGORY(Statement);
|
|
CARBON_NODE_CATEGORY(IntConst);
|
|
CARBON_NODE_CATEGORY(Requirement);
|
|
CARBON_NODE_CATEGORY(NonExprIdentifierName);
|
|
CARBON_NODE_CATEGORY(PackageName);
|
|
CARBON_NODE_CATEGORY(None);
|
|
#undef CARBON_NODE_CATEGORY
|
|
}
|
|
value &= ~lowest_bit;
|
|
} while (value);
|
|
}
|
|
|
|
CARBON_DEFINE_ENUM_CLASS_NAMES(NodeKind) = {
|
|
#define CARBON_PARSE_NODE_KIND(Name) CARBON_ENUM_CLASS_NAME_STRING(Name)
|
|
#include "toolchain/parse/node_kind.def"
|
|
};
|
|
|
|
// Check that each typed node defines a `token` member.
|
|
#define CARBON_PARSE_NODE_KIND(Name) \
|
|
static_assert(requires(Name node) { node.token; });
|
|
#include "toolchain/parse/node_kind.def"
|
|
|
|
auto NodeKind::has_bracket() const -> bool {
|
|
return definition().has_bracket();
|
|
}
|
|
|
|
auto NodeKind::bracket() const -> NodeKind { return definition().bracket(); }
|
|
|
|
auto NodeKind::has_child_count() const -> bool {
|
|
return definition().has_child_count();
|
|
}
|
|
|
|
auto NodeKind::child_count() const -> int32_t {
|
|
return definition().child_count();
|
|
}
|
|
|
|
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
|