dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 1 | // Copyright 2020 The Tint Authors. |
| 2 | // |
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | // you may not use this file except in compliance with the License. |
| 5 | // You may obtain a copy of the License at |
| 6 | // |
| 7 | // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | // |
| 9 | // Unless required by applicable law or agreed to in writing, software |
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | // See the License for the specific language governing permissions and |
| 13 | // limitations under the License. |
| 14 | |
| 15 | #include "src/type_determiner.h" |
| 16 | |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 17 | #include <algorithm> |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 18 | #include <memory> |
| 19 | #include <utility> |
dan sinclair | fd5d4ca | 2020-04-20 15:46:18 +0000 | [diff] [blame] | 20 | #include <vector> |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 21 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 22 | #include "gmock/gmock.h" |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 23 | #include "gtest/gtest.h" |
dan sinclair | 973bd6a | 2020-04-07 12:57:42 +0000 | [diff] [blame] | 24 | #include "src/ast/array_accessor_expression.h" |
dan sinclair | 6c498fc | 2020-04-07 12:47:23 +0000 | [diff] [blame] | 25 | #include "src/ast/assignment_statement.h" |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 26 | #include "src/ast/binary_expression.h" |
dan sinclair | a7d498e | 2020-09-22 22:07:13 +0000 | [diff] [blame] | 27 | #include "src/ast/bitcast_expression.h" |
dan sinclair | 0975dd5 | 2020-07-27 15:25:00 +0000 | [diff] [blame] | 28 | #include "src/ast/block_statement.h" |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 29 | #include "src/ast/bool_literal.h" |
dan sinclair | b7ea6e2 | 2020-04-07 12:54:10 +0000 | [diff] [blame] | 30 | #include "src/ast/break_statement.h" |
dan sinclair | 3ca8746 | 2020-04-07 16:41:10 +0000 | [diff] [blame] | 31 | #include "src/ast/call_expression.h" |
dan sinclair | 50080b7 | 2020-07-21 13:42:13 +0000 | [diff] [blame] | 32 | #include "src/ast/call_statement.h" |
dan sinclair | 6010b29 | 2020-04-07 12:54:20 +0000 | [diff] [blame] | 33 | #include "src/ast/case_statement.h" |
dan sinclair | aec965e | 2020-04-07 12:54:29 +0000 | [diff] [blame] | 34 | #include "src/ast/continue_statement.h" |
dan sinclair | 0cf685f | 2020-04-07 12:54:37 +0000 | [diff] [blame] | 35 | #include "src/ast/else_statement.h" |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 36 | #include "src/ast/float_literal.h" |
dan sinclair | cab0e73 | 2020-04-07 12:57:27 +0000 | [diff] [blame] | 37 | #include "src/ast/identifier_expression.h" |
dan sinclair | 91c44a5 | 2020-04-07 12:55:25 +0000 | [diff] [blame] | 38 | #include "src/ast/if_statement.h" |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 39 | #include "src/ast/intrinsic_texture_helper_test.h" |
dan sinclair | bc71eda | 2020-04-07 12:55:51 +0000 | [diff] [blame] | 40 | #include "src/ast/loop_statement.h" |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 41 | #include "src/ast/member_accessor_expression.h" |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 42 | #include "src/ast/pipeline_stage.h" |
dan sinclair | bf0fff8 | 2020-04-07 12:56:24 +0000 | [diff] [blame] | 43 | #include "src/ast/return_statement.h" |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 44 | #include "src/ast/scalar_constructor_expression.h" |
dan sinclair | c6f2947 | 2020-06-02 20:11:44 +0000 | [diff] [blame] | 45 | #include "src/ast/sint_literal.h" |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 46 | #include "src/ast/stage_decoration.h" |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 47 | #include "src/ast/struct.h" |
| 48 | #include "src/ast/struct_member.h" |
dan sinclair | 18b3285 | 2020-04-07 12:56:45 +0000 | [diff] [blame] | 49 | #include "src/ast/switch_statement.h" |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 50 | #include "src/ast/type_constructor_expression.h" |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 51 | #include "src/ast/uint_literal.h" |
dan sinclair | 327ed1b | 2020-04-07 19:27:21 +0000 | [diff] [blame] | 52 | #include "src/ast/unary_op_expression.h" |
dan sinclair | ca893e3 | 2020-04-07 12:57:12 +0000 | [diff] [blame] | 53 | #include "src/ast/variable_decl_statement.h" |
Ben Clayton | a6b9a8e | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 54 | #include "src/program_builder.h" |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 55 | #include "src/semantic/call.h" |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 56 | #include "src/semantic/expression.h" |
Ben Clayton | 87c78dd | 2021-02-03 16:43:20 +0000 | [diff] [blame] | 57 | #include "src/semantic/function.h" |
Ben Clayton | 6d612ad | 2021-02-24 14:15:02 +0000 | [diff] [blame] | 58 | #include "src/semantic/member_accessor_expression.h" |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 59 | #include "src/semantic/statement.h" |
Ben Clayton | b17aea1 | 2021-02-03 17:51:09 +0000 | [diff] [blame] | 60 | #include "src/semantic/variable.h" |
Ben Clayton | faca02d | 2021-02-10 21:34:25 +0000 | [diff] [blame] | 61 | #include "src/type/access_control_type.h" |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 62 | #include "src/type/alias_type.h" |
| 63 | #include "src/type/array_type.h" |
| 64 | #include "src/type/bool_type.h" |
| 65 | #include "src/type/depth_texture_type.h" |
| 66 | #include "src/type/f32_type.h" |
| 67 | #include "src/type/i32_type.h" |
| 68 | #include "src/type/matrix_type.h" |
| 69 | #include "src/type/multisampled_texture_type.h" |
| 70 | #include "src/type/pointer_type.h" |
| 71 | #include "src/type/sampled_texture_type.h" |
| 72 | #include "src/type/sampler_type.h" |
| 73 | #include "src/type/storage_texture_type.h" |
| 74 | #include "src/type/struct_type.h" |
| 75 | #include "src/type/texture_type.h" |
| 76 | #include "src/type/u32_type.h" |
| 77 | #include "src/type/vector_type.h" |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 78 | |
Ben Clayton | 6d612ad | 2021-02-24 14:15:02 +0000 | [diff] [blame] | 79 | using ::testing::ElementsAre; |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 80 | using ::testing::HasSubstr; |
| 81 | |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 82 | namespace tint { |
| 83 | namespace { |
| 84 | |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 85 | using IntrinsicType = semantic::IntrinsicType; |
| 86 | |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 87 | class FakeStmt : public ast::Statement { |
| 88 | public: |
Ben Clayton | 604bc72 | 2020-12-12 01:24:53 +0000 | [diff] [blame] | 89 | explicit FakeStmt(Source source) : ast::Statement(source) {} |
Ben Clayton | 1e29f4b | 2021-01-21 16:20:40 +0000 | [diff] [blame] | 90 | FakeStmt* Clone(CloneContext*) const override { return nullptr; } |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 91 | bool IsValid() const override { return true; } |
Ben Clayton | dd1b6fc | 2021-01-29 10:55:40 +0000 | [diff] [blame] | 92 | void to_str(const semantic::Info&, std::ostream& out, size_t) const override { |
| 93 | out << "Fake"; |
| 94 | } |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 95 | }; |
| 96 | |
| 97 | class FakeExpr : public ast::Expression { |
| 98 | public: |
Ben Clayton | 604bc72 | 2020-12-12 01:24:53 +0000 | [diff] [blame] | 99 | explicit FakeExpr(Source source) : ast::Expression(source) {} |
Ben Clayton | 1e29f4b | 2021-01-21 16:20:40 +0000 | [diff] [blame] | 100 | FakeExpr* Clone(CloneContext*) const override { return nullptr; } |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 101 | bool IsValid() const override { return true; } |
Ben Clayton | dd1b6fc | 2021-01-29 10:55:40 +0000 | [diff] [blame] | 102 | void to_str(const semantic::Info&, std::ostream&, size_t) const override {} |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 103 | }; |
| 104 | |
Ben Clayton | a6b9a8e | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 105 | class TypeDeterminerHelper : public ProgramBuilder { |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 106 | public: |
Ben Clayton | a6b9a8e | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 107 | TypeDeterminerHelper() : td_(std::make_unique<TypeDeterminer>(this)) {} |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 108 | |
| 109 | TypeDeterminer* td() const { return td_.get(); } |
Ben Clayton | 6262592 | 2020-11-13 22:09:38 +0000 | [diff] [blame] | 110 | |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 111 | ast::Statement* StmtOf(ast::Expression* expr) { |
| 112 | auto* sem_stmt = Sem().Get(expr)->Stmt(); |
| 113 | return sem_stmt ? sem_stmt->Declaration() : nullptr; |
| 114 | } |
| 115 | |
James Price | c9af597 | 2021-02-16 21:15:01 +0000 | [diff] [blame] | 116 | bool CheckVarUsers(ast::Variable* var, |
| 117 | std::vector<ast::Expression*>&& expected_users) { |
| 118 | auto& var_users = Sem().Get(var)->Users(); |
| 119 | if (var_users.size() != expected_users.size()) { |
| 120 | return false; |
| 121 | } |
| 122 | for (size_t i = 0; i < var_users.size(); i++) { |
| 123 | if (var_users[i]->Declaration() != expected_users[i]) { |
| 124 | return false; |
| 125 | } |
| 126 | } |
| 127 | return true; |
| 128 | } |
| 129 | |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 130 | private: |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 131 | std::unique_ptr<TypeDeterminer> td_; |
| 132 | }; |
| 133 | |
dan sinclair | cd077b0 | 2020-04-20 14:19:04 +0000 | [diff] [blame] | 134 | class TypeDeterminerTest : public TypeDeterminerHelper, public testing::Test {}; |
| 135 | |
| 136 | template <typename T> |
| 137 | class TypeDeterminerTestWithParam : public TypeDeterminerHelper, |
| 138 | public testing::TestWithParam<T> {}; |
| 139 | |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 140 | TEST_F(TypeDeterminerTest, Error_WithEmptySource) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 141 | auto* s = create<FakeStmt>(); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 142 | WrapInFunction(s); |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 143 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 144 | EXPECT_FALSE(td()->Determine()); |
| 145 | |
dan sinclair | 0975dd5 | 2020-07-27 15:25:00 +0000 | [diff] [blame] | 146 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 147 | "error: unknown statement type for type determination: Fake"); |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 148 | } |
| 149 | |
| 150 | TEST_F(TypeDeterminerTest, Stmt_Error_Unknown) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 151 | auto* s = create<FakeStmt>(Source{Source::Location{2, 30}}); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 152 | WrapInFunction(s); |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 153 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 154 | EXPECT_FALSE(td()->Determine()); |
| 155 | |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 156 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 157 | "2:30 error: unknown statement type for type determination: Fake"); |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 158 | } |
| 159 | |
dan sinclair | 6c498fc | 2020-04-07 12:47:23 +0000 | [diff] [blame] | 160 | TEST_F(TypeDeterminerTest, Stmt_Assign) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 161 | auto* lhs = Expr(2); |
| 162 | auto* rhs = Expr(2.3f); |
dan sinclair | 6c498fc | 2020-04-07 12:47:23 +0000 | [diff] [blame] | 163 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 164 | auto* assign = create<ast::AssignmentStatement>(lhs, rhs); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 165 | WrapInFunction(assign); |
dan sinclair | 6c498fc | 2020-04-07 12:47:23 +0000 | [diff] [blame] | 166 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 167 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 168 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 169 | ASSERT_NE(TypeOf(lhs), nullptr); |
| 170 | ASSERT_NE(TypeOf(rhs), nullptr); |
dan sinclair | 6c498fc | 2020-04-07 12:47:23 +0000 | [diff] [blame] | 171 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 172 | EXPECT_TRUE(TypeOf(lhs)->Is<type::I32>()); |
| 173 | EXPECT_TRUE(TypeOf(rhs)->Is<type::F32>()); |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 174 | EXPECT_EQ(StmtOf(lhs), assign); |
| 175 | EXPECT_EQ(StmtOf(rhs), assign); |
dan sinclair | 6c498fc | 2020-04-07 12:47:23 +0000 | [diff] [blame] | 176 | } |
| 177 | |
dan sinclair | 6010b29 | 2020-04-07 12:54:20 +0000 | [diff] [blame] | 178 | TEST_F(TypeDeterminerTest, Stmt_Case) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 179 | auto* lhs = Expr(2); |
| 180 | auto* rhs = Expr(2.3f); |
dan sinclair | 6010b29 | 2020-04-07 12:54:20 +0000 | [diff] [blame] | 181 | |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 182 | auto* assign = create<ast::AssignmentStatement>(lhs, rhs); |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 183 | auto* body = create<ast::BlockStatement>(ast::StatementList{ |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 184 | assign, |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 185 | }); |
dan sinclair | 1aadbd4 | 2020-06-01 16:56:46 +0000 | [diff] [blame] | 186 | ast::CaseSelectorList lit; |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 187 | lit.push_back(create<ast::SintLiteral>(ty.i32(), 3)); |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 188 | auto* cse = create<ast::CaseStatement>(lit, body); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 189 | WrapInFunction(cse); |
dan sinclair | 6010b29 | 2020-04-07 12:54:20 +0000 | [diff] [blame] | 190 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 191 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 192 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 193 | ASSERT_NE(TypeOf(lhs), nullptr); |
| 194 | ASSERT_NE(TypeOf(rhs), nullptr); |
| 195 | EXPECT_TRUE(TypeOf(lhs)->Is<type::I32>()); |
| 196 | EXPECT_TRUE(TypeOf(rhs)->Is<type::F32>()); |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 197 | EXPECT_EQ(StmtOf(lhs), assign); |
| 198 | EXPECT_EQ(StmtOf(rhs), assign); |
dan sinclair | 6010b29 | 2020-04-07 12:54:20 +0000 | [diff] [blame] | 199 | } |
| 200 | |
dan sinclair | 0975dd5 | 2020-07-27 15:25:00 +0000 | [diff] [blame] | 201 | TEST_F(TypeDeterminerTest, Stmt_Block) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 202 | auto* lhs = Expr(2); |
| 203 | auto* rhs = Expr(2.3f); |
dan sinclair | 0975dd5 | 2020-07-27 15:25:00 +0000 | [diff] [blame] | 204 | |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 205 | auto* assign = create<ast::AssignmentStatement>(lhs, rhs); |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 206 | auto* block = create<ast::BlockStatement>(ast::StatementList{ |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 207 | assign, |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 208 | }); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 209 | WrapInFunction(block); |
dan sinclair | 0975dd5 | 2020-07-27 15:25:00 +0000 | [diff] [blame] | 210 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 211 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 212 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 213 | ASSERT_NE(TypeOf(lhs), nullptr); |
| 214 | ASSERT_NE(TypeOf(rhs), nullptr); |
| 215 | EXPECT_TRUE(TypeOf(lhs)->Is<type::I32>()); |
| 216 | EXPECT_TRUE(TypeOf(rhs)->Is<type::F32>()); |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 217 | EXPECT_EQ(StmtOf(lhs), assign); |
| 218 | EXPECT_EQ(StmtOf(rhs), assign); |
dan sinclair | 0975dd5 | 2020-07-27 15:25:00 +0000 | [diff] [blame] | 219 | } |
| 220 | |
dan sinclair | 0cf685f | 2020-04-07 12:54:37 +0000 | [diff] [blame] | 221 | TEST_F(TypeDeterminerTest, Stmt_Else) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 222 | auto* lhs = Expr(2); |
| 223 | auto* rhs = Expr(2.3f); |
dan sinclair | 0cf685f | 2020-04-07 12:54:37 +0000 | [diff] [blame] | 224 | |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 225 | auto* assign = create<ast::AssignmentStatement>(lhs, rhs); |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 226 | auto* body = create<ast::BlockStatement>(ast::StatementList{ |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 227 | assign, |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 228 | }); |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 229 | auto* cond = Expr(3); |
| 230 | auto* stmt = create<ast::ElseStatement>(cond, body); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 231 | WrapInFunction(stmt); |
dan sinclair | 0cf685f | 2020-04-07 12:54:37 +0000 | [diff] [blame] | 232 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 233 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 234 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 235 | ASSERT_NE(TypeOf(stmt->condition()), nullptr); |
| 236 | ASSERT_NE(TypeOf(lhs), nullptr); |
| 237 | ASSERT_NE(TypeOf(rhs), nullptr); |
| 238 | EXPECT_TRUE(TypeOf(stmt->condition())->Is<type::I32>()); |
| 239 | EXPECT_TRUE(TypeOf(lhs)->Is<type::I32>()); |
| 240 | EXPECT_TRUE(TypeOf(rhs)->Is<type::F32>()); |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 241 | EXPECT_EQ(StmtOf(lhs), assign); |
| 242 | EXPECT_EQ(StmtOf(rhs), assign); |
| 243 | EXPECT_EQ(StmtOf(cond), stmt); |
dan sinclair | 0cf685f | 2020-04-07 12:54:37 +0000 | [diff] [blame] | 244 | } |
| 245 | |
dan sinclair | 91c44a5 | 2020-04-07 12:55:25 +0000 | [diff] [blame] | 246 | TEST_F(TypeDeterminerTest, Stmt_If) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 247 | auto* else_lhs = Expr(2); |
| 248 | auto* else_rhs = Expr(2.3f); |
dan sinclair | 91c44a5 | 2020-04-07 12:55:25 +0000 | [diff] [blame] | 249 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 250 | auto* else_body = create<ast::BlockStatement>(ast::StatementList{ |
| 251 | create<ast::AssignmentStatement>(else_lhs, else_rhs), |
| 252 | }); |
dan sinclair | 91c44a5 | 2020-04-07 12:55:25 +0000 | [diff] [blame] | 253 | |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 254 | auto* else_cond = Expr(3); |
| 255 | auto* else_stmt = create<ast::ElseStatement>(else_cond, else_body); |
dan sinclair | 91c44a5 | 2020-04-07 12:55:25 +0000 | [diff] [blame] | 256 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 257 | auto* lhs = Expr(2); |
| 258 | auto* rhs = Expr(2.3f); |
dan sinclair | 91c44a5 | 2020-04-07 12:55:25 +0000 | [diff] [blame] | 259 | |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 260 | auto* assign = create<ast::AssignmentStatement>(lhs, rhs); |
| 261 | auto* body = create<ast::BlockStatement>(ast::StatementList{assign}); |
| 262 | auto* cond = Expr(3); |
| 263 | auto* stmt = |
| 264 | create<ast::IfStatement>(cond, body, ast::ElseStatementList{else_stmt}); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 265 | WrapInFunction(stmt); |
dan sinclair | 91c44a5 | 2020-04-07 12:55:25 +0000 | [diff] [blame] | 266 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 267 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 268 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 269 | ASSERT_NE(TypeOf(stmt->condition()), nullptr); |
| 270 | ASSERT_NE(TypeOf(else_lhs), nullptr); |
| 271 | ASSERT_NE(TypeOf(else_rhs), nullptr); |
| 272 | ASSERT_NE(TypeOf(lhs), nullptr); |
| 273 | ASSERT_NE(TypeOf(rhs), nullptr); |
| 274 | EXPECT_TRUE(TypeOf(stmt->condition())->Is<type::I32>()); |
| 275 | EXPECT_TRUE(TypeOf(else_lhs)->Is<type::I32>()); |
| 276 | EXPECT_TRUE(TypeOf(else_rhs)->Is<type::F32>()); |
| 277 | EXPECT_TRUE(TypeOf(lhs)->Is<type::I32>()); |
| 278 | EXPECT_TRUE(TypeOf(rhs)->Is<type::F32>()); |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 279 | EXPECT_EQ(StmtOf(lhs), assign); |
| 280 | EXPECT_EQ(StmtOf(rhs), assign); |
| 281 | EXPECT_EQ(StmtOf(cond), stmt); |
| 282 | EXPECT_EQ(StmtOf(else_cond), else_stmt); |
dan sinclair | 91c44a5 | 2020-04-07 12:55:25 +0000 | [diff] [blame] | 283 | } |
| 284 | |
dan sinclair | bc71eda | 2020-04-07 12:55:51 +0000 | [diff] [blame] | 285 | TEST_F(TypeDeterminerTest, Stmt_Loop) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 286 | auto* body_lhs = Expr(2); |
| 287 | auto* body_rhs = Expr(2.3f); |
dan sinclair | bc71eda | 2020-04-07 12:55:51 +0000 | [diff] [blame] | 288 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 289 | auto* body = create<ast::BlockStatement>(ast::StatementList{ |
| 290 | create<ast::AssignmentStatement>(body_lhs, body_rhs), |
| 291 | }); |
| 292 | auto* continuing_lhs = Expr(2); |
| 293 | auto* continuing_rhs = Expr(2.3f); |
dan sinclair | bc71eda | 2020-04-07 12:55:51 +0000 | [diff] [blame] | 294 | |
Ben Clayton | db5ce65 | 2020-12-14 20:25:27 +0000 | [diff] [blame] | 295 | auto* continuing = create<ast::BlockStatement>( |
dan sinclair | bc71eda | 2020-04-07 12:55:51 +0000 | [diff] [blame] | 296 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 297 | ast::StatementList{ |
| 298 | create<ast::AssignmentStatement>(continuing_lhs, continuing_rhs), |
| 299 | }); |
| 300 | auto* stmt = create<ast::LoopStatement>(body, continuing); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 301 | WrapInFunction(stmt); |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 302 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 303 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 304 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 305 | ASSERT_NE(TypeOf(body_lhs), nullptr); |
| 306 | ASSERT_NE(TypeOf(body_rhs), nullptr); |
| 307 | ASSERT_NE(TypeOf(continuing_lhs), nullptr); |
| 308 | ASSERT_NE(TypeOf(continuing_rhs), nullptr); |
| 309 | EXPECT_TRUE(TypeOf(body_lhs)->Is<type::I32>()); |
| 310 | EXPECT_TRUE(TypeOf(body_rhs)->Is<type::F32>()); |
| 311 | EXPECT_TRUE(TypeOf(continuing_lhs)->Is<type::I32>()); |
| 312 | EXPECT_TRUE(TypeOf(continuing_rhs)->Is<type::F32>()); |
dan sinclair | bc71eda | 2020-04-07 12:55:51 +0000 | [diff] [blame] | 313 | } |
| 314 | |
dan sinclair | bf0fff8 | 2020-04-07 12:56:24 +0000 | [diff] [blame] | 315 | TEST_F(TypeDeterminerTest, Stmt_Return) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 316 | auto* cond = Expr(2); |
dan sinclair | bf0fff8 | 2020-04-07 12:56:24 +0000 | [diff] [blame] | 317 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 318 | auto* ret = create<ast::ReturnStatement>(cond); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 319 | WrapInFunction(ret); |
dan sinclair | bf0fff8 | 2020-04-07 12:56:24 +0000 | [diff] [blame] | 320 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 321 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 322 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 323 | ASSERT_NE(TypeOf(cond), nullptr); |
| 324 | EXPECT_TRUE(TypeOf(cond)->Is<type::I32>()); |
dan sinclair | bf0fff8 | 2020-04-07 12:56:24 +0000 | [diff] [blame] | 325 | } |
| 326 | |
dan sinclair | 327ed1b | 2020-04-07 19:27:21 +0000 | [diff] [blame] | 327 | TEST_F(TypeDeterminerTest, Stmt_Return_WithoutValue) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 328 | auto* ret = create<ast::ReturnStatement>(); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 329 | WrapInFunction(ret); |
| 330 | |
| 331 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | 327ed1b | 2020-04-07 19:27:21 +0000 | [diff] [blame] | 332 | } |
| 333 | |
dan sinclair | 18b3285 | 2020-04-07 12:56:45 +0000 | [diff] [blame] | 334 | TEST_F(TypeDeterminerTest, Stmt_Switch) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 335 | auto* lhs = Expr(2); |
| 336 | auto* rhs = Expr(2.3f); |
dan sinclair | 18b3285 | 2020-04-07 12:56:45 +0000 | [diff] [blame] | 337 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 338 | auto* body = create<ast::BlockStatement>(ast::StatementList{ |
| 339 | create<ast::AssignmentStatement>(lhs, rhs), |
| 340 | }); |
dan sinclair | 1aadbd4 | 2020-06-01 16:56:46 +0000 | [diff] [blame] | 341 | ast::CaseSelectorList lit; |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 342 | lit.push_back(create<ast::SintLiteral>(ty.i32(), 3)); |
dan sinclair | 1aadbd4 | 2020-06-01 16:56:46 +0000 | [diff] [blame] | 343 | |
dan sinclair | 18b3285 | 2020-04-07 12:56:45 +0000 | [diff] [blame] | 344 | ast::CaseStatementList cases; |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 345 | cases.push_back(create<ast::CaseStatement>(lit, body)); |
dan sinclair | 18b3285 | 2020-04-07 12:56:45 +0000 | [diff] [blame] | 346 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 347 | auto* stmt = create<ast::SwitchStatement>(Expr(2), cases); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 348 | WrapInFunction(stmt); |
dan sinclair | 18b3285 | 2020-04-07 12:56:45 +0000 | [diff] [blame] | 349 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 350 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 351 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 352 | ASSERT_NE(TypeOf(stmt->condition()), nullptr); |
| 353 | ASSERT_NE(TypeOf(lhs), nullptr); |
| 354 | ASSERT_NE(TypeOf(rhs), nullptr); |
dan sinclair | 18b3285 | 2020-04-07 12:56:45 +0000 | [diff] [blame] | 355 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 356 | EXPECT_TRUE(TypeOf(stmt->condition())->Is<type::I32>()); |
| 357 | EXPECT_TRUE(TypeOf(lhs)->Is<type::I32>()); |
| 358 | EXPECT_TRUE(TypeOf(rhs)->Is<type::F32>()); |
dan sinclair | 18b3285 | 2020-04-07 12:56:45 +0000 | [diff] [blame] | 359 | } |
| 360 | |
dan sinclair | 50080b7 | 2020-07-21 13:42:13 +0000 | [diff] [blame] | 361 | TEST_F(TypeDeterminerTest, Stmt_Call) { |
dan sinclair | 50080b7 | 2020-07-21 13:42:13 +0000 | [diff] [blame] | 362 | ast::VariableList params; |
Ben Clayton | 42d1e09 | 2021-02-02 14:29:15 +0000 | [diff] [blame] | 363 | Func("my_func", params, ty.f32(), ast::StatementList{}, |
| 364 | ast::FunctionDecorationList{}); |
dan sinclair | 50080b7 | 2020-07-21 13:42:13 +0000 | [diff] [blame] | 365 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 366 | auto* expr = Call("my_func"); |
dan sinclair | 50080b7 | 2020-07-21 13:42:13 +0000 | [diff] [blame] | 367 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 368 | auto* call = create<ast::CallStatement>(expr); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 369 | WrapInFunction(call); |
| 370 | |
| 371 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 372 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 373 | ASSERT_NE(TypeOf(expr), nullptr); |
| 374 | EXPECT_TRUE(TypeOf(expr)->Is<type::F32>()); |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 375 | EXPECT_EQ(StmtOf(expr), call); |
dan sinclair | 50080b7 | 2020-07-21 13:42:13 +0000 | [diff] [blame] | 376 | } |
| 377 | |
Sarah Mashayekhi | 844f632 | 2020-08-18 02:10:03 +0000 | [diff] [blame] | 378 | TEST_F(TypeDeterminerTest, Stmt_Call_undeclared) { |
| 379 | // fn main() -> void {func(); return; } |
| 380 | // fn func() -> void { return; } |
Ben Clayton | db5ce65 | 2020-12-14 20:25:27 +0000 | [diff] [blame] | 381 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 382 | SetSource(Source::Location{12, 34}); |
| 383 | auto* call_expr = Call("func"); |
| 384 | ast::VariableList params0; |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 385 | |
Ben Clayton | 42d1e09 | 2021-02-02 14:29:15 +0000 | [diff] [blame] | 386 | Func("main", params0, ty.f32(), |
| 387 | ast::StatementList{ |
| 388 | create<ast::CallStatement>(call_expr), |
| 389 | create<ast::ReturnStatement>(), |
| 390 | }, |
| 391 | ast::FunctionDecorationList{}); |
Sarah Mashayekhi | 844f632 | 2020-08-18 02:10:03 +0000 | [diff] [blame] | 392 | |
Ben Clayton | 42d1e09 | 2021-02-02 14:29:15 +0000 | [diff] [blame] | 393 | Func("func", params0, ty.f32(), |
| 394 | ast::StatementList{ |
| 395 | create<ast::ReturnStatement>(), |
| 396 | }, |
| 397 | ast::FunctionDecorationList{}); |
Sarah Mashayekhi | 844f632 | 2020-08-18 02:10:03 +0000 | [diff] [blame] | 398 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 399 | EXPECT_FALSE(td()->Determine()); |
| 400 | |
Ben Clayton | 33a8cdd | 2021-02-24 13:31:22 +0000 | [diff] [blame] | 401 | EXPECT_EQ(td()->error(), |
| 402 | "12:34 error: v-0006: unable to find called function: func"); |
Sarah Mashayekhi | 844f632 | 2020-08-18 02:10:03 +0000 | [diff] [blame] | 403 | } |
| 404 | |
dan sinclair | f8fa6cf | 2021-02-24 22:11:34 +0000 | [diff] [blame] | 405 | TEST_F(TypeDeterminerTest, Stmt_Call_recursive) { |
| 406 | // fn main() -> void {main(); } |
| 407 | |
| 408 | SetSource(Source::Location{12, 34}); |
| 409 | auto* call_expr = Call("main"); |
| 410 | ast::VariableList params0; |
| 411 | |
| 412 | Func("main", params0, ty.f32(), |
| 413 | ast::StatementList{ |
| 414 | create<ast::CallStatement>(call_expr), |
| 415 | }, |
| 416 | ast::FunctionDecorationList{ |
| 417 | create<ast::StageDecoration>(ast::PipelineStage::kVertex), |
| 418 | }); |
| 419 | |
| 420 | EXPECT_FALSE(td()->Determine()); |
| 421 | |
| 422 | EXPECT_EQ(td()->error(), |
| 423 | "12:34 error: recursion is not permitted. 'main' attempted to call " |
| 424 | "itself."); |
| 425 | } |
| 426 | |
dan sinclair | ca893e3 | 2020-04-07 12:57:12 +0000 | [diff] [blame] | 427 | TEST_F(TypeDeterminerTest, Stmt_VariableDecl) { |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 428 | auto* var = Var("my_var", ty.i32(), ast::StorageClass::kNone, Expr(2)); |
Ben Clayton | 4bfe461 | 2020-11-16 16:41:47 +0000 | [diff] [blame] | 429 | auto* init = var->constructor(); |
dan sinclair | ca893e3 | 2020-04-07 12:57:12 +0000 | [diff] [blame] | 430 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 431 | auto* decl = create<ast::VariableDeclStatement>(var); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 432 | WrapInFunction(decl); |
dan sinclair | ca893e3 | 2020-04-07 12:57:12 +0000 | [diff] [blame] | 433 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 434 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 435 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 436 | ASSERT_NE(TypeOf(init), nullptr); |
| 437 | EXPECT_TRUE(TypeOf(init)->Is<type::I32>()); |
dan sinclair | ca893e3 | 2020-04-07 12:57:12 +0000 | [diff] [blame] | 438 | } |
| 439 | |
Antonio Maiorano | 6ce2edf | 2021-02-26 18:25:56 +0000 | [diff] [blame] | 440 | TEST_F(TypeDeterminerTest, Stmt_VariableDecl_Alias) { |
| 441 | auto* my_int = ty.alias("MyInt", ty.i32()); |
| 442 | auto* var = Var("my_var", my_int, ast::StorageClass::kNone, Expr(2)); |
| 443 | auto* init = var->constructor(); |
| 444 | |
| 445 | auto* decl = create<ast::VariableDeclStatement>(var); |
| 446 | WrapInFunction(decl); |
| 447 | |
| 448 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 449 | |
| 450 | ASSERT_NE(TypeOf(init), nullptr); |
| 451 | EXPECT_TRUE(TypeOf(init)->Is<type::I32>()); |
| 452 | } |
| 453 | |
| 454 | TEST_F(TypeDeterminerTest, Stmt_VariableDecl_MismatchedTypeScalarConstructor) { |
| 455 | u32 unsigned_value = 2u; // Type does not match variable type |
| 456 | auto* var = |
| 457 | Var("my_var", ty.i32(), ast::StorageClass::kNone, Expr(unsigned_value)); |
| 458 | |
| 459 | auto* decl = |
| 460 | create<ast::VariableDeclStatement>(Source{{{3, 3}, {3, 22}}}, var); |
| 461 | WrapInFunction(decl); |
| 462 | |
| 463 | EXPECT_FALSE(td()->Determine()); |
| 464 | EXPECT_EQ( |
| 465 | td()->error(), |
| 466 | R"(3:3 error: constructor expression type does not match variable type)"); |
| 467 | } |
| 468 | |
| 469 | TEST_F(TypeDeterminerTest, |
| 470 | Stmt_VariableDecl_MismatchedTypeScalarConstructor_Alias) { |
| 471 | auto* my_int = ty.alias("MyInt", ty.i32()); |
| 472 | u32 unsigned_value = 2u; // Type does not match variable type |
| 473 | auto* var = |
| 474 | Var("my_var", my_int, ast::StorageClass::kNone, Expr(unsigned_value)); |
| 475 | |
| 476 | auto* decl = |
| 477 | create<ast::VariableDeclStatement>(Source{{{3, 3}, {3, 22}}}, var); |
| 478 | WrapInFunction(decl); |
| 479 | |
| 480 | EXPECT_FALSE(td()->Determine()); |
| 481 | EXPECT_EQ( |
| 482 | td()->error(), |
| 483 | R"(3:3 error: constructor expression type does not match variable type)"); |
| 484 | } |
| 485 | |
dan sinclair | 7be237a | 2020-06-15 20:55:09 +0000 | [diff] [blame] | 486 | TEST_F(TypeDeterminerTest, Stmt_VariableDecl_ModuleScope) { |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 487 | auto* init = Expr(2); |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 488 | Global("my_var", ty.i32(), ast::StorageClass::kNone, init); |
dan sinclair | 7be237a | 2020-06-15 20:55:09 +0000 | [diff] [blame] | 489 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 490 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | 7be237a | 2020-06-15 20:55:09 +0000 | [diff] [blame] | 491 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 492 | ASSERT_NE(TypeOf(init), nullptr); |
| 493 | EXPECT_TRUE(TypeOf(init)->Is<type::I32>()); |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 494 | EXPECT_EQ(StmtOf(init), nullptr); |
dan sinclair | 7be237a | 2020-06-15 20:55:09 +0000 | [diff] [blame] | 495 | } |
| 496 | |
James Price | 77f4f02 | 2021-02-04 16:33:20 +0000 | [diff] [blame] | 497 | TEST_F(TypeDeterminerTest, Stmt_VariableDecl_OuterScopeAfterInnerScope) { |
| 498 | // fn func_i32() -> i32 { |
| 499 | // { |
| 500 | // var foo : i32 = 2; |
| 501 | // var bar : i32 = foo; |
| 502 | // } |
| 503 | // var foo : f32 = 2.0; |
| 504 | // var bar : f32 = foo; |
| 505 | // } |
| 506 | |
| 507 | ast::VariableList params; |
| 508 | |
| 509 | // Declare i32 "foo" inside a block |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 510 | auto* foo_i32 = Var("foo", ty.i32(), ast::StorageClass::kNone, Expr(2)); |
James Price | 77f4f02 | 2021-02-04 16:33:20 +0000 | [diff] [blame] | 511 | auto* foo_i32_init = foo_i32->constructor(); |
| 512 | auto* foo_i32_decl = create<ast::VariableDeclStatement>(foo_i32); |
| 513 | |
| 514 | // Reference "foo" inside the block |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 515 | auto* bar_i32 = Var("bar", ty.i32(), ast::StorageClass::kNone, Expr("foo")); |
James Price | 77f4f02 | 2021-02-04 16:33:20 +0000 | [diff] [blame] | 516 | auto* bar_i32_init = bar_i32->constructor(); |
| 517 | auto* bar_i32_decl = create<ast::VariableDeclStatement>(bar_i32); |
| 518 | |
| 519 | auto* inner = create<ast::BlockStatement>( |
| 520 | ast::StatementList{foo_i32_decl, bar_i32_decl}); |
| 521 | |
| 522 | // Declare f32 "foo" at function scope |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 523 | auto* foo_f32 = Var("foo", ty.f32(), ast::StorageClass::kNone, Expr(2.f)); |
James Price | 77f4f02 | 2021-02-04 16:33:20 +0000 | [diff] [blame] | 524 | auto* foo_f32_init = foo_f32->constructor(); |
| 525 | auto* foo_f32_decl = create<ast::VariableDeclStatement>(foo_f32); |
| 526 | |
| 527 | // Reference "foo" at function scope |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 528 | auto* bar_f32 = Var("bar", ty.f32(), ast::StorageClass::kNone, Expr("foo")); |
James Price | 77f4f02 | 2021-02-04 16:33:20 +0000 | [diff] [blame] | 529 | auto* bar_f32_init = bar_f32->constructor(); |
| 530 | auto* bar_f32_decl = create<ast::VariableDeclStatement>(bar_f32); |
| 531 | |
| 532 | Func("func", params, ty.f32(), |
| 533 | ast::StatementList{inner, foo_f32_decl, bar_f32_decl}, |
| 534 | ast::FunctionDecorationList{}); |
| 535 | |
| 536 | EXPECT_TRUE(td()->Determine()); |
| 537 | ASSERT_NE(TypeOf(foo_i32_init), nullptr); |
| 538 | EXPECT_TRUE(TypeOf(foo_i32_init)->Is<type::I32>()); |
| 539 | ASSERT_NE(TypeOf(foo_f32_init), nullptr); |
| 540 | EXPECT_TRUE(TypeOf(foo_f32_init)->Is<type::F32>()); |
| 541 | ASSERT_NE(TypeOf(bar_i32_init), nullptr); |
| 542 | EXPECT_TRUE(TypeOf(bar_i32_init)->UnwrapAll()->Is<type::I32>()); |
| 543 | ASSERT_NE(TypeOf(bar_f32_init), nullptr); |
| 544 | EXPECT_TRUE(TypeOf(bar_f32_init)->UnwrapAll()->Is<type::F32>()); |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 545 | EXPECT_EQ(StmtOf(foo_i32_init), foo_i32_decl); |
| 546 | EXPECT_EQ(StmtOf(bar_i32_init), bar_i32_decl); |
| 547 | EXPECT_EQ(StmtOf(foo_f32_init), foo_f32_decl); |
| 548 | EXPECT_EQ(StmtOf(bar_f32_init), bar_f32_decl); |
James Price | c9af597 | 2021-02-16 21:15:01 +0000 | [diff] [blame] | 549 | EXPECT_TRUE(CheckVarUsers(foo_i32, {bar_i32->constructor()})); |
| 550 | EXPECT_TRUE(CheckVarUsers(foo_f32, {bar_f32->constructor()})); |
James Price | 77f4f02 | 2021-02-04 16:33:20 +0000 | [diff] [blame] | 551 | } |
| 552 | |
| 553 | TEST_F(TypeDeterminerTest, Stmt_VariableDecl_ModuleScopeAfterFunctionScope) { |
| 554 | // fn func_i32() -> i32 { |
| 555 | // var foo : i32 = 2; |
| 556 | // } |
| 557 | // var foo : f32 = 2.0; |
| 558 | // fn func_f32() -> f32 { |
| 559 | // var bar : f32 = foo; |
| 560 | // } |
| 561 | |
| 562 | ast::VariableList params; |
| 563 | |
| 564 | // Declare i32 "foo" inside a function |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 565 | auto* fn_i32 = Var("foo", ty.i32(), ast::StorageClass::kNone, Expr(2)); |
James Price | 77f4f02 | 2021-02-04 16:33:20 +0000 | [diff] [blame] | 566 | auto* fn_i32_init = fn_i32->constructor(); |
| 567 | auto* fn_i32_decl = create<ast::VariableDeclStatement>(fn_i32); |
| 568 | Func("func_i32", params, ty.i32(), ast::StatementList{fn_i32_decl}, |
| 569 | ast::FunctionDecorationList{}); |
| 570 | |
| 571 | // Declare f32 "foo" at module scope |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 572 | auto* mod_f32 = Var("foo", ty.f32(), ast::StorageClass::kNone, Expr(2.f)); |
James Price | 77f4f02 | 2021-02-04 16:33:20 +0000 | [diff] [blame] | 573 | auto* mod_init = mod_f32->constructor(); |
| 574 | AST().AddGlobalVariable(mod_f32); |
| 575 | |
| 576 | // Reference "foo" in another function |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 577 | auto* fn_f32 = Var("bar", ty.f32(), ast::StorageClass::kNone, Expr("foo")); |
James Price | 77f4f02 | 2021-02-04 16:33:20 +0000 | [diff] [blame] | 578 | auto* fn_f32_init = fn_f32->constructor(); |
| 579 | auto* fn_f32_decl = create<ast::VariableDeclStatement>(fn_f32); |
| 580 | Func("func_f32", params, ty.f32(), ast::StatementList{fn_f32_decl}, |
| 581 | ast::FunctionDecorationList{}); |
| 582 | |
| 583 | EXPECT_TRUE(td()->Determine()); |
| 584 | ASSERT_NE(TypeOf(mod_init), nullptr); |
| 585 | EXPECT_TRUE(TypeOf(mod_init)->Is<type::F32>()); |
| 586 | ASSERT_NE(TypeOf(fn_i32_init), nullptr); |
| 587 | EXPECT_TRUE(TypeOf(fn_i32_init)->Is<type::I32>()); |
| 588 | ASSERT_NE(TypeOf(fn_f32_init), nullptr); |
| 589 | EXPECT_TRUE(TypeOf(fn_f32_init)->UnwrapAll()->Is<type::F32>()); |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 590 | EXPECT_EQ(StmtOf(fn_i32_init), fn_i32_decl); |
| 591 | EXPECT_EQ(StmtOf(mod_init), nullptr); |
| 592 | EXPECT_EQ(StmtOf(fn_f32_init), fn_f32_decl); |
James Price | c9af597 | 2021-02-16 21:15:01 +0000 | [diff] [blame] | 593 | EXPECT_TRUE(CheckVarUsers(fn_i32, {})); |
| 594 | EXPECT_TRUE(CheckVarUsers(mod_f32, {fn_f32->constructor()})); |
James Price | 77f4f02 | 2021-02-04 16:33:20 +0000 | [diff] [blame] | 595 | } |
| 596 | |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 597 | TEST_F(TypeDeterminerTest, Expr_Error_Unknown) { |
Ben Clayton | 604bc72 | 2020-12-12 01:24:53 +0000 | [diff] [blame] | 598 | FakeExpr e(Source{Source::Location{2, 30}}); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 599 | WrapInFunction(&e); |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 600 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 601 | EXPECT_FALSE(td()->Determine()); |
| 602 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 603 | EXPECT_EQ(td()->error(), |
| 604 | "2:30 error: unknown expression for type determination"); |
dan sinclair | 7456f42 | 2020-04-08 19:58:35 +0000 | [diff] [blame] | 605 | } |
| 606 | |
dan sinclair | 973bd6a | 2020-04-07 12:57:42 +0000 | [diff] [blame] | 607 | TEST_F(TypeDeterminerTest, Expr_ArrayAccessor_Array) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 608 | auto* idx = Expr(2); |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 609 | Global("my_var", ty.array<f32, 3>(), ast::StorageClass::kFunction); |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 610 | |
dan sinclair | 8b40a67 | 2020-12-16 11:49:10 +0000 | [diff] [blame] | 611 | auto* acc = IndexAccessor("my_var", idx); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 612 | WrapInFunction(acc); |
| 613 | |
| 614 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 615 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 616 | ASSERT_NE(TypeOf(acc), nullptr); |
| 617 | ASSERT_TRUE(TypeOf(acc)->Is<type::Pointer>()); |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 618 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 619 | auto* ptr = TypeOf(acc)->As<type::Pointer>(); |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 620 | EXPECT_TRUE(ptr->type()->Is<type::F32>()); |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 621 | } |
| 622 | |
David Neto | 9c88ea5 | 2020-06-22 20:33:12 +0000 | [diff] [blame] | 623 | TEST_F(TypeDeterminerTest, Expr_ArrayAccessor_Alias_Array) { |
Ben Clayton | 1637cbb | 2021-01-05 15:44:39 +0000 | [diff] [blame] | 624 | auto* aary = ty.alias("myarrty", ty.array<f32, 3>()); |
David Neto | 9c88ea5 | 2020-06-22 20:33:12 +0000 | [diff] [blame] | 625 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 626 | Global("my_var", aary, ast::StorageClass::kFunction); |
David Neto | 9c88ea5 | 2020-06-22 20:33:12 +0000 | [diff] [blame] | 627 | |
dan sinclair | 8b40a67 | 2020-12-16 11:49:10 +0000 | [diff] [blame] | 628 | auto* acc = IndexAccessor("my_var", 2); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 629 | WrapInFunction(acc); |
| 630 | |
| 631 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 632 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 633 | ASSERT_NE(TypeOf(acc), nullptr); |
| 634 | ASSERT_TRUE(TypeOf(acc)->Is<type::Pointer>()); |
David Neto | 9c88ea5 | 2020-06-22 20:33:12 +0000 | [diff] [blame] | 635 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 636 | auto* ptr = TypeOf(acc)->As<type::Pointer>(); |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 637 | EXPECT_TRUE(ptr->type()->Is<type::F32>()); |
David Neto | 9c88ea5 | 2020-06-22 20:33:12 +0000 | [diff] [blame] | 638 | } |
| 639 | |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 640 | TEST_F(TypeDeterminerTest, Expr_ArrayAccessor_Array_Constant) { |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 641 | GlobalConst("my_var", ty.array<f32, 3>()); |
dan sinclair | 973bd6a | 2020-04-07 12:57:42 +0000 | [diff] [blame] | 642 | |
dan sinclair | 8b40a67 | 2020-12-16 11:49:10 +0000 | [diff] [blame] | 643 | auto* acc = IndexAccessor("my_var", 2); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 644 | WrapInFunction(acc); |
| 645 | |
| 646 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 647 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 648 | ASSERT_NE(TypeOf(acc), nullptr); |
| 649 | EXPECT_TRUE(TypeOf(acc)->Is<type::F32>()) << TypeOf(acc)->type_name(); |
dan sinclair | 973bd6a | 2020-04-07 12:57:42 +0000 | [diff] [blame] | 650 | } |
| 651 | |
| 652 | TEST_F(TypeDeterminerTest, Expr_ArrayAccessor_Matrix) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 653 | Global("my_var", ty.mat2x3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 973bd6a | 2020-04-07 12:57:42 +0000 | [diff] [blame] | 654 | |
dan sinclair | 8b40a67 | 2020-12-16 11:49:10 +0000 | [diff] [blame] | 655 | auto* acc = IndexAccessor("my_var", 2); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 656 | WrapInFunction(acc); |
| 657 | |
| 658 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 659 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 660 | ASSERT_NE(TypeOf(acc), nullptr); |
| 661 | ASSERT_TRUE(TypeOf(acc)->Is<type::Pointer>()); |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 662 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 663 | auto* ptr = TypeOf(acc)->As<type::Pointer>(); |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 664 | ASSERT_TRUE(ptr->type()->Is<type::Vector>()); |
| 665 | EXPECT_EQ(ptr->type()->As<type::Vector>()->size(), 3u); |
dan sinclair | 973bd6a | 2020-04-07 12:57:42 +0000 | [diff] [blame] | 666 | } |
| 667 | |
| 668 | TEST_F(TypeDeterminerTest, Expr_ArrayAccessor_Matrix_BothDimensions) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 669 | Global("my_var", ty.mat2x3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 973bd6a | 2020-04-07 12:57:42 +0000 | [diff] [blame] | 670 | |
dan sinclair | 8b40a67 | 2020-12-16 11:49:10 +0000 | [diff] [blame] | 671 | auto* acc = IndexAccessor(IndexAccessor("my_var", 2), 1); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 672 | WrapInFunction(acc); |
dan sinclair | 973bd6a | 2020-04-07 12:57:42 +0000 | [diff] [blame] | 673 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 674 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 675 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 676 | ASSERT_NE(TypeOf(acc), nullptr); |
| 677 | ASSERT_TRUE(TypeOf(acc)->Is<type::Pointer>()); |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 678 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 679 | auto* ptr = TypeOf(acc)->As<type::Pointer>(); |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 680 | EXPECT_TRUE(ptr->type()->Is<type::F32>()); |
dan sinclair | 973bd6a | 2020-04-07 12:57:42 +0000 | [diff] [blame] | 681 | } |
| 682 | |
| 683 | TEST_F(TypeDeterminerTest, Expr_ArrayAccessor_Vector) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 684 | Global("my_var", ty.vec3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 973bd6a | 2020-04-07 12:57:42 +0000 | [diff] [blame] | 685 | |
dan sinclair | 8b40a67 | 2020-12-16 11:49:10 +0000 | [diff] [blame] | 686 | auto* acc = IndexAccessor("my_var", 2); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 687 | WrapInFunction(acc); |
| 688 | |
| 689 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 690 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 691 | ASSERT_NE(TypeOf(acc), nullptr); |
| 692 | ASSERT_TRUE(TypeOf(acc)->Is<type::Pointer>()); |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 693 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 694 | auto* ptr = TypeOf(acc)->As<type::Pointer>(); |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 695 | EXPECT_TRUE(ptr->type()->Is<type::F32>()); |
dan sinclair | 973bd6a | 2020-04-07 12:57:42 +0000 | [diff] [blame] | 696 | } |
| 697 | |
dan sinclair | a7d498e | 2020-09-22 22:07:13 +0000 | [diff] [blame] | 698 | TEST_F(TypeDeterminerTest, Expr_Bitcast) { |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 699 | auto* bitcast = create<ast::BitcastExpression>(ty.f32(), Expr("name")); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 700 | WrapInFunction(bitcast); |
dan sinclair | a01777c | 2020-04-07 12:57:52 +0000 | [diff] [blame] | 701 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 702 | Global("name", ty.f32(), ast::StorageClass::kPrivate); |
dan sinclair | ff267ca | 2020-10-14 18:26:31 +0000 | [diff] [blame] | 703 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 704 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 705 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 706 | ASSERT_NE(TypeOf(bitcast), nullptr); |
| 707 | EXPECT_TRUE(TypeOf(bitcast)->Is<type::F32>()); |
dan sinclair | a01777c | 2020-04-07 12:57:52 +0000 | [diff] [blame] | 708 | } |
| 709 | |
dan sinclair | 3ca8746 | 2020-04-07 16:41:10 +0000 | [diff] [blame] | 710 | TEST_F(TypeDeterminerTest, Expr_Call) { |
dan sinclair | 3ca8746 | 2020-04-07 16:41:10 +0000 | [diff] [blame] | 711 | ast::VariableList params; |
Ben Clayton | 42d1e09 | 2021-02-02 14:29:15 +0000 | [diff] [blame] | 712 | Func("my_func", params, ty.f32(), ast::StatementList{}, |
| 713 | ast::FunctionDecorationList{}); |
dan sinclair | 3ca8746 | 2020-04-07 16:41:10 +0000 | [diff] [blame] | 714 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 715 | auto* call = Call("my_func"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 716 | WrapInFunction(call); |
| 717 | |
| 718 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 719 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 720 | ASSERT_NE(TypeOf(call), nullptr); |
| 721 | EXPECT_TRUE(TypeOf(call)->Is<type::F32>()); |
dan sinclair | 3ca8746 | 2020-04-07 16:41:10 +0000 | [diff] [blame] | 722 | } |
| 723 | |
Ben Clayton | 7b7d698 | 2021-02-09 17:38:05 +0000 | [diff] [blame] | 724 | TEST_F(TypeDeterminerTest, Expr_Call_InBinaryOp) { |
| 725 | ast::VariableList params; |
| 726 | Func("func", params, ty.f32(), ast::StatementList{}, |
| 727 | ast::FunctionDecorationList{}); |
| 728 | |
| 729 | auto* expr = Add(Call("func"), Call("func")); |
| 730 | WrapInFunction(expr); |
| 731 | |
| 732 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 733 | |
| 734 | ASSERT_NE(TypeOf(expr), nullptr); |
| 735 | EXPECT_TRUE(TypeOf(expr)->Is<type::F32>()); |
| 736 | } |
| 737 | |
dan sinclair | ccb52dc | 2020-04-20 14:18:54 +0000 | [diff] [blame] | 738 | TEST_F(TypeDeterminerTest, Expr_Call_WithParams) { |
dan sinclair | ccb52dc | 2020-04-20 14:18:54 +0000 | [diff] [blame] | 739 | ast::VariableList params; |
Ben Clayton | 42d1e09 | 2021-02-02 14:29:15 +0000 | [diff] [blame] | 740 | Func("my_func", params, ty.f32(), ast::StatementList{}, |
| 741 | ast::FunctionDecorationList{}); |
dan sinclair | ccb52dc | 2020-04-20 14:18:54 +0000 | [diff] [blame] | 742 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 743 | auto* param = Expr(2.4f); |
dan sinclair | ccb52dc | 2020-04-20 14:18:54 +0000 | [diff] [blame] | 744 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 745 | auto* call = Call("my_func", param); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 746 | WrapInFunction(call); |
| 747 | |
| 748 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 749 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 750 | ASSERT_NE(TypeOf(param), nullptr); |
| 751 | EXPECT_TRUE(TypeOf(param)->Is<type::F32>()); |
dan sinclair | ccb52dc | 2020-04-20 14:18:54 +0000 | [diff] [blame] | 752 | } |
| 753 | |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 754 | TEST_F(TypeDeterminerTest, Expr_Call_Intrinsic) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 755 | auto* call = Call("round", 2.4f); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 756 | WrapInFunction(call); |
dan sinclair | fd5d4ca | 2020-04-20 15:46:18 +0000 | [diff] [blame] | 757 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 758 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 759 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 760 | ASSERT_NE(TypeOf(call), nullptr); |
| 761 | EXPECT_TRUE(TypeOf(call)->Is<type::F32>()); |
dan sinclair | fd5d4ca | 2020-04-20 15:46:18 +0000 | [diff] [blame] | 762 | } |
| 763 | |
Ben Clayton | 33a8cdd | 2021-02-24 13:31:22 +0000 | [diff] [blame] | 764 | TEST_F(TypeDeterminerTest, Expr_DontCall_Function) { |
| 765 | Func("func", {}, ty.void_(), {}, {}); |
| 766 | auto* ident = create<ast::IdentifierExpression>( |
| 767 | Source{{Source::Location{3, 3}, Source::Location{3, 8}}}, |
| 768 | Symbols().Register("func")); |
| 769 | WrapInFunction(ident); |
| 770 | |
| 771 | EXPECT_FALSE(td()->Determine()); |
| 772 | EXPECT_EQ(td()->error(), "3:8 error: missing '(' for function call"); |
| 773 | } |
| 774 | |
| 775 | TEST_F(TypeDeterminerTest, Expr_DontCall_Intrinsic) { |
| 776 | auto* ident = create<ast::IdentifierExpression>( |
| 777 | Source{{Source::Location{3, 3}, Source::Location{3, 8}}}, |
| 778 | Symbols().Register("round")); |
| 779 | WrapInFunction(ident); |
| 780 | |
| 781 | EXPECT_FALSE(td()->Determine()); |
| 782 | EXPECT_EQ(td()->error(), "3:8 error: missing '(' for intrinsic call"); |
| 783 | } |
| 784 | |
dan sinclair | 4e80795 | 2020-04-07 16:41:23 +0000 | [diff] [blame] | 785 | TEST_F(TypeDeterminerTest, Expr_Cast) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 786 | Global("name", ty.f32(), ast::StorageClass::kPrivate); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 787 | |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 788 | auto* cast = Construct(ty.f32(), "name"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 789 | WrapInFunction(cast); |
dan sinclair | 3c02592 | 2020-09-24 14:38:44 +0000 | [diff] [blame] | 790 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 791 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | 4e80795 | 2020-04-07 16:41:23 +0000 | [diff] [blame] | 792 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 793 | ASSERT_NE(TypeOf(cast), nullptr); |
| 794 | EXPECT_TRUE(TypeOf(cast)->Is<type::F32>()); |
dan sinclair | 4e80795 | 2020-04-07 16:41:23 +0000 | [diff] [blame] | 795 | } |
| 796 | |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 797 | TEST_F(TypeDeterminerTest, Expr_Constructor_Scalar) { |
Ben Clayton | 1637cbb | 2021-01-05 15:44:39 +0000 | [diff] [blame] | 798 | auto* s = Expr(1.0f); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 799 | WrapInFunction(s); |
| 800 | |
| 801 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 802 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 803 | ASSERT_NE(TypeOf(s), nullptr); |
| 804 | EXPECT_TRUE(TypeOf(s)->Is<type::F32>()); |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 805 | } |
| 806 | |
| 807 | TEST_F(TypeDeterminerTest, Expr_Constructor_Type) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 808 | auto* tc = vec3<f32>(1.0f, 1.0f, 3.0f); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 809 | WrapInFunction(tc); |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 810 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 811 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 812 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 813 | ASSERT_NE(TypeOf(tc), nullptr); |
| 814 | ASSERT_TRUE(TypeOf(tc)->Is<type::Vector>()); |
| 815 | EXPECT_TRUE(TypeOf(tc)->As<type::Vector>()->type()->Is<type::F32>()); |
| 816 | EXPECT_EQ(TypeOf(tc)->As<type::Vector>()->size(), 3u); |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 817 | } |
| 818 | |
dan sinclair | cab0e73 | 2020-04-07 12:57:27 +0000 | [diff] [blame] | 819 | TEST_F(TypeDeterminerTest, Expr_Identifier_GlobalVariable) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 820 | auto* my_var = Global("my_var", ty.f32(), ast::StorageClass::kNone); |
dan sinclair | cab0e73 | 2020-04-07 12:57:27 +0000 | [diff] [blame] | 821 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 822 | auto* ident = Expr("my_var"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 823 | WrapInFunction(ident); |
| 824 | |
| 825 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 826 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 827 | ASSERT_NE(TypeOf(ident), nullptr); |
| 828 | EXPECT_TRUE(TypeOf(ident)->Is<type::Pointer>()); |
| 829 | EXPECT_TRUE(TypeOf(ident)->As<type::Pointer>()->type()->Is<type::F32>()); |
James Price | c9af597 | 2021-02-16 21:15:01 +0000 | [diff] [blame] | 830 | EXPECT_TRUE(CheckVarUsers(my_var, {ident})); |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 831 | } |
| 832 | |
| 833 | TEST_F(TypeDeterminerTest, Expr_Identifier_GlobalConstant) { |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 834 | auto* my_var = GlobalConst("my_var", ty.f32()); |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 835 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 836 | auto* ident = Expr("my_var"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 837 | WrapInFunction(ident); |
| 838 | |
| 839 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 840 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 841 | ASSERT_NE(TypeOf(ident), nullptr); |
| 842 | EXPECT_TRUE(TypeOf(ident)->Is<type::F32>()); |
James Price | c9af597 | 2021-02-16 21:15:01 +0000 | [diff] [blame] | 843 | EXPECT_TRUE(CheckVarUsers(my_var, {ident})); |
dan sinclair | cab0e73 | 2020-04-07 12:57:27 +0000 | [diff] [blame] | 844 | } |
| 845 | |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 846 | TEST_F(TypeDeterminerTest, Expr_Identifier_FunctionVariable_Const) { |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 847 | auto* my_var_a = Expr("my_var"); |
| 848 | auto* my_var_b = Expr("my_var"); |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 849 | auto* var = Const("my_var", ty.f32()); |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 850 | auto* assign = create<ast::AssignmentStatement>(my_var_a, my_var_b); |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 851 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 852 | Func("my_func", ast::VariableList{}, ty.f32(), |
| 853 | ast::StatementList{ |
| 854 | create<ast::VariableDeclStatement>(var), |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 855 | assign, |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 856 | }, |
| 857 | ast::FunctionDecorationList{}); |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 858 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 859 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 860 | |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 861 | ASSERT_NE(TypeOf(my_var_a), nullptr); |
| 862 | EXPECT_TRUE(TypeOf(my_var_a)->Is<type::F32>()); |
| 863 | EXPECT_EQ(StmtOf(my_var_a), assign); |
| 864 | ASSERT_NE(TypeOf(my_var_b), nullptr); |
| 865 | EXPECT_TRUE(TypeOf(my_var_b)->Is<type::F32>()); |
| 866 | EXPECT_EQ(StmtOf(my_var_b), assign); |
James Price | c9af597 | 2021-02-16 21:15:01 +0000 | [diff] [blame] | 867 | EXPECT_TRUE(CheckVarUsers(var, {my_var_a, my_var_b})); |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 868 | } |
| 869 | |
dan sinclair | cab0e73 | 2020-04-07 12:57:27 +0000 | [diff] [blame] | 870 | TEST_F(TypeDeterminerTest, Expr_Identifier_FunctionVariable) { |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 871 | auto* my_var_a = Expr("my_var"); |
| 872 | auto* my_var_b = Expr("my_var"); |
| 873 | auto* assign = create<ast::AssignmentStatement>(my_var_a, my_var_b); |
dan sinclair | cab0e73 | 2020-04-07 12:57:27 +0000 | [diff] [blame] | 874 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 875 | auto* var = Var("my_var", ty.f32(), ast::StorageClass::kNone); |
James Price | c9af597 | 2021-02-16 21:15:01 +0000 | [diff] [blame] | 876 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 877 | Func("my_func", ast::VariableList{}, ty.f32(), |
| 878 | ast::StatementList{ |
James Price | c9af597 | 2021-02-16 21:15:01 +0000 | [diff] [blame] | 879 | create<ast::VariableDeclStatement>(var), |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 880 | assign, |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 881 | }, |
| 882 | ast::FunctionDecorationList{}); |
dan sinclair | cab0e73 | 2020-04-07 12:57:27 +0000 | [diff] [blame] | 883 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 884 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | cab0e73 | 2020-04-07 12:57:27 +0000 | [diff] [blame] | 885 | |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 886 | ASSERT_NE(TypeOf(my_var_a), nullptr); |
| 887 | EXPECT_TRUE(TypeOf(my_var_a)->Is<type::Pointer>()); |
| 888 | EXPECT_TRUE(TypeOf(my_var_a)->As<type::Pointer>()->type()->Is<type::F32>()); |
| 889 | EXPECT_EQ(StmtOf(my_var_a), assign); |
| 890 | ASSERT_NE(TypeOf(my_var_b), nullptr); |
| 891 | EXPECT_TRUE(TypeOf(my_var_b)->Is<type::Pointer>()); |
| 892 | EXPECT_TRUE(TypeOf(my_var_b)->As<type::Pointer>()->type()->Is<type::F32>()); |
| 893 | EXPECT_EQ(StmtOf(my_var_b), assign); |
James Price | c9af597 | 2021-02-16 21:15:01 +0000 | [diff] [blame] | 894 | EXPECT_TRUE(CheckVarUsers(var, {my_var_a, my_var_b})); |
dan sinclair | cab0e73 | 2020-04-07 12:57:27 +0000 | [diff] [blame] | 895 | } |
| 896 | |
dan sinclair | 5e98930 | 2020-09-16 21:20:36 +0000 | [diff] [blame] | 897 | TEST_F(TypeDeterminerTest, Expr_Identifier_Function_Ptr) { |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 898 | auto* my_var_a = Expr("my_var"); |
| 899 | auto* my_var_b = Expr("my_var"); |
| 900 | auto* assign = create<ast::AssignmentStatement>(my_var_a, my_var_b); |
dan sinclair | 5e98930 | 2020-09-16 21:20:36 +0000 | [diff] [blame] | 901 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 902 | Func("my_func", ast::VariableList{}, ty.f32(), |
| 903 | ast::StatementList{ |
| 904 | create<ast::VariableDeclStatement>( |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 905 | Var("my_var", ty.pointer<f32>(ast::StorageClass::kFunction), |
| 906 | ast::StorageClass::kNone)), |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 907 | assign, |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 908 | }, |
| 909 | ast::FunctionDecorationList{}); |
dan sinclair | 5e98930 | 2020-09-16 21:20:36 +0000 | [diff] [blame] | 910 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 911 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | 5e98930 | 2020-09-16 21:20:36 +0000 | [diff] [blame] | 912 | |
Ben Clayton | f97b9c9 | 2021-02-16 18:45:45 +0000 | [diff] [blame] | 913 | ASSERT_NE(TypeOf(my_var_a), nullptr); |
| 914 | EXPECT_TRUE(TypeOf(my_var_a)->Is<type::Pointer>()); |
| 915 | EXPECT_TRUE(TypeOf(my_var_a)->As<type::Pointer>()->type()->Is<type::F32>()); |
| 916 | EXPECT_EQ(StmtOf(my_var_a), assign); |
| 917 | ASSERT_NE(TypeOf(my_var_b), nullptr); |
| 918 | EXPECT_TRUE(TypeOf(my_var_b)->Is<type::Pointer>()); |
| 919 | EXPECT_TRUE(TypeOf(my_var_b)->As<type::Pointer>()->type()->Is<type::F32>()); |
| 920 | EXPECT_EQ(StmtOf(my_var_b), assign); |
dan sinclair | 5e98930 | 2020-09-16 21:20:36 +0000 | [diff] [blame] | 921 | } |
| 922 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 923 | TEST_F(TypeDeterminerTest, Expr_Call_Function) { |
Ben Clayton | 42d1e09 | 2021-02-02 14:29:15 +0000 | [diff] [blame] | 924 | Func("my_func", ast::VariableList{}, ty.f32(), ast::StatementList{}, |
| 925 | ast::FunctionDecorationList{}); |
dan sinclair | cab0e73 | 2020-04-07 12:57:27 +0000 | [diff] [blame] | 926 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 927 | auto* call = Call("my_func"); |
| 928 | WrapInFunction(call); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 929 | |
| 930 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 931 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 932 | ASSERT_NE(TypeOf(call), nullptr); |
| 933 | EXPECT_TRUE(TypeOf(call)->Is<type::F32>()); |
dan sinclair | cab0e73 | 2020-04-07 12:57:27 +0000 | [diff] [blame] | 934 | } |
| 935 | |
dan sinclair | ff267ca | 2020-10-14 18:26:31 +0000 | [diff] [blame] | 936 | TEST_F(TypeDeterminerTest, Expr_Identifier_Unknown) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 937 | auto* a = Expr("a"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 938 | WrapInFunction(a); |
| 939 | |
| 940 | EXPECT_FALSE(td()->Determine()); |
dan sinclair | ff267ca | 2020-10-14 18:26:31 +0000 | [diff] [blame] | 941 | } |
| 942 | |
dan sinclair | 13d2a3b | 2020-06-22 20:52:24 +0000 | [diff] [blame] | 943 | TEST_F(TypeDeterminerTest, Function_RegisterInputOutputVariables) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 944 | auto* in_var = Global("in_var", ty.f32(), ast::StorageClass::kInput); |
| 945 | auto* out_var = Global("out_var", ty.f32(), ast::StorageClass::kOutput); |
| 946 | auto* sb_var = Global("sb_var", ty.f32(), ast::StorageClass::kStorage); |
| 947 | auto* wg_var = Global("wg_var", ty.f32(), ast::StorageClass::kWorkgroup); |
| 948 | auto* priv_var = Global("priv_var", ty.f32(), ast::StorageClass::kPrivate); |
dan sinclair | 13d2a3b | 2020-06-22 20:52:24 +0000 | [diff] [blame] | 949 | |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 950 | auto* func = Func( |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 951 | "my_func", ast::VariableList{}, ty.f32(), |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 952 | ast::StatementList{ |
| 953 | create<ast::AssignmentStatement>(Expr("out_var"), Expr("in_var")), |
| 954 | create<ast::AssignmentStatement>(Expr("wg_var"), Expr("wg_var")), |
| 955 | create<ast::AssignmentStatement>(Expr("sb_var"), Expr("sb_var")), |
| 956 | create<ast::AssignmentStatement>(Expr("priv_var"), Expr("priv_var")), |
| 957 | }, |
| 958 | ast::FunctionDecorationList{}); |
dan sinclair | 13d2a3b | 2020-06-22 20:52:24 +0000 | [diff] [blame] | 959 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 960 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | 13d2a3b | 2020-06-22 20:52:24 +0000 | [diff] [blame] | 961 | |
Ben Clayton | 87c78dd | 2021-02-03 16:43:20 +0000 | [diff] [blame] | 962 | auto* func_sem = Sem().Get(func); |
| 963 | ASSERT_NE(func_sem, nullptr); |
| 964 | |
| 965 | const auto& vars = func_sem->ReferencedModuleVariables(); |
Ryan Harrison | 9a452c1 | 2020-06-23 16:38:47 +0000 | [diff] [blame] | 966 | ASSERT_EQ(vars.size(), 5u); |
Ben Clayton | b17aea1 | 2021-02-03 17:51:09 +0000 | [diff] [blame] | 967 | EXPECT_EQ(vars[0]->Declaration(), out_var); |
| 968 | EXPECT_EQ(vars[1]->Declaration(), in_var); |
| 969 | EXPECT_EQ(vars[2]->Declaration(), wg_var); |
| 970 | EXPECT_EQ(vars[3]->Declaration(), sb_var); |
| 971 | EXPECT_EQ(vars[4]->Declaration(), priv_var); |
dan sinclair | 13d2a3b | 2020-06-22 20:52:24 +0000 | [diff] [blame] | 972 | } |
| 973 | |
dan sinclair | de2dd68 | 2020-07-14 20:37:28 +0000 | [diff] [blame] | 974 | TEST_F(TypeDeterminerTest, Function_RegisterInputOutputVariables_SubFunction) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 975 | auto* in_var = Global("in_var", ty.f32(), ast::StorageClass::kInput); |
| 976 | auto* out_var = Global("out_var", ty.f32(), ast::StorageClass::kOutput); |
| 977 | auto* sb_var = Global("sb_var", ty.f32(), ast::StorageClass::kStorage); |
| 978 | auto* wg_var = Global("wg_var", ty.f32(), ast::StorageClass::kWorkgroup); |
| 979 | auto* priv_var = Global("priv_var", ty.f32(), ast::StorageClass::kPrivate); |
dan sinclair | de2dd68 | 2020-07-14 20:37:28 +0000 | [diff] [blame] | 980 | |
Ben Clayton | 42d1e09 | 2021-02-02 14:29:15 +0000 | [diff] [blame] | 981 | Func("my_func", ast::VariableList{}, ty.f32(), |
| 982 | ast::StatementList{ |
| 983 | create<ast::AssignmentStatement>(Expr("out_var"), Expr("in_var")), |
| 984 | create<ast::AssignmentStatement>(Expr("wg_var"), Expr("wg_var")), |
| 985 | create<ast::AssignmentStatement>(Expr("sb_var"), Expr("sb_var")), |
| 986 | create<ast::AssignmentStatement>(Expr("priv_var"), Expr("priv_var")), |
| 987 | }, |
| 988 | ast::FunctionDecorationList{}); |
dan sinclair | de2dd68 | 2020-07-14 20:37:28 +0000 | [diff] [blame] | 989 | |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 990 | auto* func2 = Func( |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 991 | "func", ast::VariableList{}, ty.f32(), |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 992 | ast::StatementList{ |
| 993 | create<ast::AssignmentStatement>(Expr("out_var"), Call("my_func")), |
| 994 | }, |
| 995 | ast::FunctionDecorationList{}); |
dan sinclair | de2dd68 | 2020-07-14 20:37:28 +0000 | [diff] [blame] | 996 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 997 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | de2dd68 | 2020-07-14 20:37:28 +0000 | [diff] [blame] | 998 | |
Ben Clayton | 87c78dd | 2021-02-03 16:43:20 +0000 | [diff] [blame] | 999 | auto* func2_sem = Sem().Get(func2); |
| 1000 | ASSERT_NE(func2_sem, nullptr); |
| 1001 | |
| 1002 | const auto& vars = func2_sem->ReferencedModuleVariables(); |
dan sinclair | de2dd68 | 2020-07-14 20:37:28 +0000 | [diff] [blame] | 1003 | ASSERT_EQ(vars.size(), 5u); |
Ben Clayton | b17aea1 | 2021-02-03 17:51:09 +0000 | [diff] [blame] | 1004 | EXPECT_EQ(vars[0]->Declaration(), out_var); |
| 1005 | EXPECT_EQ(vars[1]->Declaration(), in_var); |
| 1006 | EXPECT_EQ(vars[2]->Declaration(), wg_var); |
| 1007 | EXPECT_EQ(vars[3]->Declaration(), sb_var); |
| 1008 | EXPECT_EQ(vars[4]->Declaration(), priv_var); |
dan sinclair | de2dd68 | 2020-07-14 20:37:28 +0000 | [diff] [blame] | 1009 | } |
| 1010 | |
dan sinclair | 13d2a3b | 2020-06-22 20:52:24 +0000 | [diff] [blame] | 1011 | TEST_F(TypeDeterminerTest, Function_NotRegisterFunctionVariable) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1012 | auto* var = Var("in_var", ty.f32(), ast::StorageClass::kFunction); |
dan sinclair | 13d2a3b | 2020-06-22 20:52:24 +0000 | [diff] [blame] | 1013 | |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 1014 | auto* func = |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 1015 | Func("my_func", ast::VariableList{}, ty.f32(), |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 1016 | ast::StatementList{ |
| 1017 | create<ast::VariableDeclStatement>(var), |
Ben Clayton | 1637cbb | 2021-01-05 15:44:39 +0000 | [diff] [blame] | 1018 | create<ast::AssignmentStatement>(Expr("var"), Expr(1.f)), |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 1019 | }, |
| 1020 | ast::FunctionDecorationList{}); |
dan sinclair | 13d2a3b | 2020-06-22 20:52:24 +0000 | [diff] [blame] | 1021 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1022 | Global("var", ty.f32(), ast::StorageClass::kFunction); |
dan sinclair | ff267ca | 2020-10-14 18:26:31 +0000 | [diff] [blame] | 1023 | |
dan sinclair | ff267ca | 2020-10-14 18:26:31 +0000 | [diff] [blame] | 1024 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | 13d2a3b | 2020-06-22 20:52:24 +0000 | [diff] [blame] | 1025 | |
Ben Clayton | 87c78dd | 2021-02-03 16:43:20 +0000 | [diff] [blame] | 1026 | auto* func_sem = Sem().Get(func); |
| 1027 | ASSERT_NE(func_sem, nullptr); |
| 1028 | |
| 1029 | EXPECT_EQ(func_sem->ReferencedModuleVariables().size(), 0u); |
dan sinclair | 13d2a3b | 2020-06-22 20:52:24 +0000 | [diff] [blame] | 1030 | } |
| 1031 | |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1032 | TEST_F(TypeDeterminerTest, Expr_MemberAccessor_Struct) { |
dan sinclair | 5e5e36e | 2020-12-16 14:41:00 +0000 | [diff] [blame] | 1033 | auto* strct = create<ast::Struct>( |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 1034 | ast::StructMemberList{Member("first_member", ty.i32()), |
| 1035 | Member("second_member", ty.f32())}, |
dan sinclair | 5e5e36e | 2020-12-16 14:41:00 +0000 | [diff] [blame] | 1036 | ast::StructDecorationList{}); |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1037 | |
dan sinclair | b583993 | 2020-12-16 21:38:40 +0000 | [diff] [blame] | 1038 | auto* st = ty.struct_("S", strct); |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1039 | Global("my_struct", st, ast::StorageClass::kNone); |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1040 | |
dan sinclair | 8b40a67 | 2020-12-16 11:49:10 +0000 | [diff] [blame] | 1041 | auto* mem = MemberAccessor("my_struct", "second_member"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1042 | WrapInFunction(mem); |
| 1043 | |
| 1044 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 1045 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1046 | ASSERT_NE(TypeOf(mem), nullptr); |
| 1047 | ASSERT_TRUE(TypeOf(mem)->Is<type::Pointer>()); |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1048 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1049 | auto* ptr = TypeOf(mem)->As<type::Pointer>(); |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1050 | EXPECT_TRUE(ptr->type()->Is<type::F32>()); |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1051 | } |
| 1052 | |
dan sinclair | b445a9b | 2020-04-24 00:40:45 +0000 | [diff] [blame] | 1053 | TEST_F(TypeDeterminerTest, Expr_MemberAccessor_Struct_Alias) { |
dan sinclair | 5e5e36e | 2020-12-16 14:41:00 +0000 | [diff] [blame] | 1054 | auto* strct = create<ast::Struct>( |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 1055 | ast::StructMemberList{Member("first_member", ty.i32()), |
| 1056 | Member("second_member", ty.f32())}, |
dan sinclair | 5e5e36e | 2020-12-16 14:41:00 +0000 | [diff] [blame] | 1057 | ast::StructDecorationList{}); |
dan sinclair | b445a9b | 2020-04-24 00:40:45 +0000 | [diff] [blame] | 1058 | |
dan sinclair | b583993 | 2020-12-16 21:38:40 +0000 | [diff] [blame] | 1059 | auto* st = ty.struct_("alias", strct); |
| 1060 | auto* alias = ty.alias("alias", st); |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1061 | Global("my_struct", alias, ast::StorageClass::kNone); |
dan sinclair | b445a9b | 2020-04-24 00:40:45 +0000 | [diff] [blame] | 1062 | |
dan sinclair | 8b40a67 | 2020-12-16 11:49:10 +0000 | [diff] [blame] | 1063 | auto* mem = MemberAccessor("my_struct", "second_member"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1064 | WrapInFunction(mem); |
| 1065 | |
| 1066 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 1067 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1068 | ASSERT_NE(TypeOf(mem), nullptr); |
| 1069 | ASSERT_TRUE(TypeOf(mem)->Is<type::Pointer>()); |
dan sinclair | b445a9b | 2020-04-24 00:40:45 +0000 | [diff] [blame] | 1070 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1071 | auto* ptr = TypeOf(mem)->As<type::Pointer>(); |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1072 | EXPECT_TRUE(ptr->type()->Is<type::F32>()); |
dan sinclair | b445a9b | 2020-04-24 00:40:45 +0000 | [diff] [blame] | 1073 | } |
| 1074 | |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1075 | TEST_F(TypeDeterminerTest, Expr_MemberAccessor_VectorSwizzle) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1076 | Global("my_vec", ty.vec3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1077 | |
Ben Clayton | 6d612ad | 2021-02-24 14:15:02 +0000 | [diff] [blame] | 1078 | auto* mem = MemberAccessor("my_vec", "xzyw"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1079 | WrapInFunction(mem); |
| 1080 | |
| 1081 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 1082 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1083 | ASSERT_NE(TypeOf(mem), nullptr); |
| 1084 | ASSERT_TRUE(TypeOf(mem)->Is<type::Vector>()); |
| 1085 | EXPECT_TRUE(TypeOf(mem)->As<type::Vector>()->type()->Is<type::F32>()); |
Ben Clayton | 6d612ad | 2021-02-24 14:15:02 +0000 | [diff] [blame] | 1086 | EXPECT_EQ(TypeOf(mem)->As<type::Vector>()->size(), 4u); |
| 1087 | EXPECT_THAT(Sem().Get(mem)->Swizzle(), ElementsAre(0, 2, 1, 3)); |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1088 | } |
| 1089 | |
dan sinclair | aac5865 | 2020-04-21 13:05:34 +0000 | [diff] [blame] | 1090 | TEST_F(TypeDeterminerTest, Expr_MemberAccessor_VectorSwizzle_SingleElement) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1091 | Global("my_vec", ty.vec3<f32>(), ast::StorageClass::kNone); |
dan sinclair | aac5865 | 2020-04-21 13:05:34 +0000 | [diff] [blame] | 1092 | |
Ben Clayton | 6d612ad | 2021-02-24 14:15:02 +0000 | [diff] [blame] | 1093 | auto* mem = MemberAccessor("my_vec", "b"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1094 | WrapInFunction(mem); |
| 1095 | |
| 1096 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 1097 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1098 | ASSERT_NE(TypeOf(mem), nullptr); |
| 1099 | ASSERT_TRUE(TypeOf(mem)->Is<type::Pointer>()); |
dan sinclair | aac5865 | 2020-04-21 13:05:34 +0000 | [diff] [blame] | 1100 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1101 | auto* ptr = TypeOf(mem)->As<type::Pointer>(); |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1102 | ASSERT_TRUE(ptr->type()->Is<type::F32>()); |
Ben Clayton | 6d612ad | 2021-02-24 14:15:02 +0000 | [diff] [blame] | 1103 | EXPECT_THAT(Sem().Get(mem)->Swizzle(), ElementsAre(2)); |
| 1104 | } |
| 1105 | |
| 1106 | TEST_F(TypeDeterminerTest, Expr_MemberAccessor_VectorSwizzle_BadChar) { |
| 1107 | Global("my_vec", ty.vec3<f32>(), ast::StorageClass::kNone); |
| 1108 | |
| 1109 | auto* ident = create<ast::IdentifierExpression>( |
| 1110 | Source{{Source::Location{3, 3}, Source::Location{3, 7}}}, |
| 1111 | Symbols().Register("xyqz")); |
| 1112 | |
| 1113 | auto* mem = MemberAccessor("my_vec", ident); |
| 1114 | WrapInFunction(mem); |
| 1115 | |
| 1116 | EXPECT_FALSE(td()->Determine()); |
| 1117 | EXPECT_EQ(td()->error(), "3:5 error: invalid vector swizzle character"); |
| 1118 | } |
| 1119 | |
Antonio Maiorano | fe2b417 | 2021-03-01 20:01:39 +0000 | [diff] [blame] | 1120 | TEST_F(TypeDeterminerTest, Expr_MemberAccessor_VectorSwizzle_MixedChars) { |
| 1121 | Global("my_vec", ty.vec3<f32>(), ast::StorageClass::kNone); |
| 1122 | |
| 1123 | auto* ident = create<ast::IdentifierExpression>( |
| 1124 | Source{{Source::Location{3, 3}, Source::Location{3, 7}}}, |
| 1125 | Symbols().Register("rgyw")); |
| 1126 | |
| 1127 | auto* mem = MemberAccessor("my_vec", ident); |
| 1128 | WrapInFunction(mem); |
| 1129 | |
| 1130 | EXPECT_FALSE(td()->Determine()); |
| 1131 | EXPECT_EQ( |
| 1132 | td()->error(), |
| 1133 | "3:3 error: invalid mixing of vector swizzle characters rgba with xyzw"); |
| 1134 | } |
| 1135 | |
Ben Clayton | 6d612ad | 2021-02-24 14:15:02 +0000 | [diff] [blame] | 1136 | TEST_F(TypeDeterminerTest, Expr_MemberAccessor_VectorSwizzle_BadLength) { |
| 1137 | Global("my_vec", ty.vec3<f32>(), ast::StorageClass::kNone); |
| 1138 | |
| 1139 | auto* ident = create<ast::IdentifierExpression>( |
| 1140 | Source{{Source::Location{3, 3}, Source::Location{3, 8}}}, |
| 1141 | Symbols().Register("zzzzz")); |
| 1142 | auto* mem = MemberAccessor("my_vec", ident); |
| 1143 | WrapInFunction(mem); |
| 1144 | |
| 1145 | EXPECT_FALSE(td()->Determine()); |
| 1146 | EXPECT_EQ(td()->error(), "3:3 error: invalid vector swizzle size"); |
dan sinclair | aac5865 | 2020-04-21 13:05:34 +0000 | [diff] [blame] | 1147 | } |
| 1148 | |
dan sinclair | 8eddb78 | 2020-04-23 22:26:52 +0000 | [diff] [blame] | 1149 | TEST_F(TypeDeterminerTest, Expr_Accessor_MultiLevel) { |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1150 | // struct b { |
| 1151 | // vec4<f32> foo |
| 1152 | // } |
| 1153 | // struct A { |
| 1154 | // vec3<struct b> mem |
| 1155 | // } |
| 1156 | // var c : A |
| 1157 | // c.mem[0].foo.yx |
| 1158 | // -> vec2<f32> |
| 1159 | // |
| 1160 | // MemberAccessor{ |
| 1161 | // MemberAccessor{ |
| 1162 | // ArrayAccessor{ |
| 1163 | // MemberAccessor{ |
| 1164 | // Identifier{c} |
| 1165 | // Identifier{mem} |
| 1166 | // } |
| 1167 | // ScalarConstructor{0} |
| 1168 | // } |
| 1169 | // Identifier{foo} |
| 1170 | // } |
| 1171 | // Identifier{yx} |
| 1172 | // } |
| 1173 | // |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1174 | |
dan sinclair | 5e5e36e | 2020-12-16 14:41:00 +0000 | [diff] [blame] | 1175 | auto* strctB = |
| 1176 | create<ast::Struct>(ast::StructMemberList{Member("foo", ty.vec4<f32>())}, |
| 1177 | ast::StructDecorationList{}); |
dan sinclair | b583993 | 2020-12-16 21:38:40 +0000 | [diff] [blame] | 1178 | auto* stB = ty.struct_("B", strctB); |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1179 | |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1180 | type::Vector vecB(stB, 3); |
dan sinclair | 5e5e36e | 2020-12-16 14:41:00 +0000 | [diff] [blame] | 1181 | auto* strctA = create<ast::Struct>( |
| 1182 | ast::StructMemberList{Member("mem", &vecB)}, ast::StructDecorationList{}); |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1183 | |
dan sinclair | b583993 | 2020-12-16 21:38:40 +0000 | [diff] [blame] | 1184 | auto* stA = ty.struct_("A", strctA); |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1185 | Global("c", stA, ast::StorageClass::kNone); |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1186 | |
dan sinclair | 8b40a67 | 2020-12-16 11:49:10 +0000 | [diff] [blame] | 1187 | auto* mem = MemberAccessor( |
| 1188 | MemberAccessor(IndexAccessor(MemberAccessor("c", "mem"), 0), "foo"), |
| 1189 | "yx"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1190 | WrapInFunction(mem); |
| 1191 | |
| 1192 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1193 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1194 | ASSERT_NE(TypeOf(mem), nullptr); |
| 1195 | ASSERT_TRUE(TypeOf(mem)->Is<type::Vector>()); |
| 1196 | EXPECT_TRUE(TypeOf(mem)->As<type::Vector>()->type()->Is<type::F32>()); |
| 1197 | EXPECT_EQ(TypeOf(mem)->As<type::Vector>()->size(), 2u); |
dan sinclair | 8ee1d22 | 2020-04-07 16:41:33 +0000 | [diff] [blame] | 1198 | } |
| 1199 | |
Ben Clayton | 7b7d698 | 2021-02-09 17:38:05 +0000 | [diff] [blame] | 1200 | TEST_F(TypeDeterminerTest, Expr_MemberAccessor_InBinaryOp) { |
| 1201 | auto* strct = create<ast::Struct>( |
| 1202 | ast::StructMemberList{Member("first_member", ty.f32()), |
| 1203 | Member("second_member", ty.f32())}, |
| 1204 | ast::StructDecorationList{}); |
| 1205 | |
| 1206 | auto* st = ty.struct_("S", strct); |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1207 | Global("my_struct", st, ast::StorageClass::kNone); |
Ben Clayton | 7b7d698 | 2021-02-09 17:38:05 +0000 | [diff] [blame] | 1208 | |
| 1209 | auto* expr = Add(MemberAccessor("my_struct", "first_member"), |
| 1210 | MemberAccessor("my_struct", "second_member")); |
| 1211 | WrapInFunction(expr); |
| 1212 | |
| 1213 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 1214 | |
| 1215 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1216 | EXPECT_TRUE(TypeOf(expr)->Is<type::F32>()); |
| 1217 | } |
| 1218 | |
dan sinclair | cd077b0 | 2020-04-20 14:19:04 +0000 | [diff] [blame] | 1219 | using Expr_Binary_BitwiseTest = TypeDeterminerTestWithParam<ast::BinaryOp>; |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1220 | TEST_P(Expr_Binary_BitwiseTest, Scalar) { |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1221 | auto op = GetParam(); |
| 1222 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1223 | Global("val", ty.i32(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1224 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1225 | auto* expr = create<ast::BinaryExpression>(op, Expr("val"), Expr("val")); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1226 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1227 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1228 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1229 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1230 | EXPECT_TRUE(TypeOf(expr)->Is<type::I32>()); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1231 | } |
| 1232 | |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1233 | TEST_P(Expr_Binary_BitwiseTest, Vector) { |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1234 | auto op = GetParam(); |
| 1235 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1236 | Global("val", ty.vec3<i32>(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1237 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1238 | auto* expr = create<ast::BinaryExpression>(op, Expr("val"), Expr("val")); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1239 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1240 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1241 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1242 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1243 | ASSERT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
| 1244 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::I32>()); |
| 1245 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 3u); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1246 | } |
| 1247 | INSTANTIATE_TEST_SUITE_P(TypeDeterminerTest, |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1248 | Expr_Binary_BitwiseTest, |
| 1249 | testing::Values(ast::BinaryOp::kAnd, |
| 1250 | ast::BinaryOp::kOr, |
| 1251 | ast::BinaryOp::kXor, |
| 1252 | ast::BinaryOp::kShiftLeft, |
| 1253 | ast::BinaryOp::kShiftRight, |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1254 | ast::BinaryOp::kAdd, |
| 1255 | ast::BinaryOp::kSubtract, |
| 1256 | ast::BinaryOp::kDivide, |
| 1257 | ast::BinaryOp::kModulo)); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1258 | |
dan sinclair | cd077b0 | 2020-04-20 14:19:04 +0000 | [diff] [blame] | 1259 | using Expr_Binary_LogicalTest = TypeDeterminerTestWithParam<ast::BinaryOp>; |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1260 | TEST_P(Expr_Binary_LogicalTest, Scalar) { |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1261 | auto op = GetParam(); |
| 1262 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1263 | Global("val", ty.bool_(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1264 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1265 | auto* expr = create<ast::BinaryExpression>(op, Expr("val"), Expr("val")); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1266 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1267 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1268 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1269 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1270 | EXPECT_TRUE(TypeOf(expr)->Is<type::Bool>()); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1271 | } |
| 1272 | |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1273 | TEST_P(Expr_Binary_LogicalTest, Vector) { |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1274 | auto op = GetParam(); |
| 1275 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1276 | Global("val", ty.vec3<bool>(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1277 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1278 | auto* expr = create<ast::BinaryExpression>(op, Expr("val"), Expr("val")); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1279 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1280 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1281 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1282 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1283 | ASSERT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
| 1284 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::Bool>()); |
| 1285 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 3u); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1286 | } |
| 1287 | INSTANTIATE_TEST_SUITE_P(TypeDeterminerTest, |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1288 | Expr_Binary_LogicalTest, |
| 1289 | testing::Values(ast::BinaryOp::kLogicalAnd, |
| 1290 | ast::BinaryOp::kLogicalOr)); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1291 | |
dan sinclair | cd077b0 | 2020-04-20 14:19:04 +0000 | [diff] [blame] | 1292 | using Expr_Binary_CompareTest = TypeDeterminerTestWithParam<ast::BinaryOp>; |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1293 | TEST_P(Expr_Binary_CompareTest, Scalar) { |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1294 | auto op = GetParam(); |
| 1295 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1296 | Global("val", ty.i32(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1297 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1298 | auto* expr = create<ast::BinaryExpression>(op, Expr("val"), Expr("val")); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1299 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1300 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1301 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1302 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1303 | EXPECT_TRUE(TypeOf(expr)->Is<type::Bool>()); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1304 | } |
| 1305 | |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1306 | TEST_P(Expr_Binary_CompareTest, Vector) { |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1307 | auto op = GetParam(); |
| 1308 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1309 | Global("val", ty.vec3<i32>(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1310 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1311 | auto* expr = create<ast::BinaryExpression>(op, Expr("val"), Expr("val")); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1312 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1313 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1314 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1315 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1316 | ASSERT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
| 1317 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::Bool>()); |
| 1318 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 3u); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1319 | } |
| 1320 | INSTANTIATE_TEST_SUITE_P(TypeDeterminerTest, |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1321 | Expr_Binary_CompareTest, |
| 1322 | testing::Values(ast::BinaryOp::kEqual, |
| 1323 | ast::BinaryOp::kNotEqual, |
| 1324 | ast::BinaryOp::kLessThan, |
| 1325 | ast::BinaryOp::kGreaterThan, |
| 1326 | ast::BinaryOp::kLessThanEqual, |
| 1327 | ast::BinaryOp::kGreaterThanEqual)); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1328 | |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1329 | TEST_F(TypeDeterminerTest, Expr_Binary_Multiply_Scalar_Scalar) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1330 | Global("val", ty.i32(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1331 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1332 | auto* expr = Mul("val", "val"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1333 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1334 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1335 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1336 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1337 | EXPECT_TRUE(TypeOf(expr)->Is<type::I32>()); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1338 | } |
| 1339 | |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1340 | TEST_F(TypeDeterminerTest, Expr_Binary_Multiply_Vector_Scalar) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1341 | Global("scalar", ty.f32(), ast::StorageClass::kNone); |
| 1342 | Global("vector", ty.vec3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1343 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1344 | auto* expr = Mul("vector", "scalar"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1345 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1346 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1347 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1348 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1349 | ASSERT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
| 1350 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::F32>()); |
| 1351 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 3u); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1352 | } |
| 1353 | |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1354 | TEST_F(TypeDeterminerTest, Expr_Binary_Multiply_Scalar_Vector) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1355 | Global("scalar", ty.f32(), ast::StorageClass::kNone); |
| 1356 | Global("vector", ty.vec3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1357 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1358 | auto* expr = Mul("scalar", "vector"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1359 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1360 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1361 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1362 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1363 | ASSERT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
| 1364 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::F32>()); |
| 1365 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 3u); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1366 | } |
| 1367 | |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1368 | TEST_F(TypeDeterminerTest, Expr_Binary_Multiply_Vector_Vector) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1369 | Global("vector", ty.vec3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1370 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1371 | auto* expr = Mul("vector", "vector"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1372 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1373 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1374 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1375 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1376 | ASSERT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
| 1377 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::F32>()); |
| 1378 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 3u); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1379 | } |
| 1380 | |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1381 | TEST_F(TypeDeterminerTest, Expr_Binary_Multiply_Matrix_Scalar) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1382 | Global("scalar", ty.f32(), ast::StorageClass::kNone); |
| 1383 | Global("matrix", ty.mat2x3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1384 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1385 | auto* expr = Mul("matrix", "scalar"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1386 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1387 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1388 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1389 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1390 | ASSERT_TRUE(TypeOf(expr)->Is<type::Matrix>()); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1391 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1392 | auto* mat = TypeOf(expr)->As<type::Matrix>(); |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1393 | EXPECT_TRUE(mat->type()->Is<type::F32>()); |
Ryan Harrison | 0a196c1 | 2020-04-17 13:18:20 +0000 | [diff] [blame] | 1394 | EXPECT_EQ(mat->rows(), 3u); |
| 1395 | EXPECT_EQ(mat->columns(), 2u); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1396 | } |
| 1397 | |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1398 | TEST_F(TypeDeterminerTest, Expr_Binary_Multiply_Scalar_Matrix) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1399 | Global("scalar", ty.f32(), ast::StorageClass::kNone); |
| 1400 | Global("matrix", ty.mat2x3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1401 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1402 | auto* expr = Mul("scalar", "matrix"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1403 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1404 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1405 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1406 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1407 | ASSERT_TRUE(TypeOf(expr)->Is<type::Matrix>()); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1408 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1409 | auto* mat = TypeOf(expr)->As<type::Matrix>(); |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1410 | EXPECT_TRUE(mat->type()->Is<type::F32>()); |
Ryan Harrison | 0a196c1 | 2020-04-17 13:18:20 +0000 | [diff] [blame] | 1411 | EXPECT_EQ(mat->rows(), 3u); |
| 1412 | EXPECT_EQ(mat->columns(), 2u); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1413 | } |
| 1414 | |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1415 | TEST_F(TypeDeterminerTest, Expr_Binary_Multiply_Matrix_Vector) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1416 | Global("vector", ty.vec3<f32>(), ast::StorageClass::kNone); |
| 1417 | Global("matrix", ty.mat2x3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1418 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1419 | auto* expr = Mul("matrix", "vector"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1420 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1421 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1422 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1423 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1424 | ASSERT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
| 1425 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::F32>()); |
| 1426 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 3u); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1427 | } |
| 1428 | |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1429 | TEST_F(TypeDeterminerTest, Expr_Binary_Multiply_Vector_Matrix) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1430 | Global("vector", ty.vec3<f32>(), ast::StorageClass::kNone); |
| 1431 | Global("matrix", ty.mat2x3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1432 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1433 | auto* expr = Mul("vector", "matrix"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1434 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1435 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1436 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1437 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1438 | ASSERT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
| 1439 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::F32>()); |
| 1440 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 2u); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1441 | } |
| 1442 | |
dan sinclair | 1c9b486 | 2020-04-07 19:27:41 +0000 | [diff] [blame] | 1443 | TEST_F(TypeDeterminerTest, Expr_Binary_Multiply_Matrix_Matrix) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1444 | Global("mat3x4", ty.mat3x4<f32>(), ast::StorageClass::kNone); |
| 1445 | Global("mat4x3", ty.mat4x3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1446 | |
Ben Clayton | 1637cbb | 2021-01-05 15:44:39 +0000 | [diff] [blame] | 1447 | auto* expr = Mul("mat3x4", "mat4x3"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1448 | WrapInFunction(expr); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1449 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1450 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1451 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1452 | ASSERT_TRUE(TypeOf(expr)->Is<type::Matrix>()); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1453 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1454 | auto* mat = TypeOf(expr)->As<type::Matrix>(); |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1455 | EXPECT_TRUE(mat->type()->Is<type::F32>()); |
Ryan Harrison | 0a196c1 | 2020-04-17 13:18:20 +0000 | [diff] [blame] | 1456 | EXPECT_EQ(mat->rows(), 4u); |
| 1457 | EXPECT_EQ(mat->columns(), 4u); |
dan sinclair | 9b97802 | 2020-04-07 19:26:39 +0000 | [diff] [blame] | 1458 | } |
| 1459 | |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1460 | using IntrinsicDerivativeTest = TypeDeterminerTestWithParam<std::string>; |
| 1461 | TEST_P(IntrinsicDerivativeTest, Scalar) { |
| 1462 | auto name = GetParam(); |
dan sinclair | b173056 | 2020-04-07 19:26:49 +0000 | [diff] [blame] | 1463 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1464 | Global("ident", ty.f32(), ast::StorageClass::kNone); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1465 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1466 | auto* expr = Call(name, "ident"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1467 | WrapInFunction(expr); |
| 1468 | |
| 1469 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1470 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1471 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1472 | ASSERT_TRUE(TypeOf(expr)->Is<type::F32>()); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1473 | } |
| 1474 | |
| 1475 | TEST_P(IntrinsicDerivativeTest, Vector) { |
| 1476 | auto name = GetParam(); |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1477 | Global("ident", ty.vec4<f32>(), ast::StorageClass::kNone); |
dan sinclair | b173056 | 2020-04-07 19:26:49 +0000 | [diff] [blame] | 1478 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1479 | auto* expr = Call(name, "ident"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1480 | WrapInFunction(expr); |
| 1481 | |
| 1482 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1483 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1484 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1485 | ASSERT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
| 1486 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::F32>()); |
| 1487 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 4u); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1488 | } |
| 1489 | |
| 1490 | TEST_P(IntrinsicDerivativeTest, MissingParam) { |
| 1491 | auto name = GetParam(); |
| 1492 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1493 | auto* expr = Call(name); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1494 | WrapInFunction(expr); |
| 1495 | |
| 1496 | EXPECT_FALSE(td()->Determine()); |
| 1497 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 1498 | EXPECT_EQ(td()->error(), "error: no matching call to " + name + |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 1499 | "()\n\n" |
| 1500 | "2 candidate functions:\n " + |
| 1501 | name + "(f32) -> f32\n " + name + |
| 1502 | "(vecN<f32>) -> vecN<f32>\n"); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1503 | } |
| 1504 | |
dan sinclair | b173056 | 2020-04-07 19:26:49 +0000 | [diff] [blame] | 1505 | INSTANTIATE_TEST_SUITE_P(TypeDeterminerTest, |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1506 | IntrinsicDerivativeTest, |
| 1507 | testing::Values("dpdx", |
Tomek Ponitka | 4e2d248 | 2020-09-08 16:30:29 +0000 | [diff] [blame] | 1508 | "dpdxCoarse", |
| 1509 | "dpdxFine", |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1510 | "dpdy", |
Tomek Ponitka | 4e2d248 | 2020-09-08 16:30:29 +0000 | [diff] [blame] | 1511 | "dpdyCoarse", |
| 1512 | "dpdyFine", |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1513 | "fwidth", |
Tomek Ponitka | 4e2d248 | 2020-09-08 16:30:29 +0000 | [diff] [blame] | 1514 | "fwidthCoarse", |
| 1515 | "fwidthFine")); |
dan sinclair | b173056 | 2020-04-07 19:26:49 +0000 | [diff] [blame] | 1516 | |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1517 | using Intrinsic = TypeDeterminerTestWithParam<std::string>; |
| 1518 | TEST_P(Intrinsic, Test) { |
| 1519 | auto name = GetParam(); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1520 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1521 | Global("my_var", ty.vec3<bool>(), ast::StorageClass::kNone); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1522 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1523 | auto* expr = Call(name, "my_var"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1524 | WrapInFunction(expr); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1525 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1526 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1527 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1528 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1529 | EXPECT_TRUE(TypeOf(expr)->Is<type::Bool>()); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1530 | } |
| 1531 | INSTANTIATE_TEST_SUITE_P(TypeDeterminerTest, |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1532 | Intrinsic, |
| 1533 | testing::Values("any", "all")); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1534 | |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1535 | using Intrinsic_FloatMethod = TypeDeterminerTestWithParam<std::string>; |
| 1536 | TEST_P(Intrinsic_FloatMethod, Vector) { |
| 1537 | auto name = GetParam(); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1538 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1539 | Global("my_var", ty.vec3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1540 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1541 | auto* expr = Call(name, "my_var"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1542 | WrapInFunction(expr); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1543 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1544 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1545 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1546 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1547 | ASSERT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
| 1548 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::Bool>()); |
| 1549 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 3u); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1550 | } |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1551 | |
| 1552 | TEST_P(Intrinsic_FloatMethod, Scalar) { |
| 1553 | auto name = GetParam(); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1554 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1555 | Global("my_var", ty.f32(), ast::StorageClass::kNone); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1556 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1557 | auto* expr = Call(name, "my_var"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1558 | WrapInFunction(expr); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1559 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1560 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 1561 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1562 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1563 | EXPECT_TRUE(TypeOf(expr)->Is<type::Bool>()); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1564 | } |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1565 | |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1566 | TEST_P(Intrinsic_FloatMethod, MissingParam) { |
| 1567 | auto name = GetParam(); |
| 1568 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1569 | Global("my_var", ty.f32(), ast::StorageClass::kNone); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1570 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1571 | auto* expr = Call(name); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1572 | WrapInFunction(expr); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1573 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1574 | EXPECT_FALSE(td()->Determine()); |
| 1575 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 1576 | EXPECT_EQ(td()->error(), "error: no matching call to " + name + |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 1577 | "()\n\n" |
| 1578 | "2 candidate functions:\n " + |
| 1579 | name + "(f32) -> bool\n " + name + |
| 1580 | "(vecN<f32>) -> vecN<bool>\n"); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1581 | } |
| 1582 | |
| 1583 | TEST_P(Intrinsic_FloatMethod, TooManyParams) { |
| 1584 | auto name = GetParam(); |
| 1585 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1586 | Global("my_var", ty.f32(), ast::StorageClass::kNone); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1587 | |
Ben Clayton | faca02d | 2021-02-10 21:34:25 +0000 | [diff] [blame] | 1588 | auto* expr = Call(name, "my_var", 1.23f); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1589 | WrapInFunction(expr); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1590 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1591 | EXPECT_FALSE(td()->Determine()); |
| 1592 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 1593 | EXPECT_EQ(td()->error(), "error: no matching call to " + name + |
Ben Clayton | faca02d | 2021-02-10 21:34:25 +0000 | [diff] [blame] | 1594 | "(ptr<f32>, f32)\n\n" |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 1595 | "2 candidate functions:\n " + |
| 1596 | name + "(f32) -> bool\n " + name + |
| 1597 | "(vecN<f32>) -> vecN<bool>\n"); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1598 | } |
| 1599 | INSTANTIATE_TEST_SUITE_P( |
| 1600 | TypeDeterminerTest, |
| 1601 | Intrinsic_FloatMethod, |
Tomek Ponitka | 4e2d248 | 2020-09-08 16:30:29 +0000 | [diff] [blame] | 1602 | testing::Values("isInf", "isNan", "isFinite", "isNormal")); |
dan sinclair | 46e959d | 2020-06-01 13:43:22 +0000 | [diff] [blame] | 1603 | |
Ben Clayton | f1b0e1e | 2020-11-30 23:30:58 +0000 | [diff] [blame] | 1604 | enum class Texture { kF32, kI32, kU32 }; |
| 1605 | inline std::ostream& operator<<(std::ostream& out, Texture data) { |
| 1606 | if (data == Texture::kF32) { |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1607 | out << "f32"; |
Ben Clayton | f1b0e1e | 2020-11-30 23:30:58 +0000 | [diff] [blame] | 1608 | } else if (data == Texture::kI32) { |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1609 | out << "i32"; |
| 1610 | } else { |
| 1611 | out << "u32"; |
| 1612 | } |
| 1613 | return out; |
| 1614 | } |
| 1615 | |
| 1616 | struct TextureTestParams { |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1617 | type::TextureDimension dim; |
Ben Clayton | f1b0e1e | 2020-11-30 23:30:58 +0000 | [diff] [blame] | 1618 | Texture type = Texture::kF32; |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1619 | type::ImageFormat format = type::ImageFormat::kR16Float; |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1620 | }; |
| 1621 | inline std::ostream& operator<<(std::ostream& out, TextureTestParams data) { |
| 1622 | out << data.dim << "_" << data.type; |
| 1623 | return out; |
| 1624 | } |
| 1625 | |
| 1626 | class Intrinsic_TextureOperation |
| 1627 | : public TypeDeterminerTestWithParam<TextureTestParams> { |
| 1628 | public: |
Ben Clayton | 3b4f3d2 | 2021-02-08 20:24:52 +0000 | [diff] [blame] | 1629 | /// Gets an appropriate type for the coords parameter depending the the |
| 1630 | /// dimensionality of the texture being sampled. |
| 1631 | /// @param dim dimensionality of the texture being sampled |
| 1632 | /// @param scalar the scalar type |
| 1633 | /// @returns a pointer to a type appropriate for the coord param |
| 1634 | type::Type* GetCoordsType(type::TextureDimension dim, type::Type* scalar) { |
| 1635 | switch (dim) { |
| 1636 | case type::TextureDimension::k1d: |
Ben Clayton | 3b4f3d2 | 2021-02-08 20:24:52 +0000 | [diff] [blame] | 1637 | return scalar; |
| 1638 | case type::TextureDimension::k2d: |
| 1639 | case type::TextureDimension::k2dArray: |
| 1640 | return create<type::Vector>(scalar, 2); |
| 1641 | case type::TextureDimension::k3d: |
| 1642 | case type::TextureDimension::kCube: |
| 1643 | case type::TextureDimension::kCubeArray: |
| 1644 | return create<type::Vector>(scalar, 3); |
| 1645 | default: |
| 1646 | [=]() { FAIL() << "Unsupported texture dimension: " << dim; }(); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1647 | } |
Ben Clayton | 3b4f3d2 | 2021-02-08 20:24:52 +0000 | [diff] [blame] | 1648 | return nullptr; |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1649 | } |
| 1650 | |
| 1651 | void add_call_param(std::string name, |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1652 | type::Type* type, |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1653 | ast::ExpressionList* call_params) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1654 | Global(name, type, ast::StorageClass::kNone); |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1655 | call_params->push_back(Expr(name)); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1656 | } |
Ben Clayton | 281b602 | 2021-01-21 16:35:10 +0000 | [diff] [blame] | 1657 | type::Type* subtype(Texture type) { |
Ben Clayton | f1b0e1e | 2020-11-30 23:30:58 +0000 | [diff] [blame] | 1658 | if (type == Texture::kF32) { |
Ben Clayton | 281b602 | 2021-01-21 16:35:10 +0000 | [diff] [blame] | 1659 | return create<type::F32>(); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1660 | } |
Ben Clayton | f1b0e1e | 2020-11-30 23:30:58 +0000 | [diff] [blame] | 1661 | if (type == Texture::kI32) { |
Ben Clayton | 281b602 | 2021-01-21 16:35:10 +0000 | [diff] [blame] | 1662 | return create<type::I32>(); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1663 | } |
Ben Clayton | 281b602 | 2021-01-21 16:35:10 +0000 | [diff] [blame] | 1664 | return create<type::U32>(); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1665 | } |
| 1666 | }; |
| 1667 | |
| 1668 | using Intrinsic_StorageTextureOperation = Intrinsic_TextureOperation; |
| 1669 | TEST_P(Intrinsic_StorageTextureOperation, TextureLoadRo) { |
| 1670 | auto dim = GetParam().dim; |
| 1671 | auto type = GetParam().type; |
| 1672 | auto format = GetParam().format; |
| 1673 | |
Ben Clayton | 3b4f3d2 | 2021-02-08 20:24:52 +0000 | [diff] [blame] | 1674 | auto* coords_type = GetCoordsType(dim, ty.i32()); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1675 | |
Ben Clayton | b57c19d | 2021-02-08 20:35:01 +0000 | [diff] [blame] | 1676 | auto* subtype = type::StorageTexture::SubtypeFor(format, Types()); |
Ben Clayton | faca02d | 2021-02-10 21:34:25 +0000 | [diff] [blame] | 1677 | auto* texture_type = create<type::StorageTexture>(dim, format, subtype); |
| 1678 | auto* ro_texture_type = |
| 1679 | create<type::AccessControl>(ast::AccessControl::kReadOnly, texture_type); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1680 | |
| 1681 | ast::ExpressionList call_params; |
| 1682 | |
Ben Clayton | faca02d | 2021-02-10 21:34:25 +0000 | [diff] [blame] | 1683 | add_call_param("texture", ro_texture_type, &call_params); |
Ben Clayton | 281b602 | 2021-01-21 16:35:10 +0000 | [diff] [blame] | 1684 | add_call_param("coords", coords_type, &call_params); |
Ben Clayton | 3b4f3d2 | 2021-02-08 20:24:52 +0000 | [diff] [blame] | 1685 | |
| 1686 | if (type::IsTextureArray(dim)) { |
| 1687 | add_call_param("array_index", ty.i32(), &call_params); |
| 1688 | } |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1689 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1690 | auto* expr = Call("textureLoad", call_params); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1691 | WrapInFunction(expr); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1692 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1693 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1694 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1695 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1696 | ASSERT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
Ben Clayton | f1b0e1e | 2020-11-30 23:30:58 +0000 | [diff] [blame] | 1697 | if (type == Texture::kF32) { |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1698 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::F32>()); |
Ben Clayton | f1b0e1e | 2020-11-30 23:30:58 +0000 | [diff] [blame] | 1699 | } else if (type == Texture::kI32) { |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1700 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::I32>()); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1701 | } else { |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1702 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::U32>()); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1703 | } |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1704 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 4u); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1705 | } |
| 1706 | |
| 1707 | INSTANTIATE_TEST_SUITE_P( |
| 1708 | TypeDeterminerTest, |
| 1709 | Intrinsic_StorageTextureOperation, |
| 1710 | testing::Values( |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1711 | TextureTestParams{type::TextureDimension::k1d, Texture::kF32, |
| 1712 | type::ImageFormat::kR16Float}, |
| 1713 | TextureTestParams{type::TextureDimension::k1d, Texture::kI32, |
| 1714 | type::ImageFormat::kR16Sint}, |
| 1715 | TextureTestParams{type::TextureDimension::k1d, Texture::kF32, |
| 1716 | type::ImageFormat::kR8Unorm}, |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1717 | TextureTestParams{type::TextureDimension::k2d, Texture::kF32, |
| 1718 | type::ImageFormat::kR16Float}, |
| 1719 | TextureTestParams{type::TextureDimension::k2d, Texture::kI32, |
| 1720 | type::ImageFormat::kR16Sint}, |
| 1721 | TextureTestParams{type::TextureDimension::k2d, Texture::kF32, |
| 1722 | type::ImageFormat::kR8Unorm}, |
| 1723 | TextureTestParams{type::TextureDimension::k2dArray, Texture::kF32, |
| 1724 | type::ImageFormat::kR16Float}, |
| 1725 | TextureTestParams{type::TextureDimension::k2dArray, Texture::kI32, |
| 1726 | type::ImageFormat::kR16Sint}, |
| 1727 | TextureTestParams{type::TextureDimension::k2dArray, Texture::kF32, |
| 1728 | type::ImageFormat::kR8Unorm}, |
| 1729 | TextureTestParams{type::TextureDimension::k3d, Texture::kF32, |
| 1730 | type::ImageFormat::kR16Float}, |
| 1731 | TextureTestParams{type::TextureDimension::k3d, Texture::kI32, |
| 1732 | type::ImageFormat::kR16Sint}, |
| 1733 | TextureTestParams{type::TextureDimension::k3d, Texture::kF32, |
| 1734 | type::ImageFormat::kR8Unorm})); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1735 | |
| 1736 | using Intrinsic_SampledTextureOperation = Intrinsic_TextureOperation; |
| 1737 | TEST_P(Intrinsic_SampledTextureOperation, TextureLoadSampled) { |
| 1738 | auto dim = GetParam().dim; |
| 1739 | auto type = GetParam().type; |
| 1740 | |
Ben Clayton | 281b602 | 2021-01-21 16:35:10 +0000 | [diff] [blame] | 1741 | type::Type* s = subtype(type); |
Ben Clayton | 3b4f3d2 | 2021-02-08 20:24:52 +0000 | [diff] [blame] | 1742 | auto* coords_type = GetCoordsType(dim, ty.i32()); |
Ben Clayton | 281b602 | 2021-01-21 16:35:10 +0000 | [diff] [blame] | 1743 | auto* texture_type = create<type::SampledTexture>(dim, s); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1744 | |
| 1745 | ast::ExpressionList call_params; |
| 1746 | |
Ben Clayton | 281b602 | 2021-01-21 16:35:10 +0000 | [diff] [blame] | 1747 | add_call_param("texture", texture_type, &call_params); |
| 1748 | add_call_param("coords", coords_type, &call_params); |
Ryan Harrison | 2f8df72 | 2021-03-02 20:31:58 +0000 | [diff] [blame^] | 1749 | if (dim == type::TextureDimension::k2dArray) { |
| 1750 | add_call_param("array_index", ty.i32(), &call_params); |
| 1751 | } |
| 1752 | add_call_param("level", ty.i32(), &call_params); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1753 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1754 | auto* expr = Call("textureLoad", call_params); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1755 | WrapInFunction(expr); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1756 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1757 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1758 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1759 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1760 | ASSERT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
Ben Clayton | f1b0e1e | 2020-11-30 23:30:58 +0000 | [diff] [blame] | 1761 | if (type == Texture::kF32) { |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1762 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::F32>()); |
Ben Clayton | f1b0e1e | 2020-11-30 23:30:58 +0000 | [diff] [blame] | 1763 | } else if (type == Texture::kI32) { |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1764 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::I32>()); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1765 | } else { |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1766 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::U32>()); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1767 | } |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1768 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 4u); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1769 | } |
| 1770 | |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1771 | INSTANTIATE_TEST_SUITE_P( |
| 1772 | TypeDeterminerTest, |
| 1773 | Intrinsic_SampledTextureOperation, |
Ben Clayton | 3b4f3d2 | 2021-02-08 20:24:52 +0000 | [diff] [blame] | 1774 | testing::Values(TextureTestParams{type::TextureDimension::k1d}, |
Ben Clayton | 3b4f3d2 | 2021-02-08 20:24:52 +0000 | [diff] [blame] | 1775 | TextureTestParams{type::TextureDimension::k2d}, |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 1776 | TextureTestParams{type::TextureDimension::k2dArray}, |
Ben Clayton | 3b4f3d2 | 2021-02-08 20:24:52 +0000 | [diff] [blame] | 1777 | TextureTestParams{type::TextureDimension::k3d})); |
Tomek Ponitka | 1a61fc4 | 2020-08-31 15:24:54 +0000 | [diff] [blame] | 1778 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1779 | TEST_F(TypeDeterminerTest, Intrinsic_Dot_Vec2) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1780 | Global("my_var", ty.vec2<f32>(), ast::StorageClass::kNone); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1781 | |
| 1782 | auto* expr = Call("dot", "my_var", "my_var"); |
| 1783 | WrapInFunction(expr); |
| 1784 | |
| 1785 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 1786 | |
| 1787 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1788 | EXPECT_TRUE(TypeOf(expr)->Is<type::F32>()); |
| 1789 | } |
| 1790 | |
| 1791 | TEST_F(TypeDeterminerTest, Intrinsic_Dot_Vec3) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1792 | Global("my_var", ty.vec3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1793 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1794 | auto* expr = Call("dot", "my_var", "my_var"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1795 | WrapInFunction(expr); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1796 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1797 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 1798 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1799 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1800 | EXPECT_TRUE(TypeOf(expr)->Is<type::F32>()); |
dan sinclair | 8dcfd10 | 2020-04-07 19:27:00 +0000 | [diff] [blame] | 1801 | } |
| 1802 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1803 | TEST_F(TypeDeterminerTest, Intrinsic_Dot_Vec4) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1804 | Global("my_var", ty.vec4<f32>(), ast::StorageClass::kNone); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1805 | |
| 1806 | auto* expr = Call("dot", "my_var", "my_var"); |
| 1807 | WrapInFunction(expr); |
| 1808 | |
| 1809 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 1810 | |
| 1811 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1812 | EXPECT_TRUE(TypeOf(expr)->Is<type::F32>()); |
| 1813 | } |
| 1814 | |
| 1815 | TEST_F(TypeDeterminerTest, Intrinsic_Dot_Error_Scalar) { |
| 1816 | auto* expr = Call("dot", 1.0f, 1.0f); |
| 1817 | WrapInFunction(expr); |
| 1818 | |
| 1819 | EXPECT_FALSE(td()->Determine()); |
| 1820 | |
| 1821 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 1822 | R"(error: no matching call to dot(f32, f32) |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1823 | |
| 1824 | 1 candidate function: |
| 1825 | dot(vecN<f32>, vecN<f32>) -> f32 |
| 1826 | )"); |
| 1827 | } |
| 1828 | |
| 1829 | TEST_F(TypeDeterminerTest, Intrinsic_Dot_Error_VectorInt) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1830 | Global("my_var", ty.vec4<i32>(), ast::StorageClass::kNone); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1831 | |
| 1832 | auto* expr = Call("dot", "my_var", "my_var"); |
| 1833 | WrapInFunction(expr); |
| 1834 | |
| 1835 | EXPECT_FALSE(td()->Determine()); |
| 1836 | |
| 1837 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 1838 | R"(error: no matching call to dot(ptr<vec4<i32>>, ptr<vec4<i32>>) |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1839 | |
| 1840 | 1 candidate function: |
| 1841 | dot(vecN<f32>, vecN<f32>) -> f32 |
| 1842 | )"); |
| 1843 | } |
| 1844 | |
dan sinclair | 16a2ea1 | 2020-07-21 17:44:44 +0000 | [diff] [blame] | 1845 | TEST_F(TypeDeterminerTest, Intrinsic_Select) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1846 | Global("my_var", ty.vec3<f32>(), ast::StorageClass::kNone); |
dan sinclair | 16a2ea1 | 2020-07-21 17:44:44 +0000 | [diff] [blame] | 1847 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1848 | Global("bool_var", ty.vec3<bool>(), ast::StorageClass::kNone); |
dan sinclair | 16a2ea1 | 2020-07-21 17:44:44 +0000 | [diff] [blame] | 1849 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1850 | auto* expr = Call("select", "my_var", "my_var", "bool_var"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1851 | WrapInFunction(expr); |
dan sinclair | 16a2ea1 | 2020-07-21 17:44:44 +0000 | [diff] [blame] | 1852 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1853 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 1854 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1855 | ASSERT_NE(TypeOf(expr), nullptr); |
| 1856 | EXPECT_TRUE(TypeOf(expr)->Is<type::Vector>()); |
| 1857 | EXPECT_EQ(TypeOf(expr)->As<type::Vector>()->size(), 3u); |
| 1858 | EXPECT_TRUE(TypeOf(expr)->As<type::Vector>()->type()->Is<type::F32>()); |
dan sinclair | 16a2ea1 | 2020-07-21 17:44:44 +0000 | [diff] [blame] | 1859 | } |
| 1860 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1861 | TEST_F(TypeDeterminerTest, Intrinsic_Select_Error_NoParams) { |
Alan Baker | 2e6a1bb | 2021-02-01 16:03:03 +0000 | [diff] [blame] | 1862 | auto* expr = Call("select"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1863 | WrapInFunction(expr); |
dan sinclair | 16a2ea1 | 2020-07-21 17:44:44 +0000 | [diff] [blame] | 1864 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1865 | EXPECT_FALSE(td()->Determine()); |
| 1866 | |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 1867 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 1868 | R"(error: no matching call to select() |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 1869 | |
| 1870 | 2 candidate functions: |
| 1871 | select(T, T, bool) -> T where: T is scalar |
| 1872 | select(vecN<T>, vecN<T>, vecN<bool>) -> vecN<T> where: T is scalar |
| 1873 | )"); |
dan sinclair | 16a2ea1 | 2020-07-21 17:44:44 +0000 | [diff] [blame] | 1874 | } |
| 1875 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1876 | TEST_F(TypeDeterminerTest, Intrinsic_Select_Error_SelectorInt) { |
| 1877 | auto* expr = Call("select", Expr(1), Expr(1), Expr(1)); |
| 1878 | WrapInFunction(expr); |
| 1879 | |
| 1880 | EXPECT_FALSE(td()->Determine()); |
| 1881 | |
| 1882 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 1883 | R"(error: no matching call to select(i32, i32, i32) |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1884 | |
| 1885 | 2 candidate functions: |
| 1886 | select(T, T, bool) -> T where: T is scalar |
| 1887 | select(vecN<T>, vecN<T>, vecN<bool>) -> vecN<T> where: T is scalar |
| 1888 | )"); |
| 1889 | } |
| 1890 | |
| 1891 | TEST_F(TypeDeterminerTest, Intrinsic_Select_Error_Matrix) { |
| 1892 | auto* mat = mat2x2<float>(vec2<float>(1.0f, 1.0f), vec2<float>(1.0f, 1.0f)); |
| 1893 | auto* expr = Call("select", mat, mat, Expr(true)); |
| 1894 | WrapInFunction(expr); |
| 1895 | |
| 1896 | EXPECT_FALSE(td()->Determine()); |
| 1897 | |
| 1898 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 1899 | R"(error: no matching call to select(mat2x2<f32>, mat2x2<f32>, bool) |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1900 | |
| 1901 | 2 candidate functions: |
| 1902 | select(T, T, bool) -> T where: T is scalar |
| 1903 | select(vecN<T>, vecN<T>, vecN<bool>) -> vecN<T> where: T is scalar |
| 1904 | )"); |
| 1905 | } |
| 1906 | |
| 1907 | TEST_F(TypeDeterminerTest, Intrinsic_Select_Error_MismatchTypes) { |
| 1908 | auto* expr = Call("select", 1.0f, vec2<float>(2.0f, 3.0f), Expr(true)); |
| 1909 | WrapInFunction(expr); |
| 1910 | |
| 1911 | EXPECT_FALSE(td()->Determine()); |
| 1912 | |
| 1913 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 1914 | R"(error: no matching call to select(f32, vec2<f32>, bool) |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1915 | |
| 1916 | 2 candidate functions: |
| 1917 | select(T, T, bool) -> T where: T is scalar |
| 1918 | select(vecN<T>, vecN<T>, vecN<bool>) -> vecN<T> where: T is scalar |
| 1919 | )"); |
| 1920 | } |
| 1921 | |
| 1922 | TEST_F(TypeDeterminerTest, Intrinsic_Select_Error_MismatchVectorSize) { |
| 1923 | auto* expr = Call("select", vec2<float>(1.0f, 2.0f), |
| 1924 | vec3<float>(3.0f, 4.0f, 5.0f), Expr(true)); |
| 1925 | WrapInFunction(expr); |
| 1926 | |
| 1927 | EXPECT_FALSE(td()->Determine()); |
| 1928 | |
| 1929 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 1930 | R"(error: no matching call to select(vec2<f32>, vec3<f32>, bool) |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 1931 | |
| 1932 | 2 candidate functions: |
| 1933 | select(T, T, bool) -> T where: T is scalar |
| 1934 | select(vecN<T>, vecN<T>, vecN<bool>) -> vecN<T> where: T is scalar |
| 1935 | )"); |
| 1936 | } |
| 1937 | |
dan sinclair | cd077b0 | 2020-04-20 14:19:04 +0000 | [diff] [blame] | 1938 | using UnaryOpExpressionTest = TypeDeterminerTestWithParam<ast::UnaryOp>; |
dan sinclair | 0e25762 | 2020-04-07 19:27:11 +0000 | [diff] [blame] | 1939 | TEST_P(UnaryOpExpressionTest, Expr_UnaryOp) { |
| 1940 | auto op = GetParam(); |
| 1941 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1942 | Global("ident", ty.vec4<f32>(), ast::StorageClass::kNone); |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1943 | auto* der = create<ast::UnaryOpExpression>(op, Expr("ident")); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1944 | WrapInFunction(der); |
| 1945 | |
| 1946 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 1947 | |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 1948 | ASSERT_NE(TypeOf(der), nullptr); |
| 1949 | ASSERT_TRUE(TypeOf(der)->Is<type::Vector>()); |
| 1950 | EXPECT_TRUE(TypeOf(der)->As<type::Vector>()->type()->Is<type::F32>()); |
| 1951 | EXPECT_EQ(TypeOf(der)->As<type::Vector>()->size(), 4u); |
dan sinclair | 0e25762 | 2020-04-07 19:27:11 +0000 | [diff] [blame] | 1952 | } |
| 1953 | INSTANTIATE_TEST_SUITE_P(TypeDeterminerTest, |
| 1954 | UnaryOpExpressionTest, |
| 1955 | testing::Values(ast::UnaryOp::kNegation, |
| 1956 | ast::UnaryOp::kNot)); |
| 1957 | |
dan sinclair | ee8ae04 | 2020-04-08 19:58:20 +0000 | [diff] [blame] | 1958 | TEST_F(TypeDeterminerTest, StorageClass_SetsIfMissing) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1959 | auto* var = Var("var", ty.i32(), ast::StorageClass::kNone); |
dan sinclair | ee8ae04 | 2020-04-08 19:58:20 +0000 | [diff] [blame] | 1960 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1961 | auto* stmt = create<ast::VariableDeclStatement>(var); |
Ben Clayton | 42d1e09 | 2021-02-02 14:29:15 +0000 | [diff] [blame] | 1962 | Func("func", ast::VariableList{}, ty.i32(), ast::StatementList{stmt}, |
| 1963 | ast::FunctionDecorationList{}); |
dan sinclair | ee8ae04 | 2020-04-08 19:58:20 +0000 | [diff] [blame] | 1964 | |
dan sinclair | b950e80 | 2020-04-20 14:20:01 +0000 | [diff] [blame] | 1965 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1966 | |
Ben Clayton | b17aea1 | 2021-02-03 17:51:09 +0000 | [diff] [blame] | 1967 | EXPECT_EQ(Sem().Get(var)->StorageClass(), ast::StorageClass::kFunction); |
dan sinclair | ee8ae04 | 2020-04-08 19:58:20 +0000 | [diff] [blame] | 1968 | } |
| 1969 | |
| 1970 | TEST_F(TypeDeterminerTest, StorageClass_DoesNotSetOnConst) { |
Ben Clayton | 81a29fe | 2021-02-17 00:26:52 +0000 | [diff] [blame] | 1971 | auto* var = Const("var", ty.i32()); |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1972 | auto* stmt = create<ast::VariableDeclStatement>(var); |
Ben Clayton | 42d1e09 | 2021-02-02 14:29:15 +0000 | [diff] [blame] | 1973 | Func("func", ast::VariableList{}, ty.i32(), ast::StatementList{stmt}, |
| 1974 | ast::FunctionDecorationList{}); |
dan sinclair | ee8ae04 | 2020-04-08 19:58:20 +0000 | [diff] [blame] | 1975 | |
dan sinclair | b950e80 | 2020-04-20 14:20:01 +0000 | [diff] [blame] | 1976 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1977 | |
Ben Clayton | b17aea1 | 2021-02-03 17:51:09 +0000 | [diff] [blame] | 1978 | EXPECT_EQ(Sem().Get(var)->StorageClass(), ast::StorageClass::kNone); |
dan sinclair | ee8ae04 | 2020-04-08 19:58:20 +0000 | [diff] [blame] | 1979 | } |
| 1980 | |
| 1981 | TEST_F(TypeDeterminerTest, StorageClass_NonFunctionClassError) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 1982 | auto* var = Var("var", ty.i32(), ast::StorageClass::kWorkgroup); |
dan sinclair | ee8ae04 | 2020-04-08 19:58:20 +0000 | [diff] [blame] | 1983 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 1984 | auto* stmt = create<ast::VariableDeclStatement>(var); |
Ben Clayton | 42d1e09 | 2021-02-02 14:29:15 +0000 | [diff] [blame] | 1985 | Func("func", ast::VariableList{}, ty.i32(), ast::StatementList{stmt}, |
| 1986 | ast::FunctionDecorationList{}); |
dan sinclair | ee8ae04 | 2020-04-08 19:58:20 +0000 | [diff] [blame] | 1987 | |
dan sinclair | b950e80 | 2020-04-20 14:20:01 +0000 | [diff] [blame] | 1988 | EXPECT_FALSE(td()->Determine()); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 1989 | |
dan sinclair | ee8ae04 | 2020-04-08 19:58:20 +0000 | [diff] [blame] | 1990 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 1991 | "error: function variable has a non-function storage class"); |
dan sinclair | ee8ae04 | 2020-04-08 19:58:20 +0000 | [diff] [blame] | 1992 | } |
| 1993 | |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 1994 | struct IntrinsicData { |
dan sinclair | ca1723e | 2020-04-20 15:47:55 +0000 | [diff] [blame] | 1995 | const char* name; |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 1996 | IntrinsicType intrinsic; |
dan sinclair | ca1723e | 2020-04-20 15:47:55 +0000 | [diff] [blame] | 1997 | }; |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 1998 | inline std::ostream& operator<<(std::ostream& out, IntrinsicData data) { |
dan sinclair | ca1723e | 2020-04-20 15:47:55 +0000 | [diff] [blame] | 1999 | out << data.name; |
| 2000 | return out; |
| 2001 | } |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2002 | using IntrinsicDataTest = TypeDeterminerTestWithParam<IntrinsicData>; |
| 2003 | TEST_P(IntrinsicDataTest, Lookup) { |
| 2004 | auto param = GetParam(); |
dan sinclair | ca1723e | 2020-04-20 15:47:55 +0000 | [diff] [blame] | 2005 | |
Ben Clayton | 316f9f6 | 2021-02-08 22:31:44 +0000 | [diff] [blame] | 2006 | EXPECT_EQ(TypeDeterminer::MatchIntrinsicType(param.name), param.intrinsic); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2007 | } |
| 2008 | INSTANTIATE_TEST_SUITE_P( |
| 2009 | TypeDeterminerTest, |
| 2010 | IntrinsicDataTest, |
| 2011 | testing::Values( |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 2012 | IntrinsicData{"abs", IntrinsicType::kAbs}, |
| 2013 | IntrinsicData{"acos", IntrinsicType::kAcos}, |
| 2014 | IntrinsicData{"all", IntrinsicType::kAll}, |
| 2015 | IntrinsicData{"any", IntrinsicType::kAny}, |
| 2016 | IntrinsicData{"arrayLength", IntrinsicType::kArrayLength}, |
| 2017 | IntrinsicData{"asin", IntrinsicType::kAsin}, |
| 2018 | IntrinsicData{"atan", IntrinsicType::kAtan}, |
| 2019 | IntrinsicData{"atan2", IntrinsicType::kAtan2}, |
| 2020 | IntrinsicData{"ceil", IntrinsicType::kCeil}, |
| 2021 | IntrinsicData{"clamp", IntrinsicType::kClamp}, |
| 2022 | IntrinsicData{"cos", IntrinsicType::kCos}, |
| 2023 | IntrinsicData{"cosh", IntrinsicType::kCosh}, |
| 2024 | IntrinsicData{"countOneBits", IntrinsicType::kCountOneBits}, |
| 2025 | IntrinsicData{"cross", IntrinsicType::kCross}, |
| 2026 | IntrinsicData{"determinant", IntrinsicType::kDeterminant}, |
| 2027 | IntrinsicData{"distance", IntrinsicType::kDistance}, |
| 2028 | IntrinsicData{"dot", IntrinsicType::kDot}, |
| 2029 | IntrinsicData{"dpdx", IntrinsicType::kDpdx}, |
| 2030 | IntrinsicData{"dpdxCoarse", IntrinsicType::kDpdxCoarse}, |
| 2031 | IntrinsicData{"dpdxFine", IntrinsicType::kDpdxFine}, |
| 2032 | IntrinsicData{"dpdy", IntrinsicType::kDpdy}, |
| 2033 | IntrinsicData{"dpdyCoarse", IntrinsicType::kDpdyCoarse}, |
| 2034 | IntrinsicData{"dpdyFine", IntrinsicType::kDpdyFine}, |
| 2035 | IntrinsicData{"exp", IntrinsicType::kExp}, |
| 2036 | IntrinsicData{"exp2", IntrinsicType::kExp2}, |
| 2037 | IntrinsicData{"faceForward", IntrinsicType::kFaceForward}, |
| 2038 | IntrinsicData{"floor", IntrinsicType::kFloor}, |
| 2039 | IntrinsicData{"fma", IntrinsicType::kFma}, |
| 2040 | IntrinsicData{"fract", IntrinsicType::kFract}, |
| 2041 | IntrinsicData{"frexp", IntrinsicType::kFrexp}, |
| 2042 | IntrinsicData{"fwidth", IntrinsicType::kFwidth}, |
| 2043 | IntrinsicData{"fwidthCoarse", IntrinsicType::kFwidthCoarse}, |
| 2044 | IntrinsicData{"fwidthFine", IntrinsicType::kFwidthFine}, |
| 2045 | IntrinsicData{"inverseSqrt", IntrinsicType::kInverseSqrt}, |
| 2046 | IntrinsicData{"isFinite", IntrinsicType::kIsFinite}, |
| 2047 | IntrinsicData{"isInf", IntrinsicType::kIsInf}, |
| 2048 | IntrinsicData{"isNan", IntrinsicType::kIsNan}, |
| 2049 | IntrinsicData{"isNormal", IntrinsicType::kIsNormal}, |
| 2050 | IntrinsicData{"ldexp", IntrinsicType::kLdexp}, |
| 2051 | IntrinsicData{"length", IntrinsicType::kLength}, |
| 2052 | IntrinsicData{"log", IntrinsicType::kLog}, |
| 2053 | IntrinsicData{"log2", IntrinsicType::kLog2}, |
| 2054 | IntrinsicData{"max", IntrinsicType::kMax}, |
| 2055 | IntrinsicData{"min", IntrinsicType::kMin}, |
| 2056 | IntrinsicData{"mix", IntrinsicType::kMix}, |
| 2057 | IntrinsicData{"modf", IntrinsicType::kModf}, |
| 2058 | IntrinsicData{"normalize", IntrinsicType::kNormalize}, |
| 2059 | IntrinsicData{"pow", IntrinsicType::kPow}, |
| 2060 | IntrinsicData{"reflect", IntrinsicType::kReflect}, |
| 2061 | IntrinsicData{"reverseBits", IntrinsicType::kReverseBits}, |
| 2062 | IntrinsicData{"round", IntrinsicType::kRound}, |
| 2063 | IntrinsicData{"select", IntrinsicType::kSelect}, |
| 2064 | IntrinsicData{"sign", IntrinsicType::kSign}, |
| 2065 | IntrinsicData{"sin", IntrinsicType::kSin}, |
| 2066 | IntrinsicData{"sinh", IntrinsicType::kSinh}, |
| 2067 | IntrinsicData{"smoothStep", IntrinsicType::kSmoothStep}, |
| 2068 | IntrinsicData{"sqrt", IntrinsicType::kSqrt}, |
| 2069 | IntrinsicData{"step", IntrinsicType::kStep}, |
| 2070 | IntrinsicData{"tan", IntrinsicType::kTan}, |
| 2071 | IntrinsicData{"tanh", IntrinsicType::kTanh}, |
| 2072 | IntrinsicData{"textureDimensions", IntrinsicType::kTextureDimensions}, |
| 2073 | IntrinsicData{"textureLoad", IntrinsicType::kTextureLoad}, |
| 2074 | IntrinsicData{"textureNumLayers", IntrinsicType::kTextureNumLayers}, |
| 2075 | IntrinsicData{"textureNumLevels", IntrinsicType::kTextureNumLevels}, |
| 2076 | IntrinsicData{"textureNumSamples", IntrinsicType::kTextureNumSamples}, |
| 2077 | IntrinsicData{"textureSample", IntrinsicType::kTextureSample}, |
| 2078 | IntrinsicData{"textureSampleBias", IntrinsicType::kTextureSampleBias}, |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2079 | IntrinsicData{"textureSampleCompare", |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 2080 | IntrinsicType::kTextureSampleCompare}, |
| 2081 | IntrinsicData{"textureSampleGrad", IntrinsicType::kTextureSampleGrad}, |
| 2082 | IntrinsicData{"textureSampleLevel", IntrinsicType::kTextureSampleLevel}, |
| 2083 | IntrinsicData{"trunc", IntrinsicType::kTrunc})); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2084 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2085 | TEST_F(TypeDeterminerTest, MatchIntrinsicNoMatch) { |
Ben Clayton | 316f9f6 | 2021-02-08 22:31:44 +0000 | [diff] [blame] | 2086 | EXPECT_EQ(TypeDeterminer::MatchIntrinsicType("not_intrinsic"), |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 2087 | IntrinsicType::kNone); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2088 | } |
| 2089 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2090 | using Intrinsic_DataPackingTest = TypeDeterminerTestWithParam<IntrinsicData>; |
| 2091 | TEST_P(Intrinsic_DataPackingTest, InferType) { |
Alan Baker | c63e1c0 | 2021-02-04 16:17:49 +0000 | [diff] [blame] | 2092 | auto param = GetParam(); |
| 2093 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2094 | bool pack4 = param.intrinsic == IntrinsicType::kPack4x8Snorm || |
| 2095 | param.intrinsic == IntrinsicType::kPack4x8Unorm; |
| 2096 | |
| 2097 | auto* call = pack4 ? Call(param.name, vec4<f32>(1.f, 2.f, 3.f, 4.f)) |
| 2098 | : Call(param.name, vec2<f32>(1.f, 2.f)); |
Alan Baker | c63e1c0 | 2021-02-04 16:17:49 +0000 | [diff] [blame] | 2099 | WrapInFunction(call); |
| 2100 | |
| 2101 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2102 | ASSERT_NE(TypeOf(call), nullptr); |
| 2103 | EXPECT_TRUE(TypeOf(call)->Is<type::U32>()); |
| 2104 | } |
| 2105 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2106 | TEST_P(Intrinsic_DataPackingTest, Error_IncorrectParamType) { |
| 2107 | auto param = GetParam(); |
| 2108 | |
| 2109 | bool pack4 = param.intrinsic == IntrinsicType::kPack4x8Snorm || |
| 2110 | param.intrinsic == IntrinsicType::kPack4x8Unorm; |
| 2111 | |
| 2112 | auto* call = pack4 ? Call(param.name, vec4<i32>(1, 2, 3, 4)) |
| 2113 | : Call(param.name, vec2<i32>(1, 2)); |
| 2114 | WrapInFunction(call); |
| 2115 | |
| 2116 | EXPECT_FALSE(td()->Determine()); |
| 2117 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2118 | EXPECT_THAT(td()->error(), HasSubstr("error: no matching call to " + |
| 2119 | std::string(param.name))); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2120 | } |
| 2121 | |
| 2122 | TEST_P(Intrinsic_DataPackingTest, Error_NoParams) { |
| 2123 | auto param = GetParam(); |
| 2124 | |
| 2125 | auto* call = Call(param.name); |
| 2126 | WrapInFunction(call); |
| 2127 | |
| 2128 | EXPECT_FALSE(td()->Determine()); |
| 2129 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2130 | EXPECT_THAT(td()->error(), HasSubstr("error: no matching call to " + |
| 2131 | std::string(param.name))); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2132 | } |
| 2133 | |
| 2134 | TEST_P(Intrinsic_DataPackingTest, Error_TooManyParams) { |
| 2135 | auto param = GetParam(); |
| 2136 | |
| 2137 | bool pack4 = param.intrinsic == IntrinsicType::kPack4x8Snorm || |
| 2138 | param.intrinsic == IntrinsicType::kPack4x8Unorm; |
| 2139 | |
| 2140 | auto* call = pack4 ? Call(param.name, vec4<f32>(1.f, 2.f, 3.f, 4.f), 1.0f) |
| 2141 | : Call(param.name, vec2<f32>(1.f, 2.f), 1.0f); |
| 2142 | WrapInFunction(call); |
| 2143 | |
| 2144 | EXPECT_FALSE(td()->Determine()); |
| 2145 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2146 | EXPECT_THAT(td()->error(), HasSubstr("error: no matching call to " + |
| 2147 | std::string(param.name))); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2148 | } |
| 2149 | |
Alan Baker | c63e1c0 | 2021-02-04 16:17:49 +0000 | [diff] [blame] | 2150 | INSTANTIATE_TEST_SUITE_P( |
| 2151 | TypeDeterminerTest, |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2152 | Intrinsic_DataPackingTest, |
Alan Baker | c63e1c0 | 2021-02-04 16:17:49 +0000 | [diff] [blame] | 2153 | testing::Values( |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 2154 | IntrinsicData{"pack4x8snorm", IntrinsicType::kPack4x8Snorm}, |
| 2155 | IntrinsicData{"pack4x8unorm", IntrinsicType::kPack4x8Unorm}, |
| 2156 | IntrinsicData{"pack2x16snorm", IntrinsicType::kPack2x16Snorm}, |
| 2157 | IntrinsicData{"pack2x16unorm", IntrinsicType::kPack2x16Unorm}, |
| 2158 | IntrinsicData{"pack2x16float", IntrinsicType::kPack2x16Float})); |
Alan Baker | c63e1c0 | 2021-02-04 16:17:49 +0000 | [diff] [blame] | 2159 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2160 | using Intrinsic_DataUnpackingTest = TypeDeterminerTestWithParam<IntrinsicData>; |
| 2161 | TEST_P(Intrinsic_DataUnpackingTest, InferType) { |
Alan Baker | cd17ea8 | 2021-02-09 21:23:00 +0000 | [diff] [blame] | 2162 | auto param = GetParam(); |
| 2163 | |
| 2164 | bool pack4 = param.intrinsic == IntrinsicType::kUnpack4x8Snorm || |
| 2165 | param.intrinsic == IntrinsicType::kUnpack4x8Unorm; |
| 2166 | |
| 2167 | auto* call = Call(param.name, 1u); |
| 2168 | WrapInFunction(call); |
| 2169 | |
| 2170 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2171 | ASSERT_NE(TypeOf(call), nullptr); |
| 2172 | EXPECT_TRUE(TypeOf(call)->is_float_vector()); |
| 2173 | if (pack4) { |
| 2174 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 4u); |
| 2175 | } else { |
| 2176 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 2u); |
| 2177 | } |
| 2178 | } |
| 2179 | |
| 2180 | INSTANTIATE_TEST_SUITE_P( |
| 2181 | TypeDeterminerTest, |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2182 | Intrinsic_DataUnpackingTest, |
Alan Baker | cd17ea8 | 2021-02-09 21:23:00 +0000 | [diff] [blame] | 2183 | testing::Values( |
| 2184 | IntrinsicData{"unpack4x8snorm", IntrinsicType::kUnpack4x8Snorm}, |
| 2185 | IntrinsicData{"unpack4x8unorm", IntrinsicType::kUnpack4x8Unorm}, |
| 2186 | IntrinsicData{"unpack2x16snorm", IntrinsicType::kUnpack2x16Snorm}, |
| 2187 | IntrinsicData{"unpack2x16unorm", IntrinsicType::kUnpack2x16Unorm}, |
| 2188 | IntrinsicData{"unpack2x16float", IntrinsicType::kUnpack2x16Float})); |
| 2189 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2190 | using Intrinsic_SingleParamTest = TypeDeterminerTestWithParam<IntrinsicData>; |
| 2191 | TEST_P(Intrinsic_SingleParamTest, Scalar) { |
dan sinclair | ca1723e | 2020-04-20 15:47:55 +0000 | [diff] [blame] | 2192 | auto param = GetParam(); |
| 2193 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2194 | auto* call = Call(param.name, 1.f); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2195 | WrapInFunction(call); |
dan sinclair | fd5d4ca | 2020-04-20 15:46:18 +0000 | [diff] [blame] | 2196 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2197 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2198 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2199 | ASSERT_NE(TypeOf(call), nullptr); |
| 2200 | EXPECT_TRUE(TypeOf(call)->is_float_scalar()); |
dan sinclair | fd5d4ca | 2020-04-20 15:46:18 +0000 | [diff] [blame] | 2201 | } |
| 2202 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2203 | TEST_P(Intrinsic_SingleParamTest, Vector) { |
dan sinclair | ca1723e | 2020-04-20 15:47:55 +0000 | [diff] [blame] | 2204 | auto param = GetParam(); |
| 2205 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2206 | auto* call = Call(param.name, vec3<f32>(1.0f, 1.0f, 3.0f)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2207 | WrapInFunction(call); |
dan sinclair | fd5d4ca | 2020-04-20 15:46:18 +0000 | [diff] [blame] | 2208 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2209 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2210 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2211 | ASSERT_NE(TypeOf(call), nullptr); |
| 2212 | EXPECT_TRUE(TypeOf(call)->is_float_vector()); |
| 2213 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | fd5d4ca | 2020-04-20 15:46:18 +0000 | [diff] [blame] | 2214 | } |
| 2215 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2216 | TEST_P(Intrinsic_SingleParamTest, Error_NoParams) { |
dan sinclair | ca1723e | 2020-04-20 15:47:55 +0000 | [diff] [blame] | 2217 | auto param = GetParam(); |
| 2218 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2219 | auto* call = Call(param.name); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2220 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2221 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2222 | EXPECT_FALSE(td()->Determine()); |
| 2223 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2224 | EXPECT_EQ(td()->error(), |
| 2225 | "error: no matching call to " + std::string(param.name) + |
| 2226 | "()\n\n" |
| 2227 | "2 candidate functions:\n " + |
| 2228 | std::string(param.name) + "(f32) -> f32\n " + |
| 2229 | std::string(param.name) + "(vecN<f32>) -> vecN<f32>\n"); |
dan sinclair | fd5d4ca | 2020-04-20 15:46:18 +0000 | [diff] [blame] | 2230 | } |
| 2231 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2232 | TEST_P(Intrinsic_SingleParamTest, Error_TooManyParams) { |
| 2233 | auto param = GetParam(); |
| 2234 | |
| 2235 | auto* call = Call(param.name, 1, 2, 3); |
| 2236 | WrapInFunction(call); |
| 2237 | |
| 2238 | EXPECT_FALSE(td()->Determine()); |
| 2239 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2240 | EXPECT_EQ(td()->error(), |
| 2241 | "error: no matching call to " + std::string(param.name) + |
| 2242 | "(i32, i32, i32)\n\n" |
| 2243 | "2 candidate functions:\n " + |
| 2244 | std::string(param.name) + "(f32) -> f32\n " + |
| 2245 | std::string(param.name) + "(vecN<f32>) -> vecN<f32>\n"); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2246 | } |
| 2247 | |
dan sinclair | a49328f | 2020-04-20 15:49:50 +0000 | [diff] [blame] | 2248 | INSTANTIATE_TEST_SUITE_P( |
| 2249 | TypeDeterminerTest, |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2250 | Intrinsic_SingleParamTest, |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 2251 | testing::Values(IntrinsicData{"acos", IntrinsicType::kAcos}, |
| 2252 | IntrinsicData{"asin", IntrinsicType::kAsin}, |
| 2253 | IntrinsicData{"atan", IntrinsicType::kAtan}, |
| 2254 | IntrinsicData{"ceil", IntrinsicType::kCeil}, |
| 2255 | IntrinsicData{"cos", IntrinsicType::kCos}, |
| 2256 | IntrinsicData{"cosh", IntrinsicType::kCosh}, |
| 2257 | IntrinsicData{"exp", IntrinsicType::kExp}, |
| 2258 | IntrinsicData{"exp2", IntrinsicType::kExp2}, |
| 2259 | IntrinsicData{"floor", IntrinsicType::kFloor}, |
| 2260 | IntrinsicData{"fract", IntrinsicType::kFract}, |
| 2261 | IntrinsicData{"inverseSqrt", IntrinsicType::kInverseSqrt}, |
| 2262 | IntrinsicData{"log", IntrinsicType::kLog}, |
| 2263 | IntrinsicData{"log2", IntrinsicType::kLog2}, |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 2264 | IntrinsicData{"round", IntrinsicType::kRound}, |
| 2265 | IntrinsicData{"sign", IntrinsicType::kSign}, |
| 2266 | IntrinsicData{"sin", IntrinsicType::kSin}, |
| 2267 | IntrinsicData{"sinh", IntrinsicType::kSinh}, |
| 2268 | IntrinsicData{"sqrt", IntrinsicType::kSqrt}, |
| 2269 | IntrinsicData{"tan", IntrinsicType::kTan}, |
| 2270 | IntrinsicData{"tanh", IntrinsicType::kTanh}, |
| 2271 | IntrinsicData{"trunc", IntrinsicType::kTrunc})); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2272 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2273 | TEST_F(IntrinsicDataTest, ArrayLength_Vector) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 2274 | Global("arr", ty.array<int>(), ast::StorageClass::kNone); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2275 | auto* call = Call("arrayLength", "arr"); |
| 2276 | WrapInFunction(call); |
| 2277 | |
| 2278 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2279 | |
| 2280 | ASSERT_NE(TypeOf(call), nullptr); |
| 2281 | EXPECT_TRUE(TypeOf(call)->Is<type::U32>()); |
| 2282 | } |
| 2283 | |
| 2284 | TEST_F(IntrinsicDataTest, ArrayLength_Error_ArraySized) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 2285 | Global("arr", ty.array<int, 4>(), ast::StorageClass::kNone); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2286 | auto* call = Call("arrayLength", "arr"); |
| 2287 | WrapInFunction(call); |
| 2288 | |
| 2289 | EXPECT_FALSE(td()->Determine()); |
| 2290 | |
| 2291 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2292 | "error: no matching call to arrayLength(ptr<array<i32, 4>>)\n\n" |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2293 | "1 candidate function:\n" |
| 2294 | " arrayLength(array<T>) -> u32\n"); |
| 2295 | } |
| 2296 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2297 | TEST_F(IntrinsicDataTest, Normalize_Vector) { |
| 2298 | auto* call = Call("normalize", vec3<f32>(1.0f, 1.0f, 3.0f)); |
| 2299 | WrapInFunction(call); |
| 2300 | |
| 2301 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2302 | |
| 2303 | ASSERT_NE(TypeOf(call), nullptr); |
| 2304 | EXPECT_TRUE(TypeOf(call)->is_float_vector()); |
| 2305 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
| 2306 | } |
| 2307 | |
| 2308 | TEST_F(IntrinsicDataTest, Normalize_Error_NoParams) { |
| 2309 | auto* call = Call("normalize"); |
| 2310 | WrapInFunction(call); |
| 2311 | |
| 2312 | EXPECT_FALSE(td()->Determine()); |
| 2313 | |
| 2314 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2315 | "error: no matching call to normalize()\n\n" |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2316 | "1 candidate function:\n" |
| 2317 | " normalize(vecN<f32>) -> vecN<f32>\n"); |
| 2318 | } |
| 2319 | |
| 2320 | TEST_F(IntrinsicDataTest, FrexpScalar) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 2321 | Global("exp", ty.i32(), ast::StorageClass::kWorkgroup); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2322 | auto* call = Call("frexp", 1.0f, "exp"); |
| 2323 | WrapInFunction(call); |
| 2324 | |
| 2325 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2326 | |
| 2327 | ASSERT_NE(TypeOf(call), nullptr); |
| 2328 | EXPECT_TRUE(TypeOf(call)->Is<type::F32>()); |
| 2329 | } |
| 2330 | |
| 2331 | TEST_F(IntrinsicDataTest, FrexpVector) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 2332 | Global("exp", ty.vec3<i32>(), ast::StorageClass::kWorkgroup); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2333 | auto* call = Call("frexp", vec3<f32>(1.0f, 2.0f, 3.0f), "exp"); |
| 2334 | WrapInFunction(call); |
| 2335 | |
| 2336 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2337 | |
| 2338 | ASSERT_NE(TypeOf(call), nullptr); |
| 2339 | EXPECT_TRUE(TypeOf(call)->Is<type::Vector>()); |
| 2340 | EXPECT_TRUE(TypeOf(call)->As<type::Vector>()->type()->Is<type::F32>()); |
| 2341 | } |
| 2342 | |
| 2343 | TEST_F(IntrinsicDataTest, Frexp_Error_FirstParamInt) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 2344 | Global("exp", ty.i32(), ast::StorageClass::kWorkgroup); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2345 | auto* call = Call("frexp", 1, "exp"); |
| 2346 | WrapInFunction(call); |
| 2347 | |
| 2348 | EXPECT_FALSE(td()->Determine()); |
| 2349 | |
| 2350 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2351 | "error: no matching call to frexp(i32, ptr<workgroup, i32>)\n\n" |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2352 | "2 candidate functions:\n" |
| 2353 | " frexp(f32, ptr<T>) -> f32 where: T is i32 or u32\n" |
| 2354 | " frexp(vecN<f32>, ptr<vecN<T>>) -> vecN<f32> " |
| 2355 | "where: T is i32 or u32\n"); |
| 2356 | } |
| 2357 | |
| 2358 | TEST_F(IntrinsicDataTest, Frexp_Error_SecondParamFloatPtr) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 2359 | Global("exp", ty.f32(), ast::StorageClass::kWorkgroup); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2360 | auto* call = Call("frexp", 1.0f, "exp"); |
| 2361 | WrapInFunction(call); |
| 2362 | |
| 2363 | EXPECT_FALSE(td()->Determine()); |
| 2364 | |
| 2365 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2366 | "error: no matching call to frexp(f32, ptr<workgroup, f32>)\n\n" |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2367 | "2 candidate functions:\n" |
| 2368 | " frexp(f32, ptr<T>) -> f32 where: T is i32 or u32\n" |
| 2369 | " frexp(vecN<f32>, ptr<vecN<T>>) -> vecN<f32> " |
| 2370 | "where: T is i32 or u32\n"); |
| 2371 | } |
| 2372 | |
| 2373 | TEST_F(IntrinsicDataTest, Frexp_Error_SecondParamNotAPointer) { |
| 2374 | auto* call = Call("frexp", 1.0f, 1); |
| 2375 | WrapInFunction(call); |
| 2376 | |
| 2377 | EXPECT_FALSE(td()->Determine()); |
| 2378 | |
| 2379 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2380 | "error: no matching call to frexp(f32, i32)\n\n" |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2381 | "2 candidate functions:\n" |
| 2382 | " frexp(f32, ptr<T>) -> f32 where: T is i32 or u32\n" |
| 2383 | " frexp(vecN<f32>, ptr<vecN<T>>) -> vecN<f32> " |
| 2384 | "where: T is i32 or u32\n"); |
| 2385 | } |
| 2386 | |
| 2387 | TEST_F(IntrinsicDataTest, Frexp_Error_VectorSizesDontMatch) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 2388 | Global("exp", ty.vec4<i32>(), ast::StorageClass::kWorkgroup); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2389 | auto* call = Call("frexp", vec2<f32>(1.0f, 2.0f), "exp"); |
| 2390 | WrapInFunction(call); |
| 2391 | |
| 2392 | EXPECT_FALSE(td()->Determine()); |
| 2393 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2394 | EXPECT_EQ(td()->error(), |
| 2395 | "error: no matching call to frexp(vec2<f32>, ptr<workgroup, " |
| 2396 | "vec4<i32>>)\n\n" |
| 2397 | "2 candidate functions:\n" |
| 2398 | " frexp(f32, ptr<T>) -> f32 where: T is i32 or u32\n" |
| 2399 | " frexp(vecN<f32>, ptr<vecN<T>>) -> vecN<f32> " |
| 2400 | "where: T is i32 or u32\n"); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2401 | } |
| 2402 | |
| 2403 | TEST_F(IntrinsicDataTest, ModfScalar) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 2404 | Global("whole", ty.f32(), ast::StorageClass::kWorkgroup); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2405 | auto* call = Call("modf", 1.0f, "whole"); |
| 2406 | WrapInFunction(call); |
| 2407 | |
| 2408 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2409 | |
| 2410 | ASSERT_NE(TypeOf(call), nullptr); |
| 2411 | EXPECT_TRUE(TypeOf(call)->Is<type::F32>()); |
| 2412 | } |
| 2413 | |
| 2414 | TEST_F(IntrinsicDataTest, ModfVector) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 2415 | Global("whole", ty.vec3<f32>(), ast::StorageClass::kWorkgroup); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2416 | auto* call = Call("modf", vec3<f32>(1.0f, 2.0f, 3.0f), "whole"); |
| 2417 | WrapInFunction(call); |
| 2418 | |
| 2419 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2420 | |
| 2421 | ASSERT_NE(TypeOf(call), nullptr); |
| 2422 | EXPECT_TRUE(TypeOf(call)->Is<type::Vector>()); |
| 2423 | EXPECT_TRUE(TypeOf(call)->As<type::Vector>()->type()->Is<type::F32>()); |
| 2424 | } |
| 2425 | |
| 2426 | TEST_F(IntrinsicDataTest, Modf_Error_FirstParamInt) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 2427 | Global("whole", ty.f32(), ast::StorageClass::kWorkgroup); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2428 | auto* call = Call("modf", 1, "whole"); |
| 2429 | WrapInFunction(call); |
| 2430 | |
| 2431 | EXPECT_FALSE(td()->Determine()); |
| 2432 | |
| 2433 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2434 | "error: no matching call to modf(i32, ptr<workgroup, f32>)\n\n" |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2435 | "2 candidate functions:\n" |
| 2436 | " modf(f32, ptr<f32>) -> f32\n" |
| 2437 | " modf(vecN<f32>, ptr<vecN<f32>>) -> vecN<f32>\n"); |
| 2438 | } |
| 2439 | |
| 2440 | TEST_F(IntrinsicDataTest, Modf_Error_SecondParamIntPtr) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 2441 | Global("whole", ty.i32(), ast::StorageClass::kWorkgroup); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2442 | auto* call = Call("modf", 1.0f, "whole"); |
| 2443 | WrapInFunction(call); |
| 2444 | |
| 2445 | EXPECT_FALSE(td()->Determine()); |
| 2446 | |
| 2447 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2448 | "error: no matching call to modf(f32, ptr<workgroup, i32>)\n\n" |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2449 | "2 candidate functions:\n" |
| 2450 | " modf(f32, ptr<f32>) -> f32\n" |
| 2451 | " modf(vecN<f32>, ptr<vecN<f32>>) -> vecN<f32>\n"); |
| 2452 | } |
| 2453 | |
| 2454 | TEST_F(IntrinsicDataTest, Modf_Error_SecondParamNotAPointer) { |
| 2455 | auto* call = Call("modf", 1.0f, 1.0f); |
| 2456 | WrapInFunction(call); |
| 2457 | |
| 2458 | EXPECT_FALSE(td()->Determine()); |
| 2459 | |
| 2460 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2461 | "error: no matching call to modf(f32, f32)\n\n" |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2462 | "2 candidate functions:\n" |
| 2463 | " modf(f32, ptr<f32>) -> f32\n" |
| 2464 | " modf(vecN<f32>, ptr<vecN<f32>>) -> vecN<f32>\n"); |
| 2465 | } |
| 2466 | |
| 2467 | TEST_F(IntrinsicDataTest, Modf_Error_VectorSizesDontMatch) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 2468 | Global("whole", ty.vec4<f32>(), ast::StorageClass::kWorkgroup); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2469 | auto* call = Call("modf", vec2<f32>(1.0f, 2.0f), "whole"); |
| 2470 | WrapInFunction(call); |
| 2471 | |
| 2472 | EXPECT_FALSE(td()->Determine()); |
| 2473 | |
| 2474 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2475 | "error: no matching call to modf(vec2<f32>, ptr<workgroup, " |
| 2476 | "vec4<f32>>)\n\n" |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2477 | "2 candidate functions:\n" |
| 2478 | " modf(vecN<f32>, ptr<vecN<f32>>) -> vecN<f32>\n" |
| 2479 | " modf(f32, ptr<f32>) -> f32\n"); |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2480 | } |
| 2481 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2482 | using Intrinsic_SingleParam_FloatOrInt_Test = |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2483 | TypeDeterminerTestWithParam<IntrinsicData>; |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2484 | TEST_P(Intrinsic_SingleParam_FloatOrInt_Test, Float_Scalar) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2485 | auto param = GetParam(); |
| 2486 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2487 | auto* call = Call(param.name, 1.f); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2488 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2489 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2490 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2491 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2492 | ASSERT_NE(TypeOf(call), nullptr); |
| 2493 | EXPECT_TRUE(TypeOf(call)->is_float_scalar()); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2494 | } |
| 2495 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2496 | TEST_P(Intrinsic_SingleParam_FloatOrInt_Test, Float_Vector) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2497 | auto param = GetParam(); |
| 2498 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2499 | auto* call = Call(param.name, vec3<f32>(1.0f, 1.0f, 3.0f)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2500 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2501 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2502 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2503 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2504 | ASSERT_NE(TypeOf(call), nullptr); |
| 2505 | EXPECT_TRUE(TypeOf(call)->is_float_vector()); |
| 2506 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2507 | } |
| 2508 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2509 | TEST_P(Intrinsic_SingleParam_FloatOrInt_Test, Sint_Scalar) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2510 | auto param = GetParam(); |
| 2511 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2512 | auto* call = Call(param.name, -1); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2513 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2514 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2515 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2516 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2517 | ASSERT_NE(TypeOf(call), nullptr); |
| 2518 | EXPECT_TRUE(TypeOf(call)->Is<type::I32>()); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2519 | } |
| 2520 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2521 | TEST_P(Intrinsic_SingleParam_FloatOrInt_Test, Sint_Vector) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2522 | auto param = GetParam(); |
| 2523 | |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2524 | ast::ExpressionList vals; |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2525 | vals.push_back(Expr(1)); |
| 2526 | vals.push_back(Expr(1)); |
| 2527 | vals.push_back(Expr(3)); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2528 | |
| 2529 | ast::ExpressionList params; |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2530 | params.push_back(vec3<i32>(vals)); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2531 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2532 | auto* call = Call(param.name, params); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2533 | WrapInFunction(call); |
Ben Clayton | 4bfe461 | 2020-11-16 16:41:47 +0000 | [diff] [blame] | 2534 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2535 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2536 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2537 | ASSERT_NE(TypeOf(call), nullptr); |
| 2538 | EXPECT_TRUE(TypeOf(call)->is_signed_integer_vector()); |
| 2539 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2540 | } |
| 2541 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2542 | TEST_P(Intrinsic_SingleParam_FloatOrInt_Test, Uint_Scalar) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2543 | auto param = GetParam(); |
| 2544 | |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2545 | ast::ExpressionList params; |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2546 | params.push_back(Expr(1u)); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2547 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2548 | auto* call = Call(param.name, params); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2549 | WrapInFunction(call); |
Ben Clayton | 4bfe461 | 2020-11-16 16:41:47 +0000 | [diff] [blame] | 2550 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2551 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2552 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2553 | ASSERT_NE(TypeOf(call), nullptr); |
| 2554 | EXPECT_TRUE(TypeOf(call)->Is<type::U32>()); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2555 | } |
| 2556 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2557 | TEST_P(Intrinsic_SingleParam_FloatOrInt_Test, Uint_Vector) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2558 | auto param = GetParam(); |
| 2559 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2560 | auto* call = Call(param.name, vec3<u32>(1u, 1u, 3u)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2561 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2562 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2563 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2564 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2565 | ASSERT_NE(TypeOf(call), nullptr); |
| 2566 | EXPECT_TRUE(TypeOf(call)->is_unsigned_integer_vector()); |
| 2567 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2568 | } |
| 2569 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2570 | TEST_P(Intrinsic_SingleParam_FloatOrInt_Test, Error_NoParams) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2571 | auto param = GetParam(); |
| 2572 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2573 | auto* call = Call(param.name); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2574 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2575 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2576 | EXPECT_FALSE(td()->Determine()); |
| 2577 | |
Alan Baker | e809fb3 | 2021-02-01 15:33:13 +0000 | [diff] [blame] | 2578 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2579 | "error: no matching call to " + std::string(param.name) + |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 2580 | "()\n\n" |
| 2581 | "2 candidate functions:\n " + |
| 2582 | std::string(param.name) + |
| 2583 | "(T) -> T where: T is f32, i32 or u32\n " + |
| 2584 | std::string(param.name) + |
| 2585 | "(vecN<T>) -> vecN<T> where: T is f32, i32 or u32\n"); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2586 | } |
| 2587 | |
| 2588 | INSTANTIATE_TEST_SUITE_P(TypeDeterminerTest, |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2589 | Intrinsic_SingleParam_FloatOrInt_Test, |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 2590 | testing::Values(IntrinsicData{"abs", |
| 2591 | IntrinsicType::kAbs})); |
dan sinclair | ca1723e | 2020-04-20 15:47:55 +0000 | [diff] [blame] | 2592 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2593 | TEST_F(TypeDeterminerTest, Intrinsic_Length_Scalar) { |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2594 | auto* call = Call("length", 1.f); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2595 | WrapInFunction(call); |
dan sinclair | 652a4b9 | 2020-04-20 21:09:14 +0000 | [diff] [blame] | 2596 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2597 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2598 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2599 | ASSERT_NE(TypeOf(call), nullptr); |
| 2600 | EXPECT_TRUE(TypeOf(call)->is_float_scalar()); |
dan sinclair | 652a4b9 | 2020-04-20 21:09:14 +0000 | [diff] [blame] | 2601 | } |
| 2602 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2603 | TEST_F(TypeDeterminerTest, Intrinsic_Length_FloatVector) { |
dan sinclair | 652a4b9 | 2020-04-20 21:09:14 +0000 | [diff] [blame] | 2604 | ast::ExpressionList params; |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2605 | params.push_back(vec3<f32>(1.0f, 1.0f, 3.0f)); |
dan sinclair | 652a4b9 | 2020-04-20 21:09:14 +0000 | [diff] [blame] | 2606 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2607 | auto* call = Call("length", params); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2608 | WrapInFunction(call); |
dan sinclair | 652a4b9 | 2020-04-20 21:09:14 +0000 | [diff] [blame] | 2609 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2610 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2611 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2612 | ASSERT_NE(TypeOf(call), nullptr); |
| 2613 | EXPECT_TRUE(TypeOf(call)->is_float_scalar()); |
dan sinclair | 652a4b9 | 2020-04-20 21:09:14 +0000 | [diff] [blame] | 2614 | } |
| 2615 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2616 | using Intrinsic_TwoParamTest = TypeDeterminerTestWithParam<IntrinsicData>; |
| 2617 | TEST_P(Intrinsic_TwoParamTest, Scalar) { |
dan sinclair | 37d62c9 | 2020-04-21 12:55:06 +0000 | [diff] [blame] | 2618 | auto param = GetParam(); |
| 2619 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2620 | auto* call = Call(param.name, 1.f, 1.f); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2621 | WrapInFunction(call); |
dan sinclair | 37d62c9 | 2020-04-21 12:55:06 +0000 | [diff] [blame] | 2622 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2623 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2624 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2625 | ASSERT_NE(TypeOf(call), nullptr); |
| 2626 | EXPECT_TRUE(TypeOf(call)->is_float_scalar()); |
dan sinclair | 37d62c9 | 2020-04-21 12:55:06 +0000 | [diff] [blame] | 2627 | } |
| 2628 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2629 | TEST_P(Intrinsic_TwoParamTest, Vector) { |
dan sinclair | 37d62c9 | 2020-04-21 12:55:06 +0000 | [diff] [blame] | 2630 | auto param = GetParam(); |
| 2631 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2632 | auto* call = Call(param.name, vec3<f32>(1.0f, 1.0f, 3.0f), |
| 2633 | vec3<f32>(1.0f, 1.0f, 3.0f)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2634 | WrapInFunction(call); |
dan sinclair | 37d62c9 | 2020-04-21 12:55:06 +0000 | [diff] [blame] | 2635 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2636 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2637 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2638 | ASSERT_NE(TypeOf(call), nullptr); |
| 2639 | EXPECT_TRUE(TypeOf(call)->is_float_vector()); |
| 2640 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | 37d62c9 | 2020-04-21 12:55:06 +0000 | [diff] [blame] | 2641 | } |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2642 | |
| 2643 | TEST_P(Intrinsic_TwoParamTest, Error_NoTooManyParams) { |
| 2644 | auto param = GetParam(); |
| 2645 | |
| 2646 | auto* call = Call(param.name, 1, 2, 3); |
| 2647 | WrapInFunction(call); |
| 2648 | |
| 2649 | EXPECT_FALSE(td()->Determine()); |
| 2650 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2651 | EXPECT_EQ(td()->error(), |
| 2652 | "error: no matching call to " + std::string(param.name) + |
| 2653 | "(i32, i32, i32)\n\n" |
| 2654 | "2 candidate functions:\n " + |
| 2655 | std::string(param.name) + "(f32, f32) -> f32\n " + |
| 2656 | std::string(param.name) + |
| 2657 | "(vecN<f32>, vecN<f32>) -> vecN<f32>\n"); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2658 | } |
| 2659 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2660 | TEST_P(Intrinsic_TwoParamTest, Error_NoParams) { |
dan sinclair | 37d62c9 | 2020-04-21 12:55:06 +0000 | [diff] [blame] | 2661 | auto param = GetParam(); |
| 2662 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2663 | auto* call = Call(param.name); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2664 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2665 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2666 | EXPECT_FALSE(td()->Determine()); |
| 2667 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2668 | EXPECT_EQ(td()->error(), |
| 2669 | "error: no matching call to " + std::string(param.name) + |
| 2670 | "()\n\n" |
| 2671 | "2 candidate functions:\n " + |
| 2672 | std::string(param.name) + "(f32, f32) -> f32\n " + |
| 2673 | std::string(param.name) + |
| 2674 | "(vecN<f32>, vecN<f32>) -> vecN<f32>\n"); |
dan sinclair | 37d62c9 | 2020-04-21 12:55:06 +0000 | [diff] [blame] | 2675 | } |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2676 | |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2677 | INSTANTIATE_TEST_SUITE_P( |
| 2678 | TypeDeterminerTest, |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2679 | Intrinsic_TwoParamTest, |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 2680 | testing::Values(IntrinsicData{"atan2", IntrinsicType::kAtan2}, |
| 2681 | IntrinsicData{"pow", IntrinsicType::kPow}, |
| 2682 | IntrinsicData{"step", IntrinsicType::kStep}, |
| 2683 | IntrinsicData{"reflect", IntrinsicType::kReflect})); |
dan sinclair | 37d62c9 | 2020-04-21 12:55:06 +0000 | [diff] [blame] | 2684 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2685 | TEST_F(TypeDeterminerTest, Intrinsic_Distance_Scalar) { |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2686 | auto* call = Call("distance", 1.f, 1.f); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2687 | WrapInFunction(call); |
dan sinclair | 5444438 | 2020-04-21 13:04:15 +0000 | [diff] [blame] | 2688 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2689 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2690 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2691 | ASSERT_NE(TypeOf(call), nullptr); |
| 2692 | EXPECT_TRUE(TypeOf(call)->is_float_scalar()); |
dan sinclair | 5444438 | 2020-04-21 13:04:15 +0000 | [diff] [blame] | 2693 | } |
| 2694 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2695 | TEST_F(TypeDeterminerTest, Intrinsic_Distance_Vector) { |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2696 | auto* call = Call("distance", vec3<f32>(1.0f, 1.0f, 3.0f), |
| 2697 | vec3<f32>(1.0f, 1.0f, 3.0f)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2698 | WrapInFunction(call); |
dan sinclair | 5444438 | 2020-04-21 13:04:15 +0000 | [diff] [blame] | 2699 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2700 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2701 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2702 | ASSERT_NE(TypeOf(call), nullptr); |
| 2703 | EXPECT_TRUE(TypeOf(call)->Is<type::F32>()); |
dan sinclair | 5444438 | 2020-04-21 13:04:15 +0000 | [diff] [blame] | 2704 | } |
| 2705 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2706 | TEST_F(TypeDeterminerTest, Intrinsic_Cross) { |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2707 | auto* call = |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2708 | Call("cross", vec3<f32>(1.0f, 2.0f, 3.0f), vec3<f32>(1.0f, 2.0f, 3.0f)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2709 | WrapInFunction(call); |
dan sinclair | ee39225 | 2020-06-08 23:48:15 +0000 | [diff] [blame] | 2710 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2711 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2712 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2713 | ASSERT_NE(TypeOf(call), nullptr); |
| 2714 | EXPECT_TRUE(TypeOf(call)->is_float_vector()); |
| 2715 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | ee39225 | 2020-06-08 23:48:15 +0000 | [diff] [blame] | 2716 | } |
| 2717 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2718 | TEST_F(TypeDeterminerTest, Intrinsic_Cross_Error_NoArgs) { |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2719 | auto* call = Call("cross"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2720 | WrapInFunction(call); |
dan sinclair | ee39225 | 2020-06-08 23:48:15 +0000 | [diff] [blame] | 2721 | |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 2722 | EXPECT_FALSE(td()->Determine()); |
| 2723 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2724 | EXPECT_EQ(td()->error(), R"(error: no matching call to cross() |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 2725 | |
| 2726 | 1 candidate function: |
| 2727 | cross(vec3<f32>, vec3<f32>) -> vec3<f32> |
| 2728 | )"); |
| 2729 | } |
| 2730 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2731 | TEST_F(TypeDeterminerTest, Intrinsic_Cross_Error_Scalar) { |
| 2732 | auto* call = Call("cross", 1.0f, 1.0f); |
| 2733 | WrapInFunction(call); |
| 2734 | |
| 2735 | EXPECT_FALSE(td()->Determine()); |
| 2736 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2737 | EXPECT_EQ(td()->error(), R"(error: no matching call to cross(f32, f32) |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2738 | |
| 2739 | 1 candidate function: |
| 2740 | cross(vec3<f32>, vec3<f32>) -> vec3<f32> |
| 2741 | )"); |
| 2742 | } |
| 2743 | |
| 2744 | TEST_F(TypeDeterminerTest, Intrinsic_Cross_Error_Vec3Int) { |
| 2745 | auto* call = Call("cross", vec3<i32>(1, 2, 3), vec3<i32>(1, 2, 3)); |
| 2746 | WrapInFunction(call); |
| 2747 | |
| 2748 | EXPECT_FALSE(td()->Determine()); |
| 2749 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2750 | EXPECT_EQ(td()->error(), |
| 2751 | R"(error: no matching call to cross(vec3<i32>, vec3<i32>) |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2752 | |
| 2753 | 1 candidate function: |
| 2754 | cross(vec3<f32>, vec3<f32>) -> vec3<f32> |
| 2755 | )"); |
| 2756 | } |
| 2757 | |
| 2758 | TEST_F(TypeDeterminerTest, Intrinsic_Cross_Error_Vec4) { |
| 2759 | auto* call = Call("cross", vec4<f32>(1.0f, 2.0f, 3.0f, 4.0f), |
| 2760 | vec4<f32>(1.0f, 2.0f, 3.0f, 4.0f)); |
| 2761 | |
| 2762 | WrapInFunction(call); |
| 2763 | |
| 2764 | EXPECT_FALSE(td()->Determine()); |
| 2765 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2766 | EXPECT_EQ(td()->error(), |
| 2767 | R"(error: no matching call to cross(vec4<f32>, vec4<f32>) |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2768 | |
| 2769 | 1 candidate function: |
| 2770 | cross(vec3<f32>, vec3<f32>) -> vec3<f32> |
| 2771 | )"); |
| 2772 | } |
| 2773 | |
| 2774 | TEST_F(TypeDeterminerTest, Intrinsic_Cross_Error_TooManyParams) { |
| 2775 | auto* call = Call("cross", vec3<f32>(1.0f, 2.0f, 3.0f), |
| 2776 | vec3<f32>(1.0f, 2.0f, 3.0f), vec3<f32>(1.0f, 2.0f, 3.0f)); |
| 2777 | |
| 2778 | WrapInFunction(call); |
| 2779 | |
| 2780 | EXPECT_FALSE(td()->Determine()); |
| 2781 | |
| 2782 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2783 | R"(error: no matching call to cross(vec3<f32>, vec3<f32>, vec3<f32>) |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 2784 | |
| 2785 | 1 candidate function: |
| 2786 | cross(vec3<f32>, vec3<f32>) -> vec3<f32> |
| 2787 | )"); |
| 2788 | } |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2789 | TEST_F(TypeDeterminerTest, Intrinsic_Normalize) { |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 2790 | auto* call = Call("normalize", vec3<f32>(1.0f, 1.0f, 3.0f)); |
| 2791 | WrapInFunction(call); |
| 2792 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2793 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2794 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2795 | ASSERT_NE(TypeOf(call), nullptr); |
| 2796 | EXPECT_TRUE(TypeOf(call)->is_float_vector()); |
| 2797 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | ee39225 | 2020-06-08 23:48:15 +0000 | [diff] [blame] | 2798 | } |
| 2799 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2800 | TEST_F(TypeDeterminerTest, Intrinsic_Normalize_NoArgs) { |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 2801 | auto* call = Call("normalize"); |
| 2802 | WrapInFunction(call); |
| 2803 | |
| 2804 | EXPECT_FALSE(td()->Determine()); |
| 2805 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2806 | EXPECT_EQ(td()->error(), R"(error: no matching call to normalize() |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 2807 | |
| 2808 | 1 candidate function: |
| 2809 | normalize(vecN<f32>) -> vecN<f32> |
| 2810 | )"); |
| 2811 | } |
| 2812 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2813 | using Intrinsic_ThreeParamTest = TypeDeterminerTestWithParam<IntrinsicData>; |
| 2814 | TEST_P(Intrinsic_ThreeParamTest, Scalar) { |
dan sinclair | 2287d01 | 2020-04-22 00:23:57 +0000 | [diff] [blame] | 2815 | auto param = GetParam(); |
| 2816 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2817 | auto* call = Call(param.name, 1.f, 1.f, 1.f); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2818 | WrapInFunction(call); |
dan sinclair | 2287d01 | 2020-04-22 00:23:57 +0000 | [diff] [blame] | 2819 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2820 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2821 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2822 | ASSERT_NE(TypeOf(call), nullptr); |
| 2823 | EXPECT_TRUE(TypeOf(call)->is_float_scalar()); |
dan sinclair | 2287d01 | 2020-04-22 00:23:57 +0000 | [diff] [blame] | 2824 | } |
| 2825 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2826 | TEST_P(Intrinsic_ThreeParamTest, Vector) { |
dan sinclair | 2287d01 | 2020-04-22 00:23:57 +0000 | [diff] [blame] | 2827 | auto param = GetParam(); |
| 2828 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2829 | auto* call = Call(param.name, vec3<f32>(1.0f, 1.0f, 3.0f), |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2830 | vec3<f32>(1.0f, 1.0f, 3.0f), vec3<f32>(1.0f, 1.0f, 3.0f)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2831 | WrapInFunction(call); |
dan sinclair | 2287d01 | 2020-04-22 00:23:57 +0000 | [diff] [blame] | 2832 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2833 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2834 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2835 | ASSERT_NE(TypeOf(call), nullptr); |
| 2836 | EXPECT_TRUE(TypeOf(call)->is_float_vector()); |
| 2837 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | 2287d01 | 2020-04-22 00:23:57 +0000 | [diff] [blame] | 2838 | } |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2839 | TEST_P(Intrinsic_ThreeParamTest, Error_NoParams) { |
dan sinclair | 2287d01 | 2020-04-22 00:23:57 +0000 | [diff] [blame] | 2840 | auto param = GetParam(); |
| 2841 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2842 | auto* call = Call(param.name); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2843 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2844 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2845 | EXPECT_FALSE(td()->Determine()); |
| 2846 | |
dan sinclair | 2287d01 | 2020-04-22 00:23:57 +0000 | [diff] [blame] | 2847 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2848 | "error: no matching call to " + std::string(param.name) + |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 2849 | "()\n\n" |
| 2850 | "2 candidate functions:\n " + |
| 2851 | std::string(param.name) + "(f32, f32, f32) -> f32\n " + |
| 2852 | std::string(param.name) + |
| 2853 | "(vecN<f32>, vecN<f32>, vecN<f32>) -> vecN<f32>\n"); |
dan sinclair | 2287d01 | 2020-04-22 00:23:57 +0000 | [diff] [blame] | 2854 | } |
| 2855 | |
| 2856 | INSTANTIATE_TEST_SUITE_P( |
| 2857 | TypeDeterminerTest, |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2858 | Intrinsic_ThreeParamTest, |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 2859 | testing::Values(IntrinsicData{"mix", IntrinsicType::kMix}, |
| 2860 | IntrinsicData{"smoothStep", IntrinsicType::kSmoothStep}, |
| 2861 | IntrinsicData{"fma", IntrinsicType::kFma}, |
| 2862 | IntrinsicData{"faceForward", IntrinsicType::kFaceForward})); |
dan sinclair | 5e5fb9c | 2020-06-08 19:01:07 +0000 | [diff] [blame] | 2863 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2864 | using Intrinsic_ThreeParam_FloatOrInt_Test = |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2865 | TypeDeterminerTestWithParam<IntrinsicData>; |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2866 | TEST_P(Intrinsic_ThreeParam_FloatOrInt_Test, Float_Scalar) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2867 | auto param = GetParam(); |
| 2868 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2869 | auto* call = Call(param.name, 1.f, 1.f, 1.f); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2870 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2871 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2872 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2873 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2874 | ASSERT_NE(TypeOf(call), nullptr); |
| 2875 | EXPECT_TRUE(TypeOf(call)->is_float_scalar()); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2876 | } |
| 2877 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2878 | TEST_P(Intrinsic_ThreeParam_FloatOrInt_Test, Float_Vector) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2879 | auto param = GetParam(); |
| 2880 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2881 | auto* call = Call(param.name, vec3<f32>(1.0f, 1.0f, 3.0f), |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2882 | vec3<f32>(1.0f, 1.0f, 3.0f), vec3<f32>(1.0f, 1.0f, 3.0f)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2883 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2884 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2885 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2886 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2887 | ASSERT_NE(TypeOf(call), nullptr); |
| 2888 | EXPECT_TRUE(TypeOf(call)->is_float_vector()); |
| 2889 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2890 | } |
| 2891 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2892 | TEST_P(Intrinsic_ThreeParam_FloatOrInt_Test, Sint_Scalar) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2893 | auto param = GetParam(); |
| 2894 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2895 | auto* call = Call(param.name, 1, 1, 1); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2896 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2897 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2898 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2899 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2900 | ASSERT_NE(TypeOf(call), nullptr); |
| 2901 | EXPECT_TRUE(TypeOf(call)->Is<type::I32>()); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2902 | } |
| 2903 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2904 | TEST_P(Intrinsic_ThreeParam_FloatOrInt_Test, Sint_Vector) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2905 | auto param = GetParam(); |
| 2906 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2907 | auto* call = Call(param.name, vec3<i32>(1, 1, 3), vec3<i32>(1, 1, 3), |
| 2908 | vec3<i32>(1, 1, 3)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2909 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2910 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2911 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2912 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2913 | ASSERT_NE(TypeOf(call), nullptr); |
| 2914 | EXPECT_TRUE(TypeOf(call)->is_signed_integer_vector()); |
| 2915 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2916 | } |
| 2917 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2918 | TEST_P(Intrinsic_ThreeParam_FloatOrInt_Test, Uint_Scalar) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2919 | auto param = GetParam(); |
| 2920 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2921 | auto* call = Call(param.name, 1u, 1u, 1u); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2922 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2923 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2924 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2925 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2926 | ASSERT_NE(TypeOf(call), nullptr); |
| 2927 | EXPECT_TRUE(TypeOf(call)->Is<type::U32>()); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2928 | } |
| 2929 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2930 | TEST_P(Intrinsic_ThreeParam_FloatOrInt_Test, Uint_Vector) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2931 | auto param = GetParam(); |
| 2932 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2933 | auto* call = Call(param.name, vec3<u32>(1u, 1u, 3u), vec3<u32>(1u, 1u, 3u), |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2934 | vec3<u32>(1u, 1u, 3u)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2935 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2936 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2937 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2938 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2939 | ASSERT_NE(TypeOf(call), nullptr); |
| 2940 | EXPECT_TRUE(TypeOf(call)->is_unsigned_integer_vector()); |
| 2941 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2942 | } |
| 2943 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2944 | TEST_P(Intrinsic_ThreeParam_FloatOrInt_Test, Error_NoParams) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2945 | auto param = GetParam(); |
| 2946 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2947 | auto* call = Call(param.name); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2948 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2949 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2950 | EXPECT_FALSE(td()->Determine()); |
| 2951 | |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2952 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 2953 | "error: no matching call to " + std::string(param.name) + |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 2954 | "()\n\n" |
| 2955 | "2 candidate functions:\n " + |
| 2956 | std::string(param.name) + |
| 2957 | "(T, T, T) -> T where: T is f32, i32 or u32\n " + |
| 2958 | std::string(param.name) + |
| 2959 | "(vecN<T>, vecN<T>, vecN<T>) -> vecN<T> where: T is f32, i32 " |
| 2960 | "or u32\n"); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2961 | } |
| 2962 | |
| 2963 | INSTANTIATE_TEST_SUITE_P(TypeDeterminerTest, |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2964 | Intrinsic_ThreeParam_FloatOrInt_Test, |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 2965 | testing::Values(IntrinsicData{"clamp", |
| 2966 | IntrinsicType::kClamp})); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2967 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2968 | using Intrinsic_Int_SingleParamTest = |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 2969 | TypeDeterminerTestWithParam<IntrinsicData>; |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2970 | TEST_P(Intrinsic_Int_SingleParamTest, Scalar) { |
dan sinclair | 5e5fb9c | 2020-06-08 19:01:07 +0000 | [diff] [blame] | 2971 | auto param = GetParam(); |
| 2972 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2973 | auto* call = Call(param.name, 1); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2974 | WrapInFunction(call); |
dan sinclair | 5e5fb9c | 2020-06-08 19:01:07 +0000 | [diff] [blame] | 2975 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2976 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2977 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2978 | ASSERT_NE(TypeOf(call), nullptr); |
| 2979 | EXPECT_TRUE(TypeOf(call)->is_integer_scalar()); |
dan sinclair | 5e5fb9c | 2020-06-08 19:01:07 +0000 | [diff] [blame] | 2980 | } |
| 2981 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2982 | TEST_P(Intrinsic_Int_SingleParamTest, Vector) { |
dan sinclair | 5e5fb9c | 2020-06-08 19:01:07 +0000 | [diff] [blame] | 2983 | auto param = GetParam(); |
| 2984 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 2985 | auto* call = Call(param.name, vec3<i32>(1, 1, 3)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2986 | WrapInFunction(call); |
dan sinclair | 5e5fb9c | 2020-06-08 19:01:07 +0000 | [diff] [blame] | 2987 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2988 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 2989 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 2990 | ASSERT_NE(TypeOf(call), nullptr); |
| 2991 | EXPECT_TRUE(TypeOf(call)->is_signed_integer_vector()); |
| 2992 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | 5e5fb9c | 2020-06-08 19:01:07 +0000 | [diff] [blame] | 2993 | } |
| 2994 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 2995 | TEST_P(Intrinsic_Int_SingleParamTest, Error_NoParams) { |
dan sinclair | 5e5fb9c | 2020-06-08 19:01:07 +0000 | [diff] [blame] | 2996 | auto param = GetParam(); |
| 2997 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 2998 | auto* call = Call(param.name); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 2999 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 3000 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3001 | EXPECT_FALSE(td()->Determine()); |
| 3002 | |
Alan Baker | e809fb3 | 2021-02-01 15:33:13 +0000 | [diff] [blame] | 3003 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 3004 | "error: no matching call to " + std::string(param.name) + |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 3005 | "()\n\n" |
| 3006 | "2 candidate functions:\n " + |
| 3007 | std::string(param.name) + |
| 3008 | "(T) -> T where: T is i32 or u32\n " + |
| 3009 | std::string(param.name) + |
| 3010 | "(vecN<T>) -> vecN<T> where: T is i32 or u32\n"); |
dan sinclair | 5e5fb9c | 2020-06-08 19:01:07 +0000 | [diff] [blame] | 3011 | } |
| 3012 | |
dan sinclair | af5df70 | 2020-06-08 23:48:26 +0000 | [diff] [blame] | 3013 | INSTANTIATE_TEST_SUITE_P( |
| 3014 | TypeDeterminerTest, |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 3015 | Intrinsic_Int_SingleParamTest, |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 3016 | testing::Values(IntrinsicData{"countOneBits", IntrinsicType::kCountOneBits}, |
| 3017 | IntrinsicData{"reverseBits", IntrinsicType::kReverseBits})); |
dan sinclair | 5e5fb9c | 2020-06-08 19:01:07 +0000 | [diff] [blame] | 3018 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 3019 | using Intrinsic_FloatOrInt_TwoParamTest = |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 3020 | TypeDeterminerTestWithParam<IntrinsicData>; |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 3021 | TEST_P(Intrinsic_FloatOrInt_TwoParamTest, Scalar_Signed) { |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3022 | auto param = GetParam(); |
| 3023 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 3024 | auto* call = Call(param.name, 1, 1); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3025 | WrapInFunction(call); |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3026 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3027 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 3028 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 3029 | ASSERT_NE(TypeOf(call), nullptr); |
| 3030 | EXPECT_TRUE(TypeOf(call)->Is<type::I32>()); |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3031 | } |
| 3032 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 3033 | TEST_P(Intrinsic_FloatOrInt_TwoParamTest, Scalar_Unsigned) { |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3034 | auto param = GetParam(); |
| 3035 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 3036 | auto* call = Call(param.name, 1u, 1u); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3037 | WrapInFunction(call); |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3038 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3039 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 3040 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 3041 | ASSERT_NE(TypeOf(call), nullptr); |
| 3042 | EXPECT_TRUE(TypeOf(call)->Is<type::U32>()); |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3043 | } |
| 3044 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 3045 | TEST_P(Intrinsic_FloatOrInt_TwoParamTest, Scalar_Float) { |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 3046 | auto param = GetParam(); |
| 3047 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 3048 | auto* call = Call(param.name, 1.0f, 1.0f); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3049 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 3050 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3051 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 3052 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 3053 | ASSERT_NE(TypeOf(call), nullptr); |
| 3054 | EXPECT_TRUE(TypeOf(call)->Is<type::F32>()); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 3055 | } |
| 3056 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 3057 | TEST_P(Intrinsic_FloatOrInt_TwoParamTest, Vector_Signed) { |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3058 | auto param = GetParam(); |
| 3059 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 3060 | auto* call = Call(param.name, vec3<i32>(1, 1, 3), vec3<i32>(1, 1, 3)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3061 | WrapInFunction(call); |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3062 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3063 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 3064 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 3065 | ASSERT_NE(TypeOf(call), nullptr); |
| 3066 | EXPECT_TRUE(TypeOf(call)->is_signed_integer_vector()); |
| 3067 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3068 | } |
| 3069 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 3070 | TEST_P(Intrinsic_FloatOrInt_TwoParamTest, Vector_Unsigned) { |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3071 | auto param = GetParam(); |
| 3072 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 3073 | auto* call = Call(param.name, vec3<u32>(1u, 1u, 3u), vec3<u32>(1u, 1u, 3u)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3074 | WrapInFunction(call); |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3075 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3076 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 3077 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 3078 | ASSERT_NE(TypeOf(call), nullptr); |
| 3079 | EXPECT_TRUE(TypeOf(call)->is_unsigned_integer_vector()); |
| 3080 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3081 | } |
| 3082 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 3083 | TEST_P(Intrinsic_FloatOrInt_TwoParamTest, Vector_Float) { |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3084 | auto param = GetParam(); |
| 3085 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 3086 | auto* call = |
| 3087 | Call(param.name, vec3<f32>(1.f, 1.f, 3.f), vec3<f32>(1.f, 1.f, 3.f)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3088 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 3089 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3090 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 3091 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 3092 | ASSERT_NE(TypeOf(call), nullptr); |
| 3093 | EXPECT_TRUE(TypeOf(call)->is_float_vector()); |
| 3094 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->size(), 3u); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 3095 | } |
| 3096 | |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 3097 | TEST_P(Intrinsic_FloatOrInt_TwoParamTest, Error_NoParams) { |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3098 | auto param = GetParam(); |
| 3099 | |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 3100 | auto* call = Call(param.name); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3101 | WrapInFunction(call); |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 3102 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3103 | EXPECT_FALSE(td()->Determine()); |
| 3104 | |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3105 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 3106 | "error: no matching call to " + std::string(param.name) + |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 3107 | "()\n\n" |
| 3108 | "2 candidate functions:\n " + |
| 3109 | std::string(param.name) + |
| 3110 | "(T, T) -> T where: T is f32, i32 or u32\n " + |
| 3111 | std::string(param.name) + |
| 3112 | "(vecN<T>, vecN<T>) -> vecN<T> where: T is f32, i32 or u32\n"); |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3113 | } |
| 3114 | |
dan sinclair | b4fee2f | 2020-09-22 19:42:13 +0000 | [diff] [blame] | 3115 | INSTANTIATE_TEST_SUITE_P( |
| 3116 | TypeDeterminerTest, |
Ben Clayton | 2522b29 | 2021-02-10 21:40:36 +0000 | [diff] [blame] | 3117 | Intrinsic_FloatOrInt_TwoParamTest, |
Ben Clayton | 052ab89 | 2021-02-08 19:53:42 +0000 | [diff] [blame] | 3118 | testing::Values(IntrinsicData{"min", IntrinsicType::kMin}, |
| 3119 | IntrinsicData{"max", IntrinsicType::kMax})); |
dan sinclair | 92bb557 | 2020-06-08 23:48:07 +0000 | [diff] [blame] | 3120 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 3121 | TEST_F(TypeDeterminerTest, Intrinsic_Determinant_2x2) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 3122 | Global("var", ty.mat2x2<f32>(), ast::StorageClass::kFunction); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 3123 | |
| 3124 | auto* call = Call("determinant", "var"); |
| 3125 | WrapInFunction(call); |
| 3126 | |
| 3127 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 3128 | |
| 3129 | ASSERT_NE(TypeOf(call), nullptr); |
| 3130 | EXPECT_TRUE(TypeOf(call)->Is<type::F32>()); |
| 3131 | } |
| 3132 | |
| 3133 | TEST_F(TypeDeterminerTest, Intrinsic_Determinant_3x3) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 3134 | Global("var", ty.mat3x3<f32>(), ast::StorageClass::kFunction); |
dan sinclair | 3819c26 | 2020-06-17 18:39:17 +0000 | [diff] [blame] | 3135 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 3136 | auto* call = Call("determinant", "var"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3137 | WrapInFunction(call); |
Ben Clayton | 4bfe461 | 2020-11-16 16:41:47 +0000 | [diff] [blame] | 3138 | |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3139 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 3140 | |
Ben Clayton | 1618f4b | 2021-02-03 21:02:25 +0000 | [diff] [blame] | 3141 | ASSERT_NE(TypeOf(call), nullptr); |
| 3142 | EXPECT_TRUE(TypeOf(call)->Is<type::F32>()); |
dan sinclair | 3819c26 | 2020-06-17 18:39:17 +0000 | [diff] [blame] | 3143 | } |
| 3144 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 3145 | TEST_F(TypeDeterminerTest, Intrinsic_Determinant_4x4) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 3146 | Global("var", ty.mat4x4<f32>(), ast::StorageClass::kFunction); |
dan sinclair | 3819c26 | 2020-06-17 18:39:17 +0000 | [diff] [blame] | 3147 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 3148 | auto* call = Call("determinant", "var"); |
| 3149 | WrapInFunction(call); |
dan sinclair | 3238eaa | 2020-06-17 20:22:08 +0000 | [diff] [blame] | 3150 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 3151 | EXPECT_TRUE(td()->Determine()) << td()->error(); |
| 3152 | |
| 3153 | ASSERT_NE(TypeOf(call), nullptr); |
| 3154 | EXPECT_TRUE(TypeOf(call)->Is<type::F32>()); |
| 3155 | } |
| 3156 | |
| 3157 | TEST_F(TypeDeterminerTest, Intrinsic_Determinant_NotSquare) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 3158 | Global("var", ty.mat2x3<f32>(), ast::StorageClass::kFunction); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 3159 | |
| 3160 | auto* call = Call("determinant", "var"); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3161 | WrapInFunction(call); |
dan sinclair | 3819c26 | 2020-06-17 18:39:17 +0000 | [diff] [blame] | 3162 | |
Ben Clayton | 59d2473 | 2021-02-08 22:42:54 +0000 | [diff] [blame] | 3163 | EXPECT_FALSE(td()->Determine()); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3164 | |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 3165 | EXPECT_EQ( |
| 3166 | td()->error(), |
| 3167 | "error: no matching call to determinant(ptr<function, mat2x3<f32>>)\n\n" |
| 3168 | "1 candidate function:\n" |
| 3169 | " determinant(matNxN<f32>) -> f32\n"); |
dan sinclair | 3819c26 | 2020-06-17 18:39:17 +0000 | [diff] [blame] | 3170 | } |
| 3171 | |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 3172 | TEST_F(TypeDeterminerTest, Intrinsic_Determinant_NotMatrix) { |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 3173 | Global("var", ty.f32(), ast::StorageClass::kFunction); |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 3174 | |
| 3175 | auto* call = Call("determinant", "var"); |
| 3176 | WrapInFunction(call); |
| 3177 | |
| 3178 | EXPECT_FALSE(td()->Determine()); |
| 3179 | |
| 3180 | EXPECT_EQ(td()->error(), |
Ben Clayton | 6b4924f | 2021-02-17 20:13:34 +0000 | [diff] [blame] | 3181 | "error: no matching call to determinant(ptr<function, f32>)\n\n" |
Ben Clayton | 47c4d18 | 2021-02-10 21:42:35 +0000 | [diff] [blame] | 3182 | "1 candidate function:\n" |
| 3183 | " determinant(matNxN<f32>) -> f32\n"); |
| 3184 | } |
dan sinclair | e9598d6 | 2020-06-18 18:03:00 +0000 | [diff] [blame] | 3185 | |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3186 | TEST_F(TypeDeterminerTest, Function_EntryPoints_StageDecoration) { |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3187 | // fn b() {} |
| 3188 | // fn c() { b(); } |
| 3189 | // fn a() { c(); } |
| 3190 | // fn ep_1() { a(); b(); } |
| 3191 | // fn ep_2() { c();} |
| 3192 | // |
| 3193 | // c -> {ep_1, ep_2} |
| 3194 | // a -> {ep_1} |
| 3195 | // b -> {ep_1, ep_2} |
| 3196 | // ep_1 -> {} |
| 3197 | // ep_2 -> {} |
| 3198 | |
| 3199 | ast::VariableList params; |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 3200 | auto* func_b = Func("b", params, ty.f32(), ast::StatementList{}, |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 3201 | ast::FunctionDecorationList{}); |
dan sinclair | a41132f | 2020-12-11 18:24:53 +0000 | [diff] [blame] | 3202 | auto* func_c = |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 3203 | Func("c", params, ty.f32(), |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 3204 | ast::StatementList{ |
| 3205 | create<ast::AssignmentStatement>(Expr("second"), Call("b")), |
| 3206 | }, |
| 3207 | ast::FunctionDecorationList{}); |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3208 | |
dan sinclair | a41132f | 2020-12-11 18:24:53 +0000 | [diff] [blame] | 3209 | auto* func_a = |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 3210 | Func("a", params, ty.f32(), |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 3211 | ast::StatementList{ |
| 3212 | create<ast::AssignmentStatement>(Expr("first"), Call("c")), |
| 3213 | }, |
| 3214 | ast::FunctionDecorationList{}); |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3215 | |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 3216 | auto* ep_1 = |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 3217 | Func("ep_1", params, ty.f32(), |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 3218 | ast::StatementList{ |
| 3219 | create<ast::AssignmentStatement>(Expr("call_a"), Call("a")), |
| 3220 | create<ast::AssignmentStatement>(Expr("call_b"), Call("b")), |
| 3221 | }, |
| 3222 | ast::FunctionDecorationList{ |
| 3223 | create<ast::StageDecoration>(ast::PipelineStage::kVertex), |
| 3224 | }); |
Ben Clayton | 7eaf4b5 | 2020-12-14 22:08:27 +0000 | [diff] [blame] | 3225 | |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 3226 | auto* ep_2 = |
Ben Clayton | 8d391f7 | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 3227 | Func("ep_2", params, ty.f32(), |
dan sinclair | 181d8ba | 2020-12-16 15:15:40 +0000 | [diff] [blame] | 3228 | ast::StatementList{ |
| 3229 | create<ast::AssignmentStatement>(Expr("call_c"), Call("c")), |
| 3230 | }, |
| 3231 | ast::FunctionDecorationList{ |
| 3232 | create<ast::StageDecoration>(ast::PipelineStage::kVertex), |
| 3233 | }); |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3234 | |
Ben Clayton | 37571bc | 2021-02-16 23:57:01 +0000 | [diff] [blame] | 3235 | Global("first", ty.f32(), ast::StorageClass::kPrivate); |
| 3236 | Global("second", ty.f32(), ast::StorageClass::kPrivate); |
| 3237 | Global("call_a", ty.f32(), ast::StorageClass::kPrivate); |
| 3238 | Global("call_b", ty.f32(), ast::StorageClass::kPrivate); |
| 3239 | Global("call_c", ty.f32(), ast::StorageClass::kPrivate); |
dan sinclair | ff267ca | 2020-10-14 18:26:31 +0000 | [diff] [blame] | 3240 | |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3241 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
| 3242 | |
Ben Clayton | 87c78dd | 2021-02-03 16:43:20 +0000 | [diff] [blame] | 3243 | auto* func_b_sem = Sem().Get(func_b); |
| 3244 | auto* func_a_sem = Sem().Get(func_a); |
| 3245 | auto* func_c_sem = Sem().Get(func_c); |
| 3246 | auto* ep_1_sem = Sem().Get(ep_1); |
| 3247 | auto* ep_2_sem = Sem().Get(ep_2); |
| 3248 | ASSERT_NE(func_b_sem, nullptr); |
| 3249 | ASSERT_NE(func_a_sem, nullptr); |
| 3250 | ASSERT_NE(func_c_sem, nullptr); |
| 3251 | ASSERT_NE(ep_1_sem, nullptr); |
| 3252 | ASSERT_NE(ep_2_sem, nullptr); |
| 3253 | |
| 3254 | const auto& b_eps = func_b_sem->AncestorEntryPoints(); |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3255 | ASSERT_EQ(2u, b_eps.size()); |
Ben Clayton | 1f7e18b | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 3256 | EXPECT_EQ(Symbols().Register("ep_1"), b_eps[0]); |
| 3257 | EXPECT_EQ(Symbols().Register("ep_2"), b_eps[1]); |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3258 | |
Ben Clayton | 87c78dd | 2021-02-03 16:43:20 +0000 | [diff] [blame] | 3259 | const auto& a_eps = func_a_sem->AncestorEntryPoints(); |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3260 | ASSERT_EQ(1u, a_eps.size()); |
Ben Clayton | 1f7e18b | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 3261 | EXPECT_EQ(Symbols().Register("ep_1"), a_eps[0]); |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3262 | |
Ben Clayton | 87c78dd | 2021-02-03 16:43:20 +0000 | [diff] [blame] | 3263 | const auto& c_eps = func_c_sem->AncestorEntryPoints(); |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3264 | ASSERT_EQ(2u, c_eps.size()); |
Ben Clayton | 1f7e18b | 2021-01-26 16:57:10 +0000 | [diff] [blame] | 3265 | EXPECT_EQ(Symbols().Register("ep_1"), c_eps[0]); |
| 3266 | EXPECT_EQ(Symbols().Register("ep_2"), c_eps[1]); |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3267 | |
Ben Clayton | 87c78dd | 2021-02-03 16:43:20 +0000 | [diff] [blame] | 3268 | EXPECT_TRUE(ep_1_sem->AncestorEntryPoints().empty()); |
| 3269 | EXPECT_TRUE(ep_2_sem->AncestorEntryPoints().empty()); |
dan sinclair | 0592643 | 2020-09-21 17:51:31 +0000 | [diff] [blame] | 3270 | } |
| 3271 | |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3272 | using TypeDeterminerTextureIntrinsicTest = |
| 3273 | TypeDeterminerTestWithParam<ast::intrinsic::test::TextureOverloadCase>; |
| 3274 | |
| 3275 | INSTANTIATE_TEST_SUITE_P( |
| 3276 | TypeDeterminerTest, |
| 3277 | TypeDeterminerTextureIntrinsicTest, |
| 3278 | testing::ValuesIn(ast::intrinsic::test::TextureOverloadCase::ValidCases())); |
| 3279 | |
| 3280 | std::string to_str(const std::string& function, |
dan sinclair | 8130244 | 2021-02-09 15:37:44 +0000 | [diff] [blame] | 3281 | const semantic::ParameterList& params) { |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3282 | std::stringstream out; |
| 3283 | out << function << "("; |
| 3284 | bool first = true; |
Ben Clayton | 316f9f6 | 2021-02-08 22:31:44 +0000 | [diff] [blame] | 3285 | for (auto& param : params) { |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3286 | if (!first) { |
| 3287 | out << ", "; |
| 3288 | } |
Ben Clayton | 316f9f6 | 2021-02-08 22:31:44 +0000 | [diff] [blame] | 3289 | out << semantic::str(param.usage); |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3290 | first = false; |
| 3291 | } |
| 3292 | out << ")"; |
| 3293 | return out.str(); |
| 3294 | } |
| 3295 | |
| 3296 | const char* expected_texture_overload( |
| 3297 | ast::intrinsic::test::ValidTextureOverload overload) { |
| 3298 | using ValidTextureOverload = ast::intrinsic::test::ValidTextureOverload; |
| 3299 | switch (overload) { |
Ben Clayton | 4a0b9f7 | 2021-01-11 21:07:32 +0000 | [diff] [blame] | 3300 | case ValidTextureOverload::kDimensions1d: |
Ben Clayton | 4a0b9f7 | 2021-01-11 21:07:32 +0000 | [diff] [blame] | 3301 | case ValidTextureOverload::kDimensions2d: |
| 3302 | case ValidTextureOverload::kDimensions2dArray: |
| 3303 | case ValidTextureOverload::kDimensions3d: |
| 3304 | case ValidTextureOverload::kDimensionsCube: |
| 3305 | case ValidTextureOverload::kDimensionsCubeArray: |
Ben Clayton | 6e5b5ec | 2021-01-14 18:09:07 +0000 | [diff] [blame] | 3306 | case ValidTextureOverload::kDimensionsMultisampled2d: |
| 3307 | case ValidTextureOverload::kDimensionsMultisampled2dArray: |
Ben Clayton | 4a0b9f7 | 2021-01-11 21:07:32 +0000 | [diff] [blame] | 3308 | case ValidTextureOverload::kDimensionsDepth2d: |
| 3309 | case ValidTextureOverload::kDimensionsDepth2dArray: |
| 3310 | case ValidTextureOverload::kDimensionsDepthCube: |
| 3311 | case ValidTextureOverload::kDimensionsDepthCubeArray: |
| 3312 | case ValidTextureOverload::kDimensionsStorageRO1d: |
Ben Clayton | 4a0b9f7 | 2021-01-11 21:07:32 +0000 | [diff] [blame] | 3313 | case ValidTextureOverload::kDimensionsStorageRO2d: |
| 3314 | case ValidTextureOverload::kDimensionsStorageRO2dArray: |
| 3315 | case ValidTextureOverload::kDimensionsStorageRO3d: |
| 3316 | case ValidTextureOverload::kDimensionsStorageWO1d: |
Ben Clayton | 4a0b9f7 | 2021-01-11 21:07:32 +0000 | [diff] [blame] | 3317 | case ValidTextureOverload::kDimensionsStorageWO2d: |
| 3318 | case ValidTextureOverload::kDimensionsStorageWO2dArray: |
| 3319 | case ValidTextureOverload::kDimensionsStorageWO3d: |
| 3320 | return R"(textureDimensions(texture))"; |
Ben Clayton | c21f1f9 | 2021-01-14 16:50:07 +0000 | [diff] [blame] | 3321 | case ValidTextureOverload::kNumLayers2dArray: |
| 3322 | case ValidTextureOverload::kNumLayersCubeArray: |
Ben Clayton | 6e5b5ec | 2021-01-14 18:09:07 +0000 | [diff] [blame] | 3323 | case ValidTextureOverload::kNumLayersMultisampled2dArray: |
Ben Clayton | c21f1f9 | 2021-01-14 16:50:07 +0000 | [diff] [blame] | 3324 | case ValidTextureOverload::kNumLayersDepth2dArray: |
| 3325 | case ValidTextureOverload::kNumLayersDepthCubeArray: |
Ben Clayton | c21f1f9 | 2021-01-14 16:50:07 +0000 | [diff] [blame] | 3326 | case ValidTextureOverload::kNumLayersStorageWO2dArray: |
| 3327 | return R"(textureNumLayers(texture))"; |
Ben Clayton | d971320 | 2021-01-14 18:06:57 +0000 | [diff] [blame] | 3328 | case ValidTextureOverload::kNumLevels2d: |
| 3329 | case ValidTextureOverload::kNumLevels2dArray: |
| 3330 | case ValidTextureOverload::kNumLevels3d: |
| 3331 | case ValidTextureOverload::kNumLevelsCube: |
| 3332 | case ValidTextureOverload::kNumLevelsCubeArray: |
| 3333 | case ValidTextureOverload::kNumLevelsDepth2d: |
| 3334 | case ValidTextureOverload::kNumLevelsDepth2dArray: |
| 3335 | case ValidTextureOverload::kNumLevelsDepthCube: |
| 3336 | case ValidTextureOverload::kNumLevelsDepthCubeArray: |
| 3337 | return R"(textureNumLevels(texture))"; |
Ben Clayton | 0a68b36 | 2021-01-14 18:11:17 +0000 | [diff] [blame] | 3338 | case ValidTextureOverload::kNumSamplesMultisampled2d: |
| 3339 | case ValidTextureOverload::kNumSamplesMultisampled2dArray: |
| 3340 | return R"(textureNumSamples(texture))"; |
Ben Clayton | 4a0b9f7 | 2021-01-11 21:07:32 +0000 | [diff] [blame] | 3341 | case ValidTextureOverload::kDimensions2dLevel: |
| 3342 | case ValidTextureOverload::kDimensions2dArrayLevel: |
| 3343 | case ValidTextureOverload::kDimensions3dLevel: |
| 3344 | case ValidTextureOverload::kDimensionsCubeLevel: |
| 3345 | case ValidTextureOverload::kDimensionsCubeArrayLevel: |
| 3346 | case ValidTextureOverload::kDimensionsDepth2dLevel: |
| 3347 | case ValidTextureOverload::kDimensionsDepth2dArrayLevel: |
| 3348 | case ValidTextureOverload::kDimensionsDepthCubeLevel: |
| 3349 | case ValidTextureOverload::kDimensionsDepthCubeArrayLevel: |
| 3350 | return R"(textureDimensions(texture, level))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3351 | case ValidTextureOverload::kSample1dF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3352 | return R"(textureSample(texture, sampler, coords))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3353 | case ValidTextureOverload::kSample2dF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3354 | return R"(textureSample(texture, sampler, coords))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3355 | case ValidTextureOverload::kSample2dOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3356 | return R"(textureSample(texture, sampler, coords, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3357 | case ValidTextureOverload::kSample2dArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3358 | return R"(textureSample(texture, sampler, coords, array_index))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3359 | case ValidTextureOverload::kSample2dArrayOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3360 | return R"(textureSample(texture, sampler, coords, array_index, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3361 | case ValidTextureOverload::kSample3dF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3362 | return R"(textureSample(texture, sampler, coords))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3363 | case ValidTextureOverload::kSample3dOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3364 | return R"(textureSample(texture, sampler, coords, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3365 | case ValidTextureOverload::kSampleCubeF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3366 | return R"(textureSample(texture, sampler, coords))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3367 | case ValidTextureOverload::kSampleCubeArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3368 | return R"(textureSample(texture, sampler, coords, array_index))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3369 | case ValidTextureOverload::kSampleDepth2dF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3370 | return R"(textureSample(texture, sampler, coords))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3371 | case ValidTextureOverload::kSampleDepth2dOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3372 | return R"(textureSample(texture, sampler, coords, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3373 | case ValidTextureOverload::kSampleDepth2dArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3374 | return R"(textureSample(texture, sampler, coords, array_index))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3375 | case ValidTextureOverload::kSampleDepth2dArrayOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3376 | return R"(textureSample(texture, sampler, coords, array_index, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3377 | case ValidTextureOverload::kSampleDepthCubeF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3378 | return R"(textureSample(texture, sampler, coords))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3379 | case ValidTextureOverload::kSampleDepthCubeArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3380 | return R"(textureSample(texture, sampler, coords, array_index))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3381 | case ValidTextureOverload::kSampleBias2dF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3382 | return R"(textureSampleBias(texture, sampler, coords, bias))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3383 | case ValidTextureOverload::kSampleBias2dOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3384 | return R"(textureSampleBias(texture, sampler, coords, bias, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3385 | case ValidTextureOverload::kSampleBias2dArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3386 | return R"(textureSampleBias(texture, sampler, coords, array_index, bias))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3387 | case ValidTextureOverload::kSampleBias2dArrayOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3388 | return R"(textureSampleBias(texture, sampler, coords, array_index, bias, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3389 | case ValidTextureOverload::kSampleBias3dF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3390 | return R"(textureSampleBias(texture, sampler, coords, bias))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3391 | case ValidTextureOverload::kSampleBias3dOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3392 | return R"(textureSampleBias(texture, sampler, coords, bias, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3393 | case ValidTextureOverload::kSampleBiasCubeF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3394 | return R"(textureSampleBias(texture, sampler, coords, bias))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3395 | case ValidTextureOverload::kSampleBiasCubeArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3396 | return R"(textureSampleBias(texture, sampler, coords, array_index, bias))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3397 | case ValidTextureOverload::kSampleLevel2dF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3398 | return R"(textureSampleLevel(texture, sampler, coords, level))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3399 | case ValidTextureOverload::kSampleLevel2dOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3400 | return R"(textureSampleLevel(texture, sampler, coords, level, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3401 | case ValidTextureOverload::kSampleLevel2dArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3402 | return R"(textureSampleLevel(texture, sampler, coords, array_index, level))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3403 | case ValidTextureOverload::kSampleLevel2dArrayOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3404 | return R"(textureSampleLevel(texture, sampler, coords, array_index, level, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3405 | case ValidTextureOverload::kSampleLevel3dF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3406 | return R"(textureSampleLevel(texture, sampler, coords, level))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3407 | case ValidTextureOverload::kSampleLevel3dOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3408 | return R"(textureSampleLevel(texture, sampler, coords, level, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3409 | case ValidTextureOverload::kSampleLevelCubeF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3410 | return R"(textureSampleLevel(texture, sampler, coords, level))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3411 | case ValidTextureOverload::kSampleLevelCubeArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3412 | return R"(textureSampleLevel(texture, sampler, coords, array_index, level))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3413 | case ValidTextureOverload::kSampleLevelDepth2dF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3414 | return R"(textureSampleLevel(texture, sampler, coords, level))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3415 | case ValidTextureOverload::kSampleLevelDepth2dOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3416 | return R"(textureSampleLevel(texture, sampler, coords, level, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3417 | case ValidTextureOverload::kSampleLevelDepth2dArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3418 | return R"(textureSampleLevel(texture, sampler, coords, array_index, level))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3419 | case ValidTextureOverload::kSampleLevelDepth2dArrayOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3420 | return R"(textureSampleLevel(texture, sampler, coords, array_index, level, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3421 | case ValidTextureOverload::kSampleLevelDepthCubeF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3422 | return R"(textureSampleLevel(texture, sampler, coords, level))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3423 | case ValidTextureOverload::kSampleLevelDepthCubeArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3424 | return R"(textureSampleLevel(texture, sampler, coords, array_index, level))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3425 | case ValidTextureOverload::kSampleGrad2dF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3426 | return R"(textureSampleGrad(texture, sampler, coords, ddx, ddy))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3427 | case ValidTextureOverload::kSampleGrad2dOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3428 | return R"(textureSampleGrad(texture, sampler, coords, ddx, ddy, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3429 | case ValidTextureOverload::kSampleGrad2dArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3430 | return R"(textureSampleGrad(texture, sampler, coords, array_index, ddx, ddy))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3431 | case ValidTextureOverload::kSampleGrad2dArrayOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3432 | return R"(textureSampleGrad(texture, sampler, coords, array_index, ddx, ddy, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3433 | case ValidTextureOverload::kSampleGrad3dF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3434 | return R"(textureSampleGrad(texture, sampler, coords, ddx, ddy))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3435 | case ValidTextureOverload::kSampleGrad3dOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3436 | return R"(textureSampleGrad(texture, sampler, coords, ddx, ddy, offset))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3437 | case ValidTextureOverload::kSampleGradCubeF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3438 | return R"(textureSampleGrad(texture, sampler, coords, ddx, ddy))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3439 | case ValidTextureOverload::kSampleGradCubeArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3440 | return R"(textureSampleGrad(texture, sampler, coords, array_index, ddx, ddy))"; |
Ben Clayton | 5716626 | 2021-01-13 16:36:51 +0000 | [diff] [blame] | 3441 | case ValidTextureOverload::kSampleCompareDepth2dF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3442 | return R"(textureSampleCompare(texture, sampler, coords, depth_ref))"; |
Ben Clayton | 5716626 | 2021-01-13 16:36:51 +0000 | [diff] [blame] | 3443 | case ValidTextureOverload::kSampleCompareDepth2dOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3444 | return R"(textureSampleCompare(texture, sampler, coords, depth_ref, offset))"; |
Ben Clayton | 5716626 | 2021-01-13 16:36:51 +0000 | [diff] [blame] | 3445 | case ValidTextureOverload::kSampleCompareDepth2dArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3446 | return R"(textureSampleCompare(texture, sampler, coords, array_index, depth_ref))"; |
Ben Clayton | 5716626 | 2021-01-13 16:36:51 +0000 | [diff] [blame] | 3447 | case ValidTextureOverload::kSampleCompareDepth2dArrayOffsetF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3448 | return R"(textureSampleCompare(texture, sampler, coords, array_index, depth_ref, offset))"; |
Ben Clayton | 5716626 | 2021-01-13 16:36:51 +0000 | [diff] [blame] | 3449 | case ValidTextureOverload::kSampleCompareDepthCubeF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3450 | return R"(textureSampleCompare(texture, sampler, coords, depth_ref))"; |
Ben Clayton | 5716626 | 2021-01-13 16:36:51 +0000 | [diff] [blame] | 3451 | case ValidTextureOverload::kSampleCompareDepthCubeArrayF32: |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3452 | return R"(textureSampleCompare(texture, sampler, coords, array_index, depth_ref))"; |
Ryan Harrison | e5c105d | 2021-02-25 23:57:28 +0000 | [diff] [blame] | 3453 | case ValidTextureOverload::kLoad1dLevelF32: |
| 3454 | return R"(textureLoad(texture, coords, level))"; |
| 3455 | case ValidTextureOverload::kLoad1dLevelU32: |
| 3456 | return R"(textureLoad(texture, coords, level))"; |
| 3457 | case ValidTextureOverload::kLoad1dLevelI32: |
| 3458 | return R"(textureLoad(texture, coords, level))"; |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3459 | case ValidTextureOverload::kLoad2dLevelF32: |
| 3460 | return R"(textureLoad(texture, coords, level))"; |
| 3461 | case ValidTextureOverload::kLoad2dLevelU32: |
| 3462 | return R"(textureLoad(texture, coords, level))"; |
| 3463 | case ValidTextureOverload::kLoad2dLevelI32: |
| 3464 | return R"(textureLoad(texture, coords, level))"; |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3465 | case ValidTextureOverload::kLoad2dArrayLevelF32: |
| 3466 | return R"(textureLoad(texture, coords, array_index, level))"; |
| 3467 | case ValidTextureOverload::kLoad2dArrayLevelU32: |
| 3468 | return R"(textureLoad(texture, coords, array_index, level))"; |
| 3469 | case ValidTextureOverload::kLoad2dArrayLevelI32: |
| 3470 | return R"(textureLoad(texture, coords, array_index, level))"; |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3471 | case ValidTextureOverload::kLoad3dLevelF32: |
| 3472 | return R"(textureLoad(texture, coords, level))"; |
| 3473 | case ValidTextureOverload::kLoad3dLevelU32: |
| 3474 | return R"(textureLoad(texture, coords, level))"; |
| 3475 | case ValidTextureOverload::kLoad3dLevelI32: |
| 3476 | return R"(textureLoad(texture, coords, level))"; |
| 3477 | case ValidTextureOverload::kLoadMultisampled2dF32: |
| 3478 | return R"(textureLoad(texture, coords, sample_index))"; |
| 3479 | case ValidTextureOverload::kLoadMultisampled2dU32: |
| 3480 | return R"(textureLoad(texture, coords, sample_index))"; |
| 3481 | case ValidTextureOverload::kLoadMultisampled2dI32: |
| 3482 | return R"(textureLoad(texture, coords, sample_index))"; |
| 3483 | case ValidTextureOverload::kLoadMultisampled2dArrayF32: |
| 3484 | return R"(textureLoad(texture, coords, array_index, sample_index))"; |
| 3485 | case ValidTextureOverload::kLoadMultisampled2dArrayU32: |
| 3486 | return R"(textureLoad(texture, coords, array_index, sample_index))"; |
| 3487 | case ValidTextureOverload::kLoadMultisampled2dArrayI32: |
| 3488 | return R"(textureLoad(texture, coords, array_index, sample_index))"; |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3489 | case ValidTextureOverload::kLoadDepth2dLevelF32: |
| 3490 | return R"(textureLoad(texture, coords, level))"; |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3491 | case ValidTextureOverload::kLoadDepth2dArrayLevelF32: |
| 3492 | return R"(textureLoad(texture, coords, array_index, level))"; |
| 3493 | case ValidTextureOverload::kLoadStorageRO1dRgba32float: |
| 3494 | return R"(textureLoad(texture, coords))"; |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3495 | case ValidTextureOverload::kLoadStorageRO2dRgba8unorm: |
| 3496 | case ValidTextureOverload::kLoadStorageRO2dRgba8snorm: |
| 3497 | case ValidTextureOverload::kLoadStorageRO2dRgba8uint: |
| 3498 | case ValidTextureOverload::kLoadStorageRO2dRgba8sint: |
| 3499 | case ValidTextureOverload::kLoadStorageRO2dRgba16uint: |
| 3500 | case ValidTextureOverload::kLoadStorageRO2dRgba16sint: |
| 3501 | case ValidTextureOverload::kLoadStorageRO2dRgba16float: |
| 3502 | case ValidTextureOverload::kLoadStorageRO2dR32uint: |
| 3503 | case ValidTextureOverload::kLoadStorageRO2dR32sint: |
| 3504 | case ValidTextureOverload::kLoadStorageRO2dR32float: |
| 3505 | case ValidTextureOverload::kLoadStorageRO2dRg32uint: |
| 3506 | case ValidTextureOverload::kLoadStorageRO2dRg32sint: |
| 3507 | case ValidTextureOverload::kLoadStorageRO2dRg32float: |
| 3508 | case ValidTextureOverload::kLoadStorageRO2dRgba32uint: |
| 3509 | case ValidTextureOverload::kLoadStorageRO2dRgba32sint: |
| 3510 | case ValidTextureOverload::kLoadStorageRO2dRgba32float: |
| 3511 | return R"(textureLoad(texture, coords))"; |
| 3512 | case ValidTextureOverload::kLoadStorageRO2dArrayRgba32float: |
| 3513 | return R"(textureLoad(texture, coords, array_index))"; |
| 3514 | case ValidTextureOverload::kLoadStorageRO3dRgba32float: |
| 3515 | return R"(textureLoad(texture, coords))"; |
Ben Clayton | 591268d | 2020-12-10 18:39:41 +0000 | [diff] [blame] | 3516 | case ValidTextureOverload::kStoreWO1dRgba32float: |
| 3517 | return R"(textureStore(texture, coords, value))"; |
Ben Clayton | 591268d | 2020-12-10 18:39:41 +0000 | [diff] [blame] | 3518 | case ValidTextureOverload::kStoreWO2dRgba32float: |
| 3519 | return R"(textureStore(texture, coords, value))"; |
| 3520 | case ValidTextureOverload::kStoreWO2dArrayRgba32float: |
| 3521 | return R"(textureStore(texture, coords, array_index, value))"; |
| 3522 | case ValidTextureOverload::kStoreWO3dRgba32float: |
| 3523 | return R"(textureStore(texture, coords, value))"; |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3524 | } |
| 3525 | return "<unmatched texture overload>"; |
| 3526 | } |
| 3527 | |
| 3528 | TEST_P(TypeDeterminerTextureIntrinsicTest, Call) { |
| 3529 | auto param = GetParam(); |
| 3530 | |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3531 | param.buildTextureVariable(this); |
| 3532 | param.buildSamplerVariable(this); |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3533 | |
Ben Clayton | a51aa1f | 2021-02-08 21:10:11 +0000 | [diff] [blame] | 3534 | auto* call = Call(param.function, param.args(this)); |
Ben Clayton | 401b96b | 2021-02-03 17:19:59 +0000 | [diff] [blame] | 3535 | WrapInFunction(call); |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3536 | |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3537 | ASSERT_TRUE(td()->Determine()) << td()->error(); |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3538 | |
Ben Clayton | 4a0b9f7 | 2021-01-11 21:07:32 +0000 | [diff] [blame] | 3539 | if (std::string(param.function) == "textureDimensions") { |
| 3540 | switch (param.texture_dimension) { |
| 3541 | default: |
| 3542 | FAIL() << "invalid texture dimensions: " << param.texture_dimension; |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 3543 | case type::TextureDimension::k1d: |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 3544 | EXPECT_EQ(TypeOf(call)->type_name(), ty.i32()->type_name()); |
Ben Clayton | 4a0b9f7 | 2021-01-11 21:07:32 +0000 | [diff] [blame] | 3545 | break; |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 3546 | case type::TextureDimension::k2d: |
| 3547 | case type::TextureDimension::k2dArray: |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 3548 | EXPECT_EQ(TypeOf(call)->type_name(), ty.vec2<i32>()->type_name()); |
Ben Clayton | 4a0b9f7 | 2021-01-11 21:07:32 +0000 | [diff] [blame] | 3549 | break; |
Ben Clayton | 207b5e2 | 2021-01-21 15:42:10 +0000 | [diff] [blame] | 3550 | case type::TextureDimension::k3d: |
| 3551 | case type::TextureDimension::kCube: |
| 3552 | case type::TextureDimension::kCubeArray: |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 3553 | EXPECT_EQ(TypeOf(call)->type_name(), ty.vec3<i32>()->type_name()); |
Ben Clayton | 4a0b9f7 | 2021-01-11 21:07:32 +0000 | [diff] [blame] | 3554 | break; |
| 3555 | } |
Ben Clayton | c21f1f9 | 2021-01-14 16:50:07 +0000 | [diff] [blame] | 3556 | } else if (std::string(param.function) == "textureNumLayers") { |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 3557 | EXPECT_EQ(TypeOf(call), ty.i32()); |
Ben Clayton | d971320 | 2021-01-14 18:06:57 +0000 | [diff] [blame] | 3558 | } else if (std::string(param.function) == "textureNumLevels") { |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 3559 | EXPECT_EQ(TypeOf(call), ty.i32()); |
Ben Clayton | 0a68b36 | 2021-01-14 18:11:17 +0000 | [diff] [blame] | 3560 | } else if (std::string(param.function) == "textureNumSamples") { |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 3561 | EXPECT_EQ(TypeOf(call), ty.i32()); |
Ben Clayton | 4a0b9f7 | 2021-01-11 21:07:32 +0000 | [diff] [blame] | 3562 | } else if (std::string(param.function) == "textureStore") { |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 3563 | EXPECT_EQ(TypeOf(call), ty.void_()); |
Ben Clayton | 591268d | 2020-12-10 18:39:41 +0000 | [diff] [blame] | 3564 | } else { |
| 3565 | switch (param.texture_kind) { |
| 3566 | case ast::intrinsic::test::TextureKind::kRegular: |
| 3567 | case ast::intrinsic::test::TextureKind::kMultisampled: |
| 3568 | case ast::intrinsic::test::TextureKind::kStorage: { |
| 3569 | auto* datatype = param.resultVectorComponentType(this); |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 3570 | ASSERT_TRUE(TypeOf(call)->Is<type::Vector>()); |
| 3571 | EXPECT_EQ(TypeOf(call)->As<type::Vector>()->type(), datatype); |
Ben Clayton | 591268d | 2020-12-10 18:39:41 +0000 | [diff] [blame] | 3572 | break; |
| 3573 | } |
| 3574 | case ast::intrinsic::test::TextureKind::kDepth: { |
Ben Clayton | 3335254 | 2021-01-29 16:43:41 +0000 | [diff] [blame] | 3575 | EXPECT_EQ(TypeOf(call), ty.f32()); |
Ben Clayton | 591268d | 2020-12-10 18:39:41 +0000 | [diff] [blame] | 3576 | break; |
| 3577 | } |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3578 | } |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3579 | } |
| 3580 | |
Ben Clayton | 316f9f6 | 2021-02-08 22:31:44 +0000 | [diff] [blame] | 3581 | auto* call_sem = Sem().Get(call); |
| 3582 | ASSERT_NE(call_sem, nullptr); |
| 3583 | auto* target = call_sem->Target(); |
| 3584 | ASSERT_NE(target, nullptr); |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3585 | |
Ben Clayton | 316f9f6 | 2021-02-08 22:31:44 +0000 | [diff] [blame] | 3586 | auto got = ::tint::to_str(param.function, target->Parameters()); |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3587 | auto* expected = expected_texture_overload(param.overload); |
Ben Clayton | 7f04e5c | 2020-12-09 15:57:00 +0000 | [diff] [blame] | 3588 | EXPECT_EQ(got, expected); |
Ben Clayton | 3ea3c99 | 2020-11-18 21:19:22 +0000 | [diff] [blame] | 3589 | } |
| 3590 | |
dan sinclair | b7edc4c | 2020-04-07 12:46:30 +0000 | [diff] [blame] | 3591 | } // namespace |
| 3592 | } // namespace tint |