Skip to content

Commit

Permalink
Merge pull request #501 from databrickslabs/polygon-expect-closed
Browse files Browse the repository at this point in the history 
expect closed linestring for polygon
  • Loading branch information
Milos Colic authored Jan 12, 2024
2 parents f734d66 + f9e3fa0 commit 21759ae
Show file tree
Hide file tree
Showing 3 changed files with 38 additions and 13 deletions.
21 changes: 18 additions & 3 deletions src/main/scala/com/databricks/labs/mosaic/core/geometry/polygon/MosaicPolygonJTS.scala
View file
Original file line number Diff line number Diff line change
Expand Up @@ -84,14 +84,29 @@ object MosaicPolygonJTS extends GeometryReader {
val newGeom = GeometryTypeEnum.fromString(geomSeq.head.getGeometryType) match {
case POINT =>
val extractedPoints = geomSeq.map(_.asInstanceOf[MosaicPointJTS])
val exteriorRing = extractedPoints.map(_.coord).toArray ++ Array(extractedPoints.head.coord)
val exteriorRing =
if (extractedPoints.head.coord == extractedPoints.last.coord) {
extractedPoints.map(_.coord).toArray
} else {
extractedPoints.map(_.coord).toArray ++ Array(extractedPoints.head.coord)
}
gf.createPolygon(exteriorRing)
case LINESTRING =>
val extractedLines = geomSeq.map(_.asInstanceOf[MosaicLineStringJTS])
val exteriorRing =
gf.createLinearRing(extractedLines.head.asSeq.map(_.coord).toArray ++ Array(extractedLines.head.asSeq.head.coord))
if (extractedLines.head.asSeq.head.coord == extractedLines.head.asSeq.last.coord) {
gf.createLinearRing(extractedLines.head.asSeq.map(_.coord).toArray)
} else {
gf.createLinearRing(extractedLines.head.asSeq.map(_.coord).toArray ++ Array(extractedLines.head.asSeq.head.coord))
}
val holes = extractedLines.tail
.map({ h: MosaicLineStringJTS => h.asSeq.map(_.coord).toArray ++ Array(h.asSeq.head.coord) })
.map({ h: MosaicLineStringJTS =>
if (h.asSeq.head.coord == h.asSeq.last.coord) {
h.asSeq.map(_.coord).toArray
} else {
h.asSeq.map(_.coord).toArray ++ Array(h.asSeq.head.coord)
}
})
.map(gf.createLinearRing)
.toArray
gf.createPolygon(exteriorRing, holes)
Expand Down
22 changes: 16 additions & 6 deletions src/test/scala/com/databricks/labs/mosaic/core/geometry/polygon/TestPolygonJTS.scala
View file
Original file line number Diff line number Diff line change
Expand Up @@ -42,11 +42,16 @@ class TestPolygonJTS extends AnyFlatSpec {

"MosaicPolygonJTS" should "be instantiable from a Seq of MosaicPointJTS" in {
val polygonReference = MosaicPolygonJTS.fromWKT("POLYGON ((30 10, 40 40, 20 40, 10 20, 30 10))")
val pointSeq = Seq("POINT (30 10)", "POINT (40 40)", "POINT (20 40)", "POINT (10 20)")
val pointSeq_open = Seq("POINT (30 10)", "POINT (40 40)", "POINT (20 40)", "POINT (10 20)")
.map(MosaicPointJTS.fromWKT)
.map(_.asInstanceOf[MosaicPointJTS])
val polygonTest = MosaicPolygonJTS.fromSeq(pointSeq)
polygonReference.equals(polygonTest) shouldBe true
val pointSeq_closed = Seq("POINT (30 10)", "POINT (40 40)", "POINT (20 40)", "POINT (10 20)", "POINT (30 10)")
.map(MosaicPointJTS.fromWKT)
.map(_.asInstanceOf[MosaicPointJTS])
val polygonTest_open = MosaicPolygonJTS.fromSeq(pointSeq_open)
val polygonTest_closed = MosaicPolygonJTS.fromSeq(pointSeq_closed)
polygonReference.equals(polygonTest_open) shouldBe true
polygonReference.equals(polygonTest_closed) shouldBe true
}

"MosaicPolygonJTS" should "not fail for empty Seq" in {
Expand All @@ -59,11 +64,16 @@ class TestPolygonJTS extends AnyFlatSpec {

"MosaicPolygonJTS" should "be instantiable from a Seq of MosaicLineStringJTS" in {
val polygonReference = MosaicPolygonJTS.fromWKT("POLYGON ((35 10, 45 45, 15 40, 10 20, 35 10), (20 30, 35 35, 30 20, 20 30))")
val linesSeq = Seq("LINESTRING (35 10, 45 45, 15 40, 10 20)", "LINESTRING (20 30, 35 35, 30 20)")
val linesSeq_open = Seq("LINESTRING (35 10, 45 45, 15 40, 10 20)", "LINESTRING (20 30, 35 35, 30 20)")
.map(MosaicLineStringJTS.fromWKT)
.map(_.asInstanceOf[MosaicLineStringJTS])
val linesSeq_closed = Seq("LINESTRING (35 10, 45 45, 15 40, 10 20, 35 10)", "LINESTRING (20 30, 35 35, 30 20, 20 30)")
.map(MosaicLineStringJTS.fromWKT)
.map(_.asInstanceOf[MosaicLineStringJTS])
val polygonTest = MosaicPolygonJTS.fromSeq(linesSeq)
polygonReference.equals(polygonTest) shouldBe true
val polygonTest_open = MosaicPolygonJTS.fromSeq(linesSeq_open)
val polygonTest_closed = MosaicPolygonJTS.fromSeq(linesSeq_closed)
polygonReference.equals(polygonTest_open) shouldBe true
polygonReference.equals(polygonTest_closed) shouldBe true
}

"MosaicPolygonJTS" should "return a Seq of MosaicLineStringJTS object when calling asSeq" in {
Expand Down
8 changes: 4 additions & 4 deletions src/test/scala/com/databricks/labs/mosaic/core/index/TestCustomIndexSystem.scala
View file
Original file line number Diff line number Diff line change
Expand Up @@ -118,16 +118,16 @@ class TestCustomIndexSystem extends AnyFunSuite {

// First quadrant
val wkt0 = grid.indexToGeometry(0 | resolutionMask, JTS).toWKT
wkt0 shouldBe "POLYGON ((0 0, 50 0, 50 50, 0 50, 0 0, 0 0))"
wkt0 shouldBe "POLYGON ((0 0, 50 0, 50 50, 0 50, 0 0))"

val wkt1 = grid.indexToGeometry(1 | resolutionMask, JTS).toWKT
wkt1 shouldBe "POLYGON ((50 0, 100 0, 100 50, 50 50, 50 0, 50 0))"
wkt1 shouldBe "POLYGON ((50 0, 100 0, 100 50, 50 50, 50 0))"

val wkt2 = grid.indexToGeometry(2 | resolutionMask, JTS).toWKT
wkt2 shouldBe "POLYGON ((0 50, 50 50, 50 100, 0 100, 0 50, 0 50))"
wkt2 shouldBe "POLYGON ((0 50, 50 50, 50 100, 0 100, 0 50))"

val wkt3 = grid.indexToGeometry(3 | resolutionMask, JTS).toWKT
wkt3 shouldBe "POLYGON ((50 50, 100 50, 100 100, 50 100, 50 50, 50 50))"
wkt3 shouldBe "POLYGON ((50 50, 100 50, 100 100, 50 100, 50 50))"
}

test("polyfill single cell") {
Expand Down

0 comments on commit 21759ae

Please sign in to comment.