File: //opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyo
�
[�3Lc @ sJ d d l Z d d l j Z d d l m Z d d l Z e j Z d d l j
Z d d l m Z d e
f d � � YZ d e
f d � � YZ d e f d � � YZ d
e j f d � � YZ d e
f d
� � YZ d e
f d � � YZ d e
f d � � YZ e d k rFe � Z e d d � \ Z Z Z e � Z e d d e � GHn d S( i����N( t Bbox( t Transformt ExtremeFinderSimplec B s# e Z d � Z d � Z d � Z RS( c C s | | | _ | _ d S( N( t nxt ny( t selfR R ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyt __init__ s c C s� t j | | | j � t j | | | j � } } t j | | � \ } } | t j | � t j | � � \ }
} |
j � |
j � } }
| j � | j � } } | j | |
| | � S( s�
get extreme values.
x1, y1, x2, y2 in image coordinates (0-based)
nx, ny : number of dvision in each axis
( t npt linspaceR R t meshgridt ravelt mint maxt _add_pad( R t transform_xyt x1t y1t x2t y2t x_t y_t xt yt lont latt lon_mint lon_maxt lat_mint lat_max( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyt __call__ s 1'c C s\ | | | j } | | | j } | | | | } } | | | | } } | | | | f S( N( R R ( R R R R R t dlont dlat( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR
! s
( t __name__t
__module__R R R
( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR
s t GridFinderBasec B sD e Z d d d � Z d � Z d � Z d � Z d � Z d � Z RS( c C sD t t | � j � | | _ | | _ | | _ | | _ | | _ d S( s�
the transData of the axes to the world coordinate.
locator1, locator2 : grid locator for 1st and 2nd axis.
Derived must define "transform_xy, inv_transform_xy"
(may use update_transform)
N( t superR" R t extreme_findert
grid_locator1t
grid_locator2t tick_formatter1t tick_formatter2( R R$ R% R& R' R( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR 0 s
c C sZ | j | j | | | | � } | \ } } } } | j | | � \ }
} } | j | | � \ }
} } | d
k r� t j |
| � } n t j |
| | � } | d
k r� t j |
| � } n t j |
| | � } | j | | | | | | � \ } } | | d } | | d } t j | | | | | | | | � } i } | | d <| | d <| | d <| j
| | |
| � | d <| j
| | |
| � | d <t � } | d d <xE d d d
d g D]1 } | d d | } | j | | | � | | <q�Wt � } | d d <xE d d d
d g D]1 } | d d | } | j
| | | � | | <q!W| S( s�
lon_values, lat_values : list of grid values. if integer is given,
rough number of grids in each direction.
g�����|�=t extremest lon_linest lat_linesR R t tick_labelst leftt bottomt rightt topt tick_levelsN( R$ t inv_transform_xyR% R& t NoneR t asarrayt _get_raw_grid_linesR t from_extentst _clip_grid_lines_and_find_tickst dictR' R( ( R R R R R R) R R R R t lon_levst lon_nt
lon_factort lat_levst lat_nt
lat_factort
lon_valuest
lat_valuesR* R+ t ddxt ddyt bbt grid_infot
tck_labelst directiont levs( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyt
get_grid_infoE sP (
c
C s� t j | | d � } t j | | d � } g | D]% } | j t j | � | | � ^ q1 }
g | D]% } | j | t j | � | � ^ qc } |
| f S( Nid ( R R R t
zeros_like(
R R? R@ R R R R t lons_it lats_iR R* R R+ ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR5 � s //c C sZ t � } g | d <g | d <t d g d g d g d g � | d <t d g d g d g d g � | d <g | d <| d } | d } x� t | | | � D]� \ \ } } }
} t | | | � \ } }
| s� q� n | d j |
� | d j | � x\ t |
d d d d g � D]? \ } } x0 | D]( } | | j | � | | j | � q"WqWq� W| S(
Nt valuest levelsR- R. R/ R0 R1 t tick_locst lines( R8 t zipt clip_line_to_rectt append( R RO RL RG RC t git
tck_levelst tck_locst lxt lyt vt levt xyt tckst tckRF t t( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR7 � s&
%%
((
c sR t � t � r0 � f d � } � f d � } n � \ } } | | _ | | _ d S( Nc s� t j | � t j | � } } t j | d d � t j f | d d � t j f f d � } � j | � } | d d � d f | d d � d f } } | | f S( Ni i ( R R4 t concatenatet newaxist transform( R R t ll1t ll2R R ( t aux_trans( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR � s
>-c s� t j | � t j | � } } t j | d d � t j f | d d � t j f f d � } � j � j | � } | d d � d f | d d � d f } } | | f S( Ni i ( R R4 R^ R_ t invertedR` ( R R Ra Rb R R ( Rc ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR2 � s
>-( t
isinstanceR R R2 ( R Rc R R2 ( ( Rc sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyt update_transform� s c K sD x= | D]5 } | d k r0 t | | | | � q t d � � q Wd S( NR$ R% R& R' R( s unknwonw update property( s extreme_finders
grid_locator1s
grid_locator2s tick_formatter1s tick_formatter2( t setattrt
ValueError( R t kwt k( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyt update� s
N( R R! R3 R RH R5 R7 Rf Rk ( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR" / s C t
GridFinderc B s e Z d d d d d d � Z RS( c C s� | d k r t d d � } n | d k r6 t � } n | d k rN t � } n | d k rf t � } n | d k r~ t � } n t t | � j | | | | | � | j | � d S( s
transform : transfrom from the image coordinate (which will be
the transData of the axes to the world coordinate.
or transform = (transform_xy, inv_transform_xy)
locator1, locator2 : grid locator for 1st and 2nd axis.
i N( R3 R t MaxNLocatort FormatterPrettyPrintR# Rl R Rf ( R R` R$ R% R& R' R( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR � s" N( R R! R3 R ( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyRl � s
Rm c B s5 e Z d d e e e d d � Z d � Z d � Z RS( i
c
C sH t j j | | d | d | d | d | d | �| j � d | _ d S( Nt stepst trimt integert symmetrict prune( t mtickerRm R t create_dummy_axisR3 t _factor( R t nbinsRo Rp Rq Rr Rs ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR � s
c C s� | j d k r^ | j | | j | | j � t j j | � } t j | � t | � | j f S| j | | � t j j | � } t j | � t | � d f Sd S( N( Rv R3 t
set_boundsRt Rm R R t arrayt len( R t v1t v2t locs( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR s c C s
| | _ d S( N( Rv ( R t f( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyt
set_factor s N( R R! R3 t Truet FalseR R R ( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyRm � s
t FixedLocatorc B s e Z d � Z d � Z RS( c C s
| | _ d S( N( t _locs( R R} ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR s c C sh t | | g � \ } } t j g | j D]$ } | | k r( | | k r( | ^ q( � } | t | � d f S( N( t sortedR Ry R� Rz R3 ( R R{ R| t lR} ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR s =( R R! R R ( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR� s Rn c B s e Z d � Z d � Z RS( c C s) t j � | _ | j j � t | _ d S( N( Rt t ScalarFormattert _fmtRu R� t _ignore_factor( R ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR ' s
c C sn | j s> | d k r d } n g | D] } | | ^ q% } n | j j | � g | D] } | j | � ^ qU S( Ng �?( R� R3 R� t set_locs( R RF t factorRL RX ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR , s ( R R! R R ( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyRn & s t
DictFormatterc B s e Z d d � Z d � Z RS( c C s) t t | � j � | | _ | | _ d S( sq
format_dict : dictionary for format strings to be used.
formatter : fall-back formatter
N( R# R� R t _format_dictt _fallback_formatter( R t format_dictt formatter( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR 7 s c C so | j r! | j | | | � } n d g t | � } g t | | � D]! \ } } | j j | | � ^ qD } | S( sG
factor is ignored if value is found in the dictionary
t ( R� Rz RP R� t get( R RF R� RL t fallback_stringsRj RX t r( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR @ s .N( R R! R3 R R ( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyR� 6 s t __main__i id R- ( t numpyR t matplotlib.cbookt cbookt mcbookt matplotlib.transformsR t clip_pathRQ t matplotlib.tickert tickerRt R t objectR R" Rl Rm R� Rn R� R t locatorR} t nlocR� t fmtR3 ( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/axisartist/grid_finder.pyt <module> s$ "�'