a ~b(|@sddlmZmZddlZddlmZddlZddlm Z ddl m Z m Z m Z mZddlmZmZdd lmZmZdd lmZGd d d eZd dZd#ddZd$ddZddZefddZddZgdfddZd%ddZdd Z d!d"Z!e"edS)&)absolute_importdivisionN) defaultdict)dimension_name)isscalarunique_iteratorpd unique_array) DataError Interface)MultiInterface ensure_ring)PandasInterfacec@seZdZeZdZdZdZeddZ eddZ edd Z ed d Z ed d Z eddZeddZedFddZeddZeddZeddZedGddZeddZedd Zed!d"ZedHd$d%Zed&d'Zed(d)Zed*d+Zegfd,d-ZedId.d/Zed0d1Zegd#fd2d3Zed4d5Z ed6d7Z!ed8d9Z"ed:d;Z#edd?Z%ed@dAZ&edBdCZ'edDdEZ(dS)KSpatialPandasInterface spatialpandasTcCs dtjvS)Nr)sysmodules)clsrr@lib/python3.9/site-packages/holoviews/core/data/spatialpandas.pyloadedszSpatialPandasInterface.loadedcCsP|s dSt||}dtjvrLdtjvrLddl}|t||j|jfO}|S)NF geopandasZgeoviewsr)r isinstance data_typesrrr GeoDataFrame GeoSeries)robjZis_sdfgpdrrrappliesszSpatialPandasInterface.appliescCsddlm}m}||fS)Nrrr)rrr)rrrrrrr(sz!SpatialPandasInterface.data_typescCsddlm}|S)Nrr)rr)rrrrr series_type-s z"SpatialPandasInterface.series_typecCsddlm}|S)Nrr)rr)rrrrr frame_type2s z!SpatialPandasInterface.frame_typecsPd}||vr&t|r&|SfddjD}|sHtd|dS)Ngeometrycsg|]}t|r|qSrr.0cdataZstypesrr =z5SpatialPandasInterface.geo_column..zXNo geometry column found in spatialpandas.GeoDataFrame, use the PandasInterface instead.r)r#rcolumns ValueError)rr,colcolsrr+r geo_column7sz!SpatialPandasInterface.geo_columncslddlm}|dur|j}|dur(|j}t||r>|}dtjvrxddl }t||j rd|}t||jrx||}t|t rdtjvrt |}t|t rt ||||}n0t||std|jnd|vr||t|tjr|jjn|jjg}|dgkrdg}||D]Dt|jvr2qtfdd |Dr|}q\q|||d ifS) Nrr$rZshapelyz)%s only support spatialpandas DataFrames.r&indexc3s"|]}|durdn|kVqdS)Nr4r)r)namekdrr hsz.SpatialPandasInterface.init..kdimsvdims)rrr:r;rr#to_framerrrrlist from_shapely from_multir%r0__name__r3r Z DataFramer4namesr5rr/anyZ reset_index)reltyper,r:r;rrZ index_namesrr6rinitCsF              zSpatialPandasInterface.initcsr|rdnd}|tdkr>tdtjtf|fdd|D}|rntdt||dS)NkeyallrzExpected %s instance to declare two key dimensions corresponding to the geometry coordinates but %d dimensions were found which did not refer to any columns.cs&g|]}|vr|jjvr|jqSrr5r,r)ddataset geom_dimsrrr-ysz3SpatialPandasInterface.validate..z`Supplied data does not contain specified dimensions, the following dimensions were not found: %s)rLlenr typer@ dimensionsrepr)rrKr;Z dim_typesZ not_foundrrJrvalidateos   zSpatialPandasInterface.validatecCsD|j|dd}|||vr6||j}|j|jjS|j|jjS)NT)strict) get_dimensionrLr3r,dtypeZsubtyper5)rrK dimensiondimr1rrrrTs  zSpatialPandasInterface.dtypec Csddlm}m}m}m}||j}|j|}t|j||fsBdS|D]L}t||rht |jdkrhdSt||rF|jD]} t | dkrxdSqxqFdS)Nr)MultiPolygonDtype PolygonDtypePolygon MultiPolygonFr T) spatialpandas.geometryrWrXrYrZr3r,rrTrM) rrKrWrXrYrZr1seriesgeomprrr has_holess      z SpatialPandasInterface.has_holescCs6g}t|js|S||j}|j|}dd|DS)NcSsg|] }t|qSr) geom_to_holes)r)r]rrrr-r.z0SpatialPandasInterface.holes..)rMr,r3)rrKholesr1r\rrrras    zSpatialPandasInterface.holesNcKs||\}}||jd||jd|j}|s8|S|durL|||}|||}||}|rt|dkrt|jdkr||jdjjdS|S)Nr r) rLpopr5r, select_maskindexedrMr;iloc)rrKZselection_mask selectionxdimydimZdfrdrrrselects  zSpatialPandasInterface.selectcCs|j||SN)base_interfacerc)rrKrfrrrrcsz"SpatialPandasInterface.select_maskcsfddjjDS)Ncsg|]}|jjvr|qSrrGrHrKrrr-s z4SpatialPandasInterface.geom_dims..r9rrKrrlrrLsz SpatialPandasInterface.geom_dimscCs||}|||jSrj)rSrTrN)rrKrVrrrdimension_types z%SpatialPandasInterface.dimension_typeFcCs\||}|||vrdS|r:tdd|j|jDS||}t|j|jdkS)z? Tests if dimension is scalar in each subpath. Fcss*|]"}t|p ttt|dkVqdSr N)rrMr=r )r)vrrrr8sz2SpatialPandasInterface.isscalar..r )rSrLrFr,r5rMunique)rrKrVZper_geomrrrrs   zSpatialPandasInterface.isscalarcCsz||}||}||vrh||j}||}|j|j}|dkrV|d|dfS|d|dfSn|j||SdS)Nrrr )rSrLr3r,r4Z total_boundsrkrange)rrKrVrLr1idxZboundsrrrrss     zSpatialPandasInterface.rangec sD||tfdd|Dr*td||jj||||fi|S)Nc3s|]}|vVqdSrjrrHZgeo_dimsrrr8r.z1SpatialPandasInterface.groupby..zESpatialPandasInterface does not allow grouping by geometry dimension.)rLrBr rkgroupby)rrKrOZcontainer_typeZ group_typekwargsrrurrvs  zSpatialPandasInterface.groupbycKstdSrjNotImplementedError)rr/rOfunctionrwrrr aggregatesz SpatialPandasInterface.aggregatecCstdSrjrx)rr/ZsamplesrrrsampleszSpatialPandasInterface.samplecCs|jSrjr,)rrKr:r;rrrreindexszSpatialPandasInterface.reindexcCs||t|fSrj)lengthrMrOrmrrrshapeszSpatialPandasInterface.shapecs:||tfdd|Dr*td||j|||S)Nc3s|]}|vVqdSrjrrHrurrr8r.z.SpatialPandasInterface.sort..zDSpatialPandasInterface does not allow sorting by geometry dimension.)rLrBr rksort)rrKZbyreverserrurrs  zSpatialPandasInterface.sortc Csddlm}m}||j}|j|}||}t|j|sP|dkrP|j |Sd}t |D].\}} t| |rx|d7}q\|t | j d7}q\|S)Nr)MultiPointDtypePointrr r) r[rrr3r, geom_typerrTrkr enumeraterM buffer_values) rrKrrZcol_namecolumnrrir]rrrrs      zSpatialPandasInterface.lengthcCstt|jdSNr )boolrMr,headrmrrrnonzeroszSpatialPandasInterface.nonzerocCs|j||Srj)rkredimrrKrOrrrrszSpatialPandasInterface.redimcCsP|j}||j}|t|j|kr2|d8}|j|vrL|||j||Sr)r,copyr3r=r/r4r5insert)rrKrUZdim_posvaluesZvdimr,geom_colrrr add_dimensions   z$SpatialPandasInterface.add_dimensioncs6ddlm}ddlm}|\}||jd}tjjt t rhdd Dn:t rt |}jgnfdd|dDtfd d Dstd |ttfd dDt jj|s2|rjj|ddfSt |r"|g}jj|fSj}dgg}} t|D]\} } tt| jd t |rވ|krkr nn@|d } | j| | d } |t| | | | qn,t |t r|jdur |jkr qRn|jdur*|jkr*q|jdur:dnt|jdd }|jdur^dnt|jd }|jdurj !d| j||}t|r | | |t| |nJfdd|D}|r | | t j"|td}|t| | j|7qRjj| #}|||<|S)Nrr")rFcSsg|] }|jqSrr5rHrrrr-*r.z/SpatialPandasInterface.iloc..csg|]}|jqSr)rSr5rHrlrrr-/r.r c3s|]}|vVqdSrjrrH)r2rrr80r.z.SpatialPandasInterface.iloc..z[Cannot index a dimension which is part of the geometry column of a spatialpandas DataFrame.cs(g|] }|vrn|qSr)r4r()r/rrLrrr-3srzL.iloc step slicing currently not supported forthe multi-tabular data format.csNg|]F}|d|ddfD]$}|kr@kr$nq$|q$qS)rr r)r)rrp)countrrrr-Ys(rT)$rrr[rrLr3r,r=r/rslicerOnprrSr5rFr rrTrerintrMrappendrNstartstopmaxminstepZparamZwarningarrayr)rrKr4rrZrowsscalarZgeomsZ new_geomsZindexesrr]rtr,rrZslicedZsub_rowsZidxsnewr)r2r/rrKrrLrrre sx                  $$       zSpatialPandasInterface.iloccCs||}|j|}|j}||v} ||j} ||dk} | rN|rN|| S| st| rb||jjSt||||| | St |st gS||} | |} || }|rt |dr|}t|j| || S)Nrcompute)rS interfacerLr,r3rr5rget_value_arrayrMrrr4hasattrrgeom_array_to_array)rrKrUexpandedZflatr keep_indexrLr,Zisgeomr is_pointsrr4Z geom_seriesrrrres(       zSpatialPandasInterface.valuesc s$ddlm}mddlm}g}t|js.gS||\fdd|j|j D} |jj d| |j} | |} |durt | | d} jjf| i} | fdd | D|j| d gd }||r||nd}|jD]0\}|dur.|fd d |jjD}|||q| }| pFt|j| }t ||d} j| dddfj| ddd fi} | fdd | D|dvr|dur||| |j<|| d<|| q| |_|dkr|jfi|}n(|dkr|jfi|}n td|||q|S)Nrr!rrPolygonscsg|]}|fvr|qSrrr)rV)rgrhrrr-s z0SpatialPandasInterface.split..)rcsi|]}|j|jqSrrrrowrr r.z0SpatialPandasInterface.split.. dictionary)datatypecs.i|]&}||dkr |gn|gqS)r&rr()rrrrrsr csi|]}|j|jqSrrrrrrrr.)rr/rrZ dataframez%s datatype not support)rrrelementrrMr,rLr:r;rer3r geom_to_arrayr5updateZcloner_raZiterrowsr/r get_geom_type _hole_keyrdframer0)rrKrendrrwrrZobjsZ value_dimsr1rarrrIZdsrargdfr]gtrr)rrrgrhrsplit}sT       (       zSpatialPandasInterface.splitcCs|r|j|S|jSdSrj)r,rrrrrrs zSpatialPandasInterface.dframecCs|jSrjr}rmrrr as_dframesz SpatialPandasInterface.as_dframe)T)N)F)NN)TTTF))r@ __module__ __qualname__rrkrmultitypes classmethodrr rr#r%r3rDrQrTr_rarircrLrnrrsrvr{r|r~rrrrrrrerrrrrrrrrs       +                   D  1 rc Cs|ddlm}m}m}m}m}m}m}||} t| j ||fr@dSt| j ||frTdSt| j ||frhdSt| j |rxdSdS)zReturn the HoloViews geometry type string for the geometry column. Args: gdf: The GeoDataFrame to get the geometry from col: The geometry column Returns: A string representing the type of geometry r) PointDtyper LineDtypeMultiLineDtyperXrW RingDtyperLinerYRingN) r[rrrrrXrWrrrT) rr1rrrrrXrWrrrrrrs $ rFcCsddlm}m}m}m}m}m} t||rd|durFt |j |j ggSt |rT|j n|j gg} n~t||||frt||r|j dn|j } t | dd} t|||frt| } |dur| n| dd|fg} nt|| r$|dur t |jddg} nt |j|ddg} ng} |j D]} t||rF| dn| } t | dd} t||rtt| } | |dur| n| dd|f|dkr.| |durtjtjggntjgq.|dkr| dd} |r| St| dkr| dSt| SdS)zConverts spatialpandas geometry to an array. Args: geom: spatialpandas geometry index: The column index to return multi: Whether to concatenate multiple arrays or not Returns: Array or list of arrays. r)rrYrrrZ MultiPointNrrr )r[rrYrrrZrrrrxyr,as_pyreshaperrrnanrM concatenate)r]r4rrrrYrrrZrarraysexteriorrgrrrrs>  "    $ *  rc Csddlm}m}t||r*|r$|jS|jSg}t||p>|dk}|D]@}t||||d} |rz|| |s|t j gqD|| qD|r|s|dd}|rt |St gSt j t|td} || dd<| SdS)zConverts spatialpandas extension arrays to a flattened array. Args: geom: spatialpandas geometry index: The column index to return Returns: Flattened array r) PointArrayMultiPointArrayr)rrNrr)r[rrrrrrextendrrrrremptyrMobject) geom_arrayr4expandrrrrZ multi_pointr]rrrrrs&      rc Cs8ddlm}m}m}m}t||rdd}|jd}|d|dksP|d|dkrTd}t|d|St|||frd}|jD]\}d}t||r|}n,|d}|d|dks|d|dkrd}|t|dd|7}q||r|dSdSd}|j}t||r(|d|dks$|d|dkr(d}t|dSdS)Nr)rYrrZ MultiLiner rr) r[rYrrZrrr,rMr) r]rYrrZroffsetrrrrrr geom_length)s.      0rcCs||j}|r|Sd}gg} } t||D]\} } |r.rcSs g|]\}}|r|dn|qSrr)r)srrrrr-tr.)r5rrerrrrMr rZfullZNaNrrrzip)r,rUrrrrrrZ all_scalarrZscalarsrr]rvalrrrrrrEs>           rc Csddlm}m}t||rfg}t|jD]6\}}|dkr:q(tt|  dd}| |q(|gSt||rg}|jD]R}g}t|D]6\}}|dkrqtt|  dd}| |q| |qz|Sdt |j vrggt|SggSdS)zExtracts holes from spatialpandas Polygon geometries. Args: geom: spatialpandas geometry Returns: List of arrays representing holes r)rYrZrrZMultiN)r[rYrZrrr,rrrrrrrNr@rM) r]rYrZrarZholepolyZ poly_holesrrrrr`xs.       r`Zpointc* sddlm}m}ddlm}m}m} m} m} m } m } m }m }m }m}ddlmtfdd|Dpl|dkr| }| |}}n>|d kr|}| |}}n&|d kr| }||}}n|}| |}}d }gg}}|D]}t|}||vs||vrtd ||||}}t|t|}}|rJ|rJt|gt|g}}n&|r^t||}n|rpt||}t||g}|| | fvrt|}tt|d d d d fdjddd}t|rt ||dn|g}|j!d }|d ur*t|t|krt"dt#ndd|D}|$||$|||urt|dksjtdd|Drp|}n|d ur|}qt|r|}q|d ur|}qt%t&} t'|||D]p\}}!}"g}#t(|!D]\}$}%|$t|!dkr|%d d}%t|%rdnd kr||urqʈr0|#$g|#d}&n|#}&|&$|%d d d d fr|"d ur|&dd|"|$D7}&q|)D]\}'}(| |'$|(q|| ur|#d*}#nb||urt+dd|#D}#nB||urfdd|#D}#n$rdd|#dDn |#d*}#| d$|#q| rl| d})|| urHt+|)})||)}r^|,}||| d<n||g| d<|| dg|dS)aConverts list of dictionary format geometries to spatialpandas line geometries. Args: data: List of dictionaries representing individual geometries xdim: Name of x-coordinates column ydim: Name of y-coordinates column columns: List of columns to add geom: The type of geometry Returns: A spatialpandas.GeoDataFrame version of the data r)rr) rrrYr LineArray PolygonArrayrMultiLineArrayMultiPolygonArrayr RingArrayrrrc3s|]}j|vVqdSrj)rrHrrrr8r.z#to_spatialpandas..rYrrNz"Could not find geometry dimensionsrfloatr )Zaxisz_Polygons with holes containing multi-geometries must declare a list of holes for each geometry.cSsg|]}dd|DqS)cSsg|]}tt|qSr)rrasarrayr)hrrrr-r./to_spatialpandas...r)r)Zhsrrrr-r.z$to_spatialpandas..css|]}t|dkVqdSro)rM)r)rrrrr8r.rcSsg|]}t|qSr)rrrrrrr-r.cSsg|] }|qSrflattenr)sprrrr-r.cs&g|]}rdd|Dn|qS)cSsg|] }|qSrr)r)Zssprrrr-r.rrr)rrrr-r.cSsg|]}t|qSr)rrrrrrrr-r.r&)r/)-rrrr[rrrYrrrrrrrrrrrBdictr0rbrrrZ full_likeZ column_stackrwhereZisnanZastypesumrMrrr rrrr=rritemsrrZoriented)*r,rgrhr/r]rrrrrYrrrrrrrrrZ single_arrayZ multi_arrayZ array_typeZ hole_arraysZ geom_arraysZxsZysZxscalarZyscalarrZsplitsZ split_geomsZ split_holesZ convertedrrapartsrrZsubpartsr*rpZ geometriesr)rrrto_spatialpandass 4         2     "   "     $  rc Csddlm}dd|ddD\}}t|trFdd|D}|S||||}|j|urb|jSi} |D]H} | ||fvrd } n|j|| } | | | } | r| d n| | | j <qn| S) aIConverts data from any list format to a dictionary based format. Args: eltype: Element type to convert data: The original data kdims: The declared key dimensions vdims: The declared value dimensions Returns: A list of dictionaries containing geometry coordinates and values. r )Datasetcss|] }|jVqdSrjrr)r7rrrr8r.zto_geom_dict..NrcSs(i|] \}}|t|r|nt|qSr)rrr)r)krprrrr!r.z to_geom_dict..Fr) rrrrrr,rOrZdimension_valuesr5) rCr,r:r;rrxnameynameZnew_elnew_dictrIrZvalsrrr to_geom_dicts      rc s*ddlm}dd|ddD\ggg}}}|D]l}|t|t||||t} d| vr~| d|vr~|| d|| t|dts:t|d|d<q:tt |d krt d |r|d|urt |}nJfd d ||D} t|d kr |d} n t |} t|| | }|S) a0Converts list formats into spatialpandas.GeoDataFrame. Args: eltype: Element type to convert data: The original data kdims: The declared key dimensions vdims: The declared value dimensions Returns: A GeoDataFrame containing in the list based format. rr$css|] }|jVqdSrjrrrrrr8?r.zfrom_multi..Nrrrr zMixed types not supportedcsg|]}|fvr|jqSrrrHrrrrr-Or.zfrom_multi..)rrrrNrrrrrMsetr r concatrr) rCr,r:r;rnew_datarZ geom_typesrIrr/r]rrrr?1s*     r?csddlm}m}ddlm|s"ntfdd|DrF||}ntfdd|Drdd|dD}|D]@}|D]2\}}||t |st |r|nt |q~qr||d |d <||}|S) zConverts shapely based data formats to spatialpandas.GeoDataFrame. Args: data: A list of shapely objects or dictionaries containing shapely objects Returns: A GeoDataFrame containing the shapely geometry data. rr! BaseGeometryc3s|]}t|VqdSrjr'rHrrrr8hr.zfrom_shapely..c3s.|]&}t|to$d|vo$t|dVqdS)r&N)rrrHrrrr8jscSsi|] }|gqSrr)r)r1rrrrlr.z from_shapely..r&) rrrZshapely.geometry.baser rFr<rrrrrr)r,rrrrIr1rrrrr>Xs   .r>)NFN)FN)N)#Z __future__rrr collectionsrZnumpyrrUrutilrrr r rr r Z multipathrrZpandasrrrrrrrr`rrr?r>registerrrrrs0   0 1 ! 3#v '