U 1e6@sbdZddlmZddlmZddlmZddlZddlm Z ddl m Z dd d d d d dddddddddddddddZ e ee e Zded<ed<ded<ded <d!ed"<d#ed$<d%ed<d%ed <d%ed<ed=ed=ed=ed=ed=ed=ed=d&d'Zd(d)Zd@d-d.ZdAd/d0Zd1d2Zd3d4Zd5d6ZdBd7d8Zd9d:ZdCdd?ZdS)Dz%Pylab (matplotlib) support utilities.)BytesIO) b2a_base64)partialN)_pngxy) flag_callsZTkAggZGTKAggZGTK3AggZGTK4AggZWXAggZQt4AggZQt5AggZQtAggZMacOSXZnbAggZWebAggaggsvgpdfpsz)module://matplotlib_inline.backend_inlinemodule://ipympl.backend_nbagg)Ztkgtkgtk3gtk4wxZqt4Zqt5Zqt6qtosxnbaggZwebaggZnotebookrrr r inlineZipymplZwidgetr ZGTKZGTKCairor Z GTK3CairorZ GTK4CairorZWXrZCocoaAggrcGslddlm}|s&|}dd|DSg}|D]4}|j|}|dkrTtd|q.||jjq.|SdS)aGet a list of matplotlib figures by figure numbers. If no arguments are given, all available figures are returned. If the argument list contains references to invalid figures, a warning is printed but the function continues pasting further figures. Parameters ---------- figs : tuple A tuple of ints giving the figure numbers of the figures to return. r)GcfcSsg|] }|jjqS)canvasfigure).0Zfmrr;/tmp/pip-unpacked-wheel-3hxk2k58/IPython/core/pylabtools.py Xszgetfigs..Nz!Warning: figure %s not available.) Zmatplotlib._pylab_helpersrZget_all_fig_managersfigsgetprintappendrr)Zfig_numsrZ fig_managersrnumfrrrgetfigsIs  r!cCsddl}||g|jd<dS)zSet the default figure size to be [sizex, sizey]. This is just an easy to remember, convenience wrapper that sets:: matplotlib.rcParams['figure.figsize'] = [sizex, sizey] rNzfigure.figsize) matplotlibrcParams)ZsizexZsizeyr"rrrfigsizedsr$pngtightFc Ks|js|jsdS|j}|dkr*|d}d}|||||d}||t}|jdkrpddlm }|||jj |f|| } |dkr| d } n|rt | d d  d } | S) aPrint a figure to an image, and return the resulting file data Returned data will be bytes unless ``fmt='svg'``, in which case it will be unicode. Any keyword args are passed to fig.canvas.print_figure, such as ``quality`` or ``bbox_inches``. If `base64` is True, return base64-encoded str instead of raw bytes for binary-encoded image formats .. versionadded:: 7.29 base64 argument Nretinar%)formatZ facecolorZ edgecolordpi bbox_inchesr)FigureCanvasBaserzutf-8Fnewlineascii)Zaxeslinesr*Z get_facecolorZ get_edgecolorupdaterrZmatplotlib.backend_basesr, print_figuregetvaluedecoder) figfmtr+base64kwargsr*kwZbytes_ior,datarrrr2os0     r2cKs^t|fddd|}|dkr"dSt|\}}|d|dd}|rVt|ddd}||fS) zformat a figure as a pixel-doubled (retina) PNG If `base64` is True, return base64-encoded str instead of raw bytes for binary-encoded image formats .. versionadded:: 7.29 base64 argument r'Fr6r7Nr()widthheightr-r/)r2rrr4)r5r7r8Zpngdatawhmetadatarrr retina_figures  rAcsfdd}|S)a\Factory to return a matplotlib-enabled runner for %run. Parameters ---------- safe_execfile : function This must be a function with the same interface as the :meth:`safe_execfile` method of IPython. Returns ------- A function suitable for use as the ``runner`` argument of the %run magic function. csddl}ddlm}|jd}|d|f|||||jjrZ|d|j_z |j}Wnt k rxYnX|dS)zmatplotlib-aware wrapper around safe_execfile. Its interface is identical to that of the :func:`execfile` builtin. This is ultimately a call to execfile(), but wrapped in safeties to properly handle interactive rendering.rN interactiveF) r"Zmatplotlib.pyplotpyplotr#rBdraw_if_interactivecalledZdrawZdraw_allAttributeError)fnamewherer9r"pltZis_interactiveda safe_execfilerr mpl_execfiles     z mpl_runner..mpl_execfiler)rLrMrrKr mpl_runners rNc CsDz|jjj}Wn*tk r8}z t|W5d}~XYnX|dS)zreshow an nbagg figureN)rmanagerreshowrFNotImplementedError)r5rPerrr_reshow_nbagg_figures rSc sddl}ddlm|jjd}|jjd}|jjd}|jjd}t|trT|h}t|}fdd |jjD| }|d ks|d kr|jj } | t d d dddddh} || } | rdddd | D} dddd | D} td| | fd |kr0| ttfd dd|d|ksDd |kr`| ttfddi|d|kstd|kr| ttfddd|d|kr| ttfddi|d|kr| ttfddd|dS)a_Select figure formats for the inline backend. Parameters ---------- shell : InteractiveShell The main IPython instance. formats : str or set One or a set of figure formats to enable: 'png', 'retina', 'jpeg', 'svg', 'pdf'. **kwargs : any Extra keyword arguments to be passed to fig.canvas.print_figure. rNFigurez image/svg+xmlz image/pngz image/jpegzapplication/pdfcsg|]}|dqS)N)poprr rTrrr sz)select_figure_formats..rr r%Zpng2xr'ZjpgZjpegrr z%s,cSsg|] }t|qSrreprrWrrrrscSsg|] }t|qSrrYrWrrrrsz supported formats are: %s not %sTr;r7r6)r"Zmatplotlib.figurerUZdisplay_formatter formatters isinstancestrsetvalues get_backendlowerZipython_display_formatterZfor_typerS differencejoin ValueErrorrr2rA)shellformatsr8r"Z svg_formatterZ png_formatterZ jpg_formatterZ pdf_formatterZ mplbackend formatter supportedbadbsZgsrrTrselect_figure_formatssN             rkcCs`ddl}|r*|dkr*t|}|dkrXd}n.|jd}t|d}|rX||krX|}t|}||fS)aGiven a gui string return the gui and mpl backend. Parameters ---------- gui : str Can be one of ('tk','gtk','wx','qt','qt4','inline','agg'). gui_select : str Can be one of ('tk','gtk','wx','qt','qt4','inline'). This is any gui already selected by the shell. Returns ------- A tuple of (gui, backend) where backend is one of ('TkAgg','GTKAgg', 'WXAgg','Qt4Agg','module://matplotlib_inline.backend_inline','agg'). rNautorbackend)r"backendsZ rcParamsOrig backend2guir)guiZ gui_selectr"rmrrrfind_gui_and_backend-s    rqcCsJddl}|d||jd<ddlm}||d|j_t|j|_dS)z7Activate the given backend and set interactive to True.rNTrm)rCF) r"rBr#rCZswitch_backendshowZ _needmainrrD)rmr"rIrrractivate_matplotlibXs    rsTcCsHd}t|||r d}t||t|d<ddlm}||d<t|d<dS) zPopulate the namespace with pylab-related values. Imports matplotlib, pylab, numpy, and everything from pylab and numpy. Also imports a few names from IPython (figsize, display, getfigs) zbimport numpy import matplotlib from matplotlib import pylab, mlab, pyplot np = numpy plt = pyplot z3from matplotlib.pylab import * from numpy import * r$r)displayrtr!N)execr$ZIPython.displayrtr!)Zuser_nsZ import_allsrtrrr import_pylabps    rwcCs*tjdtddddlm}|||dS)a  .. deprecated:: 7.23 use `matplotlib_inline.backend_inline.configure_inline_support()` Configure an IPython shell object for matplotlib use. Parameters ---------- shell : InteractiveShell instance backend : matplotlib backend z`configure_inline_support` is deprecated since IPython 7.23, directly use `matplotlib_inline.backend_inline.configure_inline_support()`r() stacklevelr)configure_inline_supportN)warningswarnDeprecationWarningZ matplotlib_inline.backend_inlinery)rermZconfigure_inline_support_origrrrrys  ry)r%r&F)F)NN)T)__doc__iorbinasciir functoolsrrzZIPython.core.displayrZIPython.utils.decoratorsrrndictzipr_keysror!r$r2rArNrSrkrqrsrwryrrrrsl      1 0 < + !