Files
carbon-lang/toolchain/check/handle_if_expression.cpp
T
Jon Ross-Perkins 1c748c0f14 Split semantics into check and sem_ir directories (#3176)
Continuing along with #3070. Note this is just a file rename, with BUILD
edits; every file previously in semantics/ should show as moved (except
maybe BUILDs, which split).
2023-08-31 19:54:32 +00:00

78 lines
2.9 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"
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();
context.node_stack().Push(if_node);
// Convert the condition to `bool`, and branch on it.
cond_value_id = context.ConvertToBoolValue(if_node, cond_value_id);
auto then_block_id =
context.AddDominatedBlockAndBranchIf(if_node, cond_value_id);
auto else_block_id = context.AddDominatedBlockAndBranch(if_node);
// Push the `else` block and `then` block, and start emitting the `then`.
context.node_block_stack().Pop();
context.node_block_stack().Push(else_block_id);
context.node_block_stack().Push(then_block_id);
context.AddCurrentCodeBlockToFunction();
return true;
}
auto HandleIfExpressionThen(Context& context, Parse::Node parse_node) -> bool {
// Alias parse_node for if/then/else consistency.
auto& then_node = parse_node;
// Convert the first operand to a value.
auto [then_value_node, then_value_id] =
context.node_stack().PopExpressionWithParseNode();
context.node_stack().Push(then_value_node,
context.ConvertToValueExpression(then_value_id));
context.node_stack().Push(then_node, context.node_block_stack().Pop());
context.AddCurrentCodeBlockToFunction();
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_node, then_end_block_id] =
context.node_stack()
.PopWithParseNode<Parse::NodeKind::IfExpressionThen>();
auto then_value_id = context.node_stack().PopExpression();
auto if_node = context.node_stack()
.PopForSoloParseNode<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 =
context.ConvertToValueOfType(else_node, else_value_id, result_type_id);
auto else_end_block_id = context.node_block_stack().Pop();
// Create a resumption block and branches to it.
auto chosen_value_id = context.AddConvergenceBlockWithArgAndPush(
if_node,
{{then_end_block_id, then_value_id}, {else_end_block_id, else_value_id}});
context.AddCurrentCodeBlockToFunction();
// Push the result value.
context.node_stack().Push(else_node, chosen_value_id);
return true;
}
} // namespace Carbon::Check