a 1$e"@sndZddlZddlZddlZddlmZmZm Z ddl m Z m Z ddl mZmZddZdd Zd d d ZdS)zO geopandas.clip ============== A module to clip vector data using GeoPandas. N)Polygon MultiPolygonbox) GeoDataFrame GeoSeries) _check_crs_crs_mismatch_warncCs"tjj|o t|ttttf S)N) pandasapitypesZ is_list_like isinstancerrrr)maskrP/var/www/html/django/DPS/env/lib/python3.9/site-packages/geopandas/tools/clip.py_mask_is_list_like_rectangles rcCst|}|rt|}n|}|j|jj|dd}|jdk}|sF|St|tr| }|r||j j |j ||j ||jf<q|j j |||j ||jf<n6| }|r|j |j |||<n|j ||||<|r||j}|S)aClip geometry to the polygon/rectangle extent. Clip an input GeoDataFrame to the polygon extent of the polygon parameter. Parameters ---------- gdf : GeoDataFrame, GeoSeries Dataframe to clip. mask : (Multi)Polygon, list-like Reference polygon/rectangle for clipping. Returns ------- GeoDataFrame The returned GeoDataFrame is a clipped subset of gdf that intersects with polygon/rectangle. Z intersects) predicatePoint)rrilocZsindexquery geom_typeanyr rcopygeometryvaluesZ clip_by_rectlocZ_geometry_column_name intersectionis_empty)gdfr Zclipping_by_rectangleZintersection_polygonZgdf_subZnon_point_maskclippedrrr_clip_gdf_with_masks4    rFcCst|ttfs tdt|t|}t|ttttfsP|sPtdt||rht |dkrhtdt|ttfrt ||st ||ddt|ttfr|j }n"|r|}n|j s|jn tjfd}|j }|d|dkr|d|dkr|d |dkr|d |dks"|jd dSt|ttfr<|jj}n|}t||}|r|jd k}|jd k} |ov| } | rtjd ddnd dg} gd} ddg} t|j| |j| |j| g}t|j| |j| |j| g}||k}|d kr.tjdddnd| s:|r|jjd}| rX|jdd}|| vrv|j|j| }n|| vr|j|j| }|S)a Clip points, lines, or polygon geometries to the mask extent. Both layers must be in the same Coordinate Reference System (CRS). The ``gdf`` will be clipped to the full extent of the clip object. If there are multiple polygons in mask, data from ``gdf`` will be clipped to the total boundary of all polygons in mask. If the ``mask`` is list-like with four elements ``(minx, miny, maxx, maxy)``, a faster rectangle clipping algorithm will be used. Note that this can lead to slightly different results in edge cases, e.g. if a line would be reduced to a point, this point might not be returned. The geometry is clipped in a fast but possibly dirty way. The output is not guaranteed to be valid. No exceptions will be raised for topological errors. Parameters ---------- gdf : GeoDataFrame or GeoSeries Vector layer (point, line, polygon) to be clipped to mask. mask : GeoDataFrame, GeoSeries, (Multi)Polygon, list-like Polygon vector layer used to clip ``gdf``. The mask's geometry is dissolved into one geometric feature and intersected with ``gdf``. If the mask is list-like with four elements ``(minx, miny, maxx, maxy)``, ``clip`` will use a faster rectangle clipping (:meth:`~GeoSeries.clip_by_rect`), possibly leading to slightly different results. keep_geom_type : boolean, default False If True, return only geometries of original type in case of intersection resulting in multiple geometry types or GeometryCollections. If False, return all resulting geometries (potentially mixed-types). Returns ------- GeoDataFrame or GeoSeries Vector data (points, lines, polygons) from ``gdf`` clipped to polygon boundary from mask. See also -------- GeoDataFrame.clip : equivalent GeoDataFrame method GeoSeries.clip : equivalent GeoSeries method Examples -------- Clip points (grocery stores) with polygons (the Near West Side community): >>> import geodatasets >>> chicago = geopandas.read_file( ... geodatasets.get_path("geoda.chicago_health") ... ) >>> near_west_side = chicago[chicago["community"] == "NEAR WEST SIDE"] >>> groceries = geopandas.read_file( ... geodatasets.get_path("geoda.groceries") ... ).to_crs(chicago.crs) >>> groceries.shape (148, 8) >>> nws_groceries = geopandas.clip(groceries, near_west_side) >>> nws_groceries.shape (7, 8) z1'gdf' should be GeoDataFrame or GeoSeries, got {}zJ'mask' should be GeoDataFrame, GeoSeries,(Multi)Polygon or list-like, got zIIf 'mask' is list-like, it must have four values (minx, miny, maxx, maxy)) stacklevelrNZGeometryCollectionzKkeep_geom_type can not be called on a GeoDataFrame with GeometryCollection.rr)Z LineStringZMultiLineStringZ LinearRingrZ MultiPointz>keep_geom_type can not be called on a mixed type GeoDataFrame.F)Z index_parts)r rr TypeErrorformattyperrrlenrrZ total_boundsrZboundsnpnanrrZ unary_unionrrrwarningswarnsumisinZexploder)rr Zkeep_geom_typeZmask_is_list_likeZbox_maskZbox_gdfZ combined_maskrZgeomcoll_concatZ geomcoll_origZnew_collectionZpolyslinesZpointsZorig_types_totalZclip_types_totalZ more_types orig_typerrrclipTs>              r1)F)__doc__r+numpyr)Zpandas.api.typesr Zshapely.geometryrrrZ geopandasrrZgeopandas.arrayrrrrr1rrrrs<