Add fast path for indices on non-invertible TransformedGrid

src/indices.jl

@@ -74,6 +74,14 @@ function indices(grid::OrthoRectilinearGrid, box::Box)
   LinearIndices(size(grid))[range] |> vec
 end
 
+function indices(grid::TransformedGrid, geometry::Geometry)
+  # construct reverse transform from revertible steps
+  revtrans = reduce(→, reverse(filter(isrevertible, transform(grid))))
+
+  # find indices in non-transformed space
+  indices(parent(grid), revtrans(geometry))
+end
+
 # ----------------
 # CARTESIAN RANGE
 # ----------------

test/indices.jl

@@ -202,4 +202,15 @@
   @test linds[2, 2] == only(indices(grid, p4))
   @test linds[10, 10] == only(indices(grid, p5))
   @test isempty(indices(grid, p6))
+
+  # transformed grid
+  grid = cartgrid(20, 20)
+  poly = PolyArea(cart.([(3, 3), (9, 9), (3, 15), (17, 15), (17, 3)]))
+  inds = indices(grid, poly)
+  linds = LinearIndices(size(grid))
+  @test linds[12, 6] ∈ indices(grid, poly)
+  trans = Rotate(Angle2d(T(π / 2))) → Morphological(c -> Cartesian(c.x + c.y, c.y, c.x - c.y))
+  tgrid = trans(grid)
+  tpoly = trans(poly)
+  @test indices(tgrid, tpoly) == indices(grid, poly)
 end