From 575132c0857dcf6c6ee7b54631f3d28f2348feda Mon Sep 17 00:00:00 2001 From: Elias Carvalho Date: Wed, 18 Dec 2024 08:28:38 -0300 Subject: [PATCH] Specialize 'Proj' for 'SubDomain' --- src/transforms/proj.jl | 2 ++ test/transforms.jl | 10 ++++++++++ 2 files changed, 12 insertions(+) diff --git a/src/transforms/proj.jl b/src/transforms/proj.jl index 6ca5ae2db..8562486ed 100644 --- 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 index c91c1a024..db313dc71 100644 --- 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 # --------------