File: //opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyc
�
[�3Lc @ s� d Z d d l m Z d d l m Z d d l m Z m Z d d l m Z d d l
m Z m Z d d l
Z d d l Z d e j f d � � YZ d
e j f d � � YZ d d l m Z d
e f d � � YZ d e j f d � � YZ d e f d � � YZ d d l Z i Z d � Z d d l m Z d d l m Z e e e � � Z d d l j Z e j! e � Z" d � Z# d � Z$ e% d k r�d d l& j' Z( e( j) d d d �Z* e* j+ � e$ e* � e( j, � n d S( s/
An experimental support for curvelinear grid.
i����( t chain( t
GridFinder( t AxisArtistHelpert GridHelperBase( t
AxisArtist( t Affine2Dt IdentityTransformNt FloatingAxisArtistHelperc B s e Z RS( ( t __name__t
__module__( ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR s t FixedAxisArtistHelperc B sJ e Z d d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C st | j | � \ } } t t | � j | | | d | �| d k rL | } n | | _ | | _ | | _ | | _ d S( s}
nth_coord = along which coordinate value varies.
nth_coord = 0 -> x axis, nth_coord = 1 -> y axis
t axis_directionN( t get_data_boundaryt superR
t __init__t Nonet nth_coord_tickst valuet grid_helpert _side( t selfR t sideR R t nth_coord( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR s c C s# | j j | � | j j | _ d S( N( R t
update_limt grid_info( R t axes( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR 1 s c C s� | j d } | j d k ra | j } | d | d d } d t | d | d � d } } nT | j d k r� | d | d d } | j } t | d | d � d d } } n | j j } | j | g | g � \ } } | j | j j � }
|
j
| d | d g � } d | d k o-d k n r�d | d k oMd k n r�| j j
| d | d g � \ } }
| j | | g | | g � \ } } | j j
| d | d g � \ } } | |
f t j | |
| | � t j
d
f Sd Sd S( Nt extremesi i i g @g g @�@i g �?g �f@( NN( R R R t absR t grid_findert transform_xyt transDatat transAxest invertedt transform_pointt npt arctan2t piR ( R R R t xx0t yy0t dxxt dyyR t xx1t yy1t trans_passingthrough_pointt pt xx1ct yy1ct xx2t yy2t xx2ct yy2c( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt get_axislabel_pos_angle8 s&
& & @&&&/c C s t � S( N( R ( R R ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt get_tick_transformW s c
s� � j j � � j d \ } } } � j d \ } } } t j | � t j | � } } | d k rz | | } d | } n | } d } | d k r� | | }
d | } n | }
d } � j j } t | d � \ }
} t | d � \ } } � j d k r | | k | | k @} | | } n2 � j d k rQ|
|
k |
| k @} |
| }
n � � f d � } � j d k r�t j | � }
|
j
� j � | |
| � \ � � |
j � } | |
| | k c | 8<| | | � \ � �
| | | | � \ � � | j � } | | | | k c | 8<| |
| � \ � � | |
| | � \ � �
� j d � g t
� | � D] \ } } | rk| ^ qk� n)� j d k r�t j |
� } | j
� j � | |
| � \ � � | j � } | | | | k c | 8<| |
| � \ � �
| |
| | � \ � � |
j � } | |
| | k c | 8<| | | � \ � � | | | | � \ � �
� j d � g t
� | � D] \ } } | r�| ^ q�� n � � � � � � � � � �
� � �
f
d
� } | � t g � f S( s9 tick_loc, tick_angle, tick_label, (optionally) tick_labelt lat_infot lon_infog����MbP?i i i c s[ � j | | � \ } } � j j t j | | g � j � � } | j � \ } } | | f S( N( R R t transformR" t arrayt transpose( t xt yt x1t y1t x2y2t x2t y2( R R ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR | s 't
lat_labelst
lon_labelsc
3 sa t j �
� � � � } t j � � � � � } �
� d k � � d k @} | | d d | | <� j � � } | � j j � } x� t � � | | � � D]� \ } } } } } | j | | f � }
d } d | |
d k o� d | k n r� d | |
d k o d | k n r� | d d } | d d } | | g | | | f Vq� q� Wd S( Ng gn��! @g @g�h㈵��>i g �?i g �f@( R" R# R4 R R t zipR! (
t ddt dd2t mmt
trans_tickt tr2axR: R; t dt d2t labt c2t deltat d1(
R t labelsR R) t xx1at xx1bt xx2at xx2bR* t yy1at yy1bt yy2at yy2b( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt f1� s .((N( R R R R" t asarrayR t _extremest sortedR t
empty_liket fillR t copyRC t iter( R R t lat_levst lat_nt
lat_factort lon_levst lon_nt
lon_factorR& t dyR% t dxRZ t xmint xmaxt ymint ymaxt maskR t xx00t yy00t lt mRX ( ( R R RO R R) RP RQ RR RS R* RT RU RV RW sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt get_tick_iteratorsZ sl
1
13c C s | j S( N( R ( R R ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt get_line_transform� s c
C su | j | � d d l m } t d d d d d d
d
d � | j \ } } | j | | \ } } | t | | � � S( Ni����( t Patht leftt
lon_lines0i t righti t bottomt
lat_lines0t top( Ru i ( Ru i ( Rx i ( Rx i ( R t matplotlib.pathRs t dictR R RC ( R R Rs t kt vt xxt yy( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt get_line� s
N(
R R R R R R3 R4 Rq Rr R� ( ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR
s n ( t ExtremeFinderSimplet ExtremeFinderFixedc B s e Z d � Z d � Z RS( c C s
| | _ d S( N( RZ ( R R ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR � s c C s | j S( s�
get extreme values.
x1, y1, x2, y2 in image coordinates (0-based)
nx, ny : number of dvision in each axis
( RZ ( R R R< R= R? R@ ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt __call__� s ( R R R R� ( ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR� � s t GridHelperCurveLinearc B sV e Z d d d d d � Z d � Z d d d d d � Z d � Z d � Z d � Z RS( c C sS d | _ | | _ t | � } t t | � j | | d | d | d | d | �d S( s�
aux_trans : a transform from the source (curved) coordinate to
target (rectlinear) 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_locator2t tick_formatter1t tick_formatter2N( R t _old_valuesRZ R� R
R� R ( R t aux_transR R� R� R� R� t extreme_finder( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR � s c
C sP | j \ } } } } t d | d f d | d f d | d f d | d f � | S( s%
return v= 0 , nth=1
Rt i Rv Rw i Ry ( RZ R{ ( R R t lon1t lon2t lat1t lat2( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR s
c C s� | d k r | j } n | d k r- | } n t | | d | �} t | | d | �} | j j t � | j j | j j � | S( NR R ( R R R
R t linet set_clip_ont Truet set_clip_boxt bbox( R t locR R t offsetR t _helpert axisline( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt new_fixed_axis+ s c C s7 | j d k r t � | _ n | j } | j } | j | j | | | | � } t | d � \ } } t | d � \ }
} | j | | � \ } }
} | j |
| � \ } } } | | |
| f | d <| |
| f | d <| | | f | d <| j d | | � | d <| j
d | | � | d <| d k r@t j | |
� } n t j | |
| � } | d k ryt j | | � } n t j | | | � } | | | k | | k @} | |
| k | | k @} | j
| | | | |
| � \ } } | | d <| | d <| j
| d | d | � \ } } | | d
<| | d <d S( Ni R R6 R5 Rw RB RA t lon_linest lat_linesRu Rx ( R R R{ R R� t inv_transform_xyR[ R� R� R� R� R" RY t _get_raw_grid_lines( R R< R= R? R@ R R R t lon_mint lon_maxt lat_mint lat_maxRc Rd Re R` Ra Rb t
lon_valuest
lat_valuest lon_values0t lat_values0R� R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt _update_gridY sL
c C sZ g } x% | j d D] } | j | g � q Wx% | j d D] } | j | g � q< W| S( NR� R� ( R t extend( R t
grid_linest gl( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt
get_gridlines� s c C sF | j \ } } } } | j } t j | | d � } t j | � t j | � } } | j | � | j | � t j | | d � } t j | � t j | � }
} |
j | � | j | � t j | d | d | d d d � |
g � } t j | d | d | d | d d d � g � }
| j t j | |
g � j � � } | S( sD
return Nx2 array of x,y coordinate of the boundary
id i����i N(
RZ t
_aux_transR" t linspaceR\ R] t concatenateR7 R8 R9 ( R t x0R< t y0R= t trR~ R& R* R R% R) t xxxt yyyt t( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt get_boundary� s
04$N( R R R R R R� R� R� R� ( ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR� � s / * A
t FloatingAxesBasec B s, e Z d � Z d � Z d � Z d � Z RS( c O s| | j d d � } | d k r- t d � � n t | d � sK t d � � n | j j | | | � | j d � | j � d S( NR s* FloatingAxes requires grid_helper argumentR� s. grid_helper must implement get_boundary methodg �?( t getR t
ValueErrort hasattrt _axes_class_floatingR t
set_aspectt adjust_axes_lim( R t klt kwargsR ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR � s
c C s4 d d l j } | j � } | j � } | j | � S( s[
Returns the patch used to draw the background of the axes. It
is also used as the clipping path for any data elements on the
axes.
In the standard axes, this is a rectangle, but in other
projections it may not be.
.. note::
Intended to be overridden by new projection types.
i����N( t matplotlib.patchest patchest get_grid_helperR� t Polygon( R t mpatchesR R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt _gen_axes_patch� s c C s� | j j | � | j j | j � | j j | � } | j | j � | j t � | j | j
� | j j | � | j j | � | | _
d S( N( R� t clat patcht
set_transformR R� t
set_figuret figuret set_visiblet FalseR t
set_clip_patht gridlinest _original_patch( R R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR� � s
c C s� | j � } | j � } | d d � d f | d d � d f } } t | � t | � } } t | � t | � } } | | d } | | d }
| j | | | | � | j | |
| |
� d S( Ni i g Y@( R� R� t mint maxt set_xlimt set_ylim( R R R� R: R; Rh Ri Rj Rk Rg Rf ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR� � s -( R R R R� R� R� ( ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyR� � s
c C sU t j | � } | d k rQ t j d | j t | f i | d 6� } | t | <n | S( Ns Floating %sR� ( t _floatingaxes_classesR� R t newt classobjR R� ( t
axes_classt new_class( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt floatingaxes_class_factory
s
( t Axes( t host_axes_class_factoryc C s� d d l } d d l } d d l m } t � j | j d d � | j � } | j d � } | j � } d d l
m } | d d d
d d g � } t | d
d. d | d | d | d d �} t | d d | �a | j t � d }
t � j d d |
� | } | d d d g � } t | d
d d d |
d |
f d | �} | j d d t �t j d <a t j d j j d � t j d j j d � x( d d g D] }
t j |
j t � q�W| j d! d" d t d# d$ �t j d% <a t j d& t d' t � t j j d( � t j j t � t j j d) � t j | � } d* d d+ d g d d, d d g } } | j | | � | j | | d- � \ } | j t j � d S(/ s5
polar projection, but in a rectangular box.
i����N( t PolarAxesg �f@g �?i ( t FixedLocatori i i i i
R i ih i R� R� R� R� io R g $@i i< iZ g @Rv R Rt s Test 1s Test 2i i R Rw t zt allt labels z = ?s 0.5iC iK i s k-( i ih i
i (! t numpyt angle_helpert matplotlib.projectionsR� R t scaleR$ t PolarTransformt
LocatorDMSt FormatterDMSR R� R� R t FloatingSubplott ax1t add_subplotR� t axisR� t set_textR� R� t new_floating_axist toggleR� R� t set_colort get_aux_axest scattert plotR� R� ( t figR" R� R� R� R� R� R� R� R t r_scalet tr2t grid_helper2t ant ax2R~ R Ro ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt curvelinear_test3* sP &
%c C s% d d l } d d l } d d l m } t � j | j d d � | j � } | j d � } | j � } d d l
m } | d d d
d d g � } t | d
d* d | d | d | d d �} t | d d | �a | j t � t j d j j d � t j d j j d � x% d g D] }
t j |
j t � qW| j d d d t d d �t j d! <a t j d" t d# t � t j j d � t j j d$ � t j j t � t j j d% � t j | � } d& d' d( d g d d d d g } }
| j | |
� | j | |
d) � \ } | j t j � d S(+ s5
polar projection, but in a rectangular box.
i����N( R� g �f@g �?i ( R� i i i i i
R ix i i R� R� R� R� io R Rt s Test 1Rv s Test 2Ry i iF R R Rw R� R� R� s z = ?s 0.5iC iZ iK s k-( ix i i
i (! R� R� R� R� R R� R$ R� R� R� R R� R� R R� R� R� R� R� R� R� R� R� R� R� t set_axis_directionR� R� R� R� R� R� R� ( R� R" R� R� R� R� R� R� R� R R� R� R~ R Ro ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt curvelinear_test4� sB &
%t __main__i t figsizei ( i i (- t __doc__t itertoolsR R R t axislinesR R t axis_artistR t matplotlib.transformsR R R� R" t grid_helper_curvelinearR R
R� R� R� t objectR� R� R� R� R� t% mpl_toolkits.axes_grid1.parasite_axesR� t FloatingAxest matplotlib.axesR t maxest subplot_class_factoryR� R� R� R t matplotlib.pyplott pyplott pltR� R� t clft show( ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/floating_axes.pyt <module> s: ��D [ @