Support tuple patterns outside parameter lists (#4923)

Parameter lists need substantially different treatment than tuple
patterns in other contexts, so this change splits them into separate
parse node kinds.

---------

Co-authored-by: Jon Ross-Perkins <jperkins@google.com>
This commit is contained in:
Geoff Romer
2025-02-20 03:17:45 +00:00
committed by GitHub
co-authored by Jon Ross-Perkins
parent 35f5a7f115
commit 74e1a9949f
201 changed files with 1680 additions and 860 deletions
+65 -37
View File
@@ -4,11 +4,14 @@
#include "toolchain/check/context.h"
#include "toolchain/check/handle.h"
#include "toolchain/check/inst.h"
#include "toolchain/check/subpattern.h"
#include "toolchain/check/type.h"
namespace Carbon::Check {
auto HandleParseNode(Context& context, Parse::ImplicitParamListStartId node_id)
// Handle the start of any kind of pattern list.
static auto HandlePatternListStart(Context& context, Parse::NodeId node_id)
-> bool {
context.node_stack().Push(node_id);
context.param_and_arg_refs_stack().Push();
@@ -16,34 +19,73 @@ auto HandleParseNode(Context& context, Parse::ImplicitParamListStartId node_id)
return true;
}
auto HandleParseNode(Context& context, Parse::ImplicitParamListStartId node_id)
-> bool {
return HandlePatternListStart(context, node_id);
}
auto HandleParseNode(Context& context, Parse::TuplePatternStartId node_id)
-> bool {
return HandlePatternListStart(context, node_id);
}
auto HandleParseNode(Context& context, Parse::ExplicitParamListStartId node_id)
-> bool {
context.full_pattern_stack().EndImplicitParamList();
return HandlePatternListStart(context, node_id);
}
// Handle the end of any kind of parameter list (tuple patterns have separate
// logic).
static auto HandleParamListEnd(Context& context, Parse::NodeId node_id,
Parse::NodeKind start_kind) -> bool {
if (context.node_stack().PeekIs(start_kind)) {
// End the subpattern started by a trailing comma, or the opening delimiter
// of an empty list.
EndSubpatternAsNonExpr(context);
}
// Note the Start node remains on the stack, where the param list handler can
// make use of it.
auto refs_id = context.param_and_arg_refs_stack().EndAndPop(start_kind);
context.node_stack().Push(node_id, refs_id);
return true;
}
auto HandleParseNode(Context& context, Parse::ImplicitParamListId node_id)
-> bool {
if (context.node_stack().PeekIs(Parse::NodeKind::ImplicitParamListStart)) {
// End the subpattern started by a trailing comma, or the opening delimiter
// of an empty list.
EndSubpatternAsEmpty(context);
}
// Note the Start node remains on the stack, where the param list handler can
// make use of it.
auto refs_id = context.param_and_arg_refs_stack().EndAndPop(
Parse::NodeKind::ImplicitParamListStart);
context.node_stack().Push(node_id, refs_id);
// The implicit parameter list's scope extends to the end of the following
// parameter list.
return true;
return HandleParamListEnd(context, node_id,
Parse::NodeKind::ImplicitParamListStart);
}
auto HandleParseNode(Context& context, Parse::TuplePatternStartId node_id)
auto HandleParseNode(Context& context, Parse::ExplicitParamListId node_id)
-> bool {
context.node_stack().Push(node_id);
context.param_and_arg_refs_stack().Push();
BeginSubpattern(context);
// TODO: Remove this branch once the parse tree differentiates between
// tuple patterns and param patterns.
if (context.full_pattern_stack().CurrentKind() ==
FullPatternStack::Kind::ImplicitParamList) {
context.full_pattern_stack().EndImplicitParamList();
return HandleParamListEnd(context, node_id,
Parse::NodeKind::ExplicitParamListStart);
}
auto HandleParseNode(Context& context, Parse::TuplePatternId node_id) -> bool {
if (context.node_stack().PeekIs(Parse::NodeKind::TuplePatternStart)) {
// End the subpattern started by a trailing comma, or the opening delimiter
// of an empty list.
EndSubpatternAsNonExpr(context);
}
auto refs_id = context.param_and_arg_refs_stack().EndAndPop(
Parse::NodeKind::TuplePatternStart);
context.node_stack()
.PopAndDiscardSoloNodeId<Parse::NodeKind::TuplePatternStart>();
const auto& inst_block = context.inst_blocks().Get(refs_id);
llvm::SmallVector<SemIR::TypeId> type_ids;
type_ids.reserve(inst_block.size());
for (auto inst : inst_block) {
type_ids.push_back(context.insts().Get(inst).type_id());
}
auto type_id = GetTupleType(context, type_ids);
context.node_stack().Push(
node_id,
AddPatternInst<SemIR::TuplePattern>(
context, node_id, {.type_id = type_id, .elements_id = refs_id}));
EndSubpatternAsNonExpr(context);
return true;
}
@@ -54,18 +96,4 @@ auto HandleParseNode(Context& context, Parse::PatternListCommaId /*node_id*/)
return true;
}
auto HandleParseNode(Context& context, Parse::TuplePatternId node_id) -> bool {
if (context.node_stack().PeekIs(Parse::NodeKind::TuplePatternStart)) {
// End the subpattern started by a trailing comma, or the opening delimiter
// of an empty list.
EndSubpatternAsEmpty(context);
}
// Note the Start node remains on the stack, where the param list handler can
// make use of it.
auto refs_id = context.param_and_arg_refs_stack().EndAndPop(
Parse::NodeKind::TuplePatternStart);
context.node_stack().Push(node_id, refs_id);
return true;
}
} // namespace Carbon::Check