jKhdZddlZddlZddlmZddlmZmZmZm Z m Z ddlZ e j j dGddZGd d ZGd d eZe j j d dddddGddee jZdZddZddZdZdS)aN The Colorizer class which handles the data to color pipeline via a normalization and a colormap. .. admonition:: Provisional status of colorizer The ``colorizer`` module and classes in this file are considered provisional and may change at any time without a deprecation period. .. seealso:: :doc:`/gallery/color/colormap_reference` for a list of builtin colormaps. :ref:`colormap-manipulation` for examples of how to make colormaps. :ref:`colormaps` for an in-depth discussion of choosing colormaps. :ref:`colormapnorms` for more details about data normalization. N)ma)_apicolorscbookartistscalezcolorizer : `~matplotlib.colorizer.Colorizer` or None, default: None The Colorizer object used to map color to data. If None, a Colorizer object is created from a *norm* and *cmap*.) colorizer_doccteZdZdZddZdZedZejdZdd Z e dd Z d Z d Z d ZedZejdZddZdZdZedZejdZedZejdZedZejdZdS) Colorizera Data to color pipeline. This pipeline is accessible via `.Colorizer.to_rgba` and executed via the `.Colorizer.norm` and `.Colorizer.cmap` attributes. Parameters ---------- cmap: colorbar.Colorbar or str or None, default: None The colormap used to color data. norm: colors.Normalize or str or None, default: None The normalization used to normalize the data Ncd|_||d|_d|_||_t jdg|_d|_dS)Nchangedsignals) _cmap _set_cmap_id_norm_normnormrCallbackRegistry callbackscolorbar)selfcmaprs BC:\Projects\DSCons\.venv\Lib\site-packages\matplotlib\colorizer.py__init__zColorizer.__init__5sR  t   / DDD c||?|||t|tjrt d||dS)a, Helper for initial scaling. Used by public functions that create a ScalarMappable and support parameters *vmin*, *vmax* and *norm*. This makes sure that a *norm* will take precedence over *vmin*, *vmax*. Note that this method does not set the norm. NzPassing a Normalize instance simultaneously with vmin/vmax is not supported. Please pass vmin/vmax as arguments to the norm object when creating it)set_clim isinstancer Normalize ValueErrorautoscale_None)rrvminvmaxAs r _scale_normzColorizer._scale_normAsp  t/ MM$ % % %$ 011 H GHHH Arc|jSN)rrs rrzColorizer.normW zrcBt||jj}||jurdS|jdu}|s$|jj|j||_|jjd|j |_|s| dSdS)N) n_componentsr ) _ensure_normr n_variatesrr disconnectrrconnectr )rrin_inits rrzColorizer.norm[sDty/CDDD 49   F)t# : I  * *4= 9 9 9  +33I48LBB   LLNNNNN  rFTct|tjr#|jdkr|||||St j|}|r||}||||}|S) Return a normalized RGBA array corresponding to *x*. In the normal case, *x* is a 1D or 2D sequence of scalars, and the corresponding `~numpy.ndarray` of RGBA values will be returned, based on the norm and colormap set for this Colorizer. There is one special case, for handling images that are already RGB or RGBA, such as might have been read from an image file. If *x* is an `~numpy.ndarray` with 3 dimensions, and the last dimension is either 3 or 4, then it will be treated as an RGB or RGBA array, and no mapping will be done. The array can be `~numpy.uint8`, or it can be floats with values in the 0-1 range; otherwise a ValueError will be raised. Any NaNs or masked elements will be set to 0 alpha. If the last dimension is 3, the *alpha* kwarg (defaulting to 1) will be used to fill in the transparency. If the last dimension is 4, the *alpha* kwarg is ignored; it does not replace the preexisting alpha. A ValueError will be raised if the third dimension is other than 3 or 4. In either case, if *bytes* is *False* (default), the RGBA array will be floats in the 0-1 range; if it is *True*, the returned RGBA array will be `~numpy.uint8` in the 0 to 255 range. If norm is False, no normalization of the input data is performed, and it is assumed to be in the range (0-1). )alphabytes) rnpndarrayndim_pass_image_datarasarrayrr)rxr5r6rrgbas rto_rgbazColorizer.to_rgbals> a $ $ @1((E5$?? ? JqMM   ! Ayy%uy55 rct|jddkr{|d}|jtjkrtj|dz}|jdd\}}tj||df|j}||ddddddf<||dddddf<n#|jddkr|}nt d|jjd krtjtj|x}rB|jddkr| }d |tj|d ddf<|r?| dks| d krt d |r"|dz tj}nQ|jtjkr%|s"| tj dz }nt d |jztj|r8d |tjtj|d df<|S)z Helper function to pass ndarray of shape (...,3) or (..., 4) through `to_rgba()`, see `to_rgba()` for docstring. r4N)shapedtypezThird dimension must be 3 or 4fr)axisz:Floating point image RGB values must be in the [0,1] rangez9Image RGB array must be uint8 or floating point; found %s)rDrEr7uint8emptyr!kindanyisnancopymaxminastypefloat32r is_masked getmaskarray)r<r5r6rmnxxnanss rr:zColorizer._pass_image_datas4 71:??}w"(""--72A2;DAqAq 999BBqqq!!!RaRxLBqqq!!!QwKK WQZ1__BB=>> > 8=C  vbhqkk)d** 071:??B./26$Q'''*+ ?AA ">??? 13h&&rx00 X ! ! 1YYrz**S08:<(CDD D 5??1   =; =s..)BBBc2|jj|jjfSzV Return the values (min, max) that are mapped to the colormap limits. )rr#r$r)s rget_climzColorizer.get_climsy~ty~--rcH|jdd|_dSz Call this whenever the mappable is changed to notify all the callbackSM listeners to the 'changed' signal. r TNrrlstaler)s rr zColorizer.changeds% y))) rc6|dS)Nrrpr)s rr#zColorizer.vmin#}}q!!rc2||dS)N)r#r)rr#s rr#zColorizer.vmin' 4      rc6|dSNrArvr)s rr$zColorizer.vmax+rwrc2||dS)N)r$ry)rr$s rr$zColorizer.vmax/rzrc|jjSr(rclipr)s rrzColorizer.clip3s y~rc||j_dSr(r)rrs rrzColorizer.clip7s rNNNFT)__name__ __module__ __qualname____doc__rr&propertyrsetterr> staticmethodr:r\r"rrrrpr r#r$rrrr r &s      ,X [[ ''''R(((\(T    $ $ $$X [[#3#3#3#3J... ""X" [!![!""X" [!![!X [[rr ceZdZdZdZddZdZddZfd Ze d Z e j d Z d Z d Z e dZej dZdZdZdZe dZej dZdZedZxZS)_ColorizerInterfaceaB Base class that contains the interface to `Colorizer` objects from a `ColorizingArtist` or `.cm.ScalarMappable`. Note: This class only contain functions that interface the .colorizer attribute. Other functions that as shared between `.ColorizingArtist` and `.cm.ScalarMappable` are not included. cJ|j||||jdSr() _colorizerr&_A)rrr#r$s rr&z_ColorizerInterface._scale_normEs& ##D$dg>>>>>rNFTc>|j||||S)r3)r5r6r)rr>)rr<r5r6rs rr>z_ColorizerInterface.to_rgbaHs#<&&qU&NNNrc4|jSro)rrpr)s rrpz_ColorizerInterface.get_climhs'')))rc<|j||dS)aO Set the norm limits for image scaling. Parameters ---------- vmin, vmax : float The limits. For scalar data, the limits may also be passed as a tuple (*vmin*, *vmax*) as a single positional argument. .. ACCEPTS: (vmin: float, vmax: float) N)rr)rr#r$s rrz_ColorizerInterface.set_climns"   t,,,,,rch tS#t$rYdSwxYwr|)super get_alphaAttributeError)r __class__s rrz_ColorizerInterface.get_alphas@ 77$$&& &   11 s # 11c|jjSr(rrr)s rrz_ColorizerInterface.cmap ##rc||j_dSr(rrhs rrz_ColorizerInterface.cmap#rc|jjS)z Return the `.Colormap` instance.rr)s rget_cmapz_ColorizerInterface.get_cmaprrc||_dS)raN)rrhs rset_cmapz_ColorizerInterface.set_cmaps rc|jjSr(rrr)s rrz_ColorizerInterface.normrrc||j_dSr(rrrs rrz_ColorizerInterface.normrrc||_dS)a_ Set the normalization instance. Parameters ---------- norm : `.Normalize` or str or None Notes ----- If there are any colorbars using the mappable for this norm, setting the norm of the mappable will reset the norm, locator, and formatters on the colorbar to default. Nrrs rset_normz_ColorizerInterface.set_norms rcD|j|jdS)rYN)rr\rr)s rr\z_ColorizerInterface.autoscales" !!$'*****rcD|j|jdS)r_N)rr"rr)s rr"z"_ColorizerInterface.autoscale_Nones" &&tw/////rc|jjS)zL The last colorbar associated with this object. May be None rrr)s rrz_ColorizerInterface.colorbars ''rc||j_dSr(r)rrs rrz_ColorizerInterface.colorbars#+   rctj|s|dStjt jrVjj}tjt j rjj jj f}n*djD}njg}|g}jj g}fdt|||D}dd |dS)Nz[]cg|] }|j Sr)N.0parts r zD_ColorizerInterface._format_cursor_data_override..s444$tv444rc Rg|]#\}}}|d|||d$S)z-#.g)_sig_digits_from_norm)rnodrUrs rrzD_ColorizerInterface._format_cursor_data_override..sR555"a?t11"a;;????555r[, ])r7rgetmaskrrr MultiNormnormsr BivarColormaprMzipjoin)rdatarn_soss` r_format_cursor_data_overridez0_ColorizerInterface._format_cursor_data_overrides 5==   $,4 di!1 2 2 IOE$)V%9:: 5y{DIK044$)444YKE6D9;-C5555!%s33555#499R==####rc||}tj|rEt|tjrxtjtj|j|z }td|dz }tj |j||dz}n|j |j krtj|j dz}np| tt||ztjddgz|z }t ||z }t!j||}nd}|S)NrrAr@g?r4)r7isfiniterr BoundaryNormargminabs boundariesrNdiffr#r$inverseintfloatarrayr _g_sig_digits) rrrUnormedcur_idx neigh_idxdelta neighbors g_sig_digitss rrz)_ColorizerInterface._sig_digits_from_norms@ d ;v   $ 344 4)BF4?T+A$B$BCC7Q;//  'A+0E FGGKKMMdi''ty2~..!LLv*++bh1v.>.>>!CEE I,--1133 .tU;;LLLrrr)rrrrr&r>rprrrrrrrrrr\r"rrrr __classcell__rs@rrr<s???OOOO@*** ----$ $$X$ [$$[$$$$$$X$ [$$[$ +++000((X( _,,_,$$$:\rrcheZdZdZd ddfd ZdZdZdZedZ ed Z xZ S) _ScalarMappablez A mixin class to map one or multiple sets of scalar data to RGBA. The ScalarMappable applies data normalization before returning RGBA colors from the given `~matplotlib.colors.Colormap`. N colorizerc tjdi|d|_|||||_d|_|jjd|j|_ tj dg|_dS)ar Parameters ---------- norm : `.Normalize` (or subclass thereof) or str or None The normalizing object which scales data, typically into the interval ``[0, 1]``. If a `str`, a `.Normalize` subclass is dynamically generated based on the scale with the corresponding name. If *None*, *norm* defaults to a *colors.Normalize* object which initializes its scaling based on the first data processed. cmap : str or `~matplotlib.colors.Colormap` The colormap used to map normalized data values to RGBA colors. N)rrrr rr) rrr_get_colorizerrrrr0r _id_colorizerrr)rrrrkwargsrs rrz_ScalarMappable.__init__s ""6"""-- SW-XX !_6>>y$,WW/ DDDrc| d|_dSt||jj}t j|d}t j|jtdsq|jj td|jd|jj D]@}t j||jtdstd|jdA||_|j s|j |dSdS)aB Set the value array from array-like *A*. Parameters ---------- A : array-like or None The values that are mapped to colors. The base class `.ScalarMappable` does not make any assumptions on the dimensionality and shape of the value array *A*. NTrM same_kindzImage data of dtype z cannot be converted to floatz, cannot be converted to a sequence of floats)r_ensure_multivariate_datarr,rsafe_masked_invalidr7can_castrErfieldsr[scaledrr")rr%keys r set_arrayz_ScalarMappable.set_array2s; 9DG F %a)? @ @  %ad 3 3 3{17E;77 Nw~%!6qw!6!6!67777>NNC;qv|UKHHN')Mqw)M)M)MNNNNy!! . O * *1 - - - - - . .rc|jS)z Return the array of values, that are mapped to colors. The base class `.ScalarMappable` does not make any assumptions on the dimensionality and shape of the array. )rr)s r get_arrayz_ScalarMappable.get_arraySs wrcJ|jd|d|_dSrrrsr)s rr z_ScalarMappable.changed\s' y$/// rc |ltd|DrCtddd|DzdSdS)zZ Raises a ValueError if any kwarg is not None while colorizer is not None Ncg|]}|duSr(r)rvals rrz@_ScalarMappable._check_exclusionary_keywords..js???CtO???rzZThe `colorizer` keyword cannot be used simultaneously with any of the following keywords: rc3"K|] }d|dV dS)`Nr)rrs r z?_ScalarMappable._check_exclusionary_keywords..ms*,Q,QCZZZZ,Q,Q,Q,Q,Q,Qr)rKvaluesr!rkeys)rrs r_check_exclusionary_keywordsz,_ScalarMappable._check_exclusionary_keywordsds  ??v}}???@@ S "I#'99,Q,Q6;;==,Q,Q,Q#Q#Q"RSSS !  S Srct|tr$tt|||St||S)N)rr)rr rr)rrrs rrz_ScalarMappable._get_colorizerosP i + +   8 84 9    t$$$rr) rrrrrrrr rrrrrs@rrrs*E$EEEEEEE,...BSS\S%%\%%%%%rrzcmap : str or `~matplotlib.colors.Colormap`, default: :rc:`image.cmap` The Colormap instance or registered colormap name used to map scalar data to colors.acmap : str, `~matplotlib.colors.Colormap`, `~matplotlib.colors.BivarColormap` or `~matplotlib.colors.MultivarColormap`, default: :rc:`image.cmap` The Colormap instance or registered colormap name used to map data values to colors. Multivariate data is only accepted if a multivariate colormap (`~matplotlib.colors.BivarColormap` or `~matplotlib.colors.MultivarColormap`) is used.anorm : str or `~matplotlib.colors.Normalize`, optional The normalization method used to scale scalar data to the [0, 1] range before mapping to colors using *cmap*. By default, a linear scaling is used, mapping the lowest value to 0 and the highest to 1. If given, this can be one of the following: - An instance of `.Normalize` or one of its subclasses (see :ref:`colormapnorms`). - A scale name, i.e. one of "linear", "log", "symlog", "logit", etc. For a list of available scales, call `matplotlib.scale.get_scale_names()`. In that case, a suitable `.Normalize` subclass is dynamically generated and instantiated.atnorm : str, `~matplotlib.colors.Normalize` or list, optional The normalization method used to scale data to the [0, 1] range before mapping to colors using *cmap*. By default, a linear scaling is used, mapping the lowest value to 0 and the highest to 1. This can be one of the following: - An instance of `.Normalize` or one of its subclasses (see :ref:`colormapnorms`). - A scale name, i.e. one of "linear", "log", "symlog", "logit", etc. For a list of available scales, call `matplotlib.scale.get_scale_names()`. In this case, a suitable `.Normalize` subclass is dynamically generated and instantiated. - A list of scale names or `.Normalize` objects matching the number of variates in the colormap, for use with `~matplotlib.colors.BivarColormap` or `~matplotlib.colors.MultivarColormap`, i.e. ``["linear", "log"]``.avmin, vmax : float, optional When using scalar data and no explicit *norm*, *vmin* and *vmax* define the data range that the colormap covers. By default, the colormap covers the complete value range of the supplied data. It is an error to use *vmin*/*vmax* when a *norm* instance is given (but using a `str` *norm* name together with *vmin*/*vmax* is acceptable).aHvmin, vmax : float or list, optional When using scalar data and no explicit *norm*, *vmin* and *vmax* define the data range that the colormap covers. By default, the colormap covers the complete value range of the supplied data. It is an error to use *vmin*/*vmax* when a *norm* instance is given (but using a `str` *norm* name together with *vmin*/*vmax* is acceptable). A list of values (vmin or vmax) can be used to define independent limits for each variate when using a `~matplotlib.colors.BivarColormap` or `~matplotlib.colors.MultivarColormap`.)cmap_docmulti_cmap_docnorm_docmulti_norm_doc vmin_vmax_docmulti_vmin_vmax_docc^eZdZdZfdZedZejdZdZxZ S)ColorizingArtistz Base class for artists that make map data to color using a `.colorizer.Colorizer`. The `.colorizer.Colorizer` applies data normalization before returning RGBA colors from a `~matplotlib.colors.Colormap`. c ttjt|tjdd|i|dS)zZ Parameters ---------- colorizer : `.colorizer.Colorizer` rrNr)rcheck_isinstancer rr)rrrrs rrzColorizingArtist.__init__sC i9====779777777rc|jSr()rr)s rrzColorizingArtist.colorizers rctjt||jj|j||_|jd|j|_dS)Nrr ) rrr rrr/rr0r )rcls rrzColorizingArtist.colorizers] i26666 !,,T-?@@@\11)T\JJrc 0|j|fi|||_dS)z[ Raises a ValueError if any kwarg is not None while colorizer is not None. N)rr)rrrs r_set_colorizer_check_keywordsz.ColorizingArtist._set_colorizer_check_keywordss* *))>>v>>>"r) rrrrrrrrr rrs@rrrs88888XKKK #######rrc( tjtj|dtj}n?#t $r2tj|tj}YnwxYwt |S)a Automatically generate a norm class from *scale_cls*. This differs from `.colors.make_norm_from_scale` in the following points: - This function is not a class decorator, but directly returns a norm class (as if decorating `.Normalize`). - The scale is automatically constructed with ``nonpositive="mask"``, if it supports such a parameter, to work around the difference in defaults between standard scales (which use "clip") and norms (which use "mask"). Note that ``make_norm_from_scale`` caches the generated norm classes (not the instances) and reuses them for later calls. For example, ``type(_auto_norm_from_scale("log")) == LogNorm``. mask) nonpositive)rmake_norm_from_scale functoolspartialr r[rd) scale_clsrs r_auto_norm_from_scalers$ >v*  iV < < <>>        5v*955      ::sAA9BBrAc|dkrtjtjtdf||tj}nLt |tr7tjtj |}t|S|S|dkrtj |s(tjtj dtf||tj dg|z}ntj |}t |tj r |j|kr|St!d|dt!d)NrArlinearz,Invalid norm for multivariate colormap with z inputszHInvalid cmap. A colorizer object must have a cmap with `n_variates` >= 1)rrrNormstrr rgetitem_checkedr_scale_mappingrr7iterablertupler,r!)rr,rs rr-r-sWq v{C6TBBBB <#%%DD c " " 6,U-AMMMI3(3355 5   {4   N  !6#3T5"A M M M M <#XJ|$;< >>rct|tjr4t|jj|kr|S|jtjtjfvr|dkrtd|d|jtjkrtjd}ntjd}tj tj | |}tj |r/|jD]'}tj j||d|j<(|S|dkr;t||kr'd|D}tjd d |D}d |jD}tj |dj| }t'|D]\}}||j|jkr+td |djd||j||||<tj ||r%tj j||||j<|S|dkr|S|dkrtdtd|d|d)a Ensure that the data has dtype with n_components. Input data of shape (n_components, n, m) is converted to an array of shape (n, m) with data type np.dtype(f'{data.dtype}, ' * n_components) Complex data is returned as a view with dtype np.dtype('float64, float64') or np.dtype('float32, float32') If n_components is 1 and data is not of type np.ndarray (i.e. PIL.Image), the data is returned unchanged. If data is None, the function returns None Parameters ---------- n_components : int Number of variates in the data. data : np.ndarray, PIL.Image or None Returns ------- np.ndarray, PIL.Image or None r@zPInvalid data entry for multivariate data. Complex numbers are incompatible with z variates.zfloat64, float64zfloat32, float32rrAcPg|]#}tj|d$S)Fr)r7rrrs rrz-_ensure_multivariate_data..ts*???$ Du --???rrcg|] }|j SrrErs rrz-_ensure_multivariate_data..us B B BTDJ B B Brcg|] }|d S)rr)r descriptors rrz-_ensure_multivariate_data..vs;;;J*Q-;;;rr*z=For multivariate data all variates must have same shape, not z and zInvalid data entry for multivariate data. The data must contain complex numbers, or have a first dimension 2, or be of a dtype with 2 fieldsz\Invalid data entry for multivariate data. The shape of the data must have a first dimension z or be of a dtype with z fields)rr7r8lenrEdescr complex64 complex128r!rrgetdataviewrRmaskedr rrIrD enumerate)rr,dt reconstructedr,rirFs rrrDs,$ ##! tz L 0 0K ZBL"-8 8 8q   "=$0"="="=>>> zR]**X011X011EKK d(;(;(@(@(D(DEEMut$$ K"$(KKJ>@elM*Q-0;; aCII55@?$??? Xdii B BT B B BCC D D;;"(;;; DGM << f%% > >DAqAw} 333 "S/3Aw}"S"SCG7="S"STTT#AwM! utAw'' >13 a a.q   ;<< <IEQII3?IIIJJ Jr)rA)F)rrnumpyr7r matplotlibrrrrrr _docstringinterpdregisterr rrArtistrrr-rcrrrrr>s*999999999999993 SSSSSSSSlGGGGGGGGTp%p%p%p%p%)p%p%p%f O8 .e ===@!#!#!#!#!# !#!#!#H8XXXX2/>/>/>/>dHJHJHJHJHJr