g|=ddlmZddlmZddlmZmZddlZddl Z dZ dZ dZ dZ dd Zdd Zddddd dddiiddddddddddddd dfd Zeeeddggdgdddgddge_y))build_dataframe)make_docstring)choropleth_mapboxscatter_mapboxNc|tjzdz }tjtj|tjzdz }||fS)zU Projects lat and lon to WGS84, used to get regular hexagons on a mapbox map )nppiarctanhsin)latlonxys a/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/plotly/figure_factory/_hexbin_mapbox.py_project_latlon_to_wgs84rsC bee cA 266#++,-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_latlonrsT c'BEE/C ryy# #beeai /3 6 >C 8Orcd}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&s`ffS255[3&'CAG,-13ubee$ruuf-11rcjdtj||z |z ztjdz S)Ngffffff?r)r r!)mapPxworldPxfractions rzoomz!_getBoundsZoomLevel..zoom+s,bffUW_x788266!9DDrrihrr)r r r#)lon_minlon_maxlat_minlat_maxmapDimscale WORLD_DIMZOOM_MAXr%r* latFractionlngDiff lngFractionlatZoomlngZooms r_getBoundsZoomLevelr8s  ;u=IH2 E'?VG_4=KG&-kGcMw#EK6(#Yx%8+FG6'?Ig$6 DG w **rc P|j}|j} |j} |j} d| |z z} || z}| | z } | |z } | | z }| dk(r |dkDr||z }n| dk(r|dk(rtdd\}}n| |z }|tjdz}tj ||z j t}| | ||zz| z dz z} ||z |z }|| z |z }tj|j t}tj|j t}tj|j t}tj|j 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}"tjj|||!||!fdtjj| ||"||"fd|,tj|||k<tj| | |k<tj|j| jg}#tj |#}$nP|d}tj"||ft$ }t'|D]}%t'|D] }&g||%|&f< tj"||ft$ } t'|D]}%t'|D] }&g| |%|&f< t't)|D]}%||%rDd||%cxkr|ksnd||%cxkr|ks)n,|||%||%fj+||%Ld||%cxkr|ks[n^d||%cxkr|ksmnp| ||%||%fj+||%t'|D]J}%t'|D]:}&||%|&f}'t)|'|k\r||'||%|&f<&tj||%|&f<<Lt'|D]J}%t'|D]:}&| |%|&f}'t)|'|k\r||'| |%|&f<&tj| |%|&f<<Ltj,|j t.j| j t.jf}#tj |#}$|#|$}(tj|dft.})tj0tj2|||)d||zdf<tj4tj2|||)d||zdf<tj0tj2|dz||)||zddf<tj4tj2||dz|)||zddf<|)dddfxx|zcc<|)dddfxx|zcc<|)dddfxx|z cc<|)dddfxx| z cc<|)|$})gd }*d tj6tj8d z z d tj:tj8d z zdtj:tj8d z zdtj6tj8d z z dtj:tj8d z zd tj:tj8d 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_hexbinr9sL ;;=D ;;=D ;;=D ;;=Dt $GGODGOD B B Qw26 "W qR1W(A.A "W bggajB 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 }88S#J'88S#J'3h39 %c 2cCi @5 H3h39 %c 2cCi @E6 I  (SWc"g.2  (SWc"g.2  -/VVHX ) *-/VVHX ) * 0(..2BCDXXe_$  I88S#Jf5s $A3Z $!#A $ $88S#Jf5s $A3Z $!#A $ $s1v >AQxA$$c!f):s):SVSV^,33E!H=A$$c!f):s):SVSV^,33E!H=  >s ,A3Z ,1~t9 )%-d^HQTN%'VVHQTN  , ,s ,A3Z ,1~t9 )%-d^HQTN%'VVHQTN  , , __U # ) ) +X__U-C-I-I-K L XXe_$ Y'hh1vu%G ii #emC0F ^ 1E2F  EMv6Fu5F" aL)A, ! ilF ~)..* 0@A *l#++D1B,CDKKT,'--.  u ,  # #E */D4;&&4;&&.27m"T']#** 0 ,1l, "" LLl$445 W?U   $ %AFKKMB - . W  !mmN7,CDN7%g.224nW6M6Q6Q6ST  !      "F   %)):$;$GT  !8 .   6   #<     "! "# $% &' () C.%)*6\"..$7H2I.J,'U U  !23  *0 !&-1 !*&: !# l''*+ ! " .3|2234 ;A ##A&++A.8?C ##A&++A.<8L ##A&++A.5JJqM&&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.rzScattermapbox marker options.)rr^r_rr) override_dict)NNNNNNNN)N)plotly.express._corerplotly.express._docrplotly.express._chart_typesrrnumpyr pandasrrrr8rrrrr__doc__rrrrs0.I+>e/R    DHN=,     "      1Wt .LM    Q %&EF# r