add country-coder json [#44]

tiles/src/main/java/com/protomaps/basemap/Basemap.java

@@ -4,6 +4,7 @@
 import com.onthegomap.planetiler.Planetiler;
 import com.onthegomap.planetiler.config.Arguments;
 import com.onthegomap.planetiler.util.Downloader;
+import com.protomaps.basemap.feature.CountryCoder;
 import com.protomaps.basemap.feature.NaturalEarthDb;
 import com.protomaps.basemap.feature.QrankDb;
 import com.protomaps.basemap.layers.Boundaries;
@@ -18,6 +19,7 @@
 import com.protomaps.basemap.layers.Water;
 import com.protomaps.basemap.postprocess.Clip;
 import com.protomaps.basemap.text.FontRegistry;
+import java.io.IOException;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.util.HashMap;
@@ -27,7 +29,7 @@
 
 public class Basemap extends ForwardingProfile {
 
-  public Basemap(NaturalEarthDb naturalEarthDb, QrankDb qrankDb, Clip clip) {
+  public Basemap(NaturalEarthDb naturalEarthDb, QrankDb qrankDb, CountryCoder countryCoder, Clip clip) {
 
     var admin = new Boundaries();
     registerHandler(admin);
@@ -54,7 +56,7 @@ public Basemap(NaturalEarthDb naturalEarthDb, QrankDb qrankDb, Clip clip) {
     registerHandler(poi);
     registerSourceHandler("osm", poi::processOsm);
 
-    var roads = new Roads();
+    var roads = new Roads(countryCoder);
     registerHandler(roads);
     registerSourceHandler("osm", roads::processOsm);
 
@@ -122,11 +124,11 @@ public Map<String, String> extraArchiveMetadata() {
     return result;
   }
 
-  public static void main(String[] args) {
+  public static void main(String[] args) throws IOException {
     run(Arguments.fromArgsOrConfigFile(args));
   }
 
-  static void run(Arguments args) {
+  static void run(Arguments args) throws IOException {
     args = args.orElse(Arguments.of("maxzoom", 15));
 
     Path dataDir = Path.of("data");
@@ -135,6 +137,8 @@ static void run(Arguments args) {
     Path nePath = sourcesDir.resolve("natural_earth_vector.sqlite.zip");
     String neUrl = "https://naciscdn.org/naturalearth/packages/natural_earth_vector.sqlite.zip";
 
+    var countryCoder = CountryCoder.fromJarResource();
+
     String area = args.getString("area", "geofabrik area to download", "monaco");
 
     var planetiler = Planetiler.create(args)
@@ -171,7 +175,8 @@ static void run(Arguments args) {
 
     fontRegistry.loadFontBundle("NotoSansDevanagari-Regular", "1", "Devanagari");
 
-    planetiler.setProfile(new Basemap(naturalEarthDb, qrankDb, clip)).setOutput(Path.of(area + ".pmtiles"))
+    planetiler.setProfile(new Basemap(naturalEarthDb, qrankDb, countryCoder, clip))
+      .setOutput(Path.of(area + ".pmtiles"))
       .run();
   }
 }

tiles/src/main/java/com/protomaps/basemap/feature/CountryCoder.java

@@ -0,0 +1,64 @@
+package com.protomaps.basemap.feature;
+
+import com.onthegomap.planetiler.reader.geojson.GeoJson;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.List;
+import java.util.Optional;
+import org.locationtech.jts.geom.*;
+import org.locationtech.jts.index.strtree.STRtree;
+
+public class CountryCoder {
+
+  public record Record(String country, String nameEn, MultiPolygon multiPolygon) {}
+
+  private STRtree tree;
+
+  public CountryCoder(STRtree tree) {
+    this.tree = tree;
+  }
+
+  public static CountryCoder fromJarResource() throws IOException {
+    InputStream inputStream = CountryCoder.class.getResourceAsStream("/borders.json");
+
+    String jsonContent = new String(inputStream.readAllBytes());
+    return fromJsonString(jsonContent);
+  }
+
+  public static CountryCoder fromJsonString(String s) {
+    STRtree tree = new STRtree();
+
+    var g = GeoJson.from(s);
+
+    for (var feature : g) {
+      var properties = feature.tags();
+
+      String country = "";
+      if (properties.containsKey("iso1A2")) {
+        country = properties.get("iso1A2").toString();
+      } else if (properties.containsKey("country")) {
+        country = properties.get("country").toString();
+      }
+
+      if (country.isBlank() || feature.geometry().getNumGeometries() == 0) {
+        continue;
+      }
+      MultiPolygon mp = (MultiPolygon) feature.geometry();
+      tree.insert(mp.getEnvelopeInternal(),
+        new Record(country, properties.get("nameEn").toString(), mp));
+    }
+    return new CountryCoder(tree);
+  }
+
+  public Optional<String> getCountryCode(Geometry geom) {
+    List<Record> results = tree.query(geom.getEnvelopeInternal());
+    if (results.isEmpty())
+      return Optional.empty();
+    var filtered = results.stream().filter(rec -> rec.multiPolygon.contains(geom)
+    ).toList();
+    if (filtered.isEmpty()) {
+      return Optional.empty();
+    }
+    return Optional.of(filtered.getFirst().country);
+  }
+}

tiles/src/main/java/com/protomaps/basemap/layers/Roads.java

@@ -8,6 +8,7 @@
 import com.onthegomap.planetiler.VectorTile;
 import com.onthegomap.planetiler.geo.GeometryException;
 import com.onthegomap.planetiler.reader.SourceFeature;
+import com.protomaps.basemap.feature.CountryCoder;
 import com.protomaps.basemap.feature.FeatureId;
 import com.protomaps.basemap.locales.CartographicLocale;
 import com.protomaps.basemap.locales.US;
@@ -16,6 +17,12 @@
 
 public class Roads implements ForwardingProfile.LayerPostProcesser {
 
+  private CountryCoder countryCoder;
+
+  public Roads(CountryCoder countryCoder) {
+    this.countryCoder = countryCoder;
+  }
+
   @Override
   public String name() {
     return "roads";
@@ -147,6 +154,12 @@ public void processOsm(SourceFeature sf, FeatureCollector features) {
         .setPixelTolerance(0)
         .setZoomRange(minZoom, maxZoom);
 
+      try {
+        var code = countryCoder.getCountryCode(sf.latLonGeometry());
+      } catch (Exception e) {
+        // do logic based on country code
+      }
+
       if (!kindDetail.isEmpty()) {
         feat.setAttr("kind_detail", kindDetail);
       } else {