Raster processing

baremaps-cli/pom.xml

@@ -43,6 +43,10 @@ limitations under the License.
       <groupId>org.apache.baremaps</groupId>
       <artifactId>baremaps-core</artifactId>
     </dependency>
+    <dependency>
+      <groupId>org.apache.baremaps</groupId>
+      <artifactId>baremaps-raster</artifactId>
+    </dependency>
     <dependency>
       <groupId>org.apache.baremaps</groupId>
       <artifactId>baremaps-server</artifactId>

baremaps-cli/src/main/java/org/apache/baremaps/cli/Baremaps.java

@@ -28,6 +28,7 @@
 import org.apache.baremaps.cli.geocoder.Geocoder;
 import org.apache.baremaps.cli.iploc.IpLoc;
 import org.apache.baremaps.cli.map.Map;
+import org.apache.baremaps.cli.raster.Raster;
 import org.apache.baremaps.cli.workflow.Workflow;
 import org.apache.logging.log4j.Level;
 import org.apache.logging.log4j.core.config.Configurator;
@@ -42,7 +43,14 @@
     name = "baremaps",
     description = "A toolkit for producing vector tiles.",
     versionProvider = VersionProvider.class,
-    subcommands = {Workflow.class, Database.class, Map.class, Geocoder.class, IpLoc.class},
+    subcommands = {
+        Workflow.class,
+        Database.class,
+        Map.class,
+        Geocoder.class,
+        IpLoc.class,
+        Raster.class
+    },
     sortOptions = false)
 @SuppressWarnings("squid:S106")
 public class Baremaps implements Callable<Integer> {

baremaps-cli/src/main/java/org/apache/baremaps/cli/raster/HillShade.java

@@ -0,0 +1,256 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to you under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.baremaps.cli.raster;
+
+import static com.google.common.net.HttpHeaders.ACCESS_CONTROL_ALLOW_ORIGIN;
+import static com.google.common.net.HttpHeaders.CONTENT_TYPE;
+import static org.apache.baremaps.utils.ObjectMapperUtils.objectMapper;
+
+import com.github.benmanes.caffeine.cache.Caffeine;
+import com.github.benmanes.caffeine.cache.LoadingCache;
+import com.linecorp.armeria.common.*;
+import com.linecorp.armeria.server.Server;
+import com.linecorp.armeria.server.annotation.Blocking;
+import com.linecorp.armeria.server.annotation.Get;
+import com.linecorp.armeria.server.annotation.JacksonResponseConverterFunction;
+import com.linecorp.armeria.server.annotation.Param;
+import com.linecorp.armeria.server.cors.CorsService;
+import com.linecorp.armeria.server.docs.DocService;
+import java.awt.*;
+import java.awt.image.BufferedImage;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.net.URL;
+import java.nio.ByteBuffer;
+import java.util.concurrent.Callable;
+import java.util.function.Function;
+import java.util.function.Supplier;
+import javax.imageio.ImageIO;
+import org.apache.baremaps.raster.elevation.ElevationUtils;
+import org.apache.baremaps.tilestore.TileCoord;
+import org.apache.baremaps.tilestore.TileStore;
+import org.apache.baremaps.tilestore.TileStoreException;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import picocli.CommandLine.Command;
+import picocli.CommandLine.Option;
+
+@Command(name = "hillshade", description = "Start a tile server that computes hillshades.")
+public class HillShade implements Callable<Integer> {
+
+  @Option(names = {"--host"}, paramLabel = "HOST", description = "The host of the server.")
+  private String host = "localhost";
+
+  @Option(names = {"--port"}, paramLabel = "PORT", description = "The port of the server.")
+  private int port = 9000;
+
+  @Override
+  public Integer call() throws Exception {
+
+    var serverBuilder = Server.builder();
+    serverBuilder.http(port);
+
+    var objectMapper = objectMapper();
+    var jsonResponseConverter = new JacksonResponseConverterFunction(objectMapper);
+    var tileStore = new HillShadeTileStore();
+
+    serverBuilder.annotatedService(new HillShadeTileResource(() -> tileStore),
+        jsonResponseConverter);
+
+    serverBuilder.decorator(CorsService.builderForAnyOrigin()
+        .allowAllRequestHeaders(true)
+        .allowRequestMethods(
+            HttpMethod.GET,
+            HttpMethod.POST,
+            HttpMethod.PUT,
+            HttpMethod.DELETE,
+            HttpMethod.OPTIONS,
+            HttpMethod.HEAD)
+        .allowCredentials()
+        .exposeHeaders(HttpHeaderNames.LOCATION)
+        .newDecorator());
+
+    serverBuilder.serviceUnder("/docs", new DocService());
+
+    serverBuilder.disableServerHeader();
+    serverBuilder.disableDateHeader();
+
+    var server = serverBuilder.build();
+
+    var startFuture = server.start();
+    startFuture.join();
+
+    var shutdownFuture = server.closeOnJvmShutdown();
+    shutdownFuture.join();
+
+    return 0;
+  }
+
+  public static class HillShadeTileStore implements TileStore {
+
+    // private String url = "https://s3.amazonaws.com/elevation-tiles-prod/geotiff/{z}/{x}/{y}.tif";
+    // private String url = "https://demotiles.maplibre.org/terrain-tiles/{z}/{x}/{y}.png";
+    private String url = "https://s3.amazonaws.com/elevation-tiles-prod/terrarium/{z}/{x}/{y}.png";
+
+    private final LoadingCache<TileCoord, BufferedImage> cache = Caffeine.newBuilder()
+        .maximumSize(1000)
+        .build(this::getImage);
+
+    public HillShadeTileStore() {
+      // Default constructor
+    }
+
+    public BufferedImage getImage(TileCoord tileCoord) throws IOException {
+      var tileUrl = new URL(this.url
+          .replace("{z}", String.valueOf(tileCoord.z()))
+          .replace("{x}", String.valueOf(tileCoord.x()))
+          .replace("{y}", String.valueOf(tileCoord.y())));
+      return ImageIO.read(tileUrl);
+    }
+
+    public BufferedImage getKernel(TileCoord tileCoord, Function<TileCoord, BufferedImage> provider)
+        throws IOException {
+      BufferedImage z1 =
+          provider.apply(new TileCoord(tileCoord.x() - 1, tileCoord.y() - 1, tileCoord.z()));
+      BufferedImage z2 =
+          provider.apply(new TileCoord(tileCoord.x(), tileCoord.y() - 1, tileCoord.z()));
+      BufferedImage z3 =
+          provider.apply(new TileCoord(tileCoord.x() + 1, tileCoord.y() - 1, tileCoord.z()));
+      BufferedImage z4 =
+          provider.apply(new TileCoord(tileCoord.x() - 1, tileCoord.y(), tileCoord.z()));
+      BufferedImage z5 = provider.apply(tileCoord);
+      BufferedImage z6 =
+          provider.apply(new TileCoord(tileCoord.x() + 1, tileCoord.y(), tileCoord.z()));
+      BufferedImage z7 =
+          provider.apply(new TileCoord(tileCoord.x() - 1, tileCoord.y() + 1, tileCoord.z()));
+      BufferedImage z8 =
+          provider.apply(new TileCoord(tileCoord.x(), tileCoord.y() + 1, tileCoord.z()));
+      BufferedImage z9 =
+          provider.apply(new TileCoord(tileCoord.x() + 1, tileCoord.y() + 1, tileCoord.z()));
+      int kernelSize = z5.getWidth() * 3;
+      BufferedImage kernel = new BufferedImage(kernelSize, kernelSize, z5.getType());
+      for (int y = 0; y < z5.getHeight(); y++) {
+        for (int x = 0; x < z5.getWidth(); x++) {
+          kernel.setRGB(x, y, z1.getRGB(x, y));
+          kernel.setRGB(x + z5.getWidth(), y, z2.getRGB(x, y));
+          kernel.setRGB(x + 2 * z5.getWidth(), y, z3.getRGB(x, y));
+          kernel.setRGB(x, y + z5.getHeight(), z4.getRGB(x, y));
+          kernel.setRGB(x + z5.getWidth(), y + z5.getHeight(), z5.getRGB(x, y));
+          kernel.setRGB(x + 2 * z5.getWidth(), y + z5.getHeight(), z6.getRGB(x, y));
+          kernel.setRGB(x, y + 2 * z5.getHeight(), z7.getRGB(x, y));
+          kernel.setRGB(x + z5.getWidth(), y + 2 * z5.getHeight(), z8.getRGB(x, y));
+          kernel.setRGB(x + 2 * z5.getWidth(), y + 2 * z5.getHeight(), z9.getRGB(x, y));
+        }
+      }
+      return kernel;
+    }
+
+    @Override
+    public ByteBuffer read(TileCoord tileCoord) throws TileStoreException {
+      try {
+
+        var image = cache.get(tileCoord);
+        var kernel = getKernel(tileCoord, cache::get);
+        var buffer = kernel.getSubimage(
+            image.getWidth() - 1,
+            image.getHeight() - 1,
+            image.getWidth() + 2,
+            image.getHeight() + 2);
+
+        var grid = ElevationUtils.imageToGrid(buffer);
+        var hillshade = org.apache.baremaps.raster.elevation.HillShade.hillShade(grid,
+            buffer.getWidth(), buffer.getHeight(), 45, 315);
+
+        // Create an output image
+        BufferedImage hillshadeImage =
+            new BufferedImage(image.getWidth(), image.getHeight(), BufferedImage.TYPE_BYTE_GRAY);
+        for (int y = 0; y < image.getHeight(); y++) {
+          for (int x = 0; x < image.getWidth(); x++) {
+            int value = (int) hillshade[(y + 1) * buffer.getHeight() + x + 1];
+            hillshadeImage.setRGB(x, y, new Color(value, value, value).getRGB());
+          }
+        }
+
+        try (var baos = new ByteArrayOutputStream()) {
+          ImageIO.write(hillshadeImage, "png", baos);
+          baos.flush();
+          return ByteBuffer.wrap(baos.toByteArray());
+        }
+      } catch (IOException e) {
+        throw new RuntimeException(e);
+      }
+    }
+
+    @Override
+    public void write(TileCoord tileCoord, ByteBuffer blob) throws TileStoreException {
+      throw new UnsupportedOperationException();
+    }
+
+    @Override
+    public void delete(TileCoord tileCoord) throws TileStoreException {
+      throw new UnsupportedOperationException();
+    }
+
+    @Override
+    public void close() throws Exception {
+      // Do nothing
+    }
+  }
+
+  public class HillShadeTileResource {
+
+    private static final Logger logger =
+        LoggerFactory.getLogger(org.apache.baremaps.server.TileResource.class);
+
+    // public static final String TILE_ENCODING = "gzip";
+
+    public static final String TILE_TYPE = "image/png";
+
+    private final Supplier<TileStore> tileStoreSupplier;
+
+    public HillShadeTileResource(Supplier<TileStore> tileStoreSupplier) {
+      this.tileStoreSupplier = tileStoreSupplier;
+    }
+
+    @Get("regex:^/tiles/(?<z>[0-9]+)/(?<x>[0-9]+)/(?<y>[0-9]+).png")
+    @Blocking
+    public HttpResponse tile(@Param("z") int z, @Param("x") int x, @Param("y") int y) {
+      TileCoord tileCoord = new TileCoord(x, y, z);
+      try {
+        TileStore tileStore = tileStoreSupplier.get();
+        ByteBuffer blob = tileStore.read(tileCoord);
+        if (blob != null) {
+          var headers = ResponseHeaders.builder(200)
+              .add(CONTENT_TYPE, TILE_TYPE)
+              // .add(CONTENT_ENCODING, TILE_ENCODING)
+              .add(ACCESS_CONTROL_ALLOW_ORIGIN, "*")
+              .build();
+          byte[] bytes = new byte[blob.remaining()];
+          blob.get(bytes);
+          HttpData data = HttpData.wrap(bytes);
+          return HttpResponse.of(headers, data);
+        } else {
+          return HttpResponse.of(204);
+        }
+      } catch (TileStoreException ex) {
+        logger.error("Error while reading tile.", ex);
+        return HttpResponse.of(404);
+      }
+    }
+  }
+}

baremaps-cli/src/main/java/org/apache/baremaps/cli/raster/Raster.java

@@ -0,0 +1,34 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to you under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.baremaps.cli.raster;
+
+
+
+import picocli.CommandLine;
+import picocli.CommandLine.Command;
+
+@Command(name = "raster", description = "Raster processing commands.",
+    subcommands = {HillShade.class},
+    sortOptions = false)
+public class Raster implements Runnable {
+
+  @Override
+  public void run() {
+    CommandLine.usage(this, System.out);
+  }
+}

baremaps-core/src/main/java/org/apache/baremaps/tilestore/TileCoord.java

@@ -191,6 +191,42 @@ public TileCoord parent() {
     return new TileCoord(x / 2, y / 2, z - 1);
   }
 
+  /**
+   * Returns the tile coordinate at the left of the current tile.
+   *
+   * @return the left tile
+   */
+  public TileCoord left() {
+    return new TileCoord(x - 1, y, z);
+  }
+
+  /**
+   * Returns the tile coordinate at the right of the current tile.
+   *
+   * @return the right tile
+   */
+  public TileCoord right() {
+    return new TileCoord(x + 1, y, z);
+  }
+
+  /**
+   * Returns the tile coordinate at the top of the current tile.
+   *
+   * @return the top tile
+   */
+  public TileCoord top() {
+    return new TileCoord(x, y - 1, z);
+  }
+
+  /**
+   * Returns the tile coordinate at the bottom of the current tile.
+   *
+   * @return the bottom tile
+   */
+  public TileCoord bottom() {
+    return new TileCoord(x, y + 1, z);
+  }
+
   /**
    * Returns the envelope of the tile coordinate.
    *

baremaps-raster/pom.xml

@@ -0,0 +1,20 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd">
+  <modelVersion>4.0.0</modelVersion>
+  <parent>
+    <groupId>org.apache.baremaps</groupId>
+    <artifactId>baremaps</artifactId>
+    <version>0.7.4-SNAPSHOT</version>
+  </parent>
+  <artifactId>baremaps-raster</artifactId>
+  <dependencies>
+    <dependency>
+      <groupId>com.twelvemonkeys.imageio</groupId>
+      <artifactId>imageio-tiff</artifactId>
+    </dependency>
+    <dependency>
+      <groupId>org.locationtech.jts</groupId>
+      <artifactId>jts-core</artifactId>
+    </dependency>
+  </dependencies>
+</project>