a ~bU@sddlmZmZmZddlZddlZddlmZddl m Z ddl m Z m Z ddlmZdd lmZmZdd lmZdd lmZmZmZmZGd d d eZGdddeeZdS))absolute_importdivisionunicode_literalsN) AnnularWedge) GridInterface)is_nandimension_sanitizer)HoloMap) ColorbarPlotCompositeElementPlot)BokehOverlaySelectionDisplay)base_propertiesline_propertiesfill_propertiestext_propertiescseZdZejddiddZejdeee fdZ ej dddZ ej dd dZ ejd d dZejd d dZed d ZgdeeeZeZeddZfddZddZddZdddZfddZfddZ Z!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.Zrect)Zsingle)cmapZcolorZdilatecsTt|tr|jn|}||djtfdddDrHddk pRddS)Nplotc3s|]}|vVqdSN).0oZoptsr?lib/python3.9/site-packages/holoviews/plotting/bokeh/heatmap.py Cz(HeatMapPlot.is_radial..) start_angle radius_inner radius_outerradialF) isinstancer ZlastZlookup_optionsoptionsanyget)clsZheatmaprr"r# is_radial?s zHeatMapPlot.is_radialcstt||j|Sr)superr _get_factorsgridded)selfelementranges __class__rr#r1FszHeatMapPlot._get_factorscCs|j|j|ddjS)NF)r5flat)Zapplyr2Tflatten)r3 transformr4r5rrr#_element_transformIszHeatMapPlot._element_transformcsdd|jddddD\}}}|jr4||}}||jd|||}d|vrtd|vrtd|d<d|d <d|d <nd |vrd |d <|js|jd |jri||d|dd|fS|j}|dd\} } |j || } d} | j dvr| | } d}np|j | dd} | j ddkrMr%z(HeatMapPlot.get_data..TlabelrrZ line_alphaZ line_widthZselection_line_alphaZnonselection_line_alphaZ 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.ZzvaluesZfieldr;)xy fill_colorZSUOr Fr8axiswidthheighthovercs"g|]}t|rdn|qS-r pprint_valuer vZvdimrr#r@scs2i|]*\}tfddtjDs|qS)c3s|]}|vVqdSrr)r gkrr#r$r%z2HeatMapPlot.get_data...)r,RadialHeatMapPlot _style_groupsvaluesrRrrVr# s z(HeatMapPlot.get_data..)rDrErFrLrK) dimensions invert_axes_get_colormappervdimsZ_uniqueparamwarning static_sourcer2Z interfacedtypekinddimension_valuesshaperZ_infer_interval_breaksnpdiffr9r:lenZ default_spanhandlesr nameitems)r3r4r5stylerDrEzcmapper aggregateZxdimZydimZxtypeZwidthsxvalsrKZedgesZytypeZheightsyvalsrLzvalsdata sanitizedrrTr#get_dataLsx$                 zHeatMapPlot.get_datarDcCs"|dus|jrdS|jddS)NzbOnly radial HeatMaps supports marks, to make theHeatMap quads for distinguishable set a line_width)r)r`ra)r3rr4marksrJrrr# _draw_markersszHeatMapPlot._draw_markerscsBtt||||||j|||jdd|j|||jdddS)NrDrIrE)r0r _init_glyphsrxxmarksymarks)r3rr4r5sourcer6rr#ryszHeatMapPlot._init_glyphscsJtt|||||jd}|j|||jdd|j|||jdddS)NrrDrIrE)r0r_update_glyphsrjrxrzr{)r3r4r5rmrr6rr#r}s zHeatMapPlot._update_glyphs)rD)"__name__ __module__ __qualname__r`Dictclipping_colorsZ ClassSelectorintfloattupleZpaddingBooleanZ show_legendr) Parameterrzr{dictZ _plot_methodsrrr style_optsrZselection_display classmethodr/r1r<rvrxryr} __classcell__rrr6r#rs*     M  rcseZdZejddiddZejejdddZ ejddd d Z ejd d d d Z ej dddZ ej dddZejdddZejdddZejdddZej dddZej dddZejdddZddddd Zgd!Zd"d#eeDd$d#eeDd%d#eeeDd&d#eDd'gZfd(d)ZdKd*d+Zed,d-Z fd.d/Z!dLd1d2Z"d3d4Z#d5d6Z$d7d8Z%d9d:Z&d;d<Z'd=d>Z(d?d@Z)edAdBZ*dCdDZ+dEdFZ,dGdHZ-fdIdJZ.Z/S)MrXrrrrrGzf 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.Zannularticksrzr{) annular_wedgetext multi_linearc)rrrrcCsg|] }d|qS)Zxmarks_rr prrr#r@r%zRadialHeatMapPlot.cCsg|] }d|qS)Zymarks_rrrrr#r@r%cCsg|] }d|qS)Zannular_rrrrr#r@r%cCsg|] }d|qS)Zticks_rrrrr#r@r%rcs&tt|j|i|d|_d|_dSr)r0rX__init__ZxaxisZyaxisr3argskwargsr6rr#rszRadialHeatMapPlot.__init__cCs|dkr|j|j}|j}n|dkr:|j}|jdtj}t||t|d}t|dd|ddgj}|r|ddd}t t ||S)z~ Map elements from given `order` array to bins of start and end values for radius or angle dimension. radiusanglerGr N) max_radiusr'r&rgpiZlinspaceriZvstackr9rzip)r3rdorderreversestartendrbinsrrr# _get_binss  zRadialHeatMapPlot._get_binscs8tfdd|D}|dddf|dddffS)zL Extract first and second value from tuples of mapped bins. csg|]}|qSr)r-r rDmapperrr#r@r%z1RadialHeatMapPlot._get_bounds..Nrr )rgarray)rrZrrrr# _get_boundsszRadialHeatMapPlot._get_boundscs"t|jtrtt|||dS)z: Limit hover tool to annular wedges only. N)r*Zglyphrr0rX_postprocess_hover)r3Zrendererr|r6rr#rs z$RadialHeatMapPlot._postprocess_hovercombinedcCs0|dvr dS|j }d|j|j}||||fS)zgSupply custom, static extents because radial heatmaps always have the same boundaries. )rtr)NNNNrG)r(r)r3viewr5Z range_typelowerupperrrr# get_extents's zRadialHeatMapPlot.get_extentscCsdS)N)NNr)r3r4rrr#_get_axis_dims2sz RadialHeatMapPlot._get_axis_dimscOsiS)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. rrrrr#_axis_properties6s z"RadialHeatMapPlot._axis_propertiesc Cs~t|j|jdddd||dd}tddd d d d }tddd d d d d}tddd}t|j|jddtjdd}|||||dS)z_Create dictionary containing default ColumnDataSource glyph to data mappings. inner_radius outer_radiusr& end_anglerC)rDrErrr&rrFrDrErrcenter)rDrErr text_alignZbottom)rDrErrrZ text_baselinexsysrrrrGr)rDrEr&rrZannular_wedge_1Ztext_1Ztext_2Z multi_line_1Zarc_1)rrrgr)r3rnroZ map_annularZ map_seg_labelZ map_ann_labelZ map_xmarksZ map_ymarksrrr#get_default_mappingBs0    z%RadialHeatMapPlot.get_default_mappingcs.|jjdvr*|j|fdd|DS|S)zLHelper function to convert values to corresponding dimension type. ZSUcsg|]}|qSr)rQrRZdimrr#r@ir%z-RadialHeatMapPlot._pprint..)rcrdr2Z get_dimension)r3r4Z dim_labelZvalsrrr#_pprintbs  zRadialHeatMapPlot._pprintcs|dkr|jd}n|dkr&|jd}trBfdd|D}nPtttfrd|||}n.rtt |t  t }|dd|}fdd |DS) zhHelper function to compute tick mappings based on `ticks` and default orders and bins. rTrFcsg|]}|r|qSrrr)rrr#r@|r%z;RadialHeatMapPlot._compute_tick_mapping..Ncsi|]}||qSrrrrrr#r[r%z;RadialHeatMapPlot._compute_tick_mapping..) xticksytickscallabler*rlistrrgceilrirastyper)r3rdrrrZtext_nthZ nth_labelr)rrr#_compute_tick_mappingnsz'RadialHeatMapPlot._compute_tick_mappingc Cs|jdurtggggdS|d||}dd|D}t|\}}t|}t||j|j}t ||j|j}t|||dtj |dS)zBGenerate ColumnDataSource dictionary for segment labels. NrDrErrrcSs(g|] \}\}}|||d|fqS)rGr)r rrrrrr#r@s z:RadialHeatMapPlot._get_seg_labels_data..g?) rrrrlrrgrsinrcosr) r3 order_segbins_segmappingrZlabelsZradianty_coordx_coordrrr#_get_seg_labels_datas    z&RadialHeatMapPlot._get_seg_labels_datac Cs|jdurtggggdS|d||}dd|D}t|\}}t|}tt|j ||j }t t|j ||j }t|||dgt |dS)zBGenerate ColumnDataSource dictionary for annular labels. NrrcSsg|]\}}||dfqS)rr)r rBrrrr#r@r%z:RadialHeatMapPlot._get_ann_labels_data..r) rrrrlrrgrrZdeg2rad yrotationrrri) r3 order_annbins_annrrZrrrrrrr#_get_ann_labels_datas    z&RadialHeatMapPlot._get_ann_labels_datacstrfddD}nTttr:fddD}n6ttrJ}n&ttt}dd|}t fdd|DS)zIHelper function to get marker positions depending on mark type. csg|]}|r|qSrrr)rwrr#r@r%z2RadialHeatMapPlot._get_markers..csg|] }|qSrrr)rrr#r@r%Ncsg|]}|dqS)r rrrrr#r@r%) rr*rrrgrrirrr)rwrrZmarkersZnth_markr)rrwrr# _get_markerss  zRadialHeatMapPlot._get_markersc Cs|jstggdS||j||}|j|j}|j}t|||j}t|||j}t|||j}t|||j} t|| } t||} tt | t | dS)LGenerate ColumnDataSource dictionary for segment separation lines. r) rzrrrr'rgrrrr) r3rrZanglesinnerZouterZy_startZy_endZx_startZx_endrrrrr#_get_xmarks_datas    z"RadialHeatMapPlot._get_xmarks_datacCs*|jstgdS||j||}t|dS)r)r)r{rr)r3rrrrrr#_get_ymarks_datas z"RadialHeatMapPlot._get_ymarks_datacs|jdddd}dd|D\}}}|jr8||}}||jd|||}|||} |jrji| |fS|j} | |} | |} | jddd} | j|d d }| j|d d }|||| } |||| } ||||}||||}| d |}t |r| || \}}| ||}ngg}}t ggggd }| d |d}t |rj| || \}}|||}ngg}}t ggggd }d|d|d|d||| || || i}d|jvr|jD]0tj}fdd| D}|||<q|||}|||}|||||d}|| |fS)NTrArcSsg|] }t|qSrr=r>rrr#r@r%z.RadialHeatMapPlot.get_data..rrGrHF)Zexpandedrrrr&rrrrMcs"g|]}t|rdn|qSrNrPrRrTrr#r@,sr)r\r]r^r_rrbr2rerrrirrrrrjr rkrr)r3r4r5rmZ dim_labelsrDrErnrorrprqrrrsrrrrrZ data_text_annrr&rZ data_text_segZ data_annularrurZZ data_xmarksZ data_ymarksrtrrTr#rvsj                 zRadialHeatMapPlot.get_datacs<tt|||||}|jr8d|jvr8|||jd|S)NZ color_mapper)r0rX _init_glyphZcolorbarrjZ_draw_colorbar)r3rrZ propertieskeyretr6rr#r;szRadialHeatMapPlot._init_glyph)F)r)0r~rrr`rrNumberrgrr&r'r(rrzr{rrr)Z show_framerrrrYZ _draw_orderrrrrrrr staticmethodrrrrrrrrrrrrrrvrrrrr6r#rXs\           JrX)Z __future__rrrr`ZnumpyrgZbokeh.models.glyphsrZ core.datarZ core.utilrr Z core.spacesr r4r r Z selectionrZstylesrrrrrrXrrrr#s