Add auto_webmercator to COG

Cargo.toml

@@ -49,6 +49,8 @@ env_logger = "0.11"
 flate2 = "1"
 flume = "0.11"
 futures = "0.3"
+image = "0.25.6"
+imageproc = "0.25.0"
 indoc = "2"
 insta = "1"
 itertools = "0.14"

martin-tile-utils/src/lib.rs

@@ -267,6 +267,30 @@ pub fn tile_index(lng: f64, lat: f64, zoom: u8) -> (u32, u32) {
     (col, row)
 }
 
+/// Convert min/max XYZ tile coordinates to a bounding box values in Web Mercator.
+#[must_use]
+pub fn xyz_to_bbox_webmercator(
+    zoom: u8,
+    min_x: u32,
+    min_y: u32,
+    max_x: u32,
+    max_y: u32,
+) -> [f64; 4] {
+    assert!(zoom <= MAX_ZOOM, "zoom {zoom} must be <= {MAX_ZOOM}");
+
+    let tile_length = EARTH_CIRCUMFERENCE / f64::from(1_u32 << zoom);
+
+    let left_down_bbox = tile_bbox(min_x, max_y, tile_length);
+    let right_top_bbox = tile_bbox(max_x, min_y, tile_length);
+
+    [
+        left_down_bbox[0],
+        left_down_bbox[1],
+        right_top_bbox[2],
+        right_top_bbox[3],
+    ]
+}
+
 /// Convert min/max XYZ tile coordinates to a bounding box values.
 ///
 /// The result is `[min_lng, min_lat, max_lng, max_lat]`

martin/Cargo.toml

@@ -53,7 +53,7 @@ harness = false
 
 [features]
 default = ["cog", "fonts", "lambda", "mbtiles", "metrics", "pmtiles", "postgres", "sprites", "styles", "webui"]
-cog = ["dep:png", "dep:tiff"]
+cog = ["dep:tiff", "dep:png", "dep:image", "dep:imageproc"]
 fonts = ["dep:bit-set", "dep:pbf_font_tools"]
 lambda = ["dep:lambda-web"]
 mbtiles = ["dep:mbtiles"]
@@ -81,6 +81,8 @@ deadpool-postgres = { workspace = true, optional = true }
 enum-display.workspace = true
 env_logger.workspace = true
 futures.workspace = true
+image = { workspace = true, optional = true }
+imageproc = { workspace = true, optional = true }
 itertools.workspace = true
 json-patch = { workspace = true, optional = true }
 lambda-web = { workspace = true, optional = true }

martin/src/cog/config.rs

@@ -13,6 +13,11 @@ use crate::file_config::{ConfigExtras, FileResult, SourceConfigExtras};
 pub struct CogConfig {
     #[serde(flatten)]
     pub unrecognized: UnrecognizedValues,
+
+    /// Default false
+    /// If enabled, martin will automatically serve COG as a [WebMercatorQuad](https://docs.ogc.org/is/17-083r2/17-083r2.html#72) service, the tiles will be cliped and merged internally to be aligned with the Web Mercator grid.
+    /// Note: Just work for COG files with a Web Mercator CRS (EPSG:3857).
+    pub auto_web: Option<bool>,
 }
 
 impl ConfigExtras for CogConfig {
@@ -27,7 +32,32 @@ impl SourceConfigExtras for CogConfig {
     }
 
     async fn new_sources(&self, id: String, path: PathBuf) -> FileResult<Box<dyn Source>> {
-        let cog = CogSource::new(id, path)?;
+        let cog = CogSource::new(id, path, self.auto_web.unwrap_or(false))?;
+        Ok(Box::new(cog))
+    }
+
+    async fn new_sources_with_config(
+        &self,
+        id: String,
+        path: PathBuf,
+        config: serde_yaml::Value,
+    ) -> FileResult<Box<dyn Source>> {
+        let source_auto_web = if let serde_yaml::Value::Mapping(map) = &config {
+            if let Some(auto_web_value) = map.get(serde_yaml::Value::String("auto_web".to_string()))
+            {
+                match auto_web_value {
+                    serde_yaml::Value::Bool(b) => Some(*b),
+                    _ => None,
+                }
+            } else {
+                None
+            }
+        } else {
+            None
+        };
+
+        let auto_web = source_auto_web.unwrap_or_else(|| self.auto_web.unwrap_or(false));
+        let cog = CogSource::new(id, path, auto_web)?;
         Ok(Box::new(cog))
     }
 

martin/src/cog/errors.rs

@@ -63,4 +63,7 @@ pub enum CogError {
         "Get full resolution failed for {0}: either a valid ModelPixelScaleTag or ModelPixelScaleTag is required"
     )]
     GetFullResolutionFailed(PathBuf),
+
+    #[error("Failed to create image buffer for {0}: {1}")]
+    ImageBufferCreationFailed(PathBuf, String),
 }