Rc@s4dZddljZddlmZddlZdefdYZ defdYZ ddl m Z m Z d e fd YZd e fd YZd efdYZdefdYZdfdYZiZdZdZedddddgdZddlmZeeZdS(s The axes_divider module provide helper classes to adjust the positions of multiple axes at the drawing time. Divider: this is the class that is used calculates the axes position. It divides the given rectangular area into several sub rectangles. You initialize the divider by setting the horizontal and vertical list of sizes that the division will be based on. You then use the new_locator method, whose return value is a callable object that can be used to set the axes_locator of the axes. iN(t SubplotBasetDividercBseZdZdddZdZdZdZedZ edZ dZ d Z d Z d Zd Zd ZdZdZedZdZdZdZdZdddddZdddZdZdddddgdZRS(s This is the class that is used calculates the axes position. It divides the given rectangular area into several sub-rectangles. You initialize the divider by setting the horizontal and vertical lists of sizes (:mod:`mpl_toolkits.axes_grid.axes_size`) that the division will be based on. You then use the new_locator method to create a callable object that can be used to as the axes_locator of the axes. tCcCsU||_||_||_||_||_||_d|_d|_d|_ dS(s :param fig: matplotlib figure :param pos: position (tuple of 4 floats) of the rectangle that will be divided. :param horizontal: list of sizes (:mod:`~mpl_toolkits.axes_grid.axes_size`) for horizontal division :param vertical: list of sizes (:mod:`~mpl_toolkits.axes_grid.axes_size`) for vertical division :param aspect: if True, the overall rectangular area is reduced so that the relative part of the horizontal and vertical scales have same scale. :param anchor: Determine how the reduced rectangle is placed when aspect is True. iN( t_figt_post _horizontalt _verticalt_anchort_aspectt _xrefindext _yrefindextNonet_locator(tselftfigtpost horizontaltverticaltaspecttanchor((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt__init__"s        cCs&g|jD]}|j|^q S(N(tget_horizontaltget_size(R trendererts((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytget_horizontal_sizes>scCs&g|jD]}|j|^q S(N(t get_verticalR(R RR((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytget_vertical_sizesAscCs>ddlm}||j}||j}||fS(Ni(tAddList(t axes_sizeRRR(R Rtvsizethsize((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytget_vsize_hsizeDscCs]d\}}x(|D] \}}||7}||7}qW|dkrU|||}|SdSdS(Ng(gg((tlt total_sizetrs_sumtas_sumt_rst_astk((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt_calc_kNs   cCsAdg}x1|D])\}}|j|d|||qW|S(Ngi(tappend(R!R'toffsetsR%R&((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt _calc_offsets^s !cCs ||_dS(s set the position of the rectangle. :param pos: position (tuple of 4 floats) of the rectangle that will be divided. N(R(R R((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt set_positionkscCs|jS(s%return the position of the rectangle.(R(R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt get_positiontscCs_|tjjjks*t|dkr6||_n%tddjtjjjdS(s :param anchor: anchor position ===== ============ value description ===== ============ 'C' Center 'SW' bottom left 'S' bottom 'SE' bottom right 'E' right 'NE' top right 'N' top 'NW' top left 'W' left ===== ============ isargument must be among %ss, N( t mtransformstBboxtcoefstkeystlenRt ValueErrortjointBBox(R R((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt set_anchorxs* cCs|jS(sreturn the anchor(R(R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt get_anchorscCs ||_dS(s :param horizontal: list of sizes (:mod:`~mpl_toolkits.axes_grid.axes_size`) for horizontal division . N(R(R th((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytset_horizontalscCs|jS(sreturn horizontal sizes(R(R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscCs ||_dS(s :param horizontal: list of sizes (:mod:`~mpl_toolkits.axes_grid.axes_size`) for horizontal division . N(R(R tv((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt set_verticalscCs|jS(sreturn vertical sizes(R(R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscCs ||_dS(s. :param anchor: True or False N(R(R R((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt set_aspectscCs|jS(s return aspect(R(R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt get_aspectscCs ||_dS(N(R (R R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt set_locatorscCs|jS(N(R (R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt get_locatorscCs0|jdkr|jS|j||jSdS(N(R R R-tbounds(R taxR((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytget_position_runtimes cCs|jj\}}|j||\} } } } |j|} |j|}|j| || }|j||| }|jrLt||}|j| |}|j||}|d|d|}|d|d|}t j j | | | | }t j j | | ||}|j |j |}|j|j}}n1|j| |}|j||}| | }}|dkr|d}n|dkr|d}n|||||||||}}|||||||||}}t j j ||||S(sT :param nx, nx1: Integers specifying the column-position of the cell. When nx1 is None, a single nx-th column is specified. Otherwise location of columns spanning between nx to nx1 (but excluding nx1-th column) is specified. :param ny, ny1: same as nx and nx1, but for row positions. iiiN(Rtget_size_inchesRBRRR(R=tminR+R.R/t from_boundstanchoredR7tx0ty0R (R tnxtnytnx1tny1taxesRtfigWtfigHtxtytwR8thsizestvsizestk_htk_vR'toxtoytwwthhtpbtpb1t pb1_anchoredRGRHtx1tw1ty1th1((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytlocates4       ))cCst|||||S(s returns a new locator (:class:`mpl_toolkits.axes_grid.axes_divider.AxesLocator`) for specified cell. :param nx, nx1: Integers specifying the column-position of the cell. When nx1 is None, a single nx-th column is specified. Otherwise location of columns spanning between nx to nx1 (but excluding nx1-th column) is specified. :param ny, ny1: same as nx and nx1, but for row positions. (t AxesLocator(R RIRJRKRL((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt new_locators cCs|dkr1|jjd||jd7_n{|dkrP|jj|n\|dkr|jjd||jd7_n+|dkr|jj|n tddS(Ntleftiitrighttbottomttops7the position must be one of left, right, bottom, or top(RtinsertR R)RR R3(R tpositiontsize((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt append_size s    g?ReRfRgRhc Cskddlm}m}m}xH|D]@}|||}||} || |} |j|| q#WdS(Ni(tPaddedt SizeFromFunctGetExtentHelper(RRmRnRoRl( R tuse_axestpadt adjust_dirsRmRnRotdthelperRkt padded_size((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytadd_auto_adjustable_areas   N(t__name__t __module__t__doc__R RRRR t staticmethodR(R+R,R-R6R7R9RR;RtFalseR<R=R>R?RBRbRdRlRv(((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRs2              1 RccBs/eZdZdddZdZdZRS(s| A simple callable object, initialized with AxesDivider class, returns the position and size of the given cell. cCs||_|j}|j}|||||_|_|dkrO|d}n|dkrh|d}n|||_|||_dS(s :param axes_divider: An instance of AxesDivider class. :param nx, nx1: Integers specifying the column-position of the cell. When nx1 is None, a single nx-th column is specified. Otherwise location of columns spanning between nx to nx1 (but excluding nx1-th column) is is specified. :param ny, ny1: same as nx and nx1, but for row positions. iN(t _axes_dividerR R t_nxt_nyR t_nx1t_ny1(R t axes_dividerRIRJRKRLR R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR*s        cCsS|jj}|jj}|jj|j||j||j||j|||S(N(R|R R RbR}R~RR(R RMRR R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt__call__Es     cCs't|jdr|jjSdSdS(Ntget_subplotspec(thasattrR|RR (R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRQs N(RwRxRyR RRR(((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRc%s (t SubplotSpectGridSpectSubplotDividercBsMeZdZdZdZdZdZdZdZdZ RS(sT The Divider class whose rectangle area is specified as a subplot geometry. c Os ||_t|dkrt|dtr>|d|_qy2tt|d}tt|\}}}Wntk rtdnXt |||d|_nt|dkrq|\}}}t|}t|}t|t rNt|dkrNg|D]}t|^q}t |||dd|d!|_qt ||t|d|_ntd|f|j |j j } |jdg} |jdg} |jd d } |jd d } |rtd ntj||| | | d | d | d S(s *fig* is a :class:`matplotlib.figure.Figure` instance. *args* is the tuple (*numRows*, *numCols*, *plotNum*), where the array of subplots in the figure has dimensions *numRows*, *numCols*, and where *plotNum* is the number of the subplot being created. *plotNum* starts at 1 in the upper left corner and increases to the right. If *numRows* <= *numCols* <= *plotNum* < 10, *args* can be the decimal integer *numRows* * 100 + *numCols* * 10 + *plotNum*. iis4Single argument to subplot must be a 3-digit integeriis"Illegal argument(s) to subplot: %sRRRRRtN(tfigureR2t isinstanceRt _subplotspectstrtinttmapR3Rttuplet update_paramstfigboxR@tpopR t ExceptionRR(R RtargstkwargsRtrowstcolstnumtnRRRRR((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR`s<     !(#  cCs|j|jjS(s$return the bounds of the subplot box(RRR@(R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR-s cCs|jj|j|_dS(s0update the subplot position from fig.subplotparsN(RR-RR(R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscCs/|jj\}}}}|||dfS(s"get the subplot geometry, eg 2,2,3i(Rt get_geometry(R RRtnum1tnum2((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscCs8t|||d|_|j|j|jdS(s2change subplot geometry, e.g., from 1,1,1 to 2,2,3iN(RRRR,R(R tnumrowstnumcolsR((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytchange_geometrys cCs|jS(sget the SubplotSpec instance(R(R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscCs ||_dS(sset the SubplotSpec instanceN(R(R t subplotspec((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytset_subplotspecs( RwRxRyRR-RRRRR(((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRZs G &    t AxesDividercBsweZdZd d dZdZd edZd edZd e dZ dZ dZ dZ d ZRS( s1 Divider based on the pre-existing axes. cCs||_|dkr*tj||_n ||_|dkrTtj||_n ||_tj|d|j ddd|jgd|jgdddddS( s# :param axes: axes RRRRRRRN( t_axesR tSizetAxesXt_xreftAxesYt_yrefRRt get_figure(R RMtxreftyref((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRs     cKsy|j}|jdd}|dkrQt|trB|j}qQt|}n||j|jdt |}|S(Nt axes_classtoriginal( RRR RRt _axes_classttypeRR-tTrue(R RRMRRA((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt _get_new_axess    cKs6|rqt|tjs3tj|d|j}n|r^|jjd||jd7_qq|jj|nt|tjstj|d|j}n|r|jjd||jd7_|j dddd}n5|jj||j dt |jddd}|j |}|j ||S(so Add a new axes on the right (or left) side of the main axes. :param size: A width of the axes. A :mod:`~mpl_toolkits.axes_grid.axes_size` instance or if float or string is given, *from_any* function is used to create one, with *ref_size* set to AxesX instance of the current axes. :param pad: pad between the axes. It takes same argument as *size*. :param pack_start: If False, the new axes is appended at the end of the list, i.e., it became the right-most axes. If True, it is inserted at the start of the list, and becomes the left-most axes. All extra keywords arguments are passed to the created axes. If *axes_class* is given, the new axes will be created as an instance of the given class. Otherwise, the same class of the main axes will be used. t fraction_refiiRIRJ( RRt_Basetfrom_anyRRRiR R)RdR2Rtset_axes_locator(R RkRqt pack_startRtlocatorRA((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytnew_horizontals(  % cKs6|rqt|tjs3tj|d|j}n|r^|jjd||jd7_qq|jj|nt|tjstj|d|j}n|r|jjd||jd7_|j dddd}n5|jj||j dddt |jd}|j |}|j ||S(sp Add a new axes on the top (or bottom) side of the main axes. :param size: A height of the axes. A :mod:`~mpl_toolkits.axes_grid.axes_size` instance or if float or string is given, *from_any* function is used to create one, with *ref_size* set to AxesX instance of the current axes. :param pad: pad between the axes. It takes same argument as *size*. :param pack_start: If False, the new axes is appended at the end of the list, i.e., it became the top-most axes. If True, it is inserted at the start of the list, and becomes the bottom-most axes. All extra keywords arguments are passed to the created axes. If *axes_class* is given, the new axes will be created as an instance of the given class. Otherwise, the same class of the main axes will be used. RiiRIRJ( RRRRRRRiR R)RdR2RR(R RkRqRRRRA((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt new_vertical>s(  % cKs|dkr*|j||dt|}n|dkrT|j||dt|}n`|dkr~|j||dt|}n6|dkr|j||dt|}n td|r|jj|n|S(s create an axes at the given *position* with the same height (or width) of the main axes. *position* ["left"|"right"|"bottom"|"top"] *size* and *pad* should be axes_grid.axes_size compatible. ReRRfRgRhs7the position must be one of left, right, bottom, or top(RRR{RR3Rtadd_axes(R RjRkRqt add_to_figureRRA((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt append_axesms     cCs@|jdkr5|jj}|dkr.tStSn|jSdS(Ntauto(RR RR=R{R(R R((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR=s  cCs6|jdkr+|jjdt}|jS|jSdS(NR(RR RR-RR@(R tbbox((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR-scCs'|jdkr|jjS|jSdS(N(RR RR7(R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR7s cCs't|jdr|jjSdSdS(NR(RRRR (R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRs N(RwRxRyR RRR{RRRRR=R-R7R(((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRs ./  t HBoxDividercBsYeZdZedZedZddZdZdddddZ RS(cOstj||||dS(N(RR(R RRR((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscCst|}ddl}|j|j|d|dfdd}|j|ddd}xKt|D]=\}\} } | |||feZdZdZdZddZdedZRS(cOs,|jj|||d|_d|_dS(N(RRR R t_locator_renderer(R tkltkw((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRbs cCs ||_dS(N(R (R R((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRiscCs|jS(N(R (R ((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytget_axes_locatorlscCsZ|jdkr(|jj||n.|j||j}|jj|d|dS(NRj(RR Rt apply_aspectR(R RjR((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRoscCs#||_|jj|||dS(N(RRtdraw(R Rtinframe((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRxs N( RwRxRRRR RR{R(((s^/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRas     cCsRtj|}|dkrNtd|jt|fi|d6}|t| s& 37