Files
carbon-lang/toolchain/check/handle_if_expression.cpp
T
Richard Smith fea6cf88fc Unify initialization and conversion logic (#3255)
Combine the initialization, implicit conversion, and value category
conversion functions into a single function.

This substantially reduces the duplication between these steps, and
ensures that we support the same set of conversions in all these
contexts. This also fixes some issues where we would not use the proper
value representation for tuples and structs after performing implicit
conversions.
2023-10-02 21:53:52 +00:00

75 lines
2.6 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/check/context.h"
#include "toolchain/check/convert.h"
namespace Carbon::Check {
auto HandleIfExpressionIf(Context& context, Parse::Node parse_node) -> bool {
// Alias parse_node for if/then/else consistency.
auto& if_node = parse_node;
auto cond_value_id = context.node_stack().PopExpression();
// Convert the condition to `bool`, and branch on it.
cond_value_id = ConvertToBoolValue(context, if_node, cond_value_id);
auto then_block_id =
context.AddDominatedBlockAndBranchIf(if_node, cond_value_id);
auto else_block_id = context.AddDominatedBlockAndBranch(if_node);
// Start emitting the `then` block.
context.node_block_stack().Pop();
context.node_block_stack().Push(then_block_id);
context.AddCurrentCodeBlockToFunction();
context.node_stack().Push(if_node, else_block_id);
return true;
}
auto HandleIfExpressionThen(Context& context, Parse::Node parse_node) -> bool {
auto then_value_id = context.node_stack().PopExpression();
auto else_block_id =
context.node_stack().Peek<Parse::NodeKind::IfExpressionIf>();
// Convert the first operand to a value.
then_value_id = ConvertToValueExpression(context, then_value_id);
// Start emitting the `else` block.
context.node_block_stack().Push(else_block_id);
context.AddCurrentCodeBlockToFunction();
context.node_stack().Push(parse_node, then_value_id);
return true;
}
auto HandleIfExpressionElse(Context& context, Parse::Node parse_node) -> bool {
// Alias parse_node for if/then/else consistency.
auto& else_node = parse_node;
auto else_value_id = context.node_stack().PopExpression();
auto then_value_id =
context.node_stack().Pop<Parse::NodeKind::IfExpressionThen>();
auto [if_node, _] =
context.node_stack().PopWithParseNode<Parse::NodeKind::IfExpressionIf>();
// Convert the `else` value to the `then` value's type, and finish the `else`
// block.
// TODO: Find a common type, and convert both operands to it instead.
auto result_type_id = context.semantics_ir().GetNode(then_value_id).type_id();
else_value_id =
ConvertToValueOfType(context, else_node, else_value_id, result_type_id);
// Create a resumption block and branches to it.
auto chosen_value_id = context.AddConvergenceBlockWithArgAndPush(
if_node, {else_value_id, then_value_id});
context.AddCurrentCodeBlockToFunction();
// Push the result value.
context.node_stack().Push(else_node, chosen_value_id);
return true;
}
} // namespace Carbon::Check