| // Copyright 2019 The Amber Authors. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or parseried. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| #include "gtest/gtest.h" |
| #include "src/amberscript/parser.h" |
| |
| namespace amber { |
| namespace amberscript { |
| |
| using AmberScriptParserTest = testing::Test; |
| |
| TEST_F(AmberScriptParserTest, Repeat) { |
| std::string in = R"( |
| SHADER compute shader GLSL |
| # shader |
| END |
| |
| PIPELINE compute my_pipeline |
| ATTACH shader |
| END |
| |
| REPEAT 4 |
| RUN my_pipeline 1 2 3 |
| RUN my_pipeline 4 5 6 |
| RUN my_pipeline 7 8 9 |
| END |
| )"; |
| |
| Parser parser; |
| Result r = parser.Parse(in); |
| ASSERT_TRUE(r.IsSuccess()) << r.Error(); |
| |
| auto script = parser.GetScript(); |
| const auto& commands = script->GetCommands(); |
| ASSERT_EQ(1U, commands.size()); |
| |
| auto* cmd = commands[0].get(); |
| ASSERT_TRUE(cmd->IsRepeat()); |
| |
| auto* repeat = cmd->AsRepeat(); |
| EXPECT_EQ(4U, repeat->GetCount()); |
| |
| const auto& repeat_cmds = repeat->GetCommands(); |
| ASSERT_EQ(3U, repeat_cmds.size()); |
| ASSERT_TRUE(repeat_cmds[0]->IsCompute()); |
| EXPECT_EQ(1U, repeat_cmds[0]->AsCompute()->GetX()); |
| ASSERT_TRUE(repeat_cmds[1]->IsCompute()); |
| EXPECT_EQ(4U, repeat_cmds[1]->AsCompute()->GetX()); |
| ASSERT_TRUE(repeat_cmds[2]->IsCompute()); |
| EXPECT_EQ(7U, repeat_cmds[2]->AsCompute()->GetX()); |
| } |
| |
| TEST_F(AmberScriptParserTest, RepeatMissingNum) { |
| std::string in = R"( |
| REPEAT |
| RUN my_pipeline 1 1 1 |
| END |
| )"; |
| |
| Parser parser; |
| Result r = parser.Parse(in); |
| ASSERT_FALSE(r.IsSuccess()); |
| EXPECT_EQ("3: missing count parameter for REPEAT command", r.Error()); |
| } |
| |
| TEST_F(AmberScriptParserTest, RepeatInvalidNum) { |
| std::string in = R"( |
| REPEAT INVALID |
| RUN my_pipeline 1 1 1 |
| END |
| )"; |
| |
| Parser parser; |
| Result r = parser.Parse(in); |
| ASSERT_FALSE(r.IsSuccess()); |
| EXPECT_EQ("2: invalid count parameter for REPEAT command: INVALID", |
| r.Error()); |
| } |
| |
| TEST_F(AmberScriptParserTest, RepeatFloatNum) { |
| std::string in = R"( |
| REPEAT 3.4 |
| RUN my_pipeline 1 1 1 |
| END |
| )"; |
| |
| Parser parser; |
| Result r = parser.Parse(in); |
| ASSERT_FALSE(r.IsSuccess()); |
| EXPECT_EQ("2: invalid count parameter for REPEAT command: 3.4", r.Error()); |
| } |
| |
| TEST_F(AmberScriptParserTest, RepeatMissingEnd) { |
| std::string in = R"( |
| SHADER compute shader GLSL |
| # shader |
| END |
| |
| PIPELINE compute my_pipeline |
| ATTACH shader |
| END |
| REPEAT 3 |
| RUN my_pipeline 1 1 1 |
| )"; |
| |
| Parser parser; |
| Result r = parser.Parse(in); |
| ASSERT_FALSE(r.IsSuccess()); |
| EXPECT_EQ("11: missing END for REPEAT command", r.Error()); |
| } |
| |
| TEST_F(AmberScriptParserTest, RepeatExtraParams) { |
| std::string in = R"( |
| REPEAT 3 EXTRA |
| RUN my_pipeline 1 1 1 |
| END |
| )"; |
| |
| Parser parser; |
| Result r = parser.Parse(in); |
| ASSERT_FALSE(r.IsSuccess()); |
| EXPECT_EQ("2: unknown token: EXTRA", r.Error()); |
| } |
| |
| TEST_F(AmberScriptParserTest, RepeatNegativeCount) { |
| std::string in = R"( |
| REPEAT -3 |
| RUN my_pipeline 1 1 1 |
| END |
| )"; |
| |
| Parser parser; |
| Result r = parser.Parse(in); |
| ASSERT_FALSE(r.IsSuccess()); |
| EXPECT_EQ("2: count parameter must be > 0 for REPEAT command", r.Error()); |
| } |
| |
| TEST_F(AmberScriptParserTest, RepeatZeroCount) { |
| std::string in = R"( |
| REPEAT 0 |
| RUN my_pipeline 1 1 1 |
| END |
| )"; |
| |
| Parser parser; |
| Result r = parser.Parse(in); |
| ASSERT_FALSE(r.IsSuccess()); |
| EXPECT_EQ("2: count parameter must be > 0 for REPEAT command", r.Error()); |
| } |
| |
| } // namespace amberscript |
| } // namespace amber |