|
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