def _validate_(self):
try:
SpatialGridDimension._validate_(self)
except ValueError:
if self._request_dataset is None:
msg = 'With no value representations (i.e. row, column, value), a data source is required.'
raise ValueError(msg)
def test_grid_without_row_and_column(self):
row = np.arange(39,42.5,0.5)
col = np.arange(-104,-95,0.5)
x,y = np.meshgrid(col,row)
value = np.zeros([2]+list(x.shape))
value = np.ma.array(value,mask=False)
value[0,:,:] = y
value[1,:,:] = x
minx,miny,maxx,maxy = x.min(),y.min(),x.max(),y.max()
grid = SpatialGridDimension(value=value)
sub = grid.get_subset_bbox(minx,miny,maxx,maxy,closed=False)
self.assertNumpyAll(sub.value,value)
def __getitem__(self, item):
ret = SpatialGridDimension.__getitem__(self, item)
if ret._src_idx is not None:
slice_row, slice_col = get_formatted_slice(item, 2)
src_idx = {}
for key, slc in izip(['row', 'col'], [slice_row, slice_col]):
src_idx[key] = np.atleast_1d(ret._src_idx[key][slc])
ret._src_idx = src_idx
return ret
def _get_value_from_source_(self):
try:
ret = SpatialGridDimension._get_value_(self)
except AttributeError:
if self.row is None or self.col is None:
ds = self._request_dataset.driver.open()
try:
slices = {k: get_formatted_slice(self._src_idx[k], 1) for k in self._src_idx.keys()}
slice_row = slices['row']
slice_col = slices['col']
variable_row = ds.variables[self.name_row]
variable_col = ds.variables[self.name_col]
# Load values ######################################################################################
value_row = np.atleast_2d(variable_row[slice_row, slice_col])
value_col = np.atleast_2d(variable_col[slice_row, slice_col])
fill = np.zeros([2] + list(value_row.shape), dtype=value_row.dtype)
try:
fill_value = value_row.fill_value
except AttributeError:
fill_value = None
fill = np.ma.array(fill, fill_value=fill_value, mask=False)
fill[0, :, :] = value_row
fill[1, :, :] = value_col
ret = fill
# Load corners #####################################################################################
try:
name_row_corners = variable_row.corners
except AttributeError:
# Likely no corners.
pass
else:
name_col_corners = variable_col.corners
value_row_corners = ds.variables[name_row_corners][slice_row, slice_col, :]
value_col_corners = ds.variables[name_col_corners][slice_row, slice_col, :]
# A reshape may be required if this is a singleton slice operation.
def _reshape_corners_(arr):
if arr.ndim < 3:
assert arr.shape == (1, 4)
arr = arr.reshape(1, 1, 4)
return arr
value_row_corners = _reshape_corners_(value_row_corners)
value_col_corners = _reshape_corners_(value_col_corners)
fill = np.zeros([2] + list(value_row_corners.shape), dtype=value_row_corners.dtype)
try:
fill_value = value_row_corners.fill_value
except AttributeError:
fill_value = None
fill = np.ma.array(fill, fill_value=fill_value, mask=False)
fill[0, :, :, :] = value_row_corners
fill[1, :, :, :] = value_col_corners
self.corners = fill
finally:
self._request_dataset.driver.close(ds)
else:
raise
return ret
def _get_uid_(self):
if self._src_idx is not None:
shp = (self._src_idx['row'].shape[0], self._src_idx['col'].shape[0])
else:
shp = None
return SpatialGridDimension._get_uid_(self, shp=shp)
