107 lines
3.1 KiB
C++
107 lines
3.1 KiB
C++
//
|
|
// Created by vptyp on 12.03.2026.
|
|
//
|
|
#include "cloud_point/rectify.h"
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <cloud_point/imageFactory.h>
|
|
|
|
class ImageTest : public ::testing::Test {
|
|
protected:
|
|
void SetUp() override {}
|
|
void TearDown() override {}
|
|
};
|
|
|
|
TEST_F(ImageTest, DefaultConstructor) {
|
|
score::Image image;
|
|
cv::Mat mat = image.get();
|
|
EXPECT_TRUE(mat.empty());
|
|
}
|
|
|
|
TEST_F(ImageTest, ConstructorWithMat) {
|
|
cv::Mat input = cv::Mat::zeros(5, 5, CV_8UC1);
|
|
score::Image image(input);
|
|
cv::Mat output = image.get();
|
|
EXPECT_EQ(output.rows, 5);
|
|
EXPECT_EQ(output.cols, 5);
|
|
EXPECT_EQ(output.type(), CV_8UC1);
|
|
}
|
|
|
|
TEST_F(ImageTest, PixelTypeMapping) {
|
|
EXPECT_EQ(score::ImageFactory::pixelType(score::ImageRPC::Type::BGR),
|
|
CV_8UC3);
|
|
EXPECT_EQ(score::ImageFactory::pixelType(score::ImageRPC::Type::RGBA),
|
|
CV_8UC4);
|
|
EXPECT_EQ(score::ImageFactory::pixelType(score::ImageRPC::Type::DEPTH),
|
|
CV_64FC1);
|
|
EXPECT_THROW(score::ImageFactory::pixelType(score::ImageRPC::Type::UNKNOWN),
|
|
std::runtime_error);
|
|
}
|
|
|
|
TEST_F(ImageTest, CreateBGR) {
|
|
score::ImageRPC rpc;
|
|
rpc.width = 10;
|
|
rpc.height = 20;
|
|
rpc.type = score::ImageRPC::Type::BGR;
|
|
rpc.data.resize(rpc.width * rpc.height * 3, 128);
|
|
|
|
score::Image image = score::ImageFactory::create(rpc);
|
|
cv::Mat mat = image.get();
|
|
|
|
EXPECT_EQ(mat.rows, 10);
|
|
EXPECT_EQ(mat.cols, 20);
|
|
EXPECT_EQ(mat.type(), CV_8UC3);
|
|
EXPECT_EQ(mat.at<cv::Vec3b>(0, 0)[0], 128);
|
|
}
|
|
|
|
TEST_F(ImageTest, CreateRGBA) {
|
|
score::ImageRPC rpc;
|
|
rpc.width = 15;
|
|
rpc.height = 25;
|
|
rpc.type = score::ImageRPC::Type::RGBA;
|
|
rpc.data.resize(rpc.width * rpc.height * 4, 255);
|
|
|
|
score::Image image = score::ImageFactory::create(rpc);
|
|
cv::Mat mat = image.get();
|
|
|
|
EXPECT_EQ(mat.rows, 15);
|
|
EXPECT_EQ(mat.cols, 25);
|
|
EXPECT_EQ(mat.type(), CV_8UC4);
|
|
EXPECT_EQ(mat.at<cv::Vec4b>(0, 0)[0], 255);
|
|
}
|
|
|
|
TEST_F(ImageTest, CreateDepth) {
|
|
score::ImageRPC rpc;
|
|
rpc.width = 5;
|
|
rpc.height = 10;
|
|
rpc.type = score::ImageRPC::Type::DEPTH;
|
|
rpc.data.resize(rpc.width * rpc.height * sizeof(double));
|
|
auto *dataPtr = reinterpret_cast<double *>(rpc.data.data());
|
|
for (int i = 0; i < 50; ++i)
|
|
dataPtr[i] = static_cast<double>(i);
|
|
|
|
score::Image image = score::ImageFactory::create(rpc);
|
|
cv::Mat mat = image.get();
|
|
|
|
EXPECT_EQ(mat.rows, 5);
|
|
EXPECT_EQ(mat.cols, 10);
|
|
EXPECT_EQ(mat.type(), CV_64FC1);
|
|
EXPECT_DOUBLE_EQ(mat.at<double>(0, 0), 0.0);
|
|
EXPECT_DOUBLE_EQ(mat.at<double>(4, 9), 49.0);
|
|
}
|
|
|
|
TEST_F(ImageTest, RectificationNoThrow) {
|
|
score::Rectify rectify;
|
|
score::ImageRPC rpc;
|
|
rpc.width = 10;
|
|
rpc.height = 20;
|
|
rpc.type = score::ImageRPC::Type::BGR;
|
|
rpc.data.resize(rpc.width * rpc.height * 3, 128);
|
|
score::Image image = score::ImageFactory::create(rpc);
|
|
double fx = 10.1, fy = 20.2, cx = 30.3, cy = 40.4;
|
|
cv::Mat cameraMatrix =
|
|
(cv::Mat_<double>(3, 3) << fx, 0, cx, 0, fy, cy, 0, 0, 1);
|
|
|
|
EXPECT_NO_THROW(rectify.perform(image, cameraMatrix));
|
|
} |