mirror of
https://github.com/opencv/opencv.git
synced 2024-12-14 17:29:17 +08:00
4d695cd2f4
GSoC: Modified PLY reader to support color attribute read/write #23805 * Modified PLY reader to support color attribute read/write * Fix bugs & Modified OBJ reader for color IO * Replace with correct test file * Fix I/O of property [w] in OBJ files ### Pull Request Readiness Checklist **Merged with https://github.com/opencv/opencv_extra/pull/1075** [Issue for GSoC 2023](https://github.com/opencv/opencv/issues/23624) The ply loader in 3D module doesn't support color attribute reading. I modified that to support color attribute reading & writing for the color attribute compression as described in proposal. See details at https://github.com/opencv/opencv/wiki/How_to_contribute#making-a-good-pull-request - [x] I agree to contribute to the project under Apache 2 License. - [x] To the best of my knowledge, the proposed patch is not based on a code under GPL or another license that is incompatible with OpenCV - [x] The PR is proposed to the proper branch - [x] There is a reference to the original bug report and related work - [x] There is accuracy test, performance test and test data in opencv_extra repository, if applicable Patch to opencv_extra has the same branch name. - [x] The feature is well documented and sample code can be built with the project CMake
206 lines
6.1 KiB
C++
206 lines
6.1 KiB
C++
// This file is part of OpenCV project.
|
|
// It is subject to the license terms in the LICENSE file found in the top-level directory
|
|
// of this distribution and at http://opencv.org/license.html.
|
|
|
|
#include <opencv2/core.hpp>
|
|
#include <vector>
|
|
#include <cstdio>
|
|
|
|
#include "test_precomp.hpp"
|
|
#include "opencv2/ts.hpp"
|
|
|
|
namespace opencv_test { namespace {
|
|
|
|
TEST(PointCloud, LoadObj)
|
|
{
|
|
std::vector<cv::Point3f> points_gold = {
|
|
{-5.93915f, -0.13257f, 2.55837f},
|
|
{-5.93915f, 1.86743f, 2.55837f},
|
|
{-5.93915f, -0.13257f, -1.16339f},
|
|
{-5.93915f, 1.86743f, -1.16339f},
|
|
{0.399941f, -0.13257f, 2.55837f},
|
|
{0.399941f, 1.86743f, 2.55837f},
|
|
{0.399941f, -0.13257f, -1.16339f},
|
|
{0.399941f, 1.86743f, -1.16339f}};
|
|
|
|
std::vector<cv::Point3f> normals_gold = {
|
|
{-1.0000f, 0.0000f, 0.0000f},
|
|
{0.0000f, 0.0000f, -1.0000f},
|
|
{1.0000f, 0.0000f, 0.0000f},
|
|
{0.0000f, 0.0000f, 1.0000f},
|
|
{0.0000f, -1.0000f, 0.0000f},
|
|
{0.0000f, 1.0000f, 0.0000f}};
|
|
|
|
std::vector<cv::Point3_<uchar>> rgb_gold = {
|
|
{19, 144, 149},
|
|
{219, 28, 216},
|
|
{218, 157, 101},
|
|
{11, 161, 78},
|
|
{248, 183, 214},
|
|
{63, 196, 6},
|
|
{165, 190, 153},
|
|
{89, 203, 11}};
|
|
|
|
std::vector<cv::Point3f> points;
|
|
std::vector<cv::Point3f> normals;
|
|
std::vector<cv::Point3_<uchar>> rgb;
|
|
|
|
auto folder = cvtest::TS::ptr()->get_data_path();
|
|
cv::loadPointCloud(folder + "pointcloudio/orig.obj", points, normals, rgb);
|
|
|
|
EXPECT_EQ(points_gold, points);
|
|
EXPECT_EQ(rgb_gold, rgb);
|
|
EXPECT_EQ(normals_gold, normals);
|
|
}
|
|
|
|
TEST(PointCloud, LoadObjNoNormals)
|
|
{
|
|
std::vector<cv::Point3f> points_gold = {
|
|
{-5.93915f, -0.13257f, 2.55837f},
|
|
{-5.93915f, 1.86743f, 2.55837f},
|
|
{-5.93915f, -0.13257f, -1.16339f},
|
|
{-5.93915f, 1.86743f, -1.16339f},
|
|
{0.399941f, -0.13257f, 2.55837f},
|
|
{0.399941f, 1.86743f, 2.55837f},
|
|
{0.399941f, -0.13257f, -1.16339f},
|
|
{0.399941f, 1.86743f, -1.16339f}};
|
|
|
|
std::vector<cv::Point3f> points;
|
|
std::vector<cv::Point3f> normals;
|
|
|
|
auto folder = cvtest::TS::ptr()->get_data_path();
|
|
cv::loadPointCloud(folder + "pointcloudio/orig_no_norms.obj", points, normals);
|
|
|
|
EXPECT_EQ(points_gold, points);
|
|
EXPECT_TRUE(normals.empty());
|
|
}
|
|
|
|
TEST(PointCloud, SaveObj)
|
|
{
|
|
std::vector<cv::Point3f> points_gold;
|
|
std::vector<cv::Point3f> normals_gold;
|
|
std::vector<cv::Point3_<uchar>> rgb_gold;
|
|
|
|
auto folder = cvtest::TS::ptr()->get_data_path();
|
|
auto new_path = tempfile("new.obj");
|
|
|
|
cv::loadPointCloud(folder + "pointcloudio/orig.obj", points_gold, normals_gold, rgb_gold);
|
|
cv::savePointCloud(new_path, points_gold, normals_gold, rgb_gold);
|
|
|
|
std::vector<cv::Point3f> points;
|
|
std::vector<cv::Point3f> normals;
|
|
std::vector<cv::Point3_<uchar>> rgb;
|
|
|
|
cv::loadPointCloud(new_path, points, normals, rgb);
|
|
|
|
EXPECT_EQ(normals, normals_gold);
|
|
EXPECT_EQ(points, points_gold);
|
|
EXPECT_EQ(rgb, rgb_gold);
|
|
std::remove(new_path.c_str());
|
|
}
|
|
|
|
TEST(PointCloud, LoadSavePly)
|
|
{
|
|
std::vector<cv::Point3f> points;
|
|
std::vector<cv::Point3f> normals;
|
|
std::vector<cv::Point3_<uchar>> rgb;
|
|
|
|
auto folder = cvtest::TS::ptr()->get_data_path();
|
|
std::string new_path = tempfile("new.ply");
|
|
|
|
cv::loadPointCloud(folder + "pointcloudio/orig.ply", points, normals, rgb);
|
|
cv::savePointCloud(new_path, points, normals, rgb);
|
|
|
|
std::vector<cv::Point3f> points_gold;
|
|
std::vector<cv::Point3f> normals_gold;
|
|
std::vector<cv::Point3_<uchar>> rgb_gold;
|
|
|
|
cv::loadPointCloud(new_path, points_gold, normals_gold, rgb_gold);
|
|
|
|
EXPECT_EQ(normals_gold, normals);
|
|
EXPECT_EQ(points_gold, points);
|
|
EXPECT_EQ(rgb_gold, rgb);
|
|
std::remove(new_path.c_str());
|
|
}
|
|
|
|
TEST(PointCloud, LoadSaveMeshObj)
|
|
{
|
|
std::vector<cv::Point3f> points;
|
|
std::vector<cv::Point3f> normals;
|
|
std::vector<std::vector<int32_t>> indices;
|
|
|
|
auto folder = cvtest::TS::ptr()->get_data_path();
|
|
std::string new_path = tempfile("new_mesh.obj");
|
|
|
|
cv::loadMesh(folder + "pointcloudio/orig.obj", points, normals, indices);
|
|
cv::saveMesh(new_path, points, normals, indices);
|
|
|
|
std::vector<cv::Point3f> points_gold;
|
|
std::vector<cv::Point3f> normals_gold;
|
|
std::vector<std::vector<int32_t>> indices_gold;
|
|
|
|
cv::loadMesh(new_path, points_gold, normals_gold, indices_gold);
|
|
|
|
EXPECT_EQ(normals_gold, normals);
|
|
EXPECT_EQ(points_gold, points);
|
|
EXPECT_EQ(indices_gold, indices);
|
|
EXPECT_TRUE(!indices.empty());
|
|
std::remove(new_path.c_str());
|
|
}
|
|
|
|
TEST(PointCloud, LoadSaveMeshPly)
|
|
{
|
|
std::vector<cv::Point3f> points;
|
|
std::vector<cv::Point3f> normals;
|
|
std::vector<std::vector<int32_t>> indices;
|
|
|
|
auto folder = cvtest::TS::ptr()->get_data_path();
|
|
std::string new_path = tempfile("new_mesh.ply");
|
|
|
|
// we don't support meshes in PLY format right now but it should exit silently
|
|
cv::loadMesh(folder + "pointcloudio/orig.ply", points, normals, indices);
|
|
EXPECT_TRUE(points.empty());
|
|
EXPECT_TRUE(normals.empty());
|
|
EXPECT_TRUE(indices.empty());
|
|
|
|
cv::saveMesh(new_path, points, normals, indices);
|
|
EXPECT_TRUE(points.empty());
|
|
EXPECT_TRUE(normals.empty());
|
|
EXPECT_TRUE(indices.empty());
|
|
|
|
std::remove(new_path.c_str());
|
|
}
|
|
|
|
TEST(PointCloud, NonexistentFile)
|
|
{
|
|
std::vector<cv::Point3f> points;
|
|
std::vector<cv::Point3f> normals;
|
|
|
|
auto folder = cvtest::TS::ptr()->get_data_path();
|
|
cv::loadPointCloud(folder + "pointcloudio/fake.obj", points, normals);
|
|
EXPECT_TRUE(points.empty());
|
|
EXPECT_TRUE(normals.empty());
|
|
}
|
|
|
|
TEST(PointCloud, LoadBadExtension)
|
|
{
|
|
std::vector<cv::Point3f> points;
|
|
std::vector<cv::Point3f> normals;
|
|
|
|
auto folder = cvtest::TS::ptr()->get_data_path();
|
|
cv::loadPointCloud(folder + "pointcloudio/fake.fake", points, normals);
|
|
EXPECT_TRUE(points.empty());
|
|
EXPECT_TRUE(normals.empty());
|
|
}
|
|
|
|
TEST(PointCloud, SaveBadExtension)
|
|
{
|
|
std::vector<cv::Point3f> points;
|
|
std::vector<cv::Point3f> normals;
|
|
|
|
auto folder = cvtest::TS::ptr()->get_data_path();
|
|
cv::savePointCloud(folder + "pointcloudio/fake.fake", points, normals);
|
|
}
|
|
|
|
}} /* namespace opencv_test */
|