ó m[c@`s–dZddlmZmZmZmZddlZddlmZddl Z ddl m Z m Z ddlmZddlmZmZdd lmZdd lmZd ejfd „ƒYZd ejfd„ƒYZddlmZdefd„ƒYZdejfd„ƒYZdefd„ƒYZiZd„Zddlm Z ddl!m"Z"ee"e ƒƒZ#ddl$j%Z&e&j'e#ƒZ(dS(u/ An experimental support for curvilinear grid. i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tzip(tAffine2DtIdentityTransformi(tgrid_helper_curvelinear(tAxisArtistHelpertGridHelperBase(t AxisArtist(t GridFindertFloatingAxisArtistHelpercB`seZRS((t__name__t __module__(((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR stFixedAxisArtistHelpercB`sJeZdd„Zd„Zd„Zd„Zd„Zd„Zd„Z RS(cC`st|j|ƒ\}}tt|ƒj|||d|ƒ|dkrL|}n||_||_||_||_dS(u} nth_coord = along which coordinate value varies. nth_coord = 0 -> x axis, nth_coord = 1 -> y axis taxis_directionN( tget_data_boundarytsuperRt__init__tNonetnth_coord_tickstvaluet grid_helpert_side(tselfRtsideRRt nth_coord((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyRs     cC`s#|jj|ƒ|jj|_dS(N(Rt update_limt grid_info(Rtaxes((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR2scC`sû|jd}|jdkra|j}|d|dd}dt|d|dƒd}}nT|jdkrµ|d|dd}|j}t|d|dƒdd}}n|jj}|j|g|gƒ\}} |j|jj ƒ} | j |d| dgƒ} d| dko-d knród| dkoMd knró|jj |d| dgƒ\} } |j||g||gƒ\}}|jj |d|dgƒ\}}| | ft j || || ƒt j d fSd SdS( Nuextremesiiig@gg@@igð?g€f@(NN(RRRtabsRt grid_findert transform_xyt transDatat transAxestinvertedttransform_pointtnptarctan2tpiR(RRtextremestxx0tyy0tdxxtdyyR txx1tyy1ttrans_passingthrough_pointtptxx1ctyy1ctxx2tyy2txx2ctyy2c((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytget_axislabel_pos_angle9s&  & & @&&&/cC`stƒS(N(R(RR((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytget_tick_transformXsc `sˆjj‰ˆjd\}}}ˆjd\}}}tj|ƒtj|ƒ}}|d k rz||}d|} n |}d} |d k r©||} d|} n |} d} ˆjj} t| d ƒ\} }t| dƒ\}}ˆjdkr||k||k@}||}n2ˆjdkrQ| | k| |k@}| |} n‡‡fd†}ˆjdkržtj |ƒ} | j ˆj ƒ|| |ƒ\‰‰ | j t dtƒ}|| | |kc| 8<|||ƒ\‰‰ ||| |ƒ\‰‰ |j t dtƒ}||| |kc| 8<|| |ƒ\‰‰ || || ƒ\‰‰ ˆjd ‰gtˆ|ƒD]\}}|r}|^q}‰n;ˆjdkrÙtj | ƒ}|j ˆj ƒ|| |ƒ\‰‰ |j t dtƒ}||| |kc| 8<|| |ƒ\‰‰ || || ƒ\‰‰ | j t dtƒ}|| | |kc| 8<|||ƒ\‰‰ ||| |ƒ\‰‰ ˆjd ‰gtˆ|ƒD]\}}|r¸|^q¸‰n‡‡‡‡‡‡‡‡‡ ‡ ‡ ‡ ‡ f d †}|ƒtgƒfS( u9tick_loc, tick_angle, tick_label, (optionally) tick_labelulat_infoulon_infogü©ñÒMbP?iiic`s[ˆj||ƒ\}}ˆjjtj||gƒjƒƒ}|jƒ\}}||fS(N(R!R"t transformR&tarrayt transpose(txtytx1ty1tx2y2tx2ty2(RR (sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR!}s'tcopyu lat_labelsu lon_labelsc 3`sdtjˆ ˆ ˆˆƒ}tjˆ ˆ ˆˆƒ}ˆ ˆ dkˆˆdk@}||tjd||<ˆjˆƒ}|ˆjjƒ}xÐtˆˆ||ˆƒD]¶\}}}}} |j||fƒ} d} d| | dkoùd| knr¦d| | dko!d| knr¦|dd} |dd}||g| || fVq¦q¦WdS( Ngigñh㈵øä>igð?ign†ðù! @g€f@(R&R'R(R9R#R$RR%( tddtdd2tmmt trans_tickttr2axR=R>tdtd2tlabtc2tdeltatd1( RtlabelsRR.txx1atxx1btxx2atxx2bR/tyy1atyy1btyy2atyy2b(sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytf1¯s.((N(RR RR&tasarrayRt _extremestsortedRt empty_liketfillRtastypetfloattTrueRtiter(RRtlat_levstlat_nt lat_factortlon_levstlon_nt lon_factorR+tdyR*tdxR[txmintxmaxtymintymaxtmaskR!txx00tyy00tltmRY((RR RPRR.RQRRRSRTR/RURVRWRXsDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytget_tick_iterators[sl           1 13cC`s|jS(N(R"(RR((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytget_line_transformÉsc C`s{|j|ƒddlm}tdd dd dd d d ƒ|j\}}|j||\}}|tj||gƒƒS(Ni(tPathtleftu lon_lines0trightitbottomu lat_lines0ttop(u lon_lines0i(u lon_lines0i(u lat_lines0i(u lat_lines0i(Rtmatplotlib.pathRvtdictRRR&t column_stack(RRRvtktvtxxtyy((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytget_lineÌs  N( R RRRRR8R9RtRuR‚(((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyRs     n (tExtremeFinderSimpletExtremeFinderFixedcB`seZd„Zd„ZRS(cC`s ||_dS(N(R[(RR)((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyRÝscC`s|jS(u get extreme values. x1, y1, x2, y2 in image coordinates (0-based) nx, ny : number of division in each axis (R[(RR!R?R@RBRC((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt__call__às(R RRR…(((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR„Üs tGridHelperCurveLinearcB`s\eZddddd„Zd„Zddddd„Zd„Zddd„Zd„ZRS( c C`sSd|_||_t|ƒ}tt|ƒj||d|d|d|d|ƒ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)* t grid_locator1t grid_locator2ttick_formatter1ttick_formatter2N(Rt _old_valuesR[R„RR†R(Rt aux_transR)R‡RˆR‰RŠtextreme_finder((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyRîs   c C`sP|j\}}}}td|dfd|dfd|dfd|dfƒ|S(u% return v= 0 , nth=1 RwiRxRyiRz(R[R|(RRtlon1tlon2tlat1tlat2((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR!s   cC`s|dkr|j}n|dkr-|}nt||d|ƒ}t||d|ƒ}|jjtƒ|jj|jjƒ|S(NRR( RRRR tlinet set_clip_onRat set_clip_boxtbbox(RtlocRRtoffsetRt_helpertaxisline((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytnew_fixed_axis,s      cC`s7|jdkrtƒ|_n|j}|j}|j|j||||ƒ}t|d ƒ\}} t|dƒ\} } |j|| ƒ\} } }|j| | ƒ\}}}|| | | f|d<| | |f|d<|||f|d<|j d|| ƒ|d<|j d||ƒ|d<|dkr@t j | | ƒ}nt j | | |ƒ}|dkryt j || ƒ}nt j || |ƒ}|||k|| k@}|| |k|| k@}|j |||| | | ƒ\}}||d<||d <|j |d |d|Œ\}}||d <||d RkRlRmRnRjRi((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR¾ús  -(R RRRÆRÇR¾(((sDlib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR¸Ès  cC`sXtj|ƒ}|dkrTttd|jƒt|fi|d6ƒ}|t|s,"  ÀÜD