B ]t\XN@sdZddlZddlZddlZddlZddlZddlmZm Z m Z m Z ddl m Z ddlmZeeZGdddeZGdddeZGd d d eZGd d d eZdS) a :mod:`~matplotlib.gridspec` is a module which specifies the location of the subplot in the figure. `GridSpec` specifies the geometry of the grid that a subplot will be placed. The number of rows and number of columns of the grid need to be set. Optionally, the subplot layout parameters (e.g., left, right, etc.) can be tuned. `SubplotSpec` specifies the location of the subplot in the given `GridSpec`. N)_pylab_helperscbook tight_layoutrcParams)Bboxc@speZdZdZdddZddZddZdd d Zdd d ZddZ ddZ ddZ ddZ dddZ ddZdS) GridSpecBasezm A base class of GridSpec that specifies the geometry of the grid that a subplot will be placed. NcCs&|||_|_||||dS)z The number of rows and number of columns of the grid need to be set. Optionally, the ratio of heights and widths of rows and columns can be specified. N)_nrows_ncolsset_height_ratiosset_width_ratios)selfnrowsncols height_ratios width_ratiosr2lib/python3.7/site-packages/matplotlib/gridspec.py__init__$s zGridSpecBase.__init__cCsN|jdk rd|jnd}|jdk r,d|jnd}dj|jj|j|j||dS)Nz, height_ratios=%rz, width_ratios=%rz&{clsname}({nrows}, {ncols}{optionals}))Zclsnamer rZ optionals)_row_height_ratios_col_width_ratiosformat __class____name__rr )r Z height_argZ width_argrrr__repr__.szGridSpecBase.__repr__cCs |j|jfS)z'get the geometry of the grid, e.g., 2,3)rr )r rrr get_geometry:szGridSpecBase.get_geometrycCsdS)Nr)r figurefigrrrget_subplot_params>szGridSpecBase.get_subplot_paramscCs(|\}}|||||||f}|S)z: create and return a SuplotSpec instance. r)r ZlocZrowspanZcolspanZloc1Zloc2Z subplotspecrrrnew_subplotspecAszGridSpecBase.new_subplotspeccCs(|dk rt||jkrtd||_dS)NzTExpected the given number of width ratios to match the number of columns of the grid)lenr ValueErrorr)r rrrrr IszGridSpecBase.set_width_ratioscCs|jS)N)r)r rrrget_width_ratiosOszGridSpecBase.get_width_ratioscCs(|dk rt||jkrtd||_dS)NzRExpected the given number of height ratios to match the number of rows of the grid)r!rr"r)r rrrrr RszGridSpecBase.set_height_ratioscCs|jS)N)r)r rrrget_height_ratiosXszGridSpecBase.get_height_ratiosFcs|\}}|r*d}d}d}d}d} d} n.||} | j}| j}| j}| j}| j} | j} ||} ||} | || |d}| |}|jdk r||t |jfdd|jD}n |g|}dg|g|d}t t ||gj }| || |d}| |}|jdk r:||t |jfdd|jD}n |g|}dg|g|d}t t ||gj }||d j\}}||d j\}}||||fS) z return lists of bottom and top position of rows, left and right positions of columns. If raw=True, then these are all in units relative to the container with no margins. (used for constrained_layout). gg?rNcsg|] }|qSrr).0r)normrr |sz3GridSpecBase.get_grid_positions..rcsg|] }|qSrr)r%r&)r'rrr(s))rrleftrightbottomtopwspacehspacersumnpZcumsumZ column_stackZflatrZreshapeT)r rrawr rr+r,r-r.r/r0Zsubplot_paramsZ tot_widthZ tot_heightZcell_hZsep_hZ cell_heightsZ sep_heightsZcell_hsZcell_wZsep_wZ cell_widthsZ sep_widthsZcell_wsfig_tops fig_bottoms fig_lefts fig_rightsr)r'rget_grid_positions[sH      zGridSpecBase.get_grid_positionsc Cs|\}}dd}t|trny |\}}Wntk rFtdYnXt||||||g||f\}}n||||\}}t|||S)z1Create and return a SuplotSpec instance. cSslt|tr0||\}}}||kr`||dfSn0|dkr@||7}d|krT|kr`nn||fStddS)Nrrz invalid index) isinstancesliceindices IndexError)keysizestartstop_rrr _normalizes z,GridSpecBase.__getitem__.._normalizezunrecognized subplot spec)rr:tupler"r2Zravel_multi_index SubplotSpec) r r>r rrCZk1Zk2num1num2rrr __getitem__s   "zGridSpecBase.__getitem__)NN)NN)rr)F)r __module__ __qualname____doc__rrrrr r r#r r$r9rHrrrrrs     6rc @s^eZdZdZdddZddddd d gZd d Zd dZddZdddZ ddZ dddZ dS)GridSpecz A class that specifies the geometry of the grid that a subplot will be placed. The location of grid is determined by similar way as the SubplotParams. Nc Cs||_||_||_||_||_| |_||_tj|||| | d|jdksR|j sZd|_ n(|j t j |jj dt |d|_ dS)a The number of rows and number of columns of the grid need to be set. Optionally, the subplot layout parameters (e.g., left, right, etc.) can be tuned. Parameters ---------- nrows : int Number of rows in grid. ncols : int Number or columns in grid. figure : ~.figure.Figure, optional left, right, top, bottom : float Extent of the subplots as a fraction of figure width or height. Left cannot be larger than right, and bottom cannot be larger than top. 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. Notes ----- See `~.figure.SubplotParams` for descriptions of the layout parameters. )rrNgridspec)parentnameartist)r+r-r,r.r/r0rrrZget_constrained_layout _layoutboxZinit_layoutbox layoutbox LayoutBoxseq_id) r r rrr+r-r,r.r/r0rrrrrrs"$   zGridSpec.__init__r+r-r,r.r/r0cCs.|j}y|dWntk r(YnX|S)NrQ)__dict__popKeyError)r staterrr __getstate__s zGridSpec.__getstate__cCs||_d|_dS)N)rUrQ)r rXrrr __setstate__szGridSpec.__setstate__cKsx:|D].\}}||jkr*t|||q td|fq WxtjjD]}x|jj j D]}t |t j j st |jt j j r|j|kr|j||jjn:t |jt j j r|j|kr|j||jjqZ|}||krZ|||jqZWqJWdS)z Update the current values. If any kwarg is None, default to the current value, if set, otherwise to rc. z%s is unknown keywordN)items _AllowedKeyssetattrAttributeErrorrZGcfZfigsvaluesZcanvasraxesr:mplZ SubplotBaseZ_sharexZget_subplotspec get_gridspecZ update_paramsZ _set_positionfigboxZ_shareyget_topmost_subplotspec)r kwargskvZ figmanagerZaxZssrrrupdates&     zGridSpec.updatecsz|dk rtjddddd|dkr&|}|dkrNddjD}tjjf|}n t|j}|jffd djD|S) z Return a dictionary of subplot layout parameters. The default parameters are from rcParams unless a figure attribute is set. Nz2.2rzkeyword argumentr)obj_type alternativecSsi|]}td||qS)zfigure.subplot.)r)r%rfrrr ,sz/GridSpec.get_subplot_params..csi|]}t||qSr)getattr)r%rf)r rrrk1s) rwarn_deprecatedr\rar SubplotParamscopy subplotparsrh)r rrkwrpr)r rr s  zGridSpec.get_subplot_paramscsfddjDS)Ncsg|]}t|r|qSr)rl)r%rf)r rrr(6sz.)r\)r r)r rlocally_modified_subplot_params5sz(GridSpec.locally_modified_subplot_paramsHzG?c Csdtj|j|d}d|kr"td|dkr4t|}tj||j||||||d}|r`|jf|dS)ay Adjust subplot parameters to give specified padding. Parameters ---------- pad : float 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. Defaults to ``pad_inches``. rect : tuple of 4 floats, optional (left, bottom, right, top) rectangle in normalized figure coordinates that the whole subplots area (including labels) will fit into. Default is (0, 0, 1, 1). )Z grid_specNzcThis figure includes Axes that are not compatible with tight_layout, so results might be incorrect.)padh_padw_padrect)rZget_subplotspec_listr`warningswarnZ get_rendererZget_tight_layout_figurerh) r rZrendererrtrurvrwZsubplotspec_listrerrrr8s    zGridSpec.tight_layout) NNNNNNNNN)NN)NrsNNN) rrIrJrKrr\rYrZrhrrrrrrrrrLs 9 rLc@s,eZdZdZd ddZd ddZddZdS) GridSpecFromSubplotSpeczx GridSpec whose subplot layout parameters are inherited from the location specified by a given SubplotSpec. Nc Cs`||_||_||_tj|||||d|j}|dkrs   .=