a ~b\@sddlmZmZmZddlmZddlmZddlZ ddl Z ddl m Z ddl mZmZmZmZmZmZmZmZddlmZmZdd lmZdd lmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0d dl1m2Z2d dl3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:d dl;mZ>ddl?m@Z@ddl;mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZIGddde4e:ZJGdddeJZKGdddeKe5ZLGdddeKe6ZMGdddeJe9ZNdS))absolute_importdivisionunicode_literals)groupby) defaultdictN)gridplot)ColumnDataSourceColumnRowDivTitleLegendAxisColorBar)PanelTabs)NoOpSelectionDisplay) OrderedDictStore AdjointLayoutNdLayoutLayoutEmpty GridSpaceHoloMapElement) SkipRendering) basestringcftime_to_timestamp cftime_typesget_method_owneris_param_methodunique_iterator wrap_tuplewrap_tuple_streams_STANDARD_CALENDARS)Link)DimensionedPlotGenericCompositePlotGenericLayoutPlotGenericElementPlotGenericOverlayPlotGenericAdjointLayoutPlot CallbackPlot)attach_streams displayablecollate) LinkCallback) TOOL_TYPESfilter_toolboxes make_axisupdate_shared_sources empty_plot decode_bytestheme_attr_jsoncds_column_replace get_defaultcseZdZdZejdddZejdddZejdddZ ej d gd d d Z ej dd ddZ ej dd ddZgdZdZdZeZeddZddZddZddZddZddZd d!Zed"d#Zed$d%Zd&d'Zd(d)Zd5fd+d, Z d6d/d0Z!d1d2Z"d3d4Z#Z$S)7 BokehPlotzy Plotting baseclass for the Bokeh backends, implementing the basic plotting interface for Bokeh based plots. T Whether Elements drawing the data from the same object should share their Bokeh data source allowing for linked brushing and other linked behaviors.defaultdocz{label} {group} {dimensions}z The formatting string for the title of this plot, allows defining a label group separator and dimension labels.NzAlias for title.above)rCZbelowleftrightNz^ The toolbar location, must be one of 'above', 'below', 'left', 'right', None.rAZobjectsrBrNz The width of the component (in pixels). This can be either fixed or preferred width, depending on width sizing policy.rAZboundsrBz The height of the component (in pixels). This can be either fixed or preferred height, depending on height sizing policy.)ZpanZbox_zoomZ box_selectZ lasso_selectZ poly_selectZypanZxpanz{title}bokehcCs|jr|jjdSdS)Nid)rootrefselfrOz.BokehPlot._postprocess_data..zConverting cftime.datetime from a non-standard calendar (%s) to a standard calendar for plotting. This may lead to subtle errors in formatting dates, for accurate tick formatting switch to the matplotlib backend.ms) itemsr:lenr[r anyparamwarningrgr)rNrbZnew_datakvaluesrOrOrPrczs  zBokehPlot._postprocess_datacs|js dS||}tdd|D}|jr|jdj|jjur|jr|s|jdjrfdd| D}| |j dSt ||r||_n |j |t|dr|jdur||dS)z> Update datasource with data for a new frame. Ncss|]}t|dkVqdSrG)rorhrOrOrPrkrlz/BokehPlot._update_datasource..rcs"i|]\}}||j dqSrf)Z _chunk_lengthrirsrjstreamrOrP rlz0BokehPlot._update_datasource..rY)ZdocumentrcallrtZ streamingrb current_frameZ _stream_dataZ _triggeringrnrwlengthr<r]rdrYra)rNsourcerbemptyrOrvrP_update_datasources&    zBokehPlot._update_datasourcec s|ddtg}fdd|jD}|D]b}|jD]V}|jD]J}|D]$\}}||jvrJ|rJ|d|j|<qJ|jr>|jd|j |_q>q4q*dS)z Iterates over all subplots and updates existing CustomJS callbacks with models that were replaced when compositing subplots into a CompositePlot and sets the plot id to match the root level bokeh model. cSs|SrfrOxrOrOrPrlz-BokehPlot._update_callbacks..c s$i|]}|tdt|iqS)type)listZselectr5)ritplotrOrPrxsz/BokehPlot._update_callbacks..rZPLACEHOLDER_PLOT_IDN) traverser, _merged_toolsrXrnargs top_levelcodereplacerJ) rNrsubplotsZ merged_toolssubplotr_cZtoolZobjsrOrrP_update_callbackss   zBokehPlot._update_callbackscCs |jdS)z The plotting state that gets updated via the update method and used by the renderer to generate output. rhandlesrMrOrOrPstateszBokehPlot.statecCsgS)g Should return a list of plot objects that have changed and should be updated. rOrMrOrOrPcurrent_handlesszBokehPlot.current_handlesc Cs|jj}ttd|ttd|ttd|ttd|ttd|ttd|ttd|d}t|}|D]&\}}t |trdd|vrd|d||<qd|S)Ntext_font_sizeZtitle_text_font_sizeZlabel_text_font_sizeZaxis_label_text_font_sizeZmajor_label_text_font_size)titleZ legend_titleZlegendlabelZticksZcticksZclabelvalue) rendererthemer=r r rrdictrnr[)rNrdefaultsZ processedrsrjrOrOrP_get_fontsize_defaultss        z BokehPlot._get_fontsize_defaultscs|ddtgD]}t|tttfs,qt|j}g|_d|_|j rR|j t|trz|j D]}||j7}| qbt |D]}fdd|jD|_qqdS)z Cleans up references to the plot after the plot has been deleted. Traverses through all plots cleaning up Callbacks and Stream subscribers. cSs|SrfrOrrOrOrPrrlz#BokehPlot.cleanup..Ncs,g|]$\}}t|r t|vr||fqSrO)r"r!)ripZ subscriberplotsrOrP s z%BokehPlot.cleanup..)rr>r[r*r,r-rr\Z _documentrclearrXcleanupsetZ _subscribers)rNrr\callbackrwrOrrPrs"        zBokehPlot.cleanupfontsizecs&tt||||}dd|DS)z[ Converts integer fontsizes to a string specifying fontsize in pt. cSs(i|] \}}|t|tr|nd|qS)z%spt)r[rrurOrOrPrxsz'BokehPlot._fontsize..)superr> _fontsizern)rNkeyrcommonsize __class__rOrPrszBokehPlot._fontsize15ptc Csd}|jr||nd}|s |St|jjd}|dd}|dd}|dd} |dd |} | |kr|d |} d | vrtt| dd d d } |j j ||| | |d} d|j vr|j d}nt |ddid}| |_ |S)Nr Z text_colorZ text_fontZArialZtext_font_styleZboldrrrZemg?)colorfont fontstylerrz white-spaceZnowrap)widthrT) show_title _format_titler;rrgetrstrfloat_title_templateformatrr text) rNrZdefault_fontsizerZ title_divrZ title_jsonrrrrZ title_tagsrOrOrP_get_title_div s2      zBokehPlot._get_title_divcCsdd}dd}|||g}tt|ddddd}g}i}g}|D]\}} t| } t| dkrHi} | D]\}} | | jdjqnt| } | D]\}} | j d} | j D] }| q| d urqn&d | vr| j| d n | j| d t | d r | jj| d | | jd<| jd<|| q|| dd| jD|t| <qH|D]d} | jrr| jrr|jd| jp~| j}|D]}|| | jq| j D]}|j|jdqqR||jd<||jd<d S)zd Syncs data sources between Elements, which draw data from the same object. cSst|jj|fSrf)rJrzrbrrOrOrPr.rlz(BokehPlot.sync_sources..cSs,|jo*|jduo*t|jjtj o*d|jvS)Nr|)shared_datasourcerzr[rbnpZndarrayrrrOrOrPr/scSs|dS)NrrOrrOrOrPr3rl)rr3r|Zglyph_rendererN data_source)r)r|viewr^cSsg|]}|qSrOrOrirrOrOrPrMrlz*BokehPlot.sync_sources..zXSupply either hooks or finalize_hooks not both; using hooks and ignoring finalize_hooks.)Zplot_idshared_sources source_cols)rrsortedrror]rrbrrrXresetZ propertiesrdrappendrJhooksZfinalize_hooksrqrrrzZ initialize)rNZ get_sourcesZ filter_fnZ data_sourcesZgrouped_sourcesrrr_groupZ source_datarZ new_sourcerrrhookrOrOrP sync_sources)sR             zBokehPlot.sync_sourcescCsbt|}g}|D]J\}}}tjdt|}|dus|jrF|durFq|||j|||q|S)NrI)r4Z find_linksr(Z _callbacksrZ_requires_targetrrK)rNlinksrXlinkZsrc_plotZtgt_plotr_rOrOrP init_links[s zBokehPlot.init_links)rT)rr)%__name__ __module__ __qualname____doc__rqBooleanrStringrZ title_formatObjectSelectortoolbarIntegerrheightrrbackendrZselection_displaypropertyrJrUrarercr~rrrrrrrrr __classcell__rOrOrrPr>)sB      2r>c@sTeZdZdZejdgdddZejddidd d Zd d Z d dZ e ddZ dS) CompositePlotz CompositePlot is an abstract baseclass for plot types that draw render multiple axes. It implements methods to add an overall title to such a plot. N)fixed stretch_widthstretch_height stretch_bothZ scale_widthZ scale_heightZ scale_bothNaS How the component should size itself. * "fixed" : Component is not responsive. It will retain its original width and height regardless of any subsequent browser window resize events. * "stretch_width" Component will responsively resize to stretch to the available width, without maintaining any aspect ratio. The height of the component depends on the type of the component and may be fixed or fit to component's contents. * "stretch_height" Component will responsively resize to stretch to the available height, without maintaining any aspect ratio. The width of the component depends on the type of the component and may be fixed or fit to component's contents. * "stretch_both" Component is completely responsive, independently in width and height, and will occupy all the available horizontal and vertical space, even if this changes the aspect ratio of the component. * "scale_width" Component will responsively resize to stretch to the available width, while maintaining the original or provided aspect ratio. * "scale_height" Component will responsively resize to stretch to the available height, while maintaining the original or provided aspect ratio. * "scale_both" Component will responsively resize to both the available width and height, while maintaining the original or provided aspect ratio. rFrrT Specifies various fontsizes of the displayed text. Finer control is available by supplying a dictionary where any unmentioned keys reverts to the default sizes, e.g: {'title': '15pt'}rAZ allow_NonerBcs0fddjD}|D]}|jdqdS)zo Should perform any linking required to update titles when dimensioned streams change. cs(g|] }tfdd|jDr|qS)c3s|]}|jvVqdSrf dimensionsrirsrMrOrPrkrlzECompositePlot._link_dimensioned_streams...)rpcontents)rir`rMrOrPrrlz;CompositePlot._link_dimensioned_streams..r3N)r\Zadd_subscriber_stream_update)rNr\r`rOrMrP_link_dimensioned_streamssz'CompositePlot._link_dimensioned_streamsc sNdd|jDtfddt|j|jD}t||j|j}||dS)NcSsg|]}|jD]}|qqSrOr)rir`rsrOrOrPrrlz0CompositePlot._stream_update..c3s"|]\}}|vrdn|VqdSrfrO)ridrsrrOrPrkrlz/CompositePlot._stream_update..)r\tupleziprZ current_keyr%r)rNkwargsrrOrrPrs zCompositePlot._stream_updatecCsd|jvr|jdgSgS)rrrrMrOrOrPrszCompositePlot.current_handles) rrrrrqr sizing_mode ParameterrrrrrrOrOrOrPrgs ' rcseZdZdZejdddZejddiddd Zej dd dZ ej d d dZ ej d d dZ ej d d dZ ejdgdddZejdgdddZejddddZejddddZejdeefddZd'fdd Zdd Zdgfd!d"Zd#d$Zed(d%d&ZZS))GridPlotz\ Plot a group of elements in a grid layout based on a GridSpace element object. 2zn Number of pixels to adjust row and column widths and height by to compensate for shared axes.r@rZ16ptTrr= Whether to merge all the tools into a single toolbarFz If enabled the x-axes of the GridSpace will be drawn from the objects inside the Grid rather than the GridSpace dimensions.zD Adds a legend based on the entries of the middle-right plot)ZbottomtopNTFzh Whether and where to display the xaxis, supported options are 'bottom', 'top' and None.rF)rDrENTFzh Whether and where to display the yaxis, supported options are 'left', 'right' and None.r)rihz& Rotation angle of the xticks.rHz& Rotation angle of the yticks.xz Defines the width and height of each plot in the grid, either as a tuple specifying width and height or an integer for a square plot.)rAclass_rBNr3c st|tstdttjf||||d||j\__ ||\_ _ j rp fddtgd|vrjddS)Nz GridPlot only accepts GridSpace.)layout layout_numrSkeyscst|jdSNr'r0hmaprrMrOrPrrlz#GridPlot.__init__.. axis_offsetzGridPlot axis_offset option is deprecated since 1.12.0 since subplots are now sized correctly and therefore no longer require an offset.)r[r Exceptionrr__init__shapecolsrows_create_subplotsrrrrr,rqrr)rNrrSrrparamsrrMrPrs zGridPlot.__init__c stjtrj\}}njj}}t}d|jrNjddnj}tfdd|Djdjd}t jddD]<\}} |j } |j } t| ts| f} j | d} | durt| t r| jn| j} | dj}nd} t| ttfvrtd t| s(t| } i}|dur>||d <|durP||d <| d krbd |d<| d kr~| d kr~d|d<| d kr| d krd|d<| d kr| d krd|d<d|d<d|vrd|d<jr | jdkr | j dkr d|d<d|d<nd|d<js"d|d<js2d|d<tjjj | d}|durp| dur̈j d| j!q|| fj"ddjj#j$jdt%|fi|}t|t&r|j'n|j(|| <||| <q||fS)Nr3csg|]}||fqSrOcompute_rangesrirZ frame_rangesrrNrOrPrsz-GridPlot._create_subplots..FZ shared_datarJTZ full_gridrzCannot plot nested Layouts.Z frame_widthZ frame_heightrendZalignxaxisyaxisZborderr show_legendrEZlegend_positionzHBokeh plotting class for %s type not found, object will not be rendered.)rrrrrKrSuniformr))r[ plot_sizerrrdynamicrclonerJ enumeraterrbrrrrlookup_optionsoptionsrrrr1r2rr shared_xaxis shared_yaxisrregistryrrrqrrrrrKrrr*rr)rNrrSrrrrcollapsed_layouticoordrrrvtypeoptsrZplotting_classrrOr rPrs         (      zGridPlot._create_subplotsc sRjjdd}t|}fddtjD}tjjddD]f\}}|j}|j}jt |d}|dur|j ||d} | |||<| | qH| dqHt |dddj jjd} | } jd} | rt| | } | jd<| jd <|jd <| jr2jrBd_jd S) Ncs g|]}ddtjDqS)cSsg|]}dqSrfrOrrOrOrPrQrlz7GridPlot.initialize_plot...)ranger)rirrMrOrPrQrlz,GridPlot.initialize_plot..Tr rSr) merge_toolsrtoolbar_locationrrr)rrrrr"rrrrr$initialize_plotrrr$rr _make_axesrr rrrrrrdrawn) rNrSr passed_plotsrrrrrrrrOrMrPr&Ns<           zGridPlot.initialize_plotc Cs|j|\}}d\}}|jjdd}|jr|j}|j}|dd} |jdddd} t t d d |D} t d || |jj d ||| | d }|j r|jjdkr|j}|j}|dd} |jdddd} t t dd |D} t d|| |jj d||| | d }|r|rt|}||gd|g} }|jrF|| } }|jrrd|_| ddd|ddd} }t| |g}nZ|r||g}|jr|ddd}t|}n,|r||g}|jr|ddd}t|}|S)NNNTr ZxlabelrxticksF)rcSsg|]}t|dqSr)r$rrOrOrPrrlz'GridPlot._make_axes..rr)fliprotationZ label_sizeZ tick_sizer3ZylabelytickscSsg|] }|dqSr3rOrrOrOrPrrly)rrrrr!)rZget_sizerrrr xrotationrrrr#r7kdimsrZndimsr yrotationr6Zmarginrr r )rNrrrZx_axisZy_axisrr-r.ZlsizeZtsizeZxfactorsZyfactorsZr1Zr2ZmodelsrOrOrPr'vsR   zGridPlot._make_axescCsf||j||}|jjddD]*}|jt|d}|dur|||q||}|rb|jddS) Update the internal state of the Plot to represent the given key tuple (where integers represent frames). Returns this state. Tr Nr) rrrrrr$ update_framerr)rNrrSrrrrOrOrPr6s zGridPlot.update_frame)Nr3N)N)rrrrrqrrrrrr$rrrrrrr2r4Z ClassSelectorintrrrrr&r'r8r6rrOrOrrPrs0  Q(*rcseZdZejdddZejdddZejdddZejdddZdfd d Z d d Z dddZ ddZ dddZ edddZZS) LayoutPlotTz3 Whether axes should be shared across plotsr@Fr?rz< Whether to display overlaid plots in separate panesNc sRttj|fd|i||\___jrNfddt gdS)Nrcst|jdSrrrrMrOrPrrlz%LayoutPlot.__init__..) rr8r _init_layoutrrpathsrrr,)rNrrrrrMrPrs zLayoutPlot.__init__csd}jdjd}ddjr6jddnj}tfdd|D}ttrnj nd}iii}}} j D]j\} } | j r| | fn| | fd\} } t| t r| nt | g} | || | f<| | | | f<dd d d }| t| d}tj|d }| | | fd\}}ttrB|rBtt||}t| jt}|s| jdurbqn2| d dtgsjd| jqn|d7}|rdn|}j| |||d\}}| dj}t|||fi|}||| | f<|r|||<q||| fS)NrFr r3csg|]}||fqSrOrr r rOrPrsz+LayoutPlot._init_layout..r*SingleZDualTriple)r3r'r positionscSs|SrfrOrrOrOrPrrlz)LayoutPlot._init_layout..z%s is empty, skipping subplot.)numr)rrJrrrrZ grid_itemsr[rr3coordsrZ transposerroAdjointLayoutPlot layout_dictrmainrrrrqrrrrr)rNrZ layout_countrrZ layout_itemslayout_dimensionsZlayout_subplotsZlayoutsr:rrrrZ layout_lens layout_typer=Z layout_keyrr}r>rZadjoint_layoutplotopts layout_plotrOr rPr9sL"       zLayoutPlot._init_layoutrc CsNi}|jd|jd}d}|D]&} || d} | dus| ddttgsLqt| s\t| } t|d} t | t rv| j n| j } t j|jj| d} || dj}i}| dkrLtj| | } | d kr| rd | jvrd nd }| r| jnd }t|j||d dgd|jd}nD| rd | jvrdnd}| r.| jnd }t|j||dgd|jd}t|fi|}|| | tur| || <d|| <qn| dur|jd| jqt|jdkr|nd }| | f|j|j||d |j |j!||jd td|j"ifi|}||| <t | t r*t#| t$r*|| j%|| <n|| j&|| <| dkr|}q||fS)< Plot all the views contained in the AdjointLayout Object using axes appropriate to the layout configuration. All the axes are supplied by LayoutPlot - the purpose of the call is to invoke subplots with correct options and styles and hide any empty axes as necessary. Fr NcSs|SrfrOrrOrOrPrrlz-LayoutPlot._create_subplots..)ZadjoinedrrBrEZbarez right-barerTr1r3)rrrZ invert_axeslabelledr+rztop-barerr)rrrrHr/rzIBokeh plotting class for %s type not found, object will not be rendered.) rrrCrSrrKrrr shared_axes)'rrJrrrrr1r2rr[rrrrrrrrrr@rrrrr]rqrrrror?rrrKrrI issubclassr*rr)rNrr=rCrSr>rZ adjoint_cloneZ main_plotposrRZ subplot_optsrZ plot_typerEZ side_optsrrrrrrOrOrPrsz          zLayoutPlot._create_subplotsc sg}t|jD]d}g}t|jD]B}|j||fd}|durBdnt|j}||dkr\dndq |t|qg}t|jD]d}g}t|jD]B}|j||fd}|durdnt|j}||dkrdndq|t|qt |} t |fddt| D} | S)z Computes an empty grid to position the plots on by expanding any AdjointLayouts into multiple rows and columns. Nr3r'csg|]}dgqSrfrO)rirrrOrPrfrlz,LayoutPlot._compute_grid..) r"rrrrrorrmaxsum) rNZwidthsrZc_widthsrr nsubplotsZheightsZ r_heightsrgridrOrLrP _compute_gridLs(zLayoutPlot._compute_gridcs||j|jdd}|jjd|j}|dur4in|j}|}|durNgn|}d}tt }g}d} d} t |j D]fdd|j D} tdd| D} | r|d 7}t |jD]H|j fd} fd d|j D}td d|D}|r|d 7<|d}| dur*q|jr:t|nd}| j||d }t|}d d|D}|j}|rtdd|D}tdd|D}| |O} | |O} |r|rd}n|rd}n |rd}|jrx| j dj|jddd}|sd|jf}|d kr|d}nN|dkr.css |]\}}t|jdkVqdS)r'NrorrRrOrOrPrk{rlz-LayoutPlot.initialize_plot..r3cs$g|]\}}|dkr||fqSr0rOrR)rrOrPrrlcss |]\}}t|jdkVqdS)r3NrTrRrOrOrPrkrlr#cSsh|]}|jdvr|jqS))Nautorr)rirSrOrOrP s z-LayoutPlot.initialize_plot..css |]}dD]}||vVq qdS))rbothNrOrimr`rOrOrPrkrlcss |]}dD]}||vVq qdS))rrXNrOrYrOrOrPrkrlrrrrBr r')r$r%r)childrenr$r%rrcSsg|] \}}|qSrOrO)rirrrOrOrPrrlcSsg|]\}}t||dqS))childr)r)rirr]rOrOrPrrl)tabsr)r\r%r$rrrVrT)(rrrr rrrrQrr7r"rrrnrprrIrr&rorr^rjoinr:rr$rrr6rrrr rrrrrr()rNrrSr Z plot_gridr)Zr_offsetZ col_offsetsZ tab_plotsrrrowZ row_paddedrcolZ col_paddedZc_offsetZ shared_plotsrrOZmodesrZresponsive_widthZresponsive_heightrrPZ plot_columnZpanelsrFrO)rrrPr&ks                  zLayoutPlot.initialize_plotcCsb||j||}|jD].\}}|j||fd}|dur|||q||}|r^||jd<dS)r5Nr)rrr?rrr6rr)rNrrSrrrrrOrOrPr6s zLayoutPlot.update_frame)N)r)NN)N)rrrrqrrIrr$r^rr9rrQr&r8r6rrOrOrrPr8s9 J r8cs8eZdZiZfddZdgfddZdddZZS) r@c s<||_||_|j|jd|_tt|jfd|i|dS)Nr=r)rrDrAview_positionsrr@r)rNrrDrrrrOrPrszAdjointLayoutPlot.__init__NcCst|dur g}g}|jD]H}|j|d}|durB|tddq||}||j||dqd|_|spdg}|S)rGNrr#T)rbrrrr9r&r()rNrSrZadjoined_plotsrKrr)rOrOrPr& s   z!AdjointLayoutPlot.initialize_plotcCs2d}dD]$}|j|}|dur|||}q|S)N)rBrEr)rrr6)rNrrSrrKrrOrOrPr6#s  zAdjointLayoutPlot.update_frame)N)rrrrrr&r6rrOrOrrPr@s r@)OZ __future__rrr itertoolsr collectionsrZnumpyrrqZ bokeh.layoutsrZ bokeh.modelsrr r r r r rrZbokeh.models.widgetsrrZ selectionrcorerrrrrrrrrZ core.optionsrZ core.utilrrr r!r"r#r$r%r&rr(rr)r*r+r,r-r.r/utilr0r1r2rXr4r5r6r7r8r9r:r;r<r=r>rrr8r@rOrOrOrPs4   ( , , $ ,@O}L