a Df0i@spddlmZddlZddlZddlmZddlmZddl m Z m Z ddl m Z ddlmZmZdd lmZmZdd lmZdd lmZmZmZmZmZmZmZdd lmZGd ddeZ GdddeZ!GdddeZ"ej#dgddZ$ej#dddZ%eddZ&Gdddej'Z(Gddde(Z)GdddZ*Gd d!d!e*Z+Gd"d#d#e*Z,Gd$d%d%e*Z-d&d'Z.dS)() namedtupleN) bothmethod)Dataset)ElementLayout) AdjointLayout) CallbackErrorStore) NdOverlayOverlay) GridSpace)CrossFilterSetDerivedPipe PlotResetSelectionExprSequence SelectModeStream) DynamicMapc@seZdZejddddZdS)_CmapNT)default allow_Noneconstant)__name__ __module__ __qualname__param Parametercmapr r `/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/holoviews/selection.pyrsrc@seZdZejddddZdS)_SelectionExprOverrideNTz= dim expression of the current selection override)rrdoc)rrrrrselection_exprr r r r!r"sr"cs4eZdZejddZfddZeddZZ S)_SelectionExprLayersT)rc s tj||gfddi|dS)NZ exclusiveT)super__init__)selfZ expr_overrideZcross_filter_setparams __class__r r!r'%sz_SelectionExprLayers.__init__cCsD|d}|d}|dddur0dd|dgiSdd|dgiSdS)Nrrr$exprsT)get)clsZ stream_values constantsZoverride_expr_valuesZcross_filter_set_valuesr r r!transform_function*s z'_SelectionExprLayers.transform_function) rrrrListr,r' classmethodr0 __classcell__r r r*r!r%"s  r%ZStylesg?colorsalphaZ RegionElement)region_elementZSelectionStreamsz'style_stream exprs_stream cmap_streams csveZdZdZejdddZejdddZefddZ d d Z d d Z d dZ dddZ eddZddZZS)_base_link_selectionsa Baseclass for linked selection functions. Subclasses override the _build_selection_streams class method to construct a _SelectionStreams namedtuple instance that includes the required streams for implementing linked selections. Subclasses also override the _expr_stream_updated method. This allows subclasses to control whether new selections override prior selections or whether they are combined with prior selections Fz@ Whether to link any streams on the input to the output.rr#Tz3 Whether to highlight the selected regions.c sDtjfi|}t|jd|_t|_i|_i|_| ||_ |S)N)mode) r&instancercross_filter_mode_cross_filter_streamr"_selection_override_selection_expr_streams_plot_reset_streams_build_selection_streams_selection_streams self_or_clsr)instr*r r!r;Qs z_base_link_selections.instancecCsJ|jdkrd|_n4|jdkr$d|_n"|jdkr6d|_n|jdkrFd|_dS)Nreplace overwriteappendunion intersectsubtractinverse)newselection_mode)r(eventr r r! _update_mode`s    z"_base_link_selections._update_modecCsddlm}t||j|j|jd}||j|<|j|t |d|j |<|j fdd}t ||szt |d}|j|jd|j |j|dgd S) z Register an Element or DynamicMap that may be capable of generating selection expressions in response to user interaction events r)Table)r:include_region index_cols)sourcecSs|r||dSN) clear_historyrO resettingstreamr r r!clear_stream_history}sz=_base_link_selections._register..clear_stream_historyr:rXN)elementrQrrN show_regionsrSr?r=Zappend_input_streamrr@Zhistory_stream isinstancerrwatchrP)r(hvobjrQZselection_expr_seqrZZ mode_streamr r r! _registerjs       z_base_link_selections._registercKs:|jjfi|tjtjvr,tdtj||S)NzCannot perform link_selections operation since the selected backend %r is not loaded. Load the plotting extension with hv.extension or import the plotting backend explicitly.)rupdater current_backendZ renderers RuntimeError_selection_transformclone)r(r_kwargsr r r!__call__s  z_base_link_selections.__call__r c s8ddlm}ttr*j}t|jdkrHtfdd|jDS|t j t r t jt jj }|jjjdjdSt j trtjdddkrtfd d|jDStjddd kr|jd }j|d |jSjd Sn tt rt jt jt }t|dddurjjd} |||jj|j|djdSSttttt t!fr0t"fdd#Dttrfdd}t|} t$%| jjdd<n} t&| dr,| } | SSdS)zs Transform an input HoloViews object into a dynamic object with linked selections enabled. r)initialize_dynamiccsg|]}|qSr rd.0elr(r r! sz>_base_link_selections._selection_transform..N)cachenameZ dynamic_mulcsg|]}j|dqS) operationsrirjrrr(r r!rnsZdynamic_operationrrqzQlinked selection: Encountered DynamicMap that we don't know how to recurse into: selection_displaylinkcs g|]\}}||fqSr ri)rkkvrsr r!rnscsjg}D]F\}}t|dD]0\}}|t|kr>|g||||fq q tfdd|DS)Nr csg|]}|qSr )re)rkn)r_r r!rnzO_base_link_selections._selection_transform..compose..)items enumeratelenrHr )argsrfrMrwrxirl)datar_r r!composes  z;_base_link_selections._selection_transform..composecollate)' plotting.utilrhr]rcallbackr}inputsr r issubclasstyperr`r registryrbrtbuild_selectionrBr?r-_cachegetattrrdapplyZ operationrwarningre link_inputsrr r rdictr{listvalueshasattr) r(r_rrrhrZchartobjr[rZ new_hvobjr )rr_rrr(r!rdsn                z*_base_link_selections._selection_transformcCs tdS)zg Subclasses should override this method to return a _SelectionStreams instance NNotImplementedError)r.rEr r r!rAsz._base_link_selections._build_selection_streamscKs tdS)a Called when one of the registered HoloViews objects produces a new selection expression. Subclasses should override this method, and they should use the input expression to update the `exprs_stream` property of the _SelectionStreams instance that was produced by the _build_selection_streams. Subclasses have the flexibility to control whether the new selection express overrides previous selections, or whether it is combined with previous selections. Nr)r(r_r$Zbboxr7rfr r r!_expr_stream_updateds z*_base_link_selections._expr_stream_updated)r )rrr__doc__rBooleanrr\rr;rPr`rgrdr2rArr3r r r*r!r8>s    H r8cseZdZdZejddgdddZejdddZej dd dZ ej dd d d Z ejgd dddZ ejdddZej dddZefddZejdd dddZddZd$ddZeddZeddZed d!Zed"d#ZZS)%link_selectionsz Operation which automatically links selections between elements in the supplied HoloViews object. Can be used a single time or be used as an instance to apply the linked selections across multiple objects. rGrJzP Determines how to combine selections across different elements.)Zobjectsrr#Na7 If provided, selection switches to index mode where all queries are expressed solely in terms of discrete values along the index_cols. All Elements given to link_selections must define the index_cols, either as explicit dimensions or by sharing an underlying Dataset that defines them.r9zi dim expression of the current selection or None to indicate that everything is selected.Tz[ Color of selected data, or None to use the original color of each element.)rrr#)rGrJrIrLzU Determines how to combine successive selections on the same element.g?z" Alpha of unselected data.z" Color of unselected data.c s>tjfi|}i|_i|_d|_g|_i|_|||S)NT)r&r;Z_obj_selectionsZ _obj_regionsZ_reset_regions _datasetsr_install_param_callbacksrCr*r r!r;s zlink_selections.instancer$)r^c Cs|j}|jD]\}}}|j}|j|i|j|<}||vrV||}|j|jdS|t |||j}|||<|j|r|jn|dq dS)Nr) r$r_plot_idrr-rOrclearSelectionDisplay_select) r(Zsel_exprpipedsrawrefds_cacherZsel_dsr r r! _update_pipes-szlink_selections._update_pipescCs@d}t|tsd}t|}t|jd}|j|||f|jjS)ao Returns a parameter which reflects the current selection when applied to the supplied data, making it easy to create a callback which depends on the current selection. Args: data: A Dataset type or data which can be cast to a Dataset Returns: A parameter which reflects the current selection FTr)r]rrrrrHr)r(rrrr r r!selection_param<s   zlink_selections.selection_paramcCsP|dur|jn|}|dur|St|t}|s4t|}|||}|rJ|S|jS)aG Filters the provided data based on the current state of the current selection expression. Args: data: A Dataset type or data which can be cast to a Dataset selection_expr: Optionally provide your own selection expression Returns: The filtered data N)r$r]rrr)r(rr$exprZ is_datasetfilteredr r r!filterPs  zlink_selections.filterc sfdd}j|dgfdd}j|dgfdd}j|d gfd d }j|d gfd d}jj|d gjD],}|fdd}td||jj|dgqdS)Ncs$jD]}|j|_q dSrU)r?rresetrNr:_rYrEr r!update_selection_modegszGlink_selections._install_param_callbacks..update_selection_moderNcsjjj_dSrU)r=rr<r:)rrr r!update_cross_filter_modeqs zJlink_selections._install_param_callbacks..update_cross_filter_moder<cs$jD]}j|_|q dSrU)r?rr\rRrOrrr r!update_show_regionyszDlink_selections._install_param_callbacks..update_show_regionr\cs<j}jj}t|t|kr8jr*d_jj|ddS)NF)r$)r$r=reprr\r>rO)rnew_selection_exprZcurrent_selection_exprrr r!update_selection_exprs zGlink_selections._install_param_callbacks..update_selection_exprr$cs$jj}tjt|kr |_dSrU)r=r$r)rrrr r!selection_expr_changedszHlink_selections._install_param_callbacks..selection_expr_changedcSs|r |dSrU)rVrWr r r!rZszFlink_selections._install_param_callbacks..clear_stream_historyzregistering reset for rX)rr^r=r?rprintZplot_reset_stream) rDrErrrrrrYrZr rr!res4       z(link_selections._install_param_callbackscsltjjgjdtjdtjdgfdd}j|gdt j j }t |dS)Nr4rcsPjjg}j|jddjjdddurLdjjddS)Nr4rrr)unselected_colorselected_colorrOunselected_alphaunselected_cmap selected_cmap)rr5 cmap_streamsrE style_streamr r! update_colorss   z?link_selections._build_selection_streams..update_colors)rrr)r exprs_streamr) _Stylesrrrrrrrr^r%r>r=_SelectionStreams)r.rErrr rr!rAs"   z(link_selections._build_selection_streamscCs|jdurdSt|jS)z= The datashader colormap for unselected data N)r_color_to_cmaprmr r r!rs zlink_selections.unselected_cmapcCs|jdurdSt|jS)z; The datashader colormap for selected data N)rrrmr r r!rszlink_selections.selected_cmap)N)rrrrrZSelectorr<r1rSrr$ZColorrrNZ Magnituderrrr;dependsrrrrr2rApropertyrrr3r r r*r!rs4      = ! rc@s0eZdZdZddZd ddZed ddZdS) rz Base class for selection display classes. Selection display classes are responsible for transforming an element (or DynamicMap that produces an element) into a HoloViews object that represents the current selection state. cCs|SrUr )r(r[r r r!rgszSelectionDisplay.__call__NcCs|dur i}tdSrUrr(selection_streamsr_rr region_streamror r r!rsz SelectionDisplay.build_selectionc Cs|dur i}ddlm}m}ddlm}t||r|j}d}|j|vrn||j}||vrd||}q||ni}||j<z|j j r|dur|j |dddd}| |j ||} n|j jr|dur|j |dddd}|j|} nxt|||fr6t|j dr6|dur|j |ddd}| |j ||} n(|durR|j |dddd }|j|d } Wnty} z/sz.)r)r(rr rr!_get_color_kwarg.s z(OverlaySelectionDisplay._get_color_kwargNcsfddlm}t|jj}|s"tSg}t|D]T} |jg} | dkrN|jddn|} |j | } | j j | g| | |dd} | | q.t|D]<} || } |j | } |j| g} | j j | | dd} | || <q|durZjrZfdd }||jg} |jddj || dd }t|tr|jnt|}t|d ddksBt||rP|d|n | |t|S) Nr) HistogramFruT)streams layer_numberro per_element)rrrcs$|d}|dur|}||S)Nr)Z _empty_region_style_region_element)r[r7r5rfrrmr r! update_regionPsz>OverlaySelectionDisplay.build_selection..update_region)Z link_datasetZ_selection_dims)r[rr}rr5r rangerrerr_build_layer_callbackrH_apply_style_callbackrr]rrrrinsertr)r(rr_rrrrorZ num_layersZlayersrrrZ cmap_streamZlayerrZregionZeltyper rmr!r2s@             z'OverlaySelectionDisplay.build_selectioncCs<g}|jD]$}t|dr$|j|d}||q |j|dS)Nrrrq)rrrr;rH)r.pipelinerrropr r r!_inject_cmap_in_pipeline`s     z0OverlaySelectionDisplay._inject_cmap_in_pipelinec KsJ|||||}|j}|dur,|||}||ur>||jS||SdSrU)rrrr) r(r[r,rrrorfrrr r r!ris  z-OverlaySelectionDisplay._build_layer_callbackc KsPi}|dkrd|d<nd}|dur*||d<|r6||nd}|j|||fi|S)NrFZcolorbarrr)_build_element_layer) r(r[rr5rr6rfoptsrr r r!rss z-OverlaySelectionDisplay._apply_style_callbackcCs tdSrUr)r(r[Z layer_colorZ layer_alphar$r r r!r~sz,OverlaySelectionDisplay._build_element_layercCs tdSrUr)r(r7rr r r!rsz-OverlaySelectionDisplay._style_region_element)rFT)NN)T) rrrrr'rrr2rrrrrr r r r!r s  .   rc@s$eZdZdZd ddZd ddZdS) ColorListSelectionDisplayzd Selection display class for elements that support coloring by a vectorized color list. rr6NcCs|g|_|g|_||_dSrU)rZ alpha_propsbackend)r(rZ alpha_proprr r r!r'sz"ColorListSelectionDisplay.__init__c sL|dur i}fdd}|j|jg}|j||d|d}|D] }||}q:|S)Ncsddlm}|j}|dd}d}|p,d}||dddfdd ddD} t|} t|gt| } tj| d d } tt dt| |D](\} }|s| | dd<q| | | |<q| | fd d j D}| |j fjdd|S)Nrlinear_gradientrz#e6e9ec#000000csg|] }|pqSr r )rkc) backup_clrr r!rnrzzWColorListSelectionDisplay.build_selection.._build_selection..Zint8)Zdtypecsi|] }|qSr r r)r5r r!rrzzWColorListSelectionDisplay.build_selection.._build_selection..T)rre)rrrr}nparrayrzerosziprrrrrr)rlr5r6r,rfrrZselection_exprsrZselected_colorsryZclrsZ color_indsrrZ color_optsrm)rr5r!_build_selections"  zCColorListSelectionDisplay.build_selection.._build_selectionT)rrro)rrr) r(rr_rrrrorZ sel_streamsrr rmr!rs   z)ColorListSelectionDisplay.build_selection)rr6N)NN)rrrrr'rr r r r!rs rcCs8ddlm}|d|dd}||ddd}|||dS)z< Create a light to dark cmap list from a base color rrz#ffffffrrr@)rr)rrZ start_colorZ end_colorr r r!rs r)/ collectionsrnumpyrrZparam.parameterizedrZ core.datarZ core.elementrrZ core.layoutrZ core.optionsr r Z core.overlayr r Z core.spacesr rrrrrrrrutilrrr"r%ZdefinerZ_RegionElementrZParameterizedFunctionr8rrrrrrr r r r!s:     $ :`@ e0