Specialize 'Proj' for 'SubDomain'

JuliaGeometry · Dec 18, 2024 · 575132c · 575132c
1 parent 4dbd2e2
commit 575132c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/transforms/proj.jl b/src/transforms/proj.jl
@@ -71,6 +71,8 @@ applycoord(t::Proj, g::RectilinearGrid) = TransformedGrid(g, t)
 
 applycoord(t::Proj, g::StructuredGrid) = TransformedGrid(g, t)
 
+applycoord(t::Proj, d::SubDomain) = TransformedDomain(d, t)
+
 # -----------
 # IO METHODS
 # -----------

diff --git a/test/transforms.jl b/test/transforms.jl
@@ -1414,6 +1414,16 @@ end
   r, c = TB.apply(f, d)
   @test r ≈ SimpleMesh(f.(vertices(d)), topology(d))
 
+  # ----------
+  # SUBDOMAIN
+  # ----------
+
+  f = Proj(Polar)
+  g = CartesianGrid((10, 10), cart(1, 1), T.((1, 1)))
+  d = view(g, 1:10)
+  r, c = TB.apply(f, d)
+  @test r ≈ view(SimpleMesh(f.(vertices(g)), topology(g)), 1:10)
+
   # --------------
   # SPECIAL CASES
   # --------------