OwlCyberSecurity - MANAGER
Edit File: cm.cpython-38.pyc
U ��]�3 � @ s� d Z ddlZddlZddlmZ ddlZddlmZ ddl m Z ddl m Z ddl mZ i Zddd�Zdd � Zd d � Zd d � Zejd Ze �dd� e �� D �� e D ]Zeee�ee< q�e�e� e� �e� ddd�Zddd�Z G dd� de �Z dS )a� Builtin colormaps, colormap handling utilities, and the `ScalarMappable` mixin. .. seealso:: :doc:`/gallery/color/colormap_reference` for a list of builtin colormaps. :doc:`/tutorials/colors/colormap-manipulation` for examples of how to make colormaps and :doc:`/tutorials/colors/colormaps` an in-depth discussion of choosing colormaps. :doc:`/tutorials/colors/colormapnorms` for more details about data normalization � N)�ma)�datad)�cmapsc C s |dkrt �t| �S | d| �S )z1Helper such that ``_reverser(f)(x) == f(1 - x)``.N� )� functools�partial� _reverser)�f�x� r �//usr/lib/python3/dist-packages/matplotlib/cm.pyr ( s r c C sF i }| � � D ]4\}}t|�r&t|�}ndd� t|�D �}|||< q |S )z:Can only handle specification *data* in dictionary format.c S s g | ]\}}}d | ||f�qS �� �?r )�.0r Zy0Zy1r r r � <listcomp>< s zrevcmap.<locals>.<listcomp>)�items�callabler �reversed)�dataZdata_r�key�valZvalnewr r r �revcmap0 s r c C s` d| kr d| d ddd� iS d| kr.t | �S tt| ��}t|d �dkrXdd� |D �}|S dS ) zRReverses cmap specification *spec*, can handle both dict and tuple type specs.�listedN����redr � c S s g | ]\}}d | |f�qS r r )r �a�br r r r M s z&_reverse_cmap_spec.<locals>.<listcomp>)r �listr �len)�specZrevspecr r r �_reverse_cmap_specA s r! c C sJ t | }d|kr t�| ||�S d|kr6t�|d | �S tj�| ||�S dS )zMGenerates the requested cmap from its *name*. The lut size is *lutsize*.r r N)r �colors�LinearSegmentedColormapZListedColormap� from_list)�nameZlutsizer r r r �_generate_cmapQ s r& � image.lutc C s i | ]\}}|d t |��qS )Z_r)r! )r �cmapnamer r r r � <dictcomp>d s � r) c C s� | dkr0z |j } W n tk r. td��Y nX t| t�sBtd��t|tj�rZ|t| < dS |dkrltj d }t� | ||�}|t| < dS )aj Add a colormap to the set recognized by :func:`get_cmap`. It can be used in two ways:: register_cmap(name='swirly', cmap=swirly_cmap) register_cmap(name='choppy', data=choppydata, lut=128) In the first case, *cmap* must be a :class:`matplotlib.colors.Colormap` instance. The *name* is optional; if absent, the name will be the :attr:`~matplotlib.colors.Colormap.name` attribute of the *cmap*. In the second case, the three arguments are passed to the :class:`~matplotlib.colors.LinearSegmentedColormap` initializer, and the resulting colormap is registered. Nz+Arguments must include a name or a Colormapz Colormap name must be a stringr' ) r% �AttributeError� ValueError� isinstance�strr" �Colormap�cmap_d�mpl�rcParamsr# )r% �cmapr �lutr r r � register_cmapt s r4 c C sh | dkrt jd } t| tj�r"| S | tkrJ|dkr:t| S t| �|�S ntd| d�t t��f ��dS )a� Get a colormap instance, defaulting to rc values if *name* is None. Colormaps added with :func:`register_cmap` take precedence over built-in colormaps. If *name* is a :class:`matplotlib.colors.Colormap` instance, it will be returned. If *lut* is not None it must be an integer giving the number of entries desired in the lookup table, and *name* must be a standard mpl colormap name. Nz image.cmapz6Colormap %s is not recognized. Possible values are: %sz, ) r0 r1 r, r" r. r/ Z _resampler+ �join�sorted)r% r3 r r r �get_cmap� s ��r7 c @ s� e Zd ZdZd#dd�Zd$dd�Zd d � Zd d � Zd d� Zdd� Z d%dd�Z dd� Z dd� Z dd� Z dd� Zdd � Zd d � Zd d � Zd!d"� ZdS )&�ScalarMappablez� This is a mixin class to support scalar data to RGBA mapping. The ScalarMappable makes use of data normalization before returning RGBA colors from the given colormap. Nc C sR t �� | _|dkrt� }|dkr(t�� }d| _|| _t|�| _d| _ ddi| _ dS )a� Parameters ---------- norm : :class:`matplotlib.colors.Normalize` instance The normalizing object which scales data, typically into the interval ``[0, 1]``. If *None*, *norm* defaults to a *colors.Normalize* object which initializes its scaling based on the first data processed. cmap : str or :class:`~matplotlib.colors.Colormap` instance The colormap used to map normalized data values to RGBA colors. N�arrayF) �cbookZCallbackRegistry� callbacksSMr7 r" � Normalize�_A�normr2 Zcolorbar� update_dict)�selfr> r2 r r r �__init__� s zScalarMappable.__init__FTc C s� �z>|j dk�r>|jd dkr�|dkr*d}|jtjkrDt�|d �}|jdd� \}}tj||df|jd�}||dd�dd�dd�f< ||dd�dd�df< n |jd dkr�|}ntd��|jjd k�r|r�|�� dks�|� � d k r�td ��|�r8|d � tj�}n4|jtjk�r*|�s8|� tj �d }ntd |j ��|W S W n t k �rV Y nX t �|�}|�rr| �|�}| j|||d �}|S )a� Return a normalized rgba array corresponding to *x*. In the normal case, *x* is a 1-D or 2-D sequence of scalars, and the corresponding ndarray of rgba values will be returned, based on the norm and colormap set for this ScalarMappable. 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 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 uint8, or it can be floating point with values in the 0-1 range; otherwise a ValueError will be raised. If it is a masked array, the mask will be ignored. 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 pre-existing 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 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). � r Nr � � )�shape�dtypez third dimension must be 3 or 4r r z:Floating point image RGB values must be in the 0..1 range.z9Image RGB array must be uint8 or floating point; found %s)�alpha�bytes)�ndimrE rF �npZuint8�emptyr+ Zkind�max�minZastypeZfloat32r* r Zasarrayr> r2 ) r@ r rG rH r> �m�nZxxZrgbar r r �to_rgba� sB � zScalarMappable.to_rgbac C s || _ d| jd< dS )zmSet the image array from numpy array *A*. Parameters ---------- A : ndarray Tr9 N)r= r? )r@ �Ar r r � set_array% s zScalarMappable.set_arrayc C s | j S )zReturn the array)r= �r@ r r r � get_array/ s zScalarMappable.get_arrayc C s | j S )zreturn the colormap)r2 rS r r r r7 3 s zScalarMappable.get_cmapc C s | j j| j jfS )z9return the min, max of the color limits for image scaling)r> �vmin�vmaxrS r r r �get_clim7 s zScalarMappable.get_climc C sf |dkr.z |\}}W n t tfk r, Y nX |dk rDt�|�| j_|dk rZt�|�| j_| �� dS )a& set the norm limits for image scaling; if *vmin* is a length2 sequence, interpret it as ``(vmin, vmax)`` which is used to support setp ACCEPTS: a length 2 sequence of floats; may be overridden in methods that have ``vmin`` and ``vmax`` kwargs. N)� TypeErrorr+ r" Z_sanitize_extremar> rU rV �changed)r@ rU rV r r r �set_clim; s zScalarMappable.set_climc C s dS )z] Returns ------- alpha : float Always returns 1. r r rS r r r � get_alphaO s zScalarMappable.get_alphac C s t |�}|| _| �� dS )z� set the colormap for luminance data Parameters ---------- cmap : colormap or registered colormap name N)r7 r2 rY )r@ r2 r r r �set_cmapY s zScalarMappable.set_cmapc C s" |dkrt �� }|| _| �� dS )aH Set the normalization instance. Parameters ---------- norm : `.Normalize` 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. N)r" r< r> rY )r@ r> r r r �set_norme s zScalarMappable.set_normc C s, | j dkrtd��| j�| j � | �� dS )zb Autoscale the scalar limits on the norm instance using the current array N�%You must first set_array for mappable)r= rX r> � autoscalerY rS r r r r_ x s zScalarMappable.autoscalec C s, | j dkrtd��| j�| j � | �� dS )z� Autoscale the scalar limits on the norm instance using the current array, changing only limits that are None Nr^ )r= rX r> �autoscale_NonerY rS r r r r` � s z ScalarMappable.autoscale_Nonec C s d| j |< dS )zz Add an entry to a dictionary of boolean flags that are set to True when the mappable is changed. FN�r? �r@ Zcheckerr r r � add_checker� s zScalarMappable.add_checkerc C s | j | rd| j |< dS dS )zf If mappable has changed since the last check, return True; else return False FTra rb r r r � check_update� s zScalarMappable.check_updatec C s. | j �d| � | jD ]}d| j|< qd| _dS )z� Call this whenever the mappable is changed to notify all the callbackSM listeners to the 'changed' signal rY TN)r; Zprocessr? Zstale)r@ r r r r rY � s zScalarMappable.changed)NN)NFT)NN)�__name__� __module__� __qualname__�__doc__rA rP rR rT r7 rW rZ r[ r\ r] r_ r` rc rd rY r r r r r8 � s F r8 )N)NNNN)NN) rh r ZnumpyrJ r Z matplotlibr0 Zmatplotlib.colorsr" Zmatplotlib.cbookr: Zmatplotlib._cmr Zmatplotlib._cm_listedr Z cmaps_listedr/ r r r! r& r1 ZLUTSIZE�updater r( �localsr4 r7 �objectr8 r r r r �<module> s0 � '