Merge pull request #12 from OvertureMaps/buildings-sources

set @height_source and @geometry_source helper attributes in building…
OvertureMaps · Jul 24, 2024 · 4bf0e83 · 4bf0e83
2 parents df45bc2 + 8c6cf0e
commit 4bf0e83
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/profiles/Buildings.java b/profiles/Buildings.java
@@ -12,6 +12,31 @@ public void processFeature(SourceFeature source, FeatureCollector features) {
         var polygon = features.polygon(layer);
 
         if (source instanceof ParquetFeature pf) {
+            var sources = pf.getStruct("sources").asList();
+
+            // set @height_source helper
+            sources.stream().filter(s -> s.get("property").asString().equals("properties/height")).findFirst().ifPresentOrElse(
+                    s -> polygon.setAttr("@height_source", s.get("dataset")),
+                    () -> {
+                        if (pf.hasTag("height")) {
+                            sources.stream().filter(s -> s.get("property").asString().isEmpty()).findFirst().ifPresent(
+                                    s -> polygon.setAttr("@height_source", s.get("dataset"))
+                            );
+                        }
+                    }
+            );
+
+            // set @geometry_source helper
+            sources.stream().filter(s -> s.get("property").asString().equals("properties/geometry")).findFirst().ifPresentOrElse(
+                    s -> polygon.setAttr("@geometry_source", s.get("dataset")),
+                    () -> {
+                        sources.stream().filter(s -> s.get("property").asString().isEmpty()).findFirst().ifPresent(
+                                s -> polygon.setAttr("@geometry_source", s.get("dataset"))
+                        );
+                    }
+            );
+
+            // deprecate me: @source_0_dataset
             var source0Dataset = pf.getStruct("sources").get(0).get("dataset");
             polygon.setAttr("@source_0_dataset", source0Dataset);
         }