diff --git a/modules/core/include/opencv2/core.hpp b/modules/core/include/opencv2/core.hpp index d134273753..12951d2a98 100644 --- a/modules/core/include/opencv2/core.hpp +++ b/modules/core/include/opencv2/core.hpp @@ -2004,8 +2004,8 @@ The function solveCubic finds the real roots of a cubic equation: The roots are stored in the roots array. @param coeffs equation coefficients, an array of 3 or 4 elements. -@param roots output array of real roots that has 1 or 3 elements. -@return number of real roots. It can be 0, 1 or 2. +@param roots output array of real roots that has 0, 1, 2 or 3 elements. +@return number of real roots. It can be -1 (all real numbers), 0, 1, 2 or 3. */ CV_EXPORTS_W int solveCubic(InputArray coeffs, OutputArray roots); diff --git a/modules/core/src/mathfuncs.cpp b/modules/core/src/mathfuncs.cpp index ac037e38c7..7af66c5723 100644 --- a/modules/core/src/mathfuncs.cpp +++ b/modules/core/src/mathfuncs.cpp @@ -1590,7 +1590,7 @@ int cv::solveCubic( InputArray _coeffs, OutputArray _roots ) { if( a1 == 0 ) { - if( a2 == 0 ) + if( a2 == 0 ) // constant n = a3 == 0 ? -1 : 0; else { @@ -1624,6 +1624,7 @@ int cv::solveCubic( InputArray _coeffs, OutputArray _roots ) } else { + // cubic equation a0 = 1./a0; a1 *= a0; a2 *= a0; diff --git a/modules/core/test/test_math.cpp b/modules/core/test/test_math.cpp index b1777a7752..dc743dcbc1 100644 --- a/modules/core/test/test_math.cpp +++ b/modules/core/test/test_math.cpp @@ -2445,6 +2445,158 @@ static void checkRoot(Mat& r, T re, T im) } GTEST_NONFATAL_FAILURE_("Can't find root") << "(" << re << ", " << im << ")"; } + +TEST(Core_SolveCubicConstant, accuracy) +{ + { + const std::vector coeffs{0., 0., 0., 1.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_EQ(num_roots, 0); + } + + { + const std::vector coeffs{0., 0., 0., 0.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_EQ(num_roots, -1); + } +} + +TEST(Core_SolveCubicLinear, accuracy) +{ + const std::vector coeffs{0., 0., 2., -2.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_EQ(num_roots, 1); + EXPECT_EQ(roots[0], 1.); +} + +TEST(Core_SolveCubicQuadratic, accuracy) +{ + { + const std::vector coeffs{0., 2., -4., 4.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_EQ(num_roots, 0); + } + + { + const std::vector coeffs{0., 2., -4., 2.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_EQ(num_roots, 1); + EXPECT_EQ(roots[0], 1.); + } + + { + const std::vector coeffs{0., 2., -6., 4.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_EQ(num_roots, 2); + EXPECT_EQ(roots[0], 2.); + EXPECT_EQ(roots[1], 1.); + } +} + +TEST(Core_SolveCubicCubic, accuracy) +{ + { + const std::vector coeffs{2., -6., 6., -2.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_EQ(num_roots, 1); + EXPECT_EQ(roots[0], 1.); + } + + { + const std::vector coeffs{2., -10., 24., -16.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_EQ(num_roots, 1); + EXPECT_EQ(roots[0], 1.); + } + + { + const std::vector coeffs{2., -10., 16., -8.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_TRUE(num_roots == 2 || num_roots == 3); + EXPECT_NEAR(roots[0], 1., 1e-8); + EXPECT_NEAR(roots[1], 2., 1e-8); + if (num_roots == 3) + { + EXPECT_NEAR(roots[2], 2., 1e-8); + } + } + + { + const std::vector coeffs{2., -12., 22., -12.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_EQ(num_roots, 3); + EXPECT_NEAR(roots[0], 1., 1e-8); + EXPECT_NEAR(roots[1], 3., 1e-8); + EXPECT_NEAR(roots[2], 2., 1e-8); + } +} + +TEST(Core_SolveCubicNormalizedCubic, accuracy) +{ + { + const std::vector coeffs{-3., 3., -1.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_EQ(num_roots, 1); + EXPECT_EQ(roots[0], 1.); + } + + { + const std::vector coeffs{-5., 12., -8.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_EQ(num_roots, 1); + EXPECT_EQ(roots[0], 1.); + } + + { + const std::vector coeffs{-5., 8., -4.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_TRUE(num_roots == 2 || num_roots == 3); + EXPECT_NEAR(roots[0], 1., 1e-8); + EXPECT_NEAR(roots[1], 2., 1e-8); + if (num_roots == 3) + { + EXPECT_NEAR(roots[2], 2., 1e-8); + } + } + + { + const std::vector coeffs{-6., 11., -6.}; + std::vector roots; + const auto num_roots = solveCubic(coeffs, roots); + + EXPECT_EQ(num_roots, 3); + EXPECT_NEAR(roots[0], 1., 1e-8); + EXPECT_NEAR(roots[1], 3., 1e-8); + EXPECT_NEAR(roots[2], 2., 1e-8); + } +} + TEST(Core_SolvePoly, regression_5599) { // x^4 - x^2 = 0, roots: 1, -1, 0, 0