1 files changed, 106 insertions, 0 deletions
diff --git a/extern/draco/dracoenc/src/draco/io/obj_encoder_test.cc b/extern/draco/dracoenc/src/draco/io/obj_encoder_test.cc
new file mode 100644
index 00000000000..79ea1712806
--- /dev/null
+++ b/extern/draco/dracoenc/src/draco/io/obj_encoder_test.cc
@@ -0,0 +1,106 @@
+// Copyright 2017 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 <fstream>
+#include <sstream>
+
+#include "draco/core/draco_test_base.h"
+#include "draco/core/draco_test_utils.h"
+#include "draco/io/obj_decoder.h"
+#include "draco/io/obj_encoder.h"
+
+namespace draco {
+
+class ObjEncoderTest : public ::testing::Test {
+ protected:
+  void CompareMeshes(const Mesh *mesh0, const Mesh *mesh1) {
+    ASSERT_EQ(mesh0->num_faces(), mesh1->num_faces());
+    ASSERT_EQ(mesh0->num_attributes(), mesh1->num_attributes());
+    for (size_t att_id = 0; att_id < mesh0->num_attributes(); ++att_id) {
+      ASSERT_EQ(mesh0->attribute(att_id)->size(),
+                mesh1->attribute(att_id)->size());
+    }
+  }
+
+  // Encode a mesh using the ObjEncoder and then decode to verify the encoding.
+  std::unique_ptr<Mesh> EncodeAndDecodeMesh(const Mesh *mesh) {
+    EncoderBuffer encoder_buffer;
+    ObjEncoder encoder;
+    if (!encoder.EncodeToBuffer(*mesh, &encoder_buffer))
+      return nullptr;
+
+    DecoderBuffer decoder_buffer;
+    decoder_buffer.Init(encoder_buffer.data(), encoder_buffer.size());
+    std::unique_ptr<Mesh> decoded_mesh(new Mesh());
+    ObjDecoder decoder;
+    decoder.set_use_metadata(true);
+    if (!decoder.DecodeFromBuffer(&decoder_buffer, decoded_mesh.get()).ok())
+      return nullptr;
+    return decoded_mesh;
+  }
+
+  void test_encoding(const std::string &file_name) {
+    const std::unique_ptr<Mesh> mesh(ReadMeshFromTestFile(file_name, true));
+
+    ASSERT_NE(mesh, nullptr) << "Failed to load test model " << file_name;
+    ASSERT_GT(mesh->num_faces(), 0);
+
+    const std::unique_ptr<Mesh> decoded_mesh = EncodeAndDecodeMesh(mesh.get());
+    CompareMeshes(mesh.get(), decoded_mesh.get());
+  }
+};
+
+TEST_F(ObjEncoderTest, HasSubObject) { test_encoding("cube_att_sub_o.obj"); }
+
+TEST_F(ObjEncoderTest, HasMaterial) {
+  const std::unique_ptr<Mesh> mesh0(ReadMeshFromTestFile("mat_test.obj", true));
+  ASSERT_NE(mesh0, nullptr);
+  const std::unique_ptr<Mesh> mesh1 = EncodeAndDecodeMesh(mesh0.get());
+  ASSERT_NE(mesh1, nullptr);
+  ASSERT_EQ(mesh0->num_faces(), mesh1->num_faces());
+  ASSERT_EQ(mesh0->num_attributes(), mesh1->num_attributes());
+  // Position attribute should be the same.
+  ASSERT_EQ(mesh0->attribute(0)->size(), mesh1->attribute(0)->size());
+  // Since |mesh1| is decoded from buffer, it has not material file. So the
+  // size of material attribute is the number of materials used in the obj
+  // file which is 7. The size of material attribute of |mesh0| decoded from
+  // the obj file will be the number of materials defined in the .mtl file.
+  ASSERT_EQ(mesh0->attribute(1)->size(), 29);
+  ASSERT_EQ(mesh1->attribute(1)->size(), 7);
+}
+
+TEST_F(ObjEncoderTest, TestObjEncodingAll) {
+  // Test decoded mesh from encoded obj file stays the same.
+  test_encoding("bunny_norm.obj");
+  test_encoding("cube_att.obj");
+  test_encoding("cube_att_partial.obj");
+  test_encoding("cube_quads.obj");
+  test_encoding("cube_subd.obj");
+  test_encoding("extra_vertex.obj");
+  test_encoding("multiple_isolated_triangles.obj");
+  test_encoding("multiple_tetrahedrons.obj");
+  test_encoding("one_face_123.obj");
+  test_encoding("one_face_312.obj");
+  test_encoding("one_face_321.obj");
+  test_encoding("sphere.obj");
+  test_encoding("test_nm.obj");
+  test_encoding("test_nm_trans.obj");
+  test_encoding("test_sphere.obj");
+  test_encoding("three_faces_123.obj");
+  test_encoding("three_faces_312.obj");
+  test_encoding("two_faces_123.obj");
+  test_encoding("two_faces_312.obj");
+}
+
+}  // namespace draco