a Df@sddlmZddlmZddlZddlZddlZddlmZddl Z ddl Z ddl ZddlmZmZddlmZzddlZddlZWneydZdZYn0GdddeZdS) ) annotations)VersionN)isnan)Exprngjit)expand_varargsc@sxeZdZdZdZeddZeddZeddZ ee d d Z ee d d Z ed dZ ddZeddZdS)GlyphzBase class for glyphs.FcCs tdS)a The number of dimensions required in the data structure this Glyph is constructed from. Or None if input data structure is irregular For example * ndims is 1 if glyph is constructed from a DataFrame * ndims is 2 if glyph is constructed from a 2D xarray DataArray * ndims is None if glyph is constructed from multiple DataFrames of different lengths N)NotImplementedError)selfr d/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/datashader/glyphs/glyph.pyndimss z Glyph.ndimscCsH|\}}t|rt|s&d\}}n||kr@|d|d}}||fS)N)gg?)npisfinite)Zboundsminvalmaxvalr r r maybe_expand_bounds*s  zGlyph.maybe_expand_boundscCstr(t|tjr(|}||fSt|tjr@t|j St|t j rt rvt|j t jrv||fSt|j Sn t|SdSN)cudf isinstanceZSeriesZ nans_to_nullsminmaxpdr_compute_bounds_numbavaluesxrZ DataArraycpdataZndarrayitemZravel)sr r r _compute_bounds3s   zGlyph._compute_boundscCs@tj}tj }|D]$}t|s||kr*|}||kr|}q||fSr)rinfr)Zarrrrxr r r rBszGlyph._compute_bounds_numbacCsltj}tj }t|jdD]F}t|jdD]2}|||}t|s.||krT|}||kr.|}q.q||fS)Nrr)rr"rangeshaper)valsrrijvr r r _compute_bounds_2dPs  zGlyph._compute_bounds_2dcst|trt|}t|trLt|tt|krLtjfdd|DddSttj tdkrj| St|ts| S| SdS)Ncsg|]}t|qSr )rarray).0cdfr r iz'Glyph.to_cupy_array..r)Zaxisz22.02) rtuplelistlensetrstackrr __version__Zto_cupyZ to_gpu_arrayZ as_gpu_matrix)r/columnsr r.r to_cupy_array`s    zGlyph.to_cupy_arraycCs|||j|jS)ai Create a decorator that can be used on functions that accept *aggs_and_cols as a variable length argument. The decorator will replace *aggs_and_cols with a fixed number of arguments. The appropriate fixed number of arguments is calculated from the input append function. Rationale: When we know the fixed length of a variable length argument, replacing it with fixed arguments can help numba better optimize the the function. If this ever causes problems in the future, this decorator can be safely removed without changing the functionality of the decorated function. Parameters ---------- append: function The append function for the current aggregator Returns ------- function Decorator function )_expand_aggs_and_colsr antialiased)r appendr r r expand_aggs_and_colsrszGlyph.expand_aggs_and_colsc CstjddrddStPtdzt|jj }Wn"t t fy^t|j }Yn0Wdn1st0Yt |}d}|pd}|||}|r|d8}t |S)NZNUMBA_DISABLE_JITcSs|Srr )fnr r r r1z-Glyph._expand_aggs_and_cols..ignorer)osenvirongetwarningscatch_warnings simplefilterinspectgetfullargspecZpy_funcargs TypeErrorAttributeErrorr4r)r<r r;Z append_argsZ append_arglenZ xy_arglenZ dim_arglenZaggs_and_cols_lenr r r r:s  0 zGlyph._expand_aggs_and_colsN)__name__ __module__ __qualname____doc__r;propertyr staticmethodrr!rrr*r9r=r:r r r r rs&      r) __future__rZpackaging.versionrrHrErBmathrnumpyrZpandasrZxarrayrZdatashader.utilsrrZdatashader.macrosrrZcupyr Exceptionrr r r r s"