g`dZddlZddlmZddlmZdZdZd Zd Z d Z dd Z d Z defdZ y)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 W/var/www/openai/venv/lib/python3.12/site-packages/pydeck/data_utils/viewport_helpers.py _squared_diffr s FrAv c t|t|k(s tdtjt t ||Dcgc]\}}t ||c}}Scc}}w)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 length)len Exceptionmathsqrtsumzipr )yy1r r s r euclideanrsW q6SW FGG 99SC2JGJ51b-2.JGH IIGsA(c t|Dcgc]}t|dc}t|z }t|Dcgc]}t|dc}t|z }||fScc}wcc}w)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 rr)rfloatr)pointspavg_xavg_ys r geometric_meanr#sj f-fqtf- .V .-s A'A,c|Dcgc]}|d }}|Dcgc]}|d }}t|}t|}t|}t|}||f||ffScc}wcc}w)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 rr)maxmin)rrxsysmax_xmax_ymin_xmin_ys r get_bboxr(5ss 1!A$B  1!A$B  GE GE GE GE ENUEN ++ s A Ac|Dcgc]}|t||f}}t|d}|Dcgc]}|d c}dt|Scc}wcc}w)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 c |dS)Nrr)r s r z%k_nearest_neighbors..as1r )keyrN)rsortedint)rcenterkptpts_with_distance sorted_ptsr s r k_nearest_neighborsr4Ksa*@FFv"iF34vF)~>J$ %*QAaD* %hA //G %s A  Ac|dk(r|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)rrfloorrr4)r proportioncentroid n_to_keeps r get_n_pctr:es@Q f%H :F 34I vx ;;r ct|dd|ddt|dd|ddz }t|dd|ddt|dd|ddz }t||}d}|dtjddz krd}|St dtj |tj d z tj dtj dz 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_levelrF}s471:tAwqz*SaT!WQZ-HHH471:tAwqz*SaT!WQZ-HHH8X&HJ5488Ar?*+  2TXXc] BtxxPUY]YaYabcYdGdefg >J r ct|r|jd}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_recordsr(r:rFr)rview_proportion view_typerArKr/instances r compute_viewrPs_$F"""/ Ifo6 7D d #D F #F&)vaytLH Or )r)__doc__rbindings.view_stater type_checkingrr rrr(r4r:rFrPrr r rTsE +'J*$,,04<06*+ir