From 67179029d6884f31a358c76b63a6ba9436540a62 Mon Sep 17 00:00:00 2001
From: Anshul Singhvi <anshulsinghvi@gmail.com>
Date: Thu, 9 Jan 2025 00:00:36 -0100
Subject: [PATCH] Add crs, reproject

---
 src/geometry_lookup.jl | 33 ++++++++++++++++++++++++++++-----
 1 file changed, 28 insertions(+), 5 deletions(-)

diff --git a/src/geometry_lookup.jl b/src/geometry_lookup.jl
index 98d017af..f070928f 100644
--- a/src/geometry_lookup.jl
+++ b/src/geometry_lookup.jl
@@ -31,13 +31,14 @@ dv[Geometry(Contains(GO.centroid(polygons[88])))] == dv[Geometry(88)] # true
 ```
 
 """
-struct GeometryLookup{T, D} <: Lookups.Lookup{T, 1}
+struct GeometryLookup{T, D, CRS} <: Lookups.Lookup{T, 1}
     data::Vector{T}
     tree::SortTileRecursiveTree.STRtree
     dims::D
+    crs::CRS
 end
 
-function GeometryLookup(data, dims = (X(), Y()); geometrycolumn = nothing)
+function GeometryLookup(data, dims = (X(), Y()); geometrycolumn = nothing, crs = nothing)
     # First, retrieve the geometries - from a table, vector of geometries, etc.
     geometries = _get_geometries(data, geometrycolumn)
     # Check that the geometries are actually geometries
@@ -56,7 +57,11 @@ function GeometryLookup(data, dims = (X(), Y()); geometrycolumn = nothing)
     end
     # Build the lookup accelerator tree
     tree = SortTileRecursiveTree.STRtree(geometries)
-    GeometryLookup(geometries, tree, dims)
+
+    true_crs = isnothing(crs) ? GI.crs(first(geometries)) : crs
+        
+
+    GeometryLookup(geometries, tree, dims, crs)
 end
 
 #=
@@ -68,6 +73,7 @@ This is broadly standard except for the `rebuild` method, which is used to updat
 =#
 
 DD.dims(l::GeometryLookup) = l.dims
+
 DD.dims(d::DD.Dimension{<: GeometryLookup}) = val(d).dims
  
 DD.order(::GeometryLookup) = Lookups.Unordered()
@@ -77,13 +83,19 @@ DD.parent(lookup::GeometryLookup) = lookup.data
 DD.Dimensions.format(l::GeometryLookup, D::Type, values, axis::AbstractRange) = l
 
 # Make sure that the tree is rebuilt if the data changes
-function DD.rebuild(lookup::GeometryLookup; data = lookup.data, tree = nothing, dims = lookup.dims)
+function DD.rebuild(lookup::GeometryLookup; data = lookup.data, tree = nokw, dims = lookup.dims, crs = nokw)
     new_tree = if data == lookup.data
         lookup.tree
     else
         SortTileRecursiveTree.STRtree(data)
     end
-    GeometryLookup(data, new_tree, dims)
+    new_crs = if isnokw(crs)
+        GI.crs(first(data))
+    else
+        crs
+    end
+
+    GeometryLookup(data, new_tree, dims, new_crs)
 end
 
 #=
@@ -215,6 +227,17 @@ _val_or_nothing(::Nothing) = nothing
 _val_or_nothing(d::DD.Dimension) = val(d)
 
 
+#=
+## Reproject
+
+Reproject just forwards to `GO.reproject`.
+=#
+
+function reproject(target::GeoFormat, l::GeometryLookup)
+    source = crs(l)
+    return rebuild(l; data = GO.reproject(l.data; source_crs = source, target_crs = target, always_xy = true))
+end
+
 # I/O utils
 function _geometry_cf_encode(::Union{GI.PolygonTrait, GI.MultiPolygonTrait}, geoms)
     n_points_per_geom_vec = GI.npoint.(geoms)