1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
// ----------------------------------------------------------------------------
// - Open3D: www.open3d.org -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.open3d.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
#include "open3d/t/io/TriangleMeshIO.h"
#include "open3d/data/Dataset.h"
#include "open3d/io/TriangleMeshIO.h"
#include "open3d/t/geometry/TriangleMesh.h"
#include "open3d/utility/FileSystem.h"
#include "tests/Tests.h"
namespace open3d {
namespace tests {
TEST(TriangleMeshIO, CreateMeshFromFile) {
data::KnotMesh knot_data;
auto mesh = t::io::CreateMeshFromFile(knot_data.GetPath());
EXPECT_EQ(mesh->GetTriangleIndices().GetLength(), 2880);
EXPECT_EQ(mesh->GetVertexPositions().GetLength(), 1440);
}
TEST(TriangleMeshIO, ReadWriteTriangleMeshPLY) {
data::KnotMesh knot_data;
t::geometry::TriangleMesh mesh, mesh_read;
EXPECT_TRUE(t::io::ReadTriangleMesh(knot_data.GetPath(), mesh));
std::string file_name =
utility::filesystem::GetTempDirectoryPath() + "/test_mesh.ply";
EXPECT_TRUE(t::io::WriteTriangleMesh(file_name, mesh));
EXPECT_TRUE(t::io::ReadTriangleMesh(file_name, mesh_read));
EXPECT_TRUE(
mesh.GetTriangleIndices().AllClose(mesh_read.GetTriangleIndices()));
EXPECT_TRUE(
mesh.GetVertexPositions().AllClose(mesh_read.GetVertexPositions()));
}
TEST(TriangleMeshIO, ReadWriteTriangleMeshOBJ) {
auto cube_mesh = t::geometry::TriangleMesh::CreateBox();
const std::string filename =
utility::filesystem::GetTempDirectoryPath() + "/cube.obj";
EXPECT_TRUE(t::io::WriteTriangleMesh(filename, cube_mesh));
t::geometry::TriangleMesh mesh, mesh_read;
EXPECT_TRUE(t::io::ReadTriangleMesh(filename, mesh));
core::Tensor vertices = core::Tensor::Init<float>({{0.0, 1.0, 0.0},
{1.0, 1.0, 1.0},
{1.0, 1.0, 0.0},
{0.0, 1.0, 1.0},
{0.0, 0.0, 0.0},
{0.0, 0.0, 1.0},
{1.0, 0.0, 0.0},
{1.0, 0.0, 1.0}});
EXPECT_TRUE(mesh.GetVertexPositions().AllClose(vertices));
core::Tensor triangles = core::Tensor::Init<int64_t>({{0, 1, 2},
{0, 3, 1},
{4, 5, 0},
{5, 3, 0},
{4, 6, 5},
{6, 7, 5},
{6, 2, 1},
{6, 1, 7},
{5, 7, 1},
{5, 1, 3},
{4, 0, 6},
{6, 0, 2}});
EXPECT_TRUE(mesh.GetTriangleIndices().AllClose(triangles));
}
TEST(TriangleMeshIO, ReadWriteTriangleMeshNPZ) {
auto cube_mesh = t::geometry::TriangleMesh::CreateBox();
const std::string filename =
utility::filesystem::GetTempDirectoryPath() + "/cube.npz";
EXPECT_TRUE(t::io::WriteTriangleMesh(filename, cube_mesh));
t::geometry::TriangleMesh mesh;
EXPECT_TRUE(t::io::ReadTriangleMesh(filename, mesh));
EXPECT_TRUE(
mesh.GetVertexPositions().AllClose(cube_mesh.GetVertexPositions()));
EXPECT_TRUE(
mesh.GetTriangleIndices().AllClose(cube_mesh.GetTriangleIndices()));
}
// TODO: Add tests for triangle_uvs, materials, triangle_material_ids and
// textures once these are supported.
TEST(TriangleMeshIO, TriangleMeshLegecyCompatibility) {
t::geometry::TriangleMesh mesh_tensor, mesh_tensor_read;
geometry::TriangleMesh mesh_legacy, mesh_legacy_read;
data::BunnyMesh bunny_mesh;
EXPECT_TRUE(t::io::ReadTriangleMesh(bunny_mesh.GetPath(), mesh_tensor));
EXPECT_TRUE(io::ReadTriangleMesh(bunny_mesh.GetPath(), mesh_legacy));
EXPECT_EQ(mesh_tensor.GetTriangleIndices().GetLength(),
static_cast<int64_t>(mesh_legacy.triangles_.size()));
EXPECT_EQ(mesh_tensor.GetVertexPositions().GetLength(),
static_cast<int64_t>(mesh_legacy.vertices_.size()));
const std::string tmp_path = utility::filesystem::GetTempDirectoryPath();
std::string file_name_tensor = tmp_path + "/test_mesh_tensor.obj";
std::string file_name_legacy = tmp_path + "/test_mesh_legacy.obj";
EXPECT_TRUE(t::io::WriteTriangleMesh(file_name_tensor, mesh_tensor));
EXPECT_TRUE(io::WriteTriangleMesh(file_name_legacy, mesh_legacy));
EXPECT_TRUE(t::io::ReadTriangleMesh(file_name_tensor, mesh_tensor_read));
EXPECT_TRUE(io::ReadTriangleMesh(file_name_legacy, mesh_legacy_read));
EXPECT_EQ(mesh_tensor_read.GetTriangleIndices().GetLength(),
static_cast<int64_t>(mesh_legacy_read.triangles_.size()));
EXPECT_EQ(mesh_tensor_read.GetVertexPositions().GetLength(),
static_cast<int64_t>(mesh_legacy_read.vertices_.size()));
}
} // namespace tests
} // namespace open3d
|
