ó ÄCõRc@ s´dZddlmZmZddlZejZddljZddl Z ddl Z de fd„ƒYZ de fd„ƒYZde fd „ƒYZd e fd „ƒ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ÿÿÿÿ(tdivisiontprint_functionNt GridSpecBasecB sweZdZd d d„Zd„Zd d„Zddd„Zd„Zd„Z d„Z d „Z d „Z d „Z RS( 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((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyt__init__!s cC s|j|jfS(s$get the geometry of the grid, eg 2,3(RR(R((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyt get_geometry.scC sdS(N((Rtfig((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pytget_subplot_params2sicC s4|\}}||||…|||…f}|S(s: create and return a SuplotSpec instance. ((Rtloctrowspantcolspantloc1tloc2t subplotspec((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pytnew_subplotspec5s $cC s ||_dS(N(t_col_width_ratios(RR ((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyR>scC s|jS(N(R(R((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pytget_width_ratiosAscC s ||_dS(N(t_row_height_ratios(RR ((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyRDscC s|jS(N(R(R((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pytget_height_ratiosGsc!C s«|jƒ\}}|j|ƒ}|j}|j}|j}|j}|j} |j} ||} ||} | || |d} | | }|jdk rà| |}t t |jƒƒ}g|jD]}|||^qÃ}n | g|}dg|g|d}t j jt jt||ƒƒƒ}| || |d}| |}|jdk rž||}t t |jƒƒ}g|jD]}|||^q}n |g|}dg|g|d}t j jt jt||ƒƒƒ}gt|ƒD]}||d|^qô}gt|ƒD]}||d|d^q}gt|ƒD]}||d|^qN}gt|ƒD]}||d|d^qy} |||| fS(so return lists of bottom and top position of rows, left and right positions of columns. iiiN(R RtlefttrighttbottomttoptwspacethspaceRtNonetfloattsumtnptaddt accumulatetraveltzipRtrange(!RRRR tsubplot_paramsRRRRRR ttotWidtht totHeighttcellHtsepHt netHeightttrtrt cellHeightst sepHeightstcellHstcellWtsepWtnetWidtht cellWidthst sepWidthstcellWstrowNumtfigTopst figBottomstcolNumtfigLeftst figRights((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pytget_grid_positionsKsB          ' $  ' $+/+/cC s |jƒ\}}||}t|tƒrxy|\}}Wntk rZtdƒ‚nXt|tƒr…|j|ƒ\}}} nQ|dkrž||7}n||ks¶|dkrÅtdƒ‚n||d}}t|tƒr|j|ƒ\} } } nQ|dkr||7}n||ks1|dkr@tdƒ‚n||d} } ||| } |d|| d} nt|tƒr¬|j|ƒ\} } } | d8} nM|dkrÅ||7}n||ksÝ|dkrìtdƒ‚n|d} } t|| | ƒS(s: create and return a SuplotSpec instance. sunrecognized subplot specisindex out of rangeiN( R t isinstancettuplet ValueErrortslicetindicest IndexErrorR!t SubplotSpec(RtkeyRR ttotaltk1tk2trow1trow2t_tcol1tcol2tnum1tnum2((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyt __getitem__ƒs@          N(t__name__t __module__t__doc__R!R R RRRRRRRART(((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyRs       8tGridSpecc B s}eZdZd d d d d d d d d„ZddddddgZd„Zd d „Zd „Zd d d d d d „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 sY||_||_||_||_||_||_tj|||d| 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. R R N(RRRRRR RR ( RRR RRRRRR R R ((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyR ¾s      RRRRRR c K sÄxO|jƒD]A\}}||jkr;t|||ƒq td|fƒ‚q Wddlm}ddlm}xK|jj j ƒD]7}x.|j j j D]}t||ƒsw|jdk rt|j|ƒr|jjƒjƒ|krt|jjƒ|j|jjƒqtq¸|jdk r¸t|j|ƒr¸|jjƒjƒ|krt|jjƒ|j|jjƒqtq¸q›|jƒjƒ}|jƒ|kr›|jƒ|j|jƒq›q›Wq…WdS(s‚ Update the current values. If any kwarg is None, default to the current value, if set, otherwise to rc. s%s is unknown keywordiÿÿÿÿ(t_pylab_helpers(t SubplotBaseN(t iteritemst _AllowedKeystsetattrtAttributeErrort matplotlibRYtmatplotlib.axesRZtGcftfigst itervaluestcanvastfiguretaxesRBt_sharexR!tget_subplotspect get_gridspect update_paramst set_positiontfigboxt_shareytget_topmost_subplotspec( RtkwargstktvRYRZt figmanagertaxtss((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pytupdateØs*! !  cC s»ddlm}ddl}|dkrgtg|jD]}|td|f^q5ƒ}||}n|j|jƒ}tg|jD]}|t||ƒf^q†ƒ}|j ||S(s– return a dictionary of subplot layout parameters. The default parameters are from rcParams unless a figure attribute is set. iÿÿÿÿ(t SubplotParamsNsfigure.subplot.( tmatplotlib.figureRvtcopyR!tdictR\trcParamst subplotparstgetattrRu(RRRvRxRptkwR{t update_kw((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyRüs  *1 cC s)g|jD]}t||ƒr |^q S(N(R\R|(RRp((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pytlocally_modified_subplot_paramssgHáz®Gñ?c C s©ddlm}m}m} ||jd|ƒ} d | krMtjdƒn|d krh| |ƒ}n|||j| |d|d|d|d|ƒ} |j| d S( s^ 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 padding (height/width) between edges of adjacent subplots. Defaults to `pad_inches`. rect : if rect is given, it is interpreted as 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). iÿÿÿÿ(tget_subplotspec_listtget_tight_layout_figuret get_renderert grid_specsgThis figure includes Axes that are not compatible with tight_layout, so its results might be incorrect.tpadth_padtw_padtrectN( t tight_layoutR€RR‚RfR!twarningstwarnRu( RRtrendererR„R…R†R‡R€RR‚tsubplotspec_listRo((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyRˆs   N( RURVRWR!R R\RuRRRˆ(((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyRX·s  $  tGridSpecFromSubplotSpeccB s8eZdZddddd„Zdd„Zd„ZRS(sx GridSpec whose subplot layout parameters are inherited from the location specified by a given SubplotSpec. cC s>||_||_||_tj|||d|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. R R N(t_wspacet_hspacet _subplot_specRR (RRR t subplot_specRR R R ((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyR ?s   c C sß|d kr#td}td}n|jj}|jj}|jd k rV|j}n|jd k rq|j}n|jj|dt ƒ}|j \}}}}ddl m } | d|d|d|d |d |d |ƒ} | S( sC return a dictionary of subplot layout parameters. sfigure.subplot.hspacesfigure.subplot.wspacet return_alliÿÿÿÿ(RvRRRRRR N( R!RzR{R RRRŽRt get_positiontFalsetextentsRwRv( RRR RRlRRRRRvtsp((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyRTs&         cC s |jjƒS(s@get the topmost SubplotSpec instance associated with the subplot(RRn(R((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyRnusN(RURVRWR!R RRn(((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyR:s  !RHcB sAeZdZdd„Zd„Zd„Zed„Zd„Z RS(sH specifies the location of the subplot in the given *GridSpec*. cC s;|jƒ\}}||}||_||_||_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 stars from 0. N(R t _gridspecRRRS(RtgridspecRRRStrowstcolsRJ((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyR s    cC s|jS(N(R—(R((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyRiscC s.|jƒjƒ\}}|||j|jfS(sc get the subplot geometry, eg 2,2,3. Unlike SuplorParams, index is 0-based (RiR RRRS(RR™Rš((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyR “scC sC|jƒ}|jƒ\}}|j|ƒ\}}}} t|j|ƒ\} } || } || } || }| | }|jdk rt|j|ƒ\}}||}||}||}| |}t| |ƒ} t||ƒ}t| |ƒ} t||ƒ}nt j j || || ƒ}|r;|| | ||fS|SdS(sB update the subplot position from fig.subplotpars N( RiR RAtdivmodRRRSR!tmintmaxt mtransformstBboxt from_extents(RRR’R˜RR R=R<R?R@R;R>t figBottomtfigToptfigLefttfigRighttrowNum2tcolNum2t figBottom2tfigTop2tfigLeft2t figRight2Rl((sM/opt/xtal/ccp4-6.5-linux32/lib/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 subplotRnN(RithasattrRn(RR˜((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyRnÅs  N( RURVRWR!R RiR R”R“Rn(((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyRHzs    )(RWt __future__RRR_Rztmatplotlib.transformst transformsRžtnumpyR$R‰tobjectRRXRRH(((sM/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/gridspec.pyts    œƒ@