-PhnKdZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZgd Zd Zd Zd Zd ZdZdZeeZGddZdZdZdZd#dZdZdZdZ dZ!dZ"dZ#dZ$dZ%d Z&e%e&d!Z'd"Z(dS)$aPScrapers for embedding images. Collect images that have been produced by code blocks. The only scraper we natively support is Matplotlib, others should live in modules that will support them (e.g., PyVista, Plotly). Scraped images are injected as rst ``image-sg`` directives into the ``.rst`` file generated for each example script. N) PurePosixPath)indent)filterwarnings)ExtensionError)optipng) save_figures figure_rstImagePathIterator clean_modulesmatplotlib_scrapercddl}|d|}t dt d|dkr"t d|ddlm }||fS)zImport matplotlib safely.rNaggignorezMatplotlib is currently using agg, which is a non-GUI backend, so cannot show the figure.|( |.)*is non-interactive, and thus cannot be shown)categorymessageaSphinx-Gallery relies on the matplotlib 'agg' backend to render figures and write them to files. You are currently using the {} backend. Sphinx-Gallery will terminate the build now, because changing backends is not well supported by matplotlib. We advise you to move sphinx_gallery imports before any matplotlib-dependent import. Moving sphinx_gallery imports at the top of your conf.py file should fix this issue) matplotlibuse get_backendlowerr UserWarningrformatmatplotlib.pyplotpyplot)rmatplotlib_backendplts W/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/sphinx_gallery/scrapers.py_import_matplotlibr'sNN5#//117799U"" 67=f=O6P6P   $##### s?c(g}t|dd}|'||gd}|jD]4}|D]/}||}|r||05d|}|S)N _suptitle)leftcenterright)loc, )getattrappendget_textaxes get_titlejoin)figtitlessuptitle title_locsaxr%text fig_titless r_matplotlib_fig_titlesr4Js FsK..H h''))***,,,Jh$$ $ $C<z&matplotlib_scraper..s?24PRWrc>g|]}tt|dzS)z )HLIST_IMAGE_MATPLOTLIBrrQs rrTz&matplotlib_scraper..s5   @E "VE7%;%; ;   r)&rrrmatplotlib.animationr6values isinstance_figzip get_fignumsr with_suffixr@getr( _anim_rstr4colorscolorConverterto_rgbacopyr'rcParamssavefigrAreplacestemsuffixparent Exceptioncloserstritemsr len HLIST_HEADERr,)block block_vars gallery_confkwargsrrr6r7 image_rstssrcsetanimsanifig_num image_pathr-animr3rb these_kwargsattrfig_attr default_attrdpi0rImultmultstnamehipathhikwargsconvert_to_multi_imagersts rr r ks6######......$%:;J . )F E+,Q/&/07799 & &C#y)) &"%ch"3??#4#46IJJ8 8 ":.. v  #//fX6F0FGGJjj!!99S>> !4    ij,GG H H H +C00 #2:{{}} . . .D2wsFTM2244H%.y4/?@Lwx  GGL$9$999d&>P>P%- T"  CK 3 3l 3 3 3&}5Dxw##E400Dj/K + + ..sC88$/HHFHHZ5FHH#*]4-@-@@?l?E4$;?? F//h///$* D!!&-KK    IIe      |$56 6 6 C OO\2H%I J J J%a...00 K KF \2H%IJJJJ  Y''        "~~m$$0!+M!:h!F  $ $ 0!+K!8!z_anim_rst..s*LLhc3c LLLLLLrffmpeg imagemagick)writerrAr9html5jshtmlz r;autoplay_repeatFloopr?r@rH/rrMc3"K|] }d|dV dS)z :: Nr)rRopts rrz_anim_rst..s*;;Cs;;;;;;r)videowidthheightoptions)rrdrWrrrZr]get_size_inchesroundminr[ isAvailablesaverl _repr_html_ to_jshtmlto_html5_videor_ANIMATION_RSTrr'r(rArospathrelpath_ANIMATION_VIDEO_RSTintr,)rzryrrrdrrr-fig_size thumb_sizeuse_dpirrBfmthtmlrrrA video_uris rr_r_s######DDDDDDDD )C''//J""$$H./JCLL#j(2K2KLLLLLMMG!!  & & ( (IIc*oof'I::: 1 2FAs D {!! <>>##D ""$$ ~~ dG$$$$T***  " "9s99 - -EIIelGtY&&v = !C hgbgooe\)5LMMNNI  & &)oo(1+#$$8A;$%%;;7;;;;; '  D Kr)rc0eZdZdZdZdZdZdZdZdS)r zIterate over image paths for a given example. Parameters ---------- image_path : str The template image path. cH||_t|_d|_dS)Ni@B)rylistpaths_stop)selfrys r__init__zImagePathIterator.__init__.s$VV  rc*t|jS)zReturn the number of image paths used. Returns ------- n_paths : int The number of paths. )rnrrs r__len__zImagePathIterator.__len__3s4:rc#Kt|jD]}|Vtd|jd)a[Iterate over paths. Returns ------- paths : iterable of str This enables the use of this Python pattern:: >>> for epoch in epochs: # doctest: +SKIP >>> print(epoch) # doctest: +SKIP Where ``epoch`` is given by successive outputs of :func:`mne.Epochs.next`. zGenerated over z images)rangernextr)riis r__iter__zImagePathIterator.__iter__=sV  ## H HB))++     !F4:!F!F!FGG Grc*|S)z9Return the next image path, with numbering starting at 1.)__next__rs rrzImagePathIterator.nextRs}}rc|jt|dz}|j||S)Nr)ryrrnrr()rrs rrzImagePathIterator.__next__Vs<%%c$ii!m44 $ rN) __name__ __module__ __qualname____doc__rrrrrrrrr r %sl HHH*rr )pngsvgjpggifwebpctj|d}tD]*}|d|}tj|rn+d}|d||fS)z5Find an image, tolerant of different file extensions.rr;r)rrsplitext_KNOWN_IMG_EXTSisfile)rext this_paths r_find_image_extrbsv 7  D ! !! $DOOcOO 7>>) $ $  E OOcOOS !!rc ddlm}|d}d}t|}||dD]}||||}t|ts%t d|dt |d|t||z } t| D]V} t|j || z\} } tj | st d |d | W||z }|S) aSave all open figures of the example code-block. Parameters ---------- block : sphinx_gallery.py_source_parser.Block The code block to be executed. Format (label, content, lineno). block_vars : dict Dict of block variables. gallery_conf : dict Contains the configuration of Sphinx-Gallery Returns ------- images_rst : str rst code to embed the images in the document. r_get_callablesr7rMimage_scraperszrst from scraper z was not a string, got type rzScraper z! did not produce expected image: ) gen_rstrrnrYrlrtyperrrrrr) rprqrrrr7all_rst prev_countscraperrn_newr current_pathrBs rr r nsY"('''''$%:;G())J!>,0@AAgeZ66#s##  :G::$(II::25:: '((:5,,  B-#)*r/:OL!7>>,// $XwXX,XX  3 NrrMc| d|D}fd|D}d}|r|}nr|rptj|dd}tj|ddd}t jd d |}|}t |}d} t|dkr.|d} |d} t| } t| || fz} nWt|dkrDt} t|D]-\} } || } t| } | t| || fzz } .| S) aGenerate reST for a list of image filenames. Depending on whether we have one or more figures, we use a single rst call to 'image' or a horizontal list. Parameters ---------- figure_list : list List of strings of the figures' absolute paths. sources_dir : str absolute path of Sphinx documentation sources fig_titles : str Titles of figures, empty string if no titles found. Currently only supported for matplotlib figures, default = ''. srcsetpaths : list or None List of dictionaries containing absolute paths. If empty, then srcset field is populated with the figure path. (see ``image_srcset`` configuration option). Otherwise, each dict is of the form {0: /images/image.png, 2.0: /images/image_2_00x.png} where the key is the multiplication factor and the contents the path to the image created above. Returns ------- images_rst : str rst code to embed the images in the document The rst code will have a custom ``image-sg`` directive that allows multiple resolution images to be served e.g.: ``:srcset: /plot_types/imgs/img_001.png, /plot_types/imgs/img_2_00x.png 2.00x`` Ncg|]}d|iS)rr)rRfls rrTzfigure_rst..s55522w555rcg|]S}tj|tjddTS)r)rrrrfseplstrip)rR figure_path sources_dirs rrTzfigure_rst..sY   [1199"&#FFMMcRRrrMrr z[-,_] ) rrsplitrrOrP_single_line_sanitizern_get_srcset_stSG_IMAGEro enumerateHLIST_SG_TEMPLATE) figure_listrr3rI figure_pathsalt file_namefile_name_noextfile_name_final images_rst figure_namehinamesrunns ` rr r sF65555 &L C GMM+a.11!4 '**955a82>&3@@  $ $CJ <A"1o a. W55c6 :: \  Q  ! (66 I IOB !"oG#K99F +{C.HH HJJ rc\d}|D]w}tj|||tjdd}|d|zz }|dkr|dz }m|d|ddz }x|dd dkr |d d}|dz }|S) aCreate the srcset string for including on the rst line. For example; `sources_dir` might be `/home/sample-proj/source`, hinames posix paths: 0: /home/sample-proj/source/plot_types/images/img1.png, 2.0: /home/sample-proj/source/plot_types/images/img1_2_00x.png, The result would be: '/plot_types/basic/images/sphx_glr_pie_001.png, /plot_types/basic/images/sphx_glr_pie_001_2_00x.png 2.00x' rMrrr&rz1.2fzx, N)keysrrrrfrr)rrsrcstkrs rrrs E \\^^%%wwqz;77??LLSSTWXX t 66 TMEE _____ $EE RSSzTcrc  RKE Lrc.|ddS)zRemove problematic newlines. r)rf)ss rrrs 99T3  rz% .. rst-class:: sphx-glr-horizontal z * zn * .. image-sg:: /%s :alt: %s :srcset: %s :class: sphx-glr-multi-img zM .. image-sg:: /%s :alt: %s :srcset: %s :class: sphx-glr-single-img z@ .. image:: /%s :alt: %s :class: sphx-glr-single-img ct\}}|tj|jtj|jtj|jdS)zReset matplotlib.N)r rcdefaults importlibreloadunitsdatesr)rrfnamemplrs r_reset_matplotlibr -s`!##HCNN SY SY S\"""""rcttjd}||dSdS)zReset seaborn.seabornN)sysmodulesr^reset_defaults)rrr seaborn_modules r_reset_seabornr6s;[__Y//N!%%'''''"!r)rrcVddlm}||dD]}tj|}t |jdkr[t |jd}|dkrtd|j d||||| |||d S) a@Remove, unload, or reset modules. After a script is executed it can load a variety of settings that one does not want to influence in other examples in the gallery. Parameters ---------- gallery_conf : dict The gallery configuration. fname : str or None The example being run. Will be None when this is called entering a directory of examples to be built. when : str Whether this module is run before or after examples. This parameter is only forwarded when the callables accept 3 parameters. rr reset_moduleswhenz3rd parameter in z( function signature must be 'when', got )rN) rrinspect signaturern parametersrr ValueErrorr)rrr rr reset_modulesig third_params rr r Cs&('''''&|_EE . .  -- s~  ! # #s~224455a8Kf$$ ) (=))&)) Lu4 8 8 8 8 8 Lu - - - - . .r)rMN))rrrrrOrpathlibrtextwraprwarningsr sphinx.errorsrutilsr__all__rr4rrr r_dict _scraper_dictr rrr r rrrorVrr SINGLE_IMAGEr r _reset_dictr rrrr*s !!!!!!######((((((      F"{{{|444n! 55555555t7 " " "'''TIIIIX6      ###((($ !.!.!.!.!.r