// Copyright 2018 The Draco 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 "draco/io/gltf_decoder.h"

#include <cmath>
#include <limits>
#include <memory>
#include <string>
#include <utility>
#include <vector>

#ifdef DRACO_TRANSCODER_SUPPORTED
#include "draco/core/draco_test_base.h"
#include "draco/core/draco_test_utils.h"
#include "draco/core/draco_types.h"
#include "draco/mesh/mesh_are_equivalent.h"
#include "draco/mesh/mesh_utils.h"
#include "draco/scene/scene_indices.h"
#include "draco/scene/scene_utils.h"
#include "draco/texture/texture_utils.h"

namespace draco {

namespace {
std::unique_ptr<Mesh> DecodeGltfFile(const std::string &file_name) {
  const std::string path = GetTestFileFullPath(file_name);
  GltfDecoder decoder;

  auto maybe_geometry = decoder.DecodeFromFile(path);
  if (!maybe_geometry.ok()) {
    return nullptr;
  }
  std::unique_ptr<Mesh> geometry = std::move(maybe_geometry).value();
  return geometry;
}

std::unique_ptr<Scene> DecodeGltfFileToScene(const std::string &file_name) {
  const std::string path = GetTestFileFullPath(file_name);
  GltfDecoder decoder;

  auto maybe_scene = decoder.DecodeFromFileToScene(path);
  if (!maybe_scene.ok()) {
    return nullptr;
  }
  std::unique_ptr<Scene> scene = std::move(maybe_scene).value();
  return scene;
}

void CompareVectorArray(const std::array<Vector3f, 3> &a,
                        const std::array<Vector3f, 3> &b) {
  for (int v = 0; v < 3; ++v) {
    for (int c = 0; c < 3; ++c) {
      EXPECT_FLOAT_EQ(a[v][c], b[v][c]) << "v:" << v << " c:" << c;
    }
  }
}
}  // namespace

// Tests multiple textures.
TEST(GltfDecoderTest, SphereGltf) {
  const std::string file_name = "sphere.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  EXPECT_EQ(mesh->num_attributes(), 4) << "Unexpected number of attributes.";
  EXPECT_EQ(mesh->num_points(), 231) << "Unexpected number of vertices.";
  EXPECT_EQ(mesh->num_faces(), 224) << "Unexpected number of faces.";
  ASSERT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 2);
}

TEST(GltfDecoderTest, TriangleGltf) {
  const std::string file_name = "one_face_123.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  EXPECT_EQ(mesh->num_attributes(), 1) << "Unexpected number of attributes.";
  EXPECT_EQ(mesh->num_points(), 3) << "Unexpected number of vertices.";
  EXPECT_EQ(mesh->num_faces(), 1) << "Unexpected number of faces.";
  ASSERT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 0);

  const auto *const pos_attribute =
      mesh->GetNamedAttribute(GeometryAttribute::POSITION);
  EXPECT_NE(pos_attribute, nullptr);
  const auto &face = mesh->face(FaceIndex(0));
  std::array<Vector3f, 3> pos;
  for (int c = 0; c < 3; ++c) {
    pos_attribute->GetMappedValue(face[c], &pos[c][0]);
  }

  // Test position values match.
  std::array<Vector3f, 3> pos_test;
  pos_test[0] = Vector3f(1, 0.0999713, 0);
  pos_test[1] = Vector3f(2.00006104, 0.01, 0);
  pos_test[2] = Vector3f(3, 0.10998169, 0);
  CompareVectorArray(pos, pos_test);
}

TEST(GltfDecoderTest, MirroredTriangleGltf) {
  const std::string file_name = "one_face_123_mirror.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  EXPECT_EQ(mesh->num_attributes(), 1) << "Unexpected number of attributes.";
  EXPECT_EQ(mesh->num_points(), 3) << "Unexpected number of vertices.";
  EXPECT_EQ(mesh->num_faces(), 1) << "Unexpected number of faces.";
  ASSERT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 0);

  const auto *const pos_attribute =
      mesh->GetNamedAttribute(GeometryAttribute::POSITION);
  EXPECT_NE(pos_attribute, nullptr);
  const auto &face = mesh->face(FaceIndex(0));
  std::array<Vector3f, 3> pos;
  for (int c = 0; c < 3; ++c) {
    pos_attribute->GetMappedValue(face[c], &pos[c][0]);
  }

  // Test position values match.
  std::array<Vector3f, 3> pos_test;
  pos_test[0] = Vector3f(-1, -0.0999713, 0);
  pos_test[1] = Vector3f(-3, -0.10998169, 0);
  pos_test[2] = Vector3f(-2.00006104, -0.01, 0);
  CompareVectorArray(pos, pos_test);
}

TEST(GltfDecoderTest, TranslateTriangleGltf) {
  const std::string file_name = "one_face_123_translated.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  EXPECT_EQ(mesh->num_attributes(), 1) << "Unexpected number of attributes.";
  EXPECT_EQ(mesh->num_points(), 3) << "Unexpected number of vertices.";
  EXPECT_EQ(mesh->num_faces(), 1) << "Unexpected number of faces.";
  ASSERT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 0);

  const auto *const pos_attribute =
      mesh->GetNamedAttribute(GeometryAttribute::POSITION);
  EXPECT_NE(pos_attribute, nullptr);
  const auto &face = mesh->face(FaceIndex(0));
  std::array<Vector3f, 3> pos;
  for (int c = 0; c < 3; ++c) {
    pos_attribute->GetMappedValue(face[c], &pos[c][0]);
  }

  // Test position values match. The glTF file contains a matrix in the main
  // node. The matrix defines a translation of (-1.5, 5.0, 2.3).
  std::array<Vector3f, 3> pos_test;
  pos_test[0] = Vector3f(1, 0.0999713, 0);
  pos_test[1] = Vector3f(2.00006104, 0.01, 0);
  pos_test[2] = Vector3f(3, 0.10998169, 0);
  const Vector3f translate(-1.5, 5.0, 2.3);
  for (int v = 0; v < 3; ++v) {
    pos_test[v] = pos_test[v] + translate;
  }
  CompareVectorArray(pos, pos_test);
}

// Tests multiple materials.
TEST(GltfDecoderTest, MilkTruckGltf) {
  const std::string file_name = "CesiumMilkTruck/glTF/CesiumMilkTruck.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  EXPECT_EQ(mesh->num_attributes(), 4) << "Unexpected number of attributes.";
  EXPECT_EQ(mesh->num_points(), 3564) << "Unexpected number of vertices.";
  EXPECT_EQ(mesh->num_faces(), 3624) << "Unexpected number of faces.";
  ASSERT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 4);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 1);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(1)->NumTextureMaps(), 0);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(2)->NumTextureMaps(), 0);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(3)->NumTextureMaps(), 1);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->GetName(), "truck");
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(1)->GetName(), "glass");
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(2)->GetName(),
            "window_trim");
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(3)->GetName(), "wheels");
}

TEST(GltfDecoderTest, SceneMilkTruckGltf) {
  const std::string file_name = "CesiumMilkTruck/glTF/CesiumMilkTruck.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));

  ASSERT_EQ(scene->NumMeshes(), 4);
  ASSERT_EQ(scene->NumMeshGroups(), 2);
  ASSERT_EQ(scene->NumNodes(), 5);
  ASSERT_EQ(scene->NumRootNodes(), 1);
  ASSERT_EQ(scene->NumLights(), 0);
  ASSERT_EQ(scene->GetMaterialLibrary().NumMaterials(), 4);
  ASSERT_EQ(scene->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 1);
  ASSERT_EQ(scene->GetMaterialLibrary().GetMaterial(1)->NumTextureMaps(), 0);
  ASSERT_EQ(scene->GetMaterialLibrary().GetMaterial(2)->NumTextureMaps(), 0);
  ASSERT_EQ(scene->GetMaterialLibrary().GetMaterial(3)->NumTextureMaps(), 1);
  ASSERT_EQ(scene->GetMaterialLibrary().GetMaterial(0)->GetName(), "truck");
  ASSERT_EQ(scene->GetMaterialLibrary().GetMaterial(1)->GetName(), "glass");
  ASSERT_EQ(scene->GetMaterialLibrary().GetMaterial(2)->GetName(),
            "window_trim");
  ASSERT_EQ(scene->GetMaterialLibrary().GetMaterial(3)->GetName(), "wheels");
  ASSERT_EQ(scene->NumAnimations(), 1);
  ASSERT_EQ(scene->NumSkins(), 0);
  for (AnimationIndex i(0); i < scene->NumAnimations(); ++i) {
    const Animation *const animation = scene->GetAnimation(i);
    ASSERT_NE(animation, nullptr);
    ASSERT_EQ(animation->NumSamplers(), 2);
    ASSERT_EQ(animation->NumChannels(), 2);
  }

  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->GetName(),
            "Cesium_Milk_Truck");
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(1))->GetName(), "Wheels");

  // Check all of the meshes do not have any materials.
  for (MeshIndex i(0); i < scene->NumMeshes(); ++i) {
    const Mesh &mesh = scene->GetMesh(i);
    ASSERT_EQ(mesh.GetMaterialLibrary().NumMaterials(), 0);
  }
}

TEST(GltfDecoderTest, AnimatedBonesGltf) {
  const std::string file_name = "CesiumMan/glTF/CesiumMan.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));

  ASSERT_EQ(scene->NumMeshes(), 1);
  ASSERT_EQ(scene->NumMeshGroups(), 1);
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->NumMaterialIndices(), 1);
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->GetMaterialIndex(0), 0);
  ASSERT_EQ(scene->NumNodes(), 22);
  ASSERT_EQ(scene->NumRootNodes(), 1);
  ASSERT_EQ(scene->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(scene->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 1);
  ASSERT_EQ(scene->NumAnimations(), 1);
  ASSERT_EQ(scene->NumSkins(), 1);
  for (AnimationIndex i(0); i < scene->NumAnimations(); ++i) {
    const Animation *const animation = scene->GetAnimation(i);
    ASSERT_NE(animation, nullptr);
    ASSERT_EQ(animation->NumSamplers(), 57);
    ASSERT_EQ(animation->NumChannels(), 57);
  }

  // Check all of the meshes do not have any materials.
  for (MeshIndex i(0); i < scene->NumMeshes(); ++i) {
    const Mesh &mesh = scene->GetMesh(i);
    ASSERT_EQ(mesh.GetMaterialLibrary().NumMaterials(), 0);
  }
}

TEST(GltfDecoderTest, AnimatedBonesGlb) {
  const std::string file_name = "CesiumMan/glTF_Binary/CesiumMan.glb";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));
  ASSERT_EQ(scene->NumMeshes(), 1);
  ASSERT_EQ(scene->NumMeshGroups(), 1);
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->NumMaterialIndices(), 1);
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->GetMaterialIndex(0), 0);
  ASSERT_EQ(scene->NumNodes(), 22);
  ASSERT_EQ(scene->NumRootNodes(), 1);
  ASSERT_EQ(scene->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(scene->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 1);
  ASSERT_EQ(scene->NumAnimations(), 1);
  ASSERT_EQ(scene->NumSkins(), 1);
  for (AnimationIndex i(0); i < scene->NumAnimations(); ++i) {
    const Animation *const animation = scene->GetAnimation(i);
    ASSERT_NE(animation, nullptr);
    ASSERT_EQ(animation->NumSamplers(), 57);
    ASSERT_EQ(animation->NumChannels(), 57);
  }

  // Check all of the meshes do not have any materials.
  for (MeshIndex i(0); i < scene->NumMeshes(); ++i) {
    const Mesh &mesh = scene->GetMesh(i);
    ASSERT_EQ(mesh.GetMaterialLibrary().NumMaterials(), 0);
  }
}

// Tests multiple primitives with the same material index.
TEST(GltfDecoderTest, LanternGltf) {
  const std::string file_name = "Lantern/glTF/Lantern.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));

  EXPECT_EQ(mesh->num_attributes(), 4) << "Unexpected number of attributes.";
  EXPECT_EQ(mesh->num_points(), 4145) << "Unexpected number of vertices.";
  EXPECT_EQ(mesh->num_faces(), 5394) << "Unexpected number of faces.";
  ASSERT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 4);
}

// Tests COLOR_0 input attribute.
TEST(GltfDecoderTest, ColorAttributeGltf) {
  const std::string file_name = "test_pos_color.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  EXPECT_EQ(mesh->num_attributes(), 2) << "Unexpected number of attributes.";
  EXPECT_EQ(mesh->num_points(), 114) << "Unexpected number of vertices.";
  EXPECT_EQ(mesh->num_faces(), 224) << "Unexpected number of faces.";
  ASSERT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 0);
}

// Tests a mesh with two sets of texture coordinates.
TEST(GltfDecoderTest, TwoTexCoordAttributesGltf) {
  const std::string file_name = "sphere_two_tex_coords.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  ASSERT_NE(mesh, nullptr);
  ASSERT_EQ(mesh->NumNamedAttributes(GeometryAttribute::TEX_COORD), 2);
}

// Tests an input with a valid tangent attribute does not auto generate the
// tangent attribute.
TEST(GltfDecoderTest, TestSceneWithTangents) {
  const std::string file_name = "Lantern/glTF/Lantern.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));
  ASSERT_NE(scene, nullptr);

  // Ensure no mesh has auto-generated tangents (and that some meshes have the
  // tangent attribute).
  int num_tangent_attributes = 0;
  for (MeshIndex mi(0); mi < scene->NumMeshes(); ++mi) {
    if (scene->GetMesh(mi).GetNamedAttribute(GeometryAttribute::TANGENT) !=
        nullptr) {
      num_tangent_attributes++;
      ASSERT_FALSE(MeshUtils::HasAutoGeneratedTangents(scene->GetMesh(mi)));
    }
  }
  ASSERT_GT(num_tangent_attributes, 0);
}

// Tests an input file where multiple textures share the same image asset.
TEST(GltfDecoderTest, SharedImages) {
  const std::string file_name = "SphereAllSame/sphere_texture_all.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  ASSERT_NE(mesh, nullptr);
  ASSERT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 5);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetTextureLibrary().NumTextures(), 4);
}

TEST(GltfDecoderTest, TextureNamesAreNotEmpty) {
  const std::string file_name = "SphereAllSame/sphere_texture_all.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  ASSERT_NE(mesh, nullptr);
  ASSERT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 5);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetTextureLibrary().NumTextures(), 4);
  const std::vector<const draco::Texture *> textures = {
      mesh->GetMaterialLibrary().GetTextureLibrary().GetTexture(0),
      mesh->GetMaterialLibrary().GetTextureLibrary().GetTexture(1),
      mesh->GetMaterialLibrary().GetTextureLibrary().GetTexture(2),
      mesh->GetMaterialLibrary().GetTextureLibrary().GetTexture(3)};
  EXPECT_EQ(TextureUtils::GetTargetStem(*textures[0]), "256x256_all_orange");
  EXPECT_EQ(TextureUtils::GetTargetStem(*textures[1]), "256x256_all_blue");
  EXPECT_EQ(TextureUtils::GetTargetStem(*textures[2]), "256x256_all_red");
  EXPECT_EQ(TextureUtils::GetTargetStem(*textures[3]), "256x256_all_green");
  EXPECT_EQ(TextureUtils::GetTargetFormat(*textures[0]), ImageFormat::PNG);
  EXPECT_EQ(TextureUtils::GetTargetFormat(*textures[1]), ImageFormat::PNG);
  EXPECT_EQ(TextureUtils::GetTargetFormat(*textures[2]), ImageFormat::PNG);
  EXPECT_EQ(TextureUtils::GetTargetFormat(*textures[3]), ImageFormat::PNG);
}

TEST(GltfDecoderTest, TestTexCoord1) {
  const std::string file_name = "MultiUVTest/glTF/MultiUVTest.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  ASSERT_NE(mesh, nullptr);
  ASSERT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 2);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetTextureLibrary().NumTextures(), 2);
  const std::vector<const draco::Texture *> textures = {
      mesh->GetMaterialLibrary().GetTextureLibrary().GetTexture(0),
      mesh->GetMaterialLibrary().GetTextureLibrary().GetTexture(1)};
  EXPECT_EQ(TextureUtils::GetTargetStem(*textures[0]), "uv0");
  EXPECT_EQ(TextureUtils::GetTargetStem(*textures[1]), "uv1");
  EXPECT_EQ(TextureUtils::GetTargetFormat(*textures[0]), ImageFormat::PNG);
  EXPECT_EQ(TextureUtils::GetTargetFormat(*textures[1]), ImageFormat::PNG);
  ASSERT_EQ(mesh->NumNamedAttributes(GeometryAttribute::TEX_COORD), 2);
  ASSERT_EQ(mesh->NumNamedAttributes(GeometryAttribute::POSITION), 1);
  ASSERT_EQ(mesh->NumNamedAttributes(GeometryAttribute::NORMAL), 1);
  ASSERT_EQ(mesh->NumNamedAttributes(GeometryAttribute::TANGENT), 1);
}

TEST(GltfDecoderTest, SimpleScene) {
  const std::string file_name = "Box/glTF/Box.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));

  ASSERT_EQ(scene->NumMeshes(), 1);
  ASSERT_EQ(scene->NumMeshGroups(), 1);
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->NumMaterialIndices(), 1);
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->GetMaterialIndex(0), 0);
  ASSERT_EQ(scene->NumNodes(), 2);
  ASSERT_EQ(scene->NumRootNodes(), 1);
  ASSERT_EQ(scene->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(scene->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 0);
  ASSERT_EQ(scene->NumSkins(), 0);
  ASSERT_EQ(scene->NumAnimations(), 0);

  // Check all of the meshes do not have any materials.
  for (MeshIndex i(0); i < scene->NumMeshes(); ++i) {
    const Mesh &mesh = scene->GetMesh(i);
    ASSERT_EQ(mesh.GetMaterialLibrary().NumMaterials(), 0);
  }

  // Check names of nodes are empty.
  EXPECT_TRUE(scene->GetNode(SceneNodeIndex(0))->GetName().empty());
  EXPECT_TRUE(scene->GetNode(SceneNodeIndex(1))->GetName().empty());
}

TEST(GltfDecoderTest, LanternScene) {
  const std::string file_name = "Lantern/glTF/Lantern.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));

  EXPECT_EQ(scene->NumMeshes(), 3);
  EXPECT_EQ(scene->NumMeshGroups(), 3);
  EXPECT_EQ(scene->NumNodes(), 4);
  EXPECT_EQ(scene->NumRootNodes(), 1);
  EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(), 1);
  EXPECT_EQ(scene->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 4);
  EXPECT_EQ(scene->GetMaterialLibrary().GetMaterial(0)->GetDoubleSided(),
            false);
  EXPECT_EQ(scene->NumSkins(), 0);
  EXPECT_EQ(scene->NumAnimations(), 0);

  // Check names of nodes have been populated.
  EXPECT_EQ(scene->GetNode(SceneNodeIndex(0))->GetName(), "Lantern");
  EXPECT_EQ(scene->GetNode(SceneNodeIndex(1))->GetName(), "LanternPole_Body");
  EXPECT_EQ(scene->GetNode(SceneNodeIndex(2))->GetName(), "LanternPole_Chain");
  EXPECT_EQ(scene->GetNode(SceneNodeIndex(3))->GetName(),
            "LanternPole_Lantern");
}

TEST(GltfDecoderTest, SimpleTriangleMesh) {
  const std::string file_name = "Triangle/glTF/Triangle.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));

  EXPECT_EQ(mesh->num_attributes(), 1) << "Unexpected number of attributes.";
  EXPECT_EQ(mesh->num_points(), 3) << "Unexpected number of vertices.";
  EXPECT_EQ(mesh->num_faces(), 1) << "Unexpected number of faces.";
  EXPECT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 0);
}

TEST(GltfDecoderTest, SimpleTriangleScene) {
  const std::string file_name = "Triangle/glTF/Triangle.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));

  EXPECT_EQ(scene->NumMeshes(), 1);
  EXPECT_EQ(scene->NumMeshGroups(), 1);
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->NumMaterialIndices(), 1);
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->GetMaterialIndex(0), 0);
  EXPECT_EQ(scene->NumNodes(), 1);
  EXPECT_EQ(scene->NumRootNodes(), 1);
  EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(), 1);
  EXPECT_EQ(scene->NumSkins(), 0);
  EXPECT_EQ(scene->NumAnimations(), 0);
}

TEST(GltfDecoderTest, ThreeMeshesOneNoMaterialScene) {
  const std::string file_name =
      "three_meshes_two_materials_one_no_material/"
      "three_meshes_two_materials_one_no_material.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));

  EXPECT_EQ(scene->NumMeshes(), 3);
  EXPECT_EQ(scene->NumMeshGroups(), 3);
  EXPECT_EQ(scene->NumNodes(), 4);
  EXPECT_EQ(scene->NumRootNodes(), 1);
  EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(), 3);
  EXPECT_EQ(scene->NumSkins(), 0);
  EXPECT_EQ(scene->NumAnimations(), 0);
}

TEST(GltfDecoderTest, ThreeMeshesOneNoMaterialMesh) {
  const std::string file_name =
      "three_meshes_two_materials_one_no_material/"
      "three_meshes_two_materials_one_no_material.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));

  EXPECT_EQ(mesh->num_attributes(), 4) << "Unexpected number of attributes.";
  EXPECT_EQ(mesh->num_points(), 72) << "Unexpected number of vertices.";
  EXPECT_EQ(mesh->num_faces(), 36) << "Unexpected number of faces.";
  EXPECT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 3);
}

TEST(GltfDecoderTest, DoubleSidedMaterial) {
  const std::string file_name = "TwoSidedPlane/glTF/TwoSidedPlane.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  EXPECT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 1);
  EXPECT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->GetDoubleSided(), true);

  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));
  EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(), 1);
  EXPECT_EQ(scene->GetMaterialLibrary().GetMaterial(0)->GetDoubleSided(), true);
}

TEST(GltfDecoderTest, VertexColorTest) {
  const std::string file_name = "VertexColorTest/glTF/VertexColorTest.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  EXPECT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 2);
  EXPECT_EQ(mesh->NumNamedAttributes(GeometryAttribute::COLOR), 1);

  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));
  EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(), 2);
  EXPECT_EQ(scene->NumMeshes(), 2);
  const Mesh &second_mesh = scene->GetMesh(MeshIndex(1));
  EXPECT_EQ(second_mesh.NumNamedAttributes(GeometryAttribute::COLOR), 1);
}

TEST(GltfDecoderTest, MorphTargets) {
  const std::string filename =
      "KhronosSampleModels/AnimatedMorphCube/glTF/AnimatedMorphCube.gltf";
  const std::string path = GetTestFileFullPath(filename);
  GltfDecoder decoder;
  const auto maybe_scene = decoder.DecodeFromFileToScene(path);
  EXPECT_FALSE(maybe_scene.ok());
  EXPECT_EQ(maybe_scene.status().code(), Status::Code::UNSUPPORTED_FEATURE);
}

TEST(GltfDecoderTest, SparseAccessors) {
  const std::string filename =
      "KhronosSampleModels/SimpleSparseAccessor/glTF/SimpleSparseAccessor.gltf";
  const std::string path = GetTestFileFullPath(filename);
  GltfDecoder decoder;
  const auto maybe_scene = decoder.DecodeFromFileToScene(path);
  EXPECT_FALSE(maybe_scene.ok());
  EXPECT_EQ(maybe_scene.status().code(), Status::Code::UNSUPPORTED_FEATURE);
}

TEST(GltfDecoderTest, PbrSpecularGlossinessExtension) {
  const std::string filename =
      "KhronosSampleModels/SpecGlossVsMetalRough/glTF/"
      "SpecGlossVsMetalRough.gltf";
  const std::string path = GetTestFileFullPath(filename);
  GltfDecoder decoder;
  const auto maybe_scene = decoder.DecodeFromFileToScene(path);
  EXPECT_FALSE(maybe_scene.ok());
  EXPECT_EQ(maybe_scene.status().code(), Status::Code::UNSUPPORTED_FEATURE);
}

TEST(GltfDecoderTest, DifferentWrappingModes) {
  const std::string filename =
      "KhronosSampleModels/TextureSettingsTest/glTF/TextureSettingsTest.gltf";
  const std::string path = GetTestFileFullPath(filename);
  GltfDecoder decoder;
  const auto maybe_scene = decoder.DecodeFromFileToScene(path);
  EXPECT_TRUE(maybe_scene.ok());
  const draco::Scene &scene = *maybe_scene.value();
  ASSERT_EQ(scene.GetMaterialLibrary().GetTextureLibrary().NumTextures(), 3);
  ASSERT_EQ(scene.GetMaterialLibrary().NumMaterials(), 10);
  const draco::Material &material = *scene.GetMaterialLibrary().GetMaterial(0);
  ASSERT_EQ(material.NumTextureMaps(), 1);
  ASSERT_EQ(material.GetTextureMapByIndex(0)->wrapping_mode().s,
            draco::TextureMap::REPEAT);
  ASSERT_EQ(material.GetTextureMapByIndex(0)->wrapping_mode().t,
            draco::TextureMap::MIRRORED_REPEAT);
}

TEST(GltfDecoderTest, KhrMaterialsUnlitExtension) {
  const std::string no_unlit_filename = "Box/glTF/Box.gltf";
  const std::unique_ptr<Scene> scene_no_unlit(
      DecodeGltfFileToScene(no_unlit_filename));
  EXPECT_EQ(scene_no_unlit->GetMaterialLibrary().NumMaterials(), 1);
  EXPECT_EQ(scene_no_unlit->GetMaterialLibrary().GetMaterial(0)->GetUnlit(),
            false);

  const std::string filename =
      "KhronosSampleModels/UnlitTest/glTF/UnlitTest.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(filename));
  EXPECT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 2);
  EXPECT_EQ(mesh->GetMaterialLibrary().GetMaterial(0)->GetUnlit(), true);
  EXPECT_EQ(mesh->GetMaterialLibrary().GetMaterial(1)->GetUnlit(), true);

  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(filename));
  EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(), 2);
  EXPECT_EQ(scene->GetMaterialLibrary().GetMaterial(0)->GetUnlit(), true);
  EXPECT_EQ(scene->GetMaterialLibrary().GetMaterial(1)->GetUnlit(), true);
}

TEST(GltfDecoderTest, KhrMaterialsSheenExtension) {
  // Check that a model with no sheen is loaded with no sheen.
  {
    const std::unique_ptr<Scene> scene(
        DecodeGltfFileToScene("Box/glTF/Box.gltf"));
    EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(), 1);

    // Check that material has no sheen.
    const Material &material = *scene->GetMaterialLibrary().GetMaterial(0);
    EXPECT_FALSE(material.HasSheen());

    // Check that sheen color and roughness factors have default values.
    EXPECT_EQ(material.GetSheenColorFactor(), Vector3f(0.f, 0.f, 0.f));
    EXPECT_EQ(material.GetSheenRoughnessFactor(), 0.f);

    // Check that sheen textures are absent.
    EXPECT_EQ(material.GetTextureMapByType(TextureMap::SHEEN_COLOR), nullptr);
    EXPECT_EQ(material.GetTextureMapByType(TextureMap::SHEEN_ROUGHNESS),
              nullptr);
  }

  // Check that a model with sheen is loaded as a mesh with sheen.
  {
    // Load model as a mesh.
    const std::unique_ptr<Mesh> mesh(
        DecodeGltfFile("KhronosSampleModels/SheenCloth/glTF/SheenCloth.gltf"));
    EXPECT_NE(mesh, nullptr);
    const Material &material = *mesh->GetMaterialLibrary().GetMaterial(0);

    // Check that material has sheen.
    EXPECT_TRUE(material.HasSheen());

    // Check that sheen color and roughness factors are present.
    EXPECT_EQ(material.GetSheenColorFactor(), Vector3f(1.f, 1.f, 1.f));
    EXPECT_EQ(material.GetSheenRoughnessFactor(), 1.f);

    // Check that sheen color and roughness textures are present.
    EXPECT_NE(material.GetTextureMapByType(TextureMap::SHEEN_COLOR), nullptr);
    EXPECT_NE(material.GetTextureMapByType(TextureMap::SHEEN_ROUGHNESS),
              nullptr);

    // Check that sheen color and roughness textures are shared.
    EXPECT_EQ(
        material.GetTextureMapByType(TextureMap::SHEEN_COLOR)->texture(),
        material.GetTextureMapByType(TextureMap::SHEEN_ROUGHNESS)->texture());
  }

  // Check that a model with sheen is loaded as a scene with sheen.
  {
    // Load model as a scene.
    const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(
        "KhronosSampleModels/SheenCloth/glTF/SheenCloth.gltf"));
    EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(), 1);
    const Material &material = *scene->GetMaterialLibrary().GetMaterial(0);

    // Check that material has sheen.
    EXPECT_TRUE(material.HasSheen());

    // Check that sheen color and roughness factors are present.
    EXPECT_EQ(material.GetSheenColorFactor(), Vector3f(1.f, 1.f, 1.f));
    EXPECT_EQ(material.GetSheenRoughnessFactor(), 1.f);

    // Check that sheen color and roughness textures are present.
    EXPECT_NE(material.GetTextureMapByType(TextureMap::SHEEN_COLOR), nullptr);
    EXPECT_NE(material.GetTextureMapByType(TextureMap::SHEEN_ROUGHNESS),
              nullptr);

    // Check that sheen color and roughness textures are shared.
    EXPECT_EQ(
        material.GetTextureMapByType(TextureMap::SHEEN_COLOR)->texture(),
        material.GetTextureMapByType(TextureMap::SHEEN_ROUGHNESS)->texture());
  }
}

TEST(GltfDecoderTest, PbrNextExtensions) {
  // Check that a model with no material extensions is loaded correctly.
  {
    const std::unique_ptr<Scene> scene(
        DecodeGltfFileToScene("Box/glTF/Box.gltf"));
    EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(), 1);
    const Material &m = *scene->GetMaterialLibrary().GetMaterial(0);

    // Check that material has no extensions.
    EXPECT_FALSE(m.HasSheen());
    EXPECT_FALSE(m.HasTransmission());
    EXPECT_FALSE(m.HasClearcoat());
    EXPECT_FALSE(m.HasVolume());
    EXPECT_FALSE(m.HasIor());
    EXPECT_FALSE(m.HasSpecular());
  }

  // Check that a model with material extensions is loaded correctly.
  {
    const std::unique_ptr<Mesh> mesh(
        DecodeGltfFile("pbr_next/sphere/glTF/sphere.gltf"));
    EXPECT_NE(mesh, nullptr);
    const Material &m = *mesh->GetMaterialLibrary().GetMaterial(0);

    // Check that material has extensions.
    EXPECT_TRUE(m.HasSheen());
    EXPECT_TRUE(m.HasTransmission());
    EXPECT_TRUE(m.HasClearcoat());
    EXPECT_TRUE(m.HasVolume());
    EXPECT_TRUE(m.HasIor());
    EXPECT_TRUE(m.HasSpecular());

    // Check that material has correct extension properties.
    EXPECT_EQ(m.GetSheenColorFactor(), Vector3f(1.0f, 0.329f, 0.1f));
    EXPECT_EQ(m.GetSheenRoughnessFactor(), 0.8f);
    EXPECT_EQ(m.GetTransmissionFactor(), 0.75f);
    EXPECT_EQ(m.GetClearcoatFactor(), 0.95f);
    EXPECT_EQ(m.GetClearcoatRoughnessFactor(), 0.03f);
    EXPECT_EQ(m.GetAttenuationColor(), Vector3f(0.921f, 0.640f, 0.064f));
    EXPECT_EQ(m.GetAttenuationDistance(), 0.155f);
    EXPECT_EQ(m.GetThicknessFactor(), 2.27f);
    EXPECT_EQ(m.GetIor(), 1.55f);
    EXPECT_EQ(m.GetSpecularFactor(), 0.3f);
    EXPECT_EQ(m.GetSpecularColorFactor(), Vector3f(0.212f, 0.521f, 0.051f));

    // Check that material has all extension textures.
    EXPECT_NE(m.GetTextureMapByType(TextureMap::SHEEN_COLOR), nullptr);
    EXPECT_NE(m.GetTextureMapByType(TextureMap::SHEEN_ROUGHNESS), nullptr);
    EXPECT_NE(m.GetTextureMapByType(TextureMap::TRANSMISSION), nullptr);
    EXPECT_NE(m.GetTextureMapByType(TextureMap::CLEARCOAT), nullptr);
    EXPECT_NE(m.GetTextureMapByType(TextureMap::CLEARCOAT_ROUGHNESS), nullptr);
    EXPECT_NE(m.GetTextureMapByType(TextureMap::CLEARCOAT_NORMAL), nullptr);
    EXPECT_NE(m.GetTextureMapByType(TextureMap::THICKNESS), nullptr);
    EXPECT_NE(m.GetTextureMapByType(TextureMap::SPECULAR), nullptr);
    EXPECT_NE(m.GetTextureMapByType(TextureMap::SPECULAR_COLOR), nullptr);
  }
}

TEST(GltfDecoderTest, TextureTransformTest) {
  const std::string filename =
      "KhronosSampleModels/TextureTransformTest/glTF/TextureTransformTest.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(filename));
  EXPECT_EQ(mesh->GetMaterialLibrary().NumMaterials(), 9);
  for (int i = 0; i < 6; ++i) {
    EXPECT_FALSE(TextureTransform::IsDefault(mesh->GetMaterialLibrary()
                                                 .GetMaterial(i)
                                                 ->GetTextureMapByIndex(0)
                                                 ->texture_transform()));
  }
  for (int i = 6; i < 9; ++i) {
    EXPECT_TRUE(TextureTransform::IsDefault(mesh->GetMaterialLibrary()
                                                .GetMaterial(i)
                                                ->GetTextureMapByIndex(0)
                                                ->texture_transform()));
  }

  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(filename));
  EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(), 9);
  for (int i = 0; i < 6; ++i) {
    EXPECT_FALSE(TextureTransform::IsDefault(scene->GetMaterialLibrary()
                                                 .GetMaterial(i)
                                                 ->GetTextureMapByIndex(0)
                                                 ->texture_transform()));
  }
  for (int i = 6; i < 9; ++i) {
    EXPECT_TRUE(TextureTransform::IsDefault(scene->GetMaterialLibrary()
                                                .GetMaterial(i)
                                                ->GetTextureMapByIndex(0)
                                                ->texture_transform()));
  }
}

TEST(GltfDecoderTest, GlbTextureSource) {
  const std::string file_name = "KhronosSampleModels/Duck/glTF_Binary/Duck.glb";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));
  EXPECT_EQ(scene->NumMeshes(), 1);
  EXPECT_EQ(scene->NumMeshGroups(), 1);
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->NumMaterialIndices(), 1);
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->GetMaterialIndex(0), 0);
  EXPECT_EQ(scene->NumNodes(), 3);
  EXPECT_EQ(scene->NumRootNodes(), 1);
  EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(), 1);
  EXPECT_EQ(scene->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 1);
  EXPECT_EQ(scene->NumAnimations(), 0);
  EXPECT_EQ(scene->NumSkins(), 0);
  EXPECT_EQ(scene->GetMaterialLibrary().GetTextureLibrary().NumTextures(), 1);
  const Texture *const texture =
      scene->GetMaterialLibrary().GetTextureLibrary().GetTexture(0);
  ASSERT_NE(texture, nullptr);
  const SourceImage &source_image = texture->source_image();
  EXPECT_EQ(source_image.encoded_data().size(), 16302);
  EXPECT_EQ(source_image.filename(), "");
  EXPECT_EQ(source_image.mime_type(), "image/png");
}

TEST(GltfDecoderTest, GltfTextureSource) {
  const std::string file_name = "KhronosSampleModels/Duck/glTF/Duck.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));
  EXPECT_EQ(scene->NumMeshes(), 1);
  EXPECT_EQ(scene->NumMeshGroups(), 1);
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->NumMaterialIndices(), 1);
  ASSERT_EQ(scene->GetMeshGroup(MeshGroupIndex(0))->GetMaterialIndex(0), 0);
  EXPECT_EQ(scene->NumNodes(), 3);
  EXPECT_EQ(scene->NumRootNodes(), 1);
  EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(), 1);
  EXPECT_EQ(scene->GetMaterialLibrary().GetMaterial(0)->NumTextureMaps(), 1);
  EXPECT_EQ(scene->NumAnimations(), 0);
  EXPECT_EQ(scene->NumSkins(), 0);
  EXPECT_EQ(scene->GetMaterialLibrary().GetTextureLibrary().NumTextures(), 1);
  const Texture *const texture =
      scene->GetMaterialLibrary().GetTextureLibrary().GetTexture(0);
  ASSERT_NE(texture, nullptr);
  const SourceImage &source_image = texture->source_image();
  EXPECT_EQ(source_image.encoded_data().size(), 0);
  EXPECT_FALSE(source_image.filename().empty());
  EXPECT_EQ(source_image.mime_type(), "");
}

TEST(GltfDecoderTest, GltfDecodeWithDraco) {
  // Tests that we can decode a glTF containing Draco compressed geometry.
  const std::string file_name = "Box/glTF_Binary/Box.glb";
  const std::string file_name_with_draco = "Box/glTF_Binary/Box_Draco.glb";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));
  const std::unique_ptr<Scene> scene_draco(
      DecodeGltfFileToScene(file_name_with_draco));
  ASSERT_NE(scene, nullptr);
  ASSERT_NE(scene_draco, nullptr);
  EXPECT_EQ(scene->NumMeshes(), scene_draco->NumMeshes());
  EXPECT_EQ(scene->NumMeshGroups(), scene_draco->NumMeshGroups());
  EXPECT_EQ(scene->NumNodes(), scene_draco->NumNodes());
  EXPECT_EQ(scene->NumRootNodes(), scene_draco->NumRootNodes());
  EXPECT_EQ(scene->GetMaterialLibrary().NumMaterials(),
            scene_draco->GetMaterialLibrary().NumMaterials());
  EXPECT_EQ(scene->NumAnimations(), scene_draco->NumAnimations());
  EXPECT_EQ(scene->NumSkins(), scene_draco->NumSkins());

  EXPECT_EQ(scene->NumMeshes(), 1);
  EXPECT_EQ(scene->GetMesh(draco::MeshIndex(0)).num_faces(),
            scene_draco->GetMesh(draco::MeshIndex(0)).num_faces());
}

TEST(GltfDecoderTest, TestAnimationNames) {
  const std::string file_name = "InterpolationTest/glTF/InterpolationTest.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));
  ASSERT_NE(scene, nullptr);

  EXPECT_EQ(scene->NumAnimations(), 9);

  const std::vector<std::string> animation_names{
      "Step Scale",           "Linear Scale",
      "CubicSpline Scale",    "Step Rotation",
      "CubicSpline Rotation", "Linear Rotation",
      "Step Translation",     "CubicSpline Translation",
      "Linear Translation"};
  for (int i = 0; i < scene->NumAnimations(); ++i) {
    const Animation *const anim = scene->GetAnimation(AnimationIndex(i));
    ASSERT_NE(anim, nullptr);
    ASSERT_EQ(anim->GetName(), animation_names[i]);
  }
}

TEST(GltfDecoderTest, DuplicatePrimitives) {
  const std::string file_name = "DuplicateMeshes/duplicate_meshes.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));
  ASSERT_NE(scene, nullptr);

  // There should be only one unique base mesh in the scene and four mesh
  // groups (instances).
  ASSERT_EQ(scene->NumMeshes(), 1);
  ASSERT_EQ(scene->NumMeshGroups(), 4);

  // There should be two materials used by the instances.
  ASSERT_EQ(scene->GetMaterialLibrary().NumMaterials(), 2);
}

TEST(GltfDecoderTest, SimpleSkin) {
  // This is a simple skin example from glTF tutorial.
  const std::string file_name = "simple_skin.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));
  ASSERT_NE(scene, nullptr);

  // Check scene size.
  ASSERT_EQ(scene->NumMeshes(), 1);
  ASSERT_EQ(scene->NumMeshGroups(), 1);
  ASSERT_EQ(scene->GetMeshGroup(draco::MeshGroupIndex(0))->NumMaterialIndices(),
            1);
  ASSERT_EQ(scene->NumNodes(), 3);
  ASSERT_EQ(scene->NumRootNodes(), 1);
  ASSERT_EQ(scene->GetMaterialLibrary().NumMaterials(), 1);
  ASSERT_EQ(scene->NumAnimations(), 1);
  ASSERT_EQ(scene->NumSkins(), 1);

  // Check animation size.
  const Animation *const animation = scene->GetAnimation(AnimationIndex(0));
  ASSERT_NE(animation, nullptr);
  ASSERT_EQ(animation->NumSamplers(), 1);
  ASSERT_EQ(animation->NumChannels(), 1);
  ASSERT_EQ(animation->NumNodeAnimationData(), 2);

  // Check animation sampler.
  const AnimationSampler *const sampler = animation->GetSampler(0);
  ASSERT_NE(sampler, nullptr);
  ASSERT_EQ(sampler->input_index, 0);
  ASSERT_EQ(sampler->interpolation_type,
            AnimationSampler::SamplerInterpolation::LINEAR);
  ASSERT_EQ(sampler->output_index, 1);

  // Check animation channel.
  const AnimationChannel *const channel = animation->GetChannel(0);
  ASSERT_NE(channel, nullptr);
  ASSERT_EQ(channel->sampler_index, 0);
  ASSERT_EQ(channel->target_index, 2);
  ASSERT_EQ(channel->transformation_type,
            AnimationChannel::ChannelTransformation::ROTATION);

  // Check the first node animation data.
  {
    const NodeAnimationData *const node_animation =
        animation->GetNodeAnimationData(0);
    ASSERT_EQ(node_animation->ComponentSize(), 4);
    ASSERT_EQ(node_animation->NumComponents(), 1);
    ASSERT_EQ(node_animation->count(), 12);
    ASSERT_EQ(node_animation->type(), NodeAnimationData::Type::SCALAR);
    ASSERT_FALSE(node_animation->normalized());
    const std::vector<float> &node_animation_data = *node_animation->GetData();
    const std::vector<float> expected_node_animation_data{
        0.0f, 0.5f, 1.0f, 1.5f, 2.0f, 2.5f, 3.0f, 3.5f, 4.0f, 4.5f, 5.0f, 5.5f};
    ASSERT_EQ(node_animation_data, expected_node_animation_data);
  }

  // Check the second node animation data.
  {
    const NodeAnimationData *const node_animation =
        animation->GetNodeAnimationData(1);
    ASSERT_EQ(node_animation->ComponentSize(), 4);
    ASSERT_EQ(node_animation->NumComponents(), 4);
    ASSERT_EQ(node_animation->count(), 12);
    ASSERT_EQ(node_animation->type(), NodeAnimationData::Type::VEC4);
    ASSERT_FALSE(node_animation->normalized());
    const std::vector<float> &node_animation_data = *node_animation->GetData();
    std::cout << std::endl;
    // clang-format off
    const std::vector<float> expected_node_animation_data{
      0.000f,  0.000f,  0.000f,  1.000f,
      0.000f,  0.000f,  0.383f,  0.924f,
      0.000f,  0.000f,  0.707f,  0.707f,
      0.000f,  0.000f,  0.707f,  0.707f,
      0.000f,  0.000f,  0.383f,  0.924f,
      0.000f,  0.000f,  0.000f,  1.000f,
      0.000f,  0.000f,  0.000f,  1.000f,
      0.000f,  0.000f, -0.383f,  0.924f,
      0.000f,  0.000f, -0.707f,  0.707f,
      0.000f,  0.000f, -0.707f,  0.707f,
      0.000f,  0.000f, -0.383f,  0.924f,
      0.000f,  0.000f,  0.000f,  1.000f};
    // clang-format on
    ASSERT_EQ(node_animation_data, expected_node_animation_data);
  }

  // Check skin.
  const Skin *const skin = scene->GetSkin(SkinIndex(0));
  ASSERT_NE(skin, nullptr);
  ASSERT_EQ(skin->NumJoints(), 2);
  ASSERT_EQ(skin->GetJointRoot(), kInvalidSceneNodeIndex);
  ASSERT_EQ(skin->GetJoint(0), SceneNodeIndex(1));
  ASSERT_EQ(skin->GetJoint(1), SceneNodeIndex(2));

  // Check inverse bind matrices.
  const NodeAnimationData &bind_matrices = skin->GetInverseBindMatrices();
  ASSERT_EQ(bind_matrices.type(), NodeAnimationData::Type::MAT4);
  ASSERT_EQ(bind_matrices.count(), 2);
  ASSERT_EQ(bind_matrices.normalized(), false);
  ASSERT_NE(bind_matrices.GetData(), nullptr);
  const std::vector<float> &bind_matrices_data = *bind_matrices.GetData();
  // clang-format off
  const std::vector<float> expected_bind_matrices_data{
      // First matrix.
      1.0f,  0.0f,  0.0f,  0.0f,
      0.0f,  1.0f,  0.0f,  0.0f,
      0.0f,  0.0f,  1.0f,  0.0f,
     -0.5f, -1.0f,  0.0f,  1.0f,
      // Second matrix.
      1.0f,  0.0f,  0.0f,  0.0f,
      0.0f,  1.0f,  0.0f,  0.0f,
      0.0f,  0.0f,  1.0f,  0.0f,
     -0.5f, -1.0f,  0.0f,  1.0f};
  // clang-format on
  ASSERT_EQ(bind_matrices_data, expected_bind_matrices_data);

  // Check mesh size.
  const Mesh &mesh = scene->GetMesh(MeshIndex(0));
  ASSERT_EQ(mesh.num_faces(), 8);
  ASSERT_EQ(mesh.num_points(), 10);
  ASSERT_EQ(mesh.num_attributes(), 3);

  // Check vertex joint indices.
  const PointAttribute *const joints_att =
      mesh.GetNamedAttribute(GeometryAttribute::JOINTS);
  ASSERT_NE(joints_att, nullptr);
  ASSERT_EQ(joints_att->data_type(), DT_UINT16);
  ASSERT_EQ(joints_att->num_components(), 4);
  ASSERT_EQ(joints_att->size(), 1);
  // clang-format off
  const std::array<uint16_t, 40> expected_joints = {
      // Each vertex is associated with four joints.
      0, 1, 0, 0,
      0, 1, 0, 0,
      0, 1, 0, 0,
      0, 1, 0, 0,
      0, 1, 0, 0,
      0, 1, 0, 0,
      0, 1, 0, 0,
      0, 1, 0, 0,
      0, 1, 0, 0,
      0, 1, 0, 0 };
  // clang-format on
  std::array<uint16_t, 40> joints;
  for (draco::PointIndex pi(0); pi < mesh.num_points(); ++pi) {
    joints_att->GetMappedValue(pi, &joints[4 * pi.value()]);
  }
  ASSERT_EQ(joints, expected_joints);

  // Check vertex joint weights.
  const PointAttribute *const weights_att =
      mesh.GetNamedAttribute(GeometryAttribute::WEIGHTS);
  ASSERT_NE(weights_att, nullptr);
  ASSERT_EQ(weights_att->data_type(), DT_FLOAT32);
  ASSERT_EQ(weights_att->num_components(), 4);
  ASSERT_EQ(weights_att->size(), 5);
  // clang-format off
  const std::array<float, 40> expected_weights = {
      // Each vertex has four joint weights.
      1.00f, 0.00f, 0.00f, 0.00f,
      1.00f, 0.00f, 0.00f, 0.00f,
      0.75f, 0.25f, 0.00f, 0.00f,
      0.75f, 0.25f, 0.00f, 0.00f,
      0.50f, 0.50f, 0.00f, 0.00f,
      0.50f, 0.50f, 0.00f, 0.00f,
      0.25f, 0.75f, 0.00f, 0.00f,
      0.25f, 0.75f, 0.00f, 0.00f,
      0.00f, 1.00f, 0.00f, 0.00f,
      0.00f, 1.00f, 0.00f, 0.00f };
  // clang-format on
  std::array<float, 40> weights;
  for (draco::PointIndex pi(0); pi < mesh.num_points(); ++pi) {
    weights_att->GetMappedValue(pi, &weights[4 * pi.value()]);
  }
  ASSERT_EQ(weights, expected_weights);
}

TEST(GltfDecoderTest, DecodeMeshWithImplicitPrimitiveIndices) {
  // Check that glTF primitives with implicit indices can be loaded as a mesh.
  const std::string file_name = "Fox/glTF/Fox.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  ASSERT_NE(mesh, nullptr);
  ASSERT_EQ(mesh->num_faces(), 576);
}

TEST(GltfDecoderTest, DecodeSceneWithImplicitPrimitiveIndices) {
  // Check that glTF primitives with implicit indices can be loaded as a scene.
  const std::string file_name = "Fox/glTF/Fox.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));
  ASSERT_NE(scene, nullptr);
  ASSERT_EQ(scene->NumMeshes(), 1);
  ASSERT_EQ(scene->GetMesh(MeshIndex(0)).num_faces(), 576);
}

TEST(GltfDecoderTest, DecodeFromBufferToMesh) {
  // Checks that a mesh can be decoded from buffer in GLB format.
  // Read GLB file contents into a buffer.
  const std::string file_name = "KhronosSampleModels/Duck/glTF_Binary/Duck.glb";
  const std::string file_path = GetTestFileFullPath(file_name);
  std::vector<char> file_data;
  ASSERT_TRUE(ReadFileToBuffer(file_path, &file_data));
  DecoderBuffer buffer;
  buffer.Init(file_data.data(), file_data.size());

  // Decode mesh from buffer.
  GltfDecoder decoder;
  DRACO_ASSIGN_OR_ASSERT(auto mesh, decoder.DecodeFromBuffer(&buffer));
  ASSERT_NE(mesh, nullptr);

  // Decode mesh from GLB file.
  const std::unique_ptr<Mesh> expected_mesh(DecodeGltfFile(file_name));
  ASSERT_NE(expected_mesh, nullptr);

  // Check that meshes decoded from the buffer and from GLB file are equivalent.
  MeshAreEquivalent eq;
  ASSERT_TRUE(eq(*mesh, *expected_mesh));
}

TEST(GltfDecoderTest, DecodeGraph) {
  // Checks that we can decode a scene with a general graph structure where a
  // node has multiple parents.
  // The input model has one root node, 4 children nodes that all point to a
  // single node that contains the cube mesh.
  const std::string file_name = "CubeScaledInstances/glTF/cube_att.gltf";
  const std::string file_path = GetTestFileFullPath(file_name);

  // First decode the scene into a tree-graph.
  draco::GltfDecoder dec_tree;
  DRACO_ASSIGN_OR_ASSERT(auto scene_tree,
                         dec_tree.DecodeFromFileToScene(file_path));
  // We expect to have 9 nodes with 4 mesh instances. The leaf node with the
  // cube is duplicated 4 times, once for each instance.
  ASSERT_EQ(scene_tree->NumNodes(), 9);
  auto instances_tree = draco::SceneUtils::ComputeAllInstances(*scene_tree);
  ASSERT_EQ(instances_tree.size(), 4);

  // Decode the scene into a scene-graph.
  draco::GltfDecoder dec_graph;
  dec_graph.SetSceneGraphMode(draco::GltfDecoder::GltfSceneGraphMode::DAG);
  DRACO_ASSIGN_OR_ASSERT(auto scene_graph,
                         dec_graph.DecodeFromFileToScene(file_path));

  // We expect to have 6 nodes with 4 mesh instances. The leaf node is shared
  // for all mesh instances.
  ASSERT_EQ(scene_graph->NumNodes(), 6);
  auto instances_graph = draco::SceneUtils::ComputeAllInstances(*scene_graph);
  ASSERT_EQ(instances_graph.size(), 4);

  // Check that all instances share the same scene node.
  for (draco::MeshInstanceIndex mii(1); mii < 4; ++mii) {
    ASSERT_EQ(instances_graph[mii - 1].scene_node_index,
              instances_graph[mii].scene_node_index);
  }
}

TEST(GltfDecoderTest, CorrectVolumeThicknessFactor) {
  // Checks that when a model is decoded as draco::Mesh the PBR material volume
  // thickness factor is corrected according to geometry transformation scale in
  // the scene graph.
  constexpr float kDragonScale = 0.25f;
  constexpr float kDragonVolumeThickness = 2.27f;

  // Read model as draco::Scene and check dragon mesh transformation scale and
  // its PBR material volume thickness factor.
  const std::unique_ptr<draco::Scene> scene = draco::ReadSceneFromTestFile(
      "KhronosSampleModels/DragonAttenuation/glTF/DragonAttenuation.gltf");
  ASSERT_NE(scene, nullptr);
  auto instances = draco::SceneUtils::ComputeAllInstances(*scene);
  ASSERT_EQ(instances.size(), 2);
  ASSERT_EQ(instances[MeshInstanceIndex(0)].transform.col(0).norm(),
            kDragonScale);
  ASSERT_EQ(scene->GetMaterialLibrary().GetMaterial(1)->GetThicknessFactor(),
            kDragonVolumeThickness);

  // Read model as draco::Mesh and check corrected volume thickness factor.
  const std::unique_ptr<draco::Mesh> mesh = draco::ReadMeshFromTestFile(
      "KhronosSampleModels/DragonAttenuation/glTF/DragonAttenuation.gltf");
  ASSERT_NE(mesh, nullptr);
  ASSERT_EQ(mesh->GetMaterialLibrary().GetMaterial(1)->GetThicknessFactor(),
            kDragonScale * kDragonVolumeThickness);
}

TEST(GltfDecoderTest, DecodeLightsIntoMesh) {
  // Checks that a model with lights can be decoded into draco::Mesh with the
  // lights discarded.
  const std::string file_name = "sphere_lights.gltf";
  const std::unique_ptr<Mesh> mesh(DecodeGltfFile(file_name));
  ASSERT_NE(mesh, nullptr);
  ASSERT_EQ(mesh->num_faces(), 224);
}

TEST(GltfDecoderTest, DecodeLightsIntoScene) {
  // Checks that a model with lights can be decoded into draco::Scene.
  const std::string file_name = "sphere_lights.gltf";
  const std::unique_ptr<Scene> scene(DecodeGltfFileToScene(file_name));
  ASSERT_NE(scene, nullptr);
  ASSERT_EQ(scene->NumLights(), 4);

  // Check spot light with all properties specified.
  Light &light = *scene->GetLight(LightIndex(0));
  ASSERT_EQ(light.GetName(), "Blue Lightsaber");
  ASSERT_EQ(light.GetColor(), draco::Vector3f(0.72f, 0.71f, 1.00f));
  ASSERT_EQ(light.GetIntensity(), 3.0);
  ASSERT_EQ(light.GetType(), draco::Light::SPOT);
  ASSERT_EQ(light.GetRange(), 100);
  ASSERT_EQ(light.GetInnerConeAngle(), 0.2);
  ASSERT_EQ(light.GetOuterConeAngle(), 0.8);

  // Check point light with all properties specified.
  light = *scene->GetLight(LightIndex(1));
  ASSERT_EQ(light.GetName(), "The Star of Earendil");
  ASSERT_EQ(light.GetColor(), draco::Vector3f(0.90f, 0.97f, 1.0f));
  ASSERT_EQ(light.GetIntensity(), 5.0);
  ASSERT_EQ(light.GetType(), draco::Light::POINT);
  ASSERT_EQ(light.GetRange(), 1000);
  ASSERT_EQ(light.GetInnerConeAngle(), 0.0);
  ASSERT_NEAR(light.GetOuterConeAngle(), M_PI / 4.0f, 1e-8);

  // Check directional light with some properties specified.
  light = *scene->GetLight(LightIndex(2));
  ASSERT_EQ(light.GetName(), "Arc Reactor");
  ASSERT_EQ(light.GetColor(), draco::Vector3f(0.9f, 0.9, 0.9f));
  ASSERT_EQ(light.GetIntensity(), 1.0);
  ASSERT_EQ(light.GetType(), draco::Light::DIRECTIONAL);
  ASSERT_EQ(light.GetRange(), 200.0);

  // Check spot light with no properties specified.
  light = *scene->GetLight(LightIndex(3));
  ASSERT_EQ(light.GetName(), "");
  ASSERT_EQ(light.GetColor(), draco::Vector3f(1.0f, 1.0f, 1.0f));
  ASSERT_EQ(light.GetIntensity(), 1.0);
  ASSERT_EQ(light.GetType(), draco::Light::SPOT);
  ASSERT_EQ(light.GetRange(), std::numeric_limits<float>::max());
  ASSERT_EQ(light.GetInnerConeAngle(), 0.0);
  ASSERT_NEAR(light.GetOuterConeAngle(), M_PI / 4.0f, 1e-8);

  // Check that lights are referenced by the scene nodes.
  ASSERT_EQ(scene->GetNode(SceneNodeIndex(0))->GetLightIndex(),
            kInvalidLightIndex);
  ASSERT_EQ(scene->GetNode(SceneNodeIndex(1))->GetLightIndex(), LightIndex(0));
  ASSERT_EQ(scene->GetNode(SceneNodeIndex(2))->GetLightIndex(), LightIndex(2));
  ASSERT_EQ(scene->GetNode(SceneNodeIndex(3))->GetLightIndex(), LightIndex(3));
  ASSERT_EQ(scene->GetNode(SceneNodeIndex(4))->GetLightIndex(), LightIndex(1));
}

}  // namespace draco
#endif  // DRACO_TRANSCODER_SUPPORTED
