B ]t\R@sZdZy ddlZWnek r,ddlZYnXddlZddlmZddlm Z m Z m Z ddl m Z mZmZddlmZmZmZmZddlmZmZddlmZmZmZmZdd lmZdd lm Z dd l!m"Z"m#Z#dd l m$Z%dd l&m'Z(ddl m)Z)ddlm*Z*e*rddl+m,Z,dZ-ddZ.GdddeZ'GdddeZ/ej0GdddeZ1dS)a An agg http://antigrain.com/ backend Features that are implemented * capstyles and join styles * dashes * linewidth * lines, rectangles, ellipses * clipping to a rectangle * output to RGBA and PNG, optionally JPEG and TIFF * alpha blending * DPI scaling properly - everything scales properly (dashes, linewidths, etc) * draw polygon * freetype2 w/ ft2font TODO: * integrate screen dpi w/ ppi and text N) OrderedDict)radianscossin)cbookrcParams __version__)_BackendFigureCanvasBaseFigureManagerBase RendererBase)findfontget_font)LOAD_FORCE_AUTOHINTLOAD_NO_HINTING LOAD_DEFAULTLOAD_NO_AUTOHINT)MathTextParser)Path)BboxBboxBase)colors) RendererAgg)_png)_has_pil)Imagezv2.2cCsttttttd}|tdS)N)TFZeitherZnativeautoZnonez text.hinting)rrrrr)mappingr>lib/python3.7/site-packages/matplotlib/backends/backend_agg.pyget_hinting_flag2sr c@seZdZdZeZddZddZddZ dd Z d d Z d1d dZ ddZ d2ddZddZd3ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd4d+d,Zd-d.Zd/d0Zd S)5rz The renderer handles all the drawing primitives using a graphics context instance that controls the colors/styles cCsdt|||_||_||_tt|t|||_g|_| t d|_ t dd|j|j|_dS)NZAggr)r __init__dpiwidthheight _RendererAggint _renderer_filter_renderers_update_methodsrmathtext_parserrZ from_boundsbbox)selfr#r$r"rrrr!Qs  zRendererAgg.__init__cCs|j|j|jdS)N)r#r$r")r#r$r")r,rrr __getstate___szRendererAgg.__getstate__cCs||d|d|ddS)Nr#r$r")r!)r,staterrr __setstate__dszRendererAgg.__setstate__cCsT|jj|_|jj|_|jj|_|jj|_|jj|_|jj|_|jj|_|jj|_dS)N) r'Zdraw_gouraud_triangleZdraw_gouraud_triangles draw_imageZ draw_markersZdraw_path_collectionZdraw_quad_meshcopy_from_bboxget_content_extents)r,rrrr)gs       zRendererAgg._update_methodscCs\|}|d|j|d|dg|d|d|j|dgg}||}t||fS)Nr)r2r$r1nparray)r,extentsr+regionrrrtostring_rgba_minimizedqs   z#RendererAgg.tostring_rgba_minimizedNc Csntd}|jjd}|dkr4||kr4|jr4|dkr4|dkr4t||}tt||}td||} t | } | ddd| dd<|| d<xt | | D]\} } |j| | ddf} |j }|dk r|| | }t j |d<t | |}y|j||||Wqtk r,tdYqXqWn6y|j||||Wntk rhtdYnXdS)z Draw the path zagg.path.chunksizerdNr3z Draw the math text using matplotlib.mathtext r3N) r*parser"rrrr6roundr'draw_text_image)r,rAxyspropangleoxoyr#r$descentZ font_imageused_charactersxdydrrr draw_mathtexts  zRendererAgg.draw_mathtextFc Cs|r|||||||St} ||} | dkr4dSt|dkr`t|dkr`| jt|| dn| j|d| d| jtdd| d} | \} } | d} | d} | t t |}| t t |}|j| t||| t||| d||dS) z! Render the text Nr3)flagsrztext.antialiased)Z antialiasedgP@)rTr _get_agg_fontlenordZ load_charset_textZdraw_glyphs_to_bitmapr get_descentZget_bitmap_offsetrrrr'rHr6rG)r,rArIrJrKrLrMismathmtextrVfontdZxoZyorRrSrrr draw_texts$   zRendererAgg.draw_textcCs|dkr@|}|}|}|j|||d\}}} ||| fS|rn|j||j|\} } } } }}}| | |fSt}||}|j|d|d| \}}| } |d}|d}| d} ||| fS)z Get the width, height, and descent (offset from the bottom to the baseline), in display coords, of the string *s* with :class:`~matplotlib.font_manager.FontProperties` *prop* )ZTeXzTeX!)rendererg)rVgP@) get_size_in_pointsget_texmanagerget_text_width_height_descentr*rFr"r rWrZZget_width_heightr[)r,rKrLr\size texmanagerZfontsizewhr_rNrOr#r$rPZfontsrQrVr^rrrrds&     z)RendererAgg.get_text_width_height_descentTeX!c Cs|} |} | || |j} t| dtj} ||||\} } }|tt |}|t t |}t ||}t ||}|j | ||||dS)Ngo@)rbrcZget_greyr"r6r7uint8rdrrrrGr'rH)r,rArIrJrKrLrMr\r]rerfZrgrhr_rRrSrrrdraw_texszRendererAgg.draw_texcCs |j|jfS)z4return the canvas width and height in display coords)r#r$)r,rrrget_canvas_width_heightsz#RendererAgg.get_canvas_width_heightcCs2t|}t|}||}|||j|S)zJ Get the font for text instance t, caching for efficiency )r rclearrbZset_sizer")r,rLfnamer^rerrrrWs zRendererAgg._get_agg_fontcCs||jdS)zj convert point measures to pixes using dpi and the pixels per inch of the display H)r")r,Zpointsrrrpoints_to_pixelsszRendererAgg.points_to_pixelscCs |jS)N)r' tostring_rgb)r,rrrrr szRendererAgg.tostring_rgbcCs |jS)N)r' tostring_argb)r,rrrrsszRendererAgg.tostring_argbcCs |jS)N)r' buffer_rgba)r,rrrrtszRendererAgg.buffer_rgbacCs|jdS)N)r'rn)r,rrrrnszRendererAgg.clearcCsdS)NTr)r,rrroption_image_nocompositesz$RendererAgg.option_image_nocompositecCsdS)zI agg backend doesn't support arbitrary scaling of image. Fr)r,rrroption_scale_imageszRendererAgg.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)a: 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 tuple of two floasts) 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 isinstancerr8r'restore_regionr&) r,r9r+xyZx1Zy1Zx2Zy2rNrOrrrrx$s   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))r,rrr start_filterGs zRendererAgg.start_filterc Cst|jt|j}}|\}\}}}}|j|_||dkr|dkrt |tj } || ||dfd|j \} } } | } | jjdkrt| dtj } | ddd} |j| || |||| | dS)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. rgo@fNr<)r&r#r$r:r(popr'r)r6Z fromstringrjZreshaper"Znew_gcZdtypeZkindZasarrayr0) r,Zpost_processingr#r$bufferlbrgrhZimgrNrOrArrr stop_filterPs  zRendererAgg.stop_filter)N)FN)riN)NN)__name__ __module__ __qualname____doc__ threadingRLocklockr!r-r/r)r:r?rTr`rdrlrmrWrqrrrsrtrnrurvrxr{rrrrrr>s0   "    # rcseZdZdZddZdddZfddZdd d Zd d ZddZ ddZ ddZ e Z ddZ ddZerd dddZeZd dddZeZZS)FigureCanvasAggz The canvas the figure renders into. Calls the draw and print fig methods, creates the renderers, etc... Attributes ---------- figure : `matplotlib.figure.Figure` A high-level Figure instance cCs|}||S)N) get_rendererr1)r,r+rarrrr1szFigureCanvasAgg.copy_from_bboxNcCs|}||||S)N)rrx)r,r9r+ryrarrrrxszFigureCanvasAgg.restore_regioncsL|jdd|_tj|j}z|j|jtWdtj XdS)z5 Draw the figure using the renderer. T)clearedN) rrarracquiretoolbarfiguredrawsuperrelease)r,r) __class__rrrs zFigureCanvasAgg.drawFcCs|jjj\}}}}|||jjf}y|j|jfWntk rHd}Yn X|j|k}|rrt|||jj|_||_n|r|j|jS)NT) rr+Zboundsr"Z_lastKeyraAttributeErrorrrn)r,rrrrgrhkeyZneed_new_rendererrrrrs   zFigureCanvasAgg.get_renderercCs |jS)zGet the image as an RGB byte string. `draw` must be called at least once before this function will work and to update the renderer for any subsequent changes to the Figure. Returns ------- bytes )rarr)r,rrrrrs zFigureCanvasAgg.tostring_rgbcCs |jS)zGet the image as an ARGB byte string `draw` must be called at least once before this function will work and to update the renderer for any subsequent changes to the Figure. Returns ------- bytes )rars)r,rrrrss zFigureCanvasAgg.tostring_argbcCs |jS)zGet the image as an RGBA byte string. `draw` must be called at least once before this function will work and to update the renderer for any subsequent changes to the Figure. Returns ------- bytes )rart)r,rrrrts zFigureCanvasAgg.buffer_rgbac Os\t||}tj||jjd.t|d}||j WdQRXWdQRXdS)N)r"wb) rrrr _setattr_cmrr" open_file_cmwriter'rt)r,filename_or_objargskwargsrafhrrr print_raws  zFigureCanvasAgg.print_rawc Ost||}dtd}td|i}|dd}|dk rH||tj||j j d6t |d}t j |j||j j |dWdQRXWdQRXdS) a Write the figure to a PNG file. Parameters ---------- filename_or_obj : str or PathLike or file-like object 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. For more details see the `PNG specification`_. .. _PNG specification: https://www.w3.org/TR/2003/REC-PNG-20031110/#11keywords zmatplotlib version z, http://matplotlib.org/ZSoftwaremetadataN)r"r)r)rrrrrr~updaterrrr"rrZ write_pngr') r,rrrraZ version_strrZ user_metadatarrrr print_pngs*      zFigureCanvasAgg.print_pngc CsNt||}tj||jjd |jt |j t |j ffSQRXdS)N)r") rrrrrrr"r'rtr&r#r$)r,rarrrprint_to_buffer s  zFigureCanvasAgg.print_to_buffer)dryrunc s|\}}|rdStd||dddd}ttd}tdd|dd D} td || } | ||fd d d D} | dtdd| kr| d| df| d<| j |fddi| S)a Write the figure to a JPEG file. Parameters ---------- filename_or_obj : str or PathLike or file-like object The file to write to. Other Parameters ---------------- quality : int The image quality, on a scale from 1 (worst) to 100 (best). The default is :rc:`savefig.jpeg_quality`. Values above 95 should be avoided; 100 completely disables the JPEG quantization stage. optimize : bool If present, indicates that the encoder should make an extra pass over the image in order to select optimal encoder settings. progressive : bool If present, indicates that this image should be stored as a progressive JPEG file. NRGBArawrr3zsavefig.facecolorcSsg|]}t|dqS))r&).0rIrrr 7sz-FigureCanvasAgg.print_jpg..r4ZRGBcsi|]}|kr||qSrr)rk)rrr :sz-FigureCanvasAgg.print_jpg..)qualityoptimizeZ progressiver"rzsavefig.jpeg_qualityr"formatZjpeg) rr frombuffermcolorsZto_rgbartuplenewZpaste setdefaultsave) r,rrrrbufreimageZrgbaZcolorZ backgroundZoptionsr)rr print_jpgs   zFigureCanvasAgg.print_jpgc OsJ|\}}|rdStd||dddd}|jj|jjf}|j|d|dS)Nrrrr3Ztiff)rr")rrrrr"r) r,rrrrrrerr"rrr print_tifFs  zFigureCanvasAgg.print_tif)NN)F)rrrrr1rxrrrrrsrtrZ print_rgbarrrrZ print_jpegrZ print_tiff __classcell__rr)rrrts"       9-rc@seZdZeZeZdS) _BackendAggN)rrrrZ FigureCanvasr Z FigureManagerrrrrrPsr)2rr ImportErrorZdummy_threadingZnumpyr6 collectionsrZmathrrrZ matplotlibrrrZmatplotlib.backend_basesr r r r Zmatplotlib.font_managerr rZmatplotlib.ft2fontrrrrZmatplotlib.mathtextrZmatplotlib.pathrZmatplotlib.transformsrrrrZ matplotlib.backends._backend_aggrr%rrZPILrZbackend_versionr rZexportrrrrrs:          8]