e,xddlZddlZddlZddlmZddlmZmZddl m Z m Z hdZ hdZ hd Zid d d gd d d gdd d gdd d gdd d gdd d gdd d gdd d gdd d gdd d gdd d gdd d gdd d gdd d gdd d gdd d gdd d gdgdgdgdgdgdgdgdgdgdgd gd!gd" Zhd#Zd$d%d&d'd(d)Zejd*Zd+Zd,Zd-Zd.Zd/Zd0Zd1Zd2Zd3Zd4Z dd:Z d;Z!dS)?N)colors)isfinite max_range)color_intervals process_cmap>pietablesankeyparcats parcoords>geopolarscenemapboxternary>rrrbarxaxisyaxisbox candlestickcarpetcontour contourcarpetheatmap heatmapgl histogram histogram2dhistogram2dcontourohlc pointcloudscatter scattercarpet scatterglviolinrrrrr) conemesh3d scatter3d streamtubesurface choropleth scattergeobarpolar scatterpolarscatterpolarglscatterternary scattermapbox>rrr arear!splomr&rr#r.rr)r% waterfallr"r-rr/rr$r2r0r1r opacityheightsymbolmaxzoomminzoom)alpha cell_heightmarkermax_zoommin_zoomz\D*(\d+)ct|}|r#t|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_valrCsubplot_numbers >lib/python3.11/site-packages/holoviews/plotting/plotly/util.py_get_subplot_numberrI|sC*   k * *E U[[^^,, c.|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 rxry) subplot_typesubplot_val_prefixs rH_get_subplot_val_prefixrQs8(w    ) rJc"|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 subplot)_subplot_prop_named_subplot)rOsubplot_prop_names rH_get_subplot_prop_namerVs%*222%( rJc|di}|ddD]}|dd}t|g}|D]i}t|}t |}|||}t |} | dkr|t | z} n|} | |vri|| <jdS)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 layoutdataNtyper#rA) setdefaultget_trace_to_subplotrVrQrIstr) figrXtrace trace_type subplot_typesrOrUrPrFrGlayout_prop_names rH_normalize_subplot_idsrds$^^Hb ) )F&&..YYvy11 )--j"== ) . .L 6| D D !8!F!F ))$57IJJK1==N!!#/#n2E2E#E  #/ v--+-'( ...rJc dtD}d|d<d|d<|dgD]}|dd}t|g}|D]^}t|}t |}|||}t |} t ||| ||<_|di} d D]} | | gD]} | d d } | d kr(t | }t |d||d<| d d}|d kr(t |}t |d||d<|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 ci|]}|dS)rrN).0rOs rH z(_get_max_subplot_ids.. s*;;;'$Q;;;rJrrrrYrZr#rX annotationsshapesimagesxrefrLpaperyrefrM)_subplot_typesr\r]rVrQrImax)r_max_subplot_idsr`rarbrOrUrPrFrGrX layout_propobjrm xref_numberro yref_numbers rH_get_max_subplot_idsrws ;;+9;;;O OG OG$$ ? ?YYvy11 )--j"== ) ? ?L 6| D D !8!F!F ))$57IJJK1==N,/ -~-?-?OL ) ) ?WWXr " "F: V V ::k2.. V VC7763''Dw1$77 +.w/G+U+U(7763''Dw1$77 +.w/G+U+U( V rJc|ddD]}|dd}t|g}|D]}t|}t|}|||}t |} | ||dz} | dkr|t | z||<z|||<|di} i} |D]}||} | dkrt| D]U}| |r>t |} | | z}|t |z}| || |<V| | |dd}|d d}t| D]}| drR| |}|d d }t ||z}|dkrd t |z|d <cd |d <i| d rQ| |}|d d }t ||z}|dkrd t |z|d <d |d <t| D]}|dd dkr| |}|dd}|rd|dd krt ||z}n |dd krt ||z}no|dkrt |nd}|d|z|d<dD]}| |gD]}|rC|dd }|dkr't |}d t ||zz|d<|rC|dd }|dkr't |}d t ||zz|d<dS)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 rYNrZr#rrArXrranchorrMrLaxismatchesrirmrnro) r\r]rVrQrIr^r[listkeys startswithpopupdate) r_offsetsr`rarbrOrUrPrFrGoffset_subplot_numberrX new_subplotsoffsetrsnew_subplot_numbernew_layout_propx_offsety_offsetrry anchor_numberrr{ matches_valmatches_numbersuffixrtrmrurorvs rH_offset_subplot_idsr0s(&&>>YYvy11 )--j"== ) > >L 6| D D "9!F!F ))$57IJJK0==N#W[[q%A%AA "%q((*S1F-G-GG'((,>'(( >$^^Hb ) )FL H H & A::  .. H HK%%l33 H!4[!A!A%3f%<"".5G1H1H"H06 ;0G0G _-  H MM,{{7A&&H{{7A&&HFKKMM**&&  ! !' * * &;'EYYx--F/77(BMq  "%M(:(:":h"%h  # #G , , &;'EYYx--F/77(BMq  "%M(:(:":h"%hFKKMM**:: qs v % %+&D((9d33K :q>S((%8%E%E%PNN ^s**%8%E%E%PNN0>0B0B^,,,"-a.6"9Y; D D ::k2.. D DC Dwwvs++7??"5d";";K"%K(,B(C(C"CCK Dwwvs++7??"5d";";K"%K(,B(C(C"CCK D D DrJc |dg}|di}fdfdfd}|D],}|dd} | tvr ||-|D]2} tD](} | | r||| )3|D]} | dr/|| } | d d d g} | | d <F| d r.|| }|d d d g}||d <|d g}|dd}|rm|di}|dd}d}t |||z zz|d<||ddddddd|d ||d <|d gD]t}|dddkr|ddzz|d<|dddkr|ddzz|d<u|dgD]}|dddkr>|d d!zz|d <|d"d#zz|d"<|dddkr>|d$d!zz|d$<|d%d#zz|d%<|d&gD]}|dddkr;|ddzz|d<|d'd z|d'<|dddkr;|ddzz|d<|d(d z|d(<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. rYrXctt|dzzdt|dzzdgSNrrAmin)rLscale_x translate_xs rHscale_translate_xz+_scale_translate..scale_translate_xAAaD7N[0!44AaD7N[0!446 6rJctt|dzzdt|dzzdgSrr)rMscale_y translate_ys rHscale_translate_yz+_scale_translate..scale_translate_yrrJc|di}|dddg}|dddg}||d<||d<dS)NdomainrLrrArM)r[r\)rtrrLrMrrs rHperform_scale_translatez1_scale_translate..perform_scale_translatesp"-- JJsQF # # JJsQF # #''**s ''**s rJrZr#rrrrArrjtitleN titlefontsize Frng?g)\(?centerbottom) text showarrowrmrorLrMxanchoryanchorfontrmrLrorMrkx0g?x1g?y0y1rlsizexsizey)r[r\_domain_trace_typesrprrroundappend)r_rrrrrYrXrr`raproprOrx_domainry_domainrjrrtitle_fontsize min_fontsizertrrs ```` @@rH_scale_translaters@ >>&" % %D ^^Hb ) )F666666666666++++++++YYvy11 , , , # #E * * *66* 6 6L|,, 6''t 555 6:: ??7 # # :4LEyyAq622H//99E(OO __W % % :4LEyyAq622H//99E(O**]B//K JJw % %E ,JJ{B// "vr22 !,#1L#@G"K#LMM &    !,}zz-,,AA 7764 G + +wwsC((72[@CH 7764 G + +wwsC((72[@CHzz(B''DD 7764 G + +d++g5 CCId++g5 CCI 7764 G + +d++g5 CCId++g5 CCIzz(B''99 7764 G + +wwsC((72[@CH777A..8CL 7764 G + +wwsC((72[@CH777A..8CL 99rJc|dg}|tj|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` rYrXN)r[extendcopydeepcopyr\ merge_layout)r_subfigrYrXs rH merge_figurersz >>&" % %DKK fjj4455666^^Hb ) )FHb1122222rJc|D]\}}t|tr||vrt|||5t|trS||dr=t||dtr||||dkr|dkr|ddr|tj|||<dS)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` Nrstylezwhite-bg) items isinstancedictrr~r\rrr)rtsubobjrvals rHrr6s\\^^++ c c4  +TS[[ TC ( ( ( (d## +ggdD!! +T1t,, + I  S ! ! ! ! W__ !2!2swww7M7M!2  _ c**CI#++rJc| tt|}t|dz |z}||z |||zz } fd|D}g}tt|D]O}||zt|d|z}t d|||z} ||| fP|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 widths 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 rAcg|]}|z SrNrN)rgw total_widths rH z,_compute_subplot_domains..os999QK999rJN)floatsumlenrangerr) widthsspacing widths_sum total_spacingrelative_spacingrelative_widthsdomainsc domain_start domain_stoprs @rH_compute_subplot_domainsrXs$s6{{##J[[1_/M},K*}"<=9999&999OG 3v;;  44++c/"1"2E.F.FF !\OA,>>??  k23333 NrJ2Fc  &'d|D}d|dD}t|} t|} d} t| D]R} t| D]@} || | }|s| |diddz} AS| rdnd }t| D]} t| D]} || | }|s|d id |}|rt||| || <|d id |}|rt||| || <|r,|| d z |zz }|t |z &&fd|D}nd&|r,|| d z |zz }|t |z ''fd|D}nd't ||}t ||}gid}t t||D]\} \}}t t||D]\} \}}|rtj |}t|t|}|r5| |d<| dkr*|d i dd|r5| |d<| dkr*|d i ddt||| rTd| z }d| z }| d krd| z nd}| d krd| z nd}||z } ||z }!t|| |!|| z|dz z|| z|dz zn|d d |'z}"|d d |&z}#|d |dz |#|| z z}|d |dz |"|| z z}t||||d|dt||| rddi|d<|r ||d d <n/| r d|d d<n!t ||| d z zz|d d <|r ||d d <n/| r d|d d<n!t ||| d z zz|d d <|r:|d D]\}$}%|$drd|%d< |r:|d D]\}$}%|$drd|%d< |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 grid 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 cg|]}dSrNrg_s rHrzfigure_grid..s,,,!2,,,rJcg|]}dSrrNrs rHrzfigure_grid..s111AR111rJrTconfig responsiveFNirXwidthr7rAcg|]}|zSrNrN)rgwicolumn_width_scales rHrzfigure_grid..sIIIR00IIIrJg?cg|]}|zSrNrN)rghirow_height_scales rHrzfigure_grid..sCCCr,,CCCrJ)rYrXrrg?g@autosizerLr|rM)rrr\rqrr enumerateziprrrdrwrrrrrr)( figures_grid row_spacingcolumn_spacing share_xaxis share_yaxisrr7 row_heights column_widthsnrowsncolsrrr fig_elementdefaultrhavailable_column_widthavailable_row_heightcolumn_domains row_domains output_figurefig_row row_domainr_ column_domainsubplot_offsetsrrpxpysxsy fig_height fig_widthrrrrs( @@rH figure_gridr {sNR-,|,,,K11a111M   E   EJ 5\\QQu Q QA&q/!,K  +//(B77;;L%PP PJJ  Q !)ddcG 5\\ 8 8u 8 8A&q/!,K "--11'7CCA <#&q-*:#;#; a "--11(GDDA 8!$Q A!7!7 A 8 !!&%!)~)E!E3c-6H6HHIIII=III   %k(AA/#k2B2BBCCCC{CCC .m^LLN*; DDK2..M$-c, .L.L$M$M(1(1  GZ'0Wn1M1M'N'N' 1' 1 #A#]& 1mC((&s+++"6}"E"EA/0OG,Avv"--11'4@@@A/0OG,Avv"--11'4@@@#C999 hG hG).#u++B).#u++B B B$S"b'!)BrE/719RPRU?SSSS!$X!2!28W!E!EHX!XJ #H 1 1'7 C CFX XI,Q/-2BByS`abScGcdG)!}z!}<kZ[nA\]G$Wg}Q/?A]C000O' 1R7#/"6 h ,2 h))  .2 h ++   {eai8 8 h)  +0 h((  .2 h ++M""^uqy%AA h(%&x06688 % %ID#w'' %!$I%&x06688 % %ID#w'' %!$I rJct|tr|nd}t|tr[t|dz }t|tr4t||kr!t d|t|fz t ||}n<#t $r/}tj |}|||cYd}~Sd}~wwxYwt|trg}tj dd|dz}t|dzD]} | dkr$| |d|| f,| |kr$| |d|dfV| || || dz f| || || f|St|trt||||f\}\}}tj|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 NrAzwThe number of colors in the colormap must match the intervals defined in the color_levels, expected %d colors found %d.rr)clip)rrDr~r ValueErrorr Exceptionr PLOTLY_SCALESr\nplinspacerrrmake_colorscale) cmaplevelscmincmaxncolorspalettee colorscalescaleis rHget_colorscaler "s4#63//9ffTG&$5f++/ dD ! ! 5c$ii7&:&:J!(T3455 5tW-- )--d33  G  &#0  Aq&(++vax : :AAvv!!58WQZ"89999f!!59gbk":;;;;!!58WQqS\":;;;!!58WQZ"89999 FD ! !0 / V4,!0!0!0$  !' * **s B C&$C CC_dimcDi}|diD]t\}}|dddkra||d}||g|dd}||f}|||u|D]k\}} t | dkr| d\} } | ddD]<\}} | | d <d | vr.d | vr*d t | d | d gD| d <=ldS) 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 rXrArzr{Nr}rrr|rc4g|]}t|r|ndS)N)r)rgvs rHrz5configure_matching_axes_from_dims..xs4(((34!.AA$(((rJ)r\rr[replacerrr) r_ matching_propaxis_mapkr$ matching_valaxis_ref axis_pairr axis_pairsmatches_reference linked_axisr{s rH!configure_matching_axes_from_dimsr/Psp"H"%%++-- 5 51 QqS6V  5555L    b 1 1 1yy,,H"1 I \ " ) )) 4 4 4"))   : z??Q   )3A&;!!""~  GAt/DO$7k#9#9((8A4=R]^eRfBg8h8h((( G$   rJcxt|}|D]}||}|dr||5t|trt |Zt|tt fr1|r/t|dtr|D]}t |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~rrrr clean_internal_figure_propertiestuple)r_ fig_propsrrels rHr1r1}sS I55$i ??3   5 GGDMMMM T " " 5 ,S 1 1 1 1 dE] + + 5 5 3q648P8P 5 5 50444455rJ)rrFFNN)NNN)r!)"rrenumpyrplotlyr core.utilrrutilrr rrprTr]legend_trace_types STYLE_ALIASEScompilerBrIrQrVrdrwrrrrrr r r/r1rNrJrHr=se ,,,,,,,,00000000IHH@??=<<) Wg.) Wg.)Wg. )  Wg. ) Wg. )Wg.)Wg.)Wg.)Wg.)Wg.)Wg.) Wg.)Wg.)Wg.) Wg.!)"Wg.#)$ Wg.%)*IIIII''!   !kZQ)))X:$ (#&& )) bj%% ::8%.%.%.P333loDoDoDdy9y9y9x3330+++D   H!!! ddddN++++++++\****Z55555rJ