U ꥡc$@sddlZddlZddlmZddlmZddlmZddlmZ ddl m Z dd l m Z dd lmZdd lmZmZd d ZdejkZGdddeZdS)N) JSONMixin)Layer) deck_to_html)settings)View) ViewState)BaseMapProvider)DARKget_from_map_identifiercCs4zddlm}|WdStk r.YdSXdS)Nr DeckGLWidgetTF)widgetr ImportErrorr r8/tmp/pip-unpacked-wheel-srr0x9ui/pydeck/bindings/deck.pyhas_jupyter_extras  rz google.colabc @seZdZdedddgededddddd dddejjdf d d Z e d d Z de dddZ ddZddZdddZddZdS)DeckNZMapViewT)type controllerrr)ZlatitudeZ longitudeZzoom100%c Csg|_t|tr|j|n |p$g|_||_||_|p:i}| |_| |_| rXt|  nd|_ ||_ t rddl m} | |_tj|j_tj|j_||j_||j_||j_| |j_ ||d}t|tr| tjjkst|||_n t|| |_| |_dS)a This is the renderer and configuration for a deck.gl visualization, similar to the `Deck `_ class from deck.gl. Pass `Deck` a Mapbox API token to display a basemap; see the notes below. Parameters ---------- layers : pydeck.Layer or list of pydeck.Layer, default None List of :class:`pydeck.bindings.layer.Layer` layers to render. views : list of pydeck.View, default ``[pydeck.View(type="MapView", controller=True)]`` List of :class:`pydeck.bindings.view.View` objects to render. api_keys : dict, default None Dictionary of geospatial API service providers, where the keys are ``mapbox``, ``google_maps``, or ``carto`` and the values are the API key. Defaults to None if not set. Any of the environment variables ``MAPBOX_API_KEY``, ``GOOGLE_MAPS_API_KEY``, and ``CARTO_API_KEY`` can be set instead of hardcoding the key here. map_provider : str, default 'carto' If multiple API keys are set (e.g., both Mapbox and Google Maps), inform pydeck which basemap provider to prefer. Values can be ``carto``, ``mapbox`` or ``google_maps`` map_style : str or dict, default 'dark' One of 'light', 'dark', 'road', 'satellite', 'dark_no_labels', and 'light_no_labels', a URI for a basemap style, which varies by provider, or a dict that follows the Mapbox style `specification `. The default is Carto's Dark Matter map. For Mapbox examples, see Mapbox's `gallery `. If not using a basemap, set ``map_provider=None``. initial_view_state : pydeck.ViewState, default ``pydeck.ViewState(latitude=0, longitude=0, zoom=1)`` Initial camera angle relative to the map, defaults to a fully zoomed out 0, 0-centered map To compute a viewport from data, see :func:`pydeck.data_utils.viewport_helpers.compute_view` height : int, default 500 Height of Jupyter notebook cell, in pixels. width : int` or string, default '100%' Width of visualization, in pixels (if a number) or as a CSS value string. tooltip : bool or dict of {str: str}, default True If ``True``/``False``, toggles a default tooltip on visualization hover. Layers must have ``pickable=True`` set in order to display a tooltip. For more advanced usage, the user can pass a dict to configure more custom tooltip features. Further documentation is `here `_. .. _Deck: https://deck.gl/docs/api-reference/core/deck .. _gallery: https://www.mapbox.com/gallery/ Nrr zQThe map_provider parameter must be 'mapbox' when map_style is provided as a dict.)layers isinstancerappendviewsinitial_view_state descriptioneffectsstrlower map_provider_tooltiprrr deck_widgetpydeck_settingscustom_libraries configurationheightwidthtooltip _set_api_keysdictr ZMAPBOXvalueAssertionError map_styler parameters)selfrrr/api_keysrr)r(r*rrr"r0rZcustom_map_style_errorrrr__init__s68        z Deck.__init__cCs|jjs dS|jjS)N)r$ selected_datar1rrrr4|szDeck.selected_data)r2cCsr|D]}|ot|qtD]R}|jd}|jd}||jpJt|}t|||trt|j||qdS)zQSets API key for base map provider for both HTML embedding and the Jupyter widget_keyZ_API_KEYN) r r-namegetosgetenvsetattrrr$)r1r2kZprovider attr_nameZprovider_env_var attr_valuerrrr+s   zDeck._set_api_keyscCs$tr|jddn||jSdS)z2Display current Deck object for a Jupyter notebookTnotebook_displayN)in_google_colabto_htmlupdater$r5rrrshowsz Deck.showcCsVtstd||j_d}g}|jD]}|jr(||d}q(|rR||j_ dS)aUpdate a deck.gl map to reflect the current configuration For example, if you've modified data passed to Layer and rendered the map using `.show()`, you can call `update` to change the data on the map. Intended for use in a Jupyter environment. zbInstall the Jupyter extra for pydeck with your package manager, e.g. `pip install pydeck[jupyter]`FTN) rrto_jsonr$Z json_inputrZuse_binary_transportextendZget_binary_dataZ data_buffer)r1Z has_binaryZbinary_data_setslayerrrrrCs  z Deck.updateFc Ks@|} t| f|j|j||||||jtjtj||d |} | S)aWrite a file and loads it to an iframe, if in a Jupyter environment; otherwise, write a file and optionally open it in a web browser Parameters ---------- filename : str, default None Name of the file. open_browser : bool, default False Whether a browser window will open or not after write. notebook_display : bool, default None Display the HTML output in an iframe if True. Set to True automatically if rendering in Jupyter. iframe_width : str or int, default '100%' Width of Jupyter notebook iframe in pixels, if rendered in a Jupyter environment. iframe_height : int, default 500 Height of Jupyter notebook iframe in pixels, if rendered in Jupyter or Colab. as_string : bool, default False Returns HTML as a string, if True and ``filename`` is None. css_background_color : str, default None Background color for visualization, specified as a string in any format accepted for CSS colors. Returns ------- str Returns absolute path of the file ) mapbox_keygoogle_maps_keyfilename open_browserr@ iframe_height iframe_widthr*r&r' as_stringoffline)rErrHrIr#r%r&r') r1rJrKr@rMrLrNrOkwargsZ deck_jsonfrrrrBs($z Deck.to_htmlcCs|jdd}t|ddS)NTr?data)rBgetattr)r1htmlrrr _repr_html_s zDeck._repr_html_)N)NFNrrFF)__name__ __module__ __qualname__rr r r ZCARTOr-r3propertyr4r,r+rDrCrBrVrrrrrs6   `   7r)r9sysZ json_toolsrrGrZio.htmlrrr%viewrZ view_stater Zbase_map_providerr Z map_stylesr r rmodulesrArrrrrs