def test_map_V_to_height():
A = Apex(date=2000, refh=300)
out_60_15_100_500 = [0.819719, 2.845114, 0.695437]
out_60_15_100_500_234 = [1.830277, 4.14345, 0.947624]
out_60_15_100_1000 = [0.924577, 3.149964, 0.851343]
out_60_15_200_500 = [0.803882, 2.793206, 0.682839]
out_60_30_100_500 = [0.761412, 2.878837, 0.736549]
out_70_15_100_500 = [0.846819, 2.592572, 0.347919]
# scalar
assert_allclose(A.map_V_to_height(60, 15, 100, 500, [1, 2, 3]), out_60_15_100_500, rtol=1e-5)
assert_allclose(A.map_V_to_height(60, 15, 100, 500, [2, 3, 4]), out_60_15_100_500_234, rtol=1e-5)
assert_allclose(A.map_V_to_height(60, 15, 100, 1000, [1, 2, 3]), out_60_15_100_1000, rtol=1e-5)
assert_allclose(A.map_V_to_height(60, 15, 200, 500, [1, 2, 3]), out_60_15_200_500, rtol=1e-5)
assert_allclose(A.map_V_to_height(60, 30, 100, 500, [1, 2, 3]), out_60_30_100_500, rtol=1e-5)
assert_allclose(A.map_V_to_height(70, 15, 100, 500, [1, 2, 3]), out_70_15_100_500, rtol=1e-5)
# vectorize lat
assert_allclose(A.map_V_to_height([60, 70], 15, 100, 500, np.array([[1, 2, 3]]*2).T),
np.array([out_60_15_100_500, out_70_15_100_500]).T, rtol=1e-5)
# vectorize lon
assert_allclose(A.map_V_to_height(60, [15, 30], 100, 500, np.array([[1, 2, 3]]*2).T),
np.array([out_60_15_100_500, out_60_30_100_500]).T, rtol=1e-5)
# vectorize height
assert_allclose(A.map_V_to_height(60, 15, [100, 200], 500, np.array([[1, 2, 3]]*2).T),
np.array([out_60_15_100_500, out_60_15_200_500]).T, rtol=1e-5)
# vectorize newheight
assert_allclose(A.map_V_to_height(60, 15, 100, [500, 1000], np.array([[1, 2, 3]]*2).T),
np.array([out_60_15_100_500, out_60_15_100_1000]).T, rtol=1e-5)
# vectorize E
assert_allclose(A.map_V_to_height(60, 15, 100, 500, np.array([[1, 2, 3], [2, 3, 4]]).T),
np.array([out_60_15_100_500, out_60_15_100_500_234]).T, rtol=1e-5)
class Field(object):
def __init__(self, *args, **kwargs):
if len(args) == 1:
self.read_config(args[0])
else:
self.apex_year = kwargs['apex_year']
self.field_coords = np.array(kwargs['field_coords'])
self.field_values = np.array(kwargs['field_values'])
# initialize Apex object
self.apex = Apex(date=self.apex_year)
self.map_velocity_field(self.field_coords, self.field_values)
self.convert_to_ECEF()
self.create_interpolators()
def read_config(self, config_file):
config = configparser.ConfigParser()
config.read(config_file)
self.apex_year = config.getint('FIELD', 'apex_year')
self.field_coords = np.array(eval(config.get('FIELD', 'field_coords')))
self.field_values = np.array(eval(config.get('FIELD', 'field_values')))
def map_velocity_field(self, coords, field):
# coords - array (N,3) of geodetic lat, lon, alt
# field - array (N,3) of geodetic E, N, U components of the velocity field at each position
# define output altitudes
altitude = np.arange(50., 1000., 50.)
# create array in proper shape to be applied to every input coordinate
self.altitude = np.repeat(altitude, coords.shape[-1])
# map to diffent altitudes manually - the current expected input/output arrays of apexpy.map_to_height makes this function difficult to use for this purpose
alat, alon = self.apex.geo2apex(coords[0], coords[1], coords[2])
# find positions at each altitude
self.latitude, self.longitude, __ = self.apex.apex2geo(
np.tile(alat, len(altitude)), np.tile(alon, len(altitude)),
self.altitude)
# map field to each altitude
f = np.array([
self.apex.map_V_to_height(alat, alon, coords[2], a, field.T).T
for a in altitude
])
self.field = f.reshape(-1, f.shape[-1])
def convert_to_ECEF(self):
self.X, self.Y, self.Z = pm.geodetic2ecef(self.latitude,
self.longitude,
self.altitude * 1000.)
self.Vx, self.Vy, self.Vz = pm.enu2uvw(self.field[:, 0], self.field[:,
1],
self.field[:, 2], self.latitude,
self.longitude)
def create_interpolators(self):
self.interpVx = interpolate.LinearNDInterpolator(
np.array([self.X, self.Y, self.Z]).T, self.Vx)
self.interpVy = interpolate.LinearNDInterpolator(
np.array([self.X, self.Y, self.Z]).T, self.Vy)
self.interpVz = interpolate.LinearNDInterpolator(
np.array([self.X, self.Y, self.Z]).T, self.Vz)
def plot_ionosphere(self):
fig = plt.figure(figsize=(10, 10))
ax = fig.add_subplot(111, projection='3d')
for x, y, z, vx, vy, vz in zip(self.X, self.Y, self.Z, self.Vx,
self.Vy, self.Vz):
ax.quiver(x,
y,
z,
vx,
vy,
vz,
length=0.4 * np.sqrt(vx**2 + vy**2 + vz**2),
color='green')
plt.show()
def test_map_V_to_height():
apex_out = Apex(date=2000, refh=300)
out_60_15_100_500 = [0.819719, 2.845114, 0.695437]
out_60_15_100_500_234 = [1.830277, 4.14345, 0.947624]
out_60_15_100_1000 = [0.924577, 3.149964, 0.851343]
out_60_15_200_500 = [0.803882, 2.793206, 0.682839]
out_60_30_100_500 = [0.761412, 2.878837, 0.736549]
out_70_15_100_500 = [0.846819, 2.592572, 0.347919]
# scalar
assert_allclose(apex_out.map_V_to_height(60, 15, 100, 500, [1, 2, 3]),
out_60_15_100_500,
rtol=1e-5)
assert_allclose(apex_out.map_V_to_height(60, 15, 100, 500, [2, 3, 4]),
out_60_15_100_500_234,
rtol=1e-5)
assert_allclose(apex_out.map_V_to_height(60, 15, 100, 1000, [1, 2, 3]),
out_60_15_100_1000,
rtol=1e-5)
assert_allclose(apex_out.map_V_to_height(60, 15, 200, 500, [1, 2, 3]),
out_60_15_200_500,
rtol=1e-5)
assert_allclose(apex_out.map_V_to_height(60, 30, 100, 500, [1, 2, 3]),
out_60_30_100_500,
rtol=1e-5)
assert_allclose(apex_out.map_V_to_height(70, 15, 100, 500, [1, 2, 3]),
out_70_15_100_500,
rtol=1e-5)
# vectorize lat
assert_allclose(apex_out.map_V_to_height([60, 70], 15, 100, 500,
np.array([[1, 2, 3]] * 2).T),
np.array([out_60_15_100_500, out_70_15_100_500]).T,
rtol=1e-5)
# vectorize lon
assert_allclose(apex_out.map_V_to_height(60, [15, 30], 100, 500,
np.array([[1, 2, 3]] * 2).T),
np.array([out_60_15_100_500, out_60_30_100_500]).T,
rtol=1e-5)
# vectorize height
assert_allclose(apex_out.map_V_to_height(60, 15, [100, 200], 500,
np.array([[1, 2, 3]] * 2).T),
np.array([out_60_15_100_500, out_60_15_200_500]).T,
rtol=1e-5)
# vectorize newheight
assert_allclose(apex_out.map_V_to_height(60, 15, 100, [500, 1000],
np.array([[1, 2, 3]] * 2).T),
np.array([out_60_15_100_500, out_60_15_100_1000]).T,
rtol=1e-5)
# vectorize E
assert_allclose(apex_out.map_V_to_height(
60, 15, 100, 500,
np.array([[1, 2, 3], [2, 3, 4]]).T),
np.array([out_60_15_100_500, out_60_15_100_500_234]).T,
rtol=1e-5)
