|
Server : Apache System : Linux cvar2.toservers.com 3.10.0-962.3.2.lve1.5.73.el7.x86_64 #1 SMP Wed Aug 24 21:31:23 UTC 2022 x86_64 User : njnconst ( 1116) PHP Version : 8.4.18 Disable Function : NONE Directory : /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/ |
Upload File : |
�
[�3Lc @ s6 d Z d d l m Z m Z m Z m Z d d l m
Z
m Z m Z m
Z
d d l m Z d d l m Z d d l m Z d d l m Z d d l Z d d l Z d d l Z d d l Z d � Z d
� Z d � Z d e j f d
� � YZ d d d � Z d e j! 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 e( f d � � YZ) 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 d d% � Z0 d& � Z1 d' � Z2 d( � Z3 d) � Z4 d* � Z5 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 k r' | d } n | S( s" Return angle between -180 and +180ih i� ( ( t a( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt
norm_angle s
c C s+ | d d } | d k r' | d } n | S( s Return angle between -90 and +90i� iZ ( ( R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt norm_text_angle s
c C s� | d k r t j d � S| d k r2 t j d � S| d k rK t j d
� S| d k rd t j d � St | � r� t | � d k r� | 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( ( sM /opt/alt/python27/lib64/python2.7/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 K s0 t j j | | | | | � | j | | � d S( s�
*x*, *y*, *z* Position of text
*text* Text string to display
*zdir* Direction of text
Keyword arguments are passed onto :func:`~matplotlib.text.Text`.
N( t mtextt Textt __init__t set_3d_properties( t selfR R R R R t kwargs( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR 9 s c C s@ | j � \ } } t j | | | f � | _ t | � | _ d S( N( t get_positionR R t _position3dR t _dir_vec( R R R R R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR D s c C s� t j | j | j | j g | j � } | d d | d d } | d d | d d } | d k r} | d k r} d } n t j t j | | � � } | j | d d | d d f � | j t
| � � t j j
| | � 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( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR, I s #( t __name__t
__module__t __doc__R R R, ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR 4 s i R c C s t | _ | j | | � d S( s" Convert a Text to a Text3D object.N( R t __class__R ( t objR R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt
text_2d_to_3dW 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 j j | g g | | � | | | f | _ d S( sU
Keyword arguments are passed onto :func:`~matplotlib.lines.Line2D`.
N( R t Line2DR t _verts3d( R t xst yst zst argsR ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR a s i R c C sg | j � } | j � } y) t | � } g | D] } | ^ q. } Wn n Xt | | | | � | _ d S( N( t get_xdatat get_ydatat floatt juggle_axesR: ( R R= R R; R< R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR h s c C s] | j \ } } } t j | | | | j � \ } } } | j | | � t j j | | � d S( N( R: R$ t proj_transformR& t set_dataR R9 R, ( R R- t xs3dt ys3dt zs3dR; R< R= ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR, s s $( R2 R3 R4 R R R, ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR8 \ s c C s t | _ | j | | � d S( s"
Convert a 2D line to 3D.
N( R8 R5 R ( t lineR= R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt
line_2d_to_3dy s c
C s� t | � s( t j t | � � | } n g } | j d t d t � } x? t | | � D]. \ \ \ } } } } | j | | | f � qV 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 t seg3d( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt path_to_3d_segment� s (1c C sk t | � s( t j t | � � | } n g } x6 t | | � D]% \ } } | j t | | | � � q> W| S( s@
Convert paths from a collection object to 3D segments.
( R R RL R RO RP RU ( t pathsR= R t segmentsR t pathz( ( sM /opt/alt/python27/lib64/python2.7/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 j | | | | � d S( sc
Keyword arguments are passed onto :func:`~matplotlib.collections.LineCollection`.
N( R R ( R RW R> R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR � s c C s | | _ t j | g � d S( s!
Set 3D segments
N( t _segments3dR t set_segments( R RW ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR\ � s c C s� g | j D] } t j | | j � ^ q
} g | D] \ } } } t | | � ^ q2 } t j | | � d } x, | D]$ \ } } } t | t | � � } qs W| S( sB
Project the points according to renderer matrix.
g e��A( R[ R$ R% R& RO R R\ t min( R R- t pointst xyslistR; R< R= t segments_2dt minz( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt do_3d_projection� s (+c C s* | r | j | � n t j | | � d S( N( Rb R R, ( R R- t project( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR, � s ( R2 R3 R4 R R\ Rb RN R, ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRZ � s
c C s2 t | j � | | � } t | _ | j | � d S( s6 Convert a LineCollection to a Line3DCollection object.N( RY t get_pathsRZ R5 R\ ( t colR= R t
segments3d( ( sM /opt/alt/python27/lib64/python2.7/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 | j d g � } | j d d � } t j | | | � | j | | � d S( NR= R R ( t popR R R ( R R> R R= R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR � s i R c C s~ t | � s( t j t | � � | } n g t | | � D]' \ \ } } } t | | | | � ^ q8 | _ t j | � | _ d S( N(
R R RL R RO RB t
_segment3dR t
get_facecolort _facecolor3d( R t vertsR= R R R R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR � s
=c C s | j S( N( t _path2d( R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt get_path� s c C s | j S( N( t _facecolor2d( R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRk � s c
C sv | j } t | � \ } } } t j | | | | j � \ } } } } t j t | | � � | _ | j | _ t
| � S( N( Rj RO R$ t proj_transform_clipR& t mpatht PathRn Rl Rp R] (
R R- t sR; R< R= t vxst vyst vzst vis( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRb � s ' c C s t j | | � d S( N( R R, ( R R- ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR, � s ( R2 R3 R4 R R Ro Rk Rb R, ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRh � s t PathPatch3Dc B s/ e Z d Z d � Z d d d � Z d � Z RS( s
3D PathPatch object.
c K sK | j d g � } | j d d � } t j | | � | j | | | � d S( NR= R R ( Ri R R R ( R R R R= R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR � s i R c C s/ t j | | j d | d | �| j | _ d S( NR= R ( Rh R t verticest codest _code3d( R R R= R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR � s c
C s| | j } t | � \ } } } t j | | | | j � \ } } } } t j t | | � | j � | _ | j | _
t | � S( N( Rj RO R$ Rq R& Rr Rs R| Rn Rl Rp R] (
R R- Rt R; R< R= Ru Rv Rw Rx ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRb � s '! ( R2 R3 R4 R R Rb ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRy � s c C sC | j � } | j � } | j | � } t | � r; | d Sg Sd S( s2 Return a list of vertices for the path of a patch.i N( t get_patch_transformRo t to_polygonsR ( t patcht transR t polygons( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt get_patch_verts s c C s, t | � } t | _ | j | | | � d S( s$ Convert a Patch to a Patch3D object.N( R� Rh R5 R ( R R R Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt patch_2d_to_3d s c C sG | j � } | j � } | j | � } t | _ | j | | | � d S( s, Convert a PathPatch to a PathPatch3D object.N( Ro R} t transform_pathRy R5 R ( t pathpatchR R R R� Rr ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt pathpatch_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 s) t j � | | � � f d � � _ d S( Nc s t j � | � S( N( R R, ( R ( R ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt <lambda>$ s ( R R t _old_draw( R R> R ( ( R sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR " s c C sR t | j � � \ } } t | | | | � | _ | j � | _ | j � | _ d S( N( RO t get_offsetsRB t
_offsets3dRk Rl t
get_edgecolort _edgecolor3d( R R= R R; R< ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR & s c C s� | j \ } } } t j | | | | j � \ } } } } d | _ | j t | j | � � | j t | j
| � � t j | t
| | � � t | � S( N( R� R$ Rq R& R t _alphat set_facecolorst zalphaRl t set_edgecolorsR� R t set_offsetsRO R] ( R R- R; R< R= Ru Rv Rw Rx ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRb , s ' c C s | j | � d S( N( R� ( R R- ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR, 7 s ( R2 R3 R4 R R Rb R, ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR� s
c C s) | j | _ t | _ | j | | � d S( s8 Convert a PatchCollection to a Patch3DCollection object.N( R, R� R� R5 R ( Re R= R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt patch_collection_2d_to_3d: s t Poly3DCollectionc B s� e Z d Z d � Z i e j d 6e j d 6e j d 6Z 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 s3 | j | j d t � � t j | | | | � d S( s
Create a Poly3DCollection.
*verts* should contain 3D coordinates.
Keyword arguments:
zsort, see set_zsort for options.
Note that this class does a bit of magic with the _facecolors
and _edgecolors properties.
t zsortN( t set_zsortRi t TrueR R ( R Rm R> R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR I s
t averageR] t maxc C sl | t k r d } n | t k rG | | j k r@ | j | } qM t Sn d } | | _ d | _ | | _ d S( s�
Set z-sorting behaviour:
boolean: if True use default 'average'
string: 'average', 'min' or 'max'
R� N( R� RN t _zsort_functionsR t _zsortt
_sort_zpost
_zsortfunc( R R� t zsortfunc( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR� ` s c C s� d } d } g } g } xD | D]< } | j | � | t | � } | j | | f � | } q Wt | � \ } } } t j t | � � }
t j | | | |
g � | _ | | _ d S( s Optimize points for projectioni N( t extendR RP RO R RL R t _vect _segis( R Rf t sit eit segisR^ t pR; R< R= RL ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt
get_vectorv s
c C s$ | j | � t j | g | � d S( s Set 3D vertices.N( R� R t set_verts( R Rm t closed( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR� � s
c C s> d | _ | j t � t j | � | _ t j | � | _ d S( N( R R� R� R� R t get_facecolorst
_facecolors3dt get_edgecolorst
_edgecolors3d( R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR � s
c C s
| | _ d S( s& Set the position to use for z-sorting.N( R� ( R t val( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt
set_sort_zpos� s c C s� | j d k r( | j � | j | _ n t j | j | j � \ } } } g | j D]0 \ } } | | | !| | | !| | | !f ^ qS } | j } | j
} t | � t | � k r� | j t | � d d �} n t | � t | � k r"t | � d k r| } n | j t | � d d �} n | j
r�g t | | | � D]< \ \ }
} } }
} | j | � t |
| � |
| f ^ q>} | j d d � � n t d � g | D] \ } } }
} | ^ q�} t j | | � g | D] \ } } }
} |
^ q�| _ t | j
� t | � k rBg | D] \ } } }
} | ^ q | _ n | j
| _ | j d k r�t j d g d g | j g d g g � } t j | | j � } | d d St j | � Sd S( s<
Perform the 3D projection for this object.
t axisi t cmpc S s t | d | d � S( Ni ( R� ( R R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR� � s t whoopsi i N( t _AR t update_scalarmappablet _facecolorsR� R$ t proj_transform_vecR� R& R� R� R t repeatR� RO R� t sortR R R� t
_facecolors2dt
_edgecolors2dR� R R R] ( R R- t txst tyst tzsR� R� t xyzlistt cfacet cedgeR; R< R= t fct ect
z_segments_2dR Rt R` t zvect ztrans( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRb � s<
!= R %(+ * c C s& t j | | � t j | � | _ d S( N( R t
set_facecolorRk R� ( R R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR� � s c C s& t j | | � t j | � | _ d S( N( R t
set_edgecolorR� R� ( R R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR� � s c C s | j S( N( R� ( R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR� � s c C s | j S( N( R� ( R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR� � s c C s t j | | � S( N( R R, ( R R- ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR, � s ( R2 R3 R4 R R R� R] R� R� R� R� R� R� R R� Rb R� R� R� R� R� Rk R� R� R, ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR� D s*
0 c C s< t | j � | | � } t | _ | j | � | j � d S( s6 Convert a PolyCollection to a Poly3DCollection object.N( RY Rd R� R5 R� R ( Re R= R t segments_3d( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt poly_collection_2d_to_3d� s
c C sf | d k r | | | f S| d k r2 | | | f S| d d k rU t | | | | � S| | | f Sd S( s�
Reorder coordinates so that 2D xs, ys can be plotted in the plane
orthogonal to zdir. zdir is normally x, y or z. However, if zdir
starts with a '-' it is interpreted as a compensation for rotate_axes.
R R i t -N( t rotate_axes( R; R< R= R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRB � s
c C su | d k r | | | f S| d k r2 | | | f S| d k rK | | | f S| d k rd | | | f S| | | f Sd S( s�
Reorder coordinates so that the axes are rotated with zdir along
the original z axis. Prepending the axis with a '-' does the
inverse transform, so zdir can be x, -x, y, -y, z or -z
R s -xR s -yN( ( R; R< R= R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR� � s
c C se yV t | � d k s' t | � d k rU t | d � r; t St | d d � rU t Sn Wn t SXt S( Ni i i t __float__( R R RN t hasattrR� ( t c( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt iscolor s $ c C s� t | � t d � k r- t j j | � } n t | � rD | g | St | � | k rZ | St | � rq | g | St | � d k r� d d d d g g | St | d � r� | 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( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt
get_colors s c C s� t | t | � � } t t | � t | � � } d | | � d } g t | | � D]2 \ } } | d | d | d | d | f ^ qT } | S( s6 Modify the alphas of the color list according to depthi gffffff�?i i i ( R� R R
R] R� RO ( R R= t normt satsR� Rt ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR� " s
H(6 R4 t
matplotlibR R R R Rr R R� t matplotlib.collectionsR R R R t
matplotlib.cmR t matplotlib.patchesR t matplotlib.colorsR
t matplotlib.cbookR t warningst numpyR R' R$ R
R R R R R7 R9 R8 RI RU RY RZ Rg Rh Ry R� R� R� R� R� R� R� RB R� R� R� R� ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt <module> sF "" #
&&
�