Files
carbon-lang/toolchain/semantics/semantics_handle_struct.cpp
T
Jon Ross-Perkins a1a7251716 Refactor NodeStack APIs to better handle the increase of distinct ID types. (#2864)
I'm looking at adding CallableId, so figured I'd do this API refactoring which should reduce code duplication. This increases the amount of cross-calls between APIs because it may not be an issue for performance after optimizations, and should simplify reading of the API.

Also note the prior static_asserts on layout were missing a couple types, which is why I'm moving them into Entry where it's going to be more obvious when something's added.
2023-06-01 16:39:02 -07:00

120 lines
4.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/semantics/semantics_context.h"
namespace Carbon {
auto SemanticsHandleStructComma(SemanticsContext& context,
ParseTree::Node /*parse_node*/) -> bool {
context.ParamOrArgComma(
/*for_args=*/context.parse_tree().node_kind(
context.node_stack().PeekParseNode()) !=
ParseNodeKind::StructFieldType);
return true;
}
auto SemanticsHandleStructFieldDesignator(SemanticsContext& context,
ParseTree::Node /*parse_node*/)
-> bool {
// This leaves the designated name on top because the `.` isn't interesting.
CARBON_CHECK(
context.parse_tree().node_kind(context.node_stack().PeekParseNode()) ==
ParseNodeKind::DesignatedName);
return true;
}
auto SemanticsHandleStructFieldType(SemanticsContext& context,
ParseTree::Node parse_node) -> bool {
auto [type_node, type_id] =
context.node_stack().PopWithParseNode<SemanticsNodeId>();
SemanticsTypeId cast_type_id = context.ExpressionAsType(type_node, type_id);
auto [name_node, name_id] =
context.node_stack().PopWithParseNode<SemanticsStringId>(
ParseNodeKind::DesignatedName);
context.AddNode(
SemanticsNode::StructTypeField::Make(name_node, cast_type_id, name_id));
context.node_stack().Push(parse_node);
return true;
}
auto SemanticsHandleStructFieldUnknown(SemanticsContext& context,
ParseTree::Node parse_node) -> bool {
return context.TODO(parse_node, "HandleStructFieldUnknown");
}
auto SemanticsHandleStructFieldValue(SemanticsContext& context,
ParseTree::Node parse_node) -> bool {
auto [value_parse_node, value_node_id] =
context.node_stack().PopWithParseNode<SemanticsNodeId>();
auto name_id = context.node_stack().Pop<SemanticsStringId>(
ParseNodeKind::DesignatedName);
// Store the name for the type.
auto type_block_id = context.args_type_info_stack().PeekForAdd();
context.semantics_ir().AddNode(
type_block_id,
SemanticsNode::StructTypeField::Make(
parse_node, context.semantics_ir().GetNode(value_node_id).type_id(),
name_id));
// Push the value back on the stack as an argument.
context.node_stack().Push(parse_node, value_node_id);
return true;
}
auto SemanticsHandleStructLiteral(SemanticsContext& context,
ParseTree::Node parse_node) -> bool {
auto refs_id = context.ParamOrArgEnd(
/*for_args=*/true, ParseNodeKind::StructLiteralOrStructTypeLiteralStart);
context.PopScope();
context.node_stack().PopAndDiscardSoloParseNode(
ParseNodeKind::StructLiteralOrStructTypeLiteralStart);
auto type_block_id = context.args_type_info_stack().Pop();
auto type_id = context.CanonicalizeStructType(parse_node, type_block_id);
auto value_id = context.AddNode(
SemanticsNode::StructValue::Make(parse_node, type_id, refs_id));
context.node_stack().Push(parse_node, value_id);
return true;
}
auto SemanticsHandleStructLiteralOrStructTypeLiteralStart(
SemanticsContext& context, ParseTree::Node parse_node) -> bool {
context.PushScope();
context.node_stack().Push(parse_node);
// At this point we aren't sure whether this will be a value or type literal,
// so we push onto args irrespective. It just won't be used for a type
// literal.
context.args_type_info_stack().Push();
context.ParamOrArgStart();
return true;
}
auto SemanticsHandleStructTypeLiteral(SemanticsContext& context,
ParseTree::Node parse_node) -> bool {
auto refs_id = context.ParamOrArgEnd(
/*for_args=*/false, ParseNodeKind::StructLiteralOrStructTypeLiteralStart);
context.PopScope();
context.node_stack().PopAndDiscardSoloParseNode(
ParseNodeKind::StructLiteralOrStructTypeLiteralStart);
// This is only used for value literals.
context.args_type_info_stack().Pop();
CARBON_CHECK(refs_id != SemanticsNodeBlockId::Empty)
<< "{} is handled by StructLiteral.";
auto type_id = context.CanonicalizeStructType(parse_node, refs_id);
context.node_stack().Push(parse_node,
context.semantics_ir().GetType(type_id));
return true;
}
} // namespace Carbon