3 ƽh=@sdZddlZddlZddlZddlZddlZddlmZddlZddl Z ddl Z ddl Z ddl mZmZmZejeZGdddZdS)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>=8.60 Backends: * \*Agg * PS * PDF 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)cbookdvireadrcParamsc@seZdZdZejjejdZ iZ dZ dOZ dPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_dZ ejdd d!ed"d#Zejded$d%Zejd&d'ZiZejded(d)Zejded*d+Zejded,d-Zejded.d/Zd0d1Zd`d3d4Zd5d6Zd7d8Zd9d:Zd;d<Z e!j"d=Z#ejdd>d?Z$d@dAZ%dBdCZ&ejddDdEZ'dFdGZ(dadHdIZ)dcdKdLZ*dddMdNZ+d2S)e 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 sans-serifcursive monospacepnc\renewcommand{\rmdefault}{pnc}pbk\renewcommand{\rmdefault}{pbk}ptm\usepackage{mathptmx}ppl\usepackage{mathpazo}pzc\usepackage{chancery}pch\usepackage{charter}cmrcmssphv\usepackage{helvet}pag\usepackage{avant}pcr\usepackage{courier}cmtt\usepackage{type1ec})znew century schoolbookZbookmantimesZpalatinoz zapf chanceryr Zcharterrz sans-serifZ helveticaz avant gardeZcourierr zcomputer modern romanzcomputer modern sans serifzcomputer modern typewriterz3.3zmatplotlib.get_cachedir()) alternativecCstjS)N)mpl get_cachedir)selfr';/tmp/pip-build-7iwl8md4/matplotlib/matplotlib/texmanager.pycachedirRszTexManager.cachedircCsiS)Nr')r&r'r'r( rgba_arraydWszTexManager.rgba_arraydcCst|jjdddtj|S)NT)parentsexist_ok)rtexcachemkdirobject__new__)clsr'r'r(r0\szTexManager.__new__cCs|jjddS)Nrrr)rr)_fontsget)r&r'r'r(rcszTexManager.serifcCs|jjddS)Nz sans-serifrr)rr)r2r3)r&r'r'r( sans_serifhszTexManager.sans_serifcCs|jjddS)Nr r\usepackage{chancery})rr5)r2r3)r&r'r'r(r mszTexManager.cursivecCs|jjddS)Nr r r)r r)r2r3)r&r'r'r(r rszTexManager.monospacecCs~td}t|dkr6|dj|jkr6|dj|_ntjddj|jd|_|jg}x|jD]}x|td|D]P}|j|jkr|j|j|j |<tj d|||j|jPqrtj d |qrWtjd ||j||j |<|j |j |dq`W|j j d }|j tj|j|j dd|j d d|j d dg}|jdkrd|j |j dddjd||_dj|S)Nz font.familyrzXfont.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.zQNo LaTeX-compatible font found for the %s font family in rcParams. Using default.zutf-8z sans-serifr r  \usepackage{type1cm}r)r8)rlenlower font_families font_family_loginfojoin font_infor2debugappendget_custom_preambleencodehashlibmd5 hexdigest_font_preamble)r&ffZ fontconfigr<fontZpreamble_bytescmdr'r'r(get_font_configws8    zTexManager.get_font_configNcCsHdj||jd||jt|p dg}tjj|jtj|j dj S)zU Return a filename based on a hash of the string, fontsize, and dpi. rz%fzutf-8) r?rLrCstrospathr-rErFrDrG)r&texfontsizedpisr'r'r( get_basefileszTexManager.get_basefilecCs|jS)zU Return a string containing font configuration for the tex preamble. )rH)r&r'r'r(get_font_preambleszTexManager.get_font_preamblecCstdS)z>Return a string containing user additions to the tex preamble.ztext.latex.preamble)r)r&r'r'r(rCszTexManager.get_custom_preamblec Cs djdd|jddd|jdgS)Nr7z\documentclass{article}z \newcommand{\mathdefault}[1]{#1}z\usepackage[utf8]{inputenc}z.\DeclareUnicodeCharacter{2212}{\ensuremath{-}}z:\usepackage[papersize=72in,body=70in,margin=1in]{geometry}zM\makeatletter\@ifpackageloaded{textcomp}{}{\usepackage{textcomp}}\makeatother)r?rHrC)r&r'r'r( _get_preambleszTexManager._get_preamblecCsV|j||}d|}dddj|jd}t|jd|j||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})z sans-serifr z{\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?zutf-8)encoding)rTr3r<r write_textrV)r&rPrQbasefiletexfilefontcmdr'r'r(make_texs   zTexManager.make_texz3MatplotlibBox:\(([\d.]+)pt\+([\d.]+)pt\)x([\d.]+)ptcCsV|j||}d|}dddj|jd}t|jd|j||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. z%s.texz{\sffamily %s}z{\ttfamily %s})z sans-serifr z{\rmfamily %s}aD %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} g?zutf-8)rW)rTr3r<rrXrV)r&rPrQrYrZr[r'r'r(make_tex_previews  zTexManager.make_tex_previewcCstjtj|ytj||jtjd}Wntk r^}zt dj |d|WYdd}~XnLtj k r}z.t dj |d|j d|j jdd|WYdd}~XnXtj||S)N)cwdstderrz?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_escapezutf-8)progrPexc)r=rArZ_pformat_subprocess subprocess check_outputr-STDOUTFileNotFoundError RuntimeErrorformatCalledProcessErrorrDoutputdecode)r&commandrPreportrar'r'r(_run_checked_subprocesss"& z"TexManager._run_checked_subprocesscCstjtdr|j||S|j||}d|}tjj|s|j||}t j ||j ddd|g|WdQRXxDt j |dD]2}|j d s~ytj|Wq~tk rYq~Xq~W|S) zm Generate a dvi file containing latex's layout of tex string. Return the file name. ztext.latex.previewz%s.dvilatexz-interaction=nonstopmodez--halt-on-errorN*dvirP)rprP)dict __getitem__rmake_dvi_previewrTrNrOexistsr\rZ _lock_pathrmglobendswithremoveOSError)r&rPrQrYdvifilerZfnamer'r'r(make_dvi&s"        zTexManager.make_dvic Cs|j||}d|}d|}tjj| s8tjj| r|j||}|jddd|g|}tjj|j d}t |dd} | j d j |j Wd QRXxDtj|d D]2} | jdsytj| Wqtk rYqXqW|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. z%s.dviz %s.baselinernz-interaction=nonstopmodez--halt-on-errorzutf-8z .baselinew NrorprPbaseline)rprPr~)rTrNrOrtr]rmr_re_vboxsearchrjopenwriter?groupsrurvrwrx) r&rPrQrYry baselinefilerZrlmfhrzr'r'r(rsAs$     zTexManager.make_dvi_previewc Cs|j|||}d|}tjj|s~|j||}ddddt|ddd||g }ttd d rrtjdj d krr|j d d |j |||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.16r6z --freetype0) rTrNrOrtr{rMgetattrr$Z_get_executable_infoversioninsertrm)r&rPrQrRrYpngfileryrKr'r'r(make_pngds     zTexManager.make_pngcCs|s td}|std}||j||f}|jj|}|dkr|j|||}tjjtj j |j |}|dddddf|j|<}|S)zReturn the alpha channel.z font.sizez savefig.dpiNr6) rrL grey_arraydr3rr$imageZimreadrNrOr?r-)r&rPrQrRkeyalpharrgbar'r'r(get_grey{s  zTexManager.get_greyrcCsB|j|||}tj|jd}tjj||dddf<||d<|S) zsz.H)rrr)stripZpoints_to_pixelsrqrrrrTrNrOrtrsrreadsplitr{rZDviwidthheightZdescent)r&rPrQZrendererrYrryrlrdepthrrppager')rr(get_text_width_height_descents        z(TexManager.get_text_width_height_descent)rrr r )r r )r r)rr)rr)rr)rr)rr)rr)rr)rr)rr)rr)r r!)rr!)rr!)r r!)N)NNrrr)NNr)N),__name__ __module__ __qualname____doc__rNrOr?r$r%r-rr<r;r@rZ deprecatedpropertyr)r* functools lru_cacher0r2rr4r r rLrTrUrCrVr\recompilerr]rmr{rsrrrrr'r'r'r(r/sf * '#  r)rrrurEloggingrNZpathlibrrrbnumpyrZ matplotlibr$rrr getLoggerrr=rr'r'r'r(s