def _create_resample_kdtree(source_lons,
source_lats,
valid_input_index,
nprocs=1):
"""Set up kd tree on input."""
source_lons_valid = source_lons[valid_input_index]
source_lats_valid = source_lats[valid_input_index]
if nprocs > 1:
cartesian = _spatial_mp.Cartesian_MP(nprocs)
else:
cartesian = _spatial_mp.Cartesian()
input_coords = cartesian.transform_lonlats(source_lons_valid,
source_lats_valid)
if input_coords.size == 0:
raise EmptyResult('No valid data points in input data')
# Build kd-tree on input
if nprocs > 1:
resample_kdtree = _spatial_mp.cKDTree_MP(input_coords, nprocs=nprocs)
else:
resample_kdtree = KDTree(input_coords)
return resample_kdtree
def _query_resample_kdtree(resample_kdtree,
source_geo_def,
target_geo_def,
radius_of_influence,
data_slice,
neighbours=8,
epsilon=0,
reduce_data=True,
nprocs=1):
"""Query kd-tree on slice of target coordinates."""
# Check validity of input
if not isinstance(target_geo_def, geometry.BaseDefinition):
raise TypeError('target_geo_def must be of geometry type')
elif not isinstance(radius_of_influence, (long, int, float)):
raise TypeError('radius_of_influence must be number')
elif not isinstance(neighbours, int):
raise TypeError('neighbours must be integer')
elif not isinstance(epsilon, (long, int, float)):
raise TypeError('epsilon must be number')
# Get sliced target coordinates
target_lons, target_lats = target_geo_def.get_lonlats(nprocs=nprocs,
data_slice=data_slice, dtype=source_geo_def.dtype)
# Find indiced of reduced target coordinates
valid_output_index = _get_valid_output_index(source_geo_def,
target_geo_def,
target_lons.ravel(),
target_lats.ravel(),
reduce_data,
radius_of_influence)
# Get cartesian target coordinates and select reduced set
if nprocs > 1:
cartesian = _spatial_mp.Cartesian_MP(nprocs)
else:
cartesian = _spatial_mp.Cartesian()
target_lons_valid = target_lons.ravel()[valid_output_index]
target_lats_valid = target_lats.ravel()[valid_output_index]
output_coords = cartesian.transform_lonlats(target_lons_valid,
target_lats_valid)
# pykdtree requires query points have same data type as kdtree.
try:
dt = resample_kdtree.data.dtype
except AttributeError:
# use a sensible default
dt = np.dtype('d')
output_coords = np.asarray(output_coords, dtype=dt)
# Query kd-tree
distance_array, index_array = resample_kdtree.query(output_coords,
k=neighbours,
eps=epsilon,
distance_upper_bound=radius_of_influence)
return valid_output_index, index_array, distance_array
def get_cartesian_coords(self, nprocs=None, data_slice=None, cache=False):
"""Retrieve cartesian coordinates of geometry definition
Parameters
----------
nprocs : int, optional
Number of processor cores to be used.
Defaults to the nprocs set when instantiating object
data_slice : slice object, optional
Calculate only cartesian coordnates for the defined slice
cache : bool, optional
Store result the result. Requires data_slice to be None
Returns
-------
cartesian_coords : numpy array
"""
if self.cartesian_coords is None:
# Coordinates are not cached
if nprocs is None:
nprocs = self.nprocs
if data_slice is None:
# Use full slice
data_slice = slice(None)
lons, lats = self.get_lonlats(nprocs=nprocs, data_slice=data_slice)
if nprocs > 1:
cartesian = _spatial_mp.Cartesian_MP(nprocs)
else:
cartesian = _spatial_mp.Cartesian()
cartesian_coords = cartesian.transform_lonlats(np.ravel(lons),
np.ravel(lats))
if isinstance(lons, np.ndarray) and lons.ndim > 1:
# Reshape to correct shape
cartesian_coords = cartesian_coords.reshape(lons.shape[0],
lons.shape[1], 3)
if cache and data_slice is None:
self.cartesian_coords = cartesian_coords
else:
# Coordinates are cached
if data_slice is None:
cartesian_coords = self.cartesian_coords
else:
cartesian_coords = self.cartesian_coords[data_slice]
return cartesian_coords
def _create_resample_kdtree(source_lons,
source_lats,
valid_input_index,
nprocs=1):
"""Set up kd tree on input"""
"""
if not isinstance(source_geo_def, geometry.BaseDefinition):
raise TypeError('source_geo_def must be of geometry type')
#Get reduced cartesian coordinates and flatten them
source_cartesian_coords = source_geo_def.get_cartesian_coords(nprocs=nprocs)
input_coords = geometry._flatten_cartesian_coords(source_cartesian_coords)
input_coords = input_coords[valid_input_index]
"""
source_lons_valid = source_lons[valid_input_index]
source_lats_valid = source_lats[valid_input_index]
if nprocs > 1:
cartesian = _spatial_mp.Cartesian_MP(nprocs)
else:
cartesian = _spatial_mp.Cartesian()
input_coords = cartesian.transform_lonlats(source_lons_valid,
source_lats_valid)
if input_coords.size == 0:
raise EmptyResult('No valid data points in input data')
# Build kd-tree on input
if kd_tree_name == 'pykdtree':
resample_kdtree = KDTree(input_coords)
elif nprocs > 1:
resample_kdtree = _spatial_mp.cKDTree_MP(input_coords, nprocs=nprocs)
else:
resample_kdtree = sp.cKDTree(input_coords)
return resample_kdtree