U \ akY@sDdZz ddlZWnek r,ddlZYnXddlmZddlmZmZm Z ddl Z ddl m Z ddlZddlmZmZddlmZddlmZmZmZmZmZdd lmZmZdd lmZmZm Z m!Z!dd l"m#Z#dd l$m%Z%dd l&m'Z'm(Z(ddl)m*Z+dZ,ddZ-GdddeZ*GdddeZ.ej/GdddeZ0dS)a An `Anti-Grain Geometry `_ (AGG) backend. Features that are implemented: * capstyles and join styles * dashes * linewidth * lines, rectangles, ellipses * clipping to a rectangle * output to RGBA and Pillow-supported image formats * alpha blending * DPI scaling properly - everything scales properly (dashes, linewidths, etc) * draw polygon * freetype2 w/ ft2font Still TODO: * integrate screen dpi w/ ppi and text N) nullcontext)radianscossin)Image)_apicbook)colors)_Backend_check_savefig_extra_argsFigureCanvasBaseFigureManagerBase RendererBase)findfontget_font)LOAD_FORCE_AUTOHINTLOAD_NO_HINTING LOAD_DEFAULTLOAD_NO_AUTOHINT)MathTextParser)Path)BboxBboxBase) RendererAggzv2.2c Cs(ttttttttttd }|tjdS)N) defaultZ no_autohintZforce_autohintZ no_hintingTFZeitherZnativeautononez text.hinting)rrrrmplrcParams)mappingr C/tmp/pip-unpacked-wheel-wjyw_3jo/matplotlib/backends/backend_agg.pyget_hinting_flag2s r"cseZdZdZeZfddZddZddZ dd Z e d d d Z e d d dZd6ddZddZddZd7ddZddZddddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd8d0d1Zd2d3Zd4d5Z Z!S)9rz The renderer handles all the drawing primitives using a graphics context instance that controls the colors/styles csdt||_||_||_tt|t|||_g|_| t d|_ t dd|j|j|_dS)NZAggr)super__init__dpiwidthheight _RendererAggint _renderer_filter_renderers_update_methodsrmathtext_parserrZ from_boundsbbox)selfr&r'r% __class__r r!r$Us  zRendererAgg.__init__cCs|j|j|jdS)Nr&r'r%r2r/r r r! __getstate__cszRendererAgg.__getstate__cCs||d|d|ddS)Nr&r'r%)r$)r/stater r r! __setstate__hszRendererAgg.__setstate__cCs@|jj|_|jj|_|jj|_|jj|_|jj|_|jj|_dSN)r*Zdraw_gouraud_triangleZdraw_gouraud_triangles draw_imageZ draw_markersZdraw_quad_meshcopy_from_bboxr3r r r!r,ks      zRendererAgg._update_methodsz3.4cCs@t|}t|d\}}|j|j|j|j|j|jfS)N.)npasarray buffer_rgbar_get_nonzero_slicesstartstop)r/orig_imgslice_yslice_xr r r!get_content_extentsvs  zRendererAgg.get_content_extentscCs\|}|d|j|d|dg|d|d|j|dgg}||}t||fS)Nrr;)rEr'r9r<array)r/extentsr.regionr r r!tostring_rgba_minimized}s  z#RendererAgg.tostring_rgba_minimizedNc Cstjd}|jjd}||kr,dkrJnn|jrJ|dkrJ|dkrJt||}tt||}t d||} t | } | ddd| dd<|| d<t | | D]\} } |j| | ddf} |j }|dk r|| | }t j|d<t | |}z|j||||Wqtk rD}ztd|W5d}~XYqXqnFz|j||||Wn.tk r}ztd|W5d}~XYnXdS)Nzagg.path.chunksizerdrFzszRendererAgg.buffer_rgbacCs"t|jjddddgddS)Nr;rrFrGZaxisr<r=r*Ztaketobytesr3r r r! tostring_argb!szRendererAgg.tostring_argbcCs t|jjdddgddS)NrrFrGrrr3r r r! tostring_rgb$szRendererAgg.tostring_rgbcCs|jdSr7)r*rr3r r r!r'szRendererAgg.clearcCsdS)NTr r3r r r!option_image_nocomposite*sz$RendererAgg.option_image_nocompositecCsdS)NFr r3r r r!option_scale_image2szRendererAgg.option_scale_imagec Cs|dk s|dk r|dkr*|\}}}}n&t|trD|j\}}}}n |\}}}}|dkrd||}} n|\}} |j|t|t|t|t|t|t| n |j|dS)a4 Restore the saved region. If bbox (instance of BboxBase, or its extents) is given, only the region specified by the bbox will be restored. *xy* (a pair of floats) optionally specifies the new position (the LLC of the original region, not the LLC of the bbox) where the region will be restored. >>> region = renderer.copy_from_bbox() >>> x1, y1, x2, y2 = region.get_extents() >>> renderer.restore_region(region, bbox=(x1+dx, y1, x2, y2), ... xy=(x1-dx, y1)) N)Z get_extents isinstancerrIr*restore_regionr)) r/rJr.xyx1y1Zx2y2rlrmr r r!r6s    zRendererAgg.restore_regioncCs6|j|jtt|jt|j|j|_|dS)zX Start filtering. It simply create a new canvas (the old one is saved). N) r+appendr*r(r)r&r'r%r,r3r r r! start_filterYs zRendererAgg.start_filterc Cst|}t|d\}}|||f}|j|_||j r||d|j \}}}| } |j j dkrt|dtj}|j| |j|t|j|j||ddddS)a Save the plot in the current canvas as a image and apply the *post_processing* function. def post_processing(image, dpi): # ny, nx, depth = image.shape # image (numpy array) has RGBA channels and has a depth of 4. ... # create a new_image (numpy array of 4 channels, size can be # different). The resulting image may have offsets from # lower-left corner of the original image return new_image, offset_x, offset_y The saved renderer is restored and the returned image from post_processing is plotted (using draw_image) on it. r:frNrM)r<r=r>rr?r+popr*r,rr%Znew_gcZdtypekindrr8r@r)r'rA) r/Zpost_processingrBrCrDZ cropped_imgimgrlrmrTr r r! stop_filterbs     zRendererAgg.stop_filter)N)FN)NN)"__name__ __module__ __qualname____doc__ threadingRLocklockr$r4r6r,r deprecatedrErKrRr`rrr}rrrrvrr>rrrrrrrr __classcell__r r r0r!rBs8     !    # rcseZdZddZd&ddZfddZd'd d Zd d Zd dZddZ e ddZ e Z e dddddZ ddZe dddgdejddddejddddejddd ddd!d"d#ZeZe dd!d$d%ZeZZS)(FigureCanvasAggcCs|}||Sr7) get_rendererr9)r/r.rr r r!r9szFigureCanvasAgg.copy_from_bboxNcCs|}||||Sr7)rr)r/rJr.rrr r r!rszFigureCanvasAgg.restore_regionc s^|jdd|_tj@|jr&|jnt|j|jt W5QRXW5QRXdS)NT)cleared) rrrrZtoolbarZ_wait_cursor_for_draw_cmrfiguredrawr#r3r0r r!rszFigureCanvasAgg.drawFcCsh|jjj\}}|||jjf}t|do4t|dd|k}|sTt|||jj|_||_n|rb|j |jS)Nr_lastKey) rr.rr%hasattrgetattrrrrr)r/rrrkeyZreuse_rendererr r r!rs  zFigureCanvasAgg.get_renderercCs |jS)z Get the image as RGB `bytes`. `draw` must be called at least once before this function will work and to update the renderer for any subsequent changes to the Figure. )rrr3r r r!rszFigureCanvasAgg.tostring_rgbcCs |jS)z Get the image as ARGB `bytes`. `draw` must be called at least once before this function will work and to update the renderer for any subsequent changes to the Figure. )rrr3r r r!rszFigureCanvasAgg.tostring_argbcCs |jS)z Get the image as a `memoryview` to the renderer's buffer. `draw` must be called at least once before this function will work and to update the renderer for any subsequent changes to the Figure. )rr>r3r r r!r>szFigureCanvasAgg.buffer_rgbac Gs<t||}t|d}||W5QRXdS)Nwb)rrrrZ open_file_cmwriter>)r/filename_or_objargsrfhr r r! print_raws zFigureCanvasAgg.print_raw)metadata pil_kwargsc Gs0t|tjj||dd|jj||ddS)a Write the figure to a PNG file. Parameters ---------- filename_or_obj : str or path-like or file-like The file to write to. metadata : dict, optional Metadata in the PNG file as key-value pairs of bytes or latin-1 encodable strings. According to the PNG specification, keys must be shorter than 79 chars. The `PNG specification`_ defines some common keywords that may be used as appropriate: - Title: Short (one line) title or caption for image. - Author: Name of image's creator. - Description: Description of image (possibly long). - Copyright: Copyright notice. - Creation Time: Time of original image creation (usually RFC 1123 format). - Software: Software used to create the image. - Disclaimer: Legal disclaimer. - Warning: Warning of nature of content. - Source: Device used to create the image. - Comment: Miscellaneous comment; conversion from other image format. Other keywords may be invented for other purposes. If 'Software' is not given, an autogenerated value for Matplotlib will be used. This can be removed by setting it to *None*. For more details see the `PNG specification`_. .. _PNG specification: https://www.w3.org/TR/2003/REC-PNG-20031110/#11keywords pil_kwargs : dict, optional Keyword arguments passed to `PIL.Image.Image.save`. If the 'pnginfo' key is present, it completely overrides *metadata*, including the default 'Software' key. Zpngupper)formatoriginr%rrN)rrrimageZimsaver>rr%)r/rrrrr r r! print_pngs1 zFigureCanvasAgg.print_pngcCs2t||}t|t|jt|jffSr7)rrrbytesr>r)r&r')r/rr r r!print_to_buffers   zFigureCanvasAgg.print_to_bufferqualityoptimize progressive) extra_kwargsr\zpil_kwargs={'quality': ...}) alternativezpil_kwargs={'optimize': ...}zpil_kwargs={'progressive': ...})rc Ost|j\}}}}z2|j|t|||gd|t|W5|j||||fX|dkrni}dD]} | |krr| | | | qrd|krt t jd} |d<| dkrtjdddd d | d |jj|jjftt|d dd fj|fddi|S)a Write the figure to a JPEG file. Parameters ---------- filename_or_obj : str or path-like or file-like The file to write to. Other Parameters ---------------- quality : int, default: :rc:`savefig.jpeg_quality` The image quality, on a scale from 1 (worst) to 95 (best). Values above 95 should be avoided; 100 disables portions of the JPEG compression algorithm, and results in large files with hardly any gain in image quality. This parameter is deprecated. optimize : bool, default: False Whether the encoder should make an extra pass over the image in order to select optimal encoder settings. This parameter is deprecated. progressive : bool, default: False Whether the image should be stored as a progressive JPEG file. This parameter is deprecated. pil_kwargs : dict, optional Additional keyword arguments that are passed to `PIL.Image.Image.save` when saving the figure. These take precedence over *quality*, *optimize* and *progressive*. rFN)rrrrzsavefig.jpeg_quality)rK_r\ZrcParamzSet the quality using `pil_kwargs={'quality': ...}`; the future default quality will be 75, matching the default of Pillow and libjpeg.)nameZobj_typeZaddendumr%.r;rZjpeg)mcolorsZto_rgbarZ get_facecolorZ set_facecolorr<rHrr setdefaultrdict __getitem__rrrr_r%r fromarrayr=r>save) r/rrrkwargsrgbakrr r r! print_jpg s8&$  zFigureCanvasAgg.print_jpgcCsTt||dkri}|d|jj|jjftt| j |fddi|S)Nr%rZtiff) rrrrr%rrr<r=r>r)r/rrr r r! print_tifMs zFigureCanvasAgg.print_tif)NN)F)rrrr9rrrrrr>r rZ print_rgbarrrZdelete_parameterrZ print_jpegrZ print_tiffrr r r0r!rs@      5 8rc@seZdZeZeZdS) _BackendAggN)rrrrZ FigureCanvasr Z FigureManagerr r r r!rYsr)1rr ImportErrorZdummy_threading contextlibrmathrrrZnumpyr<ZPILrZ matplotlibrrrr rZmatplotlib.backend_basesr r r r rZmatplotlib.font_managerrrZmatplotlib.ft2fontrrrrZmatplotlib.mathtextrZmatplotlib.pathrZmatplotlib.transformsrrZ matplotlib.backends._backend_aggrr(Zbackend_versionr"rexportrr r r r!s6       DV