/** * Copyright (c) 2023 OceanBase * OceanBase CE is licensed under Mulan PubL v2. * You can use this software according to the terms and conditions of the Mulan PubL v2. * You may obtain a copy of Mulan PubL v2 at: * http://license.coscl.org.cn/MulanPubL-2.0 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE. * See the Mulan PubL v2 for more details. */ #include #include #include #define private public #include "lib/geo/ob_geo_bin.h" #include "lib/geo/ob_geo_tree.h" #include "lib/geo/ob_geo_tree_traits.h" #include "lib/geo/ob_geo_bin_traits.h" #include "lib/json_type/ob_json_common.h" #include "lib/random/ob_random.h" #include "lib/geo/ob_geo_utils.h" #include "lib/geo/ob_geo_box_clip_visitor.h" #include "lib/geo/ob_geo_to_tree_visitor.h" #include "lib/geo/ob_geo_to_wkt_visitor.h" #include "sql/engine/expr/ob_geo_expr_utils.h" #include "lib/geo/ob_wkt_parser.h" #include "lib/geo/ob_geo_common.h" #undef private #include #include #include #include namespace bg = boost::geometry; namespace oceanbase { namespace common { class TestGeoTree : public ::testing::Test { public: TestGeoTree() {} ~TestGeoTree() {} virtual void SetUp() {} virtual void TearDown() {} static void SetUpTestCase() {} static void TearDownTestCase() {} private: // disallow copy DISALLOW_COPY_AND_ASSIGN(TestGeoTree); }; TEST_F(TestGeoTree, iters) { ObArenaAllocator allocator(ObModIds::TEST); ModulePageAllocator page_allocator_(allocator, common::ObModIds::OB_MODULE_PAGE_ALLOCATOR); ObGeomVector gv(page_allocator_); const ObGeomVector &c_gv = gv; gv.push_back(0); gv.push_back(1); gv.push_back(2); gv.push_back(3); gv.push_back(4); gv.push_back(5); gv.push_back(6); gv.push_back(7); gv.push_back(8); gv.push_back(9); ASSERT_EQ(gv.size(), 10); ObGeomVector::iterator iter = gv.begin(); ObGeomVector::const_iterator c_iter = c_gv.begin(); ObGeomVector::iterator begin = gv.begin(); ObGeomVector::const_iterator c_begin = c_gv.begin(); ObGeomVector::iterator end = gv.end(); ObGeomVector::const_iterator c_end = c_gv.end(); ObGeomVector::iterator tmp_iter; ObGeomVector::const_iterator tmp_c_iter; // test ++, -- ASSERT_EQ(iter, begin); ASSERT_EQ(c_iter, c_begin); ASSERT_NE(iter, end); ASSERT_NE(c_iter, c_end); int i = 0; for (; i < 10; i++) { ASSERT_EQ(*iter, i); ASSERT_EQ(*c_iter, i); iter++; c_iter++; } ASSERT_EQ(iter, end); ASSERT_EQ(c_iter, c_end); iter = begin; c_iter = c_begin; for (i = 0; i < 10; i++) { ASSERT_EQ(*iter, i); ASSERT_EQ(*c_iter, i); iter++; c_iter++; } ASSERT_EQ(iter, end); ASSERT_EQ(c_iter, c_end); ASSERT_EQ(i, 10); i--; iter--; c_iter--; for (; i > 0; i--) { ASSERT_EQ(*iter, i); ASSERT_EQ(*c_iter, i); iter--; c_iter--; } ASSERT_EQ(iter, begin); ASSERT_EQ(c_iter, c_begin); tmp_iter = iter + 3; tmp_c_iter = c_iter + 3; ASSERT_EQ(*tmp_iter, 3); ASSERT_EQ(*tmp_c_iter, 3); tmp_iter = 4 + iter; tmp_c_iter = 4 + c_iter; ASSERT_EQ(*tmp_iter, 4); ASSERT_EQ(*tmp_c_iter, 4); tmp_iter = end - 3; tmp_c_iter = c_end - 3; ASSERT_EQ(*tmp_iter, *tmp_c_iter); ASSERT_EQ(*tmp_iter, 7); ASSERT_EQ(*tmp_c_iter, 7); tmp_iter += 2; tmp_c_iter += 2; ASSERT_EQ(*tmp_iter, *tmp_c_iter); ASSERT_EQ(*tmp_iter, 9); ASSERT_EQ(*tmp_c_iter, 9); tmp_iter -= 2; tmp_c_iter -= 2; ASSERT_EQ(*tmp_iter, *tmp_c_iter); ASSERT_EQ(*tmp_iter, 7); ASSERT_EQ(*tmp_c_iter, 7); tmp_iter = ++iter; tmp_c_iter = ++c_iter; ASSERT_EQ(*tmp_iter, *tmp_c_iter); ASSERT_EQ(*tmp_iter, 1); ASSERT_EQ(*tmp_c_iter, 1); tmp_iter = iter++; tmp_c_iter = c_iter++; ASSERT_EQ(*tmp_iter, *tmp_c_iter); ASSERT_EQ(*tmp_iter, 1); ASSERT_EQ(*tmp_c_iter, 1); ASSERT_EQ(*iter, *c_iter); ASSERT_EQ(*iter, 2); ASSERT_EQ(*c_iter, 2); ASSERT_EQ(c_end - c_begin, end - begin); ASSERT_TRUE(c_end != c_begin); ASSERT_TRUE(end != begin); ASSERT_FALSE((c_end < c_begin)); ASSERT_FALSE(end < begin); ASSERT_TRUE(begin < end); ASSERT_TRUE(c_begin < c_end); } TEST_F(TestGeoTree, ObGeomVector) { ObArenaAllocator allocator(ObModIds::TEST); ModulePageAllocator page_allocator_(allocator, common::ObModIds::OB_MODULE_PAGE_ALLOCATOR); ObGeomVector gv(page_allocator_); ASSERT_TRUE(gv.empty()); gv.push_back(0); gv.push_back(1); gv.push_back(2); gv.push_back(3); gv.push_back(4); gv.push_back(5); gv.push_back(6); gv.push_back(7); gv.push_back(8); gv.push_back(9); const ObGeomVector &c_gv = gv; ASSERT_EQ(c_gv.front(), 0); ASSERT_EQ(c_gv.back(), 9); for (int i = 0; i < 10; i++) { ASSERT_EQ(c_gv[i], i); } ASSERT_FALSE(gv.empty()); ASSERT_EQ(gv.size(), 10); gv.pop_front(); ASSERT_EQ(gv.front(), 1); gv.pop_front(); ASSERT_EQ(gv.front(), 2); ASSERT_EQ(gv.back(), 9); gv.resize(3); ASSERT_EQ(gv.size(), 3); ASSERT_EQ(gv.front(), 2); ASSERT_EQ(gv[1], 3); ASSERT_EQ(gv.back(), 4); gv.resize(10); ASSERT_EQ(gv.size(), 10); gv.clear(); ASSERT_TRUE(gv.empty()); } TEST_F(TestGeoTree, point) { ObArenaAllocator allocator(ObModIds::TEST); ObCartesianPoint p(1.12, 2.32, 0); ASSERT_EQ(1.12, p.get<0>()); ASSERT_EQ(2.32, p.get<1>()); ASSERT_EQ(ObGeoCRS::Cartesian, p.crs()); p.set<0>(3.321); p.set<1>(4.444); ASSERT_EQ(3.321, p.get<0>()); ASSERT_EQ(4.444, p.get<1>()); p.x(-3.123); p.y(-4.567); ASSERT_EQ(-3.123, p.get<0>()); ASSERT_EQ(-4.567, p.get<1>()); p.x(1.7976931348623157e308); p.y(-1.7976931348623157e308); EXPECT_EQ(1.7976931348623157e308, p.x()); EXPECT_EQ(-1.7976931348623157e308, p.y()); ASSERT_FALSE(p.is_empty()); ObCartesianPoint p1(0, &allocator); ASSERT_TRUE(std::isnan(p1.x())); ASSERT_TRUE(std::isnan(p1.y())); ASSERT_TRUE(p1.is_empty()); } TEST_F(TestGeoTree, linestring) { ObArenaAllocator allocator(ObModIds::TEST); ObLineString *l = NULL; ASSERT_EQ(OB_SUCCESS, ObLineString::create_linestring(ObGeoCRS::Cartesian, 0, allocator, l)); ObCartesianLineString *line = static_cast(l); ASSERT_EQ(ObGeoType::LINESTRING, line->type()); ASSERT_EQ(1, line->dimension()); uint32_t num = 10000; common::ObVector xv; common::ObVector yv; for (int i = 0; i < num; i++) { double x = static_cast(rand())/static_cast(rand()); double y = static_cast(rand())/static_cast(rand()); ASSERT_EQ(OB_SUCCESS, line->push_back(ObWkbGeomInnerPoint(x, y))); xv.push_back(x); yv.push_back(y); } ASSERT_EQ(10000, line->size()); ObCartesianLineString::iterator iter = line->begin(); for (int i = 0; iter != line->end(); ++iter, i++) { ASSERT_EQ(xv[i], iter->get<0>()); ASSERT_EQ(yv[i], iter->get<1>()); } line->clear(); ASSERT_EQ(0, line->size()); ASSERT_TRUE(line->is_empty()); for (int i = 0; i < 4; i++) { ASSERT_EQ(OB_SUCCESS, line->push_back(ObWkbGeomInnerPoint(i, i+1))); } for (int i = 0; i < 4; i++) { ASSERT_EQ(i, (*line)[i].get<0>()); ASSERT_EQ(i+1, (*line)[i].get<1>()); } ASSERT_EQ(0, line->front().get<0>()); ASSERT_EQ(1, line->front().get<1>()); ASSERT_EQ(3, line->back().get<0>()); ASSERT_EQ(4, line->back().get<1>()); line->pop_front(); ASSERT_EQ(1, line->front().get<0>()); ASSERT_EQ(2, line->front().get<1>()); } TEST_F(TestGeoTree, linesarring) { ObArenaAllocator allocator(ObModIds::TEST); ObLinearring *r = NULL; ASSERT_EQ(OB_SUCCESS, ObLinearring::create_linearring(ObGeoCRS::Cartesian, 0, allocator, r)); ASSERT_EQ(ObGeoType::LINESTRING, r->type()); ASSERT_EQ(1, r->dimension()); ObCartesianLinearring *ring = static_cast(r); ASSERT_EQ(0, ring->size()); ASSERT_TRUE(ring->is_empty()); ObWkbGeomInnerPoint p1(0.0, 0.0); ASSERT_EQ(OB_SUCCESS, ring->push_back(p1)); ObWkbGeomInnerPoint p2(0, 10); ASSERT_EQ(OB_SUCCESS, ring->push_back(p2)); ObWkbGeomInnerPoint p3(10, 10); ASSERT_EQ(OB_SUCCESS, ring->push_back(p3)); ObWkbGeomInnerPoint p4(10, 0.0); ASSERT_EQ(OB_SUCCESS, ring->push_back(p4)); ObCartesianLinearring::iterator iter = ring->begin(); for (int i = 0; iter != ring->end(); ++iter, i++) { std::cout << "(" << iter->get<0>() << ", " << iter->get<1>() << ")" << std::endl; } ASSERT_EQ(0, (*ring)[1].get<0>()); ASSERT_EQ(10, (*ring)[1].get<1>()); ASSERT_EQ(10, (*ring)[2].get<0>()); ASSERT_EQ(10, (*ring)[2].get<1>()); ASSERT_EQ(0, ring->front().get<0>()); ASSERT_EQ(0, ring->front().get<1>()); ASSERT_EQ(10, ring->back().get<0>()); ASSERT_EQ(0, ring->back().get<1>()); ring->pop_front(); ASSERT_EQ(3, ring->size()); ASSERT_EQ(0, ring->front().get<0>()); ASSERT_EQ(10, ring->front().get<1>()); } TEST_F(TestGeoTree, Polygon) { ObPolygon *poly = NULL; ObArenaAllocator allocator(ObModIds::TEST); ASSERT_EQ(OB_SUCCESS, ObPolygon::create_polygon(ObGeoCRS::Cartesian, 0, allocator, poly)); ASSERT_EQ(ObGeoType::POLYGON, poly->type()); ASSERT_EQ(ObGeoCRS::Cartesian, poly->crs()); ASSERT_EQ(0U, poly->size()); ASSERT_TRUE(poly->empty()); ASSERT_TRUE(poly->is_empty()); ASSERT_EQ(ObGeoType::POLYGON, poly->type()); ASSERT_EQ(ObGeoCRS::Cartesian, poly->crs()); ASSERT_EQ(0U, poly->size()); ASSERT_TRUE(poly->empty()); ASSERT_TRUE(poly->is_empty()); ObCartesianLinearring outer_ring(0, allocator); outer_ring.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); outer_ring.push_back(ObWkbGeomInnerPoint(10.0, 0.0)); outer_ring.push_back(ObWkbGeomInnerPoint(10.0, 10.0)); outer_ring.push_back(ObWkbGeomInnerPoint(0.0, 10.0)); outer_ring.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); poly->push_back(outer_ring); ASSERT_FALSE(poly->empty()); ASSERT_FALSE(poly->is_empty()); ObCartesianLinearring inner_ring(0, allocator); inner_ring.push_back(ObWkbGeomInnerPoint(2.0, 2.0)); inner_ring.push_back(ObWkbGeomInnerPoint(2.0, 8.0)); inner_ring.push_back(ObWkbGeomInnerPoint(8.0, 8.0)); inner_ring.push_back(ObWkbGeomInnerPoint(8.0, 2.0)); inner_ring.push_back(ObWkbGeomInnerPoint(2.0, 2.0)); poly->push_back(inner_ring); ObCartesianPolygon *polygon = static_cast(poly); ASSERT_EQ(2U, polygon->size()); ASSERT_FALSE(polygon->empty()); ASSERT_EQ(1U, polygon->inner_ring_size()); ASSERT_FALSE(polygon->inner_ring(0).empty()); ASSERT_EQ(10, polygon->exterior_ring()[1].get<0>()); ASSERT_EQ(0, polygon->exterior_ring()[1].get<1>()); ASSERT_EQ(10, polygon->exterior_ring()[2].get<0>()); ASSERT_EQ(10, polygon->exterior_ring()[2].get<1>()); ASSERT_EQ(2, polygon->inner_ring(0)[1].get<0>()); ASSERT_EQ(8, polygon->inner_ring(0)[1].get<1>()); ASSERT_EQ(8, polygon->inner_ring(0)[2].get<0>()); ASSERT_EQ(8, polygon->inner_ring(0)[2].get<1>()); } TEST_F(TestGeoTree, Geometrycollection) { ObArenaAllocator allocator(ObModIds::TEST); ObGeometrycollection* gc = NULL; ASSERT_EQ(OB_SUCCESS, ObGeometrycollection::create_collection(ObGeoCRS::Cartesian, 0, allocator, gc)); ASSERT_EQ(ObGeoType::GEOMETRYCOLLECTION, gc->type()); ASSERT_EQ(ObGeoCRS::Cartesian, gc->crs()); ASSERT_TRUE(gc->empty()); ASSERT_TRUE(gc->is_empty()); ObCartesianGeometrycollection *cartesianGc = static_cast(gc); cartesianGc->push_back(ObCartesianGeometrycollection(0, allocator)); ASSERT_FALSE(cartesianGc->empty()); ASSERT_TRUE(cartesianGc->is_empty()); cartesianGc->push_back(ObCartesianPoint(0.0, 0.0, 0)); cartesianGc->push_back(ObCartesianPoint(10.0, 0.0, 0)); cartesianGc->push_back(ObCartesianPoint(10.0, 10.0, 0)); cartesianGc->push_back(ObCartesianPoint(0.0, 10.0, 0)); cartesianGc->push_back(ObCartesianPoint(0.0, 0.0, 0)); ObCartesianLineString ls(0, allocator); ls.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); ls.push_back(ObWkbGeomInnerPoint(10.0, 0.0)); ls.push_back(ObWkbGeomInnerPoint(10.0, 10.0)); ls.push_back(ObWkbGeomInnerPoint(0.0, 10.0)); ls.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); cartesianGc->push_back(ls); ObCartesianLinearring outer_ring(0, allocator); outer_ring.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); outer_ring.push_back(ObWkbGeomInnerPoint(10.0, 0.0)); outer_ring.push_back(ObWkbGeomInnerPoint(10.0, 10.0)); outer_ring.push_back(ObWkbGeomInnerPoint(0.0, 10.0)); outer_ring.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); ObCartesianPolygon py(0, allocator); py.push_back(outer_ring); cartesianGc->push_back(py); ObCartesianMultipoint mpt(0, allocator); mpt.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); cartesianGc->push_back(mpt); ObCartesianLineString ls2(0, allocator); ls2.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); ls2.push_back(ObWkbGeomInnerPoint(1.0, 1.0)); ObCartesianMultilinestring mls; mls.push_back(ls2); cartesianGc->push_back(mls); ObCartesianMultipolygon mpy(0, allocator); cartesianGc->push_back(mpy); ObCartesianGeometrycollection inner_gc(0, allocator); cartesianGc->push_back(inner_gc); ASSERT_EQ(12U, cartesianGc->size()); ASSERT_FALSE(cartesianGc->empty()); ASSERT_EQ(ObGeoType::GEOMETRYCOLLECTION, cartesianGc->front().type()); cartesianGc->pop_front(); cartesianGc->pop_front(); ASSERT_EQ(10U, cartesianGc->size()); ASSERT_EQ(ObGeoType::POINT, cartesianGc->front().type()); } TEST_F(TestGeoTree, Multipoint) { ObArenaAllocator allocator(ObModIds::TEST); ObMultipoint *mpt_ptr = NULL; ASSERT_EQ(OB_SUCCESS, ObMultipoint::create_multipoint(ObGeoCRS::Cartesian, 0, allocator, mpt_ptr)); ASSERT_EQ(ObGeoType::MULTIPOINT, mpt_ptr->type()); ASSERT_EQ(ObGeoCRS::Cartesian, mpt_ptr->crs()); ASSERT_TRUE(mpt_ptr->empty()); ASSERT_TRUE(mpt_ptr->is_empty()); ObCartesianMultipoint *cartMtp = static_cast(mpt_ptr); cartMtp->push_back(ObWkbGeomInnerPoint(0.0, 0.0)); ASSERT_EQ(1U, cartMtp->size()); ASSERT_FALSE(cartMtp->empty()); ASSERT_FALSE(cartMtp->is_empty()); cartMtp->push_back(ObWkbGeomInnerPoint(10.0, 10.0)); ASSERT_EQ(2U, cartMtp->size()); ASSERT_EQ(0.0, cartMtp->front().get<0>()); ASSERT_EQ(0.0, cartMtp->front().get<1>()); cartMtp->pop_front(); ASSERT_EQ(1U, cartMtp->size()); ASSERT_EQ(10.0, cartMtp->front().get<0>()); ASSERT_EQ(10.0, cartMtp->front().get<1>()); } TEST_F(TestGeoTree, Multilinestring) { ObArenaAllocator allocator(ObModIds::TEST); ObMultilinestring *mls_ptr = NULL; ASSERT_EQ(OB_SUCCESS, ObMultilinestring::create_multilinestring(ObGeoCRS::Cartesian, 0, allocator, mls_ptr)); ASSERT_EQ(ObGeoType::MULTILINESTRING, mls_ptr->type()); ASSERT_EQ(ObGeoCRS::Cartesian, mls_ptr->crs()); ASSERT_TRUE(mls_ptr->empty()); ASSERT_TRUE(mls_ptr->is_empty()); ObCartesianMultilinestring *cartMls = static_cast(mls_ptr); ObCartesianLineString ls(0, allocator); ls.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); ls.push_back(ObWkbGeomInnerPoint(10.0, 0.0)); ls.push_back(ObWkbGeomInnerPoint(10.0, 10.0)); ls.push_back(ObWkbGeomInnerPoint(0.0, 10.0)); ls.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); cartMls->push_back(ls); ASSERT_EQ(1U, cartMls->size()); ObCartesianMultilinestring *mls = cartMls; const ObCartesianMultilinestring *cmls = cartMls; ObCartesianMultilinestring::iterator iter = mls->begin(); ObCartesianMultilinestring::const_iterator citer = cmls->begin(); auto x = boost::begin(*mls); auto y = boost::begin(*cmls); ObCartesianLineString lsa(0, allocator); ASSERT_FALSE(cartMls->empty()); ASSERT_FALSE(cartMls->is_empty()); ObCartesianLineString ls2(0, allocator); ls2.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); ls2.push_back(ObWkbGeomInnerPoint(20.0, 20.0)); cartMls->push_back(ls2); ASSERT_EQ(2U, cartMls->size()); ASSERT_EQ(5U, cartMls->front().size()); ASSERT_EQ(10, cartMls->front()[2].get<0>()); ASSERT_EQ(10, cartMls->front()[2].get<1>()); cartMls->pop_front(); ASSERT_EQ(1U, cartMls->size()); ASSERT_EQ(2U, cartMls->front().size()); } TEST_F(TestGeoTree, Multipolygon) { ObArenaAllocator allocator(ObModIds::TEST); ObMultipolygon *mpy_ptr = NULL; ASSERT_EQ(OB_SUCCESS, ObMultipolygon::create_multipolygon(ObGeoCRS::Cartesian, 0, allocator, mpy_ptr)); ObCartesianMultipolygon *cartMpy = static_cast(mpy_ptr); ASSERT_EQ(ObGeoType::MULTIPOLYGON, cartMpy->type()); ASSERT_EQ(ObGeoCRS::Cartesian, cartMpy->crs()); ASSERT_TRUE(cartMpy->empty()); ASSERT_TRUE(cartMpy->is_empty()); ObCartesianLinearring outer_ring(0, allocator); outer_ring.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); outer_ring.push_back(ObWkbGeomInnerPoint(10.0, 0.0)); outer_ring.push_back(ObWkbGeomInnerPoint(10.0, 10.0)); outer_ring.push_back(ObWkbGeomInnerPoint(0.0, 10.0)); outer_ring.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); ObCartesianLinearring inner_ring(0, allocator); inner_ring.push_back(ObWkbGeomInnerPoint(2.0, 2.0)); inner_ring.push_back(ObWkbGeomInnerPoint(2.0, 8.0)); inner_ring.push_back(ObWkbGeomInnerPoint(8.0, 8.0)); inner_ring.push_back(ObWkbGeomInnerPoint(8.0, 2.0)); inner_ring.push_back(ObWkbGeomInnerPoint(2.0, 2.0)); ObCartesianPolygon py(0, allocator); py.push_back(outer_ring); py.push_back(inner_ring); cartMpy->push_back(py); ASSERT_EQ(1U, cartMpy->size()); ASSERT_FALSE(cartMpy->empty()); ASSERT_FALSE(cartMpy->is_empty()); cartMpy->push_back(ObCartesianPolygon(0, allocator)); ASSERT_EQ(2U, cartMpy->size()); ASSERT_EQ(2U, cartMpy->front().size()); cartMpy->pop_front(); ASSERT_EQ(1U, cartMpy->size()); ASSERT_EQ(0U, cartMpy->front().size()); } TEST_F(TestGeoTree, intersection_op) { ObArenaAllocator allocator(ObModIds::TEST); ObCartesianLineString ls1(0, allocator); ls1.push_back(ObWkbGeomInnerPoint(1.0, 1.0)); ls1.push_back(ObWkbGeomInnerPoint(3.0, 3.0)); ObCartesianLineString ls2(0, allocator); ls2.push_back(ObWkbGeomInnerPoint(1.0, 3.0)); ls2.push_back(ObWkbGeomInnerPoint(3.0, 1.0)); ObCartesianMultilinestring ml(0, allocator); bool d = bg::intersection(ls1, ls2, ml); for (int i = 0; i < ml.size(); i++) { for (int j = 0; j < ml[i].size(); j++) { ASSERT_EQ(2U, ml[i][j].get<0>()); ASSERT_EQ(2U, ml[i][j].get<1>()); } } ObCartesianLinearring outer_ring1(0, allocator); outer_ring1.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); outer_ring1.push_back(ObWkbGeomInnerPoint(10.0, 0.0)); outer_ring1.push_back(ObWkbGeomInnerPoint(10.0, 10.0)); outer_ring1.push_back(ObWkbGeomInnerPoint(0.0, 10.0)); outer_ring1.push_back(ObWkbGeomInnerPoint(0.0, 0.0)); ObCartesianPolygon py1(0, allocator); py1.push_back(outer_ring1); ObCartesianLinearring outer_ring2(0, allocator); outer_ring2.push_back(ObWkbGeomInnerPoint(5.0, 0.0)); outer_ring2.push_back(ObWkbGeomInnerPoint(15.0, 0.0)); outer_ring2.push_back(ObWkbGeomInnerPoint(15.0, 10.0)); outer_ring2.push_back(ObWkbGeomInnerPoint(5.0, 10.0)); outer_ring2.push_back(ObWkbGeomInnerPoint(5.0, 0.0)); ObCartesianPolygon py2(0, allocator); py2.push_back(outer_ring2); ObCartesianMultipolygon mpy(0, allocator); bool d1 = bg::intersection(py1, py2, mpy); ASSERT_EQ(1U, mpy.size()); ASSERT_EQ(1U, mpy[0].size()); int i = 0; BOOST_FOREACH(ObCartesianPolygon const& p, mpy) { std::cout << i++ << ": " << boost::geometry::area(p) << std::endl; } } void wkt_to_tree_geo(const ObString &wkt, ObArenaAllocator &allocator, ObGeometry *&geo_tree) { ObGeometry *geo = nullptr; ASSERT_EQ(ObWktParser::parse_wkt(allocator, wkt, geo, true, false), OB_SUCCESS); ObGeoToTreeVisitor tree_visitor(&allocator); ASSERT_EQ(geo->do_visit(tree_visitor), OB_SUCCESS); geo_tree = tree_visitor.get_geometry(); } void tree_geo_to_wkt(ObArenaAllocator &allocator, ObGeometry *geo_tree, ObString &wkt_cal) { bool is_geo_empty = false; ASSERT_EQ(sql::ObGeoExprUtils::check_empty(geo_tree, is_geo_empty), OB_SUCCESS); if (is_geo_empty) { wkt_cal = "EMPTY"; } else { ObGeometry *geo_bin = NULL; ASSERT_EQ(ObGeoTypeUtil::tree_to_bin(allocator, geo_tree, geo_bin, nullptr), OB_SUCCESS); ObGeoToWktVisitor visitor(&allocator); ASSERT_EQ(geo_bin->do_visit(visitor), OB_SUCCESS); visitor.get_wkt(wkt_cal); } } void clip_visitor_test(const ObString &wkt, const ObString &wkt_res, ObGeogBox &box) { ObArenaAllocator allocator(ObModIds::TEST); ObGeometry *geo_tree = nullptr; wkt_to_tree_geo(wkt, allocator, geo_tree); ObGeometry *geo_res = nullptr; lib::MemoryContext mem_context; ASSERT_EQ(CURRENT_CONTEXT->CREATE_CONTEXT(mem_context, lib::ContextParam().set_label("GIS_UT")), OB_SUCCESS); ObGeoBoxClipVisitor clip_visitor(box, mem_context); ASSERT_EQ(geo_tree->do_visit(clip_visitor), OB_SUCCESS); ASSERT_EQ(clip_visitor.get_geometry(geo_res), OB_SUCCESS); ASSERT_EQ(geo_res == nullptr, false); ObString wkt_cal; tree_geo_to_wkt(allocator, geo_res, wkt_cal); ASSERT_EQ(wkt_cal == wkt_res, true); } TEST_F(TestGeoTree, clip_visitor) { ObGeogBox box1 = {0, 10, 0, 10, 0, 0}; // linestring clip_visitor_test("LINESTRING(1 1,1 9,9 9,9 1)", "LINESTRING(1 1,1 9,9 9,9 1)", box1); clip_visitor_test("LINESTRING(-1 -9,-1 11,9 11)", "EMPTY", box1); clip_visitor_test("LINESTRING(-1 5,5 5,9 9)", "LINESTRING(0 5,5 5,9 9)", box1); clip_visitor_test("LINESTRING(5 5,8 5,12 5)", "LINESTRING(5 5,8 5,10 5)", box1); clip_visitor_test("LINESTRING(5 -1,5 5,1 2,-3 2,1 6)", "MULTILINESTRING((5 0,5 5,1 2,0 2),(0 5,1 6))", box1); clip_visitor_test("LINESTRING(0 3,0 5,0 7)", "EMPTY", box1); clip_visitor_test("LINESTRING(0 3,0 5,-1 7)", "EMPTY", box1); clip_visitor_test("LINESTRING(0 3,0 5,2 7)", "LINESTRING(0 5,2 7)", box1); clip_visitor_test("LINESTRING(2 1,0 0,1 2)", "LINESTRING(2 1,0 0,1 2)", box1); clip_visitor_test("LINESTRING(3 3,0 3,0 5,2 7)", "MULTILINESTRING((3 3,0 3),(0 5,2 7))", box1); clip_visitor_test("LINESTRING(5 5,10 5,20 5)", "LINESTRING(5 5,10 5)", box1); clip_visitor_test("LINESTRING(3 3,0 6,3 9)", "LINESTRING(3 3,0 6,3 9)", box1); clip_visitor_test("LINESTRING(-1 -1,3 1,3 3,-1 -1)", "LINESTRING(1 0,3 1,3 3,0 0)", box1); // polygon clip_visitor_test("POLYGON((5 5,5 6,6 6,6 5,5 5))", "POLYGON((5 5,5 6,6 6,6 5,5 5))", box1); clip_visitor_test("POLYGON((15 15,15 16,16 16,16 15,15 15))", "EMPTY", box1); clip_visitor_test("POLYGON((-1 -1,-1 11,11 11,11 -1,-1 -1))", "POLYGON((0 0,0 10,10 10,10 0,0 0))", box1); clip_visitor_test("POLYGON((-1 -1,-1 5,5 5,5 -1,-1 -1))", "POLYGON((0 0,0 5,5 5,5 0,0 0))", box1); clip_visitor_test("POLYGON((-2 -2,-2 5,5 5,5 -2,-2 -2), (3 3,4 4,4 2,3 3))", "POLYGON((0 0,0 5,5 5,5 0,0 0),(3 3,4 4,4 2,3 3))", box1); clip_visitor_test("POLYGON((-2 -2,-2 5,5 5,5 -2,-2 -2), (-1 -1,3 1,3 3,-1 -1))", "POLYGON((0 0,0 5,5 5,5 0,1 0,3 1,3 3,0 0))", box1); clip_visitor_test("POLYGON((0 0,10 0,5 10,0 0))", "POLYGON((0 0,5 10,10 0,0 0))", box1); clip_visitor_test("POLYGON((5 10,0 0,10 0,5 10))", "POLYGON((0 0,5 10,10 0,0 0))", box1); clip_visitor_test("POLYGON((-5 -5,5 5,5 -5,-5 -5))", "POLYGON((0 0,5 5,5 0,0 0))", box1); clip_visitor_test("POLYGON((0 0,0 10,10 10,0 0))", "POLYGON((0 0,0 10,10 10,0 0))", box1); clip_visitor_test("POLYGON((0 5,0 10,10 10,0 5))", "POLYGON((0 5,0 10,10 10,0 5))", box1); clip_visitor_test("POLYGON((0 10,10 10,5 0,0 10))", "POLYGON((0 10,10 10,5 0,0 10))", box1); clip_visitor_test("POLYGON((0 10,10 10,5 5,0 10))", "POLYGON((0 10,10 10,5 5,0 10))", box1); clip_visitor_test("POLYGON((0 10,5 10,0 5,0 10))", "POLYGON((0 5,0 10,5 10,0 5))", box1); clip_visitor_test("POLYGON((0 10,10 5,0 5,0 10))", "POLYGON((0 5,0 10,10 5,0 5))", box1); clip_visitor_test("POLYGON((0 10,10 0,0 5,0 10))", "POLYGON((0 5,0 10,10 0,0 5))", box1); clip_visitor_test("POLYGON((0 10,5 0,0 5,0 10))", "POLYGON((0 5,0 10,5 0,0 5))", box1); clip_visitor_test("POLYGON((0 10,5 5,0 5,0 10))", "POLYGON((0 5,0 10,5 5,0 5))", box1); clip_visitor_test("POLYGON((0 10,7 7,3 3,0 10))", "POLYGON((0 10,7 7,3 3,0 10))", box1); clip_visitor_test("POLYGON((0 10,5 5,5 0,0 10))", "POLYGON((0 10,5 5,5 0,0 10))", box1); clip_visitor_test("POLYGON((0 10,10 5,5 0,0 10))", "POLYGON((0 10,10 5,5 0,0 10))", box1); clip_visitor_test("POLYGON((2 5,5 10,7 5,2 5))", "POLYGON((2 5,5 10,7 5,2 5))", box1); clip_visitor_test("POLYGON((0 5,5 10,5 5,0 5))", "POLYGON((0 5,5 10,5 5,0 5))", box1); clip_visitor_test("POLYGON((0 5,5 10,10 5,0 5))", "POLYGON((0 5,5 10,10 5,0 5))", box1); clip_visitor_test("POLYGON((0 5,5 7,10 5,0 5))", "POLYGON((0 5,5 7,10 5,0 5))", box1); clip_visitor_test("POLYGON((-5 10,0 15,0 10,-5 10))", "EMPTY", box1); clip_visitor_test("POLYGON((-5 10,0 5,-5 0,-5 10))", "EMPTY", box1); clip_visitor_test("POLYGON((-5 5,0 10,0 0,-5 5))", "EMPTY", box1); clip_visitor_test("POLYGON((-5 5,0 10,0 5,-5 5))", "EMPTY", box1); clip_visitor_test("POLYGON((-5 5,0 7,0 3,-5 5))", "EMPTY", box1); clip_visitor_test("POLYGON((5 5,-5 0,-5 10,5 5))", "POLYGON((0 2.5,0 7.5,5 5,0 2.5))", box1); clip_visitor_test("POLYGON((5 0,-5 0,-5 10,5 0))", "POLYGON((0 0,0 5,5 0,0 0))", box1); clip_visitor_test("POLYGON((10 0,-10 0,-10 10,10 0))", "POLYGON((0 0,0 5,10 0,0 0))", box1); clip_visitor_test("POLYGON((5 0,-5 5,-5 10,5 0))", "POLYGON((0 2.5,0 5,5 0,0 2.5))", box1); clip_visitor_test("POLYGON((10 5,-10 0,-10 10,10 5))", "POLYGON((0 2.5,0 7.5,10 5,0 2.5))", box1); clip_visitor_test("POLYGON((10 10,-10 0,-10 5,10 10))", "POLYGON((0 5,0 7.5,10 10,0 5))", box1); clip_visitor_test("POLYGON((5 5,-5 -5,-5 15,5 5))", "POLYGON((0 0,0 10,5 5,0 0))", box1); clip_visitor_test("POLYGON((10 5,-10 -5,-10 15,10 5))", "POLYGON((0 0,0 10,10 5,0 0))", box1); clip_visitor_test("POLYGON((5 0,-5 0,-5 20,5 0))", "POLYGON((0 0,0 10,5 0,0 0))", box1); clip_visitor_test("POLYGON((10 0,-10 0,-10 20,10 0))", "POLYGON((0 0,0 10,10 0,0 0))", box1); clip_visitor_test("POLYGON((5 5,-10 5,0 15,5 5))", "POLYGON((0 5,0 10,2.5 10,5 5,0 5))", box1); clip_visitor_test("POLYGON((5 5,-5 -5,0 15,5 5))", "POLYGON((0 0,0 10,2.5 10,5 5,0 0))", box1); clip_visitor_test("POLYGON((5 5,-15 -20,-15 30,5 5))", "POLYGON((0 0,0 10,1 10,5 5,1 0,0 0))", box1); clip_visitor_test("POLYGON((5 7,5 3,-5 5,5 7))", "POLYGON((0 4,0 6,5 7,5 3,0 4))", box1); clip_visitor_test("POLYGON((5 7,5 3,-5 13,5 7))", "POLYGON((0 8,0 10,5 7,5 3,0 8))", box1); clip_visitor_test("POLYGON((6 6,4 4,-4 14,6 6))", "POLYGON((0 9,0 10,1.0000000000000007 10,6 6,4 4,0 9))", box1); clip_visitor_test("POLYGON((-2 -2,-2 12,12 12,12 -2,-2 -2),(-1 -1,11 -1,11 11,-1 11,-1 -1))", "EMPTY", box1); clip_visitor_test("POLYGON((-2 -2,-2 12,12 12,12 -2,-2 -2),(1 1,9 1,9 9,1 9,1 1))", "POLYGON((0 0,0 10,10 10,10 0,0 0),(1 1,9 1,9 9,1 9,1 1))", box1); clip_visitor_test("POLYGON((5 5,15 5,15 -5,5 -5,5 5),(8 1,8 -1,9 -1,9 1,8 1))", "POLYGON((5 0,5 5,10 5,10 0,9 0,9 1,8 1,8 0,5 0))", box1); clip_visitor_test("POLYGON((-6 5,5 5,5 -6,-6 5))", "POLYGON((0 0,0 5,5 5,5 0,0 0))", box1); clip_visitor_test("POLYGON((-15 -15,-15 15,15 15,15 -15,-15 -15),(-5 5,-5 -5,5 -5,5 5,-5 5))", "POLYGON((0 5,0 10,10 10,10 0,5 0,5 5,0 5))", box1); clip_visitor_test("POLYGON((-15 -15,-15 15,15 15,15 -15,-15 -15),(-6 5,5 -6,5 5,-6 5))", "POLYGON((0 5,0 10,10 10,10 0,5 0,5 5,0 5))", box1); clip_visitor_test("POLYGON((-15 -15,-15 15,15 15,15 -15,-15 -15),(-5 5,-6 5,-6 6,-5 6,-5 5))", "POLYGON((0 0,0 10,10 10,10 0,0 0))", box1); clip_visitor_test("POLYGON((-15 -15,-15 15,15 15,15 -15,-15 -15),(0 5,-1 5,-1 6,0 6,0 5))", "POLYGON((0 0,0 10,10 10,10 0,0 0))", box1); ObGeogBox box2 = {10, 100, 10, 100, 0, 0}; clip_visitor_test("POLYGON((50 50,200 50,200 200,50 200,50 50))", // CCW "POLYGON((50 50,50 100,100 100,100 50,50 50))", box2); clip_visitor_test("POLYGON((50 50,50 200,200 200,200 50,50 50))", // CW "POLYGON((50 50,50 100,100 100,100 50,50 50))", box2); // box1 clip_visitor_test("POLYGON(" "(-10 2,-10 8,8 8,8 2,-10 2)," // CW "(-5 6,-5 4,5 4,5 6,-5 6)" // CCW ")", "POLYGON((0 2,0 4,5 4,5 6,0 6,0 8,8 8,8 2,0 2))", box1); clip_visitor_test("POLYGON(" "(-10 2,-10 8,8 8,8 2,-10 2)," // CW "(-5 6,5 6,5 4,-5 4,-5 6)" // CW ")", "POLYGON((0 2,0 4,5 4,5 6,0 6,0 8,8 8,8 2,0 2))", box1); clip_visitor_test("POLYGON(" "(-10 2,8 2,8 8,-10 8,-10 2)," // CCW "(-5 6,5 6,5 4,-5 4,-5 6)" // CW ")", "POLYGON((0 2,0 4,5 4,5 6,0 6,0 8,8 8,8 2,0 2))", box1); clip_visitor_test("POLYGON(" "(-10 -10,-10 20,20 20,20 -10,-10 -10)," // CW "(-5 -5,0 -5,0 0,-5 0,-5 -5)" // CCW ")", "POLYGON((0 0,0 10,10 10,10 0,0 0))", box1); ObGeogBox box3 = {3.0481343214686657e-14, 20000000.000000, -20000000.000000, -1.000000, 0, 0}; clip_visitor_test("POLYGON((3.0481343214686657e-14 -20000000, 200000000 -20000000, 200000000 -1, 3.0481343214686657e-14 -1, 3.0481343214686657e-14 -20000000))", "POLYGON((0.000000000000030481343214686657 -20000000,0.000000000000030481343214686657 -1,20000000 -1,20000000 -20000000,0.000000000000030481343214686657 -20000000))", box3); // EMPTY clip_visitor_test("GEOMETRYCOLLECTION EMPTY", "EMPTY", box1); } void affine_visitor_test(const ObString &wkt, const ObString &wkt_res, ObAffineMatrix &affine) { ObArenaAllocator allocator(ObModIds::TEST); ObGeometry *geo_tree = nullptr; wkt_to_tree_geo(wkt, allocator, geo_tree); ASSERT_EQ(ObGeoMVTUtil::affine_transformation(geo_tree, affine), OB_SUCCESS); ObString wkt_cal; tree_geo_to_wkt(allocator, geo_tree, wkt_cal); ASSERT_EQ(wkt_cal == wkt_res, true); } TEST_F(TestGeoTree, affine_visitor) { ObAffineMatrix affine = {1, 0, 0, 0, -1, 0, 0, 0, 1, 0, 4096, 0}; affine_visitor_test("MULTILINESTRING((1 1, 501 501, 1001 1001),(2 2, 502 502, 1002 1002))", "MULTILINESTRING((1 4095,501 3595,1001 3095),(2 4094,502 3594,1002 3094))", affine); affine_visitor_test("POLYGON ((0 0, 10 0, 10 5, 0 -5, 0 0))", "POLYGON((0 4096,10 4096,10 4091,0 4101,0 4096))", affine); ObAffineMatrix affine2 = {1, 0, 0, 0, -1, 0, 0, 0, 1, 0, 100, 0}; affine_visitor_test("LINESTRING(0 0, 2 20, -2 40, -4 60, 4 80, 0 100)", "LINESTRING(0 100,2 80,-2 60,-4 40,4 20,0 0)", affine2); ObAffineMatrix affine3 = {1, 0, 0, 0, -1, 0, 0, 0, 1, 0, 10, 0}; affine_visitor_test("POLYGON((10 10, 10 0, 0 0, 0 10, 10 10),(9 9, 1 9, 1 1, 9 1, 9 9),(8 8, 8 2, 2 2, 2 8, 8 8),(7 7, 7 3, 3 3, 3 7, 7 7))", "POLYGON((10 0,10 10,0 10,0 0,10 0),(9 1,1 1,1 9,9 9,9 1),(8 2,8 8,2 8,2 2,8 2),(7 3,7 7,3 7,3 3,7 3))", affine3); // POLYGON((-8231396.69199339 4979982.17443372,-8231396.69199339 4980355.83678553,-8231365.02893734 4980355.83678553,-8231365.02893734 4979982.17443372,-8231396.69199339 4979982.17443372)) ObAffineMatrix affine4 = {0.41539995335806329, 0, 0, 0, -0.41539995335802404, 0, 0, 0, 1, 3422112.7441855143, 2068718.1395343679, 0}; affine_visitor_test("MULTIPOLYGON(((-8231365.02893734 4980355.83678553,-8231394.82332406 4980186.31880185,-8231367.43081065 4979982.17443372,-8231396.69199339 4980227.59327083,-8231365.02893734 4980355.83678553)))", "MULTIPOLYGON(((2804.095091749914 -121.44277270394377,2791.718504895922 -51.02501018997282,2803.097353688907 33.77655080938712,2790.942259743344 -68.17042267904617,2804.095091749914 -121.44277270394377)))", affine4); } void grid_visitor_test(const ObString &wkt, const ObString &wkt_res, double size) { ObArenaAllocator allocator(ObModIds::TEST); ObGeometry *geo_tree = nullptr; wkt_to_tree_geo(wkt, allocator, geo_tree); ObGeoGrid grid = {0, 0, 0, 0, 0, 0}; grid.x_size = grid.y_size = size; ASSERT_EQ(ObGeoMVTUtil::snap_to_grid(geo_tree, grid, false), OB_SUCCESS); ObString wkt_cal; tree_geo_to_wkt(allocator, geo_tree, wkt_cal); ASSERT_EQ(wkt_cal == wkt_res, true); } TEST_F(TestGeoTree, grid_visitor) { grid_visitor_test("LINESTRING(0 100,2 80,-2 60,-4 40,4 20,0 0)", "LINESTRING(0 100,2 80,-2 60,-4 40,4 20,0 0)", 1); grid_visitor_test("LINESTRING(0 0,1 1,2 2,3 3,4 4,5 5)", "LINESTRING(0 0,2 2,4 4)", 2); grid_visitor_test("POINT(5.1423999999 5.1423999999)", "POINT(5.1424 5.1424)", 0.0001); grid_visitor_test("POINT(5 5)", "POINT(0 0)", 20); grid_visitor_test("MULTIPOLYGON(((0 0,10 0,10 10,0 10,0 0)))", "EMPTY", 20); // different from PG, PG keeps duplicate point in multipoint grid_visitor_test("MULTIPOINT(0 0,1 1, 2 2, 3 3, 4 4, 5 5)", "MULTIPOINT((0 0),(2 2),(4 4))", 2); grid_visitor_test("MULTIPOLYGON(((0 0,10 0,10 10,0 10,0 0),(4 4, 4 5, 5 5, 5 4, 4 4)))", "MULTIPOLYGON(((0 0,10 0,10 10,0 10,0 0)))", 2); grid_visitor_test("MULTIPOLYGON(((0 0,10 0,10 10,0 10,0 0),(4 4, 4 5, 5 5, 5 4, 4 4)))", "EMPTY", 20); grid_visitor_test("MULTIPOLYGON(((2804.095091749914 -121.44277270394377,2791.718504895922 -51.02501018997282,2803.097353688907 33.77655080938712,2790.942259743344 -68.17042267904617,2804.095091749914 -121.44277270394377)))", "MULTIPOLYGON(((2804 -121,2792 -51,2803 34,2791 -68,2804 -121)))", 1); } void simplify_visitor_test(const ObString &wkt, const ObString &wkt_res, double tolerance, bool keep_collapsed) { ObArenaAllocator allocator(ObModIds::TEST); ObGeometry *geo_tree = nullptr; wkt_to_tree_geo(wkt, allocator, geo_tree); ASSERT_EQ(ObGeoMVTUtil::simplify_geometry(geo_tree, tolerance, keep_collapsed), OB_SUCCESS); ObString wkt_cal; tree_geo_to_wkt(allocator, geo_tree, wkt_cal); ASSERT_EQ(wkt_cal == wkt_res, true); } TEST_F(TestGeoTree, simplify_visitor) { simplify_visitor_test("POLYGON((10 0,10 10,0 10,0 0,10 0),(9 1,1 1,1 9,9 9,9 1),(8 2,8 8,2 8,2 2,8 2),(7 3,7 7,3 7,3 3,7 3))", "POLYGON((10 0,10 10,0 10,0 0,10 0),(9 1,1 1,1 9,9 9,9 1),(8 2,8 8,2 8,2 2,8 2),(7 3,7 7,3 7,3 3,7 3))", 0, false); simplify_visitor_test("MULTILINESTRING((1 4095,501 3595,1001 3095),(2 4094,502 3594,1002 3094))", "MULTILINESTRING((1 4095,1001 3095),(2 4094,1002 3094))", 0, false); simplify_visitor_test("LINESTRING(0 0, 1 0, 1 1, 0 1, 0 0)", "LINESTRING(0 0,0 0)", 10, true); simplify_visitor_test("POLYGON((0 0, 1 0, 1 1, 0 1, 0 0))", "POLYGON((0 0,1 0,1 1,0 0))", 10, true); simplify_visitor_test("LINESTRING(0 0, 1 0, 1 1, 0 1, 0 0)", "EMPTY", 10, false); simplify_visitor_test("POLYGON((0 0, 1 0, 1 1, 0 1, 0 0))", "EMPTY", 10, false); simplify_visitor_test("LINESTRING(0 0, 50 1.00001, 100 0)", "LINESTRING(0 0,50 1.00001,100 0)", 1, false); simplify_visitor_test("LINESTRING(0 0,50 0.99999,100 0)", "LINESTRING(0 0,100 0)", 1, false); simplify_visitor_test("POLYGON(" "(0 0, 100 0, 100 100, 0 100, 0 0)," "(1 1, 1 5, 5 5, 5 1, 1 1)," "(20 20, 20 40, 40 40, 40 20, 20 20)" ")", "POLYGON((0 0,100 0,100 100,0 100,0 0),(20 20,20 40,40 40,40 20,20 20))", 10, false); simplify_visitor_test("POLYGON(" "(0 0, 100 0, 100 100, 0 100, 0 0)," "(20 20, 20 40, 40 40, 40 20, 20 20)," "(1 1, 1 5, 5 5, 5 1, 1 1)" ")", "POLYGON((0 0,100 0,100 100,0 100,0 0),(20 20,20 40,40 40,40 20,20 20))", 10, false); simplify_visitor_test("POLYGON(" "(0 0, 100 0, 100 100, 0 100, 0 0)," "(20 20, 20 40, 40 40, 40 20, 20 20)," "(1 1, 1 5, 5 5, 5 1, 1 1)" ")", "EMPTY", 100, false); } TEST_F(TestGeoTree, ewkt_with_null) { ObArenaAllocator allocator(ObModIds::TEST); ObGeometry *geo_tree = nullptr; wkt_to_tree_geo("POINT(1 1)", allocator, geo_tree); geo_tree->set_srid(UINT32_MAX); ObGeometry *geo_bin = NULL; ASSERT_EQ(ObGeoTypeUtil::tree_to_bin(allocator, geo_tree, geo_bin, nullptr), OB_SUCCESS); ObWkbBuffer buffer(allocator); ASSERT_EQ(buffer.append(static_cast(UINT32_MAX)), OB_SUCCESS); ASSERT_EQ(buffer.append(geo_bin->val(), geo_bin->length()), OB_SUCCESS); ObString wkt_cal; ASSERT_EQ(ObGeoTypeUtil::geo_to_ewkt(buffer.string(), wkt_cal, allocator, 14), OB_SUCCESS); ObString wkt_res = "SRID=NULL;POINT(1 1)"; ASSERT_EQ(wkt_cal == wkt_res, true); } TEST_F(TestGeoTree, ewkt_with_0) { ObArenaAllocator allocator(ObModIds::TEST); ObGeometry *geo_tree = nullptr; wkt_to_tree_geo("POINT(1 1)", allocator, geo_tree); geo_tree->set_srid(0); ObGeometry *geo_bin = NULL; ASSERT_EQ(ObGeoTypeUtil::tree_to_bin(allocator, geo_tree, geo_bin, nullptr), OB_SUCCESS); ObWkbBuffer buffer(allocator); ASSERT_EQ(buffer.append(static_cast(0)), OB_SUCCESS); ASSERT_EQ(buffer.append(geo_bin->val(), geo_bin->length()), OB_SUCCESS); ObString wkt_cal; ASSERT_EQ(ObGeoTypeUtil::geo_to_ewkt(buffer.string(), wkt_cal, allocator, 14, true), OB_SUCCESS); ObString wkt_res = "SRID=0;POINT(1 1)"; ASSERT_EQ(wkt_cal == wkt_res, true); } } // namespace common } // namespace oceanbase int main(int argc, char** argv) { ::testing::InitGoogleTest(&argc, argv); // system("rm -f test_geo_tree.log"); // OB_LOGGER.set_file_name("test_geo_tree.log"); OB_LOGGER.set_log_level("DEBUG"); return RUN_ALL_TESTS(); }