h3BddlZddlmZddlZddlZddlmZm Z ddl m Z ddl m Z mZmZmZmZdZdZdZd Zd Zd Zdd Zd Zy)N)reduce) GeoDataFrame GeoSeries) PANDAS_GE_30)LINE_GEOM_TYPESPOINT_GEOM_TYPESPOLYGON_GEOM_TYPES _check_crs_crs_mismatch_warnc|jdk(sotr4d|jvr|jdd}|j d}|Sd|jvr|jddd|j dd|S)z|Ensure that the geometry column is called 'geometry'. If another column with that name exists, it will be dropped. geometryaxisTrinplace)r)_geometry_column_namercolumnsdroprename_geometry)dfs J/var/www/html/immo/lib/python3.12/site-packages/geopandas/tools/overlay.py_ensure_geometry_columnrs  # #z 1 RZZ'WWZaW0##J/B IRZZ' D9   z4  8 Ic|jj|jdd\}}|jdkDrq|jdkDra|jj |}|j dd|jj |}|j dd|j |}|jjt}||j|j|<tj||d}|} |j d}|j d}|j|j|j d d d } | j|j|j d d dd } t#| | |j$S|j&ddj|j&ddj|jj(d ddd } t*j,| d <t*j,| d <| | j.j|jj(j1|jj(gzS)z8Overlay Intersection operation used in overlay function. intersectsT predicatesortrrr)__idx1__idx2rrrr!)left_on right_indexr"_1_2)r$r%suffixesr crsN left_indexr%r))sindexqueryr sizetake reset_index intersection geom_typeisinr make_validlocpd DataFramemergerrrr+ilocnamenpnanrtolist) df1df2idx1idx2leftright intersectionspoly_ixpairs_intersectgeom_intersectdfinterresults r_overlay_intersectionrL#s6!!#,,,T!RJD$ yy1}Q||  & dD1 !!$' tT2))%0 ))../AB%27%;%F%F%H '",,$$'GH&oo4o(oo4o(!'' HHS..QH 7(  -- HHS..QH 7!   Gn#''JJ"1## HHRaL  cll//a  8! $  66x66x NN   1 1 2 9 9 ;s||?P?P>Q Q  rc j|jj|jdd\}}tj|d\}}tj ||dd}t |jjDcgc]}||vr|jdng}}g} t|j|D]G\} } td| gt|jj| z} | j| It| |j|j } | j"j%t&}| |j)| j*|<| | j,j/}|| j,j/}|||j0<|Scc}w) z6Overlay Difference operation used in overlay function.rTr) return_indexrNrc$|j|S)N) difference)xys rz%_overlay_difference..asar)indexr+)r.r/r r=uniquesplitranger0popziprlistr;appendrrTr+r4r5r r6r7is_emptycopyr)r@rArBrC idx1_uniqueidx1_unique_indices idx2_splitidxsidxnew_ggeom neighboursnew differencesrG geom_diffdfdiffs r_overlay_differencerjSs!!#,,,T!RJD$')yyD'I$K$$ 3AB 78J**+  !K/ qR7 D  E d3j (4&4 8I8I*8U3V*V   S  E@K##(();>@KOOG[1112779I +&&& ' , , .F+4F6 ' '( M# sF0c~t||}t||}t|}|jDcgc]}||jvr|n|d}}||_t j ||gdd}t |j}|jd|jd|j|Scc}w)z4Overlay Identity operation used in overlay function.r'TF ignore_indexrr r) rLrjrrr8concatrZremover[reindex)r@rAdfintersection dfdifferencecol new_columnsrKrs r_overlay_identityrvms*34N&sC0L*<8L  '' n,,,SE*<K'LYY 5Du UF>))*G NN: NN: >>'> **sB:ct||}t||}tt||d<tt||d<tj|d<tj|d<t |}t |}|j |ddgdd}|jj}d|_ |j|jjdf|j|jj<|jddgd d |jd d t|||j }|S)z@Overlay Symmetric Difference operation used in overlay function.r!r"outerr&)onhowr)r geometry_2 geometry_1rTrr r*)rjrWlenr=r>rr:r|r]r<r7isnullrr2rr+)r@rAdfdiff1dfdiff2dfsymr s r_overlay_symmetric_diffrs.!#s+G!#s+Gc'l+GHc'l+GHGHGH%g.G%g.G MMXx(g   E$$&HHM.3ii !<//HLL!!((*+ JJ l+!TJB 4. sww ?E Lrct||}t||}tj||gdd}t |j }|j d|jd|j|S)z1Overlay Union operation used in overlay function.TFrlr rn) rLrr8rorZrrpr[rq)r@rArJrdfunionrs r_overlay_unionrsh#C-G #C -Eii%(t%HG7??#G NN: NN: ??7? ++rcHgd}||vrtd|d|t|tst|tr tdt ||st ||d|d}d}nd }t ||gD]\}}|jjtj} |jjtj} |jjtj} t| | | gd kDstd |d zd |d k(r|j} |j} | d| dkr!| d| dkr| d | dkr | d | dks|jddj!|jddj#|j$j&d ddd}||j(j#|j$j&j+|j$j&gzSfd}|r|jjd}||}||}t-j.5t-j0dd|dk(r t3||}nG|d k(r t5||}n5|dk(r t7||}n#|dk(r t9||}n|dk(r t;||}|dvrj#ddgd dddd|r t=|}j?dd|S#1swY-xYw) anPerform spatial overlay between two GeoDataFrames. Currently only supports data GeoDataFrames with uniform geometry types, i.e. containing only (Multi)Polygons, or only (Multi)Points, or a combination of (Multi)LineString and LinearRing shapes. Implements several methods that are all effectively subsets of the union. See the User Guide page :doc:`../../user_guide/set_operations` for details. Parameters ---------- df1 : GeoDataFrame df2 : GeoDataFrame how : string Method of spatial overlay: 'intersection', 'union', 'identity', 'symmetric_difference' or 'difference'. keep_geom_type : bool If True, return only geometries of the same geometry type as df1 has, if False, return all resulting geometries. Default is None, which will set keep_geom_type to True but warn upon dropping geometries. make_valid : bool, default True If True, any invalid input geometries are corrected with a call to make_valid(), if False, a `ValueError` is raised if any input geometries are invalid. Returns ------- df : GeoDataFrame GeoDataFrame with new set of polygons and attributes resulting from the overlay Examples -------- >>> from shapely.geometry import Polygon >>> polys1 = geopandas.GeoSeries([Polygon([(0,0), (2,0), (2,2), (0,2)]), ... Polygon([(2,2), (4,2), (4,4), (2,4)])]) >>> polys2 = geopandas.GeoSeries([Polygon([(1,1), (3,1), (3,3), (1,3)]), ... Polygon([(3,3), (5,3), (5,5), (3,5)])]) >>> df1 = geopandas.GeoDataFrame({'geometry': polys1, 'df1_data':[1,2]}) >>> df2 = geopandas.GeoDataFrame({'geometry': polys2, 'df2_data':[1,2]}) >>> geopandas.overlay(df1, df2, how='union') df1_data df2_data geometry 0 1.0 1.0 POLYGON ((2 2, 2 1, 1 1, 1 2, 2 2)) 1 2.0 1.0 POLYGON ((2 2, 2 3, 3 3, 3 2, 2 2)) 2 2.0 2.0 POLYGON ((4 4, 4 3, 3 3, 3 4, 4 4)) 3 1.0 NaN POLYGON ((2 0, 0 0, 0 2, 1 2, 1 1, 2 1, 2 0)) 4 2.0 NaN MULTIPOLYGON (((3 4, 3 3, 2 3, 2 4, 3 4)), ((4... 5 NaN 1.0 MULTIPOLYGON (((2 3, 2 2, 1 2, 1 3, 2 3)), ((3... 6 NaN 2.0 POLYGON ((3 5, 5 5, 5 3, 4 3, 4 4, 3 4, 3 5)) >>> geopandas.overlay(df1, df2, how='intersection') df1_data df2_data geometry 0 1 1 POLYGON ((2 2, 2 1, 1 1, 1 2, 2 2)) 1 2 1 POLYGON ((2 2, 2 3, 3 3, 3 2, 2 2)) 2 2 2 POLYGON ((4 4, 4 3, 3 3, 3 4, 4 4)) >>> geopandas.overlay(df1, df2, how='symmetric_difference') df1_data df2_data geometry 0 1.0 NaN POLYGON ((2 0, 0 0, 0 2, 1 2, 1 1, 2 1, 2 0)) 1 2.0 NaN MULTIPOLYGON (((3 4, 3 3, 2 3, 2 4, 3 4)), ((4... 2 NaN 1.0 MULTIPOLYGON (((2 3, 2 2, 1 2, 1 3, 2 3)), ((3... 3 NaN 2.0 POLYGON ((3 5, 5 5, 5 3, 4 3, 4 4, 3 4, 3 5)) >>> geopandas.overlay(df1, df2, how='difference') geometry df1_data 0 POLYGON ((2 0, 0 0, 0 2, 1 2, 1 1, 2 1, 2 0)) 1 1 MULTIPOLYGON (((3 4, 3 3, 2 3, 2 4, 3 4)), ((4... 2 >>> geopandas.overlay(df1, df2, how='identity') df1_data df2_data geometry 0 1 1.0 POLYGON ((2 2, 2 1, 1 1, 1 2, 2 2)) 1 2 1.0 POLYGON ((2 2, 2 3, 3 3, 3 2, 2 2)) 2 2 2.0 POLYGON ((4 4, 4 3, 3 3, 3 4, 4 4)) 3 1 NaN POLYGON ((2 0, 0 0, 0 2, 1 2, 1 1, 2 1, 2 0)) 4 2 NaN MULTIPOLYGON (((3 4, 3 3, 2 3, 2 4, 3 4)), ((4... See Also -------- sjoin : spatial join GeoDataFrame.overlay : equivalent method Notes ----- Every operation in GeoPandas is planar, i.e. the potential third dimension is not taken into account. )r3unionidentitysymmetric_differencerPz `how` was 'z' but is expected to be in z4overlay currently only implemented for GeoDataFrames stacklevelNTFrrz contains mixed geometry types.r3rrr&r,c|j}|jjtj r|j j }|j}rN|j||fj|j||f<|jrt|dd}|S|jrtd|jd|S)NPolygonF)r4keep_geom_type_warningz,You have passed make_valid=False along with zo invalid input geometries. Use make_valid=True or make sure that all geometries are valid before using overlay.)r]r4r5r allr is_validrr7r6any_collection_extract ValueErrorsum)rmaskrtr6s r _make_validzoverlay.._make_valid8s WWY <<  / 0 4 4 6KK(((D**C$&FF49$5$@$@$BtSy!88:,iB  Bxxzl#66  rignorezCRS mismatch between the CRS)messagerPrrr)r3rrrr!r"rr ) r isinstancerNotImplementedErrorr r enumerater4r5r rrrr total_boundsr;r:rr r<rr?warningscatch_warningsfilterwarningsrjrLrrrvrr2)r@rArzkeep_geom_typer6 allowed_howsrir poly_check lines_check points_checkbox_gdf1box_gdf2rKrr4s ` roverlayrs1rL ,;se+F|nUVV#y!ZY%?! B   c3 32!%!&C:&S2\\&&'9:>>@ ll''8<<> ||(()9:>>@  K6 7! ;%1q5'1P&QR R S n####qkXa[(x{hqk/I1+!,8A;(1+3MXXbq\''! !!#,,"3"3!!< % (F ##CLL$5$56==?3<&B&B&D D %%%3 .7.A.H.H =(*+!" \\!_N ZZ((--j9 :F 6<<?2Ka1A5;Q ;3345     Mr)r3NT)r functoolsrnumpyr=pandasr8 geopandasrrgeopandas._compatrgeopandas.arrayrrr r r rrLrjrvrrrrrrrsL-*"- `4+06 ,}@;r