uebdZddlZddlmZddlmZdZdZd Zd Z d Z dd Z d Z defdZ dS)zO Functions that make it easier to provide a default centering for a view state N) ViewState) is_pandas_dfc||z ||z zS)N)xx0s Blib/python3.11/site-packages/pydeck/data_utils/viewport_helpers.py _squared_diffr s FrAv c t|t|kstdtjt dt ||DS)a-Euclidean distance in n-dimensions Parameters ---------- y : tuple of float A point in n-dimensions y1 : tuple of float A point in n-dimensions Examples -------- >>> EPSILON = 0.001 >>> euclidean((3, 6, 5), (7, -5, 1)) - 12.369 < EPSILON True z,Input coordinates must be of the same lengthc4g|]\}}t||Sr)r ).0r r s r zeuclidean.. s&GGG51b-2..GGGr )len Exceptionmathsqrtsumzip)yy1s r euclideanrsY q66SWW  FGGG 9SGGC2JJGGGHH I IIr ctd|Dt|z }td|Dt|z }||fS)zGets centroid in a series of points Parameters ---------- points : list of list of float List of (x, y) coordinates Returns ------- tuple The centroid of a list of points c8g|]}t|dSrfloatrps r rz"geometric_mean..0"---qt---r c8g|]}t|dSrrr s r rz"geometric_mean..1r"r )rr)pointsavg_xavg_ys r geometric_meanr(#s\ --f--- . .V r cd|D}d|D}t|}t|}t|}t|}||f||ffS)aGet the bounding box around the data, Parameters ---------- points : list of list of float List of (x, y) coordinates Returns ------- dict Dictionary containing the top left and bottom right points of a bounding box cg|] }|d Srrr s r rzget_bbox..B   1!A$   r cg|] }|d Sr$rr s r rzget_bbox..Cr+r )maxmin)r%xsysmax_xmax_ymin_xmin_ys r get_bboxr55si    B     B GGE GGE GGE GGE ENUEN ++r cfd|D}t|d}d|Ddt|S)aGets the k furthest points from the center Parameters ---------- points : list of list of float List of (x, y) coordinates center : list of list of float Center point k : int Number of points Returns ------- list Index of the k furthest points Todo --- Currently implemently naively, needs to be more efficient c4g|]}|t|fSr)r)rptcenters r rz'k_nearest_neighbors..`s(FFF"iF334FFFr c|dS)Nrr)r s r z%k_nearest_neighbors..as 1r )keycg|] }|d Srr)rr s r rz'k_nearest_neighbors..bs % % %QAaD % % %r N)sortedint)r%r9kpts_with_distance sorted_ptss ` r k_nearest_neighborsrCKsX*GFFFvFFF)~~>>>J % %* % % %hAh //r c|dkr|St|}tj|t|z}t |||S)asComputes the bounding box of the maximum zoom for the specified list of points Parameters ---------- points : list of list of float List of (x, y) coordinates proportion : float, default 1 Value between 0 and 1 representing the minimum proportion of data to be captured Returns ------- list k nearest data points r)r(rfloorrrC)r% proportioncentroid n_to_keeps r get_n_pctrIesKQ f%%H :F 344I vx ; ;;r ct|dd|ddt|dd|ddz }t|dd|ddt|dd|ddz }t||}d}|dtjddz krd}nkt dtj|tjd z tjdtjdz z z}|dkrd}|S) aJComputes the zoom level of a lat/lng bounding box Parameters ---------- bbox : list of list of float Northwest and southeast corners of a bounding box, given as two points in a list Returns ------- int Zoom level of map in a WGS84 Mercator projection (e.g., like that of Google Maps) rrNgv@rg@)r-r.rpowr?log)bboxlat_difflng_diffmax_diff zoom_levels r bbox_to_zoom_levelrU}s471:tAwqz**SaT!WQZ-H-HHH471:tAwqz**SaT!WQZ-H-HHH8X&&HJ548Ar??*++  2 2TXc]] BtxPUY]YabcYdYdGdefgg >>J r ct|r|d}tt||}t |}t |}||d|d|}|S)a\Automatically computes a zoom level for the points passed in. Parameters ---------- points : list of list of float or pandas.DataFrame A list of points view_propotion : float, default 1 Proportion of the data that is meaningful to plot view_type : class constructor for pydeck.ViewState, default :class:`pydeck.bindings.view_state.ViewState` Class constructor for a viewport. In the current version of pydeck, users most likely do not have to modify this attribute. Returns ------- pydeck.Viewport Viewport fitted to the data F)indexrr)latitude longitudezoom)r to_recordsr5rIrUr()r%view_proportion view_typerPrZr9instances r compute_viewr_s~$F0"""// Ifo66 7 7D d # #D F # #Fy&)vaytLLLH Or r$)__doc__rbindings.view_stater type_checkingrr rr(r5rCrIrUr_rr r rcs ++++++''''''JJJ*$,,,,0004<<<<06*+ir