def test_ENU_tofrom_ECEF():
center_lat = -30.7215261207 * np.pi / 180.0
center_lon = 21.4283038269 * np.pi / 180.0
center_alt = 1051.7
lats = np.array([
-30.72218216, -30.72138101, -30.7212785, -30.7210011, -30.72159853,
-30.72206199, -30.72174614, -30.72188775, -30.72183915, -30.72100138
]) * np.pi / 180.0
lons = np.array([
21.42728211, 21.42811727, 21.42814544, 21.42795736, 21.42686739,
21.42918772, 21.42785662, 21.4286408, 21.42750933, 21.42896567
]) * np.pi / 180.0
alts = np.array([
1052.25, 1051.35, 1051.2, 1051., 1051.45, 1052.04, 1051.68, 1051.87,
1051.77, 1051.06
])
# used pymap3d, which implements matlab code, as a reference.
x = [
5109327.46674067, 5109339.76407785, 5109344.06370947, 5109365.11297147,
5109372.115673, 5109266.94314734, 5109329.89620962, 5109295.13656657,
5109337.21810468, 5109329.85680612
]
y = [
2005130.57953031, 2005221.35184577, 2005225.93775268, 2005214.8436201,
2005105.42364036, 2005302.93158317, 2005190.65566222, 2005257.71335575,
2005157.78980089, 2005304.7729239
]
z = [
-3239991.24516348, -3239914.4185286, -3239904.57048431,
-3239878.02656316, -3239935.20415493, -3239979.68381865,
-3239949.39266985, -3239962.98805772, -3239958.30386264,
-3239878.08403833
]
east = [
-97.87631659, -17.87126443, -15.17316938, -33.19049252, -137.60520964,
84.67346748, -42.84049408, 32.28083937, -76.1094745, 63.40285935
]
north = [
-72.7437482, 16.09066646, 27.45724573, 58.21544651, -8.02964511,
-59.41961437, -24.39698388, -40.09891961, -34.70965816, 58.18410876
]
up = [
0.54883333, -0.35004539, -0.50007736, -0.70035299, -0.25148791,
0.33916067, -0.02019057, 0.16979185, 0.06945155, -0.64058124
]
xyz = pyuvdata.XYZ_from_LatLonAlt(lats, lons, alts)
nt.assert_true(np.allclose(np.stack((x, y, z)), xyz, atol=1e-3))
enu = pyuvdata.ENU_from_ECEF(xyz, center_lat, center_lon, center_alt)
nt.assert_true(np.allclose(np.stack((east, north, up)), enu, atol=1e-3))
# check that a round trip gives the original value.
xyz_from_enu = pyuvdata.ECEF_from_ENU(enu, center_lat, center_lon,
center_alt)
nt.assert_true(np.allclose(xyz, xyz_from_enu, atol=1e-3))
# check passing a single value
enu_single = pyuvdata.ENU_from_ECEF(xyz[:, 0], center_lat, center_lon,
center_alt)
nt.assert_true(
np.allclose(np.stack((east[0], north[0], up[0])), enu[:, 0],
atol=1e-3))
xyz_from_enu = pyuvdata.ECEF_from_ENU(enu_single, center_lat, center_lon,
center_alt)
nt.assert_true(np.allclose(xyz[:, 0], xyz_from_enu, atol=1e-3))
# error checking
nt.assert_raises(ValueError, pyuvdata.ENU_from_ECEF, xyz[0:1, :],
center_lat, center_lon, center_alt)
nt.assert_raises(ValueError, pyuvdata.ECEF_from_ENU, enu[0:1, :],
center_lat, center_lon, center_alt)
nt.assert_raises(ValueError, pyuvdata.ENU_from_ECEF, xyz / 2., center_lat,
center_lon, center_alt)
for uvfile in args:
uvofile = uvfile + 'P'
print(uvfile, '->', uvofile)
if os.path.exists(uvofile):
print(uvofile, 'exists, skipping.')
continue
uvi = aipy.miriad.UV(uvfile)
uvo = aipy.miriad.UV(uvofile, status='new')
#aa = aipy.phs.ArrayLocation(('-30:43:17.5','21:25:41.9'))
#get the positions in the rotated ECEF coordinates
aa = aipy.cal.get_aa(opts.cal, np.array([0.15]))
ENU_antpos = aa.antpos_ideal.T
ECEF_antpos = pyuvdata.ECEF_from_ENU(ENU_antpos, lat, lon, alt)
rotECEF_antpos = pyuvdata.rotECEF_from_ECEF(ECEF_antpos.T,
lon) * 10**9 / const.c
nints = 0
curtime = None
#format it so unused ants are at 0, 0, 0, as pyuvdata expects
for i in range(len(ENU_antpos[0])):
if (ENU_antpos.T[i] == [-1., -1., -1.]).all():
rotECEF_antpos[i] = [0., 0., 0.]
#make sure that there are the appropriate number of antennae in the position file
nants = uvi['nants']
cal_nants = rotECEF_antpos.shape[0]
if nants > cal_nants:
for i in range(nants - cal_nants):