/*
 * Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
 *
 * SPDX-License-Identifier: MIT
 */

#include <hip_test_common.hh>

#include "kernels.hh"
#include "test_fixture.hh"

/**
 * @addtogroup texCubemapLayeredGrad texCubemapLayeredGrad
 * @{
 * @ingroup TextureTest
 */

/**
 * Test Description
 * ------------------------
 *    - Test texture fetching with `texCubemapLayeredGrad` and read mode set to
 * `hipReadModeElementType`. The test is performed with:
 *      - normalized coordinates
 *      - non-normalized coordinates
 *      - Nearest-point sampling
 *      - Linear filtering
 *      - All combinations of different addressing modes.
 * Test source
 * ------------------------
 *    - unit/texture/texCubemapLayeredGrad.cc
 * Test requirements
 * ------------------------
 *    - HIP_VERSION >= 5.7
 */
HIP_TEMPLATE_TEST_CASE(Unit_texCubemapLayeredGrad_Positive_ReadModeElementType, char,
                   unsigned char, short, unsigned short, int, unsigned int, float) {
  CHECK_IMAGE_SUPPORT;
  HIP_SKIP_TEST("texCubemap isn't supported.");
  TextureTestParams<TestType> params = {};
  params.extent = make_hipExtent(2, 2, 6);
  params.num_subdivisions = 4;
  params.layers = 1;
  params.cubemap = true;
  params.GenerateTextureDesc();

  TextureTestFixture<TestType, false, true> fixture{params};

  const auto [num_threads_x, num_blocks_x] = GetLaunchConfig(10, params.NumItersX());
  const auto [num_threads_y, num_blocks_y] = GetLaunchConfig(10, params.NumItersY());
  const auto [num_threads_z, num_blocks_z] = GetLaunchConfig(10, params.NumItersZ());

  dim3 dim_grid;
  dim_grid.x = num_blocks_x;
  dim_grid.y = num_blocks_y;
  dim_grid.z = num_blocks_z;

  dim3 dim_block;
  dim_block.x = num_threads_x;
  dim_block.y = num_threads_y;
  dim_block.z = num_threads_z;

  for (auto layer = 0u; layer < params.layers; ++layer) {
    texCubemapLayeredGradKernel<vec4<TestType>><<<dim_grid, dim_block>>>(
        fixture.out_alloc_d.ptr(), params.NumItersX(), params.NumItersY(), params.NumItersZ(),
        fixture.tex.object(), params.Width(), params.Height(), params.Depth(),
        params.num_subdivisions, params.tex_desc.normalizedCoords, layer, float4{}, float4{});
    HIP_CHECK(hipGetLastError());

    fixture.LoadOutput();

    for (auto k = 0u; k < params.NumItersZ(); ++k) {
      for (auto j = 0u; j < params.NumItersY(); ++j) {
        for (auto i = 0u; i < params.NumItersX(); ++i) {
          float x = GetCoordinate(i, params.NumItersX(), params.Width(), params.num_subdivisions,
                                  params.tex_desc.normalizedCoords);
          float y = GetCoordinate(j, params.NumItersY(), params.Height(), params.num_subdivisions,
                                  params.tex_desc.normalizedCoords);
          float z = GetCoordinate(k, params.NumItersZ(), params.Depth(), params.num_subdivisions,
                                  params.tex_desc.normalizedCoords);
          auto index = k * params.NumItersX() * params.NumItersY() + j * params.NumItersX() + i;
          const auto ref_val = fixture.tex_h.TexCubemap(x, y, z, params.tex_desc);
          if (!fixture.Verify(fixture.out_alloc_h[index], ref_val)) {
            INFO("Layer: " << layer);
            INFO("i: " << i);
            INFO("j: " << j);
            INFO("k: " << k);
            INFO("Filtering mode: " << FilteringModeToString(params.tex_desc.filterMode));
            INFO("Normalized coordinates: " << std::boolalpha << params.tex_desc.normalizedCoords);
            INFO("Address mode X: " << AddressModeToString(params.tex_desc.addressMode[0]));
            INFO("Address mode Y: " << AddressModeToString(params.tex_desc.addressMode[1]));
            INFO("Address mode Z: " << AddressModeToString(params.tex_desc.addressMode[2]));
            INFO("x: " << std::fixed << std::setprecision(16) << x);
            INFO("y: " << std::fixed << std::setprecision(16) << y);
            INFO("z: " << std::fixed << std::setprecision(16) << z);
            REQUIRE(false);
          }
        }
      }
    }
  }
}

/**
 * Test Description
 * ------------------------
 *    - Test texture fetching with `texCubemapLayeredGrad` and read mode set to
 * `hipReadModeNormalizedFloat`. The test is performed with:
 *      - normalized coordinates
 *      - non-normalized coordinates
 *      - Nearest-point sampling
 *      - Linear filtering
 *      - All combinations of different addressing modes.
 * Test source
 * ------------------------
 *    - unit/texture/texCubemapLayeredGrad.cc
 * Test requirements
 * ------------------------
 *    - HIP_VERSION >= 5.7
 */
HIP_TEMPLATE_TEST_CASE(Unit_texCubemapLayeredGrad_Positive_ReadModeNormalizedFloat, char,
                   unsigned char, short, unsigned short) {
  CHECK_IMAGE_SUPPORT;
  HIP_SKIP_TEST("texCubemap isn't supported.");
  TextureTestParams<TestType> params = {};
  params.extent = make_hipExtent(2, 2, 6);
  params.num_subdivisions = 4;
  params.layers = 1;
  params.cubemap = true;
  params.GenerateTextureDesc(hipReadModeNormalizedFloat);

  TextureTestFixture<TestType, true, true> fixture{params};

  const auto [num_threads_x, num_blocks_x] = GetLaunchConfig(10, params.NumItersX());
  const auto [num_threads_y, num_blocks_y] = GetLaunchConfig(10, params.NumItersY());
  const auto [num_threads_z, num_blocks_z] = GetLaunchConfig(10, params.NumItersZ());

  dim3 dim_grid;
  dim_grid.x = num_blocks_x;
  dim_grid.y = num_blocks_y;
  dim_grid.z = num_blocks_z;

  dim3 dim_block;
  dim_block.x = num_threads_x;
  dim_block.y = num_threads_y;
  dim_block.z = num_threads_z;

  for (auto layer = 0u; layer < params.layers; ++layer) {
    texCubemapLayeredGradKernel<vec4<float>><<<dim_grid, dim_block>>>(
        fixture.out_alloc_d.ptr(), params.NumItersX(), params.NumItersY(), params.NumItersZ(),
        fixture.tex.object(), params.Width(), params.Height(), params.Depth(),
        params.num_subdivisions, params.tex_desc.normalizedCoords, layer, float4{}, float4{});
    HIP_CHECK(hipGetLastError());

    fixture.LoadOutput();

    for (auto k = 0u; k < params.NumItersZ(); ++k) {
      for (auto j = 0u; j < params.NumItersY(); ++j) {
        for (auto i = 0u; i < params.NumItersX(); ++i) {
          float x = GetCoordinate(i, params.NumItersX(), params.Width(), params.num_subdivisions,
                                  params.tex_desc.normalizedCoords);
          float y = GetCoordinate(j, params.NumItersY(), params.Height(), params.num_subdivisions,
                                  params.tex_desc.normalizedCoords);
          float z = GetCoordinate(k, params.NumItersZ(), params.Depth(), params.num_subdivisions,
                                  params.tex_desc.normalizedCoords);
          auto index = k * params.NumItersX() * params.NumItersY() + j * params.NumItersX() + i;
          auto ref_val = fixture.tex_h.TexCubemap(x, y, z, params.tex_desc);
          if (!fixture.Verify(fixture.out_alloc_h[index], ref_val)) {
            INFO("Layer: " << layer);
            INFO("i: " << i);
            INFO("j: " << j);
            INFO("k: " << k);
            INFO("Filtering mode: " << FilteringModeToString(params.tex_desc.filterMode));
            INFO("Normalized coordinates: " << std::boolalpha << params.tex_desc.normalizedCoords);
            INFO("Address mode X: " << AddressModeToString(params.tex_desc.addressMode[0]));
            INFO("Address mode Y: " << AddressModeToString(params.tex_desc.addressMode[1]));
            INFO("Address mode Z: " << AddressModeToString(params.tex_desc.addressMode[2]));
            INFO("x: " << std::fixed << std::setprecision(16) << x);
            INFO("y: " << std::fixed << std::setprecision(16) << y);
            INFO("z: " << std::fixed << std::setprecision(16) << z);
            REQUIRE(false;)
          }
        }
      }
    }
  }
}

/**
 * End doxygen group TextureTest.
 * @}
 */