a ضaL@s*dZddlmZddlZddlmZddlZddlZddlZ ddl m Z ddl m Z mZmZmZmZddlmZmZddlmZeeZe jGd d d ZejZee_Gd d d ZGd ddeZ e j!dddGddde Z"e j!dddGdddeZ#e j!dddGdddeZ$e j!dddGdddeZ%GdddeZ&e j!dddGdddeZ'dgdd ej()Dd!d"d ej*)Dd#d$d ej+)Dd%d&D]*Z,e !de-e,e.ee,fie/e,<qGd'd(d(e0Z1e !dd)d*Z2Gd+d,d,Z3d/d-d.Z4dS)0a0 A module for parsing a subset of the TeX math syntax and rendering it to a Matplotlib backend. For a tutorial of its usage, see :doc:`/tutorials/text/mathtext`. This document is primarily concerned with implementation details. The module uses pyparsing_ to parse the TeX expression. .. _pyparsing: https://pypi.org/project/pyparsing/ The Bakoma distribution of the TeX Computer Modern fonts, and STIX fonts are supported. There is experimental support for using arbitrary fonts, but results may vary without proper tweaking and metrics for those fonts. ) namedtupleN)StringIO)Image)_apicolorsrcParams _mathtext_mathtext_data)FT2ImageLOAD_NO_HINTING)FontPropertiesc@s6eZdZeddegdfegdffDdS) __getattr__c Cs8i|]0\}}|D]"}|tdt||fddqqS)3.4cSs t||SN)getattr)selfZ_mod_namerc/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/matplotlib/mathtext.py'z__getattr__..)r deprecatedproperty).0modnamesnamerrr %s  z__getattr__.)Z SHRINK_FACTORZ GROW_FACTORZNUM_SIZE_LEVELS)Zlatex_to_bakomaZ latex_to_cmexZlatex_to_standardZstix_virtual_fontsZtex2uniN)__name__ __module__ __qualname__localsupdaterr rrrrr #s    r c@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)MathtextBackenda The base class for the mathtext backend-specific code. `MathtextBackend` subclasses interface between mathtext and specific Matplotlib graphics backends. Subclasses need to override the following: - :meth:`render_glyph` - :meth:`render_rect_filled` - :meth:`get_results` And optionally, if you need to use a FreeType hinting style: - :meth:`get_hinting_type` cCsd|_d|_d|_dSNrwidthheightdepthrrrr__init__DszMathtextBackend.__init__cCs||_||_||_dS)z(Set the dimension of the drawing canvas.Nr%rwhdrrrset_canvas_sizeIszMathtextBackend.set_canvas_sizecCs tdS)z_ Draw a glyph described by *info* to the reference point (*ox*, *oy*). NNotImplementedErrorroxoyinforrr render_glyphOszMathtextBackend.render_glyphcCs tdS)zR Draw a filled black rectangle from (*x1*, *y1*) to (*x2*, *y2*). Nr0rx1y1x2y2rrrrender_rect_filledVsz"MathtextBackend.render_rect_filledcCs tdS)zp Return a backend-specific tuple to return to the backend after all processing is done. Nr0)rboxrrr get_results\szMathtextBackend.get_resultscCstS)z\ Get the FreeType hinting type to use with this particular backend. )r r)rrrget_hinting_typecsz MathtextBackend.get_hinting_typeN) rrr __doc__r*r/r6r<r>r?rrrrr#4sr#csTeZdZdZfddZddZfddZdd Zd d Zd d Z ddZ Z S)MathtextBackendAggz Render glyphs and rectangles to an FTImage buffer, which is later transferred to the Agg image by the Agg backend. cs0d|_d|_d|_d|_gd|_tdS)Nrbbox)rrrr)r3r4imagemoderBsuperr*r) __class__rrr*ps  zMathtextBackendAgg.__init__cCsBt|jd|t|jd|t|jd|t|jd|g|_dS)Nr)minrBmaxr7rrr _update_bboxxs zMathtextBackendAgg._update_bboxcs@t||||jdkrs., zMathtextBackendAgg.get_resultscCsddlm}|S)Nr backend_agg)matplotlib.backendsrZZget_hinting_flag)rrZrrrr?s z#MathtextBackendAgg.get_hinting_type) rrr r@r*rMr/r6r<r>r? __classcell__rrrFrrAks    rArmathtext.math_to_image alternativecseZdZfddZZS)MathtextBackendBitmapc s$t||\}}}}}}} ||fSr)rEr>) rr=rWr3r4r&r'r(rC charactersrFrrr>s z!MathtextBackendBitmap.get_results)rrr r>r\rrrFrr`sr`MathtextBackendPathcsBeZdZdZeddZfddZddZdd Zd d Z Z S) MathtextBackendPszT Store information to write a mathtext rendering to the PostScript backend. _PSResultz+width height depth pswriter used_characterscstt|_d|_dSr)rEr*rpswriterlastfontr)rFrrr*s zMathtextBackendPs.__init__cCsv|j||j}|j}|j}||f|jkrN||f|_|jd|d|d|j|dd|dd|jddS)N/z findfont z scalefont setfont f z moveto /z glyphshow )r'offsetpostscript_namefontsizerfrewrite symbol_name)rr3r4r5rkrlrrrr6s zMathtextBackendPs.render_glyphcCs.d||j|||||f}|j|dS)Nz%f %f %f %f rectfill )r'rerm)rr8r9r:r;psrrrr<sz$MathtextBackendPs.render_rect_filledcCs.tdd|||j|j|j|j|j|Sr$)rrVrdr&r'r(rerr=rWrrrr>s zMathtextBackendPs.get_results) rrr r@rrdr*r6r<r>r\rrrFrrcs rccsBeZdZdZeddZfddZddZdd Zd d Z Z S) MathtextBackendPdfzCStore information to write a mathtext rendering to the PDF backend. _PDFResultz/width height depth glyphs rects used_characterscstg|_g|_dSrrEr*glyphsrectsr)rFrrr*s zMathtextBackendPdf.__init__cCs:|jj}|j||j}|j||||j|j|jfdSr) rQfnamer'rjrtappendrlnumrn)rr3r4r5filenamerrrr6s zMathtextBackendPdf.render_glyphcCs&|j||j|||||fdSrrurwr'r7rrrr<sz%MathtextBackendPdf.render_rect_filledcCs2tdd|||j|j|j|j|j|j|Sr$)rrVrrr&r'r(rtrurprrrr>s zMathtextBackendPdf.get_results) rrr r@rrrr*r6r<r>r\rrrFrrqs rqcs8eZdZdZfddZddZddZdd ZZS) MathtextBackendSvgzQ Store information to write a mathtext rendering to the SVG backend. cstg|_g|_dSr)rEr* svg_glyphs svg_rectsr)rFrrr*s zMathtextBackendSvg.__init__cCs4|j||j}|j|j|j|j|||jfdSr)r'rjr|rwrQrlrxrPr2rrrr6szMathtextBackendSvg.render_glyphcCs*|j||j|d||||fdS)NrH)r}rwr'r7rrrr<sz%MathtextBackendSvg.render_rect_filledcCs:tdd|tj|j|jd}|j|j|j|j||fS)Nr)r|r}) rrVtypesSimpleNamespacer|r}r&r'r()rr=rWZ svg_elementsrrrr>s zMathtextBackendSvg.get_results rrr r@r*r6r<r>r\rrrFrr{ s  r{csBeZdZdZeddZfddZddZdd Zd d Z Z S) rbzY Store information to write a mathtext rendering to the text path machinery. _Resultzwidth height depth glyphs rectscstg|_g|_dSrrsr)rFrrr*2s zMathtextBackendPath.__init__cCs0|j||j}|j|j|j|j||fdSr)r'rjrtrwrQrlrxr2rrrr67sz MathtextBackendPath.render_glyphcCs&|j||j|||||fdSrrzr7rrrr<;sz&MathtextBackendPath.render_rect_filledcCs0tdd|||j|j|j|j|j|jSr$)rrVrr&r'r(rtrurprrrr>>s zMathtextBackendPath.get_results) rrr r@rrr*r6r<r>r\rrrFrrb*s   cs8eZdZdZfddZddZddZdd ZZS) MathtextBackendCairozS Store information to write a mathtext rendering to the Cairo backend. cstg|_g|_dSrrsr)rFrrr*Ns zMathtextBackendCairo.__init__cCs8||j|j}t|j}|j|j|j|||fdSr)rjr'chrrxrtrwrQrl)rr3r4r5Zthetextrrrr6Ss  z!MathtextBackendCairo.render_glyphcCs&|j|||j||||fdSrrzr7rrrr<Ysz'MathtextBackendCairo.render_rect_filledcCs,tdd||j|j|j|j|j|jfSr$)rrVr&r'r(rtrurprrrr>]s z MathtextBackendCairo.get_resultsrrrrFrrGs  rFontscCsg|] }|jqSrrrcrrr hrrFontConstantsBasecCsg|] }|jqSrrrrrrrjrNodecCsg|] }|jqSrrrrrrrlrZShipParserc@s eZdZdS)MathTextWarningN)rrr rrrrrssrcCst|||dSr)rrV)r3r4r=rrrrVwsrVc @seZdZdZeeeeee e edZ e j e je je je je jdZddZdddd d Zed d d ZejddddddZejddddddZejddddddZejdddd ddZdS)!MathTextParserN)bitmapZaggroZpdfsvgpathZcairomacosx)cmZ dejavuserifZ dejavusansZstixZstixsansZcustomcCs||_dS)z7Create a MathTextParser for the given backend *output*.N)lower_output)routputrrrr*szMathTextParser.__init__HF)_force_standard_ps_fontscCs$|rtjdddd|||||S)aZ Parse the given math expression *s* at the given *dpi*. If *prop* is provided, it is a `.FontProperties` object specifying the "default" font to use in the math expression, used for all non-math text. The results are cached, so multiple calls to `parse` with the same expression should be fast. rz3.5zMathtext using only standard PostScript fonts has been likely to produce wrong output for a while, has been deprecated in %(since)s and will be removed in %(removal)s, after which ps.useafm will have no effect on mathtext.)Zremovalmessage)rZwarn_deprecated _parse_cached)rsdpiproprrrrparses zMathTextParser.parse2c Cs|durt}|rtjntj|j|d}|j|j}|||}| }|j durbt |j _ |j ||||} || j| j| j|| S)N)Zfontset)r rZStandardPsFontsrZ check_getitem_font_type_mappingZget_math_fontfamily_backend_mappingrZget_size_in_points_parserrrGrr/r&r'r(r>) rrrrZforce_standard_ps_fontsZ fontset_classbackendZ font_outputrlr=rrrrs     zMathTextParser._parse_cachedrr]r^xcCs:|jdksJt|d}|j|||d\}}t||fS)a  Convert a mathtext string to a grayscale array and depth. Parameters ---------- texstr : str A valid mathtext string, e.g., r'IQ: $\sigma_i=15$'. dpi : float The dots-per-inch setting used to render the text. fontsize : int The font size in points Returns ------- array : 2D uint8 alpha Mask array of rasterized tex. depth : int Offset of the baseline from the bottom of the image, in pixels. rsizerr)rr rrNZasarrayrtexstrrrlrZftimager(rrrto_masks zMathTextParser.to_maskblackc Cs|j|||d\}}t|\}}} } tj|jd|jddftjd} d|| dddddf<d|| dddddf<d| | dddddf<|| ddddd f<| |fS) ao Convert a mathtext string to an RGBA array and depth. Parameters ---------- texstr : str A valid mathtext string, e.g., r'IQ: $\sigma_i=15$'. color : color The text color. dpi : float The dots-per-inch setting used to render the text. fontsize : int The font size in points. Returns ------- array : (M, N, 4) array RGBA color values of rasterized tex, colorized with *color*. depth : int Offset of the baseline from the bottom of the image, in pixels. )rrlrrH)ZdtypeNrIrJ)rmcolorsto_rgbarNZzerosshapeZuint8) rrcolorrrlxr(rgbaZRGBArrrrs"zMathTextParser.to_rgbacCs.|j||||d\}}t|j|dd|S)a/ Render a tex expression to a PNG file. Parameters ---------- filename A writable filename or fileobject. texstr : str A valid mathtext string, e.g., r'IQ: $\sigma_i=15$'. color : color The text color. dpi : float The dots-per-inch setting used to render the text. fontsize : int The font size in points. Returns ------- int Offset of the baseline from the bottom of the image, in pixels. )rrrlZpng)format)rrZ fromarraysave)rryrrrrlZrgbar(rrrto_pngs  zMathTextParser.to_pngcCs0|jdksJt|d}|j|||d\}}|S)a~ Get the depth of a mathtext string. Parameters ---------- texstr : str A valid mathtext string, e.g., r'IQ: $\sigma_i=15$'. dpi : float The dots-per-inch setting used to render the text. Returns ------- int Offset of the baseline from the bottom of the image, in pixels. rrr)rr rrrrr get_depth#s zMathTextParser.get_depth)rN)rr)rrr)rrr)rr)rrr rr`rArcrqr{rbrrrZ BakomaFontsZDejaVuSerifFontsZDejaVuSansFontsZ StixFontsZ StixSansFontsZ UnicodeFontsrr*r functools lru_cacherrrrrrrrrrrrs:         rc Csddlm}ddlm}|dur&t}td}|j|d|d\}} } } } |j|d| dfd } | jd| | ||d | | | j |||d | S) a Given a math expression, renders it in a closely-clipped bounding box to an image file. Parameters ---------- s : str A math expression. The math portion must be enclosed in dollar signs. filename_or_obj : str or path-like or file-like Where to write the image data. prop : `.FontProperties`, optional The size and style of the text. dpi : float, optional The output dpi. If not set, the dpi is determined as for `.Figure.savefig`. format : str, optional The output format, e.g., 'svg', 'pdf', 'ps' or 'png'. If not set, the format is determined as for `.Figure.savefig`. r)figurerYNrrrgR@)Zfigsize)Zfontproperties)rr) matplotlibrr[rZr rrZFiguretextZFigureCanvasAggZsavefig) rZfilename_or_objrrrrrZparserr&r'r(_Zfigrrr math_to_image:s   r)NNN)5r@ collectionsrriorloggingr~ZnumpyrNZPILrrrrrrrr Zmatplotlib.ft2fontr r Zmatplotlib.font_managerr getLoggerr_logZcaching_module_getattrr Zget_unicode_indexrr#rArr`rcrqr{rbrr__subclasses__rr _cls_nametyperglobalsWarningrrVrrrrrrsj      7K  +     ;