1+ using jlmie
2+ using Plots
3+
4+ # # xz plot range
5+ # x = (-150:5:150)
6+ # y = 0.0
7+ # z = (-150:5:150)
8+ # # calculation using broadcast
9+ # r, theta, phi = cart2sph(x',y,z)
10+ # # heatmap(x,z,r)
11+ # # heatmap(x,z,theta)
12+ # # heatmap(x,z,phi)
13+
14+ # meshgrid type calculation
15+ # mat_x = x' .* ones(length(z))
16+ # mat_z = ones(length(x))' .* z
17+ # mat_r,mat_theta,mat_phi = cart2sph(mat_x,y,mat_z)
18+ # calculation using broadcast
19+ # r, theta, phi = cart2sph(x',y,z)
20+ # heatmap(x,z,r)
21+ # heatmap(x,z,theta)
22+ # heatmap(x,z,phi)
23+
24+ x = (- 150 : 1 : 150 )
25+ y = (- 150 : 1 : 150 )
26+ z = (- 150 : 1 : 150 )
27+
28+ xyz = 2 # 1:x, 2:y, 3:z
29+ if xyz == 1
30+ # xz plot range
31+ mat_x = x' .* ones(length(z))
32+ mat_z = ones(length(x))' .* z
33+ mat_r,mat_theta,mat_phi = cart2sph(mat_x,0 ,mat_z)
34+ p1 = heatmap(x,z,mat_r)
35+ p2 = heatmap(x,z,mat_theta)
36+ p3 = heatmap(x,z,mat_phi)
37+ elseif xyz == 2
38+ # yz plot range
39+ mat_y = y' .* ones(length(y))
40+ mat_z = ones(length(y))' .* z
41+ mat_r,mat_theta,mat_phi = cart2sph(0 ,mat_y,mat_z)
42+ p1 = heatmap(y,z,mat_r)
43+ p2 = heatmap(y,z,mat_theta)
44+ p3 = heatmap(y,z,mat_phi)
45+ elseif xyz == 3
46+ # xy plot range
47+ mat_x = x' .* ones(length(y))
48+ mat_y = ones(length(x))' .* y
49+ mat_r,mat_theta,mat_phi = cart2sph(mat_x,mat_y,0 )
50+ p1 = heatmap(x,y,mat_r)
51+ p2 = heatmap(x,y,mat_theta)
52+ p3 = heatmap(x,y,mat_phi)
53+ end
54+ plot(p1,p2,p3)
0 commit comments