def arrayinpolygon(a, polygon, x=None, y=None):
'''
Set array element value as 1 if inside a polygon or set value as -1.
:param a: (*array_like*) The array.
:param polygon: (*PolygonShape*) The polygon.
:param x: (*float*) X coordinate of the point. Default is ``None``, for DimArray
:param y: (*float*) Y coordinate of the point. Default is ``None``, for DimArray
:returns: (*array_like*) Result array.
'''
if isinstance(a, DimArray):
if x is None or y is None:
x = self.dimvalue(1)
y = self.dimvalue(0)
if not x is None and not y is None:
if isinstance(polygon, tuple):
x_p = polygon[0]
y_p = polygon[1]
if isinstance(x_p, NDArray):
x_p = x_p.aslist()
if isinstance(y_p, NDArray):
y_p = y_p.aslist()
return NDArray(GeometryUtil.inPolygon(a.asarray(), x.aslist(), y.aslist(), x_p, y_p))
else:
if isinstance(polygon, MILayer):
polygon = polygon.layer
return NDArray(GeometryUtil.inPolygon(a.asarray(), x.aslist(), y.aslist(), polygon))
else:
return None
def inpolygon(x, y, polygon):
'''
Check if x/y points are inside a polygon or not.
:param x: (*array_like*) X coordinate of the points.
:param y: (*array_like*) Y coordinate of the points.
:param polygon: (*PolygonShape list*) The polygon list.
:returns: (*boolean array*) Inside or not.
'''
if isinstance(x, numbers.Number):
return GeoComputation.pointInPolygon(polygon, x, y)
if isinstance(x, (list, tuple)):
x = np.array(x)
if isinstance(y, (list, tuple)):
y = np.array(y)
if isinstance(polygon, tuple):
x_p = polygon[0]
y_p = polygon[1]
if isinstance(x_p, (list, tuple)):
x_p = np.array(x_p)
if isinstance(y_p, (list, tuple)):
y_p = np.array(y_p)
return NDArray(GeometryUtil.inPolygon(x._array, y._array, x_p._array, y_p._array))
else:
if isinstance(polygon, MILayer):
polygon = polygon.shapes()
elif isinstance(polygon, PolygonShape):
polygon = [polygon]
return NDArray(GeometryUtil.inPolygon(x._array, y._array, polygon))
def z_coord(self):
'''
Get Z coordinate array.
:return: Z coordinate array
'''
if self.isvectorlayer():
if self._coord_array is None:
self._coord_array = GeometryUtil.getCoordinates(self.layer)
return np.array(self._coord_array[2])
else:
return None
def convexhull(*args):
'''
Computes the smallest convex Polygon that contains all the points in the Geometry.
'''
if len(args) == 1:
a = args[0]
ap = asshape(a)
r = ap.convexHull()
return r
else:
x = args[0]
y = args[1]
r = GeometryUtil.convexHull(x.asarray(), y.asarray())
return r
def rmaskin(data, x, y, mask):
"""
Maskin data by polygons - the elements inside polygons will be removed
:param data: (*array_like*) Array data for maskin.
:param x: (*array_like*) X coordinate array.
:param y: (*array_like*) Y coordinate array.
:param mask: (*list*) Polygon list as mask borders.
:returns: (*list*) Masked data, x and y array list.
"""
if not isinstance(mask, (list, ArrayList)):
mask = [mask]
r = GeometryUtil.maskin_Remove(data._array, x._array, y._array, mask)
return NDArray(r[0]), NDArray(r[1]), NDArray(r[2])
def maskout(data, mask, x=None, y=None):
"""
Maskout data by polygons - NaN values of elements outside polygons.
:param data: (*array_like*) Array data for maskout.
:param mask: (*list*) Polygon list as maskout borders.
:param x: (*array_like*) X coordinate array.
:param y: (*array_like*) Y coordinate array.
:returns: (*array_like*) Maskouted data array.
"""
if mask is None:
return data
elif isinstance(mask, (NDArray, DimArray)):
r = ArrayMath.maskout(data.asarray(), mask.asarray())
if isinstance(data, DimArray):
return DimArray(r, data.dims, data.fill_value, data.proj)
else:
return NDArray(r)
if x is None or y is None:
if isinstance(data, DimArray):
x = data.dimvalue(data.ndim - 1)
y = data.dimvalue(data.ndim - 2)
else:
return None
if not isinstance(mask, (list, ArrayList)):
mask = [mask]
if data.ndim == 2 and x.ndim == 1 and y.ndim == 1:
x, y = np.meshgrid(x, y)
r = GeometryUtil.maskout(data._array, x._array, y._array, mask)
if isinstance(data, DimArray):
return DimArray(r, data.dims, data.fill_value, data.proj)
else:
return NDArray(r)
