ó m[c@`s dZddlmZmZmZddlZddlZddlZddlZddl Z ddl Z ddl m Z mZmZddlmZddljZddlmZejeƒZdefd„ƒYZd efd „ƒYZd efd „ƒYZd efd„ƒYZdS(sČ :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`. i(tabsolute_importtdivisiontprint_functionN(t_pylab_helperst tight_layouttrcParams(tBbox(tmplDeprecationt GridSpecBasecB`s}eZdZd d d„Zd„Zd d d„Zddd„Zd„Zd„Z d„Z d „Z e d „Z d „ZRS( sm A base class of GridSpec that specifies the geometry of the grid that a subplot will be placed. cC`s1|||_|_|j|ƒ|j|ƒdS(sø 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(t_nrowst_ncolstset_height_ratiostset_width_ratios(tselftnrowstncolst height_ratiost width_ratios((s2lib/python2.7/site-packages/matplotlib/gridspec.pyt__init__)s cC`s|j|jfS(s'get the geometry of the grid, e.g., 2,3(R R (R ((s2lib/python2.7/site-packages/matplotlib/gridspec.pyt get_geometry3scC`sdS(N((R tfiguretfig((s2lib/python2.7/site-packages/matplotlib/gridspec.pytget_subplot_params7sicC`s4|\}}||||…|||…f}|S(s: create and return a SuplotSpec instance. ((R tloctrowspantcolspantloc1tloc2t subplotspec((s2lib/python2.7/site-packages/matplotlib/gridspec.pytnew_subplotspec:s $cC`s=|dk r0t|ƒ|jkr0tdƒ‚n||_dS(NsTExpected the given number of width ratios to match the number of columns of the grid(tNonetlenR t ValueErrort_col_width_ratios(R R((s2lib/python2.7/site-packages/matplotlib/gridspec.pyR Bs!cC`s|jS(N(R!(R ((s2lib/python2.7/site-packages/matplotlib/gridspec.pytget_width_ratiosHscC`s=|dk r0t|ƒ|jkr0tdƒ‚n||_dS(NsRExpected the given number of height ratios to match the number of rows of the grid(RRR R t_row_height_ratios(R R((s2lib/python2.7/site-packages/matplotlib/gridspec.pyR Ks!cC`s|jS(N(R#(R ((s2lib/python2.7/site-packages/matplotlib/gridspec.pytget_height_ratiosQscC`s>|jƒ\}}|r?d}d}d}d}d} d} nE|j|ƒ} | j}| j}| j}| j}| j} | j} ||} ||} | || |d}| |}|jdk r||t |jƒ}g|jD]}||^qč}n |g|}dg|g|d}t j t j ||gƒjƒ}| || |d}| |}|jdk r°||t |jƒ}g|jD]}||^q—}n |g|}dg|g|d}t j t j ||gƒjƒ}||jdƒj\}}||jd ƒj\}}||||fS( sō 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š?iii’’’’iN(i’’’’i(i’’’’i(RRtlefttrighttbottomttoptwspacethspaceR#Rtsumtnptcumsumt column_stacktflatR!treshapetT(R RtrawRRR%R&R'R(R)R*tsubplot_paramst tot_widtht tot_heighttcell_htsep_htnormtrt cell_heightst sep_heightstcell_hstcell_wtsep_wt cell_widthst sep_widthstcell_wstfig_topst fig_bottomst fig_leftst fig_rights((s2lib/python2.7/site-packages/matplotlib/gridspec.pytget_grid_positionsTsH          # ! # !c C`s¼|jƒ\}}d„}t|tƒr“y|\}}Wntk rYtdƒ‚nXtj|||ƒ|||ƒg||fƒ\}}n||||ƒ\}}t|||ƒS(s1Create and return a SuplotSpec instance. cS`s“t|tƒrD|j|ƒ\}}}||krƒ||dfSn?|dkr]||7}nd|kot|knrƒ||fStdƒ‚dS(Niis invalid index(t isinstancetslicetindicest IndexError(tkeytsizetstarttstopt_((s2lib/python2.7/site-packages/matplotlib/gridspec.pyt _normalizes    sunrecognized subplot spec(RRGttupleR R,travel_multi_indext SubplotSpec( R RKRRRPtk1tk2tnum1tnum2((s2lib/python2.7/site-packages/matplotlib/gridspec.pyt __getitem__Šs  3N(t__name__t __module__t__doc__RRRRRR R"R R$tFalseRFRX(((s2lib/python2.7/site-packages/matplotlib/gridspec.pyR#s      6tGridSpecc B`s•eZdZdddddddddd„ ZddddddgZd„Zd „Zd „Zddd „Z d „Z dd dddd„Z RS(sØ 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. c C`sĖ||_||_||_||_||_| |_||_tj|||d| d| ƒ|jdks}|jj ƒ r‰d|_ n>|jj ƒt jd|jj ddt jƒd|ƒ|_ dS(sĄ 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. Notes ----- See `~.figure.SubplotParams` for descriptions of the layout parameters. RRtparenttnametgridspectartistN(R%R'R&R(R)R*RRRRtget_constrained_layoutt _layoutboxtinit_layoutboxt layoutboxt LayoutBoxtseq_id( R RRRR%R'R&R(R)R*RR((s2lib/python2.7/site-packages/matplotlib/gridspec.pyRÆs"           R%R'R&R(R)R*cC`s2|j}y|jdƒWntk r-nX|S(NRc(t__dict__tpoptKeyError(R tstate((s2lib/python2.7/site-packages/matplotlib/gridspec.pyt __getstate__Žs   cC`s||_d|_dS(N(RhRRc(R Rk((s2lib/python2.7/site-packages/matplotlib/gridspec.pyt __setstate__ęs cK`sžxRtj|ƒD]A\}}||jkr>t|||ƒqtd|fƒ‚qWxBtjtjjƒD]+}x"|j j j D]}t |t j jƒsQt |jt j jƒrõ|jjƒjƒ|krN|jjƒ|j|jjƒqNq’t |jt j jƒr’|jjƒjƒ|krN|jjƒ|j|jjƒqNq’q|jƒjƒ}|jƒ|kr|jƒ|j|jƒqqWqkWdS(s‚ Update the current values. If any kwarg is None, default to the current value, if set, otherwise to rc. s%s is unknown keywordN(tsixt iteritemst _AllowedKeystsetattrtAttributeErrort itervaluesRtGcftfigstcanvasRtaxesRGtmplt SubplotBaset_sharextget_subplotspect get_gridspect update_paramst _set_positiontfigboxt_shareytget_topmost_subplotspec(R tkwargstktvt figmanagertaxtss((s2lib/python2.7/site-packages/matplotlib/gridspec.pytupdateės&   c`s¤|dk rtjdtƒn|dkr4|}n|dkrhd„ˆjDƒ}tjj|}ntj|j ƒ}‡fd†ˆjDƒ}|j ||S(s– Return a dictionary of subplot layout parameters. The default parameters are from rcParams unless a figure attribute is set. s2the 'fig' kwarg is deprecated use 'figure' insteadcS`s!i|]}td||“qS(sfigure.subplot.(R(t.0Rƒ((s2lib/python2.7/site-packages/matplotlib/gridspec.pys s c`s"i|]}tˆ|ƒ|“qS((tgetattr(R‰Rƒ(R (s2lib/python2.7/site-packages/matplotlib/gridspec.pys s N( RtwarningstwarnRRpRxRt SubplotParamstcopyt subplotparsRˆ(R RRtkwRt update_kw((R s2lib/python2.7/site-packages/matplotlib/gridspec.pyR s       cC`s)g|jD]}t||ƒr |^q S(N(RpRŠ(R Rƒ((s2lib/python2.7/site-packages/matplotlib/gridspec.pytlocally_modified_subplot_params sgHįz®Gń?c C`s–tj|jd|ƒ}d|kr4tjdƒn|dkrRtj|ƒ}ntj||j||d|d|d|d|ƒ}|j|dS(sy 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). t grid_specscThis figure includes Axes that are not compatible with tight_layout, so results might be incorrect.tpadth_padtw_padtrectN( Rtget_subplotspec_listRwRR‹RŒt get_renderertget_tight_layout_figureRˆ( R RtrendererR”R•R–R—tsubplotspec_listR‚((s2lib/python2.7/site-packages/matplotlib/gridspec.pyR#s  N( RYRZR[RRRpRlRmRˆRR’R(((s2lib/python2.7/site-packages/matplotlib/gridspec.pyR]Øs *    tGridSpecFromSubplotSpeccB`s;eZdZddddd„Zddd„Zd„ZRS(sx GridSpec whose subplot layout parameters are inherited from the location specified by a given SubplotSpec. c C`sŽ||_||_||_tj|||d|d|ƒ|j}|dkr[d|_n/|j|d|jdt j ƒd|ƒ|_dS(s` The number of rows and number of columns of the grid need to be set. An instance of SubplotSpec is also needed to be set from which the layout parameters will be inherited. The wspace and hspace of the layout can be optionally specified or the default values (from the figure or rcParams) will be used. RRR_s .gridspecRaN( t_wspacet_hspacet _subplot_specRRRcRtlayout_from_subplotspecR_ReRg( R RRt subplot_specR)R*RRt subspeclb((s2lib/python2.7/site-packages/matplotlib/gridspec.pyRKs      c C`s|d k rtjdtƒn|d kr4|}n|jd k rL|jn|d k rd|jjntd}|jd k r†|jn|d k rž|jj ntd}|j j |ƒ}|j \}}}} t jjd|d|d|d| d|d |ƒS( s:Return a dictionary of subplot layout parameters. s2the 'fig' kwarg is deprecated use 'figure' insteadsfigure.subplot.hspacesfigure.subplot.wspaceR%R&R'R(R)R*N(RR‹RŒRRŸRR*RRžR)R t get_positiontextentsRxRR( R RRR*R)RR%R'R&R(((s2lib/python2.7/site-packages/matplotlib/gridspec.pyRgs         cC`s |jjƒS(sAGet the topmost SubplotSpec instance associated with the subplot.(R R(R ((s2lib/python2.7/site-packages/matplotlib/gridspec.pyR~sN(RYRZR[RRRR(((s2lib/python2.7/site-packages/matplotlib/gridspec.pyRFs RScB`sƒeZdZd d„Zd„Zd„Zd„Zd„Zd„Z e d„Z d„Z d „Z ejrxd „Znd „ZRS( sCSpecifies the location of the subplot in the given `GridSpec`. cC`su||_||_||_|jdk rh|j}tjd|d|jdtjƒd|ƒ|_n d|_dS(sÖ The subplot will occupy the num1-th cell of the given gridspec. If num2 is provided, the subplot will span between num1-th cell and num2-th cell. The index starts from 0. R^R_s.ssRaN( t _gridspecRVRWRcRReRfR_Rg(R R`RVRWtglb((s2lib/python2.7/site-packages/matplotlib/gridspec.pyR‡s     cC`s2|j}y|jdƒWntk r-nX|S(NRc(RhRiRj(R Rk((s2lib/python2.7/site-packages/matplotlib/gridspec.pyRlŸs   cC`s||_d|_dS(N(RhRRc(R Rk((s2lib/python2.7/site-packages/matplotlib/gridspec.pyRm§s cC`s|jS(N(R¦(R ((s2lib/python2.7/site-packages/matplotlib/gridspec.pyR|¬scC`s.|jƒjƒ\}}|||j|jfS(s” Get the subplot geometry (``n_rows, n_cols, start, stop``). start and stop are the index of the start and stop of the subplot. (R|RRVRW(R trowstcols((s2lib/python2.7/site-packages/matplotlib/gridspec.pyRÆscC`s‚|jƒ}|jƒ\}}t|j|ƒ\}}|jdk r`t|j|ƒ\}}n |}|}||||||fS(s€ Get the subplot row and column numbers: (``n_rows, n_cols, row_start, row_stop, col_start, col_stop``) N(R|RtdivmodRVRWR(R R`RRt row_startt col_starttrow_stoptcol_stop((s2lib/python2.7/site-packages/matplotlib/gridspec.pytget_rows_columns¹s cC`s’|jƒ}|jƒ\}}tj|jdkr?|jgn|j|jg||fƒ\}}|j|ƒ\}} } } ||jƒ} | |j ƒ} | |jƒ}| |j ƒ}t j || || ƒ}|r÷||d|d||fS|SdS(sAUpdate the subplot position from ``figure.subplotpars``. iN( R|RR,t unravel_indexRWRRVRFtmintmaxRt from_extents(R Rt return_allR`RRRØR©RCRBRDREt fig_bottomtfig_toptfig_leftt fig_rightR((s2lib/python2.7/site-packages/matplotlib/gridspec.pyR¤Čs *cC`s-|jƒ}t|dƒr%|jƒS|SdS(s@get the topmost SubplotSpec instance associated with the subplotRN(R|thasattrR(R R`((s2lib/python2.7/site-packages/matplotlib/gridspec.pyRŽs  cC`sR|j|j|jft|dtƒƒt|dtƒƒt|dtƒƒfkS(NR¦RVRW(R¦RVRWRŠtobject(R tother((s2lib/python2.7/site-packages/matplotlib/gridspec.pyt__eq__ęscC`s ||k S(N((R R»((s2lib/python2.7/site-packages/matplotlib/gridspec.pyt__ne__īscC`st|j|j|jfƒS(N(thashR¦RVRW(R ((s2lib/python2.7/site-packages/matplotlib/gridspec.pyt__hash__ńsN(RYRZR[RRRlRmR|RRÆR\R¤RR¼RntPY2R½Ræ(((s2lib/python2.7/site-packages/matplotlib/gridspec.pyRSƒs          (R[t __future__RRRRnRŽtloggingR‹tnumpyR,t matplotlibRxRRRtmatplotlib.transformsRtmatplotlib._layoutboxRcRetmatplotlib.cbookRt getLoggerRYt_logRŗRR]RRS(((s2lib/python2.7/site-packages/matplotlib/gridspec.pyts       …ž=