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)