ST_Scale

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from geoanalytics.sql import functions as ST

data = [
    ("POINT (0.5 0.5)",),
    ("MULTIPOINT (4.0 4.5, 5.0 4.5, 5.5 5.0)",),
    ("LINESTRING (2.0 0.0, 3.0 1.0, 4.0 0.5)", ),
    ("POLYGON ((5.0 2.0, 6.0 1.0, 5.5 1.0, 4.0 2.0, 5.0 2.0))", )
]

df = spark.createDataFrame(data, ["wkt"]) \
          .withColumn("geometry", ST.geom_from_text("wkt"))

df_scale = df.withColumn("scale", ST.scale(geometry="geometry",
                                           x_scale_factor=2,
                                           y_scale_factor=2))

ax = df_scale.st.plot("geometry", facecolor="none", edgecolor="red")
df_scale.st.plot("scale", ax=ax, facecolor="none", edgecolor="blue")

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import com.esri.geoanalytics.sql.{functions => ST}

case class GeometryRow(wkt: String)
val data = Seq(GeometryRow("POINT (0.5 0.5)"),
               GeometryRow("MULTIPOINT (4.0 4.5, 5.0 4.5, 5.5 5.0)"),
               GeometryRow("LINESTRING (2.0 0.0, 3.0 1.0, 4.0 0.5)"),
               GeometryRow("POLYGON ((5.0 2.0, 6.0 1.0, 5.5 1.0, 4.0 2.0, 5.0 2.0))"))

val df = spark.createDataFrame(data)
              .withColumn("geometry", ST.geomFromText($"wkt"))
              .withColumn("scale", ST.scale($"geometry", 2, 2))

df.select("scale").show(truncate = false)

1
2
3
4
5
6
7
8
+-----------------------------------------------+
|scale                                          |
+-----------------------------------------------+
|{"x":1,"y":1}                                  |
|{"points":[[8,9],[10,9],[11,10]]}              |
|{"paths":[[[4,0],[6,2],[8,1]]]}                |
|{"rings":[[[10,4],[12,2],[11,2],[8,4],[10,4]]]}|
+-----------------------------------------------+