ó m[c@`sdZddlmZmZmZmZddlZddlmZddl m Z ddl m Z ddl mZmZdd lmZdd lmZmZddlZdd lmZd ejfd „ƒYZdejfd„ƒYZdefd„ƒYZdS(u/ An experimental support for curvilinear grid. i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tzip(tchaini(t GridFinder(tAxisArtistHelpertGridHelperBase(t AxisArtist(tAffine2DtIdentityTransform(tPathtFixedAxisArtistHelpercB`sAeZdZdd„Zd„Zdd„Zd„Zd„ZRS(u( Helper class for a fixed axis. cC`sYtt|ƒjd|ƒ||_|dkr:|j}n||_||_t|_ dS(u} nth_coord = along which coordinate value varies. nth_coord = 0 -> x axis, nth_coord = 1 -> y axis tlocN( tsuperR t__init__t grid_helpertNonet nth_coordtnth_coord_tickstsidetFalset_limits_inverted(tselfRRR((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyRs     cC`sk|jj|ƒ|jdkr4|jƒ\}}n|jƒ\}}||kr^t|_n t|_dS(Ni(Rt update_limRtget_ylimtget_xlimtTrueRR(Rtaxestxy1txy2((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyR)s  cC`sG|dkrd|j|_n$|dkr7||_n tdƒ‚dS(Niiuwrong coord number(ii(RRt Exception(Rt coord_number((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pytchange_tick_coord7s    cC`s|jS(N(t transData(RR((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pytget_tick_transform@scC`s”|j}|jr>idd6dd6dd6dd6|j}n |j}|j|j|ƒ}|jd|j|dtƒ}t||ƒtgƒfS(u tick_loc, tick_angle, tick_labelurightuleftubottomutopitminor(RRRtget_tick_iteratorRRRtiter(RRtgRtti1tti2((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pytget_tick_iteratorsCs   N( t__name__t __module__t__doc__RRRR"R$R+(((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyR s    tFloatingAxisArtistHelpercB`s\eZd d„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d„Z RS( cC`sJtt|ƒj||ƒ||_||_d|_d|_d|_dS(u} nth_coord = along which coordinate value varies. nth_coord = 0 -> x axis, nth_coord = 1 -> y axis idN(NN( RR/RtvalueRRt _extremest_get_line_patht_line_num_points(RRRR0taxis_direction((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyRYs    cC`s||f|_dS(N(R1(Rte1te2((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyt set_extremesiscC`sá|jj|ƒ|jƒ\}}|jƒ\}}|jj}|j|j||||ƒ}t|ƒ}|j\}} |j dkr×|dk r®t ||dƒ|digð?ign†ðù! @g€f@(RCR_R`R$R\R]RR^( tddtdd2tmmt trans_tickttr2axRqRrtdtd2tlabtc2tdeltatd1( RtlabelsRRctxx1atxx1btxx2atxx2bRdtyy1atyy1btyy2atyy2b(sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pytf1s .((N(RR8RFRCtasarrayRR1tsortedtinfRt empty_liketfillR0tcopyRR'(RRRSRTRURWtdyRPRQRRRVtdxte0R5tmaskREtxx00tltmRˆ((RR8RRRcR€RR‚RƒRdR„R…R†R‡sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyR+Êsn               1  13cC`s|jS(N(R#(RR((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pytget_line_transform7scC`s_|j|ƒ|jd\}}|jdkrHttj||gƒƒS|j|||ƒSdS(Nuline_xy(RRFR2RR RCt column_stack(RRRqRr((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pytget_line:s  N( R,R-RRR7RRZRmR$R+R–R˜(((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyR/Ws   8   m tGridHelperCurveLinearcB`s}eZd d d d d d„Zd d„Zd„Zd d d d d„Zd dd„Zd„Zddd „Z e d „Z RS( cC`sPtt|ƒjƒd|_d|_||_t||||||ƒ|_dS(uÇ aux_trans : a transform from the source (curved) coordinate to target (rectilinear) coordinate. An instance of MPL's Transform (inverse transform should be defined) or a tuple of two callable objects which defines the transform and its inverse. The callables need take two arguments of array of source coordinates and should return two target coordinates. e.g., ``x2, y2 = trans(x1, y1)`` N( RR™RRRFt _old_valuest _aux_transRR8(Rt aux_transR9R?R@RARB((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyRHs   cK`s=|dk r|jj|ƒn|jj||jƒdS(N(RR8tupdate_transformtupdatet invalidate(RRœtkw((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pytupdate_grid_findergs cC`sc|jƒr+|j||||fkr+dS|j||||ƒ||||f|_t|_dS(u!bbox in 0-based image coordinatesN(tvalidRšt _update_gridRt _force_update(RRGRHRIRJ((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyt_updateps 'cC`s[|dkr|j}n|dkr-|}nt||d|ƒ}t||d|ƒ}|S(NRR4(RRR R (RRRR4toffsetRt_helpertaxisline((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pytnew_fixed_axis~s      ubottomcC`sf|dkr|j}nt||||ƒ}t||ƒ}|jjtƒ|jj|jjƒ|S(N( RRR/R tlinet set_clip_onRt set_clip_boxtbbox(RRR0RR4R§R¨((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pytnew_floating_axis”s   cC`s"|jj||||ƒ|_dS(N(R8t get_grid_infoRF(RRGRIRHRJ((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyR£´sumajorubothcC`szg}|dkr>x)|jddD]}|j|ƒq$Wn|dkrvx)|jddD]}|j|ƒq\Wn|S( Nubothuxulonulinesuyulat(ubothux(ubothuy(RFtextend(Rtwhichtaxist grid_linestgl((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyt get_gridlines¹s  c `sutddddddddƒˆ‰ddg|‰|sV‡‡‡‡fd †}n‡‡‡‡fd †}|ƒS( NtleftiZtrighttbottomittopulonulatc3`s_xXtˆjˆdˆˆjˆdˆƒD])\\}}}|}||ˆ|fVq.WdS(Nu tick_locsu tick_labels(RRF(txytaR”t angle_normal(t angle_tangentt axis_sidet lon_or_latR(sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pytfÍs(c3`s_xXtˆjˆdˆˆjˆdˆƒD])\\}}}|}||ˆdfVq.WdS(Nu tick_locsu tick_labelsu(RRF(RºR»R”R¼(R½R¾R¿R(sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyRÀÓs((R>(RRR¾R%RÀ((R½R¾R¿RsNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyR&Æs %N( R,R-RRR¡R¥R©R®R£RµRR&(((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyR™Fs       (R.t __future__RRRRtsixt six.movesRt itertoolsRR8Rt axislinesRRt axis_artistR tmatplotlib.transformsR R tnumpyRCtmatplotlib.pathR tFixedR tFloatingR/R™(((sNlib/python2.7/site-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyts"  Cï