| // Copyright 2023 The Dawn & Tint Authors |
| // |
| // Redistribution and use in source and binary forms, with or without |
| // modification, are permitted provided that the following conditions are met: |
| // |
| // 1. Redistributions of source code must retain the above copyright notice, this |
| // list of conditions and the following disclaimer. |
| // |
| // 2. Redistributions in binary form must reproduce the above copyright notice, |
| // this list of conditions and the following disclaimer in the documentation |
| // and/or other materials provided with the distribution. |
| // |
| // 3. Neither the name of the copyright holder nor the names of its |
| // contributors may be used to endorse or promote products derived from |
| // this software without specific prior written permission. |
| // |
| // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
| // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
| // DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE |
| // FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| // DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR |
| // SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER |
| // CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, |
| // OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| #include "src/tint/lang/core/ir/var.h" |
| |
| #include "gmock/gmock.h" |
| #include "gtest/gtest-spi.h" |
| #include "src/tint/lang/core/ir/builder.h" |
| #include "src/tint/lang/core/ir/instruction.h" |
| #include "src/tint/lang/core/ir/ir_helper_test.h" |
| |
| namespace tint::core::ir { |
| namespace { |
| |
| using namespace tint::core::fluent_types; // NOLINT |
| using namespace tint::core::number_suffixes; // NOLINT |
| |
| using IR_VarTest = IRTestHelper; |
| |
| TEST_F(IR_VarTest, Fail_NullType) { |
| EXPECT_FATAL_FAILURE( |
| { |
| Module mod; |
| Builder b{mod}; |
| b.Var(nullptr); |
| }, |
| ""); |
| } |
| |
| TEST_F(IR_VarTest, Results) { |
| auto* var = b.Var(ty.ptr<function, f32>()); |
| EXPECT_EQ(var->Results().Length(), 1u); |
| EXPECT_TRUE(var->Result(0)->Is<InstructionResult>()); |
| EXPECT_EQ(var->Result(0)->Instruction(), var); |
| } |
| |
| TEST_F(IR_VarTest, Initializer_Usage) { |
| Module mod; |
| Builder b{mod}; |
| auto* var = b.Var(ty.ptr<function, f32>()); |
| auto* init = b.Constant(1_f); |
| var->SetInitializer(init); |
| |
| EXPECT_THAT(init->Usages(), testing::UnorderedElementsAre(Usage{var, 0u})); |
| var->SetInitializer(nullptr); |
| EXPECT_TRUE(init->Usages().IsEmpty()); |
| } |
| |
| TEST_F(IR_VarTest, Clone) { |
| auto* v = b.Var(mod.Types().ptr(core::AddressSpace::kFunction, mod.Types().f32())); |
| v->SetInitializer(b.Constant(4_f)); |
| v->SetBindingPoint(1, 2); |
| v->SetAttributes(IOAttributes{ |
| 3, 4, core::BuiltinValue::kFragDepth, |
| Interpolation{core::InterpolationType::kFlat, core::InterpolationSampling::kCentroid}, |
| true}); |
| |
| auto* new_v = clone_ctx.Clone(v); |
| |
| EXPECT_NE(v, new_v); |
| ASSERT_NE(nullptr, new_v->Result(0)); |
| EXPECT_NE(v->Result(0), new_v->Result(0)); |
| EXPECT_EQ(new_v->Result(0)->Type(), |
| mod.Types().ptr(core::AddressSpace::kFunction, mod.Types().f32())); |
| |
| auto new_val = v->Initializer()->As<Constant>()->Value(); |
| ASSERT_TRUE(new_val->Is<core::constant::Scalar<f32>>()); |
| EXPECT_FLOAT_EQ(4_f, new_val->As<core::constant::Scalar<f32>>()->ValueAs<f32>()); |
| |
| EXPECT_TRUE(new_v->BindingPoint().has_value()); |
| EXPECT_EQ(1u, new_v->BindingPoint()->group); |
| EXPECT_EQ(2u, new_v->BindingPoint()->binding); |
| |
| auto& attrs = new_v->Attributes(); |
| EXPECT_TRUE(attrs.location.has_value()); |
| EXPECT_EQ(3u, attrs.location.value()); |
| |
| EXPECT_TRUE(attrs.index.has_value()); |
| EXPECT_EQ(4u, attrs.index.value()); |
| |
| EXPECT_TRUE(attrs.builtin.has_value()); |
| EXPECT_EQ(core::BuiltinValue::kFragDepth, attrs.builtin.value()); |
| |
| EXPECT_TRUE(attrs.interpolation.has_value()); |
| EXPECT_EQ(core::InterpolationType::kFlat, attrs.interpolation->type); |
| EXPECT_EQ(core::InterpolationSampling::kCentroid, attrs.interpolation->sampling); |
| |
| EXPECT_TRUE(attrs.invariant); |
| } |
| |
| TEST_F(IR_VarTest, CloneWithName) { |
| auto* v = b.Var("v", mod.Types().ptr(core::AddressSpace::kFunction, mod.Types().f32())); |
| auto* new_v = clone_ctx.Clone(v); |
| |
| EXPECT_EQ(std::string("v"), mod.NameOf(new_v).Name()); |
| } |
| |
| } // namespace |
| } // namespace tint::core::ir |