a ~bwx@sddlmZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z hdZhd Zhd Zd d gd d gd d gd d gd d gd d gd d gd d gd d gd d gd d gd d gd d gd d gd d gd d gd d gd gd gd gd gd gdgdgdgdgdgdgdgdZhdZddddddZedZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd:d1d2Z d;d3d4Z!dZ parcoordsZsankeypietableZparcats>geoternaryscenepolarmapbox>rr rxaxisyaxisrr rr r)barbox candlestickZcarpetcontour contourcarpetZheatmapZ heatmapgl histogramZ histogram2dhistogram2dcontourohlc pointcloudscatter scattercarpet scatterglviolinZconeZmesh3d scatter3dZ streamtubeZsurfaceZ choropleth scattergeobarpolar scatterpolarscatterpolarglscatterternary scattermapbox>rrr&r r!r%rrr r#rZ waterfallr"rrZsplomr$rZarearrrrrZopacityheightsymbolZmaxzoomZminzoom)ZalphaZ cell_heightZmarkerZmax_zoomZmin_zoomz\D*(\d+)cCs&t|}|rt|d}nd}|S)aj Extract the subplot number from a subplot value string. 'x3' -> 3 'polar2' -> 2 'scene' -> 1 'y' -> 1 Note: the absence of a subplot number (e.g. 'y') is treated by plotly as a subplot number of 1 Parameters ---------- subplot_val: str Subplot string value (e.g. 'scene4') Returns ------- int ) _subplot_rematchintgroup) subplot_valr+subplot_numberr0=lib/python3.9/site-packages/holoviews/plotting/plotly/util.py_get_subplot_number~s  r2cCs$|dkrd}n|dkrd}n|}|S)a] Get the subplot value prefix for a subplot type. For most subplot types this is equal to the subplot type string itself. For example, a `scatter3d.scene` value of `scene2` is used to associate the scatter3d trace with the `layout.scene2` subplot. However, the `xaxis`/`yaxis` subplot types are exceptions to this pattern. For example, a `scatter.xaxis` value of `x2` is used to associate the scatter trace with the `layout.xaxis2` subplot. Parameters ---------- subplot_type: str Subplot string value (e.g. 'scene4') Returns ------- str rxryr0) subplot_typesubplot_val_prefixr0r0r1_get_subplot_val_prefixs r7cCs|tvrd}n|}|S)a Get the name of the trace property used to associate a trace with a particular subplot type. For most subplot types this is equal to the subplot type string. For example, the `scatter3d.scene` property is used to associate a `scatter3d` trace with a particular `scene` subplot. However, for some subplot types the trace property is not named after the subplot type. For example, the `scatterpolar.subplot` property is used to associate a `scatterpolar` trace with a particular `polar` subplot. Parameters ---------- subplot_type: str Subplot string value (e.g. 'scene4') Returns ------- str Zsubplot)_subplot_prop_named_subplot)r5subplot_prop_namer0r0r1_get_subplot_prop_namesr:c Cs|di}|ddD]t}|dd}t|g}|D]R}t|}t|}|||}t|} | dkrv|t| } n|} | |vr8i|| <q8qdS)a Make sure a layout subplot property is initialized for every subplot that is referenced by a trace in the figure. For example, if a figure contains a `scatterpolar` trace with the `subplot` property set to `polar3`, this function will make sure the figure's layout has a `polar3` property, and will initialize it to an empty dict if it does not Note: This function mutates the input figure dict Parameters ---------- fig: dict A plotly figure dict layoutdataNtyperr)) setdefaultget_trace_to_subplotr:r7r2str) figr;trace trace_type subplot_typesr5r9r6r.r/Zlayout_prop_namer0r0r1_normalize_subplot_idss    rFcCsddtD}d|d<d|d<|dgD]\}|dd}t|g}|D]:}t|}t|}|||}t|} t||| ||<qJq*|d i} d D]r} | | gD]`} | d d } | d krt| }t|d||d<| dd}|d krt|}t|d||d<qq|S)aj Given an input figure, return a dict containing the max subplot number for each subplot type in the figure Parameters ---------- fig: dict A plotly figure dict Returns ------- dict A dict from subplot type strings to integers indicating the largest subplot number in the figure of that subplot type cSsi|] }|dqS)rr0).0r5r0r0r1 sz(_get_max_subplot_ids..rrrr<r=rr; annotationsshapesimagesxrefr3paperyrefr4)_subplot_typesr?r@r:r7r2max)rBZmax_subplot_idsrCrDrEr5r9r6r.r/r; layout_propobjrM xref_numberrO yref_numberr0r0r1_get_max_subplot_idss8       rVc Cs |ddD]|}|dd}t|g}|D]Z}t|}t|}|||}t|} | ||d} | dkr~|t| ||<q,|||<q,q |di} i} |D]\}||} | dkrqt| D]8}| |rt|} | | }|t|}| || |<qq| | |dd}|d d}t| D]}| dr| |}|d d }t||}|dkrvd t||d <nd |d <nP| d r*| |}|d d }t||}|dkrd t||d <nd |d <q*t| D]}|dd dkr| |}|dd}|r|dd kr,t||}n |dd krt||}nq|dkr^t|nd}|d||d<qdD]}| |gD]v}|r|dd }|dkrt|}d t|||d<|r|dd }|dkrt|}d t|||d<qqzdS)aO Apply offsets to the subplot id numbers in a figure. Note: This function mutates the input figure dict Note: This function assumes that the normalize_subplot_ids function has already been run on the figure, so that all layout subplot properties in use are explicitly present in the figure's layout. Parameters ---------- fig: dict A plotly figure dict offsets: dict A dict from subplot types to the offset to be applied for each subplot type. This dict matches the form of the dict returned by get_max_subplot_ids r<Nr=rrr)r;rranchorr4r3axismatchesrIrMrNrO) r?r@r:r7r2rAr>listkeys startswithpopupdate) rBZoffsetsrCrDrEr5r9r6r.r/Zoffset_subplot_numberr;Z new_subplotsoffsetrRZnew_subplot_numberZnew_layout_propZx_offsetZy_offsetrrWZ anchor_numberrrYZ matches_valZmatches_numbersuffixrSrMrTrOrUr0r0r1_offset_subplot_ids2s                          rcc s|dg}|di}fddfddfdd}|D] }|d d } | tvrF||qF|D]$} tD]} | | rt||| qtql|D]b} | d r|| } | d d dg} | | d <q| dr|| }|d d dg}||d <q|dg}|dd}|rr|di}|dd}d}t||||d<||ddddddd|d ||d<|dgD]Z}|dddkr|dd|d<|d ddkr~|d!d|d!<q~|d"gD]}|dddkr,|d#d$|d#<|d%d&|d%<|d ddkr|d'd$|d'<|d(d&|d(<q|d)gD]}|dddkr|dd|d<|d*d |d*<|d ddkr~|d!d|d!<|d+d |d+<q~dS),a Scale a figure and translate it to sub-region of the original figure canvas. Note: If the input figure has a title, this title is converted into an annotation and scaled along with the rest of the figure. Note: This function mutates the input fig dict Note: This function assumes that the normalize_subplot_ids function has already been run on the figure, so that all layout subplot properties in use are explicitly present in the figure's layout. Parameters ---------- fig: dict A plotly figure dict scale_x: float Factor by which to scale the figure in the x-direction. This will typically be a value < 1. E.g. a value of 0.5 will cause the resulting figure to be half as wide as the original. scale_y: float Factor by which to scale the figure in the y-direction. This will typically be a value < 1 translate_x: float Factor by which to translate the scaled figure in the x-direction in normalized coordinates. translate_y: float Factor by which to translate the scaled figure in the x-direction in normalized coordinates. r<r;cs,t|ddt|ddgSNrr)min)r3)scale_x translate_xr0r1scale_translate_xsz+_scale_translate..scale_translate_xcs,t|ddt|ddgSrdre)r4)scale_y translate_yr0r1scale_translate_ysz+_scale_translate..scale_translate_ycsH|di}|dddg}|dddg}||d<||d<dS)Ndomainr3rr)r4)r>r?)rSrmr3r4)rirlr0r1perform_scale_translates   z1_scale_translate..perform_scale_translater=rrrmrr)rrJtitleN titlefontsize FrNg?g)\(?centerZbottom) textZ showarrowrMrOr3r4ZxanchorZyanchorZfontrMr3rOr4rKZx0g?Zx1g?Zy0Zy1rLZsizexZsizey)r>r?_domain_trace_typesrPr^r_roundappend)rBrgrjrhrkr<r;rnrCrDpropr5rZx_domainrZy_domainrJrorpZtitle_fontsizeZ min_fontsizerSr0)rirlrgrjrhrkr1_scale_translates|              rzcCsF|dg}|t|dg|di}t||didS)ae Merge a sub-figure into a parent figure Note: This function mutates the input fig dict, but it does not mutate the subfig dict Parameters ---------- fig: dict The plotly figure dict into which the sub figure will be merged subfig: dict The plotly figure dict that will be copied and then merged into `fig` r<r;N)r>extendcopydeepcopyr? merge_layout)rBZsubfigr<r;r0r0r1 merge_figure s  rcCs|D]\}}t|tr2||vr2t|||qt|trj||drjt||dtrj|||q|dkr|dkr|ddrq|durt|||<qdS)a: Merge layout objects recursively Note: This function mutates the input obj dict, but it does not mutate the subobj dict Parameters ---------- obj: dict dict into which the sub-figure dict will be merged subobj: dict dict that sill be copied and merged into `obj` NrZstylezwhite-bg) items isinstancedictr~r\r?r{r|r})rSZsubobjryvalr0r0r1r~8s  r~c stt|}t|d|}|||||}fdd|D}g}tt|D]<}||t|d|}td|||} ||| fqR|S)a Compute normalized domain tuples for a list of widths and a subplot spacing value Parameters ---------- widths: list of float List of the desired withs of each subplot. The length of this list is also the specification of the number of desired subplots spacing: float Spacing between subplots in normalized coordinates Returns ------- list of tuple of float r)csg|] }|qSr0r0)rGwZ total_widthr0r1 rz,_compute_subplot_domains..N)floatsumlenrangerfrx) ZwidthsZspacingZ widths_sumZ total_spacingZrelative_spacingZrelative_widthsZdomainscZ domain_startZ domain_stopr0rr1_compute_subplot_domains[s  r2Fc& s\dd|D}dd|dD}t|} t|} d} t| D]<} t| D].} || | }|s^qH| |diddM} qHq<| rd nd }t| D]x} t| D]j} || | }|sq|d id |}|rt||| || <|d id |}|rt||| || <qq|r>|| d|}|t|fdd|D}nd|rx|| d|}|t|fdd|D}ndt||}t||}gid}tt||D]\} \}}tt||D]\} \}}|rt |}t |t |}|r| |d<| dkr|d i dd |rJ| |d<| dkrJ|d i dd t ||| rd| }d| }| dkr|d| nd}| dkrd| nd}||} ||}!t|| |!|| |d|| |dnx|d d |}"|d d |}#|d|d|#|| }|d|d|"|| }t||||d|dt||qĐq| rlddi|d<|r||d d <n0| rd|d d<nt||| d|d d <|r||d d <n0| rd|d d<nt||| d|d d <|r&|d D]\}$}%|$drd|%d<q|rX|d D]\}$}%|$dr8d|%d<q8|S)a Construct a figure from a 2D grid of sub-figures Parameters ---------- figures_grid: list of list of (dict or None) 2D list of plotly figure dicts that will be combined in a grid to produce the resulting figure. None values maybe used to leave empty grid cells row_spacing: float (default 50) Vertical spacing between rows in the gird in pixels column_spacing: float (default 50) Horizontal spacing between columns in the grid in pixels coordinates share_xaxis: bool (default False) Share x-axis between sub-figures in the same column. Also link all x-axes in the figure. This will only work if each sub-figure has a single x-axis share_yaxis: bool (default False) Share y-axis between sub-figures in the same row. Also link all y-axes in the figure. This will only work if each subfigure has a single y-axis width: int (default None) Final figure width. If not specified, width is the sum of the max width of the figures in each column height: int (default None) Final figure width. If not specified, height is the sum of the max height of the figures in each row Returns ------- dict A plotly figure dict cSsg|]}dqSr0rG_r0r0r1rrzfigure_grid..cSsg|]}dqSrr0rr0r0r1rrrTZconfig responsiveFNir;widthr'r)csg|] }|qSr0r0)rGZwi)column_width_scaler0r1rrg?csg|] }|qSr0r0)rGhi)row_height_scaler0r1rr)r<r;rrg?g@Zautosizer3rZr4)rrr?rQrr enumeratezipr|r}rFrVr_rcrzrrr^)&Z figures_gridZ row_spacingZcolumn_spacingZ share_xaxisZ share_yaxisrr'Z row_heightsZ column_widthsZnrowsZncolsrrrZ fig_elementdefaultrhZavailable_column_widthZavailable_row_heightZcolumn_domainsZ row_domainsZ output_figureZfig_rowZ row_domainrBZ column_domainZsubplot_offsetsrgrjZpxpyZsxZsyZ fig_heightZ fig_widthryrr0)rrr1 figure_grid~s)                *       rc Cs~t|tr|nd}t|trRt|d}t|trRt||krRtd|t|fzt||}Wn@ty}z(tj |}|dur||WYd}~Sd}~00t|trNg}t dd|d}t |dD]x} | dkr| |d|| fq| |kr| |d|dfq| || || df| || || fq|St|trtt||||fd\}\}}t|S)aHConverts a cmap spec to a plotly colorscale Args: cmap: A recognized colormap by name or list of colors levels: A list or integer declaring the color-levels cmin: The lower bound of the color range cmax: The upper bound of the color range Returns: A valid plotly colorscale Nr)zwThe number of colors in the colormap must match the intervals defined in the color_levels, expected %d colors found %d.rr)Zclip)rr,r\r ValueErrorr ExceptionrZ PLOTLY_SCALESr?npZlinspacerrxrZmake_colorscale) ZcmapZlevelsZcminZcmaxZncolorsZpaletteeZ colorscaleZscaleir0r0r1get_colorscale%s<          r_dimc Csi}|diD]R\}}|dddkr||d}||g|dd}||f}|||q|D]r\}} t| dkrqp| d\} } | ddD]B\}} | | d <d | vrd | vrd d t| d | d gD| d <qqpdS) a Configure matching axes for a figure Note: This function mutates the input figure Parameters ---------- fig: dict The figure dictionary to process. matching_prop: str The name of the axis property that should be used to determine that two axes should be matched together. If the property is missing or None, axes will not be matched r;r)rXrYNr[rrrZrcSsg|]}t|r|ndqS)N)r)rGvr0r0r1r{sz5configure_matching_axes_from_dims..)r?rr>replacerxrr) rBZ matching_propZaxis_mapkrZ matching_valZaxis_refZ axis_pairrZ axis_pairsZmatches_referenceZ linked_axisrYr0r0r1!configure_matching_axes_from_dimsSs$     rcCszt|}|D]h}||}|dr.||q t|trBt|q t|ttfr |r t|dtr |D] }t|qfq dS)z Remove all HoloViews internal properties (those with leading underscores) from the input figure. Note: This function mutates the input figure Parameters ---------- fig: dict The figure dictionary to process. rrN)r\r^r_rr clean_internal_figure_propertiestuple)rBZ fig_propsryrelr0r0r1rs      r)rrFFNN)NNN)r)$Z __future__rr|reZnumpyrZplotlyrZ core.utilrrutilrr rvrPr8r@Zlegend_trace_typesZ STYLE_ALIASEScompiler*r2r7r:rFrVrcrzrr~rrrrrr0r0r0r1s    , (6r|#$ ( . -