// Copyright 2020 The Tint Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "src/writer/hlsl/test_helper.h"

namespace tint {
namespace writer {
namespace hlsl {
namespace {

using HlslGeneratorImplTest_MemberAccessor = TestHelper;

TEST_F(HlslGeneratorImplTest_MemberAccessor, EmitExpression_MemberAccessor) {
  auto* s = Structure("Data", {Member("mem", ty.f32())});
  auto* str_var = Global("str", s, ast::StorageClass::kPrivate);

  auto* expr = MemberAccessor("str", "mem");
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(str_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "str.mem");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Load) {
  // struct Data {
  //   a : i32;
  //   b : f32;
  // };
  // var<storage> data : Data;
  // data.b;
  //
  // -> asfloat(data.Load(4));

  auto* s = Structure("data", {
                                  Member("a", ty.i32()),
                                  Member("b", ty.f32()),
                              });

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* expr = MemberAccessor("data", "b");
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "asfloat(data.Load(4))");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Load_Int) {
  // struct Data {
  //   a : i32;
  //   b : f32;
  // };
  // var<storage> data : Data;
  // data.a;
  //
  // -> asint(data.Load(0));

  auto* s = Structure("data", {
                                  Member("a", ty.i32()),
                                  Member("b", ty.f32()),
                              });

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* expr = MemberAccessor("data", "a");
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "asint(data.Load(0))");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Store_Matrix) {
  // struct Data {
  //   z : f32;
  //   a : mat2x3<f32>;
  // };
  // var<storage> data : Data;
  // mat2x3<f32> b;
  // data.a = b;
  //
  // -> float2x3 _tint_tmp = b;
  //    data.Store3(4 + 0, asuint(_tint_tmp[0]));
  //    data.Store3(4 + 16, asuint(_tint_tmp[1]));

  auto* s =
      Structure("Data", {Member("z", ty.i32()), Member("a", ty.mat2x3<f32>())});

  auto* b_var = Global("b", ty.mat2x3<f32>(), ast::StorageClass::kPrivate);
  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* lhs = MemberAccessor("data", "a");
  auto* rhs = Expr("b");

  auto* assign = create<ast::AssignmentStatement>(lhs, rhs);
  WrapInFunction(assign);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);
  gen.register_global(b_var);

  ASSERT_TRUE(gen.EmitStatement(out, assign)) << gen.error();
  EXPECT_EQ(result(), R"(float2x3 _tint_tmp = b;
data.Store3(16 + 0, asuint(_tint_tmp[0]));
data.Store3(16 + 16, asuint(_tint_tmp[1]));
)");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Store_Matrix_Empty) {
  // struct Data {
  //   z : f32;
  //   a : mat2x3<f32>;
  // };
  // var<storage> data : Data;
  // data.a = mat2x3<f32>();
  //
  // -> float2x3 _tint_tmp = float2x3(0.0f, 0.0f, 0.0f,
  // 0.0f, 0.0f, 0.0f);
  //    data.Store3(16 + 0, asuint(_tint_tmp[0]);
  //    data.Store3(16 + 16, asuint(_tint_tmp[1]));

  auto* s =
      Structure("Data", {Member("z", ty.i32()), Member("a", ty.mat2x3<f32>())});

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* lhs = MemberAccessor("data", "a");
  auto* rhs = Construct(ty.mat2x3<f32>(), ast::ExpressionList{});

  auto* assign = create<ast::AssignmentStatement>(lhs, rhs);
  WrapInFunction(assign);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitStatement(out, assign)) << gen.error();
  EXPECT_EQ(
      result(),
      R"(float2x3 _tint_tmp = float2x3(0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f);
data.Store3(16 + 0, asuint(_tint_tmp[0]));
data.Store3(16 + 16, asuint(_tint_tmp[1]));
)");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Load_Matrix) {
  // struct Data {
  //   z : f32;
  //   a : mat3x2<f32>;
  // };
  // var<storage> data : Data;
  // data.a;
  //
  // -> asfloat(uint2x3(data.Load2(4 + 0), data.Load2(4 + 8),
  // data.Load2(4 + 16)));

  auto* s =
      Structure("Data", {Member("z", ty.i32()), Member("a", ty.mat3x2<f32>())});

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* expr = MemberAccessor("data", "a");
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(),
            "asfloat(uint2x3(data.Load2(8 + 0), data.Load2(8 + 8), "
            "data.Load2(8 + 16)))");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Load_Matrix_Nested) {
  // struct Data {
  //   z : f32;
  //   a : mat2x3<f32>
  // };
  // var<storage> data : Outer;
  // data.b.a;
  //
  // -> asfloat(uint3x2(data.Load3(4 + 0), data.Load3(16 + 16)));

  auto* s = Structure("Data", {
                                  Member("z", ty.i32()),
                                  Member("a", ty.mat2x3<f32>()),
                              });

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* expr = MemberAccessor("data", "a");
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(),
            "asfloat(uint3x2(data.Load3(16 + 0), data.Load3(16 + 16)))");
}

TEST_F(
    HlslGeneratorImplTest_MemberAccessor,
    EmitExpression_MemberAccessor_StorageBuffer_Load_Matrix_By3_Is_16_Bytes) {
  // struct Data {
  //   a : mat3x3<f32>
  // };
  // var<storage> data : Data;
  // data.a;
  //
  // -> asfloat(uint3x3(data.Load3(0), data.Load3(16),
  // data.Load3(32)));

  auto* s = Structure("Data", {
                                  Member("a", ty.mat3x3<f32>()),
                              });

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* expr = MemberAccessor("data", "a");
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(),
            "asfloat(uint3x3(data.Load3(0 + 0), data.Load3(0 + 16), "
            "data.Load3(0 + 32)))");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Load_Matrix_Single_Element) {
  // struct Data {
  //   z : f32;
  //   a : mat4x3<f32>;
  // };
  // var<storage> data : Data;
  // data.a[2][1];
  //
  // -> asfloat(data.Load((2 * 16) + (1 * 4) + 16)))

  auto* s = Structure("Data", {
                                  Member("z", ty.i32()),
                                  Member("a", ty.mat4x3<f32>()),
                              });

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* expr = IndexAccessor(
      IndexAccessor(MemberAccessor("data", "a"), Expr(2)), Expr(1));
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "asfloat(data.Load((4 * 1) + (16 * 2) + 16))");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_ArrayAccessor_StorageBuffer_Load_Int_FromArray) {
  // struct Data {
  //   a : [[stride(4)]] array<i32, 5>;
  // };
  // var<storage> data : Data;
  // data.a[2];
  //
  // -> asint(data.Load((2 * 4));
  type::Array ary(ty.i32(), 5,
                  ast::DecorationList{
                      create<ast::StrideDecoration>(4),
                  });

  auto* s = Structure("Data", {Member("a", &ary)});

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* expr = IndexAccessor(MemberAccessor("data", "a"), Expr(2));
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "asint(data.Load((4 * 2) + 0))");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_ArrayAccessor_StorageBuffer_Load_Int_FromArray_ExprIdx) {
  // struct Data {
  //   a : [[stride(4)]] array<i32, 5>;
  // };
  // var<storage> data : Data;
  // data.a[(2 + 4) - 3];
  //
  // -> asint(data.Load((4 * ((2 + 4) - 3)));
  type::Array ary(ty.i32(), 5,
                  ast::DecorationList{
                      create<ast::StrideDecoration>(4),
                  });

  auto* s = Structure("Data", {Member("a", &ary)});

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* expr = IndexAccessor(MemberAccessor("data", "a"),
                             Sub(Add(Expr(2), Expr(4)), Expr(3)));
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "asint(data.Load((4 * ((2 + 4) - 3)) + 0))");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Store) {
  // struct Data {
  //   a : i32;
  //   b : f32;
  // };
  // var<storage> data : Data;
  // data.b = 2.3f;
  //
  // -> data.Store(0, asuint(2.0f));

  auto* s = Structure("data", {
                                  Member("a", ty.i32()),
                                  Member("b", ty.f32()),
                              });

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* lhs = MemberAccessor("data", "b");
  auto* rhs = Expr(2.0f);
  auto* assign = create<ast::AssignmentStatement>(lhs, rhs);
  WrapInFunction(assign);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitStatement(out, assign)) << gen.error();
  EXPECT_EQ(result(), R"(data.Store(4, asuint(2.0f));
)");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Store_ToArray) {
  // struct Data {
  //   a : [[stride(4)]] array<i32, 5>;
  // };
  // var<storage> data : Data;
  // data.a[2] = 2;
  //
  // -> data.Store((2 * 4), asuint(2.3f));

  type::Array ary(ty.i32(), 5,
                  ast::DecorationList{
                      create<ast::StrideDecoration>(4),
                  });

  auto* s = Structure("Data", {Member("a", &ary)});

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* lhs = IndexAccessor(MemberAccessor("data", "a"), Expr(2));
  auto* rhs = Expr(2);
  auto* assign = create<ast::AssignmentStatement>(lhs, rhs);
  WrapInFunction(assign);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitStatement(out, assign)) << gen.error();
  EXPECT_EQ(result(), R"(data.Store((4 * 2) + 0, asuint(2));
)");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Store_Int) {
  // struct Data {
  //   a : i32;
  //   b : f32;
  // };
  // var<storage> data : Data;
  // data.a = 2;
  //
  // -> data.Store(0, asuint(2));

  auto* s = Structure("data", {
                                  Member("a", ty.i32()),
                                  Member("b", ty.f32()),
                              });

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* lhs = MemberAccessor("data", "a");
  auto* rhs = Expr(2);
  auto* assign = create<ast::AssignmentStatement>(lhs, rhs);
  WrapInFunction(assign);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitStatement(out, assign)) << gen.error();
  EXPECT_EQ(result(), R"(data.Store(0, asuint(2));
)");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Load_Vec3) {
  // struct Data {
  //   a : vec3<i32>;
  //   b : vec3<f32>;
  // };
  // var<storage> data : Data;
  // data.b;
  //
  // -> asfloat(data.Load(16));

  auto* s = Structure("Data", {
                                  Member("a", ty.vec3<i32>()),
                                  Member("b", ty.vec3<f32>()),
                              });

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* expr = MemberAccessor("data", "b");
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "asfloat(data.Load3(16))");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Store_Vec3) {
  // struct Data {
  //   a : vec3<i32>;
  //   b : vec3<f32>;
  // };
  // var<storage> data : Data;
  // data.b = vec3<f32>(2.3f, 1.2f, 0.2f);
  //
  // -> data.Store(16, asuint(float3(2.3f, 1.2f, 0.2f)));

  auto* s = Structure("Data", {
                                  Member("a", ty.vec3<i32>()),
                                  Member("b", ty.vec3<f32>()),
                              });

  auto* coord_var = Global("data", s, ast::StorageClass::kStorage);

  auto* lhs = MemberAccessor("data", "b");
  auto* rhs = vec3<f32>(1.f, 2.f, 3.f);

  auto* assign = create<ast::AssignmentStatement>(lhs, rhs);

  WrapInFunction(assign);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitStatement(out, assign)) << gen.error();
  EXPECT_EQ(result(),
            R"(data.Store3(16, asuint(float3(1.0f, 2.0f, 3.0f)));
)");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Load_MultiLevel) {
  // struct Data {
  //   a : vec3<i32>;
  //   b : vec3<f32>;
  // };
  // struct Pre {
  //   var c : [[stride(32)]] array<Data, 4>;
  // };
  //
  // var<storage> data : Pre;
  // data.c[2].b
  //
  // -> asfloat(data.Load3(16 + (2 * 32)))

  auto* data = Structure("Data", {
                                     Member("a", ty.vec3<i32>()),
                                     Member("b", ty.vec3<f32>()),
                                 });

  type::Array ary(data, 4,
                  ast::DecorationList{
                      create<ast::StrideDecoration>(32),
                  });

  auto* pre_struct = Structure("Pre", {Member("c", &ary)});

  auto* coord_var = Global("data", pre_struct, ast::StorageClass::kStorage);

  auto* expr =
      MemberAccessor(IndexAccessor(MemberAccessor("data", "c"), Expr(2)), "b");
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "asfloat(data.Load3(16 + (32 * 2) + 0))");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Load_MultiLevel_Swizzle) {
  // struct Data {
  //   a : vec3<i32>;
  //   b : vec3<f32>;
  // };
  // struct Pre {
  //   var c : [[stride(32)]] array<Data, 4>;
  // };
  //
  // var<storage> data : Pre;
  // data.c[2].b.xy
  //
  // -> asfloat(data.Load3(16 + (2 * 32))).xy

  auto* data = Structure("Data", {
                                     Member("a", ty.vec3<i32>()),
                                     Member("b", ty.vec3<f32>()),
                                 });

  type::Array ary(data, 4,
                  ast::DecorationList{create<ast::StrideDecoration>(32)});

  auto* pre_struct = Structure("Pre", {Member("c", &ary)});

  auto* coord_var = Global("data", pre_struct, ast::StorageClass::kStorage);

  auto* expr = MemberAccessor(
      MemberAccessor(IndexAccessor(MemberAccessor("data", "c"), Expr(2)), "b"),
      "xy");
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "asfloat(data.Load3(16 + (32 * 2) + 0)).xy");
}

TEST_F(
    HlslGeneratorImplTest_MemberAccessor,
    EmitExpression_MemberAccessor_StorageBuffer_Load_MultiLevel_Swizzle_SingleLetter) {  // NOLINT
  // struct Data {
  //   a : vec3<i32>;
  //   b : vec3<f32>;
  // };
  // struct Pre {
  //   var c : [[stride(32)]] array<Data, 4>;
  // };
  //
  // var<storage> data : Pre;
  // data.c[2].b.g
  //
  // -> asfloat(data.Load((4 * 1) + 16 + (2 * 32) + 0))

  auto* data = Structure("Data", {
                                     Member("a", ty.vec3<i32>()),
                                     Member("b", ty.vec3<f32>()),
                                 });

  type::Array ary(data, 4,
                  ast::DecorationList{
                      create<ast::StrideDecoration>(32),
                  });

  auto* pre_struct = Structure("Pre", {Member("c", &ary)});

  auto* coord_var = Global("data", pre_struct, ast::StorageClass::kStorage);

  auto* expr = MemberAccessor(
      MemberAccessor(IndexAccessor(MemberAccessor("data", "c"), Expr(2)), "b"),
      "g");
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "asfloat(data.Load((4 * 1) + 16 + (32 * 2) + 0))");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Load_MultiLevel_Index) {
  // struct Data {
  //   a : vec3<i32>;
  //   b : vec3<f32>;
  // };
  // struct Pre {
  //   var c : [[stride(32)]] array<Data, 4>;
  // };
  //
  // var<storage> data : Pre;
  // data.c[2].b[1]
  //
  // -> asfloat(data.Load(4 + 16 + (2 * 32)))

  auto* data = Structure("Data", {
                                     Member("a", ty.vec3<i32>()),
                                     Member("b", ty.vec3<f32>()),
                                 });

  type::Array ary(data, 4,
                  ast::DecorationList{
                      create<ast::StrideDecoration>(32),
                  });

  auto* pre_struct = Structure("Pre", {Member("c", &ary)});

  auto* coord_var = Global("data", pre_struct, ast::StorageClass::kStorage);

  auto* expr = IndexAccessor(
      MemberAccessor(IndexAccessor(MemberAccessor("data", "c"), Expr(2)), "b"),
      Expr(1));
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "asfloat(data.Load((4 * 1) + 16 + (32 * 2) + 0))");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Store_MultiLevel) {
  // struct Data {
  //   a : vec3<i32>;
  //   b : vec3<f32>;
  // };
  // struct Pre {
  //   var c : [[stride(32)]] array<Data, 4>;
  // };
  //
  // var<storage> data : Pre;
  // data.c[2].b = vec3<f32>(1.f, 2.f, 3.f);
  //
  // -> data.Store3(16 + (2 * 32), asuint(float3(1.0f, 2.0f, 3.0f)));

  auto* data = Structure("Data", {
                                     Member("a", ty.vec3<i32>()),
                                     Member("b", ty.vec3<f32>()),
                                 });

  type::Array ary(data, 4,
                  ast::DecorationList{
                      create<ast::StrideDecoration>(32),
                  });

  auto* pre_struct = Structure("Pre", {Member("c", &ary)});

  auto* coord_var = Global("data", pre_struct, ast::StorageClass::kStorage);

  auto* lhs =
      MemberAccessor(IndexAccessor(MemberAccessor("data", "c"), Expr(2)), "b");

  auto* assign =
      create<ast::AssignmentStatement>(lhs, vec3<f32>(1.f, 2.f, 3.f));

  WrapInFunction(assign);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitStatement(out, assign)) << gen.error();
  EXPECT_EQ(result(),
            R"(data.Store3(16 + (32 * 2) + 0, asuint(float3(1.0f, 2.0f, 3.0f)));
)");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_StorageBuffer_Store_Swizzle_SingleLetter) {
  // struct Data {
  //   a : vec3<i32>;
  //   b : vec3<f32>;
  // };
  // struct Pre {
  //   var c : [[stride(32)]] array<Data, 4>;
  // };
  //
  // var<storage> data : Pre;
  // data.c[2].b.y = 1.f;
  //
  // -> data.Store((4 * 1) + 16 + (2 * 32) + 0, asuint(1.0f));

  auto* data = Structure("Data", {
                                     Member("a", ty.vec3<i32>()),
                                     Member("b", ty.vec3<f32>()),
                                 });

  type::Array ary(data, 4,
                  ast::DecorationList{
                      create<ast::StrideDecoration>(32),
                  });

  auto* pre_struct = Structure("Pre", {Member("c", &ary)});

  auto* coord_var = Global("data", pre_struct, ast::StorageClass::kStorage);

  auto* lhs = MemberAccessor(
      MemberAccessor(IndexAccessor(MemberAccessor("data", "c"), Expr(2)), "b"),
      "y");
  auto* rhs = Expr(1.f);

  auto* assign = create<ast::AssignmentStatement>(lhs, rhs);

  WrapInFunction(assign);

  GeneratorImpl& gen = Build();

  gen.register_global(coord_var);

  ASSERT_TRUE(gen.EmitStatement(out, assign)) << gen.error();
  EXPECT_EQ(result(),
            R"(data.Store((4 * 1) + 16 + (32 * 2) + 0, asuint(1.0f));
)");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_Swizzle_xyz) {
  Global("my_vec", ty.vec4<f32>(), ast::StorageClass::kPrivate);

  auto* expr = MemberAccessor("my_vec", "xyz");
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();
  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "my_vec.xyz");
}

TEST_F(HlslGeneratorImplTest_MemberAccessor,
       EmitExpression_MemberAccessor_Swizzle_gbr) {
  Global("my_vec", ty.vec4<f32>(), ast::StorageClass::kPrivate);

  auto* expr = MemberAccessor("my_vec", "gbr");
  WrapInFunction(expr);

  GeneratorImpl& gen = Build();
  ASSERT_TRUE(gen.EmitExpression(pre, out, expr)) << gen.error();
  EXPECT_EQ(result(), "my_vec.gbr");
}

}  // namespace
}  // namespace hlsl
}  // namespace writer
}  // namespace tint