U SZf|=@sddlmZddlmZddlmZmZddlZddl Z ddZ ddZ d d Z d d Zd#d dZd$ddZddddddddiiddddddddddddddfddZeeeddgdddgddddgddgdd gd!d"e_dS)%)build_dataframe)make_docstring)choropleth_mapboxscatter_mapboxNcCs0|tjd}tt|tjd}||fS)zU Projects lat and lon to WGS84, used to get regular hexagons on a mapbox map )nppiZarctanhsin)latlonxyrH/tmp/pip-unpacked-wheel-5ksk5baj/plotly/figure_factory/_hexbin_mapbox.py_project_latlon_to_wgs84srcCs>|dtj}dtt|tjddtj}||fS)zU Projects WGS84 to lat and lon, used to get regular hexagons on a mapbox map r)rrZarctanexp)r r r r rrr_project_wgs84_to_latlons(rcCsd}d|d|d}d}dd}dd} ||||tj} ||} | d krX| d n| d } | |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 rheightwidthcSsHt|tjd}td|d|d}tt|tjtj dS)Nrr)rr rlogmaxmin)r r ZradX2rrrlatRad&sz#_getBoundsZoomLevel..latRadcSs dt|||tdS)Ngffffff?r)rr)ZmapPxZworldPxfractionrrrzoom+sz!_getBoundsZoomLevel..zoomrihrr)rrr)Zlon_minZlon_maxZlat_minZlat_maxmapDimZscaleZ WORLD_DIMZZOOM_MAXrrZ latFractionZlngDiffZ lngFractionZlatZoomZlngZoomrrr_getBoundsZoomLevelsr!c/ Cs|}|} |} |} d| |} || 8}| | 7} | |} | | }| dkrf|dkrf||}n(| dkr|dkrtdd\}}n| |}|td}t||t}| | ||| d8} |||}|| |}t|t}t|t}t |t}t |t}|d}|d}|}|}||||}||dd||d}||ddd||dd}||k}|dkrt ||f}t ||f} d|k||k@d|k@||k@|@}!d|k||k@d|k@||k@|@}"tj |||!||!fdtj | ||"||"fd|dk r^tj |||k<tj | | |k<t|| g}#t|#}$nH|dkrd}tj||ftd }t|D]"}%t|D]}&g||%|&f<qqtj||ftd } t|D]"}%t|D]}&g| |%|&f<qqtt|D]}%||%rd||%krD|krnn>d||%krd|krnn|||%||%f||%n^d||%kr|krnn>d||%kr|krnn| ||%||%f||%qt|D]P}%t|D]@}&||%|&f}'t|'|kr.||'||%|&f<ntj ||%|&f<qqt|D]P}%t|D]@}&| |%|&f}'t|'|kr||'| |%|&f<ntj | |%|&f<qXqLt|t| tf}#t|#}$|#|$}(t |dft})tt|||)d||df<tt|||)d||df<tt|d||)||ddf<tt||d|)||ddf<|)dddf|9<|)dddf|9<|)dddf|7<|)dddf| 7<|)|$})ddddd d g}*d ttjd d ttjd dttjd dttjd dttjd d ttjd g}+t|)},t|*g|,|t|)dddf}-t|+g|,|tdt|)dddf}.|-|.|)|(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& .>rrrg@g?N)Zdtypeg) rrrrsqrtceilastypeintroundfloorzerosaddatnanZ concatenateZravelisnanemptyobjectrangelenappendZhstackfloatrepeatarangeZtilecosrtanarrayZvstack)/r r x_rangey_rangecolornxagg_func min_countZxminZxmaxZyminZymaxpaddingZDxZDyZdx_ZdynyZix1Ziy1Zix2Ziy2Znx1Zny1Znx2Zny2nd1Zd2bdistZlattice1Zlattice2Zc1c2accumZ good_idxsijvalsagreggated_valuecentershxZhymhxshysrrr_compute_hexbin9s&    $ $&       @ @"       ""&& *4rQc Cst||\}} |dkr,t||g}|dkrJt||g}t||\} } t|| | | ||||\} } }}t| | \}}|t}t |dddfdt |dddf}||||fS)a Computes the lat-lon aggregation at hexagonal bin level. Latitude and longitude need to be projected to WGS84 before aggregating in order to display regular hexagons on the map. Parameters ---------- lat : np.ndarray Array of latitudes (shape N) lon : np.ndarray Array of longitudes (shape N) lat_range : np.ndarray Min and max latitudes (shape 2) lon_range : np.ndarray Min and max longitudes (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 Lat coordinates of each hexagon (shape M x 6) np.ndarray Lon coordinates of each hexagon (shape M x 6) pd.Series Unique id for each hexagon, to be used in the geojson data (shape M) np.ndarray Aggregated value in each hexagon (shape M) Nr,r) rrr9rrrQrr&strpdZSeries)r r lat_range lon_ranger<r=r>r?r r r:r;rOrPrLrK hexagons_lats hexagons_lons hexagons_idsrrr_compute_wgs84_hexbins(0  0rZc Cs~g}|dkrtt|}t|||D]J\}}}t||gj}||d|td|td|gddq&td|dS) zc Creates a geojson of hexagonal features based on the outputs of _compute_wgs84_hexbin NrZFeatureZPolygon)typeZ coordinates)r[idZgeometryZFeatureCollection)r[features) rr6r2zipr9Ttolistr3dict)rWrXidsr]r r idxZpointsrrr_hexagons_to_geojson(s rdFc,Csttdd}|dkrtj}|d|dddgj}|d|dddgj}t|d|dj|d|dj||d|||d\}}}}t|||}|dkr0|dkr|dkrtd d d } nL|dkr|dk rtd |d } n.|dk r|dkrt||d } n t||d } t |d |d |d |d | }|dkrNt||d }|ddk rr|d |dj }!nd |dj i}!g}"|! D]\}#}$|dj|$}%t|%|dj|%|dj|||dr|%|djnd|||d\}&}&}}'|"tjtj||'fddgdqtj|"d |!djddd}(t|(d|(d<| dkrj|(d|(dg} t|(|dddddd|ddk rdnd||| | | | | |||||||d})|rt|ddk r|dj|ddn|d|d|d|dd}*d|*jd _d|*jd _||*jd _ |)!|*jd |ddk rt"t#|*j$D]d}+d|*j$|+jd _d|*j$|+jd _||*j$|+jd _ |)j$|+jd |*j$|+jd g|)j$|+_qL|)S)zO Returns a figure aggregating scattered points into connected hexagons N)args constructor data_framer rrr )r r rUrVr<r=r>r?irrr)r r animation_framer< locations)columns)Zaxiskeys)frameindex)rnrmTF)r<rjrm)rhgeojsonrjr<Z hover_dataricolor_discrete_sequencecolor_discrete_maplabelscolor_continuous_scale range_colorcolor_continuous_midpointopacityrcenter mapbox_styletitletemplaterr)Zby)rhr r riskip)%rlocalsrZmeanZaggvaluesrZrdrar!groupbygroupsrnitemslocr3rTZ DataFrameZc_concatrlZ rename_axisZ reset_indexZ to_numericrrrrZ sort_valuesdataZ hoverinfoZ hovertemplatemarkerZ add_tracer1r2frames),rhr r r< nx_hexagonr>rirprqrrrsrtrurvrrwrxryrzrrr?show_original_dataoriginal_data_markerrfrUrVrWrXrYcountror rZagg_data_frame_listrmrnZdfrAZaggregated_valueZagg_data_frameZfigZ original_figrHrrrcreate_hexbin_mapbox=s              rr'z/Number of hexagons (horizontally) to be createdfunctionz8Numpy array aggregator, it must take as input a 1D arrayzand output a scalar value.z=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.razScattermapbox marker options.)rr>r?rr)Z override_dict)NNNNNNNN)N)Zplotly.express._corerZplotly.express._docrZplotly.express._chart_typesrrZnumpyrZpandasrTrrr!rQrZrdrra__doc__rrrrs    * G