This is an automated email from the ASF dual-hosted git repository.
jiayu pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/sedona.git
The following commit(s) were added to refs/heads/master by this push:
new 74114ad1 [SEDONA-299] Add ST_FrechetDistance (#869)
74114ad1 is described below
commit 74114ad14a747a5d8fa8408293780f8cbfb743f3
Author: Nilesh Gajwani <[email protected]>
AuthorDate: Mon Jun 26 06:44:58 2023 -0700
[SEDONA-299] Add ST_FrechetDistance (#869)
---
.../java/org/apache/sedona/common/Functions.java | 4 ++
.../org/apache/sedona/common/utils/GeomUtils.java | 7 +++-
.../org/apache/sedona/common/FunctionsTest.java | 43 +++++++++++++++++++++-
docs/api/flink/Function.md | 20 ++++++++++
docs/api/sql/Function.md | 20 ++++++++++
.../main/java/org/apache/sedona/flink/Catalog.java | 1 +
.../apache/sedona/flink/expressions/Functions.java | 9 +++++
.../java/org/apache/sedona/flink/FunctionTest.java | 8 ++++
python/sedona/sql/st_functions.py | 13 +++++++
python/tests/sql/test_dataframe_api.py | 1 +
python/tests/sql/test_function.py | 7 ++++
.../scala/org/apache/sedona/sql/UDF/Catalog.scala | 1 +
.../sql/sedona_sql/expressions/Functions.scala | 7 ++++
.../sql/sedona_sql/expressions/st_functions.scala | 3 ++
.../apache/sedona/sql/dataFrameAPITestScala.scala | 10 ++++-
.../org/apache/sedona/sql/functionTestScala.scala | 16 ++++++++
16 files changed, 166 insertions(+), 4 deletions(-)
diff --git a/common/src/main/java/org/apache/sedona/common/Functions.java
b/common/src/main/java/org/apache/sedona/common/Functions.java
index 13cfc619..5b8ea3e6 100644
--- a/common/src/main/java/org/apache/sedona/common/Functions.java
+++ b/common/src/main/java/org/apache/sedona/common/Functions.java
@@ -955,6 +955,10 @@ public class Functions {
return geometricMedian(geometry, DEFAULT_TOLERANCE, DEFAULT_MAX_ITER,
false);
}
+ public static double frechetDistance(Geometry g1, Geometry g2) {
+ return GeomUtils.getFrechetDistance(g1, g2);
+ }
+
public static Geometry boundingDiagonal(Geometry geometry) {
if (geometry.isEmpty()) {
return GEOMETRY_FACTORY.createLineString();
diff --git a/common/src/main/java/org/apache/sedona/common/utils/GeomUtils.java
b/common/src/main/java/org/apache/sedona/common/utils/GeomUtils.java
index 2a210f56..1a3c5baf 100644
--- a/common/src/main/java/org/apache/sedona/common/utils/GeomUtils.java
+++ b/common/src/main/java/org/apache/sedona/common/utils/GeomUtils.java
@@ -26,6 +26,7 @@ import org.locationtech.jts.io.WKBWriter;
import org.locationtech.jts.io.WKTWriter;
import org.locationtech.jts.operation.polygonize.Polygonizer;
import org.locationtech.jts.operation.union.UnaryUnionOp;
+import org.locationtech.jts.algorithm.distance.DiscreteFrechetDistance;
import org.locationtech.jts.algorithm.distance.DiscreteHausdorffDistance;
import java.nio.ByteOrder;
@@ -424,7 +425,6 @@ public class GeomUtils {
return geometries;
}
-
public static Geometry get3DGeom(Geometry geometry, double zValue) {
Coordinate[] coordinates = geometry.getCoordinates();
if (coordinates.length == 0) return geometry;
@@ -481,6 +481,11 @@ public class GeomUtils {
geometry.geometryChanged();
}
+ public static double getFrechetDistance(Geometry g1, Geometry g2) {
+ if (g1.isEmpty() || g2.isEmpty()) return 0.0;
+ return DiscreteFrechetDistance.distance(g1, g2);
+ }
+
public static Double getHausdorffDistance(Geometry g1, Geometry g2, double
densityFrac) throws Exception {
if (g1.isEmpty() || g2.isEmpty()) return 0.0;
DiscreteHausdorffDistance hausdorffDistanceObj = new
DiscreteHausdorffDistance(g1, g2);
diff --git a/common/src/test/java/org/apache/sedona/common/FunctionsTest.java
b/common/src/test/java/org/apache/sedona/common/FunctionsTest.java
index 3dde5180..16970a47 100644
--- a/common/src/test/java/org/apache/sedona/common/FunctionsTest.java
+++ b/common/src/test/java/org/apache/sedona/common/FunctionsTest.java
@@ -918,10 +918,10 @@ public class FunctionsTest {
public void translateHybridGeomCollectionDeltaZ() {
Polygon polygon = GEOMETRY_FACTORY.createPolygon(coordArray(1, 0, 1,
1, 2, 1, 2, 0, 1, 0));
Polygon polygon3D = GEOMETRY_FACTORY.createPolygon(coordArray3d(1, 0,
1, 2, 0, 2, 2, 1, 2, 1, 0, 1));
- MultiPolygon multiPolygon = GEOMETRY_FACTORY.createMultiPolygon(new
Polygon[] {polygon3D, polygon});
+ MultiPolygon multiPolygon = GEOMETRY_FACTORY.createMultiPolygon(new
Polygon[]{polygon3D, polygon});
Point point3D = GEOMETRY_FACTORY.createPoint(new Coordinate(1, 1, 1));
LineString emptyLineString = GEOMETRY_FACTORY.createLineString();
- Geometry geomCollection =
GEOMETRY_FACTORY.createGeometryCollection(new Geometry[]
{GEOMETRY_FACTORY.createGeometryCollection(new Geometry[] {multiPolygon,
point3D, emptyLineString})});
+ Geometry geomCollection =
GEOMETRY_FACTORY.createGeometryCollection(new
Geometry[]{GEOMETRY_FACTORY.createGeometryCollection(new
Geometry[]{multiPolygon, point3D, emptyLineString})});
Polygon expectedPolygon = GEOMETRY_FACTORY.createPolygon(coordArray(2,
3, 2, 4, 3, 4, 3, 3, 2, 3));
Polygon expectedPolygon3D =
GEOMETRY_FACTORY.createPolygon(coordArray3d(2, 3, 6, 3, 3, 7, 3, 4, 7, 2, 3,
6));
Point expectedPoint3D = GEOMETRY_FACTORY.createPoint(new Coordinate(2,
4, 6));
@@ -934,6 +934,45 @@ public class FunctionsTest {
assertEquals(emptyLineString.toText(),
actualGeometry.getGeometryN(0).getGeometryN(2).toText());
}
+ @Test
+ public void testFrechetGeom2D() {
+ LineString lineString1 =
GEOMETRY_FACTORY.createLineString(coordArray(0, 0, 100, 0));
+ LineString lineString2 =
GEOMETRY_FACTORY.createLineString(coordArray(0, 0, 50, 50, 100, 0));
+ double expected = 70.7106781186548;
+ double actual = Functions.frechetDistance(lineString1, lineString2);
+ assertEquals(expected, actual, 1e-9);
+ }
+
+ @Test
+ public void testFrechetGeom3D() {
+ LineString lineString =
GEOMETRY_FACTORY.createLineString(coordArray3d(1, 0, 1, 2, 2, 2, 3, 3, 3));
+ Polygon polygon = GEOMETRY_FACTORY.createPolygon(coordArray3d(1, 0, 0,
1, 1, 1, 2, 1, 1, 1, 0, 0));
+ double expected = 3.605551275463989;
+ double actual = Functions.frechetDistance(lineString, polygon);
+ assertEquals(expected, actual, 1e-9);
+ }
+
+ @Test
+ public void testFrechetGeomCollection() {
+ Geometry point = GEOMETRY_FACTORY.createPoint(new Coordinate(1, 2));
+ Geometry lineString1 = GEOMETRY_FACTORY.createLineString(coordArray(2,
2, 3, 3, 4, 4));
+ Geometry lineString2 =
GEOMETRY_FACTORY.createLineString(coordArray(-1, -1, -4, -4, -10, -10));
+ Geometry geometryCollection =
GEOMETRY_FACTORY.createGeometryCollection(new Geometry[] {point, lineString1,
lineString2});
+ Polygon polygon = GEOMETRY_FACTORY.createPolygon(coordArray(1, 0, 1,
1, 2, 1, 2, 0, 1, 0));
+ double expected = 14.866068747318506;
+ double actual = Functions.frechetDistance(polygon, geometryCollection);
+ assertEquals(expected, actual, 1e-9);
+ }
+
+ @Test
+ public void testFrechetGeomEmpty() {
+ Polygon p1 = GEOMETRY_FACTORY.createPolygon(coordArray(1, 0, 1, 1, 2,
1, 2, 0, 1, 0));
+ LineString emptyPoint = GEOMETRY_FACTORY.createLineString();
+ double expected = 0.0;
+ double actual = Functions.frechetDistance(p1, emptyPoint);
+ assertEquals(expected, actual, 1e-9);
+ }
+
@Test
public void boundingDiagonalGeom2D() {
Polygon polygon = GEOMETRY_FACTORY.createPolygon(coordArray(1, 0, 1,
1, 2, 1, 2, 2, 2, 0, 1, 0));
diff --git a/docs/api/flink/Function.md b/docs/api/flink/Function.md
index 9b172dd5..52e8fe54 100644
--- a/docs/api/flink/Function.md
+++ b/docs/api/flink/Function.md
@@ -544,6 +544,26 @@ Input: `LINESTRING EMPTY`
Output: `LINESTRING EMPTY`
+## ST_FrechetDistance
+
+Introduction: Computes and returns discrete [Frechet
Distance](https://en.wikipedia.org/wiki/Fr%C3%A9chet_distance) between the
given two geometrie,
+based on [Computing Discrete Frechet
Distance](http://www.kr.tuwien.ac.at/staff/eiter/et-archive/cdtr9464.pdf)
+
+If any of the geometries is empty, returns 0.0
+
+Format: `ST_FrechetDistance(g1: geomtry, g2: geometry)`
+
+Since: `1.5.0`
+
+Example:
+```sql
+SELECT ST_FrechetDistance(g1, g2)
+```
+
+Input: `g1: POINT (0 1), g2: LINESTRING (0 0, 1 0, 2 0, 3 0, 4 0, 5 0)`
+
+Output: `5.0990195135927845`
+
## ST_GeoHash
Introduction: Returns GeoHash of the geometry with given precision
diff --git a/docs/api/sql/Function.md b/docs/api/sql/Function.md
index 2ea607b2..931b6f66 100644
--- a/docs/api/sql/Function.md
+++ b/docs/api/sql/Function.md
@@ -730,6 +730,26 @@ Input: `LINESTRING EMPTY`
Output: `LINESTRING EMPTY`
+## ST_FrechetDistance
+
+Introduction: Computes and returns discrete [Frechet
Distance](https://en.wikipedia.org/wiki/Fr%C3%A9chet_distance) between the
given two geometrie,
+based on [Computing Discrete Frechet
Distance](http://www.kr.tuwien.ac.at/staff/eiter/et-archive/cdtr9464.pdf)
+
+If any of the geometries is empty, returns 0.0
+
+Format: `ST_FrechetDistance(g1: geomtry, g2: geometry)`
+
+Since: `1.5.0`
+
+Example:
+```sql
+SELECT ST_FrechetDistance(g1, g2)
+```
+
+Input: `g1: POINT (0 1), g2: LINESTRING (0 0, 1 0, 2 0, 3 0, 4 0, 5 0)`
+
+Output: `5.0990195135927845`
+
## ST_GeoHash
Introduction: Returns GeoHash of the geometry with given precision
diff --git a/flink/src/main/java/org/apache/sedona/flink/Catalog.java
b/flink/src/main/java/org/apache/sedona/flink/Catalog.java
index d9ee14e8..2f2657b9 100644
--- a/flink/src/main/java/org/apache/sedona/flink/Catalog.java
+++ b/flink/src/main/java/org/apache/sedona/flink/Catalog.java
@@ -100,6 +100,7 @@ public class Catalog {
new Functions.ST_Force3D(),
new Functions.ST_NRings(),
new Functions.ST_Translate(),
+ new Functions.ST_FrechetDistance(),
new Functions.ST_Affine(),
new Functions.ST_BoundingDiagonal(),
new Functions.ST_HausdorffDistance(),
diff --git
a/flink/src/main/java/org/apache/sedona/flink/expressions/Functions.java
b/flink/src/main/java/org/apache/sedona/flink/expressions/Functions.java
index e1a22bb7..7c6c132d 100644
--- a/flink/src/main/java/org/apache/sedona/flink/expressions/Functions.java
+++ b/flink/src/main/java/org/apache/sedona/flink/expressions/Functions.java
@@ -581,7 +581,16 @@ public class Functions {
Geometry geometry = (Geometry) o;
return
org.apache.sedona.common.Functions.geometricMedian(geometry, tolerance,
maxIter, failIfNotConverged);
}
+ }
+ public static class ST_FrechetDistance extends ScalarFunction {
+ @DataTypeHint("Double")
+ public Double eval(@DataTypeHint(value = "RAW", bridgedTo =
org.locationtech.jts.geom.Geometry.class) Object g1,
+ @DataTypeHint(value = "RAW", bridgedTo =
org.locationtech.jts.geom.Geometry.class) Object g2) {
+ Geometry geom1 = (Geometry) g1;
+ Geometry geom2 = (Geometry) g2;
+ return org.apache.sedona.common.Functions.frechetDistance(geom1,
geom2);
+ }
}
public static class ST_NumPoints extends ScalarFunction {
diff --git a/flink/src/test/java/org/apache/sedona/flink/FunctionTest.java
b/flink/src/test/java/org/apache/sedona/flink/FunctionTest.java
index 37c9b305..b26f75ad 100644
--- a/flink/src/test/java/org/apache/sedona/flink/FunctionTest.java
+++ b/flink/src/test/java/org/apache/sedona/flink/FunctionTest.java
@@ -747,6 +747,14 @@ public class FunctionTest extends TestBase{
}
@Test
+ public void testFrechet() {
+ Table polyTable = tableEnv.sqlQuery("SELECT ST_GeomFromWKT('POINT (1
2)') AS g1, ST_GeomFromWKT('POINT (10 10)') as g2");
+ polyTable =
polyTable.select(call(Functions.ST_FrechetDistance.class.getSimpleName(),
$("g1"), $("g2")));
+ Double expected = 12.041594578792296;
+ Double actual = (Double) first(polyTable).getField(0);
+ assertEquals(expected, actual);
+ }
+
public void testBoundingDiagonal() {
Table polyTable = tableEnv.sqlQuery("SELECT
ST_BoundingDiagonal(ST_GeomFromWKT('POLYGON ((1 0, 1 1, 2 1, 2 0, 1 0))'))" +"
AS " + polygonColNames[0]);
polyTable =
polyTable.select(call(Functions.ST_AsText.class.getSimpleName(),
$(polygonColNames[0])));
diff --git a/python/sedona/sql/st_functions.py
b/python/sedona/sql/st_functions.py
index 6ee6c64f..e20f1912 100644
--- a/python/sedona/sql/st_functions.py
+++ b/python/sedona/sql/st_functions.py
@@ -113,6 +113,7 @@ __all__ = [
"ST_Force3D",
"ST_NRings",
"ST_Translate",
+ "ST_FrechetDistance",
"ST_Affine",
"ST_BoundingDiagonal"
]
@@ -1290,6 +1291,18 @@ def ST_Translate(geometry: ColumnOrName, deltaX:
Union[ColumnOrName, float], del
args = (geometry, deltaX, deltaY, deltaZ)
return _call_st_function("ST_Translate", args)
+def ST_FrechetDistance(g1: ColumnOrName, g2: ColumnOrName) -> Column:
+ """
+ Computes discrete frechet distance between the two geometries.
+ If any of the geometry is empty, ST_FrechetDistance returns 0
+ :param g1:
+ :param g2:
+ :return: Computed Discrete Frechet Distance between g1 and g2
+ """
+
+ args = (g1, g2)
+ return _call_st_function("ST_FrechetDistance", args)
+
@validate_argument_types
def ST_Affine(geometry: ColumnOrName, a: Union[ColumnOrName, float], b:
Union[ColumnOrName, float], d: Union[ColumnOrName, float],
e: Union[ColumnOrName, float], xOff: Union[ColumnOrName,
float], yOff: Union[ColumnOrName, float], c: Optional[Union[ColumnOrName,
float]] = None, f: Optional[Union[ColumnOrName, float]] = None,
diff --git a/python/tests/sql/test_dataframe_api.py
b/python/tests/sql/test_dataframe_api.py
index d61bc4c1..69961a43 100644
--- a/python/tests/sql/test_dataframe_api.py
+++ b/python/tests/sql/test_dataframe_api.py
@@ -89,6 +89,7 @@ test_configurations = [
(stf.ST_FlipCoordinates, ("point",), "point_geom", "", "POINT (1 0)"),
(stf.ST_Force_2D, ("point",), "point_geom", "", "POINT (0 1)"),
(stf.ST_Force3D, ("point", 1.0), "point_geom", "", "POINT Z (0 1 1)"),
+ (stf.ST_FrechetDistance, ("point", "line",), "point_and_line", "",
5.0990195135927845),
(stf.ST_GeometricMedian, ("multipoint",), "multipoint_geom", "", "POINT
(22.500002656424286 21.250001168173426)"),
(stf.ST_GeometryN, ("geom", 0), "multipoint", "", "POINT (0 0)"),
(stf.ST_GeometryType, ("point",), "point_geom", "", "ST_Point"),
diff --git a/python/tests/sql/test_function.py
b/python/tests/sql/test_function.py
index a161a6dd..7c2967fd 100644
--- a/python/tests/sql/test_function.py
+++ b/python/tests/sql/test_function.py
@@ -1129,6 +1129,13 @@ class TestPredicateJoin(TestBase):
actual = actual_df.selectExpr("ST_AsText(geom)").take(1)[0][0]
assert expected == actual
+ def test_frechetDistance(self):
+ expected = 5.0990195135927845
+ actual_df = self.spark.sql("SELECT
ST_FrechetDistance(ST_GeomFromText('LINESTRING (0 0, 1 0, 2 0, 3 0, 4 0, "
+ "5 0)'), ST_GeomFromText('POINT (0 1)'))")
+ actual = actual_df.take(1)[0][0]
+ assert expected == actual
+
def test_affine(self):
expected = "POLYGON Z((2 3 1, 4 5 1, 7 8 2, 2 3 1))"
actual_df = self.spark.sql("SELECT ST_Affine(ST_GeomFromText('POLYGON
((1 0 1, 1 1 1, 2 2 2, 1 0 1))'), 1, 2, "
diff --git a/sql/common/src/main/scala/org/apache/sedona/sql/UDF/Catalog.scala
b/sql/common/src/main/scala/org/apache/sedona/sql/UDF/Catalog.scala
index c74513bd..bec5a76e 100644
--- a/sql/common/src/main/scala/org/apache/sedona/sql/UDF/Catalog.scala
+++ b/sql/common/src/main/scala/org/apache/sedona/sql/UDF/Catalog.scala
@@ -152,6 +152,7 @@ object Catalog {
function[ST_Force3D](0.0),
function[ST_NRings](),
function[ST_Translate](0.0),
+ function[ST_FrechetDistance](),
function[ST_Affine](null, null, null, null, null, null),
function[ST_BoundingDiagonal](),
function[ST_HausdorffDistance](-1),
diff --git
a/sql/common/src/main/scala/org/apache/spark/sql/sedona_sql/expressions/Functions.scala
b/sql/common/src/main/scala/org/apache/spark/sql/sedona_sql/expressions/Functions.scala
index 6ea96af1..904920c9 100644
---
a/sql/common/src/main/scala/org/apache/spark/sql/sedona_sql/expressions/Functions.scala
+++
b/sql/common/src/main/scala/org/apache/spark/sql/sedona_sql/expressions/Functions.scala
@@ -1005,6 +1005,13 @@ case class ST_Translate(inputExpressions:
Seq[Expression])
}
}
+case class ST_FrechetDistance(inputExpressions: Seq[Expression])
+ extends InferredExpression(Functions.frechetDistance _) with
FoldableExpression {
+ protected def withNewChildrenInternal(newChildren: IndexedSeq[Expression]) =
{
+ copy(inputExpressions = newChildren)
+ }
+}
+
case class ST_Affine(inputExpressions: Seq[Expression])
extends
InferredExpression(InferrableFunction.allowSixRightNull(Functions.affine _))
with FoldableExpression {
protected def withNewChildrenInternal(newChildren: IndexedSeq[Expression]) =
{
diff --git
a/sql/common/src/main/scala/org/apache/spark/sql/sedona_sql/expressions/st_functions.scala
b/sql/common/src/main/scala/org/apache/spark/sql/sedona_sql/expressions/st_functions.scala
index 562f3629..fadde48c 100644
---
a/sql/common/src/main/scala/org/apache/spark/sql/sedona_sql/expressions/st_functions.scala
+++
b/sql/common/src/main/scala/org/apache/spark/sql/sedona_sql/expressions/st_functions.scala
@@ -329,6 +329,9 @@ object st_functions extends DataFrameAPI {
def ST_Translate(geometry: String, deltaX: Double, deltaY: Double): Column =
wrapExpression[ST_Translate](geometry, deltaX, deltaY, 0.0)
+ def ST_FrechetDistance(g1: Column, g2: Column): Column =
wrapExpression[ST_FrechetDistance](g1, g2)
+
+ def ST_FrechetDistance(g1: String, g2: String): Column =
wrapExpression[ST_FrechetDistance](g1, g2)
def ST_Affine(geometry: Column, a: Column, b: Column, d: Column, e: Column,
xOff: Column, yOff: Column, c: Column, f: Column, g: Column, h: Column, i:
Column, zOff: Column): Column =
wrapExpression[ST_Affine](geometry, a, b, d, e, xOff, yOff, c, f, g, h, i,
zOff)
diff --git
a/sql/common/src/test/scala/org/apache/sedona/sql/dataFrameAPITestScala.scala
b/sql/common/src/test/scala/org/apache/sedona/sql/dataFrameAPITestScala.scala
index 62006f7b..a1c2ebf7 100644
---
a/sql/common/src/test/scala/org/apache/sedona/sql/dataFrameAPITestScala.scala
+++
b/sql/common/src/test/scala/org/apache/sedona/sql/dataFrameAPITestScala.scala
@@ -1003,6 +1003,14 @@ class dataFrameAPITestScala extends TestBaseScala {
assert(expectedDefaultValue == actualDefaultValue)
}
+ it("Passed ST_FrechetDistance") {
+ val polyDf = sparkSession.sql("SELECT ST_GeomFromWKT('POINT (1 2)') as
g1, ST_GeomFromWKT('POINT (100 230)') as g2")
+ val df = polyDf.select(ST_FrechetDistance("g1", "g2"))
+ val expected = 248.5658866377283
+ val actual = df.take(1)(0).get(0).asInstanceOf[Double]
+ assertEquals(expected, actual, 1e-9)
+ }
+
it("Passed ST_Affine") {
val polyDf = sparkSession.sql("SELECT ST_GeomFromWKT('POLYGON ((2 3 1, 4
5 1, 7 8 2, 2 3 1))') AS geom")
val df = polyDf.select(ST_Affine("geom", 1, 2, 3, 4, 1, 2, 3, 4, 1, 4,
2, 1));
@@ -1029,7 +1037,7 @@ class dataFrameAPITestScala extends TestBaseScala {
it("Passed ST_HausdorffDistance") {
val polyDf = sparkSession.sql("SELECT ST_GeomFromWKT('POLYGON ((1 2, 2
1, 2 0, 4 1, 1 2))') AS g1, " +
- "ST_GeomFromWKT('MULTILINESTRING ((1 1, 2 1, 4 4, 5 5), (10 10,
11 11, 12 12, 14 14), (-11 -20, -11 -21, -15 -19))') AS g2")
+ "ST_GeomFromWKT('MULTILINESTRING ((1 1, 2 1, 4 4, 5 5), (10 10, 11 11,
12 12, 14 14), (-11 -20, -11 -21, -15 -19))') AS g2")
val df = polyDf.select(ST_HausdorffDistance("g1", "g2", 0.05))
val dfDefaultValue = polyDf.select(ST_HausdorffDistance("g1", "g2"))
val expected = 25.495097567963924
diff --git
a/sql/common/src/test/scala/org/apache/sedona/sql/functionTestScala.scala
b/sql/common/src/test/scala/org/apache/sedona/sql/functionTestScala.scala
index 4200b3e6..d31849ea 100644
--- a/sql/common/src/test/scala/org/apache/sedona/sql/functionTestScala.scala
+++ b/sql/common/src/test/scala/org/apache/sedona/sql/functionTestScala.scala
@@ -2005,6 +2005,22 @@ class functionTestScala extends TestBaseScala with
Matchers with GeometrySample
}
}
+ it ("should pass ST_FrechetDistance") {
+ val geomTestCases = Map(
+ ("'POINT (1 2)'", "'POINT (10 10)'") -> 12.041594578792296d,
+ ("'LINESTRING (0 0, 100 0)'", "'LINESTRING (0 0, 50 50, 100 0)'") ->
70.7106781186548d
+ )
+ for (((geom), expectedResult) <- geomTestCases) {
+ val g1 = geom._1
+ val g2 = geom._2
+ val df = sparkSession.sql(s"SELECT
ST_FrechetDistance(ST_GeomFromWKT($g1), ST_GeomFromWKT($g2))")
+ val actual = df.take(1)(0).get(0).asInstanceOf[Double]
+ val expected = expectedResult
+ assertEquals(expected, actual, 1e-9)
+
+ }
+ }
+
it ("should pass ST_Affine") {
val geomTestCases = Map (
("'POLYGON ((1 0 1, 1 1 1, 2 2 2, 1 0 1))'")-> ("'POLYGON Z((5 8 16, 7 9
20, 13 16 37, 5 8 16))'", "'POLYGON Z((2 3 1, 4 5 1, 7 8 2, 2 3 1))'"),
