OwlCyberSecurity - MANAGER
Edit File: figure.cpython-38.pyc
U ��]̅ � @ s� d Z ddlZddlmZ ddlZddlmZ ddlmZm Z m Z ddlm Z ddlm Z ddlmZ ddlmZmZ dd lmZ ddlmZ ddlmZ ddlmZ dd l m Z m Z m Z dd l!m"Z"m#Z# dd l$m%Z% ddl&m'Z( dd l)m*Z* ddl+m,Z,m-Z- ddl.m/Z/m0Z0 ddl1m2Z2m3Z3m4Z4m5Z5 ddl6m7Z8 ddlm9Z9 e�:e;�Z<e j=j>e,� d� dd� Z?G dd� dej@�ZAG dd� deB�ZCG dd� de�ZDdd � ZEe j=j>e�FeD�d � dS ) a5 The figure module provides the top-level :class:`~matplotlib.artist.Artist`, the :class:`Figure`, which contains all the plot elements. The following classes are defined :class:`SubplotParams` control the default spacing of the subplots :class:`Figure` Top level container for all plot elements. � N)�Integral)�rcParams)�backends� docstring� projections)� __version__)� get_backend)�Artist�allow_rasterization)�FigureCanvasBase)�Axes� SubplotBase�subplot_class_factory)�BlockingMouseInput�BlockingKeyMouseInput)�GridSpec)� Rectangle)�get_projection_names� process_projection_requirements)�Text� TextWithDash)�Affine2D�Bbox�BboxTransformTo�TransformedBbox)�NonGuiException)Zprojection_namesc C s | j r|| j _d S �N)�figure�stale��self�val� r" �3/usr/lib/python3/dist-packages/matplotlib/figure.py�_stale_figure_callback1 s r$ c st e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Z� fd d �Z� fd d �Z � fdd�Z dd� Z dd� Z dd� Z � ZS )� AxesStacka Specialization of the `.Stack` to handle all tracking of `~matplotlib.axes.Axes` in a `.Figure`. This stack stores ``key, (ind, axes)`` pairs, where: * **key** should be a hash of the args and kwargs used in generating the Axes. * **ind** is a serial number for tracking the order in which axes were added. The AxesStack is a callable, where ``ax_stack()`` returns the current axes. Alternatively the :meth:`current_key_axes` will return the current key and associated axes. c s t � �� d| _d S )Nr )�super�__init__�_ind�r �� __class__r" r# r' F s zAxesStack.__init__c C s& dd� | j D �}|�� dd� |D �S )zY Return a list of the Axes instances that have been added to the figure. c S s g | ] \}}|�qS r" r" )�.0�k�ar" r" r# � <listcomp>N s z%AxesStack.as_list.<locals>.<listcomp>c S s g | ] \}}|�qS r" r" )r, �ir. r" r" r# r/ P s )� _elements�sort)r Zia_listr" r" r# �as_listJ s zAxesStack.as_listc C s2 t | j��|�}|dkr dS tjddd� |d S )zr Return the Axes instance that was added with *key*. If it is not present, return *None*. Nz2.1a) Adding an axes using the same arguments as a previous axes currently reuses the earlier instance. In a future version, a new instance will always be created and returned. Meanwhile, this warning can be suppressed, and the future behavior ensured, by passing a unique label to each axes instance.)�message� )�dictr1 �get�cbook�warn_deprecated)r �key�itemr" r" r# r7 R s �z AxesStack.getc C s$ dd� | j D �| \}}|||ffS )Nc S s i | ]\}\}}|||f�qS r" r" )r, r- �indr. r" r" r# � <dictcomp>e s z.AxesStack._entry_from_axes.<locals>.<dictcomp>)r1 )r �er< r- r" r" r# �_entry_from_axesd s zAxesStack._entry_from_axesc s t � �| �|�� dS )z Remove the axes from the stack.N)r& �remover? �r r. r* r" r# r@ h s zAxesStack.removec s t � �| �|��S )za Move the given axes, which must already exist in the stack, to the top. )r& �bubbler? rA r* r" r# rB l s zAxesStack.bubblec s� t |t�std�|���z t|� W n tk r> t� }Y nX | �|�}|dk rrt� � ||f� t � d�|�� || kr~dS | j d7 _ t� � || j |ff�S )z� Add Axes *a*, with key *key*, to the stack, and return the stack. If *key* is unhashable, replace it by a unique, arbitrary object. If *a* is already on the stack, don't add it again, but return *None*. z%second argument, {!r}, is not an AxesNz0key {!r} already existed; Axes is being replacedr5 )� isinstancer � ValueError�format�hash� TypeError�objectr7 r& r@ r8 �_warn_externalr( �push)r r: r. Z a_existingr* r" r# �adds s �z AxesStack.addc C s6 t | j�s| j| jfS | j| j \}\}}||fS dS )z� Return a tuple of ``(key, axes)`` for the active axes. If no axes exists on the stack, then returns ``(None, None)``. N)�lenr1 Z_defaultZ_pos)r r: �index�axesr" r" r# �current_key_axes� s zAxesStack.current_key_axesc C s | � � d S �Nr5 )rO r) r" r" r# �__call__� s zAxesStack.__call__c C s || � � kS r )r3 rA r" r" r# � __contains__� s zAxesStack.__contains__)�__name__� __module__� __qualname__�__doc__r' r3 r7 r? r@ rB rK rO rQ rR � __classcell__r" r" r* r# r% 6 s r% c @ s, e Zd ZdZd dd�Zd dd�Zdd� ZdS ) � SubplotParamsz7 A class to hold the parameters for a subplot. Nc C s d| _ | �||||||� dS )a All dimensions are fractions of the figure width or height. Defaults are given by :rc:`figure.subplot.[name]`. Parameters ---------- left : float The left side of the subplots of the figure. right : float The right side of the subplots of the figure. bottom : float The bottom of the subplots of the figure. top : float The top of the subplots of the figure. wspace : float The amount of width reserved for space between subplots, expressed as a fraction of the average axis width. hspace : float The amount of height reserved for space between subplots, expressed as a fraction of the average axis height. TN)�validate�update)r �left�bottom�right�top�wspace�hspacer" r" r# r' � s zSubplotParams.__init__c s� t � dd��t � dd��t � dd��t � dd��t � dd��t � dd��� �d|� � �d|� � �d|� � �d|� � �d|� � �d|� � ������fdd �}� jr� j� jkr�|� td ��� j� jkr�|� td ��dS ) zY Update the dimensions of the passed parameters. *None* means unchanged. r[ Nr] r^ r\ r_ r` c s( �� _ �� _�� _�� _�� _�� _d S r )r[ r] r^ r\ r_ r` r" �r Z thisbottomZ thishspaceZthisleftZ thisrightZthistopZ thiswspacer" r# �reset� s z#SubplotParams.update.<locals>.resetzleft cannot be >= rightzbottom cannot be >= top)�getattr� _update_thisrY r[ r] rD r\ r^ )r r[ r\ r] r^ r_ r` rb r" ra r# rZ � s( zSubplotParams.updatec C s< |d kr,t | |d �}|d kr,d| }t| }t| ||� d S )Nzfigure.subplot.)rc r �setattr)r �sr! r: r" r" r# rd � s zSubplotParams._update_this)NNNNNN)NNNNNN)rS rT rU rV r' rZ rd r" r" r" r# rX � s � � $rX c s� e Zd ZdZdd� Zdd� Zd�� fdd � Zd d � Zd�d d�Zdd� Z e e dd�Z dd� Z d�dd�Z e e e dd�Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd�d$d%�Zd�d)d*�Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd�d6d7�Zd�d8d9�Z d:d;� Z d<d=� Z d>d?� Z d@dA� Z dBdC� Z!dDdE� Z"dFdG� Z#dHdI� Z$dJdK� Z%dLdM� Z&d�dNdO�Z'd�dPdQ�Z(dRdS� Z)e e#e)�Z*dTdU� Z+d�dVdW�Z,dXdY� Z-d#ddZ�d[d\�Z.e/j0d]d^� �Z1e/j0d_d`� �Z2dadb� Z3d�ddde�Z4dfdg� Z5d�dhdi�Z6d�djdk�Z7e8dldm� �Z9dndo� Z:dpdq� Z;e/j0drds� �Z<e=�>dtdu�e/j0d�dvdw���Z?dxdy� Z@e/j0dzd{� �ZAd|d}� ZBd~d� ZC� fd�d��ZDd�d�� ZEd�d�� ZFdd��d�d��ZGe/j0d�d�d���ZHd�d�d��ZId�d�d��ZJd�d�d��ZKd�d�� ZLd�d�d��ZMd�d�� ZNd�d�d��ZOd�d�d��ZPd�d�d��ZQd�d�d��ZRd�d�d��ZSd�d�� ZT� ZUS )��Figurea� The top level container for all the plot elements. The Figure instance supports callbacks through a *callbacks* attribute which is a `.CallbackRegistry` instance. The events you can connect to are 'dpi_changed', and the callback will be called with ``func(fig)`` where fig is the `Figure` instance. Attributes ---------- patch The `.Rectangle` instance representing the figure background patch. suppressComposite For multiple figure images, the figure will make composite images depending on the renderer option_image_nocomposite function. If *suppressComposite* is a boolean, this will override the renderer. c C s dt | jj� S )Nz Figure(%gx%g))�tuple�bbox�sizer) r" r" r# �__str__ s zFigure.__str__c C s, dj | jj| jjd | jjd t| j�d�S )Nz.<{clsname} size {h:g}x{w:g} with {naxes} Axes>r r5 )Zclsname�h�wZnaxes)rE r+ rS ri rj rL rN r) r" r" r# �__repr__ s �zFigure.__repr__N� c sh t � �� | `t�� | _|dkr(td }|dkr8td }|dkrHtd }|dkrXtd }|dkrhtd }t�|�� � s�t d� |���t j d |�� | _t� �||�| _|| _t| j| j�| _t| j�| _td d d ||||d �| _| �| j� | j�d � t| � d| _ |dk�rt � }|| _ d| _ | � | � | �!|� t"� | _#| �$� d| _%t�&� | _'t�&� | _(g | _)dS )a� Parameters ---------- figsize : 2-tuple of floats, default: :rc:`figure.figsize` Figure dimension ``(width, height)`` in inches. dpi : float, default: :rc:`figure.dpi` Dots per inch. facecolor : default: :rc:`figure.facecolor` The figure patch facecolor. edgecolor : default: :rc:`figure.edgecolor` The figure patch edge color. linewidth : float The linewidth of the frame (i.e. the edge linewidth of the figure patch). frameon : bool, default: :rc:`figure.frameon` If ``False``, suppress drawing the figure background patch. subplotpars : :class:`SubplotParams` Subplot parameters. If not given, the default subplot parameters :rc:`figure.subplot.*` are used. tight_layout : bool or dict, default: :rc:`figure.autolayout` If ``False`` use *subplotpars*. If ``True`` adjust subplot parameters using `.tight_layout` with default padding. When providing a dict containing the keys ``pad``, ``w_pad``, ``h_pad``, and ``rect``, the default `.tight_layout` paddings will be overridden. constrained_layout : bool If ``True`` use constrained layout to adjust positioning of plot elements. Like ``tight_layout``, but designed to be more flexible. See :doc:`/tutorials/intermediate/constrainedlayout_guide` for examples. (Note: does not work with :meth:`.subplot` or :meth:`.subplot2grid`.) Defaults to :rc:`figure.constrained_layout.use`. N�figure.figsizez figure.dpizfigure.facecolorzfigure.edgecolorzfigure.frameonz!figure size must be finite not {}r �r r r5 )Zxy�width�height� facecolor� edgecolor� linewidthZvisibleF)r r )*r& r' Z_axesr8 �CallbackRegistry� callbacksr �np�isfinite�allrD rE r Z from_bounds� bbox_inchesr �scale�dpi_scale_trans�_dpir ri r � transFigurer �patch�_set_artist_propsZset_antialiasedr � _suptitlerX � subplotpars� _layoutbox�set_constrained_layout�set_tight_layoutr% �_axstack�clf�_cachedRendererZGrouper�_align_xlabel_grp�_align_ylabel_grp� _gridspecs) r �figsize�dpirt ru rv �frameonr� � tight_layoutZconstrained_layoutr* r" r# r' s^ 6 � � zFigure.__init__c C s* dt | j�jkr&ddlm} |�| �S d S )NZWebAggr )�backend_webagg)�type�canvasrS Zmatplotlib.backendsr� Zipython_inline_display)r r� r" r" r# � _repr_html_� s zFigure._repr_html_Tc C s� zt | jd�}W n. tk r> } ztd| ��W 5 d}~X Y nX |dk rlz|�� W dS tk rj Y nX t�� dkr�|r�t�dt � � dS )a. If using a GUI backend with pyplot, display the figure window. If the figure was not created using :func:`~matplotlib.pyplot.figure`, it will lack a :class:`~matplotlib.backend_bases.FigureManagerBase`, and will raise an AttributeError. .. warning:: This does not manage an GUI event loop. Consequently, the figure may only be shown briefly or not shown at all if you or your environment are not managing an event loop. Proper use cases for `.Figure.show` include running this from a GUI application or an IPython shell. If you're running a pure python shell or executing a non-GUI python script, you should use `matplotlib.pyplot.show` instead, which takes care of managing the event loop for you. Parameters ---------- warn : bool If ``True`` and we are not running headless (i.e. on Linux with an unset DISPLAY), issue warning when called on a non-GUI backend. �managerz]%s Figure.show works only for figures managed by pyplot, normally created by pyplot.figure().NZheadlesszXMatplotlib is currently using %s, which is a non-GUI backend, so cannot show the figure.) rc r� �AttributeError�showr r Z"_get_running_interactive_frameworkr8 rI r )r �warnr� �errr" r" r# r� � s$ � ��z Figure.showc C s | j �� S r )r� r3 r) r" r" r# � _get_axes� s zFigure._get_axesz�List of axes in the Figure. You can access the axes in the Figure through this list. Do not modify the list itself. Instead, use `~Figure.add_axes`, `~.Figure.subplot` or `~.Figure.delaxes` to add or remove an axes.)�fget�docc C s | j S r )r r) r" r" r# �_get_dpi� s zFigure._get_dpic C sF || _ | j�� �||� | �� \}}| j|||d� | j�d| � dS )z� Parameters ---------- dpi : float forward : bool Passed on to `~.Figure.set_size_inches` ��forwardZ dpi_changedN)r r~ �clearr} �get_size_inches�set_size_inchesrx Zprocess)r r� r� rm rl r" r" r# �_set_dpi� s zFigure._set_dpiz The resolution in dots per inch.)r� c C s | j S )z6Return whether `.tight_layout` is called when drawing.)�_tightr) r" r" r# �get_tight_layout� s zFigure.get_tight_layoutc C s8 |dkrt d }t|�| _t|t�r(|ni | _d| _dS )a� Set whether and how `.tight_layout` is called when drawing. Parameters ---------- tight : bool or dict with keys "pad", "w_pad", "h_pad", "rect" or None If a bool, sets whether to call `.tight_layout` upon drawing. If ``None``, use the ``figure.autolayout`` rcparam instead. If a dict, pass it as kwargs to `.tight_layout`, overriding the default paddings. Nzfigure.autolayoutT)r �boolr� rC r6 �_tight_parametersr )r Ztightr" r" r# r� � s zFigure.set_tight_layoutc C s | j S )z� Return a boolean: True means constrained layout is being used. See :doc:`/tutorials/intermediate/constrainedlayout_guide`. )� _constrainedr) r" r" r# �get_constrained_layout� s z Figure.get_constrained_layoutc C st t � | _d| jd< d| jd< d| jd< d| jd< |dkr@td }t|�| _t|t �rb| jf |� n| �� d| _dS )aG Set whether ``constrained_layout`` is used upon drawing. If None, the rcParams['figure.constrained_layout.use'] value will be used. When providing a dict containing the keys `w_pad`, `h_pad` the default ``constrained_layout`` paddings will be overridden. These pads are in inches and default to 3.0/72.0. ``w_pad`` is the width padding and ``h_pad`` is the height padding. See :doc:`/tutorials/intermediate/constrainedlayout_guide`. Parameters ---------- constrained : bool or dict or None N�w_pad�h_padr_ r` z figure.constrained_layout.useT)r6 �_constrained_layout_padsr r� r� rC �set_constrained_layout_padsr )r Z constrainedr" r" r# r� s z Figure.set_constrained_layoutc K sP ddddg}|D ]:}||kr8|| dk r8|| | j |< qtd| | j |< qdS )a� Set padding for ``constrained_layout``. Note the kwargs can be passed as a dictionary ``fig.set_constrained_layout(**paddict)``. See :doc:`/tutorials/intermediate/constrainedlayout_guide`. Parameters ---------- w_pad : scalar Width padding in inches. This is the pad around axes and is meant to make sure there is enough room for fonts to look good. Defaults to 3 pts = 0.04167 inches h_pad : scalar Height padding in inches. Defaults to 3 pts. wspace : scalar Width padding between subplots, expressed as a fraction of the subplot width. The total padding ends up being w_pad + wspace. hspace : scalar Height padding between subplots, expressed as a fraction of the subplot width. The total padding ends up being h_pad + hspace. r� r� r_ r` Nzfigure.constrained_layout.)r� r )r �kwargsZtodoZtdr" r" r# r� s �z"Figure.set_constrained_layout_padsFc C st | j d }| j d }| j d }| j d }|rh|dk s<|dk rht�| �}|j}|| |j }|| |j }||||fS )ax Get padding for ``constrained_layout``. Returns a list of `w_pad, h_pad` in inches and `wspace` and `hspace` as fractions of the subplot. See :doc:`/tutorials/intermediate/constrainedlayout_guide`. Parameters ---------- relative : boolean If `True`, then convert from inches to figure relative. r� r� r_ r` N)r� � layoutbox� get_rendererr� rr rs )r �relativer� r� r_ r` Z renderer0r� r" r" r# �get_constrained_layout_padsC s z"Figure.get_constrained_layout_pads皙�����?� r] c C s� t dd� | jD ��}t| j�dkrP| jd j|d�D ]}|�|� |�|� q4nh|r�| �� D ]Z}|�� r�|j|d�D ]}|�|� |�|� qtq\|j|d�D ]}|�d� q�|� d� q\|r�| j |d� d | _ d S ) aD Date ticklabels often overlap, so it is useful to rotate them and right align them. Also, a common use case is a number of subplots with shared xaxes where the x-axis is date data. The ticklabels are often long, and it helps to rotate them on the bottom subplot and turn them off on other subplots, as well as turn off xlabels. Parameters ---------- bottom : scalar The bottom of the subplots for :meth:`subplots_adjust`. rotation : angle in degrees The rotation of the xtick labels. ha : string The horizontal alignment of the xticklabels. which : {None, 'major', 'minor', 'both'} Selects which ticklabels to rotate. Default is None which works the same as major. c s s | ]}t |d �V qdS )� is_last_rowN)�hasattr)r, �axr" r" r# � <genexpr>w s z'Figure.autofmt_xdate.<locals>.<genexpr>r5 r )�whichF� )r\ TN) r{ rN rL Zget_xticklabelsZset_haZ set_rotation�get_axesr� � set_visibleZ set_xlabel�subplots_adjustr )r r\ Zrotation�har� Z allsubplotsZlabelr� r" r" r# � autofmt_xdate_ s" zFigure.autofmt_xdatec C s& | j f| j| j| j| j| j| j| j�S )z.Get a list of artists contained in the figure.)r� �artistsrN �lines�patches�texts�images�legendsr) r" r" r# � get_children� s �zFigure.get_childrenc C s0 | j dk r| � | |�S | j�|j|j�}|i fS )z| Test whether the mouse event occurred on the figure. Returns ------- bool, {} N)Z _containsri �contains�x�y)r Z mouseeventZinsider" r" r# r� � s zFigure.containsc O s | j S )zY Return the figure bounding box in display space. Arguments are ignored. )ri )r �argsr� r" r" r# �get_window_extent� s zFigure.get_window_extentc K s� d|kpd|k}|� dd�}|� dd�}d|kr@d|kr@d|d< d|krXd |krXd |d< d |kr�d |kr|d |kr|td |d < d|kr�d|kr�td |d< | j|||f|�}| jdk r�| j�|� | j�||f� | j�|� |�� n�|| _d| j_| jdk �rt|�st| j dd�\}}} } | j} t j | | j| j d d�| j_| j D ]0} | | jjk �rBt j| jj| g|d dd� �qBd| _| jS )aV Add a centered title to the figure. Parameters ---------- t : str The title text. x : float, default 0.5 The x location of the text in figure coordinates. y : float, default 0.98 The y location of the text in figure coordinates. horizontalalignment, ha : {'center', 'left', right'}, default: 'center' The horizontal alignment of the text relative to (*x*, *y*). verticalalignment, va : {'top', 'center', 'bottom', 'baseline'}, default: 'top' The vertical alignment of the text relative to (*x*, *y*). fontsize, size : default: :rc:`figure.titlesize` The font size of the text. See `.Text.set_size` for possible values. fontweight, weight : default: :rc:`figure.titleweight` The font weight of the text. See `.Text.set_weight` for possible values. Returns ------- text The `.Text` instance of the title. Other Parameters ---------------- fontproperties : None or dict, optional A dict of font properties. If *fontproperties* is given the default values for font size and weight are taken from the `FontProperties` defaults. :rc:`figure.titlesize` and :rc:`figure.titleweight` are ignored in this case. **kwargs Additional kwargs are :class:`matplotlib.text.Text` properties. Examples -------- >>> fig.suptitle('This is the figure title', fontsize=12) r� r� g �?g\��(\�?Zhorizontalalignmentr� �centerZverticalalignmentZvar^ ZfontpropertiesZfontsizerj zfigure.titlesizeZ fontweightZweightzfigure.titleweightNT)r� z .suptitle)�parent�artist�name� @Zrequired)ZpaddingZstrength)�popr �textr� Zset_text� set_positionZ update_fromr@ r� r� r� � LayoutBoxr� ZchildrenZvstackr ) r �tr� Zmanual_positionr� r� Zsupr� r� r_ r` �figlb�childr" r" r# �suptitle� sN 6 � � � � zFigure.suptitlec C s || _ dS )z~ Set the canvas that contains the figure Parameters ---------- canvas : FigureCanvas N)r� )r r� r" r" r# � set_canvas s zFigure.set_canvasr c s� | r<| � � � � fdd�|jd |jd fD �} | j| dd� tj| ||||| f| �} t| _| �|� | �|� |dkr�| � ||� | j � | � | j j | _ d| _| S )a� Add a non-resampled image to the figure. The image is attached to the lower or upper left corner depending on *origin*. Parameters ---------- X The image data. This is an array of one of the following shapes: - MxN: luminance (grayscale) values - MxNx3: RGB values - MxNx4: RGBA values xo, yo : int The *x*/*y* image offset in pixels. alpha : None or float The alpha blending value. norm : :class:`matplotlib.colors.Normalize` A :class:`.Normalize` instance to map the luminance to the interval [0, 1]. cmap : str or :class:`matplotlib.colors.Colormap` The colormap to use. Default: :rc:`image.cmap`. vmin, vmax : scalar If *norm* is not given, these values set the data limits for the colormap. origin : {'upper', 'lower'} Indicates where the [0, 0] index of the array is in the upper left or lower left corner of the axes. Defaults to :rc:`image.origin`. resize : bool If *True*, resize the figure to match the given image size. Returns ------- :class:`matplotlib.image.FigureImage` Other Parameters ---------------- **kwargs Additional kwargs are `.Artist` kwargs passed on to `.FigureImage`. Notes ----- figimage complements the axes image (:meth:`~matplotlib.axes.Axes.imshow`) which will be resampled to fit the current axes. If you want a resampled image to fill the entire figure, you can define an :class:`~matplotlib.axes.Axes` with extent [0,0,1,1]. Examples:: f = plt.figure() nx = int(f.get_figwidth() * f.dpi) ny = int(f.get_figheight() * f.dpi) data = np.random.random((ny, nx)) f.figimage(data) plt.show() c s g | ] }|� �qS r" r" )r, r� �r� r" r# r/ \ s z#Figure.figimage.<locals>.<listcomp>r5 r Tr� N)�get_dpi�shaper� �mimageZ FigureImager$ �stale_callbackZ set_arrayZ set_alphaZset_climr� �appendr@ �_remove_methodr )r �XZxoZyoZalphaZnormZcmapZvminZvmax�origin�resizer� r� �imr" r� r# �figimage s E" zFigure.figimagec C s� |dkr|\}}t dd� ||fD ��s6td�||���||f| j_|r�t| d�}|dk r�t| jdd�}| j| }|| }|| }t| jdd�} | dk r�| �t |�t |�� d | _ dS ) a� Set the figure size in inches. Call signatures:: fig.set_size_inches(w, h) # OR fig.set_size_inches((w, h)) optional kwarg *forward=True* will cause the canvas size to be automatically updated; e.g., you can resize the figure window from the shell ACCEPTS: a (w, h) tuple with w, h in inches See Also -------- matplotlib.Figure.get_size_inches Nc s s | ]}t �|�V qd S r )ry rz )r, �_r" r" r# r� � s z)Figure.set_size_inches.<locals>.<genexpr>z'figure size must be finite not ({}, {})r� Z _dpi_ratior5 r� T) r{ rD rE r| �p1rc r� r� r� �intr ) r rm rl r� r� ZratioZdpivalZcanvaswZcanvashr� r" r" r# r� k s& � zFigure.set_size_inchesc C s t �| jj�S )a Returns the current size of the figure in inches. Returns ------- size : ndarray The size (width, height) of the figure in inches. See Also -------- matplotlib.Figure.set_size_inches )ry �arrayr| r� r) r" r" r# r� � s zFigure.get_size_inchesc C s | j �� S )z+Get the edge color of the Figure rectangle.)r� � get_edgecolorr) r" r" r# r� � s zFigure.get_edgecolorc C s | j �� S )z+Get the face color of the Figure rectangle.)r� � get_facecolorr) r" r" r# r� � s zFigure.get_facecolorc C s | j jS )z#Return the figure width as a float.)r| rr r) r" r" r# � get_figwidth� s zFigure.get_figwidthc C s | j jS )z$Return the figure height as a float.)r| rs r) r" r" r# � get_figheight� s zFigure.get_figheightc C s | j S )z2Return the resolution in dots per inch as a float.r� r) r" r" r# r� � s zFigure.get_dpic C s | j �� S )z� Return the figure's background patch visibility, i.e. whether the figure background will be drawn. Equivalent to ``Figure.patch.get_visible()``. )r� � get_visibler) r" r" r# � get_frameon� s zFigure.get_frameonc C s | j �|� dS )zz Set the edge color of the Figure rectangle. Parameters ---------- color : color N)r� � set_edgecolor�r Zcolorr" r" r# r� � s zFigure.set_edgecolorc C s | j �|� dS )zz Set the face color of the Figure rectangle. Parameters ---------- color : color N)r� � set_facecolorr� r" r" r# r� � s zFigure.set_facecolorc C s || _ d| _dS )z Set the resolution of the figure in dots-per-inch. Parameters ---------- val : float TN)r� r r r" r" r# �set_dpi� s zFigure.set_dpic C s | j || �� |d� dS )z� Set the width of the figure in inches. Parameters ---------- val : float forward : bool r� N)r� r� �r r! r� r" r" r# � set_figwidth� s zFigure.set_figwidthc C s | j | �� ||d� dS )z� Set the height of the figure in inches. Parameters ---------- val : float forward : bool r� N)r� r� r� r" r" r# � set_figheight� s zFigure.set_figheightc C s | j �|� d| _dS )z� Set the figure's background patch visibility, i.e. whether the figure background will be drawn. Equivalent to ``Figure.patch.set_visible()``. Parameters ---------- b : bool TN)r� r� r )r �br" r" r# � set_frameon� s zFigure.set_frameonc C s* | j �|� | jD ] }|| � qd| _dS )zn Remove the `~matplotlib.axes.Axes` *ax* from the figure and update the current axes. TN)r� r@ � _axobserversr )r r� �funcr" r" r# �delaxes s zFigure.delaxesc C sN |� | � | j�|� | jj|_|�� s4|�| j� |rD|�| j � d| _ |S )aT Add any :class:`~matplotlib.artist.Artist` to the figure. Usually artists are added to axes objects using :meth:`matplotlib.axes.Axes.add_artist`, but use this method in the rare cases that adding directly to the figure is necessary. Parameters ---------- artist : `~matplotlib.artist.Artist` The artist to add to the figure. If the added artist has no transform previously set, its transform will be set to ``figure.transFigure``. clip : bool, optional, default ``False`` An optional parameter ``clip`` determines whether the added artist should be clipped by the figure patch. Default is *False*, i.e. no clipping. Returns ------- artist : The added `~matplotlib.artist.Artist` T) � set_figurer� r� r@ r� Zis_transform_set� set_transformr� Z set_clip_pathr� r )r r� �clipr" r" r# � add_artist s zFigure.add_artistc O s( dd� }dd� }||�||� � �f}|S )z+Make a hashable key out of args and kwargs.c S sJ g }| D ]8\}}z t |�}W n tk r0 Y nX |�||f� qt |�S r )rh � Exceptionr� )�items�retr- �vr" r" r# �fixitems3 s z"Figure._make_key.<locals>.fixitemsc S s2 g }| D ] }t �|�r t|�}|�|� qt|�S r )ry �iterablerh r� )r� r r. r" r" r# �fixlistC s z!Figure._make_key.<locals>.fixlist)r )r r� r� r r r: r" r" r# � _make_key0 s zFigure._make_key)�polar� projectionc O s� |r$|dk r |dkr t d| ��d}t|t�s6|dkrBt�|�}n0t|d�rf|�� \}}|jf |� n td| ��| j ||�}|||fS )z� Handle the args/kwargs to add_axes/add_subplot/gca, returning:: (axes_proj_class, proj_class_kwargs, proj_stack_key) which can be used for new axes initialization/identification. Nr zNpolar=True, yet projection=%r. Only one of these arguments should be supplied.� _as_mpl_axeszLprojection must be a string, None or implement a _as_mpl_axes method. Got %r) rD rC �strr Zget_projection_classr� r rZ rG r )r r r r� r� �projection_classZ extra_kwargsr: r" r" r# � _process_projection_requirementsN s$ �� � z'Figure._process_projection_requirementsc O s� t |�s dS | j||�}| j�|�}|dk r:| �|� |S t|d t�rf|d }|�� | k r�td��nh|d }t � |�� � s�td� |���| j ||�\}}}| j�|�}t||�r�| �|� |S || |f|�}| �||�S )a~ Add an axes to the figure. Call signatures:: add_axes(rect, projection=None, polar=False, **kwargs) add_axes(ax) Parameters ---------- rect : sequence of float The dimensions [left, bottom, width, height] of the new axes. All quantities are in fractions of figure width and height. projection : {None, 'aitoff', 'hammer', 'lambert', 'mollweide', 'polar', 'rectilinear', str}, optional The projection type of the `~.axes.Axes`. *str* is the name of a custom projection, see `~matplotlib.projections`. The default None results in a 'rectilinear' projection. polar : boolean, optional If True, equivalent to projection='polar'. sharex, sharey : `~.axes.Axes`, optional Share the x or y `~matplotlib.axis` with sharex and/or sharey. The axis will have the same limits, ticks, and scale as the axis of the shared axes. label : str A label for the returned axes. Other Parameters ---------------- **kwargs This method also takes the keyword arguments for the returned axes class. The keyword arguments for the rectilinear axes class `~.axes.Axes` can be found in the following table but there might also be other keyword arguments if another projection is used, see the actual axes class. %(Axes)s Returns ------- axes : `~.axes.Axes` (or a subclass of `~.axes.Axes`) The returned axes class depends on the projection used. It is `~.axes.Axes` if rectilinear projection are used and `.projections.polar.PolarAxes` if polar projection are used. Notes ----- If the figure already has an axes with key (*args*, *kwargs*) then it will simply make that axes current and return it. This behavior is deprecated. Meanwhile, if you do not want this behavior (i.e., you want to force the creation of a new axes), you must use a unique set of args and kwargs. The axes *label* attribute has been exposed for this purpose: if you want two axes that are otherwise identical to be added to the figure, make sure you give them unique labels. In rare circumstances, `.add_axes` may be called with a single argument, a axes instance already created in the present figure but not in the figure's list of axes. See Also -------- .Figure.add_subplot .pyplot.subplot .pyplot.axes .Figure.subplots .pyplot.subplots Examples -------- Some simple examples:: rect = l, b, w, h fig = plt.figure() fig.add_axes(rect,label=label1) fig.add_axes(rect,label=label2) fig.add_axes(rect, frameon=False, facecolor='g') fig.add_axes(rect, polar=True) ax=fig.add_axes(rect, projection='polar') fig.delaxes(ax) fig.add_axes(ax) Nr z5The Axes must have been created in the present figurez)all entries in rect must be finite not {})rL r r� r7 �scarC r � get_figurerD ry rz r{ rE r �_add_axes_internal)r r� r� r: r� r. �rectr r" r" r# �add_axesn s4 [ �� � zFigure.add_axesc O s t |�s d}t |�dkrht|d t�rhd|d kr>dksRn td�|d ���tttt|d ���}t|d t �r�|d }|� � | k r�td��| j ||�}n^| j ||�\}}}| j �|�}|dk r�t||�r�| �|� |S | j �|� t|�| f|�|�}| �||�S ) a� Add an `~.axes.Axes` to the figure as part of a subplot arrangement. Call signatures:: add_subplot(nrows, ncols, index, **kwargs) add_subplot(pos, **kwargs) add_subplot(ax) add_subplot() Parameters ---------- *args Either a 3-digit integer or three separate integers describing the position of the subplot. If the three integers are *nrows*, *ncols*, and *index* in order, the subplot will take the *index* position on a grid with *nrows* rows and *ncols* columns. *index* starts at 1 in the upper left corner and increases to the right. *pos* is a three digit integer, where the first digit is the number of rows, the second the number of columns, and the third the index of the subplot. i.e. fig.add_subplot(235) is the same as fig.add_subplot(2, 3, 5). Note that all integers must be less than 10 for this form to work. If no positional arguments are passed, defaults to (1, 1, 1). In rare circumstances, `.add_subplot` may be called with a single argument, a subplot axes instance already created in the present figure but not in the figure's list of axes. projection : {None, 'aitoff', 'hammer', 'lambert', 'mollweide', 'polar', 'rectilinear', str}, optional The projection type of the subplot (`~.axes.Axes`). *str* is the name of a custom projection, see `~matplotlib.projections`. The default None results in a 'rectilinear' projection. polar : boolean, optional If True, equivalent to projection='polar'. sharex, sharey : `~.axes.Axes`, optional Share the x or y `~matplotlib.axis` with sharex and/or sharey. The axis will have the same limits, ticks, and scale as the axis of the shared axes. label : str A label for the returned axes. Other Parameters ---------------- **kwargs This method also takes the keyword arguments for the returned axes base class. The keyword arguments for the rectilinear base class `~.axes.Axes` can be found in the following table but there might also be other keyword arguments if another projection is used. %(Axes)s Returns ------- axes : `.axes.SubplotBase`, or another subclass of `~.axes.Axes` The axes of the subplot. The returned axes base class depends on the projection used. It is `~.axes.Axes` if rectilinear projection are used and `.projections.polar.PolarAxes` if polar projection are used. The returned axes is then a subplot subclass of the base class. Notes ----- If the figure already has a subplot with key (*args*, *kwargs*) then it will simply make that subplot current and return it. This behavior is deprecated. Meanwhile, if you do not want this behavior (i.e., you want to force the creation of a new subplot), you must use a unique set of args and kwargs. The axes *label* attribute has been exposed for this purpose: if you want two subplots that are otherwise identical to be added to the figure, make sure you give them unique labels. See Also -------- .Figure.add_axes .pyplot.subplot .pyplot.axes .Figure.subplots .pyplot.subplots Examples -------- :: fig = plt.figure() fig.add_subplot(221) # equivalent but more general ax1 = fig.add_subplot(2, 2, 1) # add a subplot with no frame ax2 = fig.add_subplot(222, frameon=False) # add a polar subplot fig.add_subplot(223, projection='polar') # add a red subplot that share the x-axis with ax1 fig.add_subplot(224, sharex=ax1, facecolor='red') #delete x2 from the figure fig.delaxes(ax2) #add x2 to the figure again fig.add_subplot(ax2) )r5 r5 r5 r5 r �d i� zBInteger subplot specification must be a three-digit number, not {}z8The Subplot must have been created in the present figureN)rL rC r rD rE rh �mapr� r r r r r r� r7 r r@ r r )r r� r� r. r: r r� r" r" r# � add_subplot� s2 s� � � zFigure.add_subplotc C s0 | j �||� | �|� | j|_d| _t|_|S )z0Private helper for `add_axes` and `add_subplot`.T)r� rK r � _remove_axr� r r$ r� )r r: r� r" r" r# r � s zFigure._add_axes_internalr5 c C s t |t�r|rdnd}t |t�r,|r(dnd}t |t�r@t�d� tjddddg||d� |dkrdi }|dkrpi }|�� }|�� }| �� r�t||fd| i|��}nt||fddi|��}| j � |� t j ||ft d �} t|�D ]j} t|�D ]\} d| d | | d f | d | f d �} | | |d < | | |d< | j|| | f f|�| | | f< q�q�|dk�r�| dd�dd�f jD ]&} | jjdddd� | jj�d� �qh|dk�r�| dd�dd�f jD ]&} | jjdddd� | jj�d� �q�|�r�| jdk�r�| �� S | �� S | S dS )a� Add a set of subplots to this figure. This utility wrapper makes it convenient to create common layouts of subplots in a single call. Parameters ---------- nrows, ncols : int, optional, default: 1 Number of rows/columns of the subplot grid. sharex, sharey : bool or {'none', 'all', 'row', 'col'}, default: False Controls sharing of properties among x (`sharex`) or y (`sharey`) axes: - True or 'all': x- or y-axis will be shared among all subplots. - False or 'none': each subplot x- or y-axis will be independent. - 'row': each subplot row will share an x- or y-axis. - 'col': each subplot column will share an x- or y-axis. When subplots have a shared x-axis along a column, only the x tick labels of the bottom subplot are created. Similarly, when subplots have a shared y-axis along a row, only the y tick labels of the first column subplot are created. To later turn other subplots' ticklabels on, use `~matplotlib.axes.Axes.tick_params`. squeeze : bool, optional, default: True - If True, extra dimensions are squeezed out from the returned array of Axes: - if only one subplot is constructed (nrows=ncols=1), the resulting single Axes object is returned as a scalar. - for Nx1 or 1xM subplots, the returned object is a 1D numpy object array of Axes objects. - for NxM, subplots with N>1 and M>1 are returned as a 2D array. - If False, no squeezing at all is done: the returned Axes object is always a 2D array containing Axes instances, even if it ends up being 1x1. subplot_kw : dict, optional Dict with keywords passed to the :meth:`~matplotlib.figure.Figure.add_subplot` call used to create each subplot. gridspec_kw : dict, optional Dict with keywords passed to the `~matplotlib.gridspec.GridSpec` constructor used to create the grid the subplots are placed on. Returns ------- ax : `~.axes.Axes` object or array of Axes objects. *ax* can be either a single `~matplotlib.axes.Axes` object or an array of Axes objects if more than one subplot was created. The dimensions of the resulting array can be controlled with the squeeze keyword, see above. Examples -------- :: # First create some toy data: x = np.linspace(0, 2*np.pi, 400) y = np.sin(x**2) # Create a figure plt.figure() # Creates a subplot ax = fig.subplots() ax.plot(x, y) ax.set_title('Simple plot') # Creates two subplots and unpacks the output array immediately ax1, ax2 = fig.subplots(1, 2, sharey=True) ax1.plot(x, y) ax1.set_title('Sharing Y axis') ax2.scatter(x, y) # Creates four polar axes, and accesses them through the # returned array axes = fig.subplots(2, 2, subplot_kw=dict(polar=True)) axes[0, 0].plot(x, y) axes[1, 1].scatter(x, y) # Share a X axis with each column of subplots fig.subplots(2, 2, sharex='col') # Share a Y axis with each row of subplots fig.subplots(2, 2, sharey='row') # Share both X and Y axes with all subplots fig.subplots(2, 2, sharex='all', sharey='all') # Note that this is the same as fig.subplots(2, 2, sharex=True, sharey=True) See Also -------- .pyplot.subplots .Figure.add_subplot .pyplot.subplot r{ �nonez]sharex argument to subplots() was an integer. Did you intend to use subplot() (without 's')?�row�col)�sharex�shareyNr )Zdtyperq r )r r{ r r r r )r r{ ���ZbothF)r� Z labelbottomZlabeltop)r r{ r5 )r� Z labelleftZ labelright)rC r� r r8 rI Z_check_in_list�copyr� r r� r� ry �emptyrH �ranger Zflat�xaxisZset_tick_paramsZ offsetTextr� �yaxisrj r; �squeeze)r �nrows�ncolsr r r# Z subplot_kwZ gridspec_kw�gsZaxarrr r Z shared_withr� r" r" r# �subplots� sb n � � � " � � zFigure.subplotsc C sZ dd� }dd� }| � |� |||j�}|d k r8||j� |||j�}|d k rV||j� d S )Nc S s< | � | �� � | �| �� � | �| �� � | �| �� � d S r )Zset_major_formatterZget_major_formatterZset_major_locatorZget_major_locatorZset_minor_formatterZget_minor_formatterZset_minor_locatorZget_minor_locator)Zaxisr" r" r# �_reset_loc_form@ s z*Figure._remove_ax.<locals>._reset_loc_formc S s> |� | �}t|�dkr:|�| � |D ]}| |k r$| S q$d S rP )Z get_siblingsrL r@ )r� ZgrouperZsiblings�last_axr" r" r# �_break_share_linkF s z,Figure._remove_ax.<locals>._break_share_link)r� Z_shared_y_axesr" Z_shared_x_axesr! )r r� r( r* r) r" r" r# r ? s zFigure._remove_axc C s� d| _ t�� | _t| j�D ]}|�� | �|� qt| j dd�}|dk rP|� � | j � � g | _ g | _g | _g | _g | _g | _|s�g | _d| _| �� r�t�| j� d| _dS )z� Clear the figure. Set *keep_observers* to True if, for example, a gui widget is tracking the axes in the figure. N�toolbarT)�suppressCompositer8 rw rx rh rN Zclar� rc r� rZ r� r� r� r� r� r� r� r� r� r� r� r� Znonetreer� r )r �keep_observersr� r+ r" r" r# r� X s* z Figure.clfc C s | j |d� dS )z> Clear the figure -- synonym for :meth:`clf`. )r- N)r� )r r- r" r" r# r� w s z Figure.clearc C s^ | � � s dS | �� }|�| j� tdd� |D �dd� d�}| jD ]p}|�� }|rf|||�}|�|� n|�� |�� D ]8}t|d�rv|�� }|r�|||�}|�|� qv|�� qvq@z�|� d � | � � r�| jr�| � |� | � � �r| j�rz| j|f| j� W n tk �r Y nX | j�|� t�|| || j� |�d � W 5 d| _ X || _| j�|� dS ) zu Render the figure using :class:`matplotlib.backend_bases.RendererBase` instance *renderer*. Nc s s | ]}|� � s|V qd S r )Z get_animated�r, r� r" r" r# r� � s z Figure.draw.<locals>.<genexpr>c S s | � � S r )Z get_zorder)r� r" r" r# �<lambda>� � z Figure.draw.<locals>.<lambda>)r: � apply_aspectFr )r� r� r@ r� �sortedrN Zget_axes_locatorr1 r� r Z open_groupr� �execute_constrained_layoutr� r� r� rD �drawr� Z _draw_list_compositing_imagesr, Z close_groupr� r� Z draw_event)r �rendererr� r� Zlocator�posr� r" r" r# r4 } sV � � �z Figure.drawc C s" | j dkrtd��|�| j � dS )z� Draw :class:`matplotlib.artist.Artist` instance *a* only. This is available only after the figure is drawn. NzLdraw_artist can only be used after an initial draw which caches the renderer)r� r� r4 rA r" r" r# � draw_artist� s zFigure.draw_artistc C s | j S )aY Return a list of axes in the Figure. You can access and modify the axes in the Figure through this list. Do not modify the list itself. Instead, use `~Figure.add_axes`, `~.Figure.subplot` or `~.Figure.delaxes` to add or remove an axes. Note: This is equivalent to the property `~.Figure.axes`. )rN r) r" r" r# r� � s zFigure.get_axesc O sZ t j| jf|�|�\}}}}t|�r$t j| ||f|�|�}| j�|� | jj|_d| _ |S )at Place a legend on the figure. To make a legend from existing artists on every axes:: legend() To make a legend for a list of lines and labels:: legend( (line1, line2, line3), ('label1', 'label2', 'label3'), loc='upper right') These can also be specified by keyword:: legend( handles=(line1, line2, line3), labels=('label1', 'label2', 'label3'), loc='upper right') Parameters ---------- handles : sequence of `.Artist`, optional A list of Artists (lines, patches) to be added to the legend. Use this together with *labels*, if you need full control on what is shown in the legend and the automatic mechanism described above is not sufficient. The length of handles and labels should be the same in this case. If they are not, they are truncated to the smaller length. labels : sequence of strings, optional A list of labels to show next to the artists. Use this together with *handles*, if you need full control on what is shown in the legend and the automatic mechanism described above is not sufficient. Other Parameters ---------------- %(_legend_kw_doc)s Returns ------- :class:`matplotlib.legend.Legend` instance Notes ----- Not all kinds of artist are supported by the legend command. See :doc:`/tutorials/intermediate/legend_guide` for details. T) �mlegendZ_parse_legend_argsrN rL ZLegendr� r� r@ r� r )r r� r� Zhandles�labelsZ extra_args�lr" r" r# �legend� s 8��� z Figure.legend�3.1�withdashc K s� t | jd�}|r,|tjjk r,t|||d�}nt|||d�}|�|� |dk rV|�|� |�|� |�| � t |_ | j � |� | j j |_d| _|S )aU Add text to figure. Parameters ---------- x, y : float The position to place the text. By default, this is in figure coordinates, floats in [0, 1]. The coordinate system can be changed using the *transform* keyword. s : str The text string. fontdict : dictionary, optional, default: None A dictionary to override the default text properties. If fontdict is None, the defaults are determined by your rc parameters. A property in *kwargs* override the same property in fontdict. withdash : boolean, optional, default: False Creates a `~matplotlib.text.TextWithDash` instance instead of a `~matplotlib.text.Text` instance. Other Parameters ---------------- **kwargs : `~matplotlib.text.Text` properties Other miscellaneous text parameters. %(Text)s Returns ------- text : `~.text.Text` See Also -------- .Axes.text .pyplot.text )Z transform)r� r� r� NT)r6 r� r8 Z deprecationZ_deprecated_parameterr r rZ r� r$ r� r� r� r@ r� r ) r r� r� rf Zfontdictr= r� �defaultr� r" r" r# r� s ( � z Figure.textc C s( || kr|� | � t|_|�| j� d S r )r� r$ r� r� r� rA r" r" r# r� Z s zFigure._set_artist_propsc K st | j �� \}}|dk rh|s |S | jf |�\}}}|dd� }|dd� }||kr^t||�r^|S t�d� | jd|�S )a Get the current axes, creating one if necessary. The following kwargs are supported for ensuring the returned axes adheres to the given projection etc., and for axes creation if the active axes does not exist: %(Axes)s Nr5 zlRequested projection is different from current axis projection, creating new axis with requested projection.)r5 r5 r5 )r� rO r rC r8 rI r )r r� Zckey�caxr r� r: r" r" r# �gca` s � z Figure.gcac C s$ | j �|� | jD ] }|| � q|S )z*Set the current axes to be a and return a.)r� rB r� )r r. r� r" r" r# r � s z Figure.scac C sZ | j �� d }|dkrdS |�� }|dk r.|S t| j�D ] }|�� }|dk r8| S q8dS )zR Helper for :func:`~matplotlib.pyplot.gci`. Do not use elsewhere. r5 N)r� rO �_gci�reversedrN )r r? r� r� r" r" r# rA � s z Figure._gcic sx t � �� }dD ]}|�|d � qt|d< ddlm} t| jdd �|jj � � krVd|d< |�dd � | j d k rtd | j _ |S ) N)r� r� r� r� �__mpl_version__r )�_pylab_helpersr� T�_restore_to_pylabr� ) r& � __getstate__r� � _mpl_version� matplotlibrD rc r� �GcfZfigs�valuesr� r� )r �stateZ attr_to_poprD r* r" r# rF � s � zFigure.__getstate__c s� |� d�}|� dd�}|tkr0t�d|� d�� || _g | _d | _d | _|r�dd lm } dd l m � |� � }|r|t |�d nd}|j�|| �� | �� r�� �| �� � � �fdd �}� j�d |�� _�j�� � || _|�� d | _d S ) NrC rE Fz.This figure was saved with matplotlib version z' and is unlikely to function correctly.r r5 c s �j �� � d S r )rI � set_active)Zevent�ZmgrZ pylab_helpersr" r# � make_active� s z(Figure.__setstate__.<locals>.make_activeZbutton_press_eventT)r� rG r8 rI �__dict__r� r� r� Zmatplotlib.pyplotZpyplotZmatplotlib._pylab_helpersrD Z get_fignums�maxZ _backend_modZ new_figure_manager_given_figureZ get_labelZset_window_titleZ mpl_connectZ_cidgcfrI rL ZnumberZdraw_if_interactiver )r rK �versionZrestore_to_pylabZpltZallnumsZnumrN r" rM r# � __setstate__� s4 � � zFigure.__setstate__c C s | j �|� dS )z>Whenever the axes state change, ``func(self)`` will be called.N)r� r� )r r� r" r" r# �add_axobserver� s zFigure.add_axobserver)� transparentc K sZ |� dtd � d|krJtjddddd� |�d�}|dkrNt�td �}nd }|dkr^td }|r�|� dd � |� d d � g }| jD ]4}|j}|� |� � |� � f� |� d � |� d � q�n |� dtd � |� d td � |r�| j�� }| j�|� | jj|f|� |�r| j�|� |�rVt| j|�D ]*\}} |j� | d � |j� | d � �q*dS )a� Save the current figure. Call signature:: savefig(fname, dpi=None, facecolor='w', edgecolor='w', orientation='portrait', papertype=None, format=None, transparent=False, bbox_inches=None, pad_inches=0.1, frameon=None, metadata=None) The output formats available depend on the backend being used. Parameters ---------- fname : str or PathLike or file-like object A path, or a Python file-like object, or possibly some backend-dependent object such as `matplotlib.backends.backend_pdf.PdfPages`. If *format* is not set, then the output format is inferred from the extension of *fname*, if any, and from :rc:`savefig.format` otherwise. If *format* is set, it determines the output format. Hence, if *fname* is not a path or has no extension, remember to specify *format* to ensure that the correct backend is used. Other Parameters ---------------- dpi : [ *None* | scalar > 0 | 'figure' ] The resolution in dots per inch. If *None*, defaults to :rc:`savefig.dpi`. If 'figure', uses the figure's dpi value. quality : [ *None* | 1 <= scalar <= 100 ] The image quality, on a scale from 1 (worst) to 95 (best). Applicable only if *format* is jpg or jpeg, ignored otherwise. If *None*, defaults to :rc:`savefig.jpeg_quality` (95 by default). Values above 95 should be avoided; 100 completely disables the JPEG quantization stage. optimize : bool If *True*, indicates that the JPEG encoder should make an extra pass over the image in order to select optimal encoder settings. Applicable only if *format* is jpg or jpeg, ignored otherwise. Is *False* by default. progressive : bool If *True*, indicates that this image should be stored as a progressive JPEG file. Applicable only if *format* is jpg or jpeg, ignored otherwise. Is *False* by default. facecolor : color spec or None, optional The facecolor of the figure; if *None*, defaults to :rc:`savefig.facecolor`. edgecolor : color spec or None, optional The edgecolor of the figure; if *None*, defaults to :rc:`savefig.edgecolor` orientation : {'landscape', 'portrait'} Currently only supported by the postscript backend. papertype : str One of 'letter', 'legal', 'executive', 'ledger', 'a0' through 'a10', 'b0' through 'b10'. Only supported for postscript output. format : str The file format, e.g. 'png', 'pdf', 'svg', ... The behavior when this is unset is documented under *fname*. transparent : bool If *True*, the axes patches will all be transparent; the figure patch will also be transparent unless facecolor and/or edgecolor are specified via kwargs. This is useful, for example, for displaying a plot on top of a colored background on a web page. The transparency of these patches will be restored to their original values upon exit of this function. bbox_inches : str or `~matplotlib.transforms.Bbox`, optional Bbox in inches. Only the given portion of the figure is saved. If 'tight', try to figure out the tight bbox of the figure. If None, use savefig.bbox pad_inches : scalar, optional Amount of padding around the figure when bbox_inches is 'tight'. If None, use savefig.pad_inches bbox_extra_artists : list of `~matplotlib.artist.Artist`, optional A list of extra artists that will be considered when the tight bbox is calculated. metadata : dict, optional Key/value pairs to store in the image metadata. The supported keys and defaults depend on the image format and backend: - 'png' with Agg backend: See the parameter ``metadata`` of `~.FigureCanvasAgg.print_png`. - 'pdf' with pdf backend: See the parameter ``metadata`` of `~.backend_pdf.PdfPages`. - 'eps' and 'ps' with PS backend: Only 'Creator' is supported. pil_kwargs : dict, optional Additional keyword arguments that are passed to `PIL.Image.save` when saving the figure. Only applicable for formats that are saved using Pillow, i.e. JPEG, TIFF, and (if the keyword is set to a non-None value) PNG. r� z savefig.dpir� r< Zkwargrt )r� Zobj_typeZ alternativeNzsavefig.frameonFzsavefig.transparentr ru zsavefig.facecolorzsavefig.edgecolorr r5 )� setdefaultr r8 r9 r� r6 � __getitem__rN r� r� r� r� r� r� r� r� r� Z print_figure�zip) r �fnamerT r� r� Zoriginal_axes_colorsr� r� Zoriginal_frameonZccr" r" r# �savefig� sD p � � zFigure.savefigc s� |dkr| � � }| � � }|dkr\|rJt|t�rJ| �� sJtj|f|�\}}ntj|f|�\}}ddddddg� � fdd �|�� D �}tj||f|�}| � |� d | _ |S ) z� Create a colorbar for a ScalarMappable instance, *mappable*. Documentation for the pyplot thin wrapper: %(colorbar_doc)s NZfraction�padZshrinkZaspectZanchorZpanchorc s i | ]\}}|� kr||�qS r" r" )r, r- r �ZNON_COLORBAR_KEYSr" r# r= � s z#Figure.colorbar.<locals>.<dictcomp>T) r@ rC r r� �cbarZmake_axes_gridspecZ make_axesr Zcolorbar_factoryr r ) r Zmappabler? r� Z use_gridspec�kwZ current_axZcb_kw�cbr" r[ r# �colorbar� s" � � zFigure.colorbarc C s� | � � r | �d� t�d� | j�||||||� | jD ]n}t|t�s�t|j t�rl|j � � |� |j j � q�t|j t�r�|j � � |� |j j � q8|� � |� |j � q8d| _dS )z� Update the :class:`SubplotParams` with *kwargs* (defaulting to rc when *None*) and update the subplot locations. Fz�This figure was using constrained_layout==True, but that is incompatible with subplots_adjust and or tight_layout: setting constrained_layout==False. TN)r� r� r8 rI r� rZ rN rC r Z_sharexZ update_paramsr� ZfigboxZ_shareyr )r r[ r\ r] r^ r_ r` r� r" r" r# r� � s zFigure.subplots_adjust� � c C s t | |||d�}||||d�S )a� Blocking call to interact with a figure. Wait until the user clicks *n* times on the figure, and return the coordinates of each click in a list. There are three possible interactions: - Add a point. - Remove the most recently added point. - Stop the interaction and return the points added so far. The actions are assigned to mouse buttons via the arguments *mouse_add*, *mouse_pop* and *mouse_stop*. Mouse buttons are defined by the numbers: - 1: left mouse button - 2: middle mouse button - 3: right mouse button - None: no mouse button Parameters ---------- n : int, optional, default: 1 Number of mouse clicks to accumulate. If negative, accumulate clicks until the input is terminated manually. timeout : scalar, optional, default: 30 Number of seconds to wait before timing out. If zero or negative will never timeout. show_clicks : bool, optional, default: True If True, show a red cross at the location of each click. mouse_add : {1, 2, 3, None}, optional, default: 1 (left click) Mouse button used to add points. mouse_pop : {1, 2, 3, None}, optional, default: 3 (right click) Mouse button used to remove the most recently added point. mouse_stop : {1, 2, 3, None}, optional, default: 2 (middle click) Mouse button used to stop input. Returns ------- points : list of tuples A list of the clicked (x, y) coordinates. Notes ----- The keyboard can also be used to select points in case your mouse does not have one or more of the buttons. The delete and backspace keys act like right clicking (i.e., remove last point), the enter key terminates input and any other key (not already used by the window manager) selects a point. )� mouse_add� mouse_pop� mouse_stop)�n�timeout� show_clicks)r )r re rf rg rb rc rd Zblocking_mouse_inputr" r" r# �ginput� s 6��z Figure.ginputr c C s t | �}||d�S )a Blocking call to interact with the figure. This will return True is a key was pressed, False if a mouse button was pressed and None if *timeout* was reached without either being pressed. If *timeout* is negative, does not timeout. )rf )r )r rf Zblocking_inputr" r" r# �waitforbuttonpress s zFigure.waitforbuttonpressc C sN dd� | � � D �}| jD ]}|�� r|�|�� � q| j|krJ|�| j� |S )Nc S s g | ]}|� � r|�� r|�qS r" )r� Z get_in_layoutr. r" r" r# r/ s �z9Figure.get_default_bbox_extra_artists.<locals>.<listcomp>)r� rN r� �extend� get_default_bbox_extra_artistsr� r@ )r Z bbox_artistsr� r" r" r# rk s z%Figure.get_default_bbox_extra_artistsc C s� g }|dkr| � � }n|}|D ]4}|�|�}|dk r |jdksH|jdkr |�|� q | jD ]H}|�� rZz|j||d�}W n tk r� |�|�}Y nX |�|� qZdd� |D �}t|�dkr�| j S t � |�}t |t � �d| j ��} | S )a� Return a (tight) bounding box of the figure in inches. Artists that have ``artist.set_in_layout(False)`` are not included in the bbox. Parameters ---------- renderer : `.RendererBase` instance renderer that will be used to draw the figures (i.e. ``fig.canvas.get_renderer()``) bbox_extra_artists : list of `.Artist` or ``None`` List of artists to include in the tight bounding box. If ``None`` (default), then all artist children of each axes are included in the tight bounding box. Returns ------- bbox : `.BboxBase` containing the bounding box (in figure inches). Nr )�bbox_extra_artistsc S s< g | ]4}t �|j�rt �|j�r|jd ks4|jd kr|�qS )r )ry rz rr rs )r, r� r" r" r# r/ M s �z(Figure.get_tightbbox.<locals>.<listcomp>� �?)rk � get_tightbboxrr rs r� rN r� rG rL r| r �unionr r r} r� ) r r5 rl Zbbr� r. ri r� Z_bboxr| r" r" r# rn s2 � �zFigure.get_tightbboxc C s2 | j dkr.tjdd| d�| _ | j �dddd� dS )z7Initialize the layoutbox for use in constrained_layout.Nr� )r� r� r� ro rm )r� r� r� Zconstrain_geometryr) r" r" r# �init_layoutbox[ s �zFigure.init_layoutboxc C s� ddl m} t�d� | jdkr.t�d� dS | �� \}}}}| }|�� \}} || }|| }|dkrpt � |�}|||||||� dS )z} Use ``layoutbox`` to determine pos positions within axes. See also `.set_constrained_layout_pads`. r )�do_constrained_layoutzExecuting constrainedlayoutNz�Calling figure.constrained_layout, but figure not setup to do constrained layout. You either called GridSpec without the fig keyword, you are using plt.subplot, or you need to call figure or subplots with the constrained_layout=True kwarg.) Z matplotlib._constrained_layoutrq �_log�debugr� r8 rI r� r� r� r� ) r r5 rq r� r� r_ r` Zfigrr rs r" r" r# r3 c s z!Figure.execute_constrained_layout�H�z�G�?c C sn ddl m}m}m} || j�} d| kr0t�d� |dkr@|| �}|| | j| |||||d�} | rj| jf | � dS )an Automatically adjust subplot parameters to give specified padding. To exclude an artist on the axes from the bounding box calculation that determines the subplot parameters (i.e. legend, or annotation), then set `a.set_in_layout(False)` for that artist. Parameters ---------- renderer : subclass of `~.backend_bases.RendererBase`, optional Defaults to the renderer for the figure. pad : float, optional Padding between the figure edge and the edges of subplots, as a fraction of the font size. h_pad, w_pad : float, optional Padding (height/width) between edges of adjacent subplots, as a fraction of the font size. Defaults to *pad*. rect : tuple (left, bottom, right, top), optional A rectangle (left, bottom, right, top) in the normalized figure coordinate that the whole subplots area (including labels) will fit into. Default is (0, 0, 1, 1). See Also -------- .Figure.set_tight_layout .pyplot.tight_layout r5 )r� �get_subplotspec_list�get_tight_layout_figureNzcThis figure includes Axes that are not compatible with tight_layout, so results might be incorrect.)rZ r� r� r )r� r� ru rv rN r8 rI r� ) r r5 rZ r� r� r r� ru rv Zsubplotspec_listr� r" r" r# r� s$ �zFigure.tight_layoutc C s� |dkr| j }t�|��� }|D ]�}t�d|�� � |�� }|�� \}}}}}} |j � � } |D ]\} | j � � | kr^| �� }|�� \}}} } }} | dkr�| |ks�| dkr^| |kr^| j � || � q^q dS )at Align the ylabels of subplots in the same subplot column if label alignment is being done automatically (i.e. the label position is not manually set). Alignment persists for draw events after this is called. If a label is on the bottom, it is aligned with labels on axes that also have their label on the bottom and that have the same bottom-most subplot row. If the label is on the top, it is aligned with labels on axes with the same top-most row. Parameters ---------- axs : list of `~matplotlib.axes.Axes` Optional list of (or ndarray) `~matplotlib.axes.Axes` to align the xlabels. Default is to align all axes on the figure. See Also -------- matplotlib.figure.Figure.align_ylabels matplotlib.figure.Figure.align_labels Notes ----- This assumes that ``axs`` are from the same `.GridSpec`, so that their `.SubplotSpec` positions correspond to figure positions. Examples -------- Example with rotated xtick labels:: fig, axs = plt.subplots(1, 2) for tick in axs[0].get_xticklabels(): tick.set_rotation(55) axs[0].set_xlabel('XLabel 0') axs[1].set_xlabel('XLabel 1') fig.align_xlabels() N� Working on: %sr\ r^ ) rN ry �asarray�ravelrr rs Z get_xlabel�get_subplotspec�get_rows_columnsr! �get_label_positionr� �join)r �axsr� �ssr$ r% �row0�row1�col0�col1�labpo�axcZrowc0Zrowc1Zcolcr" r" r# � align_xlabels� s& , ���zFigure.align_xlabelsc C s� |dkr| j }t�|��� }|D ]�}t�d|�� � |�� }|�� \}}}}}} |j � � } |D ]d} | |kr^| j � � | kr^| �� }|�� \}}}}} } | dkr�| |ks�| dkr^| | kr^| j � || � q^q dS )aO Align the ylabels of subplots in the same subplot column if label alignment is being done automatically (i.e. the label position is not manually set). Alignment persists for draw events after this is called. If a label is on the left, it is aligned with labels on axes that also have their label on the left and that have the same left-most subplot column. If the label is on the right, it is aligned with labels on axes with the same right-most column. Parameters ---------- axs : list of `~matplotlib.axes.Axes` Optional list (or ndarray) of `~matplotlib.axes.Axes` to align the ylabels. Default is to align all axes on the figure. See Also -------- matplotlib.figure.Figure.align_xlabels matplotlib.figure.Figure.align_labels Notes ----- This assumes that ``axs`` are from the same `.GridSpec`, so that their `.SubplotSpec` positions correspond to figure positions. Examples -------- Example with large yticks labels:: fig, axs = plt.subplots(2, 1) axs[0].plot(np.arange(0, 1000, 50)) axs[0].set_ylabel('YLabel 0') axs[1].set_ylabel('YLabel 1') fig.align_ylabels() Nrw r[ r] ) rN ry rx ry rr rs Z get_ylabelrz r{ r" r| r� r} )r r~ r� r r$ r% r� r� r� r� r� r� Zcolc0Zcolc1r" r" r# � align_ylabels� s( + ���zFigure.align_ylabelsc C s | j |d� | j|d� dS )a� Align the xlabels and ylabels of subplots with the same subplots row or column (respectively) if label alignment is being done automatically (i.e. the label position is not manually set). Alignment persists for draw events after this is called. Parameters ---------- axs : list of `~matplotlib.axes.Axes` Optional list (or ndarray) of `~matplotlib.axes.Axes` to align the labels. Default is to align all axes on the figure. See Also -------- matplotlib.figure.Figure.align_xlabels matplotlib.figure.Figure.align_ylabels )r~ N)r� r� )r r~ r" r" r# � align_labels9 s zFigure.align_labelsc K s2 |� dd�}tf ||| d�|��}| j�|� |S )a^ Return a `.GridSpec` that has this figure as a parent. This allows complex layout of axes in the figure. Parameters ---------- nrows : int Number of rows in grid. ncols : int Number or columns in grid. Returns ------- gridspec : `.GridSpec` Other Parameters ---------------- **kwargs Keyword arguments are passed to `.GridSpec`. See Also -------- matplotlib.pyplot.subplots Examples -------- Adding a subplot that spans two rows:: fig = plt.figure() gs = fig.add_gridspec(2, 2) ax1 = fig.add_subplot(gs[0, 0]) ax2 = fig.add_subplot(gs[1, 0]) # spans two rows: ax3 = fig.add_subplot(gs[:, 1]) r N)r$ r% r )r� r r� r� )r r$ r% r� r� r& r" r" r# � add_gridspecQ s ' zFigure.add_gridspec) NNNNro NNNN)T)T)F)r� r� r] N) r r NNNNNNF)NT)T)T)F)r5 r5 FFTNN)F)F)NF)NNT)NNNNNN)r5 r� Tr5 r` ra )r )N)N)Nrt NNN)N)N)N)VrS rT rU rV rk rn r' r� r� r� �propertyrN r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r Zdedent_interpdr r r r' r r� r� r r4 r7 r� r; r8 Z_delete_parameterr� r� r@ r rA rF rR rS rY r_ r� rh ri rk rn rp r3 r� r� r� r� r� rW r" r" r* r# rg � s� �{ /� $ - a � V ( $ � ~ � - 8 L ; - + � � = ; � 1 E D rg c C s� t | d�ot�| � }t�d�}t�d�}|rJ| jdd� \}}|| }n| }td d }t�|| |f�}|td || �� }|td || �� }t�|||�}|S ) a� Calculate the width and height for a figure with a specified aspect ratio. While the height is taken from :rc:`figure.figsize`, the width is adjusted to match the desired aspect ratio. Additionally, it is ensured that the width is in the range [4., 16.] and the height is in the range [2., 16.]. If necessary, the default height is adjusted to ensure this. Parameters ---------- arg : scalar or 2d array If a scalar, this defines the aspect ratio (i.e. the ratio height / width). In case of an array the aspect ratio is number of rows / number of columns, so that the array could be fitted in the figure undistorted. Returns ------- width, height The figure size in inches. Notes ----- If you want to create an axes within the figure, that still preserves the aspect ratio, be sure to create it with equal width and height. See examples below. Thanks to Fernando Perez for this function. Examples -------- Make a figure twice as tall as it is wide:: w, h = figaspect(2.) fig = Figure(figsize=(w, h)) ax = fig.add_axes([0.1, 0.1, 0.8, 0.8]) ax.imshow(A, **kwargs) Make a figure with the proper aspect for an array:: A = rand(5,3) w, h = figaspect(A) fig = Figure(figsize=(w, h)) ax = fig.add_axes([0.1, 0.1, 0.8, 0.8]) ax.imshow(A, **kwargs) r� )g @r� )� 0@r� Nra rp r5 rm )rm )rm ) r� ry Zisscalarr� r� r �minrP r� ) �argZisarrayZ figsize_minZ figsize_maxZnrZncZ arr_ratioZ fig_heightZnewsizer" r" r# � figaspect~ s 0 r� )rg )GrV ZloggingZnumbersr Znumpyry rH r r r r r rG r Zmatplotlib.artistr� Zmartistr r Zmatplotlib.backend_basesr Zmatplotlib.cbookr8 Zmatplotlib.colorbarr_ r\ Zmatplotlib.imageZimager� Zmatplotlib.axesr r r Zmatplotlib.blocking_inputr r Zmatplotlib.gridspecr Zmatplotlib.legendr; r8 Zmatplotlib.patchesr Zmatplotlib.projectionsr r Zmatplotlib.textr r Zmatplotlib.transformsr r r r Zmatplotlib._layoutboxr� r� r Z getLoggerrS rr ZinterpdrZ r$ ZStackr% rH rX rg r� Zkwdocr" r" r" r# �<module> sd nQ O