Ńň
ÝZ}Jc @ s8 d Z d d k l Z d d k l Z d d k Z d d k i Z d Z d Z
d Z d Z d Z
d
Z d Z d Z d
Z d Z d Z d Z d Z d Z d Z e Z d Z d Z d Z d d Z d d Z d Z d Z d Z! e" d j o e n d S( s,
Various transforms used for by the 3D code
i˙˙˙˙( t LineCollection( t CircleNc C s | d \ } } | d \ } } | | j o d } d } | } nK | | j o d } d } | } n' | | } | | } | | | | } | | | f S( s<
Return 2D equation of line in the form ax+by+c = 0
i i˙˙˙˙i i ( ( t p0t p1t x0t y0t x1t y1t at bt c( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt line2d s
c C sN | \ } } } | \ } } t | | | | | t i | d | d S( sO
Distance from line to point
line is a tuple of coefficients a,b,c
i ( t abst npt sqrt( t lt pR R R
R R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt line2d_dist% s c C sĚ | d | d } | d | d } t i | d | d } t i | d | d } | | | | t t | d | d } t i | d d } t i | | | d | | | d } | S( sŇ distance(s) from line defined by p1 - p2 to point(s) p0
p0[0] = x(s)
p0[1] = y(s)
intersection point p = p1 + u*(p2-p1)
and intersection point lies within segement if u is between 0 and 1
i i i ( R
t asarrayt floatR t clipR ( R t p2R t x21t y21t x01t y01t ut d( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt line2d_seg_dist/ s
.+c C s* d d k } | i } d
d } } | i | | t | | } | \ } } d d
} } | i | | t | | | d | d f } t | | t i | | f } xK t | | | D]7 \ } } }
t | | f |
d d } | i | qÁ W| i
d d | i d d | i d S( Ni˙˙˙˙i i i i id iČ t filli8˙˙˙( i i ( i i ( i i i i ( id i i iČ (
t pylabt gcat plott zipt scatterR R
t arrayR t add_patcht xlimt ylimt show( R t axt xst yst pointsR R t distt xt yR R
( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt test_lines_distsD s"
! c C s- t i | d d | d d | d d S( s 3d vector lengthi i i ( R
R ( t v( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt modZ s c C s | | | | | | } } } t i d | d d | | g d d | d | | g d d d | | | g d d d d g g S( Ng đ?i ( R
R# ( t xmint xmaxt ymint ymaxt zmint zmaxt dxt dyt dz( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt world_transformation^ s c C sH d \ } } d \ } } d \ } } t | | | | | | } | GHd S( Nid ix i˙˙˙gš?gÉ?( id ix ( i˙˙˙id ( gš?gÉ?( R; ( R2 R3 R4 R5 R6 R7 t M( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt
test_worldh s
c C s | | } | t | } t i | | } | t | } t i | | } | d | d | d d g | d | d | d d g | d | d | d d g d d d d g g } d d d | d g d d d | d g d d d | d g d d d d g g } t i | | S( Ni i i ( R1 R
t crosst dot( t Et Rt Vt nR R0 t Mrt Mt( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt view_transformationo s
c C sq | | | | } d | | | | } t i d d d d g d d d d g d d | | g d d d d g g S( Niţ˙˙˙i i i˙˙˙˙( R
R# ( t zfrontt zbackR R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt persp_transformation s c C sU t i | | } | d } | d | | d | | d | } } } | | | f S( Ni i i i ( R
R? ( t vecR< t vecwt wt txst tyst tzs( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt proj_transform_vec s
,c C sś t i | | } | d } | d | | d | | d | } } } | d d j | d d j | d d j | d d j } t i | o | d d j } n | | | | f S( Ni i i i ( R
R? t sometrue( RJ R< RK RL RM RN RO t tis( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt proj_transform_vec_clip s
,:c C st t i | } t | | | } t i | | } y | | d } Wn t j
o n X| d | d | d f S( Ni i i i ( t linalgt invt vec_pad_onesR
R? t
OverflowError( R) R* t zsR< t iMRJ t vecr( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt
inv_transformŠ s c C sŞ yu y+ t i | | | t i | i g } WnC t t f j
o1 t i | | | t i t | g } n XWn. t j
o" t i | | | d g } n X| S( Ni ( R
R# t onest shapet AttributeErrort TypeErrort len( R) R* RX RJ ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyRV ł s +3 c C s t | | | } t | | S( s7
Transform the points by the projection matrix
( RV RP ( R) R* RX R< RJ ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt proj_transform˝ s c C s t | | | } t | | S( sv
Transform the points by the projection matrix
and return the clipping result
returns txs,tys,tzs,tis
( RV RS ( R) R* RX R< RJ ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt proj_transform_clipÄ s c C s t t | | S( N( R! t proj_trans_points( R+ R< ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt proj_pointsÎ s c C s( t | \ } } } t | | | | S( N( R! Ra ( R+ R< R) R* RX ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyRc Ń s c C s( t | \ } } } t | | | | S( N( R! Rb ( R+ R< R) R* RX ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt proj_trans_clip_pointsŐ s i c C sG d d k } d | d d g d d | d g d d d | g } } } t | | | | \ } } } | d | d f | d | d f | d | d f | d | d f f \ } }
} } | |
f | | f | | f g }
| i }
t |
} |
i | xB t | | d d d d g D]" \ } } } | i | | | q Wd S(
Ni˙˙˙˙i i i i t oR- R. t z( R Ra R R t add_collectionR! t text( R< t sR R) R* RX RM RN RO Rf R( t ayt azt linest linecR- R. t t( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt test_proj_draw_axesŮ s 8 "4!
c C s | p t i d d d g d } t i d d d g d } t i d d d g } t | | | } t d d } t i | | } | S( Ni i˙˙˙˙i ič id i i˙˙˙( R
R# RF RI R? ( R@ RA RB t viewMt perspMR< ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt test_proj_make_Mç s # c C sĺ d d k } t } g } d d d d d d d d d d g
D] } | d
| q> ~ } d d d d d d d d d d g
d d d d d d d d d d g
d d d d d d d d d d g
} } } g } | | | f D] } | t i | d qÚ ~ \ } } } t | d d
t | | | | \ }
} } t |
| | | \ }
} } | i |
| d | | i |
| d d x6 t |
| | D]" \ } } } | i
| | | qW| i d d | i d d | i
d S( Ni˙˙˙˙i i i i i i i i s %di, Rj i R
t rgÉżgÉ?( R Rs R
R# Rp Ra R[ R" R R! Ri R% R& R' ( R R< t _[1]t it tsR) R* RX t _[2]R0 RM RN RO t ixst iyst izsR- R. Ro ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt test_projň s" CB,@ c C s{ t i | t i | } } t i d d d d g d | | d g d | | d g d d d d g g } t i | | S( Ni i ( R
t cost sinR# R? ( RB t alphat cosat sinat M1( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt rot_x s c C sR d d d d g } t | t i d GHd d d d g } t | t i d GHd S( Ni i i ( R R
t pi( RB ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt test_rot s t __main__(# t __doc__t matplotlib.collectionsR t matplotlib.patchesR t numpyR
t numpy.linalgRT R R R R/ R1 R; R= RF RI RP RS R[ RV Ra Rb t transformRd Rc Re Rp t NoneRs R| R R
t __name__( ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/proj3d.pyt s<