File: //opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyo
�
�)�Uc @ s\ d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z y d d l m Z Wn! e
k
r� d d l m Z n Xd d l m Z y d d l
m Z Wn* e
k
r� d d l
m Z e j Z n Xd d l Z e j j d � Z e g e d D]" Z e e j d e � d
� ^ q� Z d d l Z d d l j Z e j d � d d l j Z d d l j Z d d l m Z d d
l! m" Z" d e# f d � � YZ$ e% e j& d � r�e j& j' Z' n e j( d � Z' d Z) d Z* d Z+ e, � Z- d � Z. e/ d � Z0 d � Z1 d � Z2 e/ d � Z3 d � Z4 d � Z5 d � Z6 d � Z7 d S( s} A special directive for including a matplotlib plot.
The source code for the plot may be included in one of two ways:
1. A path to a source file as the argument to the directive::
.. plot:: path/to/plot.py
When a path to a source file is given, the content of the
directive may optionally contain a caption for the plot::
.. plot:: path/to/plot.py
This is the caption for the plot
Additionally, one my specify the name of a function to call (with
no arguments) immediately after importing the module::
.. plot:: path/to/plot.py plot_function1
2. Included as inline content to the directive::
.. plot::
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np
img = mpimg.imread('_static/stinkbug.png')
imgplot = plt.imshow(img)
In HTML output, `plot` will include a .png file with a link to a high-res
.png and .pdf. In LaTeX output, it will include a .pdf.
To customize the size of the plot, this directive supports all of the
options of the `image` directive, except for `target` (since plot will
add its own target). These include `alt`, `height`, `width`, `scale`,
`align` and `class`.
Additionally, if the `:include-source:` option is provided, the
literal source will be displayed inline in the text, (as well as a
link to the source in HTML). If this source file is in a non-UTF8 or
non-ASCII encoding, the encoding must be specified using the
`:encoding:` option.
If the `:context:` option is plotted, the code will be run in the
context of all previous plot directives for which the context option
was specified. This only applies to inline code plot directives, not
those run from files.
If the ``:nofigs:`` option is specified, the code block will be run,
but no figures will be inserted. This is usually useful with the
``:context:`` option.
The set of file formats to generate can be specified with the
`plot_formats` configuration variable.
Error handling:
Any errors generated during the running of the code are emitted as warnings
using the Python `warnings` module, using a custom category called
`PlotWarning`. To turn the warnings into fatal errors that stop the
documentation build, after adjusting your `sys.path` in your `conf.py` Sphinx
configuration file, use::
import plot_directive
warnings.simplefilter('error', plot_directive.PlotWarning)
i����N( t md5( t
directives( t align( t Imaget .i s [a-z]i t Agg( t _pylab_helpers( t only_directivest PlotWarningc B s e Z d Z RS( so Warning category for all warnings generated by this directive.
By printing our warnings with this category, it becomes possible to turn
them into errors by using in your conf.py::
warnings.simplefilter('error', plot_directive.PlotWarning)
This way, you can ensure that your docs only build if all your examples
actually run successfully.
( t __name__t
__module__t __doc__( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyR e s
t relpathc C s] t j j | � s" t d | � n t j j | � sD t d | � n t j j | � j t j � } t j j | � j t j � } t j d k r� | d | d k r� t d | d j � d | d j � � n xK t
t t | � t | � � � D] } | | | | k r� Pq� q� W| d 7} t j
g t | � | | | } | rUt j j | � Sd
Sd S(
s�
Return a relative path to the target from either the current dir or an optional base dir.
Base can be a directory specified either as absolute or relative to current dir.
s Target does not exist: s+ Base is not a directory or does not exist: t ntt dost os2i s0 Target is on a different drive to base. Target: s , base: i t N( s ntR s os2( t ost patht existst OSErrort isdirt abspatht splitt sept namet uppert ranget mint lent pardirt join( t targett baset base_listt target_listt it rel_list( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyR y s #,(
"s�
.. htmlonly::
%(links)s
.. figure:: %(prefix)s%(tmpdir)s/%(outname)s.png
%(options)s
%(caption)s
.. latexonly::
.. figure:: %(prefix)s%(tmpdir)s/%(outname)s.pdf
%(options)s
%(caption)s
sh
.. htmlonly::
[`source code <%(linkdir)s/%(basename)s.py>`__]
Exception occurred rendering plot.
s c C sG t j j | � pF t j j | � oF t j | � j t j | � j k S( sh
Returns True if derivative is out-of-date wrt original,
both of which are full file paths.
( R R R t statt st_mtime( t originalt derived( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt out_of_date� s c B s= | d k r; d | } | r0 | d e f d Uq9| d Un� e j � } e j j | � \ } } e j j d e j j | � � e j } e
j � e _ e j | � d } z4 e
| � } e j d | | d d e j f � }
Wd e j d =e j | � | e _ | d k r| j � n X| d k r9e |
| � � n d S( ss
Import a Python module from a path, and run the function given by
name, if function_name is not None.
s6 import numpy as np; import matplotlib.pyplot as plt
%sNi t __plot__t pyt r( t Nonet plot_contextR t getcwdR R t syst insertR t stdoutt cStringIOt StringIOt chdirt opent impt load_modulet PY_SOURCEt closet getattr( t plot_patht
function_namet plot_codet contextt exec_codet pwdR t fnameR3 t fdt module( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt run_code� s.
"
c C s t j j � } xt | � D]� \ } } x� t | � D]� \ } \ } }
t | � d k rb | } n d | | f } | d | } t j j | | � } y | j j j
| d |
�Wn+ t j d | � }
t
j |
t � d SX| d k r5 t j | t j j | | � � q5 q5 Wq Wt | � S( s�
Once a plot script has been imported, this function runs savefig
on all of the figures in all of the desired formats.
i s %s_%02dR t dpis Exception saving plot %si ( R t Gcft get_all_fig_managerst enumerateR R R R t canvast figuret savefigt cbookt exception_to_strt warningst warnR t shutilt copyfile( R= t basenamet tmpdirt destdirt formatst fig_managersR$ t figmant jt formatRG t outnamet outpatht s( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt run_savefig� s" *c C s t j d � t j � d S( Nt all( t pltR; t
matplotlibt rc_file_defaults( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt clear_state s
c C s� t | � } t j j | � \ } } t j j | � \ } }
t } xL | D]D \ } }
t j j | d | | f � } t | | � rI t } PqI qI W| r� | r� d Sd } xv t r t } xO | D]G \ } }
t j j | d | | | f � } t | | � r� t } Pq� q� W| r | d 7} q� Pq� W| r8| d k r8| S| sHt � n y t
| | | d | �Wn+ t j d | � } t
j | t � d SX| s�t | | | | | � } d t j k r�t j d =n | Sd Sd S( sY
Run a pyplot script and save the low and high res PNGs and a PDF
in outdir.
s %s.%si i s
%s_%02d.%sR@ s Exception running plot %sR+ N( t strR R R t splitextt TrueR R* t FalseRd RF RN RO RP RQ R R_ R1 t modules( R= R> R? RU RV RW R@ t basedirRC RT t extt
all_existsR[ RG R\ R$ R^ t num_figs( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt render_figures sN
c
C s� | j d � } | r | d =n | j d � } | r> | d =n t j j } t | � t k sn t | � t k r} t | � } n t j j
| � }
t j j |
� \ } } t j j | j
j d � \ }
} t j j d | � } t t j |
� } t j j | | � } t j j d | � } t j j | � } t d k rAd } n d } t j j | � sit j | � n t j j t j j t j j j | � � } t j j | � s�t j | � n d
j d � | j d
� D� � } t | | | | | | d | �} g } | d k r,t j | t j j | |
� � n | j d � r | d k r�t d k r\d } n t j j j } | j d t j j | | � d g � | j d � r| j d | d � | d =qnF | j d d g � | j g | j d
� D] } d | j! � ^ q�� | j d � | d =n g } | s}| d k r]g | j" � D] \ } } d t# | | f ^ qD} d
j | � } x� t$ | � D]� } | d
k r�| } n d | | f } g } | d k r�| j d � n x5 | d
D]) \ } } | j d | | | | f � q�Wt% | � r3d d j | � t& � } n d } | j t' t& � j d
� � q�Wq}| j t( t& � j d
� � n t% | � r�| j) | | j* j+ d � � n g S( NR@ t nofigst sourcet plot_directivet buildi i R t /s
c s s | ] } t | j � Vq d S( N( t template_content_indentt strip( t .0t line( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pys <genexpr>� s s include-sourcei s .. include:: %ss
:literal:t encodings :encoding: %ss ::s %ss %s:%s: %ss %s_%02ds- `source code <%(linkdir)s/%(basename)s.py>`__s `%s <%s/%s.%s>`__s [%s]s , ( i i ( i (, t has_keyt setupt configt plot_formatst typeRe t unicodet evalR R RT Rf R t documentt
attributesR R t confdirR t sphinx_versionR RN t mkdirst appt buildert outdirRn R. RR RS t srcdirt extendt appendt rstript itemsRt R R t localst templatet exception_templatet insert_inputt input_linesRp ( R= Rj R> R? t captiont optionst
state_machineR@ Ro RW RC RT Rk t rstdirt rstfileR� t reldirt linkdirRU t prefixRV Rm t linest include_prefixt rowt keyt valR$ R\ t linksR[ RG ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt _plot_directiveP s�
$" * "
*
/ ! $ c C s� t | � r� t j | d � } t t j j | � t j j j
� }
d j | � } t | � d k rq | d } n d } t
| |
| d | | | � Sd j | � }
t |
� j � d } t
| d d |
d | | � Sd S( sd
Handle the arguments to the plot directive. The real work happens
in _plot_directive.
i s
i i i����t inlineR N( R R t uriR R R t dirnameRz R� R� R� R R. R� R t hexdigest( R t argumentsR� t contentt linenot content_offsett
block_textt stateR� R= Rj R� R> R? ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyRq � s $
c C s1 x*| j j � D]\ } } | s( q n | j | } | d k rG q n | j | } | j d k r x� | D]� } | j d k rj | } x- | D]% } | j d k r� | j � } Pq� q� W| d j | � | d j | � | d j | � | d j | � | j j
j | | f | j j
j | <Pqj qj Wq q Wd S( s�
To make plots referenceable, we need to move the reference from
the "htmlonly" (or "latexonly") node to the actual figure node
itself.
t html_onlyt
latex_onlyRL R� t idst namesN( R� R� (
t nametypest iteritemst nameidsR. R� t tagnamet astextt removeR� t settingst envt docnamet labels( R� R� R t explicitt labelidt nodet nt sectnamet c( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt mark_plot_labels� s*
%c C s� | t _ | j t _ | j t _ i
t j d 6t j d 6t j d 6t j d 6t d 6t j
d 6t j d 6t j d 6t j d 6t j d
6} | j
d t t d | � | j d d d d g t � | j d t � d S( Nt altt heightt widtht scaleR t classs include-sourceR@ Ro Rx t ploti i R| t pngiP s hires.pngi� t pdfi2 s doctree-read( i i i ( R� iP ( s hires.pngi� ( R� i2 ( Rz R� R{ R� R t unchangedt length_or_unitlesst length_or_percentage_or_unitlesst nonnegative_intR t class_optiont flagRx t
add_directiveRq Rg t add_config_valuet connectR� ( R� R� ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyRz s&
(8 R R1 R RR R8 RP R4 t ret hashlibR t ImportErrort docutils.parsers.rstR t& docutils.parsers.rst.directives.imagesR R t sphinxt __version__R R� t tuplet xt intRb t matplotlib.cbookRN t uset matplotlib.pyplott pyplotRa t matplotlib.imaget imageR t matplotlib.sphinxextR t WarningR t hasattrR R t curdirR� R� Rt t dictR/ R* Rh RF R_ Rd Rn R� Rq R� Rz ( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt <module>D sN T
3
6 ( @ z #