disk plotting works

Author: dlfivefifty

examples/diskhelmholtz.jl

@@ -1,7 +1,32 @@
-using MultivariateOrthogonalPolynomials, FastTransforms, BlockBandedMatrices, WGLMakie
-import MultivariateOrthogonalPolynomials: ZernikeITransform, grid
+using MultivariateOrthogonalPolynomials, FastTransforms, BlockBandedMatrices, Plots
+plotly()
+import MultivariateOrthogonalPolynomials: ZernikeITransform, grid, plotgrid
+
 
 Z = Zernike()[:,Block.(Base.OneTo(10))]
+g = grid(Z)
+θ = getproperty.(g[1,:],:θ)
+[permutedims(RadialCoordinate.(1,θ)); g; permutedims(RadialCoordinate.(0,θ))]
+import Mul
+
+plotgrid(Z)
+
+
+Z = Zernike()
+xy = axes(Z,1)
+x,y = first.(xy),last.(xy)
+u = Z * (Z \ @.(cos(10x*y)))
+surface(u)
+
+@.(cos(10x*y))
+
+plot(u)
+
+g = plotgrid(Z[:,Block.(Base.OneTo(10))])
+surface(first.(g), last.(g), u[g])
+plot(u)
+
+
 G = grid(Z)
 
 contourf(first.(G), last.(G), ones(size(G)...))

src/MultivariateOrthogonalPolynomials.jl

@@ -8,7 +8,7 @@ import Base: axes, in, ==, *, ^, \, copy, OneTo, getindex, size
 import DomainSets: boundary
 
 import QuasiArrays: LazyQuasiMatrix, LazyQuasiArrayStyle
-import ContinuumArrays: @simplify, Weight, grid, TransformFactorization, Expansion
+import ContinuumArrays: @simplify, Weight, grid, plotgrid, TransformFactorization, Expansion
 
 import ArrayLayouts: MemoryLayout
 import BlockArrays: block, blockindex, BlockSlice, viewblock

src/disk.jl

@@ -178,6 +178,21 @@ function grid(S::FiniteZernike{T}) where T
     RadialCoordinate.(r, π*θ')
 end
 
+_angle(rθ::RadialCoordinate) = rθ.θ
+
+function plotgrid(S::FiniteZernike{T}) where T
+    N = blocksize(S,2) ÷ 2 + 1 # polynomial degree
+    g = grid(parent(S)[:,Block.(OneTo(2N))]) # double sampling
+    θ = [map(_angle,g[1,:]); 0]
+    [permutedims(RadialCoordinate.(1,θ)); g g[:,1]; permutedims(RadialCoordinate.(0,θ))]
+end
+
+function plotgrid(S::SubQuasiArray{<:Any,2,<:Zernike})
+    kr,jr = parentindices(S)
+    Z = parent(S)
+    plotgrid(Z[kr,Block.(OneTo(Int(findblock(axes(Z,2),maximum(jr)))))])
+end
+
 struct ZernikeTransform{T} <: Plan{T}
     N::Int
     disk2cxf::FastTransforms.FTPlan{T,2,FastTransforms.DISK}