OwlCyberSecurity - MANAGER
Edit File: texmanager.cpython-38.pyc
U ��]�@ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl Z ddl Z ddl Z ddl Zddl mZmZmZ e�e�ZG dd� de�ZdS )a� This module supports 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)�cbook�dviread�rcParamsc @ s8 e Zd ZdZe�� Zedk r>ej� ed�Z e e �j ddd� ndZ i Z i ZdZdZdZd Zd Zd Zd d ddd d ddddddddddd�ZdZdedd� eD �� Ze�� dd� �Zdd� Z d8d d �Z d d � Z d d!� Z d"d#� Z d$d%� Z!e"�#d&�Z$d'd(� Z%d)d*� Z&d+d,� Z'd-d.� Z(d/d0� Z)d9d1d2�Z*d:d4d5�Z+d;d6d7�Z,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. Nz tex.cacheT��parents�exist_ok)Zcmr� )Zcmssr )Zcmttr )Zpzcz\usepackage{chancery}�serif)r � sans-serif�cursive� monospace)Zpncz \renewcommand{\rmdefault}{pnc})Zpbkz \renewcommand{\rmdefault}{pbk})Zptmz\usepackage{mathptmx})Zpplz\usepackage{mathpazo})Zpchz\usepackage{charter})Zphvz\usepackage{helvet})Zpagz\usepackage{avant})Zpcrz\usepackage{courier})znew century schoolbookZbookman�timesZpalatinoz zapf chanceryr Zcharterr r Z helveticaz avant gardeZcourierr zcomputer modern romanzcomputer modern sans serifzcomputer modern typewriter)�text.latex.preamble�text.latex.unicode�text.latex.preview� font.familyc c s | ]}d | V qdS )�font.N� )�.0�nr r �7/usr/lib/python3/dist-packages/matplotlib/texmanager.py� <genexpr>a s zTexManager.<genexpr>c C s t �| �}|�� |S )N)�object�__new__�_reinit)�cls�selfr r r r c s zTexManager.__new__c C s� | j d krtd��t| j �jddd� td }t|�dkr\|d �� | jkr\|d �� | _n>t |t �r�|�� | jkr�|�� | _nt � dd� | j�� d | _| jg}| jD ]�}|�d d �}td | D ]X}|�� | jk�rt| || j|�� � t �d ||| j|�� � �q<q�t �d|� q�t � d|� t| || j| � |�t| |�d � q�| �� �d�}|�t�|��� � d� |�| _| jd | jd | j d g}| jdk�r�|�| j d � d� dg| dg �| _ d S )NzBCannot create TexManager, as there is no cache directory availableTr r � r zXfont.family must be one of (%s) when text.usetex is True. serif will be used by default.z, r �-�_r z family: %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 � z\usepackage{type1cm}z\usepackage{textcomp}) �texcache� RuntimeErrorr �mkdirr �len�lower� font_families� font_family� isinstance�str�_log�info�join�replace� font_info�setattr�debug�append�getattr�get_custom_preamble�encode�hashlib�md5� hexdigest� _fontconfigr � sans_serifr r �_font_preamble)r ZffZ fontconfigr* Zfont_family_attrZfontZpreamble_bytes�cmdr r r r i sV � � ��� �zTexManager._reinitc C sH d� || �� d| | �� t|p d�g�}tj� | jt�|� d��� � �S )zU Return a filename based on a hash of the string, fontsize, and dpi. r z%fr" ) r/ �get_font_configr6 r, �os�pathr$ r8 r9 r7 r: )r �tex�fontsize�dpi�sr r r � get_basefile� s � �zTexManager.get_basefilec s� � j dkrt�� j�� _ � fdd�� jD �}|r�t�d|� |D ]2}t�d|� j | t| � t�t| �� j |< q@t�d� j � � � � t�d� j � � j S )z8Reinitializes self if relevant rcParams on have changed.Nc s" g | ]}t | � j| kr|�qS r )r � _rc_cache)r Zpar�r r r � <listcomp>� s �z.TexManager.get_font_config.<locals>.<listcomp>zfollowing keys changed: %sz%-20s: %-10s -> %-10szRE-INIT old fontconfig: %szfontconfig: %s) rG �dict�fromkeys�_rc_cache_keysr- r3 r �copy�deepcopyr; r )r Zchanged�kr rH r r? � s �zTexManager.get_font_configc C s | j S )zU Return a string containing font configuration for the tex preamble. )r= rH r r r �get_font_preamble� s z TexManager.get_font_preamblec C s t d S )z>Return a string containing user additions to the tex preamble.r )r rH r r r r6 � s z TexManager.get_custom_preamblec C s� | � ||�}d| }| �� }ddd��| jd�}|| }td rFd}nd}d | j||||d |f }t|d ��d} td r�| �|�d �� nDz| �|�d �� W n. t k r� } zt � d� � W 5 d} ~ X Y nX W 5 Q R X |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}r � \usepackage[utf8]{inputenc}r z� \documentclass{article} %s %s %s \usepackage[papersize={72in,72in},body={70in,70in},margin={1in,1in}]{geometry} \pagestyle{empty} \begin{document} \fontsize{%f}{%f}%s \end{document} � �?�wb�utf8�ascii�WYou are using unicode and latex, but have not enabled the 'text.latex.unicode' rcParam.N� rF r6 �getr* r r= �open�writer7 �UnicodeEncodeErrorr- r. � r rB rC �basefile�texfileZcustom_preambleZfontcmdZunicode_preamblerE �fh�errr r r �make_tex� s: �� �� zTexManager.make_texz3MatplotlibBox:\(([\d.]+)pt\+([\d.]+)pt\)x([\d.]+)ptc C s� | � ||�}d| }| �� }ddd��| jd�}|| }td rFd}nd}d | j||||d |f }t|d ��d} td r�| �|�d �� nDz| �|�d �� W n. t k r� } zt � d� � W 5 d} ~ X Y nX W 5 Q R X |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. rQ rR rS rT rU r rV r a� \documentclass{article} %s %s %s \usepackage[active,showbox,tightpage]{preview} \usepackage[papersize={72in,72in},body={70in,70in},margin={1in,1in}]{geometry} %% 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} rW rX rY rZ r[ Nr\ ra r r r �make_tex_preview� s: �� �� zTexManager.make_tex_previewc C s� t �|� ztj|| jtjd�}W n� tk rX } ztd�|d ��|�W 5 d }~X Y nL tj k r� } z,tdj|d |� d�|j � d�d��|�W 5 d }~X Y nX t �|� |S )N)�cwd�stderrz?Failed to process string with tex because {} could not be foundr zr{prog} was not able to process the following string: {tex!r} Here is the full report generated by {prog}: {exc} Zunicode_escaper" )�progrB �exc) r- r3 � subprocessZ check_outputr$ ZSTDOUT�FileNotFoundErrorr% �formatZCalledProcessErrorr7 �output�decode)r ZcommandrB �reportrk r r r �_run_checked_subprocess+ s4 � ��� ��� z"TexManager._run_checked_subprocessc C s� t d r| �||�S | �||�}d| }tj�|�s�| �||�}t�|�� | � ddd|g|� W 5 Q R X t � |d �D ]2}|� d�sxzt� |� W qx t k r� Y qxX qx|S )zm Generate a dvi file containing latex's layout of tex string. Return the file name. r �%s.dvi�latex�-interaction=nonstopmode�--halt-on-error�*)�dvirB )r �make_dvi_previewrF r@ rA �existsrf r Z _lock_pathrr �glob�endswith�remove�OSError)r rB rC rb �dvifilerc �fnamer r r �make_dviA s( �� zTexManager.make_dvic C s� | � ||�}d| }d| }tj�|�r4tj�|�s�| �||�}| �ddd|g|�}tj�|� d��}t |d d��} | � d � |� � �� W 5 Q R X t�|d �D ]2} | �d �s�zt�| � W q� tk r� Y q�X q�|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. rs � %s.baselinert ru rv r" z .baseline�w� rw )rx rB Zbaseline)rF r@ rA rz rg rr r �_re_vbox�searchrp r^ r_ r/ �groupsr{ r| r} r~ ) r rB rC rb r � baselinefilerc rq �mrd r� r r r ry \ s* �� zTexManager.make_dvi_previewc C sV | � |||�}d| }tj�|�sR| �||�}| �ddddt|�ddd||g |� |S ) zp Generate a png file containing latex's rendering of tex string. Return the file name. z%s.pngZdvipngz-bgZ Transparentz-Dz-TZtightz-o)rF r@ rA rz r� rr r, )r rB rC rD rb �pngfiler r r r �make_png~ s ��zTexManager.make_pngc C sx ddl m} || �� ||f}| j�|�}|dkrt| �|||�}|�tj� | j |��}|dd�dd�df | j|< }|S )zReturn the alpha channel.r )�_pngN���) � matplotlibr� r? � grey_arraydr] r� Zread_pngr@ rA r/ r$ ) r rB rC rD r� �key�alphar� �Xr r r �get_grey� s zTexManager.get_grey�r r r c C sz |s t d }|st d }|\}}}|| �� ||t|�f}| j�|�} | dkrv| �|||�} t�|||| g�} | | j|< | S )z<Return latex's rendering of the tex string as an rgba array.z font.sizez savefig.dpiN)r r? �tuple� rgba_arraydr] r� �npZdstack) r rB rC rD Zrgb�r�g�br� �Zr� r r r �get_rgba� s zTexManager.get_rgbac s� |� � dkrdS |r |�d�nd� td r�| �||�}d| }tj�|�sV| �||�}t|��}|� � � � }W 5 Q R X � fdd�|D �\} } } | | | | fS | � ||�}t � |d � �� } | \} W 5 Q R X | j| j| j | jfS d S ) z-Return width, height and descent of the text.r r� g �?r r r� c s g | ]}t |�� �qS r )�float)r �l1�Z dpi_fractionr r rI � s z<TexManager.get_text_width_height_descent.<locals>.<listcomp>�H N)�stripZpoints_to_pixelsr rF r@ rA rz ry r^ �read�splitr� r ZDvi�width�heightZdescent)r rB rC Zrendererrb r� r rd �lr� Zdepthr� rx Zpager r� r � get_text_width_height_descent� s z(TexManager.get_text_width_height_descent)N)NN)NNr� )N)-�__name__� __module__� __qualname__�__doc__�mplZ get_cachedirZcachedirr@ rA r/ r$ r r& r� r� r r< r r r* r) r1 rG r� rL � functools� lru_cacher r rF r? rP r6 rf �re�compiler� rg rr r� ry r� r� r� r� r r r r r 1 sn ��� 2 -�;" r )r� rM r� r{ r8 Zloggingr@ �pathlibr r� rl Znumpyr� r� r� r r r Z getLoggerr� r- r r r r r r �<module> s