image-match/python/opencv/modules/imgproc/test/test_subdivision2d.cpp

/*M///////////////////////////////////////////////////////////////////////////////////////
// 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.
//M*/
#include "test_precomp.hpp"

namespace opencv_test { namespace {
TEST(Imgproc_Subdiv2D_getTriangleList, regression_5788)
{
    const float points[65][2] = {
        { 390,  802}, { 397,  883}, { 414,  963 }, { 439, 1042 }, { 472, 1113},
        { 521, 1181}, { 591, 1238}, { 678, 1284 }, { 771, 1292 }, { 853, 1281},
        { 921, 1243}, { 982, 1191}, {1030, 1121 }, {1059, 1038 }, {1072,  945},
        {1081,  849}, {1082,  749}, { 459,  734 }, { 502,  704 }, { 554,  696},
        { 609,  698}, { 660,  707}, { 818,  688 }, { 874,  661 }, { 929,  646},
        { 982,  653}, {1026,  682}, { 740,  771 }, { 748,  834 }, { 756,  897},
        { 762,  960}, { 700,  998}, { 733, 1006 }, { 766, 1011 }, { 797,  999},
        { 825,  987}, { 528,  796}, { 566,  766 }, { 617,  763 }, { 659,  794},
        { 619,  808}, { 569,  812}, { 834,  777 }, { 870,  735 }, { 918,  729},
        { 958,  750}, { 929,  773}, { 882,  780 }, { 652, 1102 }, { 701, 1079},
        { 743, 1063}, { 774, 1068}, { 807, 1057 }, { 852, 1065 }, { 896, 1077},
        { 860, 1117}, { 820, 1135}, { 783, 1141 }, { 751, 1140 }, { 706, 1130},
        { 675, 1102}, { 743, 1094}, { 774, 1094 }, { 809, 1088 }, { 878, 1082}
    };
    std::vector<cv::Point2f> pts;
    cv::Rect rect(0, 0, 1500, 2000);
    cv::Subdiv2D subdiv(rect);
    for( int i = 0; i < 65; i++ )
    {
        cv::Point2f pt(points[i][0], points[i][1]);
        pts.push_back(pt);
    }

    subdiv.insert(pts);

    std::vector<cv::Vec6f> triangles;
    subdiv.getTriangleList(triangles);

    int trig_cnt = 0;
    for( std::vector<cv::Vec6f>::const_iterator it = triangles.begin(); it != triangles.end(); it++, trig_cnt++ )
    {
        EXPECT_TRUE( (0 <= triangles.at(trig_cnt).val[0] && triangles.at(trig_cnt).val[0] < 1500) &&
                     (0 <= triangles.at(trig_cnt).val[1] && triangles.at(trig_cnt).val[1] < 2000) &&
                     (0 <= triangles.at(trig_cnt).val[2] && triangles.at(trig_cnt).val[2] < 1500) &&
                     (0 <= triangles.at(trig_cnt).val[3] && triangles.at(trig_cnt).val[3] < 2000) &&
                     (0 <= triangles.at(trig_cnt).val[4] && triangles.at(trig_cnt).val[4] < 1500) &&
                     (0 <= triangles.at(trig_cnt).val[5] && triangles.at(trig_cnt).val[5] < 2000) );
    }
    EXPECT_EQ(trig_cnt, 105);
}

TEST(Imgproc_Subdiv2D, issue_25696) {
    std::vector<cv::Point2f> points{
        {0, 0}, {40, 40}, {84, 104}, {86, 108}
    };

    cv::Rect subdivRect{cv::Point{-10, -10}, cv::Point{96, 118}};
    cv::Subdiv2D subdiv{subdivRect};
    subdiv.insert(points);

    std::vector<cv::Vec6f> triangles;
    subdiv.getTriangleList(triangles);

    ASSERT_EQ(static_cast<size_t>(2), triangles.size());
}

// Initialization test
TEST(Imgproc_Subdiv2D, rect2f_constructor_and_init)
{
    cv::Rect2f rect_f(0.5f, 1.5f, 100.7f, 200.3f);
    cv::Subdiv2D subdiv_f(rect_f);

    cv::Point2f pt1(50.2f, 80.1f);
    cv::Point2f pt2(75.8f, 120.9f);
    cv::Point2f pt3(25.5f, 150.3f);

    EXPECT_NO_THROW(subdiv_f.insert(pt1));
    EXPECT_NO_THROW(subdiv_f.insert(pt2));
    EXPECT_NO_THROW(subdiv_f.insert(pt3));

    cv::Subdiv2D subdiv_init;

    EXPECT_NO_THROW(subdiv_init.initDelaunay(rect_f));
    EXPECT_NO_THROW(subdiv_init.insert(pt1));
    EXPECT_NO_THROW(subdiv_init.insert(pt2));
    EXPECT_NO_THROW(subdiv_init.insert(pt3));

    std::vector<cv::Vec6f> triangles;

    EXPECT_NO_THROW(subdiv_f.getTriangleList(triangles));
    EXPECT_GT(triangles.size(), 0u);
}

// test with small coordinates
TEST(Imgproc_Subdiv2D, rect2f_edge_cases)
{
    cv::Rect2f small_rect(0.0f, 0.0f, 0.1f, 0.1f);
    cv::Subdiv2D subdiv_small(small_rect);

    cv::Point2f small_pt(0.05f, 0.05f);
    EXPECT_NO_THROW(subdiv_small.insert(small_pt));
    cv::Rect2f float_rect(10.25f, 20.75f, 50.5f, 30.25f);

    cv::Subdiv2D subdiv_float(float_rect);

    cv::Point2f float_pt1(35.125f, 35.875f);
    cv::Point2f float_pt2(45.375f, 25.625f);
    cv::Point2f float_pt3(55.750f, 45.125f);

    EXPECT_NO_THROW(subdiv_float.insert(float_pt1));
    EXPECT_NO_THROW(subdiv_float.insert(float_pt2));
    EXPECT_NO_THROW(subdiv_float.insert(float_pt3));

    std::vector<cv::Vec6f> triangles;

    subdiv_float.getTriangleList(triangles);
    EXPECT_GT(triangles.size(), 0u);
}
}}