LeBddlmZddlZddlZddlmZmZddlm Z m Z ddl m Z ddl mZddlmZmZd d lmZd d lmZd d lmZd dlmZGddeeZGddeZdS))productN)LineCollectionPatchCollection)CircleWedge) GridInterface)HoloMap)dimension_sanitizeris_nan) HeatMapMixin) ColorbarPlot) QuadMeshPlot) filter_stylescteZdZejddidZejdeee fZ ej ddZ ej dd Z ejd d Zejd d Zejd d Zejd dZedZdZdZdZddZdZdZd S) HeatMapPlotNaNwhitea 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 color hex string of the form #FFFFFF or #FFFFFFFF or a length 3 or length 4 tuple specifying values in the range 0-1 or a named HTML color.defaultdocr)rclass_F- Whether the HeatMap should be radial7 Whether to annotate each pixel with its value.Na- 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. Ticks along x-axis/segments specified as an integer, explicit list of ticks or function. If `None`, no ticks are shown. Ticks along y-axis/annulars specified as an integer, explicit list of ticks or function. If `None`, no ticks are shown.c t|tr|jn|}||djt fddDrddk pddS)Nplotc3 K|]}|vV dSN).0ooptss >lib/python3.11/site-packages/holoviews/plotting/mpl/heatmap.py z(HeatMapPlot.is_radial..Cs'XX1Q$YXXXXXX) start_angle radius_inner radius_outerradialF) isinstancer lastlookup_optionsoptionsanyget)clsheatmapr's @r( is_radialzHeatMapPlot.is_radial?s",Wg">">K',,G!!'622:XXXX(WXXXXX7((8,,56U;?88He;T;T Vr*c|jdiD]}|i}|D]"\}}|||ddd||<#|S)N annotationsdatacenter)xyxycoordshorizontalalignmentverticalalignment)handlesr4valuesremoveitemsannotate)selfaxr9ar@ plot_coordtexts r(_annotate_plotzHeatMapPlot._annotate_plotFs!!-44;;==  A HHJJJJ + 1 1 3 3 J J J"$++dz7=BJ@H#.#J#JGJ  r*c|jd}||}|ddtj|dz z}|ddtj|dz z}t ||}i} t ||D]0\} } t| rdn|| } | | | <1| S)Nr@-) vdimsdimension_valuesflattennpdiffrzipr pprint_value) rEelementxvalsyvalsval_dimvalsxposypos plot_coordsr9rHvrIs r(_annotate_valueszHeatMapPlot._annotate_valuesRs-"''0088::SbSzBGENN2--SbSzBGENN2--dD))   d33 + +MJ ))@33)=)=a)@)@D&*K # #r*ct |j\ |jr c ||dj}|d}|j|ddt j|dz z}|s|j d} fd|D} tt|| }|d} | j|ddt j|dz z} |s|jd}fd|D} tt| | } || fS) Nr!xticksrLrMFc:g|]}|Sr$rU)r%kxdims r( z.HeatMapPlot._compute_ticks..k'>>>t((++>>>r*yticksc:g|]}|Sr$rc)r%rdydims r(rfz.HeatMapPlot._compute_ticks..srgr*) kdims invert_axesr1r2r4rRrSgriddedrPlistrT)rErVrWrXxfactorsyfactorsr'rar[xlabelsrhr\ylabelsrerjs @@r(_compute_tickszHeatMapPlot._compute_ticks_sQ] d   $tJD$""7F33;(## >":r 11D I"?;;D%HH>>>>X>>>G#dG,,--F(## >":r 11D I"?;;D%HH>>>>X>>>G#dG,,--Fv~r*xcP||jrdS|jddS)NzuOnly radial HeatMaps supports marks, to make theHeatMap quads more distinguishable set linewidthsto a non-zero value.)r.paramwarning)rErFrVmarksrAfactorsaxiss r( _draw_markerszHeatMapPlot._draw_markersxs; =DK= F 2 3 3 3 3 3r*c|d}|d}|dd}gdfd|D}|j|i|}|jr |r||||jd<|||j|j|d|d|||j|j |d |d d |iS) Nrorpr9annularxmarksymarkscTi|]#\}tfdD |$S)c3 K|]}|vV dSr#r$)r%prds r(r)z6HeatMapPlot.init_artists...s'!;!;Q!q&!;!;!;!;!;!;r*)r3)r%r^rdprefixess @r( z,HeatMapPlot.init_artists..sQ===1!!;!;!;!;(!;!;!;;;=q!===r*rrt)rzryartist) poprC pcolormesh show_valuesrJr@r{ current_framerr) rErF plot_args plot_kwargsrorpr9rrs @r( init_artistszHeatMapPlot.init_artistss-??:..??:..!oomT:: 222==== (9(9(;(;===  9[99   O  O*.*=*=b+*N*NDL ' 2t14;$Q<  = = = 2t14;$Q<  = = =&!!r*c||j\}}|j}|js|jd|dd}t j|t j t j |}|j r||}}|j ddddddf}|j ||}|jdvr&t j|jddzd z } n+||d } t%j| } |j ||} | jdvr&t j|jd dzd z } n+||d } t%j| } t)||jid g} t)||jid g} ||| | | | \}}| |d <| |d<|jr||j| | |d<|jd }|||||d|vr-|d|df|d<| | |f|||dfS)NzHeatMap 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.r Fflat)maskrLSUOr?expandedrryrorpr9vminvmaxclimrarh)rkrm_uniquervrwrPrRmaarray logical_notisfiniterlT interfacedtypekindarangeshaper _infer_interval_breaksrnr4namersrr_rO _norm_kwargsr)rErVrangesstylererj aggregater:xtyperWytyperXrorprarhvdims r(get_datazHeatMapPlot.get_datas] dO  L J   K L L L ))!%)88u{{4bnR[5F5F&G&G{HH   &t$D6$$B$"*%D#)))T:: :  IdjmAo..s2EE..te.DDE!8??E#)))T:: :  IdjmAo..s2EE..te.DDE!8??E 49b1155iDDEE 49b1155iDDEE,,WeUHhWW$j$j   X#'#8#8%QV#W#WE- }Q '65$777 U??!IIf--uyy/@/@@E&Mud#Uv,P,PPPr*)rt)__name__ __module__ __qualname__rvDictclipping_colors ClassSelectorintfloattuplepaddingBooleanr.r Parameterrrrarh classmethodr7rJr_rsr{rrr$r*r(rrs ej%)9@0111O"e!!S%4GHHHG U]5/0111F %-4:;;;KU_T0FU_T0FU_R.=>>>FU_R.=>>>FVV[V      23333"""$+Q+Q+Q+Q+Qr*rc.eZdZejejdz dZejddZejddd Z ejd d d Z ej d dZ ej ddZ ejddZejddZejddZejddZejddgZgdZgdZed#dZedZedZedZd$d Zd!Zd"ZdS)%RadialHeatMapPlotr z_ Define starting angle of the first annulars. By default, beings at 12 o clock.rrzX Define the maximum radius which is used for the x and y range extents. g?)rrz: Define the radius fraction of inner, empty space.)rboundsrg?)rrzH Define the radius fraction of outer space including the labels.TrFrNa/ 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.rrpolar)robjectsr})annular_edgecolorsannular_linewidthxmarks_linewidthxmarks_edgecolorcmapymarks_linewidthymarks_edgecolorct|}tj|||dz}|r |ddd}tt |ddtjdzz|}|S)z]Map elements from given `order` array to bins ranging from `start` to `end`. rNrLr )lenrRlinspacernrTpi)startendorderreversesizermappings r(_map_order_to_ticksz%RadialHeatMapPlot._map_order_to_tickssl 5zzUC22  "DDbD\Fs6#2#;a0%8899r*c$fd|DS)zJCompute x and y positions for separation lines for given angles. cDg|]}tj|g|ggSr$)rRr)r%rGinnerouters r(rfz:RadialHeatMapPlot._compute_separations.. s0CCCq1e*q%j122CCCr*r$)rrangless`` r(_compute_separationsz&RadialHeatMapPlot._compute_separationss$ DCCCCFCCCCr*ctrfd|D}nttrcratt jt |z tdg}d|dd|D}n&ttrfd|D}ng}|S)Nc0g|]\}}||Sr$r$r%r^lmarkers r(rfz2RadialHeatMapPlot._get_markers..s*66641aFF1II6Q666r*rcg|]\}}|Sr$r$)r%r^rs r(rfz2RadialHeatMapPlot._get_markers..s55541aQ555r*c"g|] \}}|v | Sr$r$rs r(rfz2RadialHeatMapPlot._get_markers..s"88841aAKKQKKKr*) callabler/rmaxrRceilrastyper)ticksrrxnth_marks ` r( _get_markerszRadialHeatMapPlot._get_markerss F   66665666EE  $ $  BGCJJ$788??DDaHIIH5558#4555EE  & & 88885888EEE r*cvtrfd|D}nttrWtt jt |z tdg}|dd|}ntttfr|tt jt |t z tdg}dt|dd|D}nrt}ng}|S)Nc4g|]\}}|||fSr$r$)r%r^rtickers r(rfz0RadialHeatMapPlot._get_ticks..s.;;;1;aV;;;r*rc"g|] \\}}}||f Sr$r$)r%r^rtls r(rfz0RadialHeatMapPlot._get_ticks..$s$NNN!QaWNNNr*) rr/rrrRrrrrrnrT)rrrs ` r( _get_tickszRadialHeatMapPlot._get_tickss! F   ;;;;;;;EE  $ $ BGCJJ$788??DDaHIIH**H*%EE  . . BGCJJV$<==DDSII1MNNHNNs583Df/M/MNNNEE  LLEEE r*combinedc n|dkrtjfdzSddtjdz|j|jzfS)Nhardrrr )rRnanr max_radiusr-)rEviewr range_typekwargss r( get_extentszRadialHeatMapPlot.get_extents,s:   F9Q; 1beAgtt/@@AAr*c z"#|ddd}d|D\}}}|jr||}}|j""|d}"|d} "dd} "fd } | ||}| || } |j} |jdt jzz} t j| | t|d z}| | | |d}d }||j z}t j||t| d z}| ||| }g}tt| D]}|||dz}tt|ddd D]m}t j |||dz}|d |d z }td|d |d |d |}||n|||j}||||}|||j}d|D}| |d<|||||jd d|vr-|d|df|d<|||d}|||j#t3|jt4s#fd#D#|||j} #| d}!|||!fS)NT)labelrc34K|]}t|VdSr#)r )r%ds r(r)z-RadialHeatMapPlot.get_data..5s+>>a&q))>>>>>>r*Frr rcl|jjdvr#|fd|DS|S)NSUc:g|]}|Sr$rc)r%r^dims r(rfz?RadialHeatMapPlot.get_data.._pprint..Cs':::((++:::r*)rr get_dimension) dim_labelrZrrs @r(_pprintz+RadialHeatMapPlot.get_data.._pprint@sEzd**--i88::::T::::Kr*rrrLrrr)widthc.g|]}td|S)r)r)r%rs r(rfz.RadialHeatMapPlot.get_data..cs"777A&Q''777r*rrrr)r~ xseparator yseparatorcXg|]&\}}|tjtz z |f'Sr$)rRrr)r%r^rras r(rfz.RadialHeatMapPlot.get_data..ms4FFFtq!q25#f++-.2FFFr*r) dimensionsrlrmrPr+rRrrrrr,rangerad2degrappendrrrrrQrrOrrrar/rrh)$rErVrr dim_labelsrtrzrWrXzvalsrr+ end_angle bins_segment segment_ticks radius_max radius_min bins_annular radius_tickspatchesjybinixbinrwedgerrradiirr:rhrrras$ @@r(rzRadialHeatMapPlot.get_data2s''d'33BQB7 >>:>>>1a  (AqQO **1u*==**1u*==**15*99     5!!5!!& $q25y0 {; 3u::a<HH 00i16>>  $"33 {:z3u::a<HH // J0577 s5zz"" & &A!A#&D3u::&&ttt, & &z,q1u"566QQj$q'47DG5QQQu%%%%  & ""=$+>>**:z6JJ!!, <<77777g '65'-2BCCC U??!IIf--uyy/@/@@E&M"&OO <<$+s++ GFFFFvFFFFt{;;!f55UE!!r*cgd}d|jD}t|d|}|d}|ddt|fd|ji|}||d|i}|d} | r`d|jD}t|d ||} | d dt | fi| } || | |d<|d } | rhd |jD}t|d ||} | d dt| fd|jd| } || | |d <|S)N)crrrnormrcg|] }|dk| S)r~r$r%gs r(rfz2RadialHeatMapPlot.init_artists..vsBBB1 >>!>>>r*r~ interpolation transformrrcg|] }|dk| S)rr$r#s r(rfz2RadialHeatMapPlot.init_artists..EEEAqH}}a}}}r*r edgecolorsrcg|] }|dk| S)rr$r#s r(rfz2RadialHeatMapPlot.init_artists..r(r*rnone) facecolorr&) _style_groupsrrr transAxesadd_collectionr)rErFrr color_optsgroups edge_optsannuliartistspaths xmark_opts xseparators ymark_opts yseparatorss r(rzRadialHeatMapPlot.init_artistsssCCC BBT/BBB!+y&AA 9% ot,,, MM2<M9MM &!!!V$,'  0EE!3EEEF&{HfjQQJ NN< . . .(==*==K   k * * *$/GL !,'  0EE!3EEEF&{HfjQQJ NN< . . .)%P646LPPDNPPK   k * * *$/GL !r*)F)r)rrrrvNumberrRrr+rr,r-rr.rrrrrarhObjectSelector projectionr- style_opts staticmethodrrrrrrrr$r*r(rrsN%,ruQw5Kc0    J 5<HC=>>>L 5<VBKLLLLU]4.0111F %-4:;;;KU_T0FU_T0FU_Q-=>>>FU_Q-=>>>F&%gyIIIJ333M:::J   \ DD\D   \   \  BBBB >">">"Br*r) itertoolsrnumpyrRrvmatplotlib.collectionsrrmatplotlib.patchesrr core.datar core.spacesr core.utilr r mixinsrrVrrasterrutilrrrr$r*r(rIs[ BBBBBBBB,,,,,,,,&&&&&&""""""44444444!!!!!!!!!!!! lQlQlQlQlQ, lQlQlQ`QQQQQ QQQQQr*