Rc@sdZddlmZddlmZddlmZmZddlm Z ddl m Z m Z ddl ZddlZdejfd YZd ejfd YZdd lmZd efdYZdejfdYZdefdYZiZdZddlmZddlmZeeeZddljZej eZ!dZ"dZ#e$dkrddl%j&Z'e'j(dddZ)e)j*e#e)e'j+ndS(s/ An experimental support for curvilinear grid. i(tchain(t GridFinder(tAxisArtistHelpertGridHelperBase(t AxisArtist(tAffine2DtIdentityTransformNtFloatingAxisArtistHelpercBseZRS((t__name__t __module__(((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyRstFixedAxisArtistHelpercBsJeZddZdZdZdZdZdZdZ RS(cCst|j|\}}tt|j|||d||dkrL|}n||_||_||_||_dS(s} nth_coord = along which coordinate value varies. nth_coord = 0 -> x axis, nth_coord = 1 -> y axis taxis_directionN( tget_data_boundarytsuperR t__init__tNonetnth_coord_tickstvaluet grid_helpert_side(tselfRtsideRRt nth_coord((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyRs     cCs#|jj||jj|_dS(N(Rt update_limt grid_info(Rtaxes((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR1scCs|jd}|jdkra|j}|d|dd}dt|d|dd}}nT|jdkr|d|dd}|j}t|d|ddd}}n|jj}|j|g|g\}} |j|jj } | j |d| dg} d| dko-d knrd| dkoMd knr|jj |d| dg\} } |j||g||g\}}|jj |d|dg\}}| | ft j || || t j d fSd SdS( Ntextremesiiig@gg@@ig?gf@(NN(RRRtabsRt grid_findert transform_xyt transDatat transAxestinvertedttransform_pointtnptarctan2tpiR(RRRtxx0tyy0tdxxtdyyRtxx1tyy1ttrans_passingthrough_pointtptxx1ctyy1ctxx2tyy2txx2ctyy2c((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytget_axislabel_pos_angle8s&  & & @&&&/cCstS(N(R(RR((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytget_tick_transformWsc sjjjd\}}}jd\}}}tj|tj|}}|d k rz||}d|} n |}d} |d k r||} d|} n |} d} jj} t| d \} }t| d\}}jdkr||k||k@}||}n2jdkrQ| | k| |k@}| |} nfd}jdkrtj |} | j j || |\ | j }|| | |kc| 8<|||\ ||| |\ |j }||| |kc| 8<|| |\ || || \ jdgt |D]\}}|rk|^qkn)jdkrtj | }|j j || |\ |j }||| |kc| 8<|| |\ || || \ | j }|| | |kc| 8<|||\ ||| |\ jd gt |D]\}}|r|^qn f d }|tgfS( s9tick_loc, tick_angle, tick_label, (optionally) tick_labeltlat_infotlon_infogMbP?iiics[j||\}}jjtj||gj}|j\}}||fS(N(RRt transformR"tarrayt transpose(txtytx1ty1tx2y2tx2ty2(RR(s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR|s't lat_labelst lon_labelsc 3satj }tj } dkdk@}||dd||<j}|jj}xt||D]\}}}}} |j||f} d} d| | dkod| knrd| | dkod| knr|dd} |dd}||g| || fVqqWdS( Nggn! @g@gh㈵>ig?igf@(R"R#R4RR tzipR!( tddtdd2tmmt trans_tickttr2axR:R;tdtd2tlabtc2tdeltatd1( RtlabelsRR)txx1atxx1btxx2atxx2bR*tyy1atyy1btyy2atyy2b(s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytf1s.((N(RRRR"tasarrayRt _extremestsortedRt empty_liketfillRtcopyRCtiter(RRtlat_levstlat_nt lat_factortlon_levstlon_nt lon_factorR&tdyR%tdxRZtxmintxmaxtymintymaxtmaskRtxx00tyy00tltmRX((RRRORR)RPRQRRRSR*RTRURVRWs_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytget_tick_iteratorsZsl             1   13cCs|jS(N(R(RR((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytget_line_transformsc Csu|j|ddlm}tdd dd dd d d|j\}}|j||\}}|t||S(Ni(tPathtleftt lon_lines0itrightitbottomt lat_lines0ttop(Rui(Rui(Rxi(Rxi(Rtmatplotlib.pathRstdictRRRC(RRRstktvtxxtyy((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytget_lines  N( RR RRRR3R4RqRrR(((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR s     n (tExtremeFinderSimpletExtremeFinderFixedcBseZdZdZRS(cCs ||_dS(N(RZ(RR((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyRscCs|jS(s get extreme values. x1, y1, x2, y2 in image coordinates (0-based) nx, ny : number of division in each axis (RZ(RRR<R=R?R@((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt__call__s(RR RR(((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyRs tGridHelperCurveLinearcBs\eZdddddZdZdddddZdZdddZdZRS( c CsSd|_||_t|}tt|j||d|d|d|d|dS(s 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_valuesRZRR RR(Rt aux_transRRRRRtextreme_finder((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyRs   c CsP|j\}}}}td|dfd|dfd|dfd|df|S(s% return v= 0 , nth=1 RtiRvRwiRy(RZR{(RRtlon1tlon2tlat1tlat2((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR s   cCs|dkr|j}n|dkr-|}nt||d|}t||d|}|jjt|jj|jj|S(NRR ( RRR Rtlinet set_clip_ontTruet set_clip_boxtbbox(RtlocRR toffsetRt_helpertaxisline((s_/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pytnew_fixed_axis+s      cCs7|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 s8  D  [ @