U 7c @sdZddlZddlZddlZddlmZddlZddlZddl Z ddl Z ddl m Z mZmZddl mZddlTddlmZddlmZddlmZddlmZmZmZmZmZm Z m!Z!e"e#Z$ej%ej%ej%ej&ej'ej(ej)ej*ej+ej,ej-ej.ej/ej0ej1ej2ej3ej4ej5d Z6e7e6Z8d0d d Z9d1d dZ:dddddZ;ddZZ?Gddde?Z@Gddde@ZAGddde@ZBGddde?ZCGdd d e?ZDd2d!d"ZEd3ddd$d%d&ZFd'd(ZGd)d*ZHd4d.d/ZIdS)5zR The image module supports basic image loading, rescaling and display operations. N)Path)_apicbookcm)_image)*)FigureCanvasBase)Affine2DBboxBaseBbox BboxTransformBboxTransformToIdentityTransformTransformedBbox) antialiasednonenearestbilinearZbicubicZspline16Zspline36hanningZhammingZhermiteZkaiserZquadricZcatromZgaussianZbesselZmitchellZsincZlanczosZblackman?c CsLt|dkr"tjdtjdddfSg}g}|D]t}|||\}}}} |dk r.||9}||9}|||||f|t||g||jd||jdggq.t|dkrtjdtjdddfSt |} tj t | j t | j dftjd} |D]>\}}}} t|| j|| j} tj|| | tjd| dq| | j|| j|fS) a- 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` magnification : float, default: 1 The additional magnification to apply for the renderer in use. Returns ------- image : uint8 array (M, N, 4) The composited RGBA image. offset_x, offset_y : float The (left, bottom) offset where the composited image should be placed in the output figure. r)rrdtypeNrFresamplealpha)lennpemptyuint8 make_imageappend_get_scalar_alphar shapeunionzerosintheightwidthr translatex0y0rrNEAREST) Zimagesrenderer magnificationpartsZbboxesimagedataxytransbboxoutputrr84/tmp/pip-unpacked-wheel-bcsgmve3/matplotlib/image.pycomposite_images9s6 &  r:cstdd|D}|dk r|n}|s.|sD|D]}|q2nlgfdd}|D]B}t|tr|r|r|s |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. css|]}t|tVqdSN) isinstance _ImageBase.0r3r8r8r9 {sz0_draw_list_compositing_images..Ncstdkrdnjtdkrt\}}}|jdkr}|j| |t |t ||| dd=dS)Nrr) rdrawr:sizenew_gcZset_clip_rectangler6Z set_clip_path get_clip_path draw_imageroundrestore)r2lbgcZ image_groupZmagparentr.r8r9 flush_imagess    z3_draw_list_compositing_images..flush_images) anyZoption_image_nocompositerAget_image_magnificationr<r= can_composite get_clip_onrDr")r.rLZartistsZsuppress_compositeZ has_imagesZ not_compositearMr8rKr9_draw_list_compositing_imagesqs(     rSrrc Cs d}|jddkrdt|jddtt|jdd}|dddd|f}t|d|}|jddkrt|jd dtt|jdd}|dd|ddf}td||}| }|d krt ddg|jd|jdgg} | | } t t | dddf} t t | dddf} | d |jdksr| |jdksr| d |jdkr| d |jdks| |jdks| d |jdkrd }nd}t||jd d|j} |dkr|}t|| |t|||||| S)a Convenience wrapper around `._image.resample` to resample *data* to *out_shape* (with a third dimension if *data* is RGBA) that takes care of allocating the output array and fetching the relevant properties from the Image object *image_obj*. zData with more than {n} cannot be accurately displayed. Downsampling to less than {n} before displaying. To remove this warning, manually downsample your data.riz 2**23 columns)nNriz 2**24 rowsrrr)r$warningswarnformatr'rceilr scaleget_interpolationarray transformabsdiffr&r get_resamplerr _interpd_get_filternorm get_filterrad)Z image_objr2 out_shaper^rrmsgstep interpolationposZdispZdispxZdispyoutr8r8r9 _resamplesR  "     rkcCsztj|jd|jddf|jd}||ddddddf<|jtjkr`d|dddddf<nd|dddddf<|S) z\ Convert an RGB image to RGBA, as required by the image resample C++ extension. rrrrNrUr)rr&r$rr )Argbar8r8r9 _rgb_to_rgbas " rocseZdZdZdZd;dddd Zd d Zfd d ZddZddZ ddZ ddZ dr=a Base class for images. 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 rNT@Finterpolation_stagec  Kstj|tj||||dkr.tjd}tjddg|d||_ | || || || | ||||_d|_|| dS)N image.originupperlower)origin)martistArtist__init__rScalarMappablemplrcParamsr check_in_listrvset_filternorm set_filterradset_interpolationset_interpolation_stage set_resampleaxes_imcache_internal_update) selfaxcmapnormrhrv filternorm filterradrrrkwargsr8r8r9rys       z_ImageBase.__init__cCsDz |}t|jd|dWStk r>t|jYSXdS)Nz(size=))get_sizetype__name__ RuntimeError)rrBr8r8r9__str__s z_ImageBase.__str__cstddiS)Nr)super __getstate__r __class__r8r9rsz_ImageBase.__getstate__cCs"|jdkrtd|jjddS)z9Return the size of the image as tuple (numrows, numcols).N"You must first set the image arrayrV)_Arr$rr8r8r9rs z_ImageBase.get_sizecCs.tj||t|dkr$tdd|_dS)z Set the alpha value used for blending - not supported on all backends. Parameters ---------- alpha : float or 2D array-like or None )rrVz5alpha must be a float, two-dimensional array, or NoneN)rwrxZ_set_alpha_for_arrayrndim TypeErrorr)rrr8r8r9 set_alpha&sz_ImageBase.set_alphacCs$|jdkst|jdkrdS|jS)a Get a scalar alpha value to be applied to the artist as a whole. If the alpha value is a matrix, the method returns 1.0 because pixels have individual alpha values (see `~._ImageBase._make_image` for details). If the alpha value is a scalar, the method returns said value to be applied to the artist as a whole because pixels do not have individual alpha values. Nrr)Z_alpharrrr8r8r9r#4s z_ImageBase._get_scalar_alphacCsd|_tj|dS)zU Call this whenever the mappable is changed so observers can update. N)rrrzchangedrr8r8r9rAsz_ImageBase.changedrc- Cs@|dkrtd|jdkr"tdt||}|dkr:dS|j|} |j|} | dks^| dkrbdS|jdkrtd|j d  dd} nt } | t |j|j d|j|j d|j |j |7} | t|j |j  |} | jrb|rb| d d ks| d d krbt| } t| }| | | }|| | }| t d |d |7} nt| } t| }|| f}|sr|jd ks|jd kr|j dd kstd|j d|jd kr|jdkr|}|}|tjjkrtdtd}}|jjdkr`t|jjdkr,tjntj }|j|jjkrt!"d|jd|dn.|#tj|#tj}|dkrtjntj }tj$||d}|j%&|t|j%j't|j%j(}t|j%j(|d }|tjkrdnd}|||}||kr d}n t|}|||}||kr0d}n t|}|dk sN|dk r`tj)||||dd}d}|j%j(|j%j'}}|tjjkr||}}tj$||g|d} ||8}| |8} |#|*}|#|*}||kr||||}| |||} ||7}| |7} t+|||| }!~|!|8}!| |8} ||krT|!|||9}!| |||9} |!|7}!| |7} t,|j%t-j.r|!#|j}!|j/j |j krt0|j/t tj1t dn t2|tj }"t+||"|| dd}#~"t3|#}$d|#|$<|4}%|%dk rt|%dkr|#t+||%|| dd9}#tj5|!|$}&| \}'}(t,|j%t-j6rV|'dkrVt7|j8}'|j%j9:.t;j<|j%|'|(d|%|&})W5QRXW5QRXn|jd kr|j%&||=|}|j d d krt>|}|?}%t+||d || |%d!}*t+|t>|d"dd f|| |%d!})|*|)d <|j=|)dd#d$})|jd kr0|?}%|)ddddd f}+|+#tj |#|%|+dd<n|j@dkr|j=|d|jd kd$|_@|j@})tA|| BC},|)tt|,jDdtt|,jEd|)j dtt|,jFdtt|,jGd|)j df})ttt|,jFdtt|,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 attributerzk_make_image must get a non-empty image. Your Artist's draw method must filter before this method is called.)NrrNrtrrgrVrUrUrzInvalid shape z for image datarnfrzCasting input data from z to z for imshow.gחArgcAg@)rj皙?g?T)r)vminvmax).rU)r.Fbytesr)HrrBr intersectionr)r(rvr r*r$r[rr+r, get_transform is_affinemathrZr'r ValueError_interpolation_stageminmaxrmaZmaskedZint32rkinditemsizefloat64float32r warn_externalastyper]rZautoscale_Nonerrclipitemrkr<mcolorsZNoNormmaskwherenanZ ones_likeisnanZ get_alphaZ masked_arrayZLogNormZfinfoeps callbacksZblockedrZ _setattr_cmto_rgbaror#rrinvertedfrozenyminymaxxminxmax)-rrmZin_bboxZout_bboxZ clip_bboxr/ unsampledround_to_pixel_borderZ clipped_bboxZout_width_baseZout_height_baset0tZ out_widthZ out_heightZ extra_widthZ extra_heightreZa_minZa_maxZ scaled_dtypedaZA_scaledZdvZvmidZfactZnewminZnewmaxoffsetfracrrZvrangeZ A_resampledrZ out_alphaZout_maskrZresampled_maskedZs_vminZs_vmaxr7Z output_alphaZ alpha_channelZsubsetr8r8r9 _make_imageHsT            (             "           z_ImageBase._make_imagecCs tddS)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 overriddenNNotImplementedError)rr.r/rr8r8r9r!Msz_ImageBase.make_imagecCsdS)z| Return whether the image is better to be drawn unsampled. The derived class needs to override it. Fr8rr8r8r9_check_unsampled_imageasz!_ImageBase._check_unsampled_imagec Os|sd|_dS|jdkr*d|_dS|}|||||| | | | r| r|jr|j|dd\}}}}|dk rt|jd|jd|}||||||n2|||\}}}}|dk r||||||d|_dS)NFrTrr) get_visiblestale get_arrayrBrCZ _set_gc_cliprr#Zset_urlget_urlZset_gidZget_gidZoption_scale_imagerrrr!r r[r$rErOrG) rr.argsrrJimrHrIr5r8r8r9rAis8   z_ImageBase.drawc Cs||\}}|dk r||fS|j|jk r2difS|}||j|jg\}}|\}}} } ||krv||}}| | kr| | } } |dk r|dk r||ko|kno| |ko| kn}nd}|ifSz7Test whether the mouse event occurred within the image.NF) _default_containsrZinaxesrrr^r3r4 get_extent) r mouseeventinsideinfor5r3r4rrrrr8r8r9containss     2z_ImageBase.containscCsF|j|jdkr|jdddn|jddd}tj|j|dddS)z$Write the image to png file *fname*.ruNrTrpng)rY)rrvrPILImageZ fromarraysave)rfnamerr8r8r9 write_pngs "z_ImageBase.write_pngcCspt|tjjrt|}tj|dd|_|jjtj krXt |jjt dsXt d |jj|jjdkr|jjddkr|jddddd f|_|jjd ks|jjdkr|jjdd kst d |jj|jjdkr`t|jjtjrd nd}|jd ks||jkr,tdt|jd ||_|jjtj kr`t|jjtjr`|jtj |_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` TcopyZ same_kindz3Image data of dtype {} cannot be converted to floatrUrrNrrVrzInvalid shape {} for image datarlzmClipping input data to the valid range for imshow with RGB data ([0..1] for floats or [0..255] for integers).)r<rr pil_to_arrayrsafe_masked_invalidrrrr Zcan_castfloatrrYrr$Z issubdtypeintegerrr_logwarningrrrr)rrmhighr8r8r9set_datas@    z_ImageBase.set_datacCs||dS)z Retained for backwards compatibility - use set_data instead. Parameters ---------- A : array-like N)rrrmr8r8r9 set_arrays z_ImageBase.set_arraycCs|jS)a@ Return the interpolation method the image uses when resizing. One of 'antialiased', 'nearest', 'bilinear', 'bicubic', 'spline16', 'spline36', 'hanning', 'hamming', 'hermite', 'kaiser', 'quadric', 'catrom', 'gaussian', 'bessel', 'mitchell', 'sinc', 'lanczos', or 'none'. )_interpolationrr8r8r9r\s z_ImageBase.get_interpolationcCs8|dkrtjd}|}tjt|d||_d|_dS)a@ Set the interpolation method the image uses when resizing. If None, use :rc:`image.interpolation`. 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 : {'antialiased', 'nearest', 'bilinear', 'bicubic', 'spline16', 'spline36', 'hanning', 'hamming', 'hermite', 'kaiser', 'quadric', 'catrom', 'gaussian', 'bessel', 'mitchell', 'sinc', 'lanczos', 'none'} or None Nzimage.interpolationrhT)r{r|rurr}rbrrrsr8r8r9rs  z_ImageBase.set_interpolationcCs.|dkr d}tjddg|d||_d|_dS)z Set when interpolation happens during the transform to RGBA. Parameters ---------- s : {'data', 'rgba'} or None Whether to apply up/downsampling interpolation in data or rgba space. Nr2rn)rT)rr}rrrr8r8r9r s z"_ImageBase.set_interpolation_stagecCs|}|jdko|jo|jS)z>Return whether the image can be composited with its neighbors.r)rrrZ is_separable)rr5r8r8r9rPs  z_ImageBase.can_compositecCs"|dkrtjd}||_d|_dS)z Set whether image resampling is used. Parameters ---------- v : bool or None If None, use :rc:`image.resample`. Nzimage.resampleT)r{r|rkr)rvr8r8r9r"s  z_ImageBase.set_resamplecCs|jS)z(Return whether image resampling is used.)rkrr8r8r9ra0sz_ImageBase.get_resamplecCst||_d|_dS)z Set whether the resize filter normalizes the weights. See help for `~.Axes.imshow`. Parameters ---------- filternorm : bool TN)bool _filternormr)rrr8r8r9r~4s z_ImageBase.set_filternormcCs|jS)z8Return whether the resize filter normalizes the weights.)rrr8r8r9rcAsz_ImageBase.get_filternormcCs(t|}|dkrtd||_d|_dS)z Set the resize filter radius only applicable to some interpolation schemes -- see help for imshow Parameters ---------- filterrad : positive float rz+The filter radius must be a positive numberTN)rr _filterradr)rrrr8r8r9rEs z_ImageBase.set_filterradcCs|jS)zReturn the filterrad setting.)rrr8r8r9rdTsz_ImageBase.get_filterrad)NNNNTrpF)rFT)rF)!r __module__ __qualname____doc__zorderryrrrrr#rrr!rrwZallow_rasterizationrArrrrr\rrrPrrar~rcrrd __classcell__r8r8rr9r=sT       0   r=c sjeZdZdZejdddddd fd d Zdd d ZdddZddZ ddZ ddZ ddZ Z S) AxesImagea> An image attached to an Axes. Parameters ---------- ax : `~.axes.Axes` The axes the image will belong to. cmap : str or `~matplotlib.colors.Colormap`, default: :rc:`image.cmap` The Colormap instance or registered colormap name used to map scalar data to colors. norm : str or `~matplotlib.colors.Normalize` Maps luminance to 0-1. interpolation : str, default: :rc:`image.interpolation` Supported values are 'none', 'antialiased', 'nearest', 'bilinear', 'bicubic', 'spline16', 'spline36', 'hanning', 'hamming', 'hermite', 'kaiser', 'quadric', 'catrom', 'gaussian', 'bessel', 'mitchell', 'sinc', 'lanczos', 'blackman'. interpolation_stage : {'data', 'rgba'}, default: 'data' If 'data', interpolation is carried out on the data provided by the user. If 'rgba', the interpolation is carried out after the colormapping has been applied (visual interpolation). origin : {'upper', 'lower'}, default: :rc:`image.origin` Place the [0, 0] index of the array in the upper left or lower left corner of the axes. The convention 'upper' is typically used for matrices and images. extent : tuple, optional The 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. filternorm : bool, default: True A parameter for the antigrain image resize filter (see the antigrain documentation). If filternorm is set, the filter normalizes integer values and corrects the rounding errors. It doesn't do anything with the source floating point values, it corrects only integers according to the rule of 1.0 which means that any sum of pixel weights must be equal to 1.0. So, the filter function must produce a graph of the proper shape. filterrad : float > 0, default: 4 The filter radius for filters that have a radius parameter, i.e. when interpolation is one of: 'sinc', 'lanczos' or 'blackman'. resample : bool, default: False When True, use a full resampling method. When False, only resample when the output image is larger than the input image. **kwargs : `.Artist` properties 3.6rnameNTrpFrqc  s0||_tj|f||||||| | d| dS)N)rrrhrvrrrrr)_extentrry) rrrrrhrvextentrrrrrrrr8r9rys zAxesImage.__init__cCs.|j\}}}}t||||g}||jjSr;)rr Z from_extentsZ transformedrZ transData)rr.r+x1r,y1r6r8r8r9get_window_extentszAxesImage.get_window_extentrc Cst|}|\}}}}tt||g||gg} t| |} |rT|pZ|jj n|j j } |j |j | | | ||dS)Nr) rrr rr]rrQZ get_clip_boxrr6figurerr) rr.r/rr5rZx2ry2r6Ztransformed_bboxrr8r8r9r!s zAxesImage.make_imagecCs |dkS)z9Return whether the image would be better drawn unsampled.r)r\rr8r8r9rsz AxesImage._check_unsampled_imagecCs||_\}}}}||f||ff}|j|||g|jjdd<||g|jjdd<|jrt|jj||fdd|jr|jj ||fddd|_ dS)a( Set the image extent. Parameters ---------- extent : 4-tuple of float The position and size of the image as tuple ``(left, right, bottom, top)`` in data coordinates. Notes ----- This updates ``ax.dataLim``, and, if autoscaling, sets ``ax.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) rrZupdate_datalimZ sticky_edgesr3r4Zget_autoscalex_onZset_xlimZget_autoscaley_onZset_ylimr)rrrrrrZcornersr8r8r9 set_extents   zAxesImage.set_extentcCsV|jdk r|jS|}|\}}|jdkr>d|d|ddfSd|dd|dfSdS)  (  zPcolorImage.make_imagecCsdS)NFr8rr8r8r9rsz"PcolorImage._check_unsampled_imagecCstj|dd}|dkr4tjd|jddtjd}nt|tj}|dkrltjd|jddtjd}nt|tj}|jdd|jd|jdfkrt d|jdd|jd|jdff|j d krt d |j d kr|jddkr|j d d }n|jddkrt d|d |dkrT|ddd }|ddddd f}|d |dkr|ddd }|ddd }||_ ||_ ||_d|_d|_dS)aq Set the grid for the rectangle boundaries, and the data values. Parameters ---------- x, y : 1D array-like, optional Monotonic arrays of length N+1 and M+1, respectively, specifying rectangle boundaries. If not given, will default to ``range(N + 1)`` and ``range(M + 1)``, respectively. A : array-like The data to be color-coded. The interpretation depends on the shape: - (M, N) ndarray or masked array: values to be colormapped - (M, N, 3): RGB array - (M, N, 4): RGBA array TrNrrrrVz2Axes don't match array shape. Got %s, expected %s.r1zA must be 2D or 3DrUrr2rz+3D arrays must have RGB or RGBA as last dim)rrrZaranger$rr]r#rBrrr3rrrrrr4r8r8r9rs>"   zPcolorImage.set_datacGs tddSr5rr6r8r8r9r&szPcolorImage.set_arraycCs|j|j}}||jdksF||jdksF||jdksF||jdkrJdSt|j|d}t|j|d}z|j||fWStk rYdSXdS)Nrrr)ZxdataZydatarrrr r IndexError)rrr3r4rrr8r8r9r)s  zPcolorImage.get_cursor_data)NNNNN)rF) rrrrrrryr!rrrrrr8r8rr9r:s % !6r:csReZdZdZdZdZejddddfdd Zd d Z dddZ ddZ Z S) FigureImagezAn image attached to a figure.rrrrrNc s:tjd|||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)rrrvr)rryroxoyrr/)rfigrrZoffsetxZoffsetyrvrrr8r9ry>s zFigureImage.__init__cCs8|\}}d|j|d|jd|j|d|jfS)r r r )rrBrC)rrrr8r8r9rYs   zFigureImage.get_extentrFc Cs|j|jj}t|j||j|g|j||jjd|j||jjdgg}|j\}}||j9}||j9}tddg||gg}|j|j||||||ddS)NrrF)r/rr) dpirr rBrCrr$Zget_size_inchesr) rr.r/rZfacr6r)r(rr8r8r9r!_s&  zFigureImage.make_imagecCstj||d|_dS)zSet the image array.TN)rrzrrrr8r8r9rpszFigureImage.set_data)NNrrN)rF) rrrrrrrrryrr!rrr8r8rr9r@7s  r@csLeZdZdZejddddfd d Zdd d Zd dZdddZ Z S) BboxImagez;The Image class whose size is determined by the given bbox.rrrNTrpFc s,tjd|||||||d| ||_dS)rAN)rrrhrvrrr)N)rryr6) rr6rrrhrvrrrrrr8r9ryys  zBboxImage.__init__cCsF|dkr|}t|jtr&|jSt|jr:||StddS)NzUnknown type of bbox)Z get_figureZ _get_rendererr<r6r callabler)rr.r8r8r9rs    zBboxImage.get_window_extentcCsT||\}}|dk r||fS|s.difS|j|j}}|||}|ifSr)rrr3r4rr)rrrrr3r4r8r8r9rszBboxImage.containsrc Csl|\}}||}|j||g_||}tddg||gg}t||_|j|j|||||dS)Nrr) Zget_canvas_width_heightrrZ_pointsr r Z _transformrr) rr.r/rr)r(Zbbox_inZbbox_outrr8r8r9r!s   zBboxImage.make_image)NNNNTrpF)N)rF) rrrrrrryrrr!rr8r8rr9rFvs  rFc Csddlm}|dkrt|trT||}t|jdkrr8r8r9r@zszimsave..RGBrYrE)*matplotlib.figurerZr<osPathLikefspathrKrrNr{r|rurZfigimagesavefig memoryviewrYrr$rrzZset_climrrr frombufferrr __version__rPZPngInfoitemsadd_textrZ _str_equalrrZto_rgbnewZpaste setdefaultr)rrrrrrYrvrErWrXrZrDrnsmZ pil_shaper1r`krZ facecolorcolorZ backgroundr8r8r9imsave s2                     rrc Cs|jdkrt|S|jdrp|d|j}|jdrHt|d}n t|d}||jddd d Sz| d }Wn,t k r}zt d |W5d}~XYnXt|SdS) a` Load a `PIL image`_ and return it as a numpy int 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. )r^ZRGBXrbLzI;16r_r)z>u2zs     $  9 0 8w(?G N x#