e/UddlZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z mZdd lmZdd lmZmZmZmZGd d e ZGd dee ZdS)N) AnnularWedge) GridInterface)HoloMap)dimension_sanitizeris_nan) ColorbarPlotCompositeElementPlot)BokehOverlaySelectionDisplay)base_propertiesfill_propertiesline_propertiestext_propertiesceZdZejddidZejdeee fZ ej ddZ ej dd Z ejd d Zejd d Zed ZgdezezezZeZedZfdZdZdZddZfdZfdZ xZ!S) HeatMapPlotNaNwhite Dictionary to specify colors for clipped values. Allows setting color for NaN values and for values above and below the min and max value. The min, max, or NaN color may specify an RGB(A) color as a either (1) a color hex string of the form #FFFFFF or #FFFFFFFF, (2) a length-3 or length-4 tuple specifying values in the range 0-1, or (3) a named HTML color.defaultdocr)rclass_Fz- Whether to show legend for the plot.- Whether the HeatMap should be radialNa- Add separation lines to the heatmap for better readability. By default, does not show any separation lines. If parameter is of type integer, draws the given amount of separations lines spread across heatmap. If parameter is of type list containing integers, show separation lines at given indices. If parameter is of type tuple, draw separation lines at given categorical values. If parameter is of type function, draw separation lines where function returns True for passed heatmap category.rect)single)cmapcolordilatec t|tr|jn|}||djt fddDrddk pddS)Nplotc3 K|]}|vV dSN).0ooptss @lib/python3.11/site-packages/holoviews/plotting/bokeh/heatmap.py z(HeatMapPlot.is_radial..>s'XX1Q$YXXXXXX) start_angle radius_inner radius_outerradialF) isinstancerlastlookup_optionsoptionsanyget)clsheatmapr's @r( is_radialzHeatMapPlot.is_radial:s",Wg">">K',,G!!'622:XXXX(WXXXXX7((8,,56U;?88He;T;T Vr*cRt|j|Sr#)super _get_factorsgridded)selfelementranges __class__s r(r:zHeatMapPlot._get_factorsAsww##GOV<<flat)applyr;Tflatten)r< transformr=r>s r(_element_transformzHeatMapPlot._element_transformDs+wvEJJLTTVVVr*cd|dddD\}}}|jr||}}||jd|||}d|vrd|vrd|d<d|d<d|d <n d |vrd |d <|js|jd |jr i||d |dd|fS|j}|dd\} } |j || } d} | j dvr| | } d}n| | d} | j ddkrDtj| d}t!j|dj} n't)| r|jg| j dzng} | j} d}|j || }d}|j dvr| | }d}n| | d}|j ddkrDtj|d}t!j|dj}n't)|r|jg|j dzng}|j}d}| dd}|j}|jr||}}|| ||d |i}| | |d<|||d<d|jvrI|jsB|jD]:t/j}fd| D||<;d|D}|||d |d||d|fS)Nc34K|]}t|VdSr#rr%ds r(r)z'HeatMapPlot.get_data..Hs+VVa&q))VVVVVVr*Tlabelrr line_alpha line_widthselection_line_alphanonselection_line_alpha line_colorrzHeatMap element index is not unique, ensure you aggregate the data before displaying it, e.g. using heatmap.aggregate(function=np.mean). Duplicate index values have been dropped.zvaluesfieldrE)xy fill_colorSUOr FrAaxiswidthheighthoverc\g|](}t|rdn|)S-r pprint_valuer%vvdims r( z(HeatMapPlot.get_data..sL#N#N#N'(+1))#M339J9J19M9M#N#N#Nr*ci|]?\}tfdtjD<|@S)c3 K|]}|vV dSr#r$)r%gks r(r)z2HeatMapPlot.get_data...s'NNQ!VNNNNNNr*)r3RadialHeatMapPlot _style_groupsvalues)r%rgrms @r( z(HeatMapPlot.get_data..sgPPP$!QNNNN%6%D%K%K%M%MNNNNNPAPPPr*)rVrWrXr_r^) dimensions invert_axes_get_colormappervdims_uniqueparamwarning static_sourcer; interfacedtypekinddimension_valuesshaper_infer_interval_breaksnpdiffrCrDlen default_spanhandlesrnameitems)r<r=r>stylerVrWzcmapper aggregatexdimydimxtypewidthsxvalsr^edgesytypeheightsyvalsr_zvalsdata sanitizedrhs @r(get_datazHeatMapPlot.get_dataGsAVV73E3ED3E3Q3QRTSTRT3UVVV1a  (AqQ'' a(8'65QQ u $ $U)B)B"#E, ,-E( )/0E+ , ,  & &")E,  L J   K L L L   iQQ X_6`6`aachh hO ))++BQB/ d#)))T:: :  ..t44EEE..t%.@@E{1~!!%sourcer?s r(rzHeatMapPlot._init_glyphsse T7FF;;; 4$+C@@@ 4$+C@@@@@r*ct||||jd}||||jd||||jddS)Nr!rVr\rW)r9_update_glyphsrrrr)r<r=r>rr!r?s r(rzHeatMapPlot._update_glyphsso w666|F# 4$+C@@@ 4$+C@@@@@r*)rV)"__name__ __module__ __qualname__rwDictclipping_colors ClassSelectorintfloattuplepaddingBoolean show_legendr. Parameterrrdict _plot_methodsr rr style_optsr selection_display classmethodr7r:rFrrrr __classcell__r?s@r(rr s ej%)9@?@@@O"e!!S%4GHHHG%-40111KU]5/0111FU_T0FU_T0FD'''M---?!"$34J5466VV[V =====WWWKAKAKAZQQQQ AAAAA AAAAAAAAAr*rceZdZejddidZejejdz dZ ejddd Z ejd d d Z ej ddZ ej ddZejddZejddZejddZej ddZej ddZejddZddddd Zgd!Zd"eezDd#eezDzd$eezezDzd%eDzd&gzZfd'Zd9d(Zed)Z fd*Z!d:d,Z"d-Z#d.Z$d/Z%d0Z&d1Z'd2Z(d3Z)ed4Z*d5Z+d6Z,d7Z-fd8Z.xZ/S);rnrrrrrYzf Define starting angle of the first annulus segment. By default, begins at 12 o'clock.g?)r?z: Define the radius fraction of inner, empty space.)rboundsrg?)rr zH Define the radius fraction of outer space including the labels.Na/ Add separation lines between segments for better readability. By default, does not show any separation lines. If parameter is of type integer, draws the given amount of separations lines spread across radial heatmap. If parameter is of type list containing integers, show separation lines at given indices. If parameter is of type tuple, draw separation lines at given segment values. If parameter is of type function, draw separation lines where function returns True for passed segment value.a/ Add separation lines between annulars for better readability. By default, does not show any separation lines. If parameter is of type integer, draws the given amount of separations lines spread across radial heatmap. If parameter is of type list containing integers, show separation lines at given indices. If parameter is of type tuple, draw separation lines at given annular values. If parameter is of type function, draw separation lines where function returns True for passed annular value.rzX Define the maximum radius which is used for the x and y range extents. TrFzA Whether or not to show a complete frame around the plot.z Ticks along x-axis/segments specified as an integer, explicit list of ticks or function. If `None`, no ticks are shown.z Ticks along y-axis/annulars specified as an integer, explicit list of ticks or function. If `None`, no ticks are shown.Zzx Define angle along which yticks/annulars are shown. By default, yticks are drawn like a regular y-axis.annularticksrr) annular_wedgetext multi_linearc)rrrrcg|]}d|zS)xmarks_r$r%ps r(rizRadialHeatMapPlot.LLLQ9q=LLLr*cg|]}d|zS)ymarks_r$rs r(rizRadialHeatMapPlot.rr*cg|]}d|zS)annular_r$rs r(rizRadialHeatMapPlot.s___a:>___r*cg|]}d|zS)ticks_r$rs r(rizRadialHeatMapPlot.s999A8a<999r*rcVtj|i|d|_d|_dSr#)r9__init__xaxisyaxis)r<argskwargsr?s r(rzRadialHeatMapPlot.__init__s1$)&)))  r*cz|dkr|j|jz}|j}n$|dkr|j}|jdtjzz}tj||t |dz}tj|dd|ddgj}|r |ddd}tt||S)z~ Map elements from given `order` array to bins of start and end values for radius or angle dimension. radiusanglerYr N) max_radiusr,r+rpilinspacervstackrCrzip)r<r|orderreversestartendrbinss r( _get_binszRadialHeatMapPlot._get_binss 8  Od&77E/CC W__$E"QY.CUCUa88y&"+vabbz2335  ":DCt$$%%%r*cttjfd|D}|dddf|dddffS)zL Extract first and second value from tuples of mapped bins. c:g|]}|Sr$)r4)r%rVmappers r(riz1RadialHeatMapPlot._get_bounds..s#888A&**Q--888r*Nrr )rarray)rrprs` r( _get_boundszRadialHeatMapPlot._get_bounds sL 888888899QQQT{E!!!Q$K''r*ct|jtr$t||dSdS)z: Limit hover tool to annular wedges only. N)r/glyphrr9_postprocess_hover)r<rendererrr?s r(rz$RadialHeatMapPlot._postprocess_hoversC hnl 3 3 9 GG & &x 8 8 8 8 8 9 9r*combinedc N|dvrdS|j }d|jz|jz}||||fS)zgSupply custom, static extents because radial heatmaps always have the same boundaries. )rr)NNNNrY)r-r)r<viewr> range_typerloweruppers r( get_extentszRadialHeatMapPlot.get_extentssC 1 1 19""DO#d&77ueU++r*cdS)N)NNr$)r<r=s r(_get_axis_dimsz RadialHeatMapPlot._get_axis_dims(s|r*ciS)zOverwrite default axis properties handling due to clashing categorical input and numerical output axes. Axis properties are handled separately for radial heatmaps because of missing radial axes in bokeh. r$)r<rrs r(_axis_propertiesz"RadialHeatMapPlot._axis_properties+s  r*c "t|j|jdddd||d}tddd d d }tddd d d d }tdd}t|j|jddtjzd}|||||dS)z_Create dictionary containing default ColumnDataSource glyph to data mappings. inner_radius outer_radiusr+ end_anglerT)rVrWrrr+rrXrVrWrrcenter)rVrWrr text_alignbottom)rVrWrrr text_baselinexsysrrrrYr)rVrWr+rrannular_wedge_1text_1text_2 multi_line_1arc_1)rrrr)r<rr map_annular map_seg_label map_ann_label map_xmarks map_ymarkss r(get_default_mappingz%RadialHeatMapPlot.get_default_mapping6s T_(6(6'4%001&H&H JJJ sc#*xAAA sc#*x+3555 Td+++ T_&'1RU78MMM $/'' *# %% %r*ct|jjdvr(|j|fd|DS|S)zT Helper function to convert values to corresponding dimension type. SUc:g|]}|Sr$)re)r%rgdims r(riz-RadialHeatMapPlot._pprint..[s'666AC$$Q''666r*)r{r|r; get_dimension)r<r= dim_labelvalsrs @r(_pprintzRadialHeatMapPlot._pprintUsH :?$ & &/// ::C6666666 6 r*c|dkr |jd}n|dkr |jd}trfd|D}nttt fr|||}nVrTtjt|tz  t}|dd|}fd|DS)zp Helper function to compute tick mappings based on `ticks` and default orders and bins. rTrFc*g|]}| |Sr$r$)r%rVrs r(riz;RadialHeatMapPlot._compute_tick_mapping..ls&555aEE!HH5555r*Nc"i|] }|| Sr$r$r%rVrs r(rqz;RadialHeatMapPlot._compute_tick_mapping..vs---q47---r*) xticksytickscallabler/rlistrrceilrrastyper)r<r|rrrtext_nth nth_labelrs ` @r(_compute_tick_mappingz'RadialHeatMapPlot._compute_tick_mapping_s 7??KEGG X  KEG E?? *55555555HH t} - - *>>$w77DHH  *E U5\\ 9::AA#FFI[[y[)H----H----r*c|jtggggS|d||}d|D}t |\}}t j|}t j||jz|jz}t j ||jz|jz}t|||dt j z|zS)zJ Generate ColumnDataSource dictionary for segment labels. NrVrWrrrc4g|]\}\}}|||z dz |zfS)rYr$)r%rrrs r(riz:RadialHeatMapPlot._get_seg_labels_data..sC===(dLUC3;!+u45===r*g?) rrr rrrrsinrcosr) r< order_segbins_segmappingrplabelsradianty_coordx_coords r(_get_seg_labels_dataz&RadialHeatMapPlot._get_seg_labels_dataxs ; "2666 6,,WiJJ==,3MMOO===v,(7##&//DO3doE&//DO3doEg+/111 1r*c |jtggggS|d||}d|D}t |\}}t j|}t jt j|j |z|j z}t j t j|j |z|j z}t|||dgt|zS)zJ Generate ColumnDataSource dictionary for annular labels. Nr"rc(g|]\}}||dfS)rr$)r%rMrs r(riz:RadialHeatMapPlot._get_ann_labels_data..s%JJJ5&)$JJJr*r) rrr rrrrr$deg2rad yrotationrr%r) r< order_annbins_annr(rpr)rr+r,s r(_get_ann_labels_dataz&RadialHeatMapPlot._get_ann_labels_datas ; "2666 6,,Xy(KKJJ'--//JJJf&!!&DN3344v=O&DN3344v=Og#c&kk/+++ +r*ctrfdD}nttrfdD}n_ttr}nGt jt z t}dd|}t j fd|DS)zQ Helper function to get marker positions depending on mark type. c*g|]}| |Sr$r$)r%rVrs r(riz2RadialHeatMapPlot._get_markers..s&444Q55884q444r*c g|] }| Sr$r$)r%rVrs r(riz2RadialHeatMapPlot._get_markers..s///AuQx///r*Nc,g|]}|dS)r r$rs r(riz2RadialHeatMapPlot._get_markers..s!555a555r*) rr/rrrrrrrr)rrrmarkersnth_marks``` r( _get_markerszRadialHeatMapPlot._get_markerss E?? (4444%444GG t $ $ (///////GG u % % (GGws5zzE12299#>>HJJhJ'Gx5555W555666r*c$|jstggS||j||}|j|jz}|j}t j||z|jz}t j||z|jz}t j||z|jz}t j||z|jz} t|| } t||} tt| t| S)T Generate ColumnDataSource dictionary for segment separation lines. r) rrr;rr,rr$r%rr) r<r&r'anglesinneroutery_starty_endx_startx_endrrs r(_get_xmarks_dataz"RadialHeatMapPlot._get_xmarks_datas{ &2"%%% %""4; 8DD$"33&..5(4?:v&8&..5(4?:v&8 %  % tBxxDHH----r*c|jstgS||j||}t|S)r=)r)rrr;)r<r2r3rs r(_get_ymarks_dataz"RadialHeatMapPlot._get_ymarks_datasF{ #r??? """4; 8DD6""""r*c|ddd}d|D\}}}|jr||}}||jd|||}|||} |jri| |fS|j} | |} | |} | dd} | |d }| |d }|||| } |||| } ||||}||||}| d |}t|r0| || \}}| ||}ngg}}tgggg }| d |d}t|r0| || \}}|||}ngg}}tgggg }d |d|d|d||| || || i}d|jvrD|jD]<t!j}fd| D}|||<=|||}|||}|||||d}|| |fS)NTrLrc34K|]}t|VdSr#rIrJs r(r)z-RadialHeatMapPlot.get_data..s+>>a&q))>>>>>>r*rrYr[F)expandedrr"rr+rrrr`c\g|](}t|rdn|)Srbrdrfs r(riz.RadialHeatMapPlot.get_data..sLEEE"(D##0A0A!0D0DEEEr*r)rrrsrtrur ryr;r}rrrrr4rr-rrrrErG)r<r=r>r dim_labelsrVrWrrr(rrrrr&r2r3rr data_text_annr'r+r data_text_seg data_annularrrp data_xmarks data_ymarksrrhs @r(rzRadialHeatMapPlot.get_datasP''d'33BQB7 >>:>>>1a  (AqQ'' a(8'(.77**1g66   &w% %O **1--**1--**14*88..q5.AA ..q5.AA  Wa// Wa//LL!Y77 LL!Y77 >>(I66 x== @)-)9)9(E)J)J &L, 55iJJMM*,b,L 2"B???M>>'9d;; x== @%)%5%5h%F%F "K 55iJJMM%'K 2"B???M&{#Y& & 5!UAu 6 dl " "  1 1/ :: EEEE#,#=#=d#C#CEEE*0 Y''++Ix@@ ++Ix@@ #/'' +$ && We##r*ct||||}|jr*d|jvr!|||jd|S)N color_mapper)r9 _init_glyphcolorbarr_draw_colorbar)r<r!r( propertieskeyretr?s r(rTzRadialHeatMapPlot._init_glyph sYgg!!$SAA = D^t|;;   dl>&B C C C r*)F)r)0rrrrwrrNumberrrr+r,r-rrrrrr. show_framerrr1ro _draw_orderr rrrrrr staticmethodrrrrrr rr r-r4r;rErGrrTrrs@r(rnrns ej%)9@?@@@O%,ruQw5K 5<HC=>>>L 5<VBKLLLLU_T0FU_T0Fc0    JU]4.0111Fu3DEEEJU_Q-=>>>FU_Q-=>>>F R.,---I '0$#+$&&M A@@KLL/O*KLLLLL/O*KLLLM__?_+L+^___`:9999:>DHEJ  &&&&,((\(99999,,,,   %%%>...2111.+++*77\7 .... # # #G$G$G$Rr*rn)numpyrrwbokeh.models.glyphsr core.datar core.spacesr core.utilrrr=r r selectionr stylesr rrrrrnr$r*r(res2 ,,,,,,&&&&&&""""""4444444477777777333333VVVVVVVVVVVVWAWAWAWAWA,WAWAWAt}}}}},l}}}}}r*