[FEAT MERGE]Patch Spatial Relationship Computation Performance Optimization & Implement of oracle spatial index

unittest/share/test_geo_bin.cpp

@@ -28,6 +28,9 @@
 #include "lib/geo/ob_geo_longtitude_correct_visitor.h"
 #include "lib/geo/ob_geo_to_tree_visitor.h"
 #include "lib/geo/ob_geo_reverse_coordinate_visitor.h"
+#include "lib/geo/ob_geo_segment_collect_visitor.h"
+#include "lib/geo/ob_geo_vertex_collect_visitor.h"
+#include "lib/geo/ob_geo_cache.h"
 #include "lib/geo/ob_srs_info.h"
 #include "lib/geo/ob_geo_utils.h"
 #include "lib/geo/ob_sdo_geo_func_to_wkb.h"
@@ -6105,6 +6108,121 @@ TEST_F(TestGeoBin, elevation_visitor) {
         "POLYGON ((1 9, 9 9, 9 1, 1 1, 1 9))",
         "POLYGON Z ((1 9 5,9 9 10,9 1 5,1 1 0,1 9 5))");
 }
+TEST_F(TestGeoBin, collect_visitor_poly)
+{
+    ObArenaAllocator allocator(ObModIds::TEST);
+    ObJsonBuffer data(&allocator);
+
+    ASSERT_EQ(OB_SUCCESS, append_bo(data));
+    ASSERT_EQ(OB_SUCCESS, append_type(data, ObGeoType::POLYGON));
+    uint32_t ring_num = 2;
+    ASSERT_EQ(OB_SUCCESS, append_uint32(data, ring_num));
+    for (uint32_t i = 0; i < ring_num; i++) {
+        ASSERT_EQ(OB_SUCCESS, append_uint32(data, 3));
+        ASSERT_EQ(OB_SUCCESS, append_double(data, 181));
+        ASSERT_EQ(OB_SUCCESS, append_double(data, 90));
+        ASSERT_EQ(OB_SUCCESS, append_double(data, 10));
+        ASSERT_EQ(OB_SUCCESS, append_double(data, 20));
+        ASSERT_EQ(OB_SUCCESS, append_double(data, 30));
+        ASSERT_EQ(OB_SUCCESS, append_double(data, 40));
+    }
+
+    ObIWkbGeogPolygon iwkb_geog;
+    iwkb_geog.set_data(data.string());
+    ObVertexes vertexes;
+    ObGeoVertexCollectVisitor vertex_visitor(vertexes);
+    ObLineSegments segments;
+    ObGeoSegmentCollectVisitor seg_visitor(&segments);
+    ASSERT_EQ(OB_SUCCESS, iwkb_geog.do_visit(vertex_visitor));
+    ASSERT_EQ(OB_SUCCESS, iwkb_geog.do_visit(seg_visitor));
+
+    ASSERT_EQ(6, vertexes.size());
+    for (uint32_t i = 0; i < ring_num; i++) {
+      ASSERT_EQ(181, vertexes[i * 3].x);
+      ASSERT_EQ(90, vertexes[i * 3].y);
+      ASSERT_EQ(10, vertexes[i * 3 + 1].x);
+      ASSERT_EQ(20, vertexes[i * 3 + 1].y);
+      ASSERT_EQ(30, vertexes[i * 3 + 2].x);
+      ASSERT_EQ(40, vertexes[i * 3 + 2].y);
+    }
+    for (uint32_t i = 0; i < segments.segs_.size(); i++) {
+      uint32_t start_idx = segments.segs_[i].begin;
+      uint32_t end_idx = segments.segs_[i].end;
+      std::cout << "linestring(" << i << ": ";
+      for (; start_idx <= end_idx; start_idx++) {
+        std::cout << segments.verts_[start_idx].x << " " <<  segments.verts_[start_idx].y << ",";
+      }
+      std::cout << ")" << std::endl;
+    }
+}
+
+
+TEST_F(TestGeoBin, linesegments_collect_visitor_poly)
+{
+    ObArenaAllocator allocator(ObModIds::TEST);
+    ObJsonBuffer data(&allocator);
+
+    /*POLYGON((121.48017238084356 31.227565780172494,
+               121.48388535915394 31.22429103925363,
+               121.48462793947535 31.221579898137527,
+               121.48242817125481 31.21807789724289,
+               121.47703361713795 31.222376670115178,
+               121.48017238084356 31.227565780172494),
+              (121.479 31.223,
+               121.480 31.2233,
+               121.48088 31.2238,
+               121.48098 31.2234,
+               121.4813 31.223,
+               121.479 31.223))*/
+
+    ASSERT_EQ(OB_SUCCESS, append_bo(data));
+    ASSERT_EQ(OB_SUCCESS, append_type(data, ObGeoType::POLYGON));
+    uint32_t ring_num = 2;
+    ASSERT_EQ(OB_SUCCESS, append_uint32(data, ring_num));
+    ASSERT_EQ(OB_SUCCESS, append_uint32(data, 6));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 121.48017238084356));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 31.227565780172494));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 121.48388535915394));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 31.22429103925363));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 121.48462793947535));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 31.221579898137527));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 121.48242817125481));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 31.21807789724289));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 121.47703361713795));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 31.222376670115178));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 121.48017238084356));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 31.227565780172494));
+
+    ASSERT_EQ(OB_SUCCESS, append_uint32(data, 6));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 121.479));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 31.223));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 121.480));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 31.2233));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 121.48088));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 31.2238));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 121.48098));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 31.2234));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 121.4813));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 31.223));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 121.479));
+    ASSERT_EQ(OB_SUCCESS, append_double(data, 31.223));
+
+    ObIWkbGeogPolygon iwkb_geog;
+    iwkb_geog.set_data(data.string());
+    ObLineSegments segments;
+    ObGeoSegmentCollectVisitor seg_visitor(&segments);
+    ASSERT_EQ(OB_SUCCESS, iwkb_geog.do_visit(seg_visitor));
+    for (uint32_t i = 0; i < segments.segs_.size(); i++) {
+      uint32_t start_idx = segments.segs_[i].begin;
+      uint32_t end_idx = segments.segs_[i].end;
+      std::cout << "linestring(" << i << ": ";
+      for (; start_idx <= end_idx; start_idx++) {
+        std::cout << segments.verts_[start_idx].x << " " <<  segments.verts_[start_idx].y << ",";
+      }
+      std::cout << ")" << std::endl;
+    }
+}
+
 } // namespace common
 } // namespace oceanbase