OwlCyberSecurity - MANAGER
Edit File: image.cpython-38.pyc
U ��]d� � @ s� d Z ddlmZ ddlmZ ddlZddlZddlmZ ddl Z ddl Z ddl mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlT dd l m Z m Z m Z m!Z!m"Z"m#Z# e�$e%�Z&e j'e j'e j(e j)e j*e j+e j,e j-e j.e j/e j0e j1e j2e j3e j4e j5e j6e j7d �Z8e9e8�Z:d*d d �Z;d+dd�Z<dd� Z=G dd� dej>ej?�Z@G dd� de@�ZAG dd� deA�ZBG dd� deA�ZCG dd� de@�ZDG d d � d e@�ZEd,d d �ZFd-d!d"�ZGd#d$� ZHd.d(d)�ZIdS )/zR The image module supports basic image loading, rescaling and display operations. � )�BytesIO)�ceilN)�Path)�rcParams)�FigureCanvasBase)�*)�Affine2D�BboxBase�Bbox� BboxTransform�IdentityTransform�TransformedBbox)�none�nearest�bilinearZbicubicZspline16Zspline36ZhanningZhammingZhermiteZkaiserZquadricZcatromZgaussianZbesselZmitchellZsincZlanczosZblackman� �?c C sP t | �dkr"tjdtjd�ddfS g }g }| D ]x}|�||�\}}}} |dk r.||9 }||9 }|�||||�� prdf� |�t||g||jd ||jd gg�� q.t |�dkr�tjdtjd�ddfS t� |�} tj t | j �t | j �dftjd�} |D ]>\}}}} t� �|| j || j �} tj|| | tjd| d � q�| | j| | j| fS ) aD Composite a number of RGBA images into one. The images are composited in the order in which they appear in the `images` list. Parameters ---------- images : list of Images Each must have a `make_image` method. For each image, `can_composite` should return `True`, though this is not enforced by this function. Each image must have a purely affine transformation with no shear. renderer : RendererBase instance magnification : float The additional magnification to apply for the renderer in use. Returns ------- tuple : image, offset_x, offset_y Returns the tuple: - image: A numpy array of the same type as the input images. - offset_x, offset_y: The offset of the image (left, bottom) in the output figure. r )r r � ��dtypeNr � r F)�resample�alpha)�len�np�empty�uint8� make_image�append� get_alphar �shape�union�zeros�int�height�widthr � translate�x0�y0�_imager �NEAREST) Zimages�renderer� magnification�partsZbboxes�image�data�x�y�trans�bbox�outputr � r4 �2/usr/lib/python3/dist-packages/matplotlib/image.py�composite_images; s6 &� � �r6 c s� t dd� |D ��}|dk r |n��� }|s.|sD|D ]}|��� q2n\g � ��� �� ���fdd�}|D ]2}t|t�r�|�� r�� �|� qf|� |��� qf|� dS )a Draw a sorted list of artists, compositing images into a single image where possible. For internal matplotlib use only: It is here to reduce duplication between `Figure.draw` and `Axes.draw`, but otherwise should not be generally useful. c s s | ]}t |t�V qd S �N)� isinstance� _ImageBase��.0r/ r4 r4 r5 � <genexpr> s z0_draw_list_compositing_images.<locals>.<genexpr>Nc s� t � �dkr � d ��� nnt � �dkr�t� ���\} }}| jdkr���� }|��j� |���� � �� |t � |�t � |�| � |� � � d d �= d S )Nr r ) r �drawr6 �size�new_gcZset_clip_rectangler2 Z set_clip_pathZ get_clip_path� draw_imager �round�restore)r. �l�b�gc�Z image_groupZmag�parentr* r4 r5 � flush_images� s z3_draw_list_compositing_images.<locals>.flush_images)�anyZoption_image_nocompositer= �get_image_magnificationr8 r9 � can_compositer )r* rG ZartistsZsuppress_compositeZ has_imagesZ not_composite�arH r4 rF r5 � _draw_list_compositing_imagesu s � rM c C sz t j| jd | jd df| jd�}| |dd�dd�dd�f< |jt jkr`d|dd�dd�df< nd|dd�dd�df< |S ) z\ Convert an RGB image to RGBA, as required by the image resample C++ extension. r r r r N� � r )r r! r r r )�A�rgbar4 r4 r5 � _rgb_to_rgba� s " rR c s� e Zd ZdZdd� Zd7dd �Z� fd d �Zd d � Zdd� Zdd� Z d8dd�Z d9dd�Z dd� Z dd� Z ejdd � �Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z � Z S ):r9 r c C s dt | jjj� S �NzAxesImage(%g,%g;%gx%g)��tuple�axesr2 �bounds��selfr4 r4 r5 �__str__� s z_ImageBase.__str__NT� @Fc K sz t j�| � tj�| ||� d| _|dkr2td }|| _| �|� | � |� | � |� | � |� || _ d| _ | �| � dS )� interpolation and cmap default to their rc settings cmap is a colors.Colormap instance norm is a colors.Normalize instance to map luminance to 0-1 extent is data axes (left, right, bottom, top) for making image plots registered with data plots. Default is to label the pixel centers with the zero-based row and column indices. Additional kwargs are matplotlib.artist properties TN� image.origin)�martist�Artist�__init__�cm�ScalarMappableZ _mouseoverr �origin�set_filternorm� set_filterrad�set_interpolation� set_resamplerV �_imcache�update) rY �ax�cmap�norm� interpolationrc � filternorm� filterradr �kwargsr4 r4 r5 r` � s z_ImageBase.__init__c s t � �� }d |d< |S )Nrh )�super� __getstate__)rY �state�� __class__r4 r5 rr � s z_ImageBase.__getstate__c C s" | j dkrtd��| j jdd� S )z+Get the numrows, numcols of the input imageN�"You must first set the image array� )�_A� RuntimeErrorr rX r4 r4 r5 �get_size� s z_ImageBase.get_sizec C s t j�| |� d| _dS )z� Set the alpha value used for blending - not supported on all backends. Parameters ---------- alpha : float N)r^ r_ � set_alpharh )rY r r4 r4 r5 r{ � s z_ImageBase.set_alphac C s d| _ d| _tj�| � dS )zb Call this whenever the mappable is changed so observers can update state N)rh � _rgbacachera rb �changedrX r4 r4 r5 r} � s z_ImageBase.changedr c % C s |dkrt d��|jdkr"t d��t�||�}|dkr:dS |j| } |j| } | dks^| dkrbdS | jdkr�t� �d|j d �� dd�} nt � } | t� � |j|j d |j|j d ��|j |j �| �� 7 } | t� �|j |j �� ||� } | j�rh|�rh| d d k�s | d d k�rhtt| ��} tt| ��}| | | }|| | }| t� � d | d | �7 } nt| �} t| �}|�s0|jd k�r�td �|j ���|jd k�r�|j}|�� }|�� }|tjjk�r�t�d�t�d� }}|j dk�r.|j}|jtj tj fk�r^|jtj k�r&t �!d�|j�� tj }n0|�"tj �|�"tj � }|dk�rXtj }ntj }tj#|j |d�}||dd�< | j$�%|� t� | j$j&�t� | j$j'� }| j$j'|d }|tj k�r�dnd}||| }||k �r�d}n t� |�}||| }||k�rd}n t� |�}|dk �s&|dk �r4t�(|||�}||8 }|�"|��)� }|�"|��)� }||k�rr||| d }|d7 }tj*|| f|jd�} t+�,|| | t-| �.� | �/� d | �0� | �1� � ~| d8 } ||k�r�| || d 9 } | |7 } t2| j$t3j4��r| �"|j�} tj#|j tj d�} |j5j |j k�rLt�6|j5t� tj7�t� d��| dd�< n d| dd�< tj*|| f| jd�} t+�,| | | t-| �.� dd| �0� | �1� � ~ t�8| �} t�9| �} d| | < | �$tj�:| | ��} n�|j d dk�r�t;|�}n |j d dk�rtd �|j ���tj*|| df|jd�} tj*|| f|jd�}!| �<� }"|"dk�rBd}"|d }#t+�,|#|!| t-| �.� | �/� |"| �0� | �1� � t;|ddd�f �}t+�,|| | t-| �.� | �/� |"| �0� | �1� � |!| d< | j=| ddd �} |jd k�r�| �<� }"|"dk�r�d}"| dd�dd�df }#t�>t�>|#tj �| |" tj?�|#dd�< n�| j@dk�rT| j=|d|jd kd �| _@| j@} tA|| �B� �C� ��B� }$| tt|$jDd��tt|$jEd | j d ���tt|$jFd��tt|$jGd | j d ���f } t� �tt|$jFd��tt|$jDd���| } | |j |j | fS ) a� Normalize, rescale, and colormap the image *A* from the given *in_bbox* (in data space), to the given *out_bbox* (in pixel space) clipped to the given *clip_bbox* (also in pixel space), and magnified by the *magnification* factor. *A* may be a greyscale image (M, N) with a dtype of float32, float64, float128, uint16 or uint8, or an (M, N, 4) RGBA image with a dtype of float32, float64, float128, or uint8. If *unsampled* is True, the image will not be scaled, but an appropriate affine transformation will be returned instead. If *round_to_pixel_border* is True, the output image size will be rounded to the nearest pixel boundary. This makes the images align correctly with the axes. It should not be used if exact scaling is needed, such as for `FigureImage`. Returns ------- image : (M, N, 4) uint8 array The RGBA image, resampled unless *unsampled* is True. x, y : float The upper left corner where the image should be drawn, in pixel space. trans : Affine2D The affine transformation from image to pixel space. Nz9You must first set the image array or the image attributer zk_make_image must get a non-empty image. Your Artist's draw method must filter before this method is called.)Nr r N�upperr ���r g �rw rN � Invalid shape {} for image datarw �fz5Casting input data from '{0}' to 'float64' for imshowg �חAr g �cAg ��@g�������?皙�����?TrN r ).rN .F��bytesrl )Hry r> r � intersectionr$ r# rc r r% r �scaler r& r' � get_transform� is_affiner" r �ndim� ValueError�formatr �min�maxr ZmaZmaskedZint32Zkind�float32Zfloat16�float64�cbookZ_warn_external�astyper rl Zautoscale_None�vmax�vmin�clip�itemr! r( r � _interpd_�get_interpolation� get_resample�get_filternorm� get_filterradr8 �mcolorsZNoNorm�mask�where�nan�array�isnanZ masked_arrayrR r �to_rgba�asarrayr rh r �frozen�inverted�ymin�ymax�xmin�xmax)%rY rP Zin_bboxZout_bboxZ clip_bboxr+ � unsampled�round_to_pixel_borderZ clipped_bboxZout_width_baseZout_height_baseZt0�tZ out_widthZ out_heightZ extra_widthZ extra_heightZ inp_dtypeZa_minZa_maxZ scaled_dtypeZdaZA_scaledZdvZvmidZfactZnewminZnewmaxZ A_resampledr� Zout_maskZ out_alphar3 Zoutput_ar Z alpha_channelZsubsetr4 r4 r5 � _make_image� s� � ���� �� � � � � � � � � � � � � � � ��� ��z_ImageBase._make_imagec C s t d��dS )am Normalize, rescale, and colormap this image's data for rendering using *renderer*, with the given *magnification*. If *unsampled* is True, the image will not be scaled, but an appropriate affine transformation will be returned instead. Returns ------- image : (M, N, 4) uint8 array The RGBA image, resampled unless *unsampled* is True. x, y : float The upper left corner where the image should be drawn, in pixel space. trans : Affine2D The affine transformation from image to pixel space. z(The make_image method must be overriddenN��NotImplementedError)rY r* r+ r� r4 r4 r5 r ) s z_ImageBase.make_imagec C sV | j |dd�\}}}}|dkr"dS t� �|jd |jd �| }|�|||||� dS )zu draw unsampled image. The renderer should support a draw_image method with scale parameter. T�r� Nr r )r r r� r r@ )rY r* rE �imrC rD r1 r4 r4 r5 �_draw_unsampled_image= s z _ImageBase._draw_unsampled_imagec C s dS )z{ return True if the image is better to be drawn unsampled. The derived class needs to override it. Fr4 �rY r* r4 r4 r5 �_check_unsampled_imageL s z!_ImageBase._check_unsampled_imagec O s� | � � sd| _d S | �� jdkr*d| _d S |�� }| �|� |�| �� � |�| � � � |� | � � � | � |�r�| � � jr�| �||� n0| �||�� �\}}}}|d k r�|�||||� |�� d| _d S )NFr )� get_visible�stale� get_arrayr> r? Z _set_gc_clipr{ r Zset_urlZget_urlZset_gidZget_gidr� r� r� r� r rJ r@ rB ) rY r* �argsrp rE r� rC rD r1 r4 r4 r5 r= S s. � � z_ImageBase.drawc C s� | j dk r| � | |�S | j|jk r*di fS |j|j }}| �� \}}}}||krZ|| }}||krl|| }}|dk r�|dk r�|| ko�|kn o�|| ko�|kn }nd}|i fS )zI Test whether the mouse event occurred within the image. NF)� _containsrV Zinaxes�xdata�ydata� get_extent) rY � mouseeventr/ r0 r� r� r� r� �insider4 r4 r5 �containsq s 2z_ImageBase.containsc C sH ddl m} | j| jdkr*| jddd� n| jddd�}|�||� dS )z&Write the image to png file with fnamer ��_png�lowerNr Tr� )� matplotlibr� r� rc rx � write_png)rY �fnamer� r� r4 r4 r5 r� � s " �z_ImageBase.write_pngc C sb zddl m} W n tk r$ Y nX t||j�r:t|�}tj|dd�| _| jjt j kr|t � | jjt d�s|t d�| jj���| jjdks�| jjdkr�| jjd d ks�t d �| jj���| jjdk�rLt �| jjt j�r�d nd }| j�� dk s�|| j�� k �rt�d� t �| jd|�| _| jjt j k�rLt �| jjt j��rL| j�t j �| _d| _d| _d| _dS )z� Set the image array. Note that this function does *not* update the normalization used. Parameters ---------- A : array-like or `PIL.Image.Image` r ��ImageT��copyZ same_kindz3Image data of dtype {} cannot be converted to floatrw rN r �rN r r� rO r zmClipping input data to the valid range for imshow with RGB data ([0..1] for floats or [0..255] for integers).N) �PILr� � ImportErrorr8 � pil_to_arrayr� �safe_masked_invalidrx r r r Zcan_cast�float� TypeErrorr� r� r Z issubdtypeZintegerr� r� �_logZwarningr� r� rh r| r� )rY rP r� Zhighr4 r4 r5 �set_data� sF �� ��� ��z_ImageBase.set_datac C s | � |� dS )z� Retained for backwards compatibility - use set_data instead. Parameters ---------- A : array-like N)r� �rY rP r4 r4 r5 � set_array� s z_ImageBase.set_arrayc C s | j S )a* Return the interpolation method the image uses when resizing. One of 'nearest', 'bilinear', 'bicubic', 'spline16', 'spline36', 'hanning', 'hamming', 'hermite', 'kaiser', 'quadric', 'catrom', 'gaussian', 'bessel', 'mitchell', 'sinc', 'lanczos', or 'none'. )�_interpolationrX r4 r4 r5 r� � s z _ImageBase.get_interpolationc C s8 |dkrt d }|�� }|tkr(td��|| _d| _dS )a( Set the interpolation method the image uses when resizing. if None, use a value from rc setting. If 'none', the image is shown as is without interpolating. 'none' is only supported in agg, ps and pdf backends and will fall back to 'nearest' mode for other backends. Parameters ---------- s : {'nearest', 'bilinear', 'bicubic', 'spline16', 'spline36', 'hanning', 'hamming', 'hermite', 'kaiser', 'quadric', 'catrom', 'gaussian', 'bessel', 'mitchell', 'sinc', 'lanczos', 'none'} Nzimage.interpolationz Illegal interpolation stringT)r r� r� r� r� r� �rY �sr4 r4 r5 rf � s z _ImageBase.set_interpolationc C s | � � }| jdko |jo |jS )z>Return whether the image can be composited with its neighbors.r )r� r� r� Z is_separable)rY r1 r4 r4 r5 rK � s ��z_ImageBase.can_compositec C s |dkrt d }|| _d| _dS )z� Set whether image resampling is used. Parameters ---------- v : bool or None If None, use :rc:`image.resample` = True. Nzimage.resampleT)r � _resampler� )rY �vr4 r4 r5 rg s z_ImageBase.set_resamplec C s | j S )z(Return whether image resampling is used.)r� rX r4 r4 r5 r� s z_ImageBase.get_resamplec C s t |�| _d| _dS )z� Set whether the resize filter normalizes the weights. See help for `~.Axes.imshow`. Parameters ---------- filternorm : bool TN)�bool� _filternormr� )rY rn r4 r4 r5 rd s z_ImageBase.set_filternormc C s | j S )z8Return whether the resize filter normalizes the weights.)r� rX r4 r4 r5 r� s z_ImageBase.get_filternormc C s( t |�}|dkrtd��|| _d| _dS )z� Set the resize filter radius only applicable to some interpolation schemes -- see help for imshow Parameters ---------- filterrad : positive float r z+The filter radius must be a positive numberTN)r� r� � _filterradr� )rY ro �rr4 r4 r5 re # s z_ImageBase.set_filterradc C s | j S )z Return the filterrad setting.)r� rX r4 r4 r5 r� 2 s z_ImageBase.get_filterrad)NNNNTr[ F)r FT)r F) �__name__� __module__� __qualname__�zorderrZ r` rr rz r{ r} r� r r� r� r^ Zallow_rasterizationr= r� r� r� r� r� rf rK rg r� rd r� re r� � __classcell__r4 r4 rt r5 r9 � sL � ' � , 2 r9 c sf e Zd Zdd� Zd� fdd� Zdd d �Zdd d �Zdd� Zdd� Zdd� Z dd� Z � fdd�Z � Z S )� AxesImagec C s dt | jjj� S rS rT rX r4 r4 r5 rZ 8 s zAxesImage.__str__Nr r[ Fc s. || _ t� j|f||||||| d�| �� dS )r\ �rk rl rm rc rn ro r N)�_extentrq r` ) rY rj rk rl rm rc �extentrn ro r rp rt r4 r5 r` ; s �� �zAxesImage.__init__c C s. | j \}}}}t�||||g�}|�| jj�S r7 )r� r Z from_extentsZ transformedrV Z transData)rY r* r& �x1r' �y1r2 r4 r4 r5 �get_window_extentc s zAxesImage.get_window_extentr c C sX | � � }| �� \}}}}tt�||g||gg��} t| |�} | j| j| | | jj ||d�S )Nr� ) r� r� r r r� r r� rx rV r2 ) rY r* r+ r� r1 r� Zx2r� Zy2r2 Ztransformed_bboxr4 r4 r5 r h s �zAxesImage.make_imagec C s | � � dko|�� S )zK Return whether the image would be better drawn unsampled. r )r� Zoption_scale_imager� r4 r4 r5 r� s s �z AxesImage._check_unsampled_imagec C s� | | _ \}}}}||f||ff}| j�|� ||g| jjdd�< ||g| jjdd�< | jjrr| jj||fdd� | jjr�| jj ||fdd� d| _ dS )ab extent is data axes (left, right, bottom, top) for making image plots This updates ax.dataLim, and, if autoscaling, sets viewLim to tightly fit the image, regardless of dataLim. Autoscaling state is not changed, so following this with ax.autoscale_view will redo the autoscaling in accord with dataLim. N)�autoT) r� rV Zupdate_datalimZ sticky_edgesr/ r0 Z _autoscaleXonZset_xlimZ _autoscaleYonZset_ylimr� )rY r� r� r� r� r� Zcornersr4 r4 r5 � set_extentz s zAxesImage.set_extentc C sV | j dk r| j S | �� }|\}}| jdkr>d|d |d dfS d|d d|d fS dS )�.Get the image extent: left, right, bottom, topNr~ � �� �?)r� rz rc )rY Zsz�numrows�numcolsr4 r4 r5 r� � s zAxesImage.get_extentc C s� | � � \}}}}| jdkr$|| }}| �� }t||g||gg�}tddg|jdd� g�}t||d�} |j|j } } | �| | g�} t t � | ��r�dS | � t �\} }d| kr�|jd k r�n n d| kr�|jd k s�n dS || |f S dS )z� Return the image value at the event position or *None* if the event is outside the image. See Also -------- matplotlib.artist.Artist.get_cursor_data r~ r Nrw )ZboxinZboxoutr )r� rc r� r r r r� r� Ztransform_pointrI r r� r� r" )rY �eventr� r� r� r� �arrZ data_extentZ array_extentr1 r0 r/ Zpoint�i�jr4 r4 r5 �get_cursor_data� s :zAxesImage.get_cursor_datac sD t �|�dkr4| jr4dt�| jj�|���� d S t� � |�S d S )Nr �[�]) r r� Zcolorbarr� Z strip_mathZ formatterZformat_data_short�striprq �format_cursor_data)rY r. rt r4 r5 r� � s ����z AxesImage.format_cursor_data)NNNNNr r[ F)N)r F) r� r� r� rZ r` r� r r� r� r� r� r� r� r4 r4 rt r5 r� 7 s" �( r� c s| e Zd Zdd�� fdd� Zdd� Zdd d �Zd d � Zd d� Zdd� Zdd� Z dd� Z dd� Z � fdd�Z � fdd�Z � ZS ) �NonUniformImager )rm c s t � j|f|� | �|� dS )z� kwargs are identical to those for AxesImage, except that 'nearest' and 'bilinear' are the only supported 'interpolation' options. N)rq r` rf )rY rj rm rp rt r4 r5 r` � s zNonUniformImage.__init__c C s dS )z; return False. Do not use unsampled image. Fr4 r� r4 r4 r5 r� � s z&NonUniformImage._check_unsampled_imager Fc C s� | j d krtd��|r td��| j }|jdkr�|jtjkrV| j|dd�}| j� � | _ n>t� |d d �d d �tj f dd�}d|d d �d d �df< d| _ n�|jtjkr�d| � tj�}|jd dk�rt�t|jd d� d ��tj�}||d d �d d �d d�f< d|d d �d d �df< |}d | _ | jjj\}}}} | jjj\} } } } t�| �d t�| �d }t�| �d t�| �d }||9 }||9 }t�| j| j|t|�t|�||| ||| ft | j �}|| | t � fS ) Nrv z*unsampled not supported on NonUniformImagerw T�r� r rO rN r Fr� )r ) rx ry r� r� r r r r� rk �is_gray� is_grayscale�repeatZnewaxisr� r r! rU rV �viewLimrW r2 �extentsrA r( Zpcolor�_Ax�_Ayr" r� r� r )rY r* r+ r� rP �Br& r' Zv_widthZv_heightrC rD r� r� r$ r# r� r4 r4 r5 r � sB " �zNonUniformImage.make_imagec C s� t �|t j�}t �|t j�}tj|dd�}|j|j krBdkr`n n|jdd� |j|j kshtd��|jdkrztd��|jd kr�|jd d kr�td ��|jd kr�|jd dkr�|jdd� |_|| _|| _ || _ d | _ d| _ d S ) a^ Set the grid for the pixel centers, and the pixel values. *x* and *y* are monotonic 1-D ndarrays of lengths N and M, respectively, specifying pixel centers *A* is an (M,N) ndarray or masked array of values to be colormapped, or a (M,N,3) RGB array, or a (M,N,4) RGBA array. Tr� r r rw z Axes don't match array shaper� zCan only plot 2D or 3D datarN )r rN r z?3D arrays must have three (RGB) or four (RGBA) color componentsN) r r� r� r� r� r� r r� rx r r rh r� �rY r/ r0 rP r4 r4 r5 r� � s 6 zNonUniformImage.set_datac G s t d��d S �NzMethod not supportedr� �rY r� r4 r4 r5 r� s zNonUniformImage.set_arrayc C s( |dk r|dkrt d��t�| |� dS )z| Parameters ---------- s : str, None Either 'nearest', 'bilinear', or ``None``. N)r r z?Only nearest neighbor and bilinear interpolations are supported)r� r� rf r� r4 r4 r5 rf s z!NonUniformImage.set_interpolationc C s6 | j d krtd��| jd | jd | jd | jd fS )NzMust set data firstr r )rx ry r r rX r4 r4 r5 r� ! s zNonUniformImage.get_extentc C s d S r7 r4 r� r4 r4 r5 rd & s z NonUniformImage.set_filternormc C s d S r7 r4 r� r4 r4 r5 re ) s z NonUniformImage.set_filterradc s" | j d k rtd��t� �|� d S �Nz'Cannot change colors after loading data)rx ry rq �set_norm)rY rl rt r4 r5 r , s zNonUniformImage.set_normc s" | j d k rtd��t� �|� d S r )rx ry rq �set_cmap)rY rk rt r4 r5 r 1 s zNonUniformImage.set_cmap)r F)r� r� r� r` r� r r� r� rf r� rd re r r r� r4 r4 rt r5 r� � s $ r� c sL e Zd ZdZd� fdd� Zddd�Zd d � Zd d � Zd d� Zdd� Z � Z S )� PcolorImagez� Make a pcolor-style plot with an irregular rectangular grid. This uses a variation of the original irregular image code, and it is used by pcolorfast for the corresponding grid type. Nc s6 t � j|||d� | �|� |dk r2| �|||� dS )z� cmap defaults to its rc setting cmap is a colors.Colormap instance norm is a colors.Normalize instance to map luminance to 0-1 Additional kwargs are matplotlib.artist properties )rl rk N)rq r` ri r� )rY rj r/ r0 rP rk rl rp rt r4 r5 r` > s zPcolorImage.__init__r Fc C s: | j d krtd��|r td��| jj�� }t�|d�}t� |�d � tj �}| jj j \}}}} t�|�d t�|�d } t�| �d t�|�d } tt�| | ��} tt�| | ��} | jd kr�| j| j dd�} | | _| j jdkr�| j�� | _n| j} | jj} t�| j| j| | | | j| j| j | j f|�}|||t � fS ) Nrv z&unsampled not supported on PColorImager rO r� Tr� rw ) rx ry r� rV ZpatchZ get_facecolorr� r� r r� r� r r2 r rA r" r| r� rk r� r� r� r( Zpcolor2r r r& r� r' r� r )rY r* r+ r� ZfcZbgrC rD r� r� r$ r# rP Zvlr� r4 r4 r5 r S s4 �zPcolorImage.make_imagec C s dS )NFr4 r� r4 r4 r5 r� q s z"PcolorImage._check_unsampled_imagec C s$ t j|dd�}|dkr4tjd|jd d tjd�}nt�|tj��� }|dkrltjd|jd d tjd�}nt�|tj��� }|jdd� |jd |jd fkr�t d|jdd� |jd |jd ff ��|j d kr�t d ��|j d k�r|jd dk�r|jdd� |_d | _ |j d k�r�|jd d k�r�|dd�dd�df |dd�dd�df k� � �r�|dd�dd�df |dd�dd�df k� � �r�d| _ nt d��|d |d k �r�|ddd� }|dd�ddd�f }|d |d k �r|ddd� }|ddd� }|| _ || _|| _d| _d| _dS )a� Set the grid for the rectangle boundaries, and the data values. *x* and *y* are monotonic 1-D ndarrays of lengths N+1 and M+1, respectively, specifying rectangle boundaries. If None, they will be created as uniform arrays from 0 through N and 0 through M, respectively. *A* is an (M,N) ndarray or masked array of values to be colormapped, or a (M,N,3) RGB array, or a (M,N,4) RGBA array. Tr� Nr r r rw z2Axes don't match array shape. Got %s, expected %s.r� zA must be 2D or 3DrN Fr� z+3D arrays must have RGB or RGBA as last dimr )r� r� r Zaranger r� r� Zravelr> r� r� r� �allrx r r r| r� r r4 r4 r5 r� t sH " �� 2.�zPcolorImage.set_datac G s t d��d S r r� r r4 r4 r5 r� � s zPcolorImage.set_arrayc C s� |j |j }}|| jd k sF|| jd ksF|| jd k sF|| jd krJd S t�| j|�d }t�| j|�d }z| j||f W S tk r� Y d S X d S )Nr r r )r� r� r r r Z searchsortedrx � IndexError)rY r� r/ r0 r� r� r4 r4 r5 r� � s � �zPcolorImage.get_cursor_data)NNNNN)r F) r� r� r� �__doc__r` r r� r� r� r� r� r4 r4 rt r5 r 7 s � 7r c s@ e Zd ZdZdZd� fdd� Zdd� Zdd d �Zd d � Z� Z S )� FigureImager r Nc s: t � jd|||d� || _|| _|| _| �|� d| _dS )�� cmap is a colors.Colormap instance norm is a colors.Normalize instance to map luminance to 0-1 kwargs are an optional list of Artist keyword args N)rl rk rc r )rq r` �figure�ox�oyri r+ )rY �figrk rl ZoffsetxZoffsetyrc rp rt r4 r5 r` � s � zFigureImage.__init__c C s8 | � � \}}d| j |d | j d| j |d | j fS )r� r� r� )rz r r )rY r� r� r4 r4 r5 r� � s �zFigureImage.get_extentr Fc C s� |j | jj }t| j| | j| g| j| | jjd | j| | jjd gg�}| j�� \}}||j 9 }||j 9 }tddg||gg�}| j| j||||| |dd�S )Nr r F)r+ r� r� ) �dpir r r r rx r Zget_size_inchesr� ) rY r* r+ r� Zfacr2 r$ r# r� r4 r4 r5 r � s& �� �zFigureImage.make_imagec C s" t j�| tj|dd�� d| _dS )zSet the image array.Tr� N)ra rb r� r� r� r� r� r4 r4 r5 r� � s �zFigureImage.set_data)NNr r N)r F) r� r� r� r� r� r` r� r r� r� r4 r4 rt r5 r � s � r c sh e Zd ZdZe�dd�d� fd d � �Ze�d�ed d � ��Z d d� Z ddd�Z dd� Z ddd�Z � ZS )� BboxImagez;The Image class whose size is determined by the given bbox.z3.1�interp_at_nativeNr r[ FTc s: t � jd|||||||d�| �� || _| | _t� | _dS )r Nr� )N)rq r` r2 �_interp_at_nativer � _transform) rY r2 rk rl rm rc rn ro r r rp rt r4 r5 r` � s �� � zBboxImage.__init__c C s | j S r7 )r rX r4 r4 r5 r s zBboxImage.interp_at_nativec C s | j S r7 )r rX r4 r4 r5 r� # s zBboxImage.get_transformc C sD |d kr| � � j}t| jt�r$| jS t| j�r8| �|�S td��d S )Nzunknown type of bbox)Z get_figureZ_cachedRendererr8 r2 r �callabler� r� r4 r4 r5 r� & s zBboxImage.get_window_extentc C sL | j dk r| � | |�S | �� s&di fS |j|j }}| �� �||�}|i fS )z7Test whether the mouse event occurred within the image.NF)r� r� r/ r0 r� r� )rY r� r/ r0 r� r4 r4 r5 r� 1 s zBboxImage.containsr c C s~ |� � \}}| �|��� }| j||g _| �|�}tddg||gg�}ttddgddgg�|�| _| j| j|||||d�S )Nr r r� ) Zget_canvas_width_heightr� r� Z_pointsr r r r� rx ) rY r* r+ r� r$ r# Zbbox_inZbbox_outr� r4 r4 r5 r > s �zBboxImage.make_image)NNNNr r[ FT)N)r F)r� r� r� r r� Z_delete_parameterr` Z deprecated�propertyr r� r� r� r r� r4 r4 rt r5 r � s$ �! r c C s� dd� }d|i}|dkr�t | t�rdtj�| �}t|j�dkrBd}q�tj� | �\}}|� � dd� }q�t | d�r�tj� | j �\}}|� � dd� }q�d}n|}||k�rzddl m} W n$ tk r� td t|� ��Y nX |�| ��}t|�W 5 Q R � S Q R X || } t | t��r�tj�| �}t|j�dk�rXdd lm} t| �| ��� �} | | �S t| d ��} | | �W 5 Q R � S Q R X n| | �S dS ) a� Read an image from a file into an array. Parameters ---------- fname : str or file-like The image file to read: a filename, a URL or a file-like object opened in read-binary mode. format : str, optional The image file format assumed for reading the data. If not given, the format is deduced from the filename. If nothing can be deduced, PNG is tried. Returns ------- imagedata : :class:`numpy.array` The image data. The returned array has shape - (M, N) for grayscale images. - (M, N, 3) for RGB images. - (M, N, 4) for RGBA images. Notes ----- Matplotlib can only read PNGs natively. Further image formats are supported via the optional dependency on Pillow. Note, URL strings are not compatible with Pillow. Check the `Pillow documentation`_ for more information. .. _Pillow documentation: http://pillow.readthedocs.io/en/latest/ c _ s ddl m} |j| |�S )Nr r� )r� r� �read_png)r� rp r� r4 r4 r5 r l s zimread.<locals>.read_png�pngNr �namer r� z_Only know how to handle extensions: %s; with Pillow installed matplotlib can handle more images)�request�rb)r8 �str�urllib�parseZurlparser Zscheme�os�path�splitextr� �hasattrr r� r� r� r� �list�openr� r r Zurlopen�read) r� r� r ZhandlersZparsedZext�basenamer� r- Zhandlerr �fdr4 r4 r5 �imreadK sB ! � r, �d c C s� ddl m} ddlm} t| tj�r.t�| �} |dkr^t| t�rRt | �j dd� nt d � � }|dkr�||dd �} | j |||||d d � | j| ||d d � �n6tj|d �} | �||� |dkr�t d }|dkr�|ddd� }| j|d d�} |dk�r| j| | |d� n�zddlm} W n6 tk �rL } ztd|� d��|�W 5 d}~X Y nX | jd | jd f}| �d|| dddd�}|dk�r�d}tdd � t�t d �D ��}| � d ||�}|� ||� |}|j | |||fd � dS ) a� Save an array as an image file. Parameters ---------- fname : str or PathLike or file-like A path or a file-like object to store the image in. If *format* is not set, then the output format is inferred from the extension of *fname*, if any, and from :rc:`savefig.format` otherwise. If *format* is set, it determines the output format. arr : array-like The image data. The shape can be one of MxN (luminance), MxNx3 (RGB) or MxNx4 (RGBA). vmin, vmax : scalar, optional *vmin* and *vmax* set the color scaling for the image by fixing the values that map to the colormap color limits. If either *vmin* or *vmax* is None, that limit is determined from the *arr* min/max value. cmap : str or `~matplotlib.colors.Colormap`, optional A Colormap instance or registered colormap name. The colormap maps scalar data to colors. It is ignored for RGB(A) data. Defaults to :rc:`image.cmap` ('viridis'). format : str, optional The file format, e.g. 'png', 'pdf', 'svg', ... The behavior when this is unset is documented under *fname*. origin : {'upper', 'lower'}, optional Indicates whether the ``(0, 0)`` index of the array is in the upper left or lower left corner of the axes. Defaults to :rc:`image.origin` ('upper'). dpi : int The DPI to store in the metadata of the file. This does not affect the resolution of the output image. r ��Figurer� Nr zsavefig.format)ZpdfZpsZepsZsvgF)r �frameonT)rk r� r� rc Zresize)r r� Z transparent)rk r] r� r r� r �r r� z Saving to z requires Pillow�RGBA�raw)Zjpg�jpegr4 c s s | ]}t |d �V qdS )rO N)r" r: r4 r4 r5 r< � s �zimsave.<locals>.<genexpr>zsavefig.facecolor�RGB)r� r ) �matplotlib.figurer/ r� r� r8 r# �PathLike�fspathr r �suffixr r� Zfigimage�savefigra rb Zset_climr� r� r� r� r� r � frombufferrU r� Zto_rgb�newZpasteZsave)r� r� r� r� rk r� rc r r/ r� r ZsmrQ r� �excZ pil_shaper- ZcolorZ backgroundr4 r4 r5 �imsave� sd # � � �� � � r> c C s� | j dkrt�| �S | j �d�rp| �d| j �}| j �d�rHt�|d�}n t�|d�}|�| jddd� �� d �S z| � d �} W n t k r� t d ��Y nX t�| �S dS ) aX Load a `PIL image`_ and return it as a numpy array. .. _PIL image: https://pillow.readthedocs.io/en/latest/reference/Image.html Returns ------- numpy.array The array shape depends on the image type: - (M, N) for grayscale images. - (M, N, 3) for RGB images. - (M, N, 4) for RGBA images. )r2 ZRGBXr5 �LzI;16r3 r z>u2z<u2Nr z=u2r2 zUnknown image mode) �moder r� � startswith�tobytes�endswithr; Zreshaper> r� Zconvertr� ry )ZpilImager3 r/ r4 r4 r5 r� � s r� r� r Fc C s� t | �}|j\}}}d} || | } || | } |rTddlm} | j| | f| d�} n$ddlm} || | f| d�} t| � | jddddgddg g d �}|j |dd |d � | j || d � | S ) aw Make a thumbnail of image in *infile* with output filename *thumbfile*. See :doc:`/gallery/misc/image_thumbnail_sgskip`. Parameters ---------- infile : str or file-like The image file -- must be PNG, or Pillow-readable if you have Pillow_ installed. .. _Pillow: http://python-pillow.org/ thumbfile : str or file-like The thumbnail filename. scale : float, optional The scale factor for the thumbnail. interpolation : str, optional The interpolation scheme used in the resampling. See the *interpolation* parameter of `~.Axes.imshow` for possible values. preview : bool, optional If True, the default backend (presumably a user interface backend) will be used which will cause a figure to be raised if `~matplotlib.pyplot.show` is called. If it is False, the figure is created using `FigureCanvasBase` and the drawing backend is selected as `~matplotlib.figure.savefig` would normally do. Returns ------- figure : `~.figure.Figure` The figure instance containing the thumbnail. r- r N)Zfigsizer r. r r� F)�aspectr0 ZxticksZyticksT)rD r rm r1 ) r, r Zmatplotlib.pyplotZpyplotr r6 r/ r Zadd_axesZimshowr: )ZinfileZ thumbfiler� rm Zpreviewr� ZrowsZcolsZdepthr r# r$ Zpltr r/ rj r4 r4 r5 � thumbnail s& & �rE )r )N)N)NNNNNr- )r� r F)Jr �ior Zmathr r# Zlogging�pathlibr Z urllib.parser! Znumpyr r� r Zmatplotlib.artistZartistr^ Zmatplotlib.backend_basesr Zmatplotlib.colorsZcolorsr� Z matplotlib.cmra Zmatplotlib.cbookr� Zmatplotlib._imager( Zmatplotlib.transformsr r r r r r Z getLoggerr� r� r) ZBILINEARZBICUBICZSPLINE16ZSPLINE36ZHANNINGZHAMMINGZHERMITEZKAISERZQUADRICZCATROMZGAUSSIANZBESSELZMITCHELLZSINCZLANCZOSZBLACKMANr� �setZinterpolations_namesr6 rM rR r_ rb r9 r� r� r r r r, r> r� rE r4 r4 r4 r5 �<module> s~ � ; � . w =R T � P# �