m[c@`smdZddlmZmZmZmZddlZddlmZddl j Z ddl m Z ddlmZdefd YZd efd YZdd lmZmZd efdYZdefdYZdefdYZdefdYZdefdYZiZdZdZededZ ddl!m"Z"ee"Z#dS(u The axes_divider module provides helper classes to adjust the positions of multiple axes at drawing time. Divider: this is the class that is used to calculate 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 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. i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tmap(t SubplotBasei(t axes_sizetDividercB`s eZdZdddZdZdZdZedZ edZ dZ d Z d Z d Zd Zd ZdZdZedZdZdZdZdZdddddZdddZdZdddZRS(u This class 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 as the axes_locator of the axes. uCcC`sU||_||_||_||_||_||_d|_d|_d|_ dS(u Parameters ---------- fig : Figure pos : tuple of 4 floats position of the rectangle that will be divided horizontal : list of :mod:`~mpl_toolkits.axes_grid.axes_size` sizes for horizontal division vertical : list of :mod:`~mpl_toolkits.axes_grid.axes_size` sizes for vertical division aspect : bool if True, the overall rectangular area is reduced so that the relative part of the horizontal and vertical scales have the same scale. anchor : {'C', 'SW', 'S', 'SE', 'E', 'NE', 'N', 'NW', 'W'} placement of the reduced rectangle when *aspect* is True iN( t_figt_post _horizontalt _verticalt_anchort_aspectt _xrefindext _yrefindextNonet_locator(tselftfigtpost horizontaltverticaltaspecttanchor((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt__init__&s        cC`s&g|jD]}|j|^q S(N(tget_horizontaltget_size(Rtrendererts((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytget_horizontal_sizesDscC`s&g|jD]}|j|^q S(N(t get_verticalR(RRR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytget_vertical_sizesGscC`s>ddlm}||j}||j}||fS(Ni(tAddList(RR!RR(RR!tvsizethsize((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytget_vsize_hsizeJscC`s]d\}}x(|D] \}}||7}||7}qW|dkrU|||}|SdSdS(Ng(gg((tlt total_sizetrs_sumtas_sumt_rst_astk((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt_calc_kSs   cC`sAdg}x1|D])\}}|j|d|||qW|S(Ngi(tappend(R%R+toffsetsR)R*((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt _calc_offsetsbs !cC`s ||_dS(u set the position of the rectangle. Parameters ---------- pos : tuple of 4 floats position of the rectangle that will be divided N(R (RR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt set_positionns cC`s|jS(u%return the position of the rectangle.(R (R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt get_positionyscC`sS|tjjks$t|dkr0||_ntddjtjjdS(u Parameters ---------- anchor : {'C', 'SW', 'S', 'SE', 'E', 'NE', 'N', 'NW', 'W'} 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 ===== ============ iuargument must be among %su, N(t mtransformstBboxtcoefstlenR t ValueErrortjointBBox(RR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt set_anchor}s$ cC`s|jS(ureturn the anchor(R (R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt get_anchorscC`s ||_dS(u Parameters ---------- h : list of :mod:`~mpl_toolkits.axes_grid.axes_size` sizes for horizontal division N(R (Rth((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytset_horizontalscC`s|jS(ureturn horizontal sizes(R (R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscC`s ||_dS(u Parameters ---------- v : list of :mod:`~mpl_toolkits.axes_grid.axes_size` sizes for vertical division N(R (Rtv((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt set_verticalscC`s|jS(ureturn vertical sizes(R (R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscC`s ||_dS(uE Parameters ---------- aspect : bool N(R (RR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt set_aspectscC`s|jS(u return aspect(R (R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt get_aspectscC`s ||_dS(N(R(RR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt set_locatorscC`s|jS(N(R(R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt get_locatorscC`s0|jdkr|jS|j||jSdS(N(RRR1tbounds(RtaxR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytget_position_runtimes cC`s|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(u Parameters ---------- nx, nx1 : int 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. ny, ny1 : int Same as *nx* and *nx1*, but for row positions. axes renderer iiiN(Rtget_size_inchesRERR R,R@tminR/R2R3t from_boundstanchoredR:tx0ty0R(Rtnxtnytnx1tny1taxesRtfigWtfigHtxtytwR;thsizestvsizestk_htk_vR+toxtoytwwthhtpbtpb1t pb1_anchoredRJRKtx1tw1ty1th1((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytlocates4      ))cC`st|||||S(u Returns a new locator (:class:`mpl_toolkits.axes_grid.axes_divider.AxesLocator`) for specified cell. Parameters ---------- nx, nx1 : int 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. ny, ny1 : int Same as *nx* and *nx1*, but for row positions. (t AxesLocator(RRLRMRNRO((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt new_locatorscC`s|dkr1|jjd||jd7_n|dkrP|jj|n`|dkr|jjd||jd7_n/|dkr|jj|ntdddS( Nuleftiiurightubottomutopu!the position must be one of left,u right, bottom, or top(R tinsertRR-R RR6(Rtpositiontsize((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt append_sizes    g?c C`s|dkr!ddddg}nddlm}m}m}xH|D]@}|||}||} || |} |j|| qDWdS(Nulefturightubottomutopi(tPaddedt SizeFromFunctGetExtentHelper(RRRlRmRnRk( Rtuse_axestpadt adjust_dirsRlRmRntdthelperRjt padded_size((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytadd_auto_adjustable_area#s   N(t__name__t __module__t__doc__RRRR R$t staticmethodR,R/R0R1R9R:R<RR>RtFalseR?R@RARBREReRgRkRu(((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRs2              2 RfcB`s/eZdZdddZdZdZRS(u| A simple callable object, initialized with AxesDivider class, returns the position and size of the given cell. cC`s||_|j}|j}|||||_|_|dkrO|d}n|dkrh|d}n|||_|||_dS(u Parameters ---------- axes_divider : AxesDivider nx, nx1 : int 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. ny, ny1 : int Same as *nx* and *nx1*, but for row positions. iN(t _axes_dividerRRt_nxt_nyRt_nx1t_ny1(Rt axes_dividerRLRMRNRORR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR6s        cC`sS|jj}|jj}|jj|j||j||j||j|||S(N(R{RRReR|R}R~R(RRPRRR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt__call__Rs     cC`s't|jdr|jjSdSdS(Nuget_subplotspec(thasattrR{tget_subplotspecR(R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR^s N(RvRwRxRRRR(((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRf1s (t SubplotSpectGridSpectSubplotDividercB`sMeZdZdZdZdZdZdZdZdZ RS(uT The Divider class whose rectangle area is specified as a subplot geometry. c O`s ||_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| dS(u+ Parameters ---------- fig : :class:`matplotlib.figure.Figure` args : tuple (*numRows*, *numCols*, *plotNum*) The array of subplots in the figure has dimensions *numRows*, *numCols*, and *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*. iiu4Single argument to subplot must be a 3-digit integeriiu"Illegal argument(s) to subplot: %su horizontaluverticaluaspectuanchoruCuRRN(tfigureR5t isinstanceRt _subplotspectstrtintRR6Rttuplet update_paramstfigboxRCtpopRt ExceptionRR(RRtargstkwargsRtrowstcolstnumtnRRRRR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRms<     !(#  cC`s|j|jjS(u$return the bounds of the subplot box(RRRC(R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR1s cC`s|jj|j|_dS(u0update the subplot position from fig.subplotparsN(RR1RR(R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscC`s/|jj\}}}}|||dfS(u%get the subplot geometry, e.g., 2,2,3i(Rt get_geometry(RRRtnum1tnum2((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscC`s8t|||d|_|j|j|jdS(u2change subplot geometry, e.g., from 1,1,1 to 2,2,3iN(RRRR0R(RtnumrowstnumcolsR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytchange_geometrys cC`s|jS(uget the SubplotSpec instance(R(R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscC`s ||_dS(uset the SubplotSpec instanceN(R(Rt subplotspec((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytset_subplotspecs( RvRwRxRR1RRRRR(((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRhs E %    t AxesDividercB`sweZdZd d dZdZd edZd edZd e dZ dZ dZ dZ d ZRS( u1 Divider based on the pre-existing axes. cC`s||_|dkr*tj||_n ||_|dkrTtj||_n ||_tj|d|j ddd|jgd|jgdddddS( uw Parameters ---------- axes : :class:`~matplotlib.axes.Axes` xref yref RRRRRRuCN( t_axesRtSizetAxesXt_xreftAxesYt_yrefRRt get_figure(RRPtxreftyref((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRs     cK`sy|j}|jdd}|dkrQt|trB|j}qQt|}n||j|jdt |}|S(Nu axes_classtoriginal( RRRRRt _axes_classttypeRR1tTrue(RRRPt axes_classRD((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt _get_new_axes s    cK`s<|rqt|tjs3tj|d|j}n|r^|jjd||jd7_qq|jj|nt|tjstj|d|j}n|r|jjd||jd7_|j ddd|j }n8|jj||j dt |jdd|j }|j |}|j ||S(u Add a new axes on the right (or left) side of the main axes. Parameters ---------- size : :mod:`~mpl_toolkits.axes_grid.axes_size` or float or string A width of the axes. If float or string is given, *from_any* function is used to create the size, with *ref_size* set to AxesX instance of the current axes. pad : :mod:`~mpl_toolkits.axes_grid.axes_size` or float or string Pad between the axes. It takes same argument as *size*. pack_start : bool 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. kwargs 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_refiiRLRM(RRt_Basetfrom_anyRR RhRR-RgRR5Rtset_axes_locator(RRjRpt pack_startRtlocatorRD((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytnew_horizontals(  ( cK`s<|rqt|tjs3tj|d|j}n|r^|jjd||jd7_qq|jj|nt|tjstj|d|j}n|r|jjd||jd7_|j d|j dd}n8|jj||j d|j dt |jd}|j |}|j ||S(u Add a new axes on the top (or bottom) side of the main axes. Parameters ---------- size : :mod:`~mpl_toolkits.axes_grid.axes_size` or float or string A height of the axes. If float or string is given, *from_any* function is used to create the size, with *ref_size* set to AxesX instance of the current axes. pad : :mod:`~mpl_toolkits.axes_grid.axes_size` or float or string Pad between the axes. It takes same argument as *size*. pack_start : bool 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. kwargs 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. RiiRLRM(RRRRRR RhRR-RgRR5RR(RRjRpRRRRD((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt new_verticalNs(  ( cK`s|dkr*|j||dt|}n|dkrT|j||dt|}nd|dkr~|j||dt|}n:|dkr|j||dt|}ntdd|r|jj|n|S(u 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. uleftRurightubottomutopu!the position must be one of left,u right, bottom, or top(RRRzRR6Rtadd_axes(RRiRjRpt add_to_figureRRD((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt append_axess     cC`s@|jdkr5|jj}|dkr.tStSn|jSdS(Nuauto(R RRR@RzR(RR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR@s  cC`s6|jdkr+|jjdt}|jS|jSdS(NR(R RRR1RRC(Rtbbox((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR1scC`s'|jdkr|jjS|jSdS(N(R RRR:(R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR:s cC`s't|jdr|jjSdSdS(Nuget_subplotspec(RRRR(R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRs N(RvRwRxRRRRzRRRRR@R1R:R(((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRs 22  t HBoxDividercB`sYeZdZedZedZddZdZdddddZ RS(cO`stj||||dS(N(RR(RRRR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscC`st|}ddl}|j|j|d|dfdd}|j|ddd}xKt|D]=\}\} } | |||fs tLocatableAxesBasecB`sGeZdZdZdZddZdedZdZ RS(cO`s,|jj|||d|_d|_dS(N(RRRRt_locator_renderer(Rtkltkw((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRvs cC`s ||_dS(N(R(RR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyR}scC`s|jS(N(R(R((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pytget_axes_locatorscC`sZ|jdkr(|jj||n.|j||j}|jj|d|dS(NRi(RRRt apply_aspectR(RRiR((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRscC`s#||_|jj|||dS(N(RRtdraw(RRtinframe((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRs cO`sd|kr'd|kr'tdnt||j|jt||}|j|j|jj||jd|jd|j j |||S(uY Need to overload so that twinx/twiny will work with these axes. usharexushareyu%Twinned Axes may share only one axis.udatalim( R6RRR1RRRRtset_adjustablet _twinned_axesR7(RRRtax2((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyt_make_twin_axess'  N( RvRwRRRRRRzRR(((sClib/python2.7/site-packages/mpl_toolkits/axes_grid1/axes_divider.pyRus     cC`sXtj|}|dkrTttd|jt|fi|d6}|t| s," 47+