a ضa7@sdZddlZddlZddlZddlZddlmZddlZddlm Z ddl Z ddl m ZddlZddlmZmZmZmZeeZdddd ZGd d d ZdS) a Support for embedded TeX expressions in Matplotlib. Requirements: * LaTeX. * \*Agg backends: dvipng>=1.6. * PS backend: PSfrag, dvips, and Ghostscript>=9.0. * PDF and SVG backends: if LuaTeX is present, it will be used to speed up some post-processing steps, but note that it is not used to parse the TeX string itself (only LaTeX is supported). To enable TeX rendering of all text in your Matplotlib figure, set :rc:`text.usetex` to True. TeX and dvipng/dvips processing results are cached in ~/.matplotlib/tex.cache for reuse between sessions. `TexManager.get_rgba` can also be used to directly obtain raster output as RGBA NumPy arrays. N)Path)TemporaryDirectory)parse)_apicbookdvireadrcParamsoptioncCs&|durd|dnd}d||dS)a$ Output LaTeX code that loads a package (possibly with an option) if it hasn't been loaded yet. LaTeX cannot load twice a package with different options, so this helper can be used to protect against users loading arbitrary packages/options in their custom preamble. N[]z]\makeatletter\@ifpackageloaded{%(package)s}{}{\usepackage%(option)s{%(package)s}}\makeatotherpackager rrre/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/matplotlib/texmanager.py_usepackage_if_not_loaded(s rc@seZdZdZejedZ iZ dZ dZ ddddd d d d d d dddddddZ ddddddddddddd ZedZedZedZedZeddZddZd6ddZd d!Zd"d#Zd$d%Zd&d'Zdd(d)d*Zd+d,Zd-d.Z d7d/d0Z!d8d2d3Z"d9d4d5Z#dS): TexManagerz Convert strings to dvi files using TeX, caching the results to a directory. Repeated calls to this constructor always return the same instance. z tex.cacheserif)r sans-serifcursive monospace)Zpncz\renewcommand{\rmdefault}{pnc})Zpbkz\renewcommand{\rmdefault}{pbk})Zptmz\usepackage{mathptmx})Zpplz\usepackage{mathpazo})Zpzcz\usepackage{chancery})Zpchz\usepackage{charter})cmrr )cmssr )Zphvz\usepackage{helvet})Zpagz\usepackage{avant})Zpcrz\usepackage{courier})Zcmtt\usepackage{type1ec})rr)rr)new century schoolbookbookmantimespalatino zapf chanceryrcharterrr helvetica avant gardecourierrcomputer modern romancomputer modern sans serifcomputer modern typewriterrrr) rrrrr r$rr!r"r%r#r&z3.5cCst|jjdddt|S)NT)parentsexist_ok)rtexcachemkdirobject__new__)clsrrrr,fszTexManager.__new__c std}t|dkr |dnd}d}t|dkrF||jvrF||_nBt|dkrn||jvrnd}|j||_ntdd |jd|_|jg}i|jD]}|r|j|kr|j||<n|td |D]T}||jvr|j||<t d |||j|q8qt d |qtd ||j||<| |dq| d }| t|fdddD}|jdkr|dd|dd t||_d |S)Nz font.familyrFTzXfont.family must be one of (%s) when text.usetex is True. serif will be used by default.z, rzfont.zfamily: %s, font: %s, info: %sz&%s font is not compatible with usetex.zPNo LaTeX-compatible font found for the %s fontfamily in rcParams. Using default.utf-8csh|]}|dqS)r.r).0familyZfontsrr sz-TexManager.get_font_config..)rrrrz\usepackage{type1cm} r )rlenlower_font_families _font_family _font_info _font_types_loginfojoindebugappendget_custom_preambleencodehashlibmd5 hexdigestaddsorted_font_preamble) selfffZff_valZreduced_notationZ fontconfig font_familyfontZpreamble_bytescmdrr2rget_font_configksV     zTexManager.get_font_configNcCsHd||d||t|p dg}tj|jt| d S)zU Return a filename based on a hash of the string, fontsize, and dpi. r z%fr/) r=rMr@strospathr)rBrCrArD)rHtexfontsizedpisrrr get_basefiles zTexManager.get_basefilecCs|jS)zU Return a string containing font configuration for the tex preamble. )rGrHrrrget_font_preambleszTexManager.get_font_preamblecCstdS)z>Return a string containing user additions to the tex preamble.ztext.latex.preamble)rrVrrrr@szTexManager.get_custom_preamblec Cs.ddd|jddd|tddd td g S) Nr4z\documentclass{article}z \newcommand{\mathdefault}[1]{#1}z\usepackage[utf8]{inputenc}z.\DeclareUnicodeCharacter{2212}{\ensuremath{-}}z1\usepackage[papersize=72in, margin=1in]{geometry}Z underscorestringsr Ztextcomp)r=rGr@rrVrrr _get_preambles zTexManager._get_preamblecCsV|||}d|}ddd|jd}t|jd|||d||fdd |S) zv Generate a tex file to render the tex string at a specific font size. Return the file name. z%s.texz{\sffamily %s}z{\ttfamily %s})rrz{\rmfamily %s}z %s \pagestyle{empty} \begin{document} %% The empty hbox ensures that a page is printed even for empty inputs, except %% when using psfrag which gets confused by it. \fontsize{%f}{%f}%% \ifdefined\psfrag\else\hbox{}\fi%% %s \end{document} g?r/)encoding)rUgetr8r write_textrY)rHrQrRbasefiletexfileZfontcmdrrrmake_texs   zTexManager.make_texcwdc Cstt|z$tj||dur$|n|jtjd}Wntyl}z t d |d|WYd}~nVd}~0tj y}z4t dj |d| d|j dd|WYd}~n d}~00t||S)N)rastderrz?Failed to process string with tex because {} could not be foundrzr{prog} was not able to process the following string: {tex!r} Here is the full report generated by {prog}: {exc} unicode_escaper/)progrQexc)r;r>rZ_pformat_subprocess subprocess check_outputr)STDOUTFileNotFoundError RuntimeErrorformatCalledProcessErrorrAoutputdecode)rHcommandrQrareportrerrr_run_checked_subprocesss4   z"TexManager._run_checked_subprocesscCs|||}d|}tj|st|||}tt|jdH}|jdddd|j g||dt|t|j  |Wdn1s0Y|S) zm Generate a dvi file containing latex's layout of tex string. Return the file name. z%s.dvi)dirZlatexz-interaction=nonstopmodez--halt-on-errorz../r`N) rUrOrPexistsrr_rparentrqnamereplace)rHrQrRr]dvifiler^Ztmpdirrrrmake_dvis    6zTexManager.make_dvic Cs||||}d|}tj|s|||}ddddt|ddd||g }td }ttd d rzt dj |krz| d d | |||S)zp Generate a png file containing latex's rendering of tex string. Return the file name. z%s.pngZdvipngz-bgZ Transparentz-Dz-TZtightz-oz1.16Z_called_from_pytestFr.z --freetype0) rUrOrPrsrxrN parse_versiongetattrmplZ_get_executable_infoversioninsertrq) rHrQrRrSr]pngfilerwrLZbad_verrrrmake_pngs      zTexManager.make_pngcCs|s td}|std}||||f}|j|}|dur||||}tjtj |j |}|dddddf|j|<}|S)zReturn the alpha channel.z font.sizez savefig.dpiN) rrM _grey_arraydr[rr{imageZimreadrOrPr=r))rHrQrRrSkeyalphar~rgbarrrget_grey-s  zTexManager.get_greyrrrcCsH||||}tg|jdR}tj||dddf<||d<|S)a* Return latex's rendering of the tex string as an rgba array. Examples -------- >>> texmanager = TexManager() >>> s = r"\TeX\ is $\displaystyle\sum_n\frac{-e^{i\pi}}{2^n}$!" >>> Z = texmanager.get_rgba(s, fontsize=12, dpi=80, rgb=(1, 0, 0)) .N).r)rnpemptyshaper{colorsZto_rgb)rHrQrRrSZrgbrrrrrget_rgba;s zTexManager.get_rgbacCsz|dkrdS|||}|r*|dnd}t|d|}|\}Wdn1sZ0Y|j|j|j|jfS)z-Return width, height and descent of the text.r rg?r.HN)striprxZpoints_to_pixelsrZDviwidthheightZdescent)rHrQrRZrendererrwZ dpi_fractionZdvipagerrrget_text_width_height_descentKs  $z(TexManager.get_text_width_height_descent)N)NN)NNr)N)$__name__ __module__ __qualname____doc__rOrPr=r{Z get_cachedirr)rr8r7r9r:rZdeprecate_privatize_attributeZ grey_arraydrJZ font_familiesZ font_info functools lru_cacher,rMrUrWr@rYr_rqrxrrrrrrrrr9s^     6   r)rrrBloggingrOpathlibrrftempfilerZnumpyrZpackaging.versionrryZ matplotlibr{rrrr getLoggerrr;rrrrrrs