From 45e4dfb247cfe26567b64c820e8760acc98e6dce Mon Sep 17 00:00:00 2001
From: Pete Gadomski <pete.gadomski@gmail.com>
Date: Mon, 6 Oct 2025 16:39:38 -0600
Subject: [PATCH] feat: render visual cogs

---
 src/app.tsx               | 17 ++++++++++++
 src/components/assets.tsx | 55 +++++++++++++++++++++++++++++++++++----
 src/components/map.tsx    | 38 ++++++++++++++++++++++++---
 src/components/value.tsx  | 10 ++++++-
 src/utils/stac.ts         | 19 +++++++++++++-
 5 files changed, 129 insertions(+), 10 deletions(-)

diff --git a/src/app.tsx b/src/app.tsx
index 937545e..b2f1d06 100644
--- a/src/app.tsx
+++ b/src/app.tsx
@@ -9,10 +9,12 @@ import useStacValue from "./hooks/stac-value";
 import type { BBox2D, Color } from "./types/map";
 import type { DatetimeBounds, StacValue } from "./types/stac";
 import {
+  isCog,
   isCollectionInBbox,
   isCollectionInDatetimeBounds,
   isItemInBbox,
   isItemInDatetimeBounds,
+  isVisual,
 } from "./utils/stac";
 
 // TODO make this configurable by the user.
@@ -30,6 +32,7 @@ export default function App() {
   const [datetimeBounds, setDatetimeBounds] = useState<DatetimeBounds>();
   const [filter, setFilter] = useState(true);
   const [stacGeoparquetItemId, setStacGeoparquetItemId] = useState<string>();
+  const [cogTileHref, setCogTileHref] = useState<string>();
 
   // Derived state
   const {
@@ -116,6 +119,17 @@ export default function App() {
     setItems(undefined);
     setDatetimeBounds(undefined);
 
+    let cogTileHref = undefined;
+    if (value && value.assets) {
+      for (const asset of Object.values(value.assets)) {
+        if (isCog(asset) && isVisual(asset)) {
+          cogTileHref = asset.href as string;
+          break;
+        }
+      }
+    }
+    setCogTileHref(cogTileHref);
+
     if (value && (value.title || value.id)) {
       document.title = "stac-map | " + (value.title || value.id);
     } else {
@@ -151,6 +165,7 @@ export default function App() {
               picked={picked}
               setPicked={setPicked}
               setStacGeoparquetItemId={setStacGeoparquetItemId}
+              cogTileHref={cogTileHref}
             ></Map>
           </FileUpload.Dropzone>
         </FileUpload.RootProvider>
@@ -184,6 +199,8 @@ export default function App() {
           filteredItems={filteredItems}
           setItems={setItems}
           setDatetimeBounds={setDatetimeBounds}
+          cogTileHref={cogTileHref}
+          setCogTileHref={setCogTileHref}
         ></Overlay>
       </Container>
       <Toaster></Toaster>
diff --git a/src/components/assets.tsx b/src/components/assets.tsx
index 03ef2ab..5c7372a 100644
--- a/src/components/assets.tsx
+++ b/src/components/assets.tsx
@@ -1,26 +1,41 @@
-import { useState } from "react";
+import { useEffect, useState } from "react";
 import { LuDownload } from "react-icons/lu";
 import {
   Button,
   ButtonGroup,
   Card,
+  Checkbox,
   Collapsible,
   DataList,
   HStack,
   Image,
+  Span,
 } from "@chakra-ui/react";
 import type { StacAsset } from "stac-ts";
 import Properties from "./properties";
 import type { StacAssets } from "../types/stac";
+import { isCog, isVisual } from "../utils/stac";
 
-export default function Assets({ assets }: { assets: StacAssets }) {
+export default function Assets({
+  assets,
+  cogTileHref,
+  setCogTileHref,
+}: {
+  assets: StacAssets;
+  cogTileHref: string | undefined;
+  setCogTileHref: (href: string | undefined) => void;
+}) {
   return (
     <DataList.Root>
       {Object.keys(assets).map((key) => (
         <DataList.Item key={"asset-" + key}>
           <DataList.ItemLabel>{key}</DataList.ItemLabel>
           <DataList.ItemValue>
-            <Asset asset={assets[key]} />
+            <Asset
+              asset={assets[key]}
+              cogTileHref={cogTileHref}
+              setCogTileHref={setCogTileHref}
+            />
           </DataList.ItemValue>
         </DataList.Item>
       ))}
@@ -28,11 +43,24 @@ export default function Assets({ assets }: { assets: StacAssets }) {
   );
 }
 
-function Asset({ asset }: { asset: StacAsset }) {
+function Asset({
+  asset,
+  cogTileHref,
+  setCogTileHref,
+}: {
+  asset: StacAsset;
+  cogTileHref: string | undefined;
+  setCogTileHref: (href: string | undefined) => void;
+}) {
   const [imageError, setImageError] = useState(false);
+  const [checked, setChecked] = useState(false);
   // eslint-disable-next-line
   const { href, roles, type, title, ...properties } = asset;
 
+  useEffect(() => {
+    setChecked(cogTileHref === asset.href);
+  }, [cogTileHref, asset.href]);
+
   return (
     <Card.Root size={"sm"} w="full">
       <Card.Header>
@@ -64,7 +92,24 @@ function Asset({ asset }: { asset: StacAsset }) {
             </Collapsible.Content>
           </Collapsible.Root>
         )}
-        <HStack justify={"right"}>
+        <HStack>
+          {isCog(asset) && isVisual(asset) && (
+            <Checkbox.Root
+              checked={checked}
+              onCheckedChange={(e) => {
+                setChecked(!!e.checked);
+                if (e.checked) setCogTileHref(asset.href);
+                else setCogTileHref(undefined);
+              }}
+            >
+              <Checkbox.HiddenInput />
+              <Checkbox.Control />
+              <Checkbox.Label>Visualize</Checkbox.Label>
+            </Checkbox.Root>
+          )}
+
+          <Span flex={"1"} />
+
           <ButtonGroup size="sm" variant="outline">
             <Button asChild>
               <a href={asset.href} target="_blank">
diff --git a/src/components/map.tsx b/src/components/map.tsx
index 579b85f..7b77caa 100644
--- a/src/components/map.tsx
+++ b/src/components/map.tsx
@@ -5,7 +5,8 @@ import {
   useControl,
 } from "react-map-gl/maplibre";
 import { type DeckProps, Layer } from "@deck.gl/core";
-import { GeoJsonLayer } from "@deck.gl/layers";
+import { TileLayer } from "@deck.gl/geo-layers";
+import { BitmapLayer, GeoJsonLayer } from "@deck.gl/layers";
 import { MapboxOverlay } from "@deck.gl/mapbox";
 import { GeoArrowPolygonLayer } from "@geoarrow/deck.gl-layers";
 import bbox from "@turf/bbox";
@@ -31,6 +32,7 @@ export default function Map({
   setPicked,
   table,
   setStacGeoparquetItemId,
+  cogTileHref,
 }: {
   value: StacValue | undefined;
   collections: StacCollection[] | undefined;
@@ -44,6 +46,7 @@ export default function Map({
   setPicked: (picked: StacValue | undefined) => void;
   table: Table | undefined;
   setStacGeoparquetItemId: (id: string | undefined) => void;
+  cogTileHref: string | undefined;
 }) {
   const mapRef = useRef<MapRef>(null);
   const mapStyle = useColorModeValue(
@@ -87,7 +90,36 @@ export default function Map({
     fillColor[3],
   ];
 
-  const layers: Layer[] = [
+  let layers: Layer[] = [];
+
+  if (cogTileHref)
+    layers.push(
+      new TileLayer({
+        id: "cog-tiles",
+        extent: value && getBbox(value, collections),
+        maxRequests: 10,
+        data:
+          cogTileHref &&
+          `https://titiler.xyz/cog/tiles/WebMercatorQuad/{z}/{x}/{y}.png?url=${cogTileHref}`,
+        renderSubLayers: (props) => {
+          const { boundingBox } = props.tile;
+
+          return new BitmapLayer(props, {
+            data: undefined,
+            image: props.data,
+            bounds: [
+              boundingBox[0][0],
+              boundingBox[0][1],
+              boundingBox[1][0],
+              boundingBox[1][1],
+            ],
+          });
+        },
+      })
+    );
+
+  layers = [
+    ...layers,
     new GeoJsonLayer({
       id: "picked",
       data: pickedGeoJson,
@@ -121,7 +153,7 @@ export default function Map({
     new GeoJsonLayer({
       id: "value",
       data: valueGeoJson,
-      filled: !items,
+      filled: !items && !cogTileHref,
       getFillColor: collections ? inverseFillColor : fillColor,
       getLineColor: collections ? inverseLineColor : lineColor,
       getLineWidth: 2,
diff --git a/src/components/value.tsx b/src/components/value.tsx
index 6372227..886b7a0 100644
--- a/src/components/value.tsx
+++ b/src/components/value.tsx
@@ -62,6 +62,8 @@ export interface SharedValueProps {
   bbox: BBox2D | undefined;
   setItems: (items: StacItem[] | undefined) => void;
   setDatetimeBounds: (bounds: DatetimeBounds | undefined) => void;
+  cogTileHref: string | undefined;
+  setCogTileHref: (href: string | undefined) => void;
 }
 
 interface ValueProps extends SharedValueProps {
@@ -84,6 +86,8 @@ export function Value({
   setFilter,
   bbox,
   setDatetimeBounds,
+  cogTileHref,
+  setCogTileHref,
 }: ValueProps) {
   const [search, setSearch] = useState<StacSearch>();
   const [numberOfCollections, setNumberOfCollections] = useState<number>();
@@ -369,7 +373,11 @@ export function Value({
             AccordionIcon={LuFiles}
             accordionValue="assets"
           >
-            <Assets assets={assets} />
+            <Assets
+              assets={assets}
+              cogTileHref={cogTileHref}
+              setCogTileHref={setCogTileHref}
+            />
           </Section>
         )}
 
diff --git a/src/utils/stac.ts b/src/utils/stac.ts
index 442953d..57ed0b6 100644
--- a/src/utils/stac.ts
+++ b/src/utils/stac.ts
@@ -1,5 +1,5 @@
 import type { UseFileUploadReturn } from "@chakra-ui/react";
-import type { StacCollection, StacItem, StacLink } from "stac-ts";
+import type { StacAsset, StacCollection, StacItem, StacLink } from "stac-ts";
 import type { BBox2D } from "../types/map";
 import type { DatetimeBounds, StacAssets, StacValue } from "../types/stac";
 
@@ -250,3 +250,20 @@ export function getImportantLinks(links: StacLink[]) {
   }
   return { rootLink, collectionsLink, nextLink, prevLink, filteredLinks };
 }
+
+export function isCog(asset: StacAsset) {
+  return (
+    asset.type === "image/tiff; application=geotiff; profile=cloud-optimized"
+  );
+}
+
+export function isVisual(asset: StacAsset) {
+  if (asset.roles) {
+    for (const role of asset.roles) {
+      if (role === "visual" || role === "thumbnail") {
+        return true;
+      }
+    }
+  }
+  return false;
+}