| @@ 269-316 (lines=48) @@ | ||
| 266 | ||
| 267 | return qlat, qlon |
|
| 268 | ||
| 269 | def _qd2apex_nonvectorized(self, qlat, qlon, height): |
|
| 270 | """Converts quasi-dipole to modified apex coordinates. |
|
| 271 | ||
| 272 | Parameters |
|
| 273 | ---------- |
|
| 274 | qlat : float |
|
| 275 | Quasi-dipole latitude |
|
| 276 | qlon : float |
|
| 277 | Quasi-dipole longitude |
|
| 278 | height : float |
|
| 279 | Altitude in km |
|
| 280 | ||
| 281 | Returns |
|
| 282 | ------- |
|
| 283 | alat : float |
|
| 284 | Modified apex latitude |
|
| 285 | alon : float |
|
| 286 | Modified apex longitude |
|
| 287 | ||
| 288 | Raises |
|
| 289 | ------ |
|
| 290 | ApexHeightError |
|
| 291 | if apex height < reference height |
|
| 292 | ||
| 293 | """ |
|
| 294 | # Evaluate the latitude |
|
| 295 | qlat = helpers.checklat(qlat, name='qlat') |
|
| 296 | ||
| 297 | # Get the longitude and apex height |
|
| 298 | alon = qlon |
|
| 299 | h_apex = self.get_apex(qlat, height) |
|
| 300 | ||
| 301 | if h_apex < self.refh: |
|
| 302 | if np.isclose(h_apex, self.refh, rtol=0, atol=1e-5): |
|
| 303 | # Allow for values that are close |
|
| 304 | h_apex = self.refh |
|
| 305 | else: |
|
| 306 | estr = ''.join(['apex height ({:.3g}) is < '.format(h_apex), |
|
| 307 | 'reference height ({:.3g})'.format(self.refh), |
|
| 308 | ' for qlat {:.3g}'.format(qlat)]) |
|
| 309 | raise ApexHeightError(estr) |
|
| 310 | ||
| 311 | # Convert the latitude |
|
| 312 | sqlat = np.sign(qlat) if qlat != 0 else 1 |
|
| 313 | alat = sqlat * np.degrees(np.arccos(np.sqrt((self.RE + self.refh) |
|
| 314 | / (self.RE + h_apex)))) |
|
| 315 | ||
| 316 | return alat, alon |
|
| 317 | ||
| 318 | def _map_EV_to_height(self, alat, alon, height, newheight, data, ev_flag): |
|
| 319 | """Maps electric field related values to a desired height |
|
| @@ 223-267 (lines=45) @@ | ||
| 220 | # ------------------------- |
|
| 221 | # Define the hidden methods |
|
| 222 | ||
| 223 | def _apex2qd_nonvectorized(self, alat, alon, height): |
|
| 224 | """Convert from apex to quasi-dipole (not-vectorised) |
|
| 225 | ||
| 226 | Parameters |
|
| 227 | ----------- |
|
| 228 | alat : (float) |
|
| 229 | Apex latitude in degrees |
|
| 230 | alon : (float) |
|
| 231 | Apex longitude in degrees |
|
| 232 | height : (float) |
|
| 233 | Height in km |
|
| 234 | ||
| 235 | Returns |
|
| 236 | --------- |
|
| 237 | qlat : (float) |
|
| 238 | Quasi-dipole latitude in degrees |
|
| 239 | qlon : (float) |
|
| 240 | Quasi-diplole longitude in degrees |
|
| 241 | ||
| 242 | """ |
|
| 243 | # Evaluate the latitude |
|
| 244 | alat = helpers.checklat(alat, name='alat') |
|
| 245 | ||
| 246 | # Convert modified apex to quasi-dipole, longitude is the same |
|
| 247 | qlon = alon |
|
| 248 | ||
| 249 | # Get the apex height |
|
| 250 | h_apex = self.get_apex(alat) |
|
| 251 | ||
| 252 | if h_apex < height: |
|
| 253 | if np.isclose(h_apex, height, rtol=0, atol=1e-5): |
|
| 254 | # Allow for values that are close |
|
| 255 | h_apex = height |
|
| 256 | else: |
|
| 257 | estr = ''.join(['height {:.3g} is > '.format(np.max(height)), |
|
| 258 | 'apex height {:.3g} for alat {:.3g}'.format( |
|
| 259 | h_apex, alat)]) |
|
| 260 | raise ApexHeightError(estr) |
|
| 261 | ||
| 262 | # Convert the latitude |
|
| 263 | salat = np.sign(alat) if alat != 0 else 1 |
|
| 264 | qlat = salat * np.degrees(np.arccos(np.sqrt((self.RE + height) |
|
| 265 | / (self.RE + h_apex)))) |
|
| 266 | ||
| 267 | return qlat, qlon |
|
| 268 | ||
| 269 | def _qd2apex_nonvectorized(self, qlat, qlon, height): |
|
| 270 | """Converts quasi-dipole to modified apex coordinates. |
|