~iMCddlmZddlmZddlmZmZddlmcm Z ddl Z ddl Z dZdZdZdZ dd Zdd Zddddd dddiiddddddddddddd dfd Zeeeddggdgdddgddge_dZdS))build_dataframe)make_docstring)choropleth_map scatter_mapNc|tjzdz }tjtj|tjzdz }||fS)zU Projects lat and lon to WGS84, used to get regular hexagons on a mapbox map )nppiarctanhsin)latlonxys U/home/jrussi/.local/lib/python3.11/site-packages/plotly/figure_factory/_hexbin_map.py_project_latlon_to_wgs84r sA be cA 26#++,,--A a4Kc|dztjz }dtjtj|ztjdz z dztjz }||fS)zU Projects WGS84 to lat and lon, used to get regular hexagons on a mapbox map r)r r arctanexp)rrrr s r_project_wgs84_to_latlonrsO c'BE/C ry## #beai /3 6 >C 8Orc$d}d|zd|zd}d}d}d} ||||z tjz } ||z } | dkr| dzn| dz } | |d |d | } | |d |d | }t| ||S) z Get the mapbox zoom level given bounds and a figure dimension Source: https://stackoverflow.com/questions/6048975/google-maps-v3-how-to-calculate-the-zoom-level-for-a-given-bounds rheightwidthctj|tjzdz }tjd|zd|z z dz }t t |tjtj dz S)Nrr)r r r logmaxmin)r r radX2s rlatRadz#_getBoundsZoomLevel..latRad'sbfS25[3&''CAG,--13ube$$ruf--11rcfdtj||z |z ztjdz S)Ngffffff?r)r r!)mapPxworldPxfractions rzoomz!_getBoundsZoomLevel..zoom,s-bfUW_x788826!99DDrrihrr)r r r#)lon_minlon_maxlat_minlat_maxmapDimscale WORLD_DIMZOOM_MAXr%r* latFractionlngDiff lngFractionlatZoomlngZooms r_getBoundsZoomLevelr8s  ;u==IH222 EEE6'??VVG__4=KG&-kkGcMMw#EKd6(#Yx%8+FFGd6'?Ig$6 DDG w * **rc ||}|} |} |} d| |z z} || z}| | z } | |z } | | z }| dkr |dkr||z }n%| dkr|dkrtdd\}}n| |z }|tjdz}tj||z t}| | ||zz| z dz z} ||z |z }|| z |z }tj|t}tj|t}tj |t}tj |t}|dz}|dz}|}|}||z||zz}||z dzd||z dzzz}||z dz dzd||z dz dzzz}||k}|@tj ||f}tj ||f} d|k||kzd|kz||kz|z}!d|k||kzd|kz||kz|z}"tj |||!||!fdtj | ||"||"fd|&tj |||k<tj | | |k<tj|| g}#tj|#}$n|d}tj||ft$ }t'|D]}%t'|D] }&g||%|&f< tj||ft$ } t'|D]}%t'|D] }&g| |%|&f< t't)|D]}%||%r\d||%cxkr|krHn d||%cxkr|kr2n6|||%||%f||%fd||%cxkr|krHn|d||%cxkr|kr2n| ||%||%f||%t'|D]S}%t'|D]A}&||%|&f}'t)|'|kr||'||%|&f<0tj ||%|&f<BTt'|D]S}%t'|D]A}&| |%|&f}'t)|'|kr||'| |%|&f<0tj | |%|&f<BTtj|t.| t.f}#tj|#}$|#|$}(tj |dft.})tjtj|||)d||zdf<tjtj|||)d||zdf<tjtj|dz||)||zddf<tjtj||dz|)||zddf<|)dddfxx|zcc<|)dddfxx|zcc<|)dddfxx|z cc<|)dddfxx| z cc<|)|$})gd }*d tjtjd z z d tjtjd z zdtjtjd z zdtjtjd z z dtjtjd z zd tjtjd z zg}+t)|)},tj|*g|,z|ztj|)dddfz}-tj|+g|,z|ztjdz tj|)dddfz}.|-|.|)|(fS) aQ Computes the aggregation at hexagonal bin level. Also defines the coordinates of the hexagons for plotting. The binning is inspired by matplotlib's implementation. Parameters ---------- x : np.ndarray Array of x values (shape N) y : np.ndarray Array of y values (shape N) x_range : np.ndarray Min and max x (shape 2) y_range : np.ndarray Min and max y (shape 2) color : np.ndarray Metric to aggregate at hexagon level (shape N) nx : int Number of hexagons horizontally agg_func : function Numpy compatible aggregator, this function must take a one-dimensional np.ndarray as input and output a scalar min_count : int Minimum number of points in the hexagon for the hexagon to be displayed Returns ------- np.ndarray X coordinates of each hexagon (shape M x 6) np.ndarray Y coordinates of each hexagon (shape M x 6) np.ndarray Centers of the hexagons (shape M x 2) np.ndarray Aggregated value in each hexagon (shape M) g& .>rr rg@?N)dtype)rr;r;rr=r=) r#r"rr sqrtceilastypeintroundfloorzerosaddatnan concatenateravelisnanemptyobjectrangelenappendhstackfloatrepeatarangetilecosr tanarrayvstack)/rrx_rangey_rangecolornxagg_func min_countxminxmaxyminymaxpaddingDxDydx_dynyix1iy1ix2iy2nx1ny1nx2ny2nd1d2bdistlattice1lattice2c1c2accum good_idxsijvalsagreggated_valuecentershxhymhxshyss/ r_compute_hexbinr:sL ;;==D ;;==D ;;==D ;;==Dt $GGODGOD B B Qww266 "W qR1WW(A..AA "W bgajjB b    % %B TBG^d "a ''D TRA TRA (1++  S ! !C (1++  S ! !C (1++  S ! !C (1++  S ! !C q&C q&C C C c C#IA c'a#SQ. .B c'C-A q3w}&: : :B GE }8S#J''8S#J''3h39 %c 2cCi @5 H3h39 %c 2cCi @E6 I  (SWc"g.222  (SWc"g.222  -/VHX ) *-/VHX ) * 0 0(..2B2BCDDXe__$  I8S#Jf555s $ $A3ZZ $ $!#A $8S#Jf555s $ $A3ZZ $ $!#A $s1vv > >AQx >A$$$$$$$$$c!f):):):):s):):):):):SVSV^,33E!H===A$$$$$$$$$c!f):):):):s):):):):):SVSV^,33E!H===s , ,A3ZZ , ,1~t99 ))%-Xd^^HQTNN%'VHQTNN  , s , ,A3ZZ , ,1~t99 ))%-Xd^^HQTNN%'VHQTNN  , __U # # ) ) + +X__U-C-C-I-I-K-K L  Xe__$ Y'h1vu%%G i #<emC000FF ^ 1E222FF  EMv666FFu555F" aL)A, ! ilF  ~)..** 0@0@AAA *   Xd,-dX C C XZZ   \*+<<>>  R/D d5k**3355 d5k**3355=A']T"--W ..77999PT- 0 0 0 ,1l, "" L!!fXL(9(99!-- "2     Y2 CCCPPfWoo""28,,QN 7 # ' ' ) ) 7 # ' ' ) )    !++---  + g   "FFF   %)):$;$GT  !8 7 .- v  65  K #<";  T v  )! "e# $% &e' (v) C.")*6 \"''-.$#( ,'ikkU U  !23    *0 !&-1 !*&: !# l'*+++ ! " .3|23344  ;A #A&+A.8?C #A&+A.<8L #A&+A.5JqM&q) '*/2& 1 "" JrrBz/Number of hexagons (horizontally) to be created)functionz8Numpy array aggregator, it must take as input a 1D arrayzand output a scalar value.)rBz=Minimum number of points in a hexagon for it to be displayed.z3If None and color is not set, display all hexagons.zDIf None and color is set, only display hexagons that contain points.boolzCWhether to show the original data on top of the hexbin aggregation.rzScattermap marker options.)rr^r_rr) override_dictctjddtd|vr|d|d<t |i|S)Nzcreate_hexbin_mapbox() is deprecated and will be removed in the next major version. Please use create_hexbin_map() instead. Learn more at: https://plotly.com/python/mapbox-to-maplibre/r) stacklevelcategory mapbox_styler)warningswarnDeprecationWarningpopr)rkwargss rcreate_hexbin_mapboxrs_ M I# $jj88{ d -f - --r) NNNNNNNNN)N)plotly.express._corerplotly.express._docrplotly.express._chart_typesrrnarwhals.stable.v1stablev1rnumpyr rrrr8rrrrr__doc__rrrrrs000000......CCCCCCCC+++>e/e/e/R    LHLHLHLH^====,     "      1vvvvr+N$LM        Q %&BC#0 . . . . .r