hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
aosc-generic-25025-dev
oceanbase/unittest/share/test_geo_tree.cpp

964 lines
36 KiB
C++
Raw Permalink Blame History

/**
* 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 <gtest/gtest.h>
#include <boost/geometry.hpp>
#include <boost/foreach.hpp>
#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 <sys/time.h>
#include <stdexcept>
#include <exception>
#include <typeinfo>
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<int> gv(page_allocator_);
const ObGeomVector<int> &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<int>::iterator iter = gv.begin();
ObGeomVector<int>::const_iterator c_iter = c_gv.begin();
ObGeomVector<int>::iterator begin = gv.begin();
ObGeomVector<int>::const_iterator c_begin = c_gv.begin();
ObGeomVector<int>::iterator end = gv.end();
ObGeomVector<int>::const_iterator c_end = c_gv.end();
ObGeomVector<int>::iterator tmp_iter;
ObGeomVector<int>::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<int> 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<int> &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<ObCartesianLineString *>(l);
ASSERT_EQ(ObGeoType::LINESTRING, line->type());
ASSERT_EQ(1, line->dimension());
uint32_t num = 10000;
common::ObVector<double> xv;
common::ObVector<double> yv;
for (int i = 0; i < num; i++) {
double x = static_cast<double>(rand())/static_cast<double>(rand());
double y = static_cast<double>(rand())/static_cast<double>(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<ObCartesianLinearring *>(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<ObCartesianPolygon *>(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<ObCartesianGeometrycollection *>(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<ObCartesianMultipoint *>(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<ObCartesianMultilinestring *>(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<ObCartesianMultipolygon *>(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_t>(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<uint32_t>(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();
}
Reference in New Issue View Git Blame Copy Permalink