Ñò
‘ Jc @ sü d Z d d k l Z l Z l Z l Z d d k l
Z
l Z l Z l
Z
d d k l Z d d k l Z d d k l Z d d k l Z d d k Z d d k Z d d k Z d „ Z d
„ Z d „ Z d e i f d
„ ƒ YZ d d d „ Z d e i f d „ ƒ YZ! d d d „ Z" d d d „ Z# d d d „ Z$ d e f d „ ƒ YZ% d d d „ Z& d e f d „ ƒ YZ' d „ Z( d d d „ Z) d e
f d „ ƒ YZ* d d d „ Z+ d e f d! „ ƒ YZ, d d d" „ Z- d# „ Z. d$ „ Z/ d% „ Z0 d& „ Z1 d S(' sx
Module containing 3D artist code and functions to convert 2D
artists into 3D versions which can be added to an Axes3D.
iÿÿÿÿ( t linest textt patht colors( t
Collectiont LineCollectiont PolyCollectiont PatchCollection( t ScalarMappable( t Patch( t Normalize( t iterableNc C s- | d d } | d j o | d } n | S( s" Return angle between -180 and +180ih i´ ( ( t a( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt
norm_angle s
c C s- | d d } | d j o | d } n | S( s Return angle between -90 and +90i´ iZ ( ( R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt norm_text_angle s
c C s¡ | d j o t i d ƒ S| d j o t i d ƒ S| d j o t i d
ƒ S| d j o t i d ƒ St | ƒ o t | ƒ d j o | St d ƒ ‚ d S( Nt xi i t yt zi s2 'x', 'y', 'z', None or vector of length 3 expected( i i i ( i i i ( i i i ( i i i ( t npt arrayt NoneR t lent
ValueError( t zdir( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt get_dir_vector$ s
t Text3Dc B s> e Z d Z d d d d d d „ Z d d d „ Z d „ Z RS( sE
Text object with 3D position and (in the future) direction.
i t R c C s- t i i | | | | ƒ | i | | ƒ d S( N( t mtextt Textt __init__t set_3d_properties( t selfR R R R R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR 7 s c C s@ | i ƒ \ } } t i | | | f ƒ | _ t | ƒ | _ d S( N( t get_positionR R t _position3dR t _dir_vec( R R R R R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR ; s c C sè t i | i | i | i g | i ƒ } | d d | d d } | d d | d d } | d j o | d j o
d } n t i t i | | ƒ ƒ } | i | d d | d d f ƒ | i t
| ƒ ƒ t i i
| | ƒ d S( Ni i g ( t proj3dt proj_trans_pointsR! R" t Mt matht degreest atan2t set_positiont set_rotationR R R t draw( R t renderert projt dxt dyt angle( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR+ @ s
#( t __name__t
__module__t __doc__R R R+ ( ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR 2 s i R c C s t | _ | i | | ƒ d S( s" Convert a Text to a Text3D object.N( R t __class__R ( t objR R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt
text_2d_to_3dN s t Line3Dc B s/ e Z d Z d „ Z d d d „ Z d „ Z RS( s
3D line object.
c O s2 t i i | g g | | Ž | | | f | _ d S( N( R t Line2DR t _verts3d( R t xst yst zst argst kwargs( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR X s i R c
C so | i ƒ } | i ƒ } y1 t | ƒ } g } | D]
} | | q2 ~ } Wn n Xt | | | | ƒ | _ d S( N( t get_xdatat get_ydatat floatt juggle_axesR9 ( R R< R R: R; t _[1]R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR \ s %c C s] | i \ } } } t i | | | | i ƒ \ } } } | i | | ƒ t i i | | ƒ d S( N( R9 R# t proj_transformR% t set_dataR R8 R+ ( R R, t xs3dt ys3dt zs3dR: R; R< ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR+ g s $( R1 R2 R3 R R R+ ( ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR7 S s c C s t | _ | i | | ƒ d S( s"
Convert a 2D line to 3D.
N( R7 R4 R ( t lineR< R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt
line_2d_to_3dm s c
C sÇ t | ƒ p t i t | ƒ ƒ | } n g } | i d t d t ƒ } x? t | | ƒ D]. \ \ \ } } } } | i | | | f ƒ qX Wg } | D]% \ } } } | t | | | | ƒ q• ~ }
|
S( s Convert a path to a 3D segment.t simplifyt curves( R R t onesR t
iter_segmentst Falset zipt appendRB ( R R< R t segt pathsegsR R t codeR RC t seg3d( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt path_to_3d_segmentt s
9c C sm t | ƒ p t i t | ƒ ƒ | } n g } x6 t | | ƒ D]% \ } } | i t | | | ƒ ƒ q@ W| S( s@
Convert paths from a collection object to 3D segments.
( R R RM R RP RQ RV ( t pathsR< R t segmentsR t pathz( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt paths_to_3d_segments s
t Line3DCollectionc B s5 e Z d Z d „ Z d „ Z d „ Z e d „ Z RS( s#
A collection of 3D lines.
c O s t i | | | | Ž d S( N( R R ( R RX R= R> ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR “ s c C s | | _ t i | g ƒ d S( s!
Set 3D segments
N( t _segments3dR t set_segments( R RX ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR] – s c C s¯ g } | i D] } | t i | | i ƒ q ~ } g } | D] \ } } } | t | | ƒ q> ~ } t i | | ƒ d }
x, | D]$ \ } } } t |
t | ƒ ƒ }
qƒ W|
S( sB
Project the points according to renderer matrix.
g eÍÍA( R\ R# R$ R% RP R R] t min( R R, RC t pointst xyslistt _[2]R: R; R< t segments_2dt minz( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt do_3d_projection s ,3 c C s, | o | i | ƒ n t i | | ƒ d S( N( Rd R R+ ( R R, t project( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR+ ¬ s ( R1 R2 R3 R R] Rd RO R+ ( ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR[ Ž s
c C s2 t | i ƒ | | ƒ } t | _ | i | ƒ d S( s6 Convert a LineCollection to a Line3DCollection object.N( RZ t get_pathsR[ R4 R] ( t colR< R t
segments3d( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt line_collection_2d_to_3d± s t Patch3Dc B sJ e Z d Z d „ Z d d d „ Z d „ Z d „ Z d „ Z d „ Z RS( s
3D patch object.
c O sK | i d g ƒ } | i d d ƒ } t i | | | Ž | i | | ƒ d S( NR< R R ( t popR R R ( R R= R> R< R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR ¼ s i R c C sˆ t | ƒ p t i t | ƒ ƒ | } n g } t | | ƒ D]( \ \ } } } | t | | | | ƒ q> ~ | _ t i | ƒ | _ d S( N(
R R RM R RP RB t
_segment3dR t
get_facecolort _facecolor3d( R t vertsR< R RC R R R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR Â s
Ac C s | i S( N( t _path2d( R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt get_pathÊ s c C s | i S( N( t _facecolor2d( R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyRm Í s c
C sv | i } t | Œ \ } } } t i | | | | i ƒ \ } } } } t i t | | ƒ ƒ | _ | i | _ t
| ƒ S( N( Rl RP R# t proj_transform_clipR% t mpatht PathRp Rn Rr R^ (
R R, t sR: R; R< t vxst vyst vzst vis( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyRd Ð s ' c C s t i | | ƒ d S( N( R R+ ( R R, ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR+ Ù s ( R1 R2 R3 R R Rq Rm Rd R+ ( ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyRj · s c C sE | i ƒ } | i ƒ } | i | ƒ } t | ƒ o | d Sg Sd S( s2 Return a list of vertices for the path of a patch.i N( t get_patch_transformRq t to_polygonsR ( t patcht transR t polygons( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt get_patch_vertsÜ s
c C s, t | ƒ } t | _ | i | | | ƒ d S( s$ Convert a Patch to a Patch3D object.N( R€ Rj R4 R ( R} R R Ro ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt patch_2d_to_3dæ s t Patch3DCollectionc B s2 e Z d Z d „ Z d „ Z d „ Z d „ Z RS( s%
A collection of 3D patches.
c O s t i | | | Ž d S( N( R R ( R R= R> ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR ñ s c C sR t | i ƒ Œ \ } } t | | | | ƒ | _ | i ƒ | _ | i ƒ | _ d S( N( RP t get_offsetsRB t
_offsets3dRm Rn t
get_edgecolort _edgecolor3d( R R< R R: R; ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR ô s c C s— | i \ } } } t i | | | | i ƒ \ } } } } d | _ | i t | i | ƒ ƒ | i t | i
| ƒ ƒ t i | t
| | ƒ ƒ t | ƒ S( N( R„ R# Rs R% R t _alphat set_facecolorst zalphaRn t set_edgecolorsR† R t set_offsetsRP R^ ( R R, R: R; R< Rw Rx Ry Rz ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyRd ú s ' c C s t i | | ƒ d S( N( R R+ ( R R, ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR+ s ( R1 R2 R3 R R Rd R+ ( ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR‚ ì s
c C s t | _ | i | | ƒ d S( s8 Convert a PatchCollection to a Patch3DCollection object.N( R‚ R4 R ( Rg R< R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt patch_collection_2d_to_3d s t Poly3DCollectionc B sŒ e Z d Z d „ Z d „ Z e d „ Z d „ Z d „ Z d „ Z d „ Z
e
Z d „ Z e Z
d „ Z e Z d
„ Z e Z d „ Z RS( s&
A collection of 3D polygons.
c O s, t i | | | | Ž d | _ d | _ d S( sÇ
Create a Poly3DCollection.
*verts* should contain 3D coordinates.
Note that this class does a bit of magic with the _facecolors
and _edgecolors properties.
i N( R R t _zsortR t
_sort_zpos( R Ro R= R> ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR s
c C s´ d } d } g } g } xD | D]< } | i | ƒ | t | ƒ } | i | | f ƒ | } q Wt | Œ \ } } } t i t | ƒ ƒ }
t i | | | |
g ƒ | _ | | _ d S( s Optimize points for projectioni N( t extendR RQ RP R RM R t _vect _segis( R Rh t sit eit segisR_ t pR: R; R< RM ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt
get_vector s
c C s$ | i | ƒ t i | g | ƒ d S( s Set 3D vertices.N( R— R t set_verts( R Ro t closed( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR˜ 0 s
c C s: d | _ d | _ t i | ƒ | _ t i | ƒ | _ d S( Ni ( RŽ R R R t get_facecolorst
_facecolors3dt get_edgecolorst
_edgecolors3d( R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR 6 s c C s
| | _ d S( N( R ( R t val( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt
set_sort_zpos< s c C sï | i d j o | i ƒ | i | _ n t i | i | i ƒ \ } } } g } | i D]1 \ } } | | | | !| | | !| | | !f qY ~ } | i } | i
}
t | ƒ t | ƒ j o | i t | ƒ d d ƒ} n t |
ƒ t | ƒ j o< t |
ƒ d j o
| }
n |
i t | ƒ d d ƒ}
n | i
ot g } t | | |
ƒ D]= \ \ } }
} } } | t i | ƒ t | |
ƒ | | f qS~ } | i d d „ ƒ n
t d ‚ g } | D] \ } } } } | | qÄ~ } t i | | ƒ g } | D] \ } } } } | | q~ | _ t | i
ƒ t | ƒ j o4 g } | D] \ } } } } | | qM~ | _ n
| i
| _ | i d j oL t i d g d g | i g d g g ƒ } t i | | i ƒ } | d d St i | ƒ Sd S( s<
Perform the 3D projection for this object.
t axisi t cmpc S s t | d | d ƒ S( i ( R¡ ( R R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt Z s t whoopsi i N( t _AR t update_scalarmappablet _facecolorsR› R# t proj_transform_vecR‘ R% R’ R R t repeatRŽ RP R t averaget sortR R R˜ t
_facecolors2dt
_edgecolors2dR R R^ ( R R, t txst tyst tzsRC R“ R” t xyzlistt cfacet cedgeRa R: R; R< t fct ect
z_segments_2dt _[3]R Rv Rb t _[4]t _[5]t zvect ztrans( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyRd ? s<
!A
V -0 4 *
c C s& t i | | ƒ t i | ƒ | _ d S( N( R t
set_facecolorRm R› ( R R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR» o s c C s& t i | | ƒ t i | ƒ | _ d S( N( R t
set_edgecolorR… R ( R R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR¼ t s c C s | i S( N( R« ( R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyRš y s c C s | i S( N( R¬ ( R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyRœ } s c C s t i | | ƒ S( N( R R+ ( R R, ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR+ s ( R1 R2 R3 R R— t TrueR˜ R RŸ Rd R» Rˆ R¼ RŠ Rš Rm Rœ R… R+ ( ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR
s 0 c C s< t | i ƒ | | ƒ } t | _ | i | ƒ | i ƒ d S( s6 Convert a PolyCollection to a Poly3DCollection object.N( RZ Rf R R4 R˜ R ( Rg R< R t segments_3d( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt poly_collection_2d_to_3d„ s
c C sG | d j o | | | f S| d j o | | | f S| | | f Sd S( s*
Reorder coordinates so that zdir
R R N( ( R: R; R< R ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyRB ‹ s
c C sV y; t | ƒ d j p t | ƒ d j o t | d d ƒ SWn t j
o t SXd S( Ni i i t __float__( R t hasattrt
IndexErrorRO ( t c( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt iscolor– s ;c C s« t | ƒ t d ƒ j o t i i | ƒ } n t | ƒ o | g | St | ƒ | j o | St | ƒ o | g | St | d ƒ o | d g | St d | ‚ d S( s= Stretch the color argument to provide the required number numt stringi s unknown color format %sN( t typet mcolorst colorConvertert to_rgbaRÄ R R ( RÃ t num( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt
get_colorsœ s
c C s˜ t | t | ƒ ƒ } t t | ƒ t | ƒ ƒ } d | | ƒ d } g } t | | ƒ D]3 \ } } | | d | d | d | d | f qX ~ } | S( s6 Modify the alphas of the color list according to depthi gffffffæ?i i i ( RË R R
R^ t maxRP ( R R< t normt satsRC RÃ Rv ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyR‰ s
P(2 R3 t
matplotlibR R R R Rt R RÇ t matplotlib.collectionsR R R R t
matplotlib.cmR t matplotlib.patchesR t matplotlib.colorsR
t matplotlib.cbookR t numpyR R& R# R
R R R R R6 R8 R7 RJ RV RZ R[ Ri Rj R€ R R‚ RŒ R R¿ RB RÄ RË R‰ ( ( ( s@ /usr/lib64/python2.6/site-packages/mpl_toolkits/mplot3d/art3d.pyt s> ""
#%
w