a ~bG@sddlmZddlZddlZddlmZmZmZddl m Z m Z m Z ddl mZmZddlmZmZmZddlmZmZdd lmZdd lmZGd d d ee eZGd ddeZGdddeZGdddeZGdddeZe e!dde"#DZ$dS))groupbyN) DimensionedViewableElementasdim) ComposableLayoutNdLayout) OrderedDict NdMapping) Overlayable NdOverlayCompositeOverlay)HoloMap GridSpace)AttrTree)get_param_valuesc@seZdZdZejdddZdZdZddZ d&d d Z d d Z d dZ ddZ ddZe Zed'ddZddZgddfddZgddfddZddZd(ddZd)d d!Zd*d"d#Zd+d$d%ZdS),Elementat Element is the atomic datastructure used to wrap some data with an associated visual representation, e.g. an element may represent a set of points, an image or a curve. Elements provide a common API for interacting with data of different types and define how the data map to a set of dimensions and how those map to the visual representation. T)defaultZconstantcKsdS)N)NNNr)selfkwargsrr5lib/python3.9/site-packages/holoviews/core/element.py$_get_selection_expr_for_stream_valuesz,Element._get_selection_expr_for_stream_valueNc Ksddlm}t|ts|g}g}|dddD]&}||f|||d|} || q.|r~|} tt|D]} | || >} qjnt|dkrt|} n|d} | S)aComputes and adjoins histogram along specified dimension(s). Defaults to first value dimension if present otherwise falls back to first key dimension. Args: dimension: Dimension(s) to compute histogram on num_bins (int, optional): Number of bins bin_range (tuple optional): Lower and upper bounds of bins adjoin (bool, optional): Whether to adjoin histogram Returns: AdjointLayout of element and histogram or just the histogram ) histogramN)num_bins bin_range dimensionrr)Z operationr isinstancelistappendrangelenr) rr rrZadjoinrrZhistsdhistlayoutZdidxrrrr'"s$     z Element.histcCs"|dkr |Stdt|jdS)Nrz%%s currently does not support getitem)NotImplementedErrortype__name__)rkeyrrr __getitem__Is zElement.__getitem__cCsdS)zIndicates whether the element is empty. Subclasses may override this to signal that the Element contains no data and can safely be dropped during indexing. Trrrrr __nonzero__PszElement.__nonzero__cCs ||vS)z&Whether element contains the Dimension) dimensions)rr rrr __contains__YszElement.__contains__cCs tddS)zDisable iterator interface.z'Iteration on Elements is not supported.Nr)r.rrr__iter__^szElement.__iter__cKstd|jdS)z Deprecated method to perform collapse operations, which may now be performed through concatenation and aggregation. z"Collapsing not implemented for %s.N)r)r+)clsdatafunctionkdimsrrrr collapse_datafszElement.collapse_datacKstdS)aYSnap list or dict of coordinates to closest position. Args: coords: List of 1D or 2D coordinates **kwargs: Coordinates specified as keyword pairs Returns: List of tuples of the snapped coordinates Raises: NotImplementedError: Raised if snapping is not supported Nr2)rZcoordsrrrrclosestos zElement.closestFcKstdS)aSamples values at supplied coordinates. Allows sampling of element with a list of coordinates matching the key dimensions, returning a new object containing just the selected samples. Supports multiple signatures: Sampling with a list of coordinates, e.g.: ds.sample([(0, 0), (0.1, 0.2), ...]) Sampling a range or grid of coordinates, e.g.: 1D: ds.sample(3) 2D: ds.sample((3, 3)) Sampling by keyword, e.g.: ds.sample(x=0) Args: samples: List of nd-coordinates to sample bounds: Bounds of the region to sample Defined as two-tuple for 1D sampling and four-tuple for 2D sampling. closest: Whether to snap to closest coordinates **kwargs: Coordinates specified as keyword pairs Keywords of dimensions and scalar coordinates Returns: Element containing the sampled coordinates Nr2)rZsamplesZboundsr9Z sample_valuesrrrsamples zElement.samplecKstdS)aApplies reduction along the specified dimension(s). Allows reducing the values along one or more key dimension with the supplied function. Supports two signatures: Reducing with a list of dimensions, e.g.: ds.reduce(['x'], np.mean) Defining a reduction using keywords, e.g.: ds.reduce(x=np.mean) Args: dimensions: Dimension(s) to apply reduction on Defaults to all key dimensions function: Reduction operation to apply, e.g. numpy.mean spreadfn: Secondary reduction to compute value spread Useful for computing a confidence interval, spread, or standard deviation. **reductions: Keyword argument defining reduction Allows reduction to be defined as keyword pair of dimension and function Returns: The element after reductions have been applied. Nr2)rr0r6ZspreadfnZ reductionrrrreduceszElement.reducecs|r|rtdtt|dkr,td|r8|}|rPfdd|D}n|shfddjD}fdd|D}ddt|d d D}|d S) NzPPass reduced dimensions either as an argument or as part of the kwargs not both.rzFCannot define reduce operations with more than one function at a time.csg|] }|fqSrr.0r&r6rr z'Element._reduce_map..csg|] }|fqSrrr<r>rrr?r@cs$g|]\}}j|ddj|fqST)strict get_dimensionname)r=r&fnr.rrr?scSs"g|]\}}|dd|DfqS)cSsg|] \}}|qSrr)r=dim_rrrr?r@z2Element._reduce_map...r)r=rFgrprrrr?r@cSs|dS)Nrr)xrrrr@z%Element._reduce_map..r) Exceptionr%setvaluesitemsr7r)rr0r6Z reduce_mapZreducedZgroupedr)r6rr _reduce_maps zElement._reduce_mapcszddl}|dur$ddD}nfdd|D}|}tfdd|D}||}|rv|fdd|D}|S)aConvert dimension values to DataFrame. Returns a pandas dataframe of columns along each dimension, either completely flat or indexed by key dimensions. Args: dimensions: Dimensions to return as columns multi_index: Convert key dimensions to (multi-)index Returns: DataFrame of columns corresponding to each dimension rNcSsg|] }|jqSr)rEr<rrrr?r@z"Element.dframe..csg|]}j|ddjqSrArCr<r.rrr?r@csg|]}||fqSr)dimension_valuesr=rGr.rrr?r@csg|]}|jvr|qSrr7r<r.rrr?r@)Zpandasr0r Z DataFrameZ set_index)rr0Z multi_indexpdZ column_namesdim_valsZdfrr.rdframes  zElement.dframecs|dur ddjjD}nfdd|D}gg}}|D]&}|}||||jjq@tt|dkrdd|D}t |S)zConvert dimension values to columnar array. Args: dimensions: List of dimensions to return Returns: Array of columns corresponding to each dimension NcSsg|]}|qSrrr<rrrr?r@z!Element.array..csg|]}j|ddqSrA)rDr<r.rrr?r@rcSsg|]}|dqS)object)Zastype)r=crrrr?r@) r7vdimsrQr#Zdtypekindr%rMnpZ column_stack)rr0dimscolumnstypesrGcolumnrr.rarrays    z Element.arraycCsT|jdt|j|r*t|ts*|g}ddlm}||fi|rLt|dniS)z4Deprecated method to convert any Element to a Table.zgThe table method is deprecated and should no longer be used. Instead cast the %s to a a Table directly.r)Table)datatype) paramwarningr*r+r!r"elementradict)rrbrarrrtable s z Element.tablec sjdt}|sj}|r.rcsg|]}|jqSrrhrRr.rrr?(s)rcrdr%r7ziprYr )rr7rYrlengthkeysrNrr.rmappings"        zElement.mapping)NrNT)NN)NF)N)N)NN)r+ __module__ __qualname____doc__rcStringgroupZ_Element__abstract_selection_streamsrr'r-r/r1r3__bool__ classmethodr8r9r:r;rPrVr`rgrlrrrrrs.  '  #   rc@s<eZdZdZdZeddZeddZddZd d Z d S) Tabularzp Baseclass to give an elements providing an API to generate a tabular representation of the object. TcCs t|dS)z*Number of rows in table (including header)r)r%r.rrrrows8sz Tabular.rowscCs t|S)zNumber of columns in table)r%r0r.rrrcols=sz Tabular.colscCs|j}||jkr$td|jdnv||jkrBtd|jdnX|dkrx||krl|jrh|j||jSdS|j|jS||}||j |d|fSdS)zFormatted contents of table cell. Args: row (int): Integer index of table row col (int): Integer index of table column Returns: Formatted table cell contents zMaximum column index is %drzMaximum row index is %drN) ndimsrwrLrvrYZ pprint_labelr7rDZ pprint_valueZiloc)rrowcolryrGrrr pprint_cellCs     zTabular.pprint_cellcCs|dkr dSdS)aType of the table cell, either 'data' or 'heading' Args: row (int): Integer index of table row col (int): Integer index of table column Returns: Type of the table cell, either 'data' or 'heading' rheadingr5r)rrzr{rrr cell_type^s zTabular.cell_typeN) r+rmrnroZ_Tabular__abstractpropertyrvrwr|r~rrrrru0s  ruc@seZdZejdddZdZdS) Element2D)NNNNz Allows overriding the extents of the Element in 2D space defined as four-tuple defining the (left, bottom, right and top) edges.rdocTN)r+rmrnrcTupleextentsZ_Element2D__abstractrrrrrlsrc@s"eZdZejdddZdZdZdS) Element3D)NNNNNNzz Allows overriding the extents of the Element in 3D space defined as (xmin, ymin, zmin, xmax, ymax, zmax).rTrN)r+rmrnrcrrZ_Element3D__abstractrrrrrrrusrcseZdZdZejgddZejdddZejgddZ ej ddZ ej dd dZ ejeeddd Zejdd dZejgd dZdZdZeeeeeefeeeefeeiZdfd d ZddZe ddZ!ddZ"Z#S)Collatoral Collator is an NdMapping type which can merge any number of HoloViews components with whatever level of nesting by inserting the Collators key dimensions on the HoloMaps. If the items in the Collator do not contain HoloMaps they will be created. Collator also supports filtering of Tree structures and dropping of constant dimensions. zV List of dimensions to drop when collating data, specified as strings.rFzY Whether to demote any non-varying key dimensions to constant dimensions.z[ List of paths to drop when collating data, specified as strings or tuples.)rNzk The progress bar instance used to report progress. Set to None to disable progress bars.)class_rZ is_instanceZ instantiatea If supplied the function will be applied on each Collator value during collation. This may be used to apply an operation to the data or load references from disk before they are collated into a displayable HoloViews object.z Collator operates on HoloViews objects, if vdims are specified a value_transform function must also be supplied.c sft|trJtt|fi|}d|vr0|j|d<d|vrB|j|d<|}tt|j |fi|dS)Nr7rY) r!rrfrr7rYrlsuperr__init__)rr5params __class__rrrs   zCollator.__init__c s>jtjd}t}tjD]\}\}}t|trJ| j }tj r|j r|t tjddd|} |}t|tstdtj|}fdd|D}fdd|D} |s| r܈||t | }|||<jd ur(t|d |d q(|} |j| d j} | D]} | | q(| S) a" Filter each Layout in the Collator with the supplied path_filters. If merge is set to True all Layouts are merged, otherwise an NdMapping containing all the Layouts is returned. Optionally a list of dimensions to be ignored can be supplied. rSvalueT)Zlabelz=Collator values must be Dimensioned objects before collation.cs0g|](\}}jr$|vr|jvr||fqSr) drop_constantdropr=r&kZ constant_dimsrrrr?sz%Collator.__call__..cs0g|](\}}|vr|jvrjr||fqSr)rrrrrrr?sNrdr)static_dimensionsr r7r% enumerater5rOr!rfilterfiltersrYvalue_transformrfrir0r ValueError_add_dimensions progress_barfloatrNZlastcloneupdate) r ndmappingZ num_elementsidxr,r5ZvargsZdim_keysZ varying_keys constant_keys componentsZ accumulatorZ componentrrr__call__s4       zCollator.__call__cCs6g}|jD]&}tt||jdkr ||q |S)z1 Return all constant dimensions. r)r7r%rMrQrEr#)rr0rGrrrrs   zCollator.static_dimensionsc s t|trd|_fdd|dddD}t|jrt|j|d}|D](\}}t|}||jvrH||d|}qHnt|jjrt |rt |\}} j| |it ||d}q|}n<|jd|d }| D]$\} } | |ddd||| <qt|trd |_|S) z Recursively descend through an Layout and NdMapping objects in order to add the supplied dimension values to all contained HoloMaps. Fcs"g|]\}}|jvr||fqSr)r)r=rGvalr.rrr?s  z,Collator._add_dimensions..Nr)cdimsr)r7r)Z shared_datarT)r!rfixed merge_typerrr7Z add_dimension _nest_orderr%rir"rOr) ritemr\rrUZnew_itemrGrr0r,rvrr.rrs,       zCollator._add_dimensions)N)$r+rmrnrorcListrZBooleanrrrprqZ ParameterrZ ClassSelectorr rrCallablerrYZ_deep_indexableZ_auxiliary_componentrrrr r rrrrrrr __classcell__rrrrrs.     ) rcCs(g|] \}}t|trt|tr|qSr)r!r* issubclassr)r=Z_kZ_vrrrr?s r?)% itertoolsrZnumpyr[rcr rrrr(rrr rr r Zoverlayr r rspacesrrZtreerutilrrrurrrr"rMlocalsrO__all__rrrrs$   #<