OpenCV_4.2.0/opencv_contrib-4.2.0/modules/phase_unwrapping/test/test_unwrapping.cpp

67 lines
1.7 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 "test_precomp.hpp"
namespace opencv_test { namespace {
class CV_Unwrapping : public cvtest::BaseTest
{
public:
CV_Unwrapping();
~CV_Unwrapping();
protected:
void run(int);
};
CV_Unwrapping::CV_Unwrapping(){}
CV_Unwrapping::~CV_Unwrapping(){}
void CV_Unwrapping::run( int )
{
int rows = 600;
int cols = 800;
int max = 50;
Mat ramp(rows, cols, CV_32FC1);
Mat wrappedRamp(rows, cols, CV_32FC1);
Mat unwrappedRamp;
Mat rowValues(1, cols, CV_32FC1);
Mat wrappedRowValues(1, cols, CV_32FC1);
for( int i = 0; i < cols; ++i )
{
float v = (float)i*(float)max/(float)cols;
rowValues.at<float>(0, i) = v;
wrappedRowValues.at<float>(0, i) = atan2(sin(v), cos(v));
}
for( int i = 0; i < rows; ++i )
{
rowValues.row(0).copyTo(ramp.row(i));
wrappedRowValues.row(0).copyTo(wrappedRamp.row(i));
}
phase_unwrapping::HistogramPhaseUnwrapping::Params params;
params.width = cols;
params.height = rows;
Ptr<phase_unwrapping::HistogramPhaseUnwrapping> phaseUnwrapping = phase_unwrapping::HistogramPhaseUnwrapping::create(params);
phaseUnwrapping->unwrapPhaseMap(wrappedRamp, unwrappedRamp);
for(int i = 0; i < rows; ++i )
{
for( int j = 0; j < cols; ++ j )
{
EXPECT_NEAR(ramp.at<float>(i, j), unwrappedRamp.at<float>(i, j), 0.001);
}
}
}
TEST( HistogramPhaseUnwrapping, unwrapPhaseMap )
{
CV_Unwrapping test;
test.safe_run();
}
}} // namespace