67 lines
1.7 KiB
C++
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
|