|
1 | 1 | using MultivariateOrthogonalPolynomials, FastTransforms, BlockBandedMatrices, Plots
|
2 | 2 | plotly()
|
3 |
| -import MultivariateOrthogonalPolynomials: ZernikeITransform, grid, plotgrid |
4 | 3 |
|
| 4 | +Z = Zernike(1) |
| 5 | +W = Weighted(Z) |
| 6 | +xy = axes(Z,1); x,y = first.(xy),last.(xy) |
| 7 | +Δ = Z \ (Laplacian(xy) * W) |
| 8 | +S = Z \ W |
| 9 | +k = 2 |
| 10 | +f = @.(cos(x*exp(y))) |
| 11 | +F = factorize(Δ + k^2 * S) |
| 12 | +c = (Z \ f) |
| 13 | +F \ c |
5 | 14 |
|
6 |
| -Z = Zernike()[:,Block.(Base.OneTo(10))] |
7 |
| -g = grid(Z) |
8 |
| -θ = getproperty.(g[1,:],:θ) |
9 |
| -[permutedims(RadialCoordinate.(1,θ)); g; permutedims(RadialCoordinate.(0,θ))] |
10 |
| -import Mul |
| 15 | +u = W * ((Δ + k^2 * S) \ (Z \ f)) |
11 | 16 |
|
12 |
| -plotgrid(Z) |
13 |
| - |
14 |
| - |
15 |
| -Z = Zernike() |
16 |
| -xy = axes(Z,1) |
17 |
| -x,y = first.(xy),last.(xy) |
18 |
| -u = Z * (Z \ @.(cos(10x*y))) |
19 |
| -surface(u) |
20 |
| - |
21 |
| -@.(cos(10x*y)) |
22 |
| - |
23 |
| -plot(u) |
24 |
| - |
25 |
| -g = plotgrid(Z[:,Block.(Base.OneTo(10))]) |
26 |
| -surface(first.(g), last.(g), u[g]) |
27 |
| -plot(u) |
28 |
| - |
29 |
| - |
30 |
| -G = grid(Z) |
31 |
| - |
32 |
| -contourf(first.(G), last.(G), ones(size(G)...)) |
33 |
| - |
34 |
| - |
35 |
| -scatter(vec(first.(G)), vec(last.(G))) |
36 |
| -G[1,1].θ |
37 |
| - |
38 |
| -WZ = Weighted(Zernike(1)) |
39 |
| -xy = axes(WZ,1) |
40 |
| -x,y = first.(xy),last.(xy) |
41 |
| -f = Zernike(1) \ @. exp(x*cos(y)) |
42 |
| - |
43 |
| -N = 50 |
44 |
| -KR = Block.(Base.OneTo(N)) |
45 |
| -Δ = (Zernike(1) \ (Laplacian(xy) * WZ))[KR,KR] |
46 |
| -C = (Zernike(1) \ WZ)[KR,KR] |
47 |
| -k = 5 |
48 |
| -L = Δ - k^2 * C |
49 |
| - |
50 |
| -v = f[KR] |
51 |
| -@time u = (L \ v); |
52 |
| - |
53 |
| -g = MultivariateOrthogonalPolynomials.grid(Zernike(1)[:,KR]) |
54 |
| -U = ZernikeITransform{Float64}(N, 0, 1) * u |
55 |
| - |
56 |
| -plot(first.(g), last.(g), U) |
57 |
| - |
58 |
| - |
59 |
| - |
60 |
| -F = factorize(Zernike(1)[:,KR]).plan |
61 |
| -F \ u |
62 |
| -u |
63 |
| - |
64 |
| - |
65 |
| -F*u |
66 |
| - |
67 |
| - |
68 |
| -F.plan \ v |
69 |
| -F |> typeof |> fieldnames |
70 |
| - |
71 |
| -grid(WZ[:,KR]) |
72 |
| - |
73 |
| -F |>typeof |> fieldnames |
74 |
| -F.F * v |
75 |
| - |
76 |
| -m = DiskTrav(v).matrix |
77 |
| - |
78 |
| -plan_disk2cxf(m, 0, 0) * m |
79 | 17 |
|
80 | 18 |
|
0 commit comments