mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-07-15 17:45:58 +00:00
53f2026fc5
This Cl adds a test for a continue inside an else case. Bug: tint:5 Change-Id: If9606b18ff8c2af6eb271aa37c404f36097f592e Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/22060 Reviewed-by: Ryan Harrison <rharrison@chromium.org>
751 lines
20 KiB
C++
751 lines
20 KiB
C++
// Copyright 2020 The Tint Authors.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include <memory>
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "src/ast/assignment_statement.h"
|
|
#include "src/ast/bool_literal.h"
|
|
#include "src/ast/break_statement.h"
|
|
#include "src/ast/continue_statement.h"
|
|
#include "src/ast/else_statement.h"
|
|
#include "src/ast/identifier_expression.h"
|
|
#include "src/ast/if_statement.h"
|
|
#include "src/ast/int_literal.h"
|
|
#include "src/ast/loop_statement.h"
|
|
#include "src/ast/return_statement.h"
|
|
#include "src/ast/scalar_constructor_expression.h"
|
|
#include "src/ast/statement_condition.h"
|
|
#include "src/ast/type/bool_type.h"
|
|
#include "src/ast/type/i32_type.h"
|
|
#include "src/context.h"
|
|
#include "src/type_determiner.h"
|
|
#include "src/writer/spirv/builder.h"
|
|
#include "src/writer/spirv/spv_dump.h"
|
|
|
|
namespace tint {
|
|
namespace writer {
|
|
namespace spirv {
|
|
namespace {
|
|
|
|
using BuilderTest = testing::Test;
|
|
|
|
TEST_F(BuilderTest, If_Empty) {
|
|
ast::type::BoolType bool_type;
|
|
|
|
// if (true) {
|
|
// }
|
|
auto cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
|
|
ast::IfStatement expr(std::move(cond), ast::StatementList{});
|
|
|
|
Context ctx;
|
|
ast::Module mod;
|
|
TypeDeterminer td(&ctx, &mod);
|
|
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
|
|
|
|
Builder b(&mod);
|
|
b.push_function(Function{});
|
|
|
|
EXPECT_TRUE(b.GenerateIfStatement(&expr)) << b.error();
|
|
EXPECT_EQ(DumpInstructions(b.types()), R"(%1 = OpTypeBool
|
|
%2 = OpConstantTrue %1
|
|
)");
|
|
EXPECT_EQ(DumpInstructions(b.functions()[0].instructions()),
|
|
R"(OpSelectionMerge %3 None
|
|
OpBranchConditional %2 %4 %3
|
|
%4 = OpLabel
|
|
OpBranch %3
|
|
%3 = OpLabel
|
|
)");
|
|
}
|
|
|
|
TEST_F(BuilderTest, If_WithStatements) {
|
|
ast::type::BoolType bool_type;
|
|
ast::type::I32Type i32;
|
|
|
|
// if (true) {
|
|
// v = 2;
|
|
// }
|
|
auto var =
|
|
std::make_unique<ast::Variable>("v", ast::StorageClass::kPrivate, &i32);
|
|
|
|
ast::StatementList body;
|
|
body.push_back(std::make_unique<ast::AssignmentStatement>(
|
|
std::make_unique<ast::IdentifierExpression>("v"),
|
|
std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::IntLiteral>(&i32, 2))));
|
|
|
|
auto cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
|
|
ast::IfStatement expr(std::move(cond), std::move(body));
|
|
|
|
Context ctx;
|
|
ast::Module mod;
|
|
TypeDeterminer td(&ctx, &mod);
|
|
td.RegisterVariableForTesting(var.get());
|
|
|
|
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
|
|
|
|
Builder b(&mod);
|
|
b.push_function(Function{});
|
|
ASSERT_TRUE(b.GenerateGlobalVariable(var.get())) << b.error();
|
|
|
|
EXPECT_TRUE(b.GenerateIfStatement(&expr)) << b.error();
|
|
EXPECT_EQ(DumpInstructions(b.types()), R"(%3 = OpTypeInt 32 1
|
|
%2 = OpTypePointer Private %3
|
|
%4 = OpConstantNull %3
|
|
%1 = OpVariable %2 Private %4
|
|
%5 = OpTypeBool
|
|
%6 = OpConstantTrue %5
|
|
%9 = OpConstant %3 2
|
|
)");
|
|
EXPECT_EQ(DumpInstructions(b.functions()[0].instructions()),
|
|
R"(OpSelectionMerge %7 None
|
|
OpBranchConditional %6 %8 %7
|
|
%8 = OpLabel
|
|
OpStore %1 %9
|
|
OpBranch %7
|
|
%7 = OpLabel
|
|
)");
|
|
}
|
|
|
|
TEST_F(BuilderTest, If_WithElse) {
|
|
ast::type::BoolType bool_type;
|
|
ast::type::I32Type i32;
|
|
|
|
// if (true) {
|
|
// v = 2;
|
|
// } else {
|
|
// v = 3;
|
|
// }
|
|
auto var =
|
|
std::make_unique<ast::Variable>("v", ast::StorageClass::kPrivate, &i32);
|
|
|
|
ast::StatementList body;
|
|
body.push_back(std::make_unique<ast::AssignmentStatement>(
|
|
std::make_unique<ast::IdentifierExpression>("v"),
|
|
std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::IntLiteral>(&i32, 2))));
|
|
|
|
ast::StatementList else_body;
|
|
else_body.push_back(std::make_unique<ast::AssignmentStatement>(
|
|
std::make_unique<ast::IdentifierExpression>("v"),
|
|
std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::IntLiteral>(&i32, 3))));
|
|
|
|
ast::ElseStatementList else_stmts;
|
|
else_stmts.push_back(
|
|
std::make_unique<ast::ElseStatement>(std::move(else_body)));
|
|
|
|
auto cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
|
|
ast::IfStatement expr(std::move(cond), std::move(body));
|
|
expr.set_else_statements(std::move(else_stmts));
|
|
|
|
Context ctx;
|
|
ast::Module mod;
|
|
TypeDeterminer td(&ctx, &mod);
|
|
td.RegisterVariableForTesting(var.get());
|
|
|
|
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
|
|
|
|
Builder b(&mod);
|
|
b.push_function(Function{});
|
|
ASSERT_TRUE(b.GenerateGlobalVariable(var.get())) << b.error();
|
|
|
|
EXPECT_TRUE(b.GenerateIfStatement(&expr)) << b.error();
|
|
EXPECT_EQ(DumpInstructions(b.types()), R"(%3 = OpTypeInt 32 1
|
|
%2 = OpTypePointer Private %3
|
|
%4 = OpConstantNull %3
|
|
%1 = OpVariable %2 Private %4
|
|
%5 = OpTypeBool
|
|
%6 = OpConstantTrue %5
|
|
%10 = OpConstant %3 2
|
|
%11 = OpConstant %3 3
|
|
)");
|
|
EXPECT_EQ(DumpInstructions(b.functions()[0].instructions()),
|
|
R"(OpSelectionMerge %7 None
|
|
OpBranchConditional %6 %8 %9
|
|
%8 = OpLabel
|
|
OpStore %1 %10
|
|
OpBranch %7
|
|
%9 = OpLabel
|
|
OpStore %1 %11
|
|
OpBranch %7
|
|
%7 = OpLabel
|
|
)");
|
|
}
|
|
|
|
TEST_F(BuilderTest, If_WithElseIf) {
|
|
ast::type::BoolType bool_type;
|
|
ast::type::I32Type i32;
|
|
|
|
// if (true) {
|
|
// v = 2;
|
|
// } elseif (true) {
|
|
// v = 3;
|
|
// }
|
|
auto var =
|
|
std::make_unique<ast::Variable>("v", ast::StorageClass::kPrivate, &i32);
|
|
|
|
ast::StatementList body;
|
|
body.push_back(std::make_unique<ast::AssignmentStatement>(
|
|
std::make_unique<ast::IdentifierExpression>("v"),
|
|
std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::IntLiteral>(&i32, 2))));
|
|
|
|
ast::StatementList else_body;
|
|
else_body.push_back(std::make_unique<ast::AssignmentStatement>(
|
|
std::make_unique<ast::IdentifierExpression>("v"),
|
|
std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::IntLiteral>(&i32, 3))));
|
|
|
|
auto else_cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
|
|
ast::ElseStatementList else_stmts;
|
|
else_stmts.push_back(std::make_unique<ast::ElseStatement>(
|
|
std::move(else_cond), std::move(else_body)));
|
|
|
|
auto cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
|
|
ast::IfStatement expr(std::move(cond), std::move(body));
|
|
expr.set_else_statements(std::move(else_stmts));
|
|
|
|
Context ctx;
|
|
ast::Module mod;
|
|
TypeDeterminer td(&ctx, &mod);
|
|
td.RegisterVariableForTesting(var.get());
|
|
|
|
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
|
|
|
|
Builder b(&mod);
|
|
b.push_function(Function{});
|
|
ASSERT_TRUE(b.GenerateGlobalVariable(var.get())) << b.error();
|
|
|
|
EXPECT_TRUE(b.GenerateIfStatement(&expr)) << b.error();
|
|
EXPECT_EQ(DumpInstructions(b.types()), R"(%3 = OpTypeInt 32 1
|
|
%2 = OpTypePointer Private %3
|
|
%4 = OpConstantNull %3
|
|
%1 = OpVariable %2 Private %4
|
|
%5 = OpTypeBool
|
|
%6 = OpConstantTrue %5
|
|
%10 = OpConstant %3 2
|
|
%13 = OpConstant %3 3
|
|
)");
|
|
EXPECT_EQ(DumpInstructions(b.functions()[0].instructions()),
|
|
R"(OpSelectionMerge %7 None
|
|
OpBranchConditional %6 %8 %9
|
|
%8 = OpLabel
|
|
OpStore %1 %10
|
|
OpBranch %7
|
|
%9 = OpLabel
|
|
OpSelectionMerge %11 None
|
|
OpBranchConditional %6 %12 %11
|
|
%12 = OpLabel
|
|
OpStore %1 %13
|
|
OpBranch %11
|
|
%11 = OpLabel
|
|
OpBranch %7
|
|
%7 = OpLabel
|
|
)");
|
|
}
|
|
|
|
TEST_F(BuilderTest, If_WithMultiple) {
|
|
ast::type::BoolType bool_type;
|
|
ast::type::I32Type i32;
|
|
|
|
// if (true) {
|
|
// v = 2;
|
|
// } elseif (true) {
|
|
// v = 3;
|
|
// } elseif (false) {
|
|
// v = 4;
|
|
// } else {
|
|
// v = 5;
|
|
// }
|
|
auto var =
|
|
std::make_unique<ast::Variable>("v", ast::StorageClass::kPrivate, &i32);
|
|
|
|
ast::StatementList body;
|
|
body.push_back(std::make_unique<ast::AssignmentStatement>(
|
|
std::make_unique<ast::IdentifierExpression>("v"),
|
|
std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::IntLiteral>(&i32, 2))));
|
|
ast::StatementList elseif_1_body;
|
|
elseif_1_body.push_back(std::make_unique<ast::AssignmentStatement>(
|
|
std::make_unique<ast::IdentifierExpression>("v"),
|
|
std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::IntLiteral>(&i32, 3))));
|
|
ast::StatementList elseif_2_body;
|
|
elseif_2_body.push_back(std::make_unique<ast::AssignmentStatement>(
|
|
std::make_unique<ast::IdentifierExpression>("v"),
|
|
std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::IntLiteral>(&i32, 4))));
|
|
ast::StatementList else_body;
|
|
else_body.push_back(std::make_unique<ast::AssignmentStatement>(
|
|
std::make_unique<ast::IdentifierExpression>("v"),
|
|
std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::IntLiteral>(&i32, 5))));
|
|
|
|
auto elseif_1_cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
auto elseif_2_cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, false));
|
|
|
|
ast::ElseStatementList else_stmts;
|
|
else_stmts.push_back(std::make_unique<ast::ElseStatement>(
|
|
std::move(elseif_1_cond), std::move(elseif_1_body)));
|
|
else_stmts.push_back(std::make_unique<ast::ElseStatement>(
|
|
std::move(elseif_2_cond), std::move(elseif_2_body)));
|
|
else_stmts.push_back(
|
|
std::make_unique<ast::ElseStatement>(std::move(else_body)));
|
|
|
|
auto cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
|
|
ast::IfStatement expr(std::move(cond), std::move(body));
|
|
expr.set_else_statements(std::move(else_stmts));
|
|
|
|
Context ctx;
|
|
ast::Module mod;
|
|
TypeDeterminer td(&ctx, &mod);
|
|
td.RegisterVariableForTesting(var.get());
|
|
|
|
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
|
|
|
|
Builder b(&mod);
|
|
b.push_function(Function{});
|
|
ASSERT_TRUE(b.GenerateGlobalVariable(var.get())) << b.error();
|
|
|
|
EXPECT_TRUE(b.GenerateIfStatement(&expr)) << b.error();
|
|
EXPECT_EQ(DumpInstructions(b.types()), R"(%3 = OpTypeInt 32 1
|
|
%2 = OpTypePointer Private %3
|
|
%4 = OpConstantNull %3
|
|
%1 = OpVariable %2 Private %4
|
|
%5 = OpTypeBool
|
|
%6 = OpConstantTrue %5
|
|
%10 = OpConstant %3 2
|
|
%14 = OpConstant %3 3
|
|
%15 = OpConstantFalse %5
|
|
%19 = OpConstant %3 4
|
|
%20 = OpConstant %3 5
|
|
)");
|
|
EXPECT_EQ(DumpInstructions(b.functions()[0].instructions()),
|
|
R"(OpSelectionMerge %7 None
|
|
OpBranchConditional %6 %8 %9
|
|
%8 = OpLabel
|
|
OpStore %1 %10
|
|
OpBranch %7
|
|
%9 = OpLabel
|
|
OpSelectionMerge %11 None
|
|
OpBranchConditional %6 %12 %13
|
|
%12 = OpLabel
|
|
OpStore %1 %14
|
|
OpBranch %11
|
|
%13 = OpLabel
|
|
OpSelectionMerge %16 None
|
|
OpBranchConditional %15 %17 %18
|
|
%17 = OpLabel
|
|
OpStore %1 %19
|
|
OpBranch %16
|
|
%18 = OpLabel
|
|
OpStore %1 %20
|
|
OpBranch %16
|
|
%16 = OpLabel
|
|
OpBranch %11
|
|
%11 = OpLabel
|
|
OpBranch %7
|
|
%7 = OpLabel
|
|
)");
|
|
}
|
|
|
|
TEST_F(BuilderTest, If_WithBreak) {
|
|
ast::type::BoolType bool_type;
|
|
// loop {
|
|
// if (true) {
|
|
// break;
|
|
// }
|
|
// }
|
|
auto cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
|
|
ast::StatementList if_body;
|
|
if_body.push_back(std::make_unique<ast::BreakStatement>());
|
|
|
|
auto if_stmt =
|
|
std::make_unique<ast::IfStatement>(std::move(cond), std::move(if_body));
|
|
|
|
ast::StatementList loop_body;
|
|
loop_body.push_back(std::move(if_stmt));
|
|
|
|
ast::LoopStatement expr(std::move(loop_body), {});
|
|
|
|
Context ctx;
|
|
ast::Module mod;
|
|
TypeDeterminer td(&ctx, &mod);
|
|
|
|
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
|
|
|
|
Builder b(&mod);
|
|
b.push_function(Function{});
|
|
|
|
EXPECT_TRUE(b.GenerateLoopStatement(&expr)) << b.error();
|
|
EXPECT_EQ(DumpInstructions(b.types()), R"(%5 = OpTypeBool
|
|
%6 = OpConstantTrue %5
|
|
)");
|
|
EXPECT_EQ(DumpInstructions(b.functions()[0].instructions()),
|
|
R"(OpBranch %1
|
|
%1 = OpLabel
|
|
OpLoopMerge %2 %3 None
|
|
OpBranch %4
|
|
%4 = OpLabel
|
|
OpSelectionMerge %7 None
|
|
OpBranchConditional %6 %8 %7
|
|
%8 = OpLabel
|
|
OpBranch %2
|
|
%7 = OpLabel
|
|
OpBranch %3
|
|
%3 = OpLabel
|
|
OpBranch %1
|
|
%2 = OpLabel
|
|
)");
|
|
}
|
|
|
|
TEST_F(BuilderTest, If_WithElseBreak) {
|
|
ast::type::BoolType bool_type;
|
|
// loop {
|
|
// if (true) {
|
|
// } else {
|
|
// break;
|
|
// }
|
|
// }
|
|
auto cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
|
|
ast::StatementList if_body;
|
|
ast::StatementList else_body;
|
|
else_body.push_back(std::make_unique<ast::BreakStatement>());
|
|
|
|
ast::ElseStatementList else_stmts;
|
|
else_stmts.push_back(
|
|
std::make_unique<ast::ElseStatement>(std::move(else_body)));
|
|
|
|
auto if_stmt =
|
|
std::make_unique<ast::IfStatement>(std::move(cond), std::move(if_body));
|
|
if_stmt->set_else_statements(std::move(else_stmts));
|
|
|
|
ast::StatementList loop_body;
|
|
loop_body.push_back(std::move(if_stmt));
|
|
|
|
ast::LoopStatement expr(std::move(loop_body), {});
|
|
|
|
Context ctx;
|
|
ast::Module mod;
|
|
TypeDeterminer td(&ctx, &mod);
|
|
|
|
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
|
|
|
|
Builder b(&mod);
|
|
b.push_function(Function{});
|
|
|
|
EXPECT_TRUE(b.GenerateLoopStatement(&expr)) << b.error();
|
|
EXPECT_EQ(DumpInstructions(b.types()), R"(%5 = OpTypeBool
|
|
%6 = OpConstantTrue %5
|
|
)");
|
|
EXPECT_EQ(DumpInstructions(b.functions()[0].instructions()),
|
|
R"(OpBranch %1
|
|
%1 = OpLabel
|
|
OpLoopMerge %2 %3 None
|
|
OpBranch %4
|
|
%4 = OpLabel
|
|
OpSelectionMerge %7 None
|
|
OpBranchConditional %6 %8 %9
|
|
%8 = OpLabel
|
|
OpBranch %7
|
|
%9 = OpLabel
|
|
OpBranch %2
|
|
%7 = OpLabel
|
|
OpBranch %3
|
|
%3 = OpLabel
|
|
OpBranch %1
|
|
%2 = OpLabel
|
|
)");
|
|
}
|
|
|
|
// This is blocked on implementing conditional break
|
|
TEST_F(BuilderTest, DISABLED_If_WithConditionalBreak) {
|
|
ast::type::BoolType bool_type;
|
|
// loop {
|
|
// if (true) {
|
|
// break if (false);
|
|
// }
|
|
// }
|
|
auto cond_true = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
auto cond_false = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, false));
|
|
|
|
ast::StatementList if_body;
|
|
if_body.push_back(std::make_unique<ast::BreakStatement>(
|
|
ast::StatementCondition::kIf, std::move(cond_false)));
|
|
|
|
auto if_stmt = std::make_unique<ast::IfStatement>(std::move(cond_true),
|
|
std::move(if_body));
|
|
|
|
ast::StatementList loop_body;
|
|
loop_body.push_back(std::move(if_stmt));
|
|
|
|
ast::LoopStatement expr(std::move(loop_body), {});
|
|
|
|
Context ctx;
|
|
ast::Module mod;
|
|
TypeDeterminer td(&ctx, &mod);
|
|
|
|
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
|
|
|
|
Builder b(&mod);
|
|
b.push_function(Function{});
|
|
|
|
EXPECT_TRUE(b.GenerateLoopStatement(&expr)) << b.error();
|
|
EXPECT_EQ(DumpInstructions(b.types()), R"(%5 = OpTypeBool
|
|
%6 = OpConstantTrue %5
|
|
%7 = OpConstantFalse %5
|
|
)");
|
|
EXPECT_EQ(DumpInstructions(b.functions()[0].instructions()),
|
|
R"(OpBranch %1
|
|
%1 = OpLabel
|
|
OpLoopMerge %2 %3 None
|
|
OpBranch %4
|
|
%4 = OpLabel
|
|
OpSelectionMerge %8 None
|
|
OpBranchConditional %6 %9 %8
|
|
%9 = OpLabel
|
|
OpBranchConditional %7 %2 %8
|
|
%8 = OpLabel
|
|
OpBranch %3
|
|
%3 = OpLabel
|
|
OpBranch %1
|
|
%2 = OpLabel
|
|
)");
|
|
}
|
|
|
|
TEST_F(BuilderTest, DISABLED_If_WithElseConditionalBreak) {
|
|
FAIL();
|
|
}
|
|
|
|
TEST_F(BuilderTest, If_WithContinue) {
|
|
ast::type::BoolType bool_type;
|
|
// loop {
|
|
// if (true) {
|
|
// continue;
|
|
// }
|
|
// }
|
|
auto cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
|
|
ast::StatementList if_body;
|
|
if_body.push_back(std::make_unique<ast::ContinueStatement>());
|
|
|
|
auto if_stmt =
|
|
std::make_unique<ast::IfStatement>(std::move(cond), std::move(if_body));
|
|
|
|
ast::StatementList loop_body;
|
|
loop_body.push_back(std::move(if_stmt));
|
|
|
|
ast::LoopStatement expr(std::move(loop_body), {});
|
|
|
|
Context ctx;
|
|
ast::Module mod;
|
|
TypeDeterminer td(&ctx, &mod);
|
|
|
|
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
|
|
|
|
Builder b(&mod);
|
|
b.push_function(Function{});
|
|
|
|
EXPECT_TRUE(b.GenerateLoopStatement(&expr)) << b.error();
|
|
EXPECT_EQ(DumpInstructions(b.types()), R"(%5 = OpTypeBool
|
|
%6 = OpConstantTrue %5
|
|
)");
|
|
EXPECT_EQ(DumpInstructions(b.functions()[0].instructions()),
|
|
R"(OpBranch %1
|
|
%1 = OpLabel
|
|
OpLoopMerge %2 %3 None
|
|
OpBranch %4
|
|
%4 = OpLabel
|
|
OpSelectionMerge %7 None
|
|
OpBranchConditional %6 %8 %7
|
|
%8 = OpLabel
|
|
OpBranch %3
|
|
%7 = OpLabel
|
|
OpBranch %3
|
|
%3 = OpLabel
|
|
OpBranch %1
|
|
%2 = OpLabel
|
|
)");
|
|
}
|
|
|
|
TEST_F(BuilderTest, DISABLED_If_WithConditionalContinue) {
|
|
FAIL();
|
|
}
|
|
|
|
TEST_F(BuilderTest, If_WithElseContinue) {
|
|
ast::type::BoolType bool_type;
|
|
// loop {
|
|
// if (true) {
|
|
// } else {
|
|
// continue;
|
|
// }
|
|
// }
|
|
auto cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
|
|
ast::StatementList if_body;
|
|
ast::StatementList else_body;
|
|
else_body.push_back(std::make_unique<ast::ContinueStatement>());
|
|
|
|
ast::ElseStatementList else_stmts;
|
|
else_stmts.push_back(
|
|
std::make_unique<ast::ElseStatement>(std::move(else_body)));
|
|
|
|
auto if_stmt =
|
|
std::make_unique<ast::IfStatement>(std::move(cond), std::move(if_body));
|
|
if_stmt->set_else_statements(std::move(else_stmts));
|
|
|
|
ast::StatementList loop_body;
|
|
loop_body.push_back(std::move(if_stmt));
|
|
|
|
ast::LoopStatement expr(std::move(loop_body), {});
|
|
|
|
Context ctx;
|
|
ast::Module mod;
|
|
TypeDeterminer td(&ctx, &mod);
|
|
|
|
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
|
|
|
|
Builder b(&mod);
|
|
b.push_function(Function{});
|
|
|
|
EXPECT_TRUE(b.GenerateLoopStatement(&expr)) << b.error();
|
|
EXPECT_EQ(DumpInstructions(b.types()), R"(%5 = OpTypeBool
|
|
%6 = OpConstantTrue %5
|
|
)");
|
|
EXPECT_EQ(DumpInstructions(b.functions()[0].instructions()),
|
|
R"(OpBranch %1
|
|
%1 = OpLabel
|
|
OpLoopMerge %2 %3 None
|
|
OpBranch %4
|
|
%4 = OpLabel
|
|
OpSelectionMerge %7 None
|
|
OpBranchConditional %6 %8 %9
|
|
%8 = OpLabel
|
|
OpBranch %7
|
|
%9 = OpLabel
|
|
OpBranch %3
|
|
%7 = OpLabel
|
|
OpBranch %3
|
|
%3 = OpLabel
|
|
OpBranch %1
|
|
%2 = OpLabel
|
|
)");
|
|
}
|
|
|
|
TEST_F(BuilderTest, DISABLED_If_WithElseConditionalContinue) {
|
|
FAIL();
|
|
}
|
|
|
|
TEST_F(BuilderTest, If_WithReturn) {
|
|
ast::type::BoolType bool_type;
|
|
// if (true) {
|
|
// return;
|
|
// }
|
|
auto cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
|
|
ast::StatementList if_body;
|
|
if_body.push_back(std::make_unique<ast::ReturnStatement>());
|
|
|
|
ast::IfStatement expr(std::move(cond), std::move(if_body));
|
|
|
|
Context ctx;
|
|
ast::Module mod;
|
|
TypeDeterminer td(&ctx, &mod);
|
|
|
|
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
|
|
|
|
Builder b(&mod);
|
|
b.push_function(Function{});
|
|
|
|
EXPECT_TRUE(b.GenerateIfStatement(&expr)) << b.error();
|
|
EXPECT_EQ(DumpInstructions(b.types()), R"(%1 = OpTypeBool
|
|
%2 = OpConstantTrue %1
|
|
)");
|
|
EXPECT_EQ(DumpInstructions(b.functions()[0].instructions()),
|
|
R"(OpSelectionMerge %3 None
|
|
OpBranchConditional %2 %4 %3
|
|
%4 = OpLabel
|
|
OpReturn
|
|
%3 = OpLabel
|
|
)");
|
|
}
|
|
|
|
TEST_F(BuilderTest, If_WithReturnValue) {
|
|
ast::type::BoolType bool_type;
|
|
// if (true) {
|
|
// return false;
|
|
// }
|
|
auto cond = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, true));
|
|
auto cond2 = std::make_unique<ast::ScalarConstructorExpression>(
|
|
std::make_unique<ast::BoolLiteral>(&bool_type, false));
|
|
|
|
ast::StatementList if_body;
|
|
if_body.push_back(std::make_unique<ast::ReturnStatement>(std::move(cond2)));
|
|
|
|
ast::IfStatement expr(std::move(cond), std::move(if_body));
|
|
|
|
Context ctx;
|
|
ast::Module mod;
|
|
TypeDeterminer td(&ctx, &mod);
|
|
|
|
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
|
|
|
|
Builder b(&mod);
|
|
b.push_function(Function{});
|
|
|
|
EXPECT_TRUE(b.GenerateIfStatement(&expr)) << b.error();
|
|
EXPECT_EQ(DumpInstructions(b.types()), R"(%1 = OpTypeBool
|
|
%2 = OpConstantTrue %1
|
|
%5 = OpConstantFalse %1
|
|
)");
|
|
EXPECT_EQ(DumpInstructions(b.functions()[0].instructions()),
|
|
R"(OpSelectionMerge %3 None
|
|
OpBranchConditional %2 %4 %3
|
|
%4 = OpLabel
|
|
OpReturnValue %5
|
|
%3 = OpLabel
|
|
)");
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace spirv
|
|
} // namespace writer
|
|
} // namespace tint
|