o ¡:´aúuã@sìdZddlZddlZddlmZddlZddlZddlmZddl m Z ddl Z ddl Z ddl Z ddlZddlZddlmZmZddlmZddlZddlZddlmZddlmZdd lmZmZmZe d ¡ej d d d ej!ƒZ!dZ"dd„Z#dd„Z$dd„Z%dd„Z&dd„Z'Gdd„deƒZ(dd„Z)dd„Z*dd „Z+ej d!d"d d#d$„ƒZ,e  d!¡d%d&„ƒZ-d'd(„Z.d)Z/d*Z0e1ƒZ2Gd+d,„d,ƒZ3d?d-d.„Z4Gd/d0„d0e5ƒZ6e  d!¡d@d1d2„ƒZ7d@d3d4„Z8dAd6d7„Z9d8d9„Z: : : dBd;d<„Z;d=d>„Z>> import matplotlib.pyplot as plt >>> plt.plot([1, 2, 3], [4, 5, 6]) Options ------- The ``plot`` directive supports the following options: format : {'python', 'doctest'} The format of the input. If unset, the format is auto-detected. include-source : bool Whether to display the source code. The default can be changed using the `plot_include_source` variable in :file:`conf.py` (which itself defaults to False). 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 or str 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. If the ``:context: reset`` option is specified, the context is reset for this and future plots, and previous figures are closed prior to running the code. ``:context: close-figs`` keeps the context but closes previous figures before running the code. nofigs : bool If specified, the code block will be run, but no figures will be inserted. This is usually useful with the ``:context:`` option. caption : str If specified, the option's argument will be used as a caption for the figure. This overwrites the caption given in the content, when the plot is generated from a file. 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 (default: False). plot_html_show_source_link Whether to show a link to the source in HTML (default: True). plot_pre_code Code that should be executed before each plot. If None (the default), it will default to a string containing:: import numpy as np from matplotlib import pyplot as plt plot_basedir Base directory, to which ``plot::`` file names are relative to. If None or empty (the default), file names are relative to the directory where the file containing the directive is. plot_formats File formats to generate (default: ['png', 'hires.png', 'pdf']). 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. When passing from the command line through sphinx_build the list should be passed as suffix:dpi,suffix:dpi, ... plot_html_show_formats Whether to show links to the files in HTML (default: True). plot_rcparams A dictionary containing any non-standard rcParams that should be applied before each plot (default: {}). plot_apply_rcparams By default, rcParams are applied when ``:context:`` option is not used in a plot directive. If set, this configuration option overrides this behavior 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 restructured text. éN)ÚStringIO)Úrelpath)ÚPath)Ú directivesÚ Directive)ÚImage)ÚFigureManagerBase)Ú_apiÚ_pylab_helpersÚcbookZaggz3.4z2docutils.parsers.rst.directives.images.Image.align)Z alternativeécCsD|r| ¡sdS| ¡ ¡dvrdS| ¡ ¡dvrdStd|ƒ‚)NT)ÚnoÚ0ZfalseF)ZyesÚ1Útruez"%s" unknown boolean)ÚstripÚlowerÚ ValueError©Úarg©rúE/usr/lib/python3/dist-packages/matplotlib/sphinxext/plot_directive.pyÚ_option_boolean±s  rcCs|dvr|Stdƒ‚)N)NÚresetú close-figsz2Argument should be None or 'reset' or 'close-figs')rrrrrÚ_option_context½srcCs t |d¡S)N)ÚpythonÚdoctest)rÚchoicerrrrÚ_option_formatÃs rcCstjddddt |¡S)Nú3.5ÚencodingÚoption)ÚnameZobj_type)r Zwarn_deprecatedrr!rrrrÚ_deprecated_option_encodingÇs r$c CsÔ|j ¡D]b\}}|s q|j|}|durq|j|}|jdvrg|D]D}|jdkrf|}|D] }|jdkr:| ¡}nq-|d |¡|d |¡|d |¡|d |¡|jj j ||f|jj j |<nq"qdS)z” To make plots referenceable, we need to move the reference from the "htmlonly" (or "latexonly") node to the actual figure node itself. N)Z html_onlyZ latex_onlyÚfigureÚcaptionÚidsÚnames) Z nametypesÚitemsZnameidsr'ZtagnameZastextÚremoveÚappendÚsettingsÚenvZdocnameÚlabels) ÚappÚdocumentr#ZexplicitZlabelidZnodeÚnZsectnameÚcrrrÚmark_plot_labelsÌs4     þ ÿó€÷r3c @sVeZdZdZdZdZdZdZej ej ej ej e jejeeeejeej dœ Zdd„Zd S) Ú PlotDirectivezEThe ``.. plot::`` directive, as documented in the module's docstring.Trr F) ÚaltÚheightÚwidthÚscaleÚalignÚclassúinclude-sourceÚformatÚcontextÚnofigsr!r&c CsHzt|j|j|j|j|j|jƒWSty#}z| t |ƒ¡‚d}~ww)zRun the plot directive.N) ÚrunÚ argumentsÚcontentÚoptionsÚ state_machineÚstateÚlinenoÚ ExceptionÚerrorÚstr)ÚselfÚerrrr?s ÿ€ÿzPlotDirective.runN)Ú__name__Ú __module__Ú __qualname__Ú__doc__Z has_contentZrequired_argumentsZoptional_argumentsZfinal_argument_whitespacerZ unchangedZlength_or_unitlessZ length_or_percentage_or_unitlessZnonnegative_intrr9Z class_optionrrrÚflagr$Z option_specr?rrrrr4ês(ô r4cCsP|dur$|jjdkr&t d¡}|jtdƒ}|jddt ||¡dSdSdS)NZhtmlz!plot_directive/plot_directive.cssZ_staticT©Úexist_ok) Úbuilderr<r Z_get_data_pathÚoutdirrÚmkdirÚshutilÚcopy)r/ÚexcÚsrcÚdstrrrÚ_copy_css_file s   ürZcCsæ|t_|jt_|jt_| dt¡| ddd¡| ddd¡| ddd¡| dgd¢d¡| d dd¡| d dd¡| d id¡| d dd¡| d dd¡| ddd¡| dt¡|  d¡| dt ¡ddt j dœ}|S)NZplotÚ plot_pre_codeTÚplot_include_sourceFÚplot_html_show_source_linkÚ plot_formats©Zpngz hires.pngZpdfÚ plot_basedirÚplot_html_show_formatsÚ plot_rcparamsÚplot_apply_rcparamsÚplot_working_directoryÚ plot_templatez doctree-readzplot_directive.csszbuild-finished)Zparallel_read_safeZparallel_write_safeÚversion) Úsetupr/ÚconfigÚconfdirZ add_directiver4Zadd_config_valueZconnectr3Z add_css_filerZÚ matplotlibÚ __version__)r/Zmetadatarrrrgs*    ÿrgcCsFz t|ddƒWdStyYnwt dtj¡}| |¡}t|ƒS)NzÚexecFz^\s*>>>)ÚcompileÚ SyntaxErrorÚreÚMÚsearchÚbool)ÚtextÚrÚmrrrÚcontains_doctest-s  ÿ rvr zdoctest.script_from_examplescCsnt|ƒs|Sd}| d¡D]'}t d|¡}|r!|| d¡d7}q | ¡r0|d| ¡d7}q |d7}q |S)z_ Extract code from a piece of text, which contains either Python code or doctests. ÚÚ z^\s*(>>>|\.\.\.) (.*)$r z# )rvÚsplitroÚmatchÚgroupr)rsÚcodeÚlinerurrrÚunescape_doctest9s  r~cCs t|ƒdS)úSplit code at plt.show().é)Ú_split_code_at_show)rsrrrÚsplit_code_at_showMs r‚cCsg}t|ƒ}g}| d¡D]'}|s| ¡dks|r/| ¡dkr/| |¡| d |¡¡g}q | |¡q d |¡ ¡rD| d |¡¡||fS)rrxz plt.show()z>>> plt.show())rvryrr+Újoin)rsÚpartsÚ is_doctestÚpartr}rrrrSs ÿ ÿ  ra( {{ 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 }}.{{ default_fmt }} {% 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 }} {# appropriate leading whitespace added beforehand #} {% endfor %} .. only:: not html {% for img in images %} .. figure:: {{ build_dir }}/{{ img.basename }}.* {% for option in options -%} {{ option }} {% endfor -%} {{ caption }} {# appropriate leading whitespace added beforehand #} {% endfor %} zi .. only:: html [`source code <%(linkdir)s/%(basename)s.py>`__] Exception occurred rendering plot. c@s$eZdZdd„Zdd„Zdd„ZdS)Ú ImageFilecCs||_||_g|_dS©N)ÚbasenameÚdirnameÚformats)rIr‰rŠrrrÚ__init__¬s zImageFile.__init__cCstj |jd|j|f¡S)Nz%s.%s)ÚosÚpathrƒrŠr‰)rIr<rrrÚfilename±szImageFile.filenamecs‡fdd„ˆjDƒS)Ncsg|]}ˆ |¡‘qSr)r)Ú.0Úfmt©rIrrÚ µsz'ImageFile.filenames..)r‹r’rr’rÚ filenames´szImageFile.filenamesN)rKrLrMrŒrr”rrrrr‡«s r‡csRtj |¡sdS|durg}|g|¢}dd„‰t |¡j‰t‡‡fdd„|DƒƒS)a4 Return whether *derived* is out-of-date relative to *original* or any of the RST files included in it using the RST include directive (*includes*). *derived* and *original* are full paths, and *includes* is optionally a list of full paths which may have been included in the *original*. TNcSstj |¡o |t |¡jkSrˆ)rrŽÚexistsÚstatÚst_mtime)ÚoriginalÚ derived_mtimerrrÚout_of_date_oneÆs ÿz$out_of_date..out_of_date_onec3s|]}ˆ|ˆƒVqdSrˆr©rÚf©r™ršrrÚ Ës€zout_of_date..)rrŽr•r–r—Úany)r˜ZderivedÚincludesZfiles_to_checkrrrÚ out_of_date¸s   r¡c@s eZdZdS)Ú PlotErrorN)rKrLrMrrrrr¢Îsr¢cCstt|ƒ|||ƒdS)ús Import a Python module from a path, and run the function given by name, if function_name is not None. N)Ú _run_coder~)r|Ú code_pathÚnsÚ function_namerrrÚrun_codeÒsr¨c Cst ¡}tjjdur=z t tjj¡Wn<ty(}z tt|ƒdƒ|‚d}~wty<}z tt|ƒdƒ|‚d}~ww|durPtj   tj   |¡¡}t |¡t j t|gt ¡gtj ¢d›t tƒ¡|zmzP|duroi}|s†tjjdur}td|ƒn tttjjƒ|ƒd|vrŽd|d<t j tdd „d t||ƒ|dur¨t|d |ƒWdƒn1s²wYWnttfyÍ}ztt ¡ƒ|‚d}~wwWt |¡nt |¡wWdƒn1såwYWdƒ|SWdƒ|S1sýwY|S) r£Nz[ `plot_working_directory` option inSphinx configuration file must be a valid directory pathzV `plot_working_directory` option in Sphinx configuration file must be a string or None)ÚargvrŽz8import numpy as np from matplotlib import pyplot as plt Ú__main__rKcSsdSrˆrr’rrrÚsz_run_code..)Zshowz())rÚgetcwdrgrhrdÚchdirÚOSErrorrHÚ TypeErrorrŽÚabspathrŠr Z _setattr_cmÚsysÚ contextlibÚredirect_stdoutrr[rlrrFÚ SystemExitr¢Ú tracebackÚ format_exc)r|r¥r¦r§ÚpwdÚerrrŠrrrr¤Ûsj þ€þ€ÿ ÿ þ ÿ €ý€€ÿ€(è ççr¤TcCs&|rt d¡t ¡tj |¡dS)NÚall)ÚpltÚcloserjZrc_file_defaultsZrcParamsÚupdate)rbr»rrrÚ clear_states r½cCs¸ddddœ}g}|j}|D]L}t|tƒr7d|vr+| d¡\}}| t|ƒt|ƒf¡q | || |d¡f¡q t|ttfƒrTt |ƒdkrT| t|dƒt|dƒf¡q t d|ƒ‚|S) NéPéÈr_ú:r rr€z)invalid image format "%r" in plot_formats) r^Ú isinstancerHryr+ÚintÚgetÚtupleÚlistÚlenr¢)rhZ default_dpir‹r^r‘ÚsuffixÚdpirrrÚget_plot_formatss    rÉFc  CsÜt|ƒ} t|ƒ\} } d} t||ƒ}| D]\}}|s#t|| |¡| dr'd} n|j |¡q| r6||gfgSg}d} t| ƒD]\\}}g}t  ¡D]F}t | ƒdkr[td|||f|ƒ}n td||f|ƒ}| D]\}}|svt|| |¡| drzd} n|j |¡qf| s‰|dk} n| |¡qH| s“n| ||f¡q>| rŸ|Sg}|r¥t ni}|r²t |j ƒt  ¡| p¶|}t| ƒD]¡\}}|rÄ|jrËt |j |ƒn|rÒt d¡t| rÚt |¡n||||ƒg}tj ¡}t|ƒD]i\}}t |ƒdkrt | ƒdkrt||ƒ}nt | ƒdkrtd||f|ƒ}n td|||f|ƒ}| |¡| D].\}}z|jjj| |¡|d WntyK}ztt ¡ƒ|‚d }~ww|j |¡q%që| ||f¡q»|rd|jrlt |j | d |S) z™ Run a pyplot script and save the images in *output_dir*. Save the images under *output_dir* with file names derived from *output_base* T)r Fr€z %s_%02d_%02dz%s_%02drr¹)rÈN)r»)rÉrr‡r¡rr‹r+Ú enumerateÚ itertoolsÚcountrÆÚ plot_contextr½rbÚclearrcrºr»r¤rÚscript_from_examplesr ZGcfZget_all_fig_managersZcanvasr%ZsavefigrFr¢rµr¶)r|r¥Z output_dirÚ output_baser=r§rhÚ context_resetÚ close_figsÚ code_includesr‹r…Z code_piecesZ all_existsÚimgr<rÈÚresultsÚiÚ code_pieceÚimagesÚjr‘r¦Z fig_managersZfigmanr¸rrrÚrender_figures)sœ   ÿ   ÿ ÿ      þ  ÿ  €ÿûrÚc3 Cs¨|j}|jjj}d|v}t|ƒ} | dd} | d|j¡d|vr+dg|d|d<n| ddg¡d|v} | s:dn|d} |jd} tj   | ¡}t |ƒrž|j s`tj   tjjjt |d¡¡}ntj   tj|j t |d¡¡}d  |¡}d |vr‚|r~td ƒ‚|d }t |ƒd kr|d }nd}t|ƒjd d}tj  |¡}n5| }t d  tt|ƒ¡¡}|j dd¡d }||jd<tj  tj  |¡¡\}}d||f}d}| d d¡}tj  |¡\}}|dvrâ|}nd}| dd¡}t |ƒ}d|vrý|ddkrûd}nd}t!|tjƒ}tj   |¡ "tj j#¡}tj   tj   tjj$¡d|¡}tj  %|¡}tj&|ddtj  'tj   tjjj(|¡¡}tj&|ddtj   t!tj|ƒ|¡ tj j#d¡}z t!||ƒ tj j#d¡}Wn tyh|}Ynw|d||}z dd„|jj)Dƒ} Wnt*y–dd„|j+j,Dƒ}!d d„|!Dƒ} Ynwz|  -|¡Wn ty¨Ynwzt.||||| ||| d!k| d"k| d# }"g}#Wn,t/yë}$z|j0j1}%|%j2d d$ 3|||$¡|d%}&|gfg}"|&g}#WYd}$~$nd}$~$wwdd  d&d'„| 4d¡Dƒ¡}g}'t5|"ƒD]t\}(\})}*|dr)|rdg|) 6¡¢}+n d(dgt 7|)d)¡ 6¡¢}+d  |+¡},nd},|r0g}*d*d„| ,¡Dƒ}-|(dkrE|j8rE|}.nd}.t9 :|j;pNt<¡j=| |||.t |*ƒd k|-|*|,|j>oct |*ƒ|d+ }/|' ?|/ 4d¡¡|' ?d¡q|'r€|j@|'|d,t|ƒjAddd-|"D])\})}*|*D]!}0|0 B¡D]}1tj   |tj  |1¡¡}2|1|2kr¯tC D|1|2¡q—q‘q‹|j8rÒt|||ƒjE|| krÍ|rÍtF G|¡n|d d|#S).Nr>rr;r:zplot-directiver=Úsourcerxr&zGCaption specified in both content and options. Please remove ambiguity.r r€zutf-8)r!Z _plot_counterz%s-%d.pyrw)z.pyz.rstz.txtÚ.ú-r<rFTZplot_directiverPú/cSs"g|] }tj t ¡|d¡‘qS©r)rrŽrƒr¬©rÚtrrrr“sÿzrun..cSs h|] }tj tj|d¡’qSrß)rrŽrƒrgriràrrrÚ sÿzrun..cSsg|] }tj |¡r|‘qSr)rrŽÚisfiler›rrrr“ s  ÿrr)rÑrÒrÓz.Exception occurred in plotting {} from {}: {})r}css|] }d| ¡VqdS)z N)r)rr}rrrrž*s€ÿzrun..z.. code-block:: pythonz cSs$g|]\}}|dvrd||f‘qS))r5r6r7r8r9r:z:%s: %sr)rÚkeyÚvalrrrr“=s þ ÿ) Ú default_fmtÚdest_dirÚ build_dirÚ source_linkZ multi_imagerBrØÚ source_codeZhtml_show_formatsr&)rÛ)ÚparentsrQ)Hr0r,r-rhrÉÚ setdefaultr\Z attributesrrŽrŠrÆr`rƒrgr/rRZsrcdirrZuririrrZ read_textr‰ÚtextwrapÚdedentÚmaprHrÃÚsplitextÚreplacervrÚlstripÚsepZ doctreedirÚnormpathÚmakedirsr°rSZ include_logÚAttributeErrorZ input_linesr)r*rÚr¢ZmemoÚreporterZsystem_messager<ryrÊÚ splitlinesÚindentr]Újinja2ZTemplatereÚTEMPLATEZrenderraÚextendZ insert_inputrTr”rUÚcopyfileZ write_textrrÏ)3r@rArBrCrDrEr0rhr>r‹ræZ keep_contextZ context_optZrst_fileZrst_dirZsource_file_namer&r§r|rÐZcounterÚbaseÚextZ source_extr…Zsource_rel_nameZsource_rel_dirrèrçZ dest_dir_linkZbuild_dir_linkréZsource_file_includesZpossible_sourcesrÕÚerrorsr¸r÷ZsmZ total_linesrÙr×rØÚlinesrêZoptsZsrc_linkÚresultrÔÚfnZdestimgrrrr?”sV     ÿ ÿ ÿ        þ ÿÿ ÿý ÿÿûÿ÷ ÿý €ù  ÿ ÿ ÿ ö     €ýÿÿÿür?rˆ)NN)T)FFN)=rNr²rÚiorrËrÚos.pathrZpathlibrrorUr±rírµZdocutils.parsers.rstrrZ&docutils.parsers.rst.directives.imagesrrúrjZmatplotlib.backend_basesrZmatplotlib.pyplotZpyplotrºr r r ZuseZ deprecatedr9rkrrrr$r3r4rZrgrvr~r‚rrûZexception_templateÚdictrÍr‡r¡Ú RuntimeErrorr¢r¨r¤r½rÉrÚr?rrrrÚst       ÿþ   4     5 ý k