Add ST_OffsetCurve to Snowflake UDFs and tests

Author: jiayuasu

snowflake-tester/src/test/java/org/apache/sedona/snowflake/snowsql/TestFunctions.java

@@ -768,6 +768,18 @@ public void test_ST_NPoints() {
         "select sedona.ST_NPoints(sedona.ST_GeomFromText('LINESTRING(1 2, 3 4, 5 6)'))", 3);
   }
 
+  @Test
+  public void test_ST_OffsetCurve() {
+    registerUDF("ST_OffsetCurve", byte[].class, double.class);
+    verifySqlSingleRes(
+        "select sedona.ST_AsText(sedona.ST_OffsetCurve(sedona.ST_GeomFromText('LINESTRING(0 0, 10 0)'), 5.0))",
+        "LINESTRING (0 5, 10 5)");
+    registerUDF("ST_OffsetCurve", byte[].class, double.class, int.class);
+    verifySqlSingleRes(
+        "select sedona.ST_AsText(sedona.ST_OffsetCurve(sedona.ST_GeomFromText('LINESTRING(0 0, 10 0)'), 5.0, 4))",
+        "LINESTRING (0 5, 10 5)");
+  }
+
   @Test
   public void test_ST_NumGeometries() {
     registerUDF("ST_NumGeometries", byte[].class);

snowflake-tester/src/test/java/org/apache/sedona/snowflake/snowsql/TestFunctionsV2.java

@@ -716,6 +716,18 @@ public void test_ST_NPoints() {
         "select sedona.ST_NPoints(ST_GeometryFromWKT('LINESTRING(1 2, 3 4, 5 6)'))", 3);
   }
 
+  @Test
+  public void test_ST_OffsetCurve() {
+    registerUDFV2("ST_OffsetCurve", String.class, double.class);
+    verifySqlSingleRes(
+        "select sedona.ST_AsText(sedona.ST_OffsetCurve(ST_GeometryFromWKT('LINESTRING(0 0, 10 0)'), 5.0))",
+        "LINESTRING (0 5, 10 5)");
+    registerUDFV2("ST_OffsetCurve", String.class, double.class, int.class);
+    verifySqlSingleRes(
+        "select sedona.ST_AsText(sedona.ST_OffsetCurve(ST_GeometryFromWKT('LINESTRING(0 0, 10 0)'), 5.0, 4))",
+        "LINESTRING (0 5, 10 5)");
+  }
+
   @Test
   public void test_ST_NumGeometries() {
     registerUDFV2("ST_NumGeometries", String.class);

snowflake/src/main/java/org/apache/sedona/snowflake/snowsql/UDFs.java

@@ -798,6 +798,18 @@ public static byte[] ST_Normalize(byte[] geometry) {
     return GeometrySerde.serialize(Functions.normalize(GeometrySerde.deserialize(geometry)));
   }
 
+  @UDFAnnotations.ParamMeta(argNames = {"geometry", "distance"})
+  public static byte[] ST_OffsetCurve(byte[] geometry, double distance) {
+    return GeometrySerde.serialize(
+        Functions.offsetCurve(GeometrySerde.deserialize(geometry), distance));
+  }
+
+  @UDFAnnotations.ParamMeta(argNames = {"geometry", "distance", "quadrantSegments"})
+  public static byte[] ST_OffsetCurve(byte[] geometry, double distance, int quadrantSegments) {
+    return GeometrySerde.serialize(
+        Functions.offsetCurve(GeometrySerde.deserialize(geometry), distance, quadrantSegments));
+  }
+
   @UDFAnnotations.ParamMeta(argNames = {"geometry"})
   public static int ST_NumGeometries(byte[] geometry) {
     return Functions.numGeometries(GeometrySerde.deserialize(geometry));

snowflake/src/main/java/org/apache/sedona/snowflake/snowsql/UDFsV2.java

@@ -961,6 +961,24 @@ public static String ST_Normalize(String geometry) {
     return GeometrySerde.serGeoJson(Functions.normalize(GeometrySerde.deserGeoJson(geometry)));
   }
 
+  @UDFAnnotations.ParamMeta(
+      argNames = {"geometry", "distance"},
+      argTypes = {"Geometry", "double"},
+      returnTypes = "Geometry")
+  public static String ST_OffsetCurve(String geometry, double distance) {
+    return GeometrySerde.serGeoJson(
+        Functions.offsetCurve(GeometrySerde.deserGeoJson(geometry), distance));
+  }
+
+  @UDFAnnotations.ParamMeta(
+      argNames = {"geometry", "distance", "quadrantSegments"},
+      argTypes = {"Geometry", "double", "int"},
+      returnTypes = "Geometry")
+  public static String ST_OffsetCurve(String geometry, double distance, int quadrantSegments) {
+    return GeometrySerde.serGeoJson(
+        Functions.offsetCurve(GeometrySerde.deserGeoJson(geometry), distance, quadrantSegments));
+  }
+
   @UDFAnnotations.ParamMeta(
       argNames = {"geometry"},
       argTypes = {"Geometry"})