9
9
pass
10
10
11
11
12
- def sht_matrix (N , azi , elev , weights = None ):
12
+ def sht_matrix (N , azi , colat , weights = None ):
13
13
r"""Matrix of spherical harmonics up to order N for given angles.
14
14
15
15
Computes a matrix of spherical harmonics up to order :math:`N`
@@ -40,8 +40,8 @@ def sht_matrix(N, azi, elev, weights=None):
40
40
Maximum order.
41
41
azi : (Q,) array_like
42
42
Azimuth.
43
- elev : (Q,) array_like
44
- Elevation .
43
+ colat : (Q,) array_like
44
+ Colatitude .
45
45
weights : (Q,) array_like, optional
46
46
Quadrature weights.
47
47
@@ -52,7 +52,7 @@ def sht_matrix(N, azi, elev, weights=None):
52
52
53
53
"""
54
54
azi = util .asarray_1d (azi )
55
- elev = util .asarray_1d (elev )
55
+ colat = util .asarray_1d (colat )
56
56
if azi .ndim == 0 :
57
57
Q = 1
58
58
else :
@@ -63,7 +63,7 @@ def sht_matrix(N, azi, elev, weights=None):
63
63
i = 0
64
64
for n in range (N + 1 ):
65
65
for m in range (- n , n + 1 ):
66
- Ymn [:, i ] = weights * special .sph_harm (m , n , azi , elev )
66
+ Ymn [:, i ] = weights * special .sph_harm (m , n , azi , colat )
67
67
i += 1
68
68
return Ymn
69
69
@@ -162,25 +162,25 @@ def grid_equal_angle(n):
162
162
-------
163
163
azi : array_like
164
164
Azimuth.
165
- elev : array_like
166
- Elevation .
165
+ colat : array_like
166
+ Colatitude .
167
167
weights : array_like
168
168
Quadrature weights.
169
169
"""
170
170
azi = np .linspace (0 , 2 * np .pi , 2 * n + 2 , endpoint = False )
171
- elev , d_elev = np .linspace (0 , np .pi , 2 * n + 2 , endpoint = False , retstep = True )
172
- elev += d_elev / 2
171
+ colat , d_colat = np .linspace (0 , np .pi , 2 * n + 2 , endpoint = False , retstep = True )
172
+ colat += d_colat / 2
173
173
174
- weights = np .zeros_like (elev )
174
+ weights = np .zeros_like (colat )
175
175
p = np .arange (1 , 2 * n + 2 , 2 )
176
- for i , theta in enumerate (elev ):
176
+ for i , theta in enumerate (colat ):
177
177
weights [i ] = 2 * np .pi / (n + 1 ) * np .sin (theta ) * np .sum (np .sin (p * theta )/ p )
178
178
179
179
azi = np .tile (azi , 2 * n + 2 )
180
- elev = np .repeat (elev , 2 * n + 2 )
180
+ colat = np .repeat (colat , 2 * n + 2 )
181
181
weights = np .repeat (weights , 2 * n + 2 )
182
182
weights /= n + 1 # sum(weights) == 4pi
183
- return azi , elev , weights
183
+ return azi , colat , weights
184
184
185
185
186
186
def grid_gauss (n ):
@@ -197,19 +197,19 @@ def grid_gauss(n):
197
197
-------
198
198
azi : array_like
199
199
Azimuth.
200
- elev : array_like
201
- Elevation .
200
+ colat : array_like
201
+ Colatitude .
202
202
weights : array_like
203
203
Quadrature weights.
204
204
"""
205
205
azi = np .linspace (0 , 2 * np .pi , 2 * n + 2 , endpoint = False )
206
206
x , weights = np .polynomial .legendre .leggauss (n + 1 )
207
- elev = np .arccos (x )
207
+ colat = np .arccos (x )
208
208
azi = np .tile (azi , n + 1 )
209
- elev = np .repeat (elev , 2 * n + 2 )
209
+ colat = np .repeat (colat , 2 * n + 2 )
210
210
weights = np .repeat (weights , 2 * n + 2 )
211
211
weights *= np .pi / (n + 1 ) # sum(weights) == 4pi
212
- return azi , elev , weights
212
+ return azi , colat , weights
213
213
214
214
215
215
def grid_equal_polar_angle (n ):
@@ -248,8 +248,8 @@ def grid_lebedev(n):
248
248
-------
249
249
azi : array_like
250
250
Azimuth.
251
- elev : array_like
252
- Elevation .
251
+ colat : array_like
252
+ Colatitude .
253
253
weights : array_like
254
254
Quadrature weights.
255
255
@@ -273,5 +273,5 @@ def available_quadrature(d):
273
273
if np .any (q .weights < 0 ):
274
274
warn ("Lebedev grid of order {} has negative weights." .format (n ))
275
275
azi = q .azimuthal_polar [:, 0 ]
276
- elev = q .azimuthal_polar [:, 1 ]
277
- return azi , elev , 4 * np .pi * q .weights
276
+ colat = q .azimuthal_polar [:, 1 ]
277
+ return azi , colat , 4 * np .pi * q .weights
0 commit comments