Rc@sdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZddlmZddlmZejZddlZejjdZeged D]"Zee jd ed ^qZyddlZd ZWn&ek r_ddlZd ZnXddlZddl j!Z!ej"d ddl#j$Z%ddlm&Z&dZyddl'm(Z(Wneek r(dej)krej*j+dZ(q)dej)krej*j+dZ(q)e,dnXdZ-ee-_dZ.dZ/dZ0dZ1dZ2dZ3dZ4dZ5dZ6dZ7e8Z9d e:fd!YZ;d"Z<d#e,fd$YZ=ddd%Z?d&Z@d'ZAd(ZBdS()s* A directive for including a matplotlib plot in a Sphinx document. By default, 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. The source code for the plot may be included in one of three 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) 3. Using **doctest** syntax:: .. plot:: A plotting example: >>> import matplotlib.pyplot as plt >>> plt.plot([1,2,3], [4,5,6]) Options ------- The ``plot`` directive supports the following options: format : {'python', 'doctest'} Specify the format of the input include-source : bool Whether to display the source code. The default can be changed using the `plot_include_source` variable in conf.py encoding : str If this source file is in a non-UTF8 or non-ASCII encoding, the encoding must be specified using the `:encoding:` option. The encoding will not be inferred using the ``-*- coding -*-`` metacomment. context : bool If provided, 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. nofigs : bool If specified, the code block will be run, but no figures will be inserted. This is usually useful with the ``:context:`` option. Additionally, 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`. Configuration options --------------------- The plot directive has the following configuration options: plot_include_source Default value for the include-source option plot_pre_code Code that should be executed before each plot. plot_basedir Base directory, to which ``plot::`` file names are relative to. (If None or empty, file names are relative to the directoly where the file containing the directive is.) plot_formats File formats to generate. List of tuples or strings:: [(suffix, dpi), suffix, ...] that determine the file format and the DPI. For entries whose DPI was omitted, sensible defaults are chosen. plot_html_show_formats Whether to show links to the files in HTML. plot_rcparams A dictionary containing any non-standard rcParams that should be applied before each plot. plot_apply_rcparams By default, rcParams are applied when `context` option is not used in a plot directive. This configuration option overrides this behaviour and applies rcParams before each plot. plot_working_directory By default, the working directory will be changed to the directory of the example, so the code can get at its data files, if any. Also its path will be added to `sys.path` so it can import any helper modules sitting beside it. This configuration option can be used to specify a central directory (also added to `sys.path`) where data files and helper modules for all code are located. plot_template Provide a customized template for preparing resturctured text. i(tprint_functionN(t directives(tnodes(tImaget.is[a-z]icKstj|j|S(N(tjinja2tTemplatetrender(ttemplatetkw((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pytformat_templatescKstj||S(N(tjinjat from_string(RR ((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyR stAgg(t_pylab_helpers(trelpathtposixc Csddlm}m}m}m}m}m}|sCtdn||j|}||j|} t ||| g} |gt || | | } | s|S|| S(s#Return a relative version of a pathi(tseptcurdirtjointabspatht commonprefixtpardirsno path specified( tos.pathRRRRRRt ValueErrortsplittlen( tpathtstartRRRRRRt start_listt path_listtitrel_list((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyRs.tntcCsddlm}m}m}m}m}m}m}|sItdn||j |} ||j |} | dj | dj kr||\} } ||\} } t | t | Artd||fqtd| d| dfnxWt t t| t| D]*}| |j | |j kr)Pq)q)W|d7}|gt| || |}|s|S||S(s#Return a relative version of a pathi(RRRRRRtsplituncsno path specifiedis,Cannot mix UNC and non-UNC paths (%s and %s)s&path is on drive %s, start on drive %si(RRRRRRRR"RRtlowertbooltrangetminR(RRRRRRRRR"RRtunc_pathtrestt unc_startRR ((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyRs*4 (  s,Unsupported platform (no relpath available!)c Cst||||||S(N(trun( tnamet argumentstoptionstcontenttlinenotcontent_offsett block_texttstatet state_machine((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pytplot_directivescCsd| s|j rtS|jjdkr4tS|jjd krPtStd|dS( Ntnot0tfalsetyest1ttrues"%s" unknown boolean(snoR6sfalse(syesR9strue(tstriptTrueR#tFalseR(targ((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt_option_booleanscCstj|dS(Ntpythontdoctest(spythonsdoctest(Rtchoice(R>((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt_option_formatscCstj|dS(Nttoptmiddletbottomtlefttcentertright(stopRERFsleftscentersright(RRB(R>((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt _option_aligns c Cs1x*|jjD]\}}|s(qn|j|}|dkrGqn|j|}|jdkrx|D]}|jdkrj|}x-|D]%}|jdkr|j}PqqW|dj||dj||dj||dj||j j j ||f|j j j |texecs^\s*>>>(tcompileR=t SyntaxErrortretMtsearchR$(ttexttrtm((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pytcontains_doctest:s cCst|s|Sd}xz|jdD]i}tjd|}|r^||jdd7}q&|jr|d|jd7}q&|d7}q&W|S(s` Extract code from a piece of text, which contains either Python code or doctests. ts s^\s*(>>>|\.\.\.) (.*)$is# (RRRtmatchtgroupR;(RtcodetlineR((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pytunescape_doctestEs  cCsg}t|}g}x|jdD]p}| rG|jdks_|r|jdkr|j||jdj|g}q(|j|q(Wdj|jr|jdj|n|S(s# Split code at plt.show() s s plt.show()s>>> plt.show()(RRR;RXR(Rtpartst is_doctesttpartR((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pytsplit_code_at_showYs   s  {{ source_code }} {{ only_html }} {% if source_link or (html_show_formats and not multi_image) %} ( {%- if source_link -%} `Source code <{{ source_link }}>`__ {%- endif -%} {%- if html_show_formats and not multi_image -%} {%- for img in images -%} {%- for fmt in img.formats -%} {%- if source_link or not loop.first -%}, {% endif -%} `{{ fmt }} <{{ dest_dir }}/{{ img.basename }}.{{ fmt }}>`__ {%- endfor -%} {%- endfor -%} {%- endif -%} ) {% endif %} {% for img in images %} .. figure:: {{ build_dir }}/{{ img.basename }}.png {%- for option in options %} {{ option }} {% endfor %} {% if html_show_formats and multi_image -%} ( {%- for fmt in img.formats -%} {%- if not loop.first -%}, {% endif -%} `{{ fmt }} <{{ dest_dir }}/{{ img.basename }}.{{ fmt }}>`__ {%- endfor -%} ) {%- endif -%} {{ caption }} {% endfor %} {{ only_latex }} {% for img in images %} .. image:: {{ build_dir }}/{{ img.basename }}.pdf {% endfor %} {{ only_texinfo }} {% for img in images %} .. image:: {{ build_dir }}/{{ img.basename }}.png {%- for option in options %} {{ option }} {% endfor %} {% endfor %} sh .. htmlonly:: [`source code <%(linkdir)s/%(basename)s.py>`__] Exception occurred rendering plot. t ImageFilecBs#eZdZdZdZRS(cCs||_||_g|_dS(N(tbasenametdirnametformats(tselfRR((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt__init__s  cCs#tjj|jd|j|fS(Ns%s.%s(tosRRRR(RRl((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pytfilenamescCs#g|jD]}|j|^q S(N(RR(Rtfmt((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt filenamess(t__name__t __module__RRR(((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyRs  cCsGtjj| pFtjj|oFtj|jtj|jkS(sh Returns True if derivative is out-of-date wrt original, both of which are full file paths. (RRtexistststattst_mtime(toriginaltderived((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt out_of_datest PlotErrorcBseZRS((RR(((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyRsc Bs$ej}eej}ejjdk ryej ejjWnQe k ro}e e |dn)e k r}e e |dnXejj dejjnM|dk rejjejj|}ej |ejj d|nej}eje_ej} |ge_zye|}|dkrUi}n|sejjdkrxd|Uqejj|Und|krd|Un||U|dk r|d|UnWn+eefk r}eejnXWdej || e_|ej(|e_X|S( ss Import a Python module from a path, and run the function given by name, if function_name is not None. s[ `plot_working_directory` option inSphinx configuration file must be a valid directory pathsV `plot_working_directory` option in Sphinx configuration file must be a string or Noneis8import numpy as np from matplotlib import pyplot as plt t__main__s__name__ = '__main__'s()N(RtgetcwdtlisttsysRR|R}RzRTtchdirtOSErrortstrt TypeErrortinsertRRtstdoutt cStringIOtStringIOtargvRRqt Exceptiont SystemExitRt tracebackt format_exc( Rt code_pathtnst function_nametpwdt old_sys_pathterrRRt old_sys_argv((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pytrun_codesN                 cCs+tjdtjtjj|dS(Ntall(tplttcloset matplotlibtrc_file_defaultstrcParamstupdate(Rx((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt clear_states  cCsmidd6dd6dd6}g}|j} t| ttfrNt| } nx| D]} t| tr|j| |j| dfqUt| tt fkrt | dkr|jt| dt | dfqUt d | qUWt |} t} t||} xF|D]>\}}t|| j|rJt} Pn| jj|qW| rt|| gfgSg}t} x t| D]\}}g}xtd D]}t | dkrtd |||f|} ntd ||f|} xF|D]>\}}t|| j|r1t} Pn| jj|qW| s[|dk} Pn|j| qW| svPn|j||fqW| r|Sg}|rt}ni}xt| D]\}}| s|jrt|jnt||||g}tjj}xt|D]\}}t |dkrbt | dkrbt||} nJt | dkrtd ||f|} ntd |||f|} |j| xr|D]j\}}y&|jjj | j|d |Wn%t!k r}t t"j#nX| jj|qWq W|j||fqW| sY|jrit|jn|S(s Run a pyplot script and save the low and high res PNGs and a PDF in outdir. Save the images under *output_dir* with file names derived from *output_base* iPRtis hires.pngRuiiis)invalid image format "%r" in plot_formatsis %s_%02d_%02ds%s_%02dtdpi($Rst isinstanceRtunicodetevalRXtgetttypettupleRRtintRRR<RRRR=Rt enumeratetxranget plot_contextRyRRxRRtGcftget_all_fig_managerstcanvasRMtsavefigRRR(RRt output_dirt output_baseRmRR}t default_dpiRRsRt code_piecest all_existstimgRlRtresultsRt code_piecetimagestjRt fig_managerstfigmanR((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pytrender_figuress   "**   $  &c8Cs[|r|rtdn|j}|jjj}|jd}|jd|j|jd} |jd} t j j | } t |r^|j st j jtjjjtj|d} n+t j jtj|j tj|d} dj|} t |dkr|d }nd}t| d }|j}WdQXt j j| }n| } tjdjtt|}|jjd dd }||jd D]F}3t j j|t j j|3}4|3|4krt?j@|3|4qqWqWqWt j j|||}5t|5d25}6| | kr>tA|}7n|}7|6jB|7WdQX|S(5Ns1plot:: directive can't have both args and contentRnsinclude-sourceRmtsourceis iiRt _plot_counters%s-%d.pyRs.pys.rsts.txtRt-RlR@R4t/s.Exception occurred in plotting %s from %s: %sRcss|]}d|jVqdS(s N(R;(t.0R((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pys ss.. code-block:: pythons %sRfRgRhRiRjRks:%s: %ss.. only:: htmls.. only:: latexs.. only:: texinfotdest_dirt build_dirt source_linkt multi_imaget only_htmlt only_latext only_texinfoR-Rt source_codethtml_show_formatsRNtw(s.pys.rsts.txt(saltsheightswidthsscalesalignsclass(Ct RuntimeErrorR^RYRZR}thas_keyt setdefaultRrt attributesRRRRRvRR|R]tbuildertsrcdirRturiR~RTtopentreadRttextwraptdedenttmapRRtsplitexttreplaceRR=R<Rt startswithRt doctreedirtnormpathRtmakedirsRtoutdirRRtmemotreportertsystem_messageRRtrstriptitemsR R{tTEMPLATERwtextendt insert_inputtcbooktmkdirsRtshutiltcopyfileRtwrite(8R,R.R-R3R2R/R^R}RnRmtrst_filetrst_dirtsource_file_nameRNRtfdRRtcountertbasetextt source_extRtsource_rel_nametsource_rel_dirRRt dest_dir_linktbuild_dir_linkRRterrorsRRtsmt total_linesRRRtlinestrowRtkeytvaltoptsRRRtsrc_linktresultRtfntdestimgt target_nametft code_escaped((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyR*~s      ! $       !         / * "     !  (Ct__doc__t __future__RRRtglobRtimptwarningsRRRRtdocutils.parsers.rstRtdocutilsRt&docutils.parsers.rst.directives.imagesRRjtsphinxt __version__Rtsphinx_versionRtxRRR t ImportErrorR Rtmatplotlib.cbookRtusetmatplotlib.pyplottpyplotRRRRtbuiltin_module_namesRRRR4R?RCRJReR|RRRRtexception_templatetdictRtobjectRRRRTRRRR*(((s]/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/sphinxext/plot_directive.pyt}sfl   3              #  R   =  j