def azibazi_numpy(a_lats, a_lons, b_lats, b_lons, implementation='c'): a_lats, a_lons, b_lats, b_lons = float_array_broadcast( a_lats, a_lons, b_lats, b_lons) assert implementation in ('c', 'python') if implementation == 'c': from pyrocko import orthodrome_ext return orthodrome_ext.azibazi_numpy(a_lats, a_lons, b_lats, b_lons) _cosdelta = cosdelta_numpy(a_lats, a_lons, b_lats, b_lons) azis = azimuth_numpy(a_lats, a_lons, b_lats, b_lons, _cosdelta) bazis = azimuth_numpy(b_lats, b_lons, a_lats, a_lons, _cosdelta) eq = num.logical_and(a_lats == b_lats, a_lons == b_lons) ii_eq = num.where(eq)[0] azis[ii_eq] = 0.0 bazis[ii_eq] = 180.0 return azis, bazis
def azibazi_numpy(a_lats, a_lons, b_lats, b_lons, implementation='c'): a_lats, a_lons, b_lats, b_lons = float_array_broadcast( a_lats, a_lons, b_lats, b_lons) assert implementation in ('c', 'python') if implementation == 'c': from pyrocko import orthodrome_ext return orthodrome_ext.azibazi_numpy(a_lats, a_lons, b_lats, b_lons) _cosdelta = cosdelta_numpy(a_lats, a_lons, b_lats, b_lons) azis = azimuth_numpy(a_lats, a_lons, b_lats, b_lons, _cosdelta) bazis = azimuth_numpy(b_lats, b_lons, a_lats, a_lons, _cosdelta) eq = num.logical_and(a_lats == b_lats, a_lons == b_lons) ii_eq = num.where(eq)[0] azis[ii_eq] = 0.0 bazis[ii_eq] = 180.0 return azis, bazis