U \ a>@sdZddlZddlZddlZddlZddlZddlmZddlZddl Z ddl m Z ddl Z ddlZddlmZmZmZmZeeZGdddZdS)a Support for embedded TeX expressions in Matplotlib via dvipng and dvips for the raster and PostScript backends. The tex and dvipng/dvips information is cached in ~/.matplotlib/tex.cache for reuse between sessions. Requirements: * LaTeX * \*Agg backends: dvipng>=1.6 * PS backend: psfrag, dvips, and Ghostscript>=9.0 For raster output, you can get RGBA numpy arrays from TeX expressions as follows:: texmanager = TexManager() s = "\TeX\ is Number $\displaystyle\sum_{n=1}^\infty\frac{-e^{i\pi}}{2^n}$!" Z = texmanager.get_rgba(s, fontsize=12, dpi=80, rgb=(1, 0, 0)) To enable TeX rendering of all text in your Matplotlib figure, set :rc:`text.usetex` to True. N)Path)TemporaryDirectory)_apicbookdvireadrcParamsc@seZdZdZejedZ iZ dZ dZ ddddd d d d d d dddddddZ ejdddeddZededdZiZededdZededdZededdZededdZeddZd d!Zd@d#d$Zd%d&Zd'd(Zd)d*Zd+d,Z e!"d-Z#edd.d/Z$d"d0d1d2Z%d3d4Z&edd5d6Z'd7d8Z(dAd9d:Z)dBdd?Z+d"S)D 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}cmrcmssr)Zphvz\usepackage{helvet})Zpagz\usepackage{avant})Zpcrz\usepackage{courier})cmtt\usepackage{type1ec})rr)rr)znew century schoolbookZbookmantimesZpalatinoz zapf chanceryr Zcharterr r Z helveticaz avant gardeZcourierr zcomputer modern romanzcomputer modern sans serifzcomputer modern typewriterz3.3zmatplotlib.get_cachedir()) alternativecCstSN)mpl get_cachedirselfr9/tmp/pip-unpacked-wheel-wjyw_3jo/matplotlib/texmanager.pyNzTexManager.cCsiSrrrrrrrOrcCs|jddS)Nr r_fontsgetrrrrrRrcCs|jddS)Nr rr rrrrrTrcCs|jddS)Nr r r rrrrrVscCs|jddS)Nr )rrr rrrrrYrcCst|jjdddt|S)NT)parentsexist_ok)rtexcachemkdirobject__new__)clsrrrr([szTexManager.__new__c Csxtd}t|dkr6|d|jkr6|d|_ntdd|jd|_|jg}|jD]}td|D]R}||jkr|j||j |<t d|||j|qqnt d |qntd ||j||j |<| |j |dq^| d }| t||j dd|j d d|j d dg}|jdkr^| |j dddd||_d|S)Nz font.familyrzXfont.family must be one of (%s) when text.usetex is True. serif will be used by default.z, r zfont.zfamily: %s, font: %s, info: %sz&%s font is not compatible with usetex.zQNo LaTeX-compatible font found for the %s font family in rcParams. Using default.utf-8r r r  \usepackage{type1cm}r)r-)rlenlower font_families font_family_loginfojoin font_infor!debugappendget_custom_preambleencodehashlibmd5 hexdigest_font_preamble)rffZ fontconfigr1fontZpreamble_bytescmdrrrget_font_config`sD       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. rz%fr+) r4rAr8strospathr%r:r;r9r<)rtexfontsizedpisrrr get_basefiles zTexManager.get_basefilecCs|jS)zU Return a string containing font configuration for the tex preamble. )r=rrrrget_font_preambleszTexManager.get_font_preamblecCstdS)z>Return a string containing user additions to the tex preamble.ztext.latex.preamble)rrrrrr8szTexManager.get_custom_preamblec Cs ddd|jddd|dgS)Nr,z\documentclass{article}z \newcommand{\mathdefault}[1]{#1}z\usepackage[utf8]{inputenc}z.\DeclareUnicodeCharacter{2212}{\ensuremath{-}}z1\usepackage[papersize=72in, margin=1in]{geometry}zM\makeatletter\@ifpackageloaded{textcomp}{}{\usepackage{textcomp}}\makeatother)r4r=r8rrrr _get_preambleszTexManager._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. %s.tex{\sffamily %s}{\ttfamily %s}r r {\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} ?r+encodingrIr"r1r write_textrKrrErFbasefiletexfileZfontcmdrrrmake_texs   zTexManager.make_texz3MatplotlibBox:\(([\d.]+)pt\+([\d.]+)pt\)x([\d.]+)ptcCsV|||}d|}ddd|jd}t|jd|||d||fdd |S) z Generate a tex file to render the tex string at a specific font size. It uses the preview.sty to determine the dimension (width, height, descent) of the output. Return the file name. rLrMrNrOrPaD %s \usepackage[active,showbox,tightpage]{preview} %% we override the default showbox as it is treated as an error and makes %% the exit status not zero \def\showbox#1%% {\immediate\write16{MatplotlibBox:(\the\ht#1+\the\dp#1)x\the\wd#1}} \begin{document} \begin{preview} {\fontsize{%f}{%f}%s} \end{preview} \end{document} rQr+rRrTrVrrrmake_tex_previews zTexManager.make_tex_previewcwdc Cstt|z$tj||dk r$|n|jtjd}Wntk rj}zt d |d|W5d}~XYnLtj k r}z,t dj |d| d|j dd|W5d}~XYnXt||S)N)r\stderrz?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} Zunicode_escaper+)progrEexc)r2r6rZ_pformat_subprocess subprocess check_outputr%STDOUTFileNotFoundError RuntimeErrorformatCalledProcessErrorr9outputdecode)rcommandrEr\reportr_rrr_run_checked_subprocesss6   z"TexManager._run_checked_subprocessc Csttdr|||S|||}d|}tj|s|||}t t |j d6}|j ddd|g||dt |t |j |W5QRX|S)zm Generate a dvi file containing latex's layout of tex string. Return the file name. text.latex.preview%s.dvi)dirlatex-interaction=nonstopmode--halt-on-errorr[)dict __getitem__rmake_dvi_previewrIrCrDexistsrYrrparentrknamereplace)rrErFrWdvifilerXZtmpdirrrrmake_dvis      "zTexManager.make_dvic Cs|||}d|}d|}tj|r4tj|s|||}|ddd|g|}tj| d}t |dd} | d | W5QRXt|d D]2} | d szt| Wqtk rYqXq|S) z Generate a dvi file containing latex's layout of tex string. It calls make_tex_preview() method and store the size information (width, height, descent) in a separate file. Return the file name. rm %s.baselinerorprqr+z .baselinew *)dvirEZbaseline)rIrCrDrurZrkr_re_vboxsearchrhopenwriter4groupsglobendswithremoveOSError) rrErFrWry baselinefilerXrjmfhfnamerrrrt)s*   zTexManager.make_dvi_previewc Cs||||}d|}tj|s~|||}ddddt|ddd||g }ttd d rrtdj d krr| 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-oZ_called_from_pytestFz1.16r*z --freetype0) rIrCrDrurzrBgetattrrZ_get_executable_infoversioninsertrk)rrErFrGrWpngfileryr@rrrmake_pngLs"     zTexManager.make_pngcCs|s td}|std}||||f}|j|}|dkr||||}tjtj |j |}|dddddf|j|<}|S)zReturn the alpha channel.z font.sizez savefig.dpiN) rrA grey_arraydr"rrimageZimreadrCrDr4r%)rrErFrGkeyalpharrgbarrrget_greycs  zTexManager.get_greyrrrcCsB||||}t|jd}tj||dddf<||d<|S)zsz.HN)stripZpoints_to_pixelsrrrsrrIrCrDrurtrreadsplitrzrZDviwidthheightZdescent)rrErFZrendererrWrryrlrdepthrrpagerrrget_text_width_height_descentys        z(TexManager.get_text_width_height_descent)N)NN)NNr)N),__name__ __module__ __qualname____doc__rCrDr4rrr%rr1r0r5r deprecatedpropertyZcachedirZ rgba_arraydr!r Z sans_serifr r functools lru_cacher(rArIrJr8rKrYrecompilerrZrkrzrtrrrrrrrrr)s~      *  & "  r)rrrr:loggingrCpathlibrrr`tempfilerZnumpyrZ matplotlibrrrrr getLoggerrr2rrrrrs