Jc @ s d Z d d k l Z l Z d d k l Z d d k l Z d d k l Z d d k Z
d d k l Z l
Z
d d k Z d d k Z d d k Z d Z d Z d
e f d YZ d d
Z d S( sZ
Module containing Axes3D, an object which can plot 3D objects on a
2D matplotlib figure.
i( t Axest rcParams( t cbook( t Bbox( t collectionsN( t Normalizet colorConverterc C sJ t | d j p t | d j o d | } | i | Sd | Sd S( s: Used to generate more comprehensible numbers in status barg @gMbP?s %1.4es %4.3fN( t abst _formatSciNotation( t selft valuet s( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt sensible_format_data s &
c C s+ t t i d d g d d g g } | S( Ni i ( R t npt array( t box( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt unit_bbox s 't Axes3Dc B s? e Z d Z d8 d Z d Z d Z d8 d Z d8 d8 d Z d8 d8 d Z d Z
d Z d Z d8 d8 d
Z
e e e d Z d Z d8 d8 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 d Z d Z! d Z" d Z# d Z$ d! Z% d8 d" Z& d8 d# Z' d8 d$ Z( e d% Z) d8 d& Z* e* Z+ d' Z, e, Z- d( Z. d) Z/ d* Z0 d+ Z1 d, d- Z2 d. d/ Z3 e3 Z4 d0 Z5 e5 Z6 d1 d2 d3 Z7 d1 d2 d4 Z8 e8 Z9 d1 d2 d5 Z: d6 d7 Z; RS(9 s
3D axes object.
c O s$ | d j o d d d d g } n | | _ g | _ | i d d } | i d d } t | _ t | _ t | _ t | _ | i | | d | _
t i | | i | d t
d g d
g | | d | _ d | _
| i | i | i | i i d | i i | d S( Ng g ?t azimit elevi i t frameont xtickst yticksi ( t Nonet figt cidst popR t
xy_viewLimt
zz_viewLimt
xy_dataLimt
zz_dataLimt view_initt _readyR t __init__t Truet Mt
mouse_initt create_axest set_top_viewt axesPatcht
set_linewidtht add_axes( R R t rectt argst kwargsR R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR! ' s,
c C s` d | i } d | i } d | i } d | i } t i | | | t i | | | d S( Ngffffff?g?( t distR t set_xlimt set_ylim( R t xdwlt xdwt ydwlt ydw( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR& F s
c C sy t i d | i i | i i | | _ t i d | i i | i i | | _ t i d | i
i | i i | | _ d S( Nt xt yt z(
t axis3dt XAxisR t intervalxR t w_xaxist YAxist intervalyt w_yaxist ZAxisR R t w_zaxis( R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR% Q s c
C s | p
| i \ } } } } } } | | | | | | | | g | | | | | | | | g | | | | | | | | g } } }
t | | |
S( N( t
get_w_limst zip( R t valst minxt maxxt minyt maxyt minzt maxzt xst yst zs( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt unit_cubeY s
%&c C s? | d j o
| i } n | i | } t i | | } | S( N( R R# RL t proj3dt proj_points( R RB R# t xyzst tcube( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
tunit_cube` s
c C s | i | | } | d | d f | d | d f | d | d f | d | d f | d | d f | d | d f | d | d f | d | d f | d | d f | d | d f | d | d f | d | d f g } | S( Ni i i i i i i i ( RQ ( R RB R# t tct edges( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt tunit_edgesg s c C s | i i | t | _ | i | _ | i | _ | i | _ | i | _ | i | _ g } | i D] } | | i
| | f qf ~ } | i | i x) t
| D] \ } \ } } | | _ q Wg } | i D] } | | i
| | f q ~ } | i | i x) t
| D] \ } \ } } | | _ q W| i i | | i i | | i i | t i | | d S( N( R' t drawt Falset _frameont get_projR# t vvect eyet get_axis_positionR t do_3d_projectiont sortt reverset enumeratet zordert patchesR: R= R? R ( R t renderert _[1]t colt zlistt iR6 t _[2]t patch( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyRU y s2 ,
,
c C s | i } | i | | i } | d d | d d j } | d d | d d j } | d d | d d j } | | | f S( Ni i i i ( R@ RQ R# ( R RB RR t xhight yhight zhigh( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR[ s c K s d S( N( ( R t xysR, ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt update_datalim s c C s t t i | | | f \ } } } y: | i | i } } | d j o | i } n Wn t j
o
n X| i i t i | | g i | | d j o* | i
i t i | | g i | n | i d S( N( t mapR
t asarrayt flattenR t AttributeErrorR t update_from_data_xyR t TR t autoscale_view( R t Xt Yt Zt had_dataR4 R5 R6 ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt auto_scale_xyz s $
&
*c C s | i | i p d S| i p d S| o | i | i i n | o | i | i i n | o | i | i i n d S( N(
R& R t get_autoscale_ont
set_xlim3dR R9 t
set_ylim3dR< t
set_zlim3dR ( R t scalext scaleyt scalez( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyRt s
c C sL | i \ } } | i \ } } | i \ } } | | | | | | f S( s Get 3d world limits.( t
get_xlim3dt
get_ylim3dt
get_zlim3d( R RC RD RE RF RG RH ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR@ s c O s\ | d j o t i | o | \ } } n | | j o | d 8} | d 7} n | | f S( Ng ?( R R t iterable( R t xmint xmaxR+ R, ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt _determine_lims s
c O s" | i | | } | | i _ | S( s Set 3D x limits.( R R R9 ( R R+ R, t lims( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR{ s c O s" | i | | } | | i _ | S( s Set 3D y limits.( R R R< ( R R+ R, R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR| s c O s" | i | | } | | i _ | S( s Set 3D z limits.( R R R9 ( R R+ R, R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR} s c C s
| i i S( s Get 3D x limits.( R R9 ( R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR s c C s
| i i S( s Get 3D y limits.( R R< ( R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR s c C s
| i i S( s Get 3D z limits.( R R9 ( R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR s c O s d S( N( R ( R R+ R, ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt clabel s c C s
d G| GHd S( Nt numsteps( ( R R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt pany s c C s
d G| GHd S( NR ( ( R R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt panpy s c C s d | _ | | _ | | _ d S( Ni
( R- R R ( R R R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR s c C s t i | i d t i | i d } } | i \ } } | i \ } } | i \ } } t i | | | | | | } t i d d d g }
|
d t i
| t i
| | i } |
d t i | t i
| | i } |
d t i | | i }
t i | | |
f } | | _
|
| | _ | i t i | i | _ t | t i d j o t i d } n t i d } | i | i } } t i | |
| } t i | | } t i | | } t i | | } | S( s Create the projection matrix from the current viewing
position.
elev stores the elevation angle in the z plane
azim stores the azimuth angle in the x,y plane
dist is the distance of the eye viewing point from the object
point.
i g ?i i i i( i i i( i i i ( R
t piR R R R R RM t world_transformationR t cosR- t sinRZ RY t modR t view_transformationt persp_transformationt dot( R t relevt razimR R t ymint ymaxt zmint zmaxt worldMt Rt xpt ypt zpt Et Vt zfrontt zbackt viewMt perspMt M0R# ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyRX s0 ) ++
c C s{ d | _ | i i } | d j oU | i d | i } | i d | i } | i d | i } | | | g | _ n d S( Nt motion_notify_eventt button_press_eventt button_release_event( R t button_pressedt figuret canvast mpl_connectt _on_movet
_button_presst _button_releaseR ( R t canvt c1t c2t c3( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR$ 3 s
c C sR x$ | i D] } | i i i | q
Wg | _ t i | | i t d d S( Ns axes3d.grid( R R R t mpl_disconnectR t clat gridR ( R t cid( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR < s
c C s) | i | _ | i | i | _ | _ d S( N( t buttonR t xdatat ydatat sxt sy( R t event( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR D s c C s
d | _ d S( N( R R ( R R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR H s c C sD y | i | SWn, t j
o | i i } t | | SXd S( s
Return x string formatted. This function will use the attribute
self.fmt_xdata if it is callable, else will fall back on the xaxis
major formatter
N( t fmt_xdatat TypeErrorR: t get_major_formatterR ( R R4 t fmt( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt format_xdataK s
c C sD y | i | SWn, t j
o | i i } t | | SXd S( s
Return y string formatted. This function will use the attribute
self.fmt_ydata if it is callable, else will fall back on the yaxis
major formatter
N( t fmt_ydataR R= R R ( R R5 R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt format_ydataW s
c C sJ y | i | SWn2 t t f j
o | i i } t | | SXd S( s
Return z string formatted. This function will use the attribute
self.fmt_zdata if it is callable, else will fall back on the yaxis
major formatter
N( t fmt_zdataRq R R? R R ( R R6 R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt format_zdatac s
c C s | i d j o d S| i d j o d | i | i f S| | f } | i } g } t | D]. \ } \ } } | t i | | | | f qc ~ } | i | d d }
| |
\ } } | \ } } }
| \ } } } t
i | | | | } t
i | | | | } | | } | | |
| | | } t i | | | | i \ } } } | i
| } | i | } | i | } d | | | f S( s
Given the 2D view coordinates attempt to guess a 3D coordinate.
Looks for the nearest edge to the point and then assumes that
the point is at the same z location as the nearest point on the edge.
t i s! azimuth=%d deg, elevation=%d deg i s x=%s, y=%s, z=%sN( R# R R R R RT R_ RM t line2d_seg_distR] R
t hypott
inv_transformR R ( R t xdt ydt pRS Rc Rf t p0t p1t ldistst edgeit x0t y0t z0t x1t y1t z1t d0t d1t dtR6 R4 R5 RI RJ RK ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt format_coordo s, A
$c C s& | i p d S| i d j o d S| i | i } } | d j o d S| | i | | i } } | i \ } } | i \ } } | | }
| | } | | | _ | _ | i d j o | d j o | d j o d St i
| i | | d | _ t i
| i | |
d | _ | i
| i i i n | i d j o n | i d j o | i \ } }
} } } } d | | | } |
| | } | | | } | | | } | i | | |
| | i | | | | | i | | | | | i
| i i i n d S( sF Mouse moving
button-1 rotates
button-3 zooms
Ni i i i i ( R R# R R R R R t get_xlimt get_ylimt art3dt
norm_angleR R RX R R RU R@ R{ R| R} ( R R R4 R5 t dxt dyR R R R t wt hRC RD RE RF RG RH t dft dz( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR sB
!!
c K sK | i i } | i | | d j o | i | n | i | | S( s Set xlabel. N( R: t get_labelt set_textR t update( R t xlabelt fontdictR, t label( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
set_xlabel s
c K sK | i i } | i | | d j o | i | n | i | | S( s Set ylabel.N( R= R R R R ( R t ylabelR R, R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
set_ylabel s
c K sK | i i } | i | | d j o | i | n | i | | S( s Set zlabel.N( R? R R R R ( R t zlabelR R, R ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
set_zlabel s
c K s
| | _ d S( s&
Set / unset 3D grid.
N( t
_draw_grid( R t onR, ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR s c C s/ t i | | | | } t i | | | | S( s Add text to the plot.( R t textR t
text_2d_to_3d( R R4 R5 R6 R t zdirR ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR s c O sy | i } | i d d } | i d d } d } t | d j oa t i | d oM t | t | d j o0 t i | d d o | | } | d 7} n@ t | d j o, t i | d o | | } | d 7} n t i | p t i t | | } n t i | | | | | | } x' | D] }
t
i |
d | d | q<W| i | | | | | S( so
Plot 2D or 3D data.
========== ================================================
Argument Description
========== ================================================
*xs*, *ys* X, y coordinates of vertices
*zs* z value(s), either one for all points or one for
each point.
*zdir* Which direction to use as z ('x', 'y' or 'z')
when plotting a 2d set.
========== ================================================
Other arguments are passed on to
:func:`~matplotlib.axes.Axes.plot`
RK i R R6 i (
t has_dataR t lenR R t is_scalarR
t onesR t plotR t
line_2d_to_3dRy ( R RI RJ R+ R, Rx RK R t argsit linest line( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyR s& '5
'
c & O s | i } | i \ } } t i | t i | t i | } }
} | i d d } | i d d }
| i d d } t i t i | } | i d d } g } g } g } xt i
d | d | D]} x{t i
d | d |
D]`} g } g } x\| | f | |
f | | f g D]9\ } } | | | t | | |
d !} | t | d | |
| t | | | d !} | t | d | | | t | | |
d d } | d d d
} | | | t | | | d d } | d d d
} | i | d | d
| d | d
g t i
| | | | f } | i | q*Wt | } t i g } g } d } x@ | D]8 }! |! | j o% | i |! |! } | |! d 7} qqW| i | | i | t | t i | d t i | d }" t i | d t i | d }# | i t i |" |# q Wq Wt i | | | }$ | d j o' |$ i t i | |$ i d n | i | | }% |$ i |% | i |$ | i | | | | |$ S(
s
Create a surface plot.
By default it will be colored in shades of a solid color,
but it also supports color mapping by supplying the *cmap*
argument.
========== ================================================
Argument Description
========== ================================================
*X*, *Y*, Data values as numpy.arrays
*Z*
*rstride* Array row stride (step size)
*cstride* Array column stride (step size)
*color* Color of the surface patches
*cmap* A colormap for the surface patches.
========== ================================================
t rstridei
t cstridet colort bt cmapi i Nig i ( R t shapeR
t transposeR R R t to_rgbat getR t aranget mint appendt concatenateRA R t crossR t Poly3DCollectiont set_arrayR( t
_shade_colorst set_facecolorst add_collectionRy (& R Ru Rv Rw R+ R, Rx t rowst colst tXt tYt tZR R R R t polyst normalst avgzt rst cst pst cornerst at tat ztopt zleftt zbaset zrightR6 t lastpt ps2t avgzsumR t v1t v2t polyct colors( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt plot_surface' sh / " "39()
$$!
c C s| g } xo | D]g } t i | d t i | d } t i | d t i | d } | i t i | | q
W| S( s
Generate normals for polygons by using the first three points.
This normal of course might not make sense for polygons with
more than three points not lying in a plane.
i i i ( R
R R R
( R t polygonsR t vertsR( R) ( ( sA /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/axes3d.pyt _generate_normalsw s $$ c
C s g } xG | D]? } | t i | d } | i t i | d d d g q
Wt i | } t i | } t | | d j op t t | | t
| | } | i } d | d s