o h@spdZddlZddlmZddlmZddZd d Zd d Zd dZ ddZ dddZ ddZ defddZ dS)zO Functions that make it easier to provide a default centering for a view state N) ViewState) is_pandas_dfcCs||||S)N)xx0rrV/var/www/vscode/kcb/lib/python3.10/site-packages/pydeck/data_utils/viewport_helpers.py _squared_diff sr cCs6t|t|ks tdttddt||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 lengthcSsg|] \}}t||qSr)r ).0rrrrr zeuclidean..)len Exceptionmathsqrtsumzip)yy1rrr euclideansrcCs<tdd|Dt|}tdd|Dt|}||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 cSg|]}t|dqSrfloatr prrr r 0z"geometric_mean..cSrrrrrrr r 1r)rr)pointsavg_xavg_yrrr geometric_mean#s r"cCsLdd|D}dd|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 cSg|]}|dqSrrrrrr r Bzget_bbox..cSr#rrrrrr r Cr$)maxmin)rxsysmax_xmax_ymin_xmin_yrrr get_bbox5s r-cs<fdd|D}t|ddd}dd|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 csg|] }|t|fqSr)r)r ptcenterrr r `r z'k_nearest_neighbors..cSs|dS)Nrr)rrrr asz%k_nearest_neighbors..)keycSr#rr)r rrrr r br$N)sortedint)rr0kpts_with_distance sorted_ptsrr/r k_nearest_neighborsKsr8cCs2|dkr|St|}t|t|}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"rfloorrr8)r proportioncentroid n_to_keeprrr get_n_pctes  r=cCst|dd|ddt|dd|dd}t|dd|ddt|dd|dd}t||}d}|dtddkrId}|Stdt|td tdtd}|dkrgd}|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&rpowr4log)bboxlat_difflng_diffmax_diff zoom_levelrrr bbox_to_zoom_level}s4 4 0rHcCsLt|r |jdd}tt||}t|}t|}||d|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_recordsr-r=rHr")rview_proportion view_typerCrLr0instancerrr compute_views rQr)__doc__rbindings.view_stater type_checkingrr rr"r-r8r=rHrQrrrr s