|
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 : /proc/self/root/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/ |
Upload File : |
�
/&XLc @ s� d Z d d l Z d d l m Z m Z d d l m Z d d l m Z d d l m Z d d l
Z d d l m
Z
m Z m Z d d l Z d d l Z d d l Z d � Z d � Z d
e f d � � YZ d d
� Z d d l j Z e j j e � d S( sZ
Module containing Axes3D, an object which can plot 3D objects on a
2D matplotlib figure.
i����N( t Axest rcParams( t cbook( t Bbox( t collections( t Normalizet colorConvertert LightSourcec C sG t | � d k s$ t | � d k r; d | } | j | � Sd | Sd S( s: Used to generate more comprehensible numbers in status barg ��@g����MbP?s %1.4es %4.3fN( t abst _formatSciNotation( t selft valuet s( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt sensible_format_data s $
c C s+ t t j d d g d d g g � � } | S( Ni i ( R t npt array( t box( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt unit_bbox s 't Axes3Dc B sx e Z d Z d Z d> d � Z d � Z d � Z d> d � Z d> d> d � Z d> d> d � Z
d � Z d � Z d
� Z
d> d> d � Z e e e d � Z d
� Z d> d> d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d> d> d � Z d � Z d d d � Z d � Z d � Z d � Z! d � Z" d! � Z# d" � Z$ d# � Z% d$ � Z& d% � Z' d> d& � Z( d> d' � Z) d> d( � Z* e d) � Z+ d> d* � Z, e, Z- e. j, Z/ d+ � Z0 e0 Z1 d, � Z2 d- � Z3 d. � Z4 d/ � Z5 d0 � Z6 d1 d2 � Z7 d3 � Z8 e8 Z9 d4 � Z: e: Z; d5 d6 d7 � Z< d5 d6 d8 � Z= e= Z> d5 d6 d9 � Z? d: d; d< � Z@ d> d= � ZA RS(? s
3D axes object.
t 3dc O s. | d k r! d d d d g } n | | _ g | _ | j d d � | _ | j d d � | _ t � | _ t � | _ t � | _ t � | _
| j | j | j � d | _ t
j | | j | d t d g d
g | | �d | _ d | _ | j � | j � | j � | j j d � | j j | � d S(
s|
Build an :class:`Axes3D` instance in
:class:`~matplotlib.figure.Figure` *fig* with
*rect=[left, bottom, width, height]* in
:class:`~matplotlib.figure.Figure` coordinates
Optional keyword arguments:
================ =========================================
Keyword Description
================ =========================================
*azim* Azimuthal viewing angle (default -60)
*elev* Elevation viewing angle (default 30)
================ =========================================
g g �?t azimi���t elevi i t frameont xtickst yticksi N( t Nonet figt _cidst popt initial_azimt initial_elevR 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 kwargs( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR% * s,
c C sl d | j } d | j } d | j } d | j } t j | | | d d �t j | | | d d �d S( Ngffffff�?g�������?t auto( t distR t set_xlimR t set_ylim( R
t xdwlt xdwt ydwlt ydw( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR* Y s
c C sy t j d | j j | j j | � | _ t j d | j j | j j | � | _ t j d | j
j | j j | � | _ d S( Nt xt yt z(
t axis3dt XAxisR t intervalxR! t w_xaxist YAxist intervalyt w_yaxist ZAxisR R" t w_zaxis( R
( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR) d s c C s� | p | j � \ } } } } } } | | | | | | | | g | | | | | | | | g | | | | | | | | g } } }
t | | |
� S( N( t
get_w_limst zip( R
t valst minxt maxxt minyt maxyt minzt maxzt xst yst zs( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt unit_cubel s
$&c C s= | d k r | j } n | j | � } t j | | � } | S( N( R R' RQ t proj3dt proj_points( R
RG R' t xyzst tcube( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
tunit_cubes s
c C s� | j | | � } | 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 ( RV ( R
RG R' t tct edges( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt tunit_edgesz s c C s� | j j | � t | _ | j � | _ | j | _ | j | _ | j | _ | j | _ g | j D] } | j
| � | f ^ qb } | j � | j � x) t
| � D] \ } \ } } | | _ q� Wg | j D] } | j
| � | f ^ q� } | j � | j � x) t
| � D] \ } \ } } | | _ qW| j | j | j f } x | D] } | j | � qMWx | D] } | j | � qkWt j | | � 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? RB RD t draw_paneR ( R
t renderert colt zlistt iR; t patcht axest ax( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRZ � s2 (
(
c C s� | j � } | j | | j � } | d d | d d k } | d d | d d k } | d d | d d k } | | | f S( Ni i i i ( RE RV R' ( R
RG RW t xhight yhight zhigh( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR` � s c K s d S( N( ( R
t xysR0 ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt update_datalim� s c C s� t t j | | | f � \ } } } y8 | j � | j � } } | d k r[ | j � } n Wn t k
rr � n X| j j t j | | g � j | � | d k r� | j
j t j | | g � j | � n | j � 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_dataR9 R: R; ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt auto_scale_xyz� s $
& )c K s | j � | j s d S| j � s' d S| rC | j | j j � n | r_ | j | j j � n | r{ | j | j j � n d S( N(
R* R$ t get_autoscale_ont
set_xlim3dR! R> t
set_ylim3dRA t
set_zlim3dR" ( R
t scalext scaleyt scalezt kw( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRz � s
c C sL | j � \ } } | j � \ } } | j � \ } } | | | | | | f S( s Get 3d world limits.( t
get_xlim3dt
get_ylim3dt
get_zlim3d( R
RH RI RJ RK RL RM ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRE � s c O sW | d k r* t j | � r* | \ } } n | | k rM | d 8} | d 7} n | | f S( Ng �?( R R t iterable( R
t xmint xmaxR/ R0 ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt _determine_lims� s
c O s" | j | | � } | | j _ | S( s Set 3D x limits.( R� R R> ( R
R/ R0 t lims( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� � s c O s" | j | | � } | | j _ | S( s Set 3D y limits.( R� R RA ( R
R/ R0 R� ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� � s c O s" | j | | � } | | j _ | S( s Set 3D z limits.( R� R R> ( R
R/ R0 R� ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� � s c C s
| j j S( s Get 3D x limits.( R R> ( R
( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� s c C s
| j j S( s Get 3D y limits.( R RA ( R
( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� s c C s
| j j S( s Get 3D z limits.( R R> ( R
( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� s c O s d S( N( R ( R
R/ R0 ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt clabel s c C s
d G| GHd S( Nt numsteps( ( R
R� ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt pany s c C s
d G| GHd S( NR� ( ( R
R� ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt panpy s c C sU d | _ | d k r$ | j | _ n | | _ | d k rH | j | _ n | | _ d S( s{
Set the elevation and azimuth of the axes.
This can be used to rotate the axes programatically.
'elev' stores the elevation angle in the z plane.
'azim' stores the azimuth angle in the x,y plane.
if elev or azim are None (default), then the initial value
is used which was specified in the :class:`Axes3D` constructor.
i
N( R2 R R R R R ( R
R R ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR# s
c C s� t j | j d t j | j d } } | j � \ } } | j � \ } } | j � \ } } t j | | | | | | � } t j d d d g � }
|
d t j
| � t j
| � | j } |
d t j | � t j
| � | j } |
d t j | � | j }
t j | | |
f � } | | _
|
| | _ | j t j | j � | _ t | � t j d k r~t j d � } n t j d � } | j | j } } t j | |
| � } t j | | � } t j | | � } t j | | � } | 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� RR t world_transformationR t cosR2 t sinR_ R^ 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' ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR] . s0 ) ++
i i c C s� d | _ | j j } | d k ru | j d | j � } | j d | j � } | j d | j � } | | | g | _ n
t j
d � t j | � | _
t j | � | _ d S( s" Initializes mouse button callbacks to enable 3D rotation of
the axes. Also optionally sets the mouse buttons for 3D rotation
and zooming.
============ =======================================================
Argument Description
============ =======================================================
*rotate_btn* The integer or list of integers specifying which mouse
button or buttons to use for 3D rotation of the axes.
Default = 1.
*zoom_btn* The integer or list of integers specifying which mouse
button or buttons to use to zoom the 3D axes.
Default = 3.
============ =======================================================
t motion_notify_eventt button_press_eventt button_release_eventsc Axes3D.figure.canvas is 'None', mouse rotation disabled. Set canvas then call Axes3D.mouse_init().N( R t button_pressedt figuret canvast mpl_connectt _on_movet
_button_presst _button_releaseR t warningst warnR t
atleast_1dt _rotate_btnt _zoom_btn( R
t
rotate_btnt zoom_btnt canvt c1t c2t c3( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR( ] s
c C s, | j � t j | � | j t d � d S( s7 Clear axes and disable mouse button callbacks.
s axes3d.gridN( t disable_mouse_rotationR t clat gridR ( R
( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� | s
c C s4 x$ | j D] } | j j j | � q
Wg | _ d S( s( Disable mouse button callbacks.
N( R R� R� t mpl_disconnect( R
t cid( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� � s c C s; | j | k r7 | j | _ | j | j | _ | _ n d S( N( t inaxest buttonR� t xdatat ydatat sxt sy( R
t event( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� � s c C s
d | _ d S( N( R R� ( R
R� ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� � s c C sB y | j | � SWn* t k
r= | j j � } 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
R9 t fmt( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt format_xdata� s
c C sB y | j | � SWn* t k
r= | j j � } 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� RB R� R
( R
R: R� ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt format_ydata� s
c C sH y | j | � SWn0 t t f k
rC | j j � } 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_zdataRw R� RD R� R
( R
R; R� ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt format_zdata� s
c C s� | j d k r d S| j | j k r9 d | j | j f S| | f } | j � } g t | � D]- \ } \ } } t j | | | � | f ^ q^ } | j
� | d d } | | \ } } | \ }
} } | \ }
} } t j |
| | | � } t j |
| | | � } | | } | | | | | | } t j
| | | | j � \ } } } | j | � } | j | � } | j | � } 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 s! azimuth=%d deg, elevation=%d deg i i s x=%s, y=%s, z=%sN( R' R R� R� R R RY Rd RR t line2d_seg_distRb R t hypott
inv_transformR� R� ( R
t xdt ydt pRX Rk t p0t p1t ldistst edgeit x0t y0t z0t x1t y1t z1t d0t d1t dtR; R9 R: RN RO RP ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt format_coord� s, =
$c C s | j s
d S| j d k r d S| j | j } } | d k rC d S| | j | | j } } | j � \ } } | j � \ } } | | }
| | } | | | _ | _ | j | j k r6| d k r� | d k r� d St
j | j | | d � | _ t
j | j
| |
d � | _
| j � | j j j � n� | j | j k r| j � \ } }
} } } } d | | | } |
| | } | | | } | | | } | j | | |
| � | j | | | | � | j | | | | � | j � | j j j � n d S( s� Mouse moving
button-1 rotates by default. Can be set explicitly in mouse_init().
button-3 zooms by default. Can be set explicitly in mouse_init().
Ni i� i ( R� R' R R� R� R� R� t get_xlimt get_ylimR� t art3dt
norm_angleR R R] R� R� RZ R� RE R� R� R� ( R
R� R9 R: t dxt dyR� R� R� R� t wt hRH RI RJ RK RL RM t dft dz( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� � s>
!!
c K sI | j j � } | j | � | d k r8 | j | � n | j | � | S( s Set xlabel.N( R? t get_labelt set_textR t update( R
t xlabelt fontdictR0 t label( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
set_xlabel s
c K sI | j j � } | j | � | d k r8 | j | � n | j | � | S( s Set ylabel.N( RB R� R� R R� ( R
t ylabelR� R0 R� ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
set_ylabel# s
c K sI | j j � } | j | � | d k r8 | j | � n | j | � | S( s Set zlabel.N( RD R� R� R R� ( R
t zlabelR� R0 R� ( ( sN /opt/alt/python27/lib64/python2.7/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 onR0 ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� 7 s c K s2 t j | | | | | � } t j | | | � | S( s�
Add text to the plot. kwargs will be passed on to Axes.text,
except for the `zdir` keyword, which sets the direction to be
used as the z direction.
( R t textR� t
text_2d_to_3d( R
R9 R: R; R t zdirR0 R ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR = s c O so | j � } | j d d � } | j d d � } d } t | � d k r� t j | d � r� t | � t | d � k r� t j | d d � r� | | } | d 7} n< t | � d k r� t j | d � r� | | } | d 7} n t j | � s t j t | � � | } n t j | | | | | | � } x' | D] }
t
j |
d | d | �q2W| j | | | | � | 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`
RP i R R; i (
t has_dataR t lenR R� t is_scalarR t onesR t plotR� t
line_2d_to_3dR ( R
RN RO R/ R0 R~ RP R t argsit linest line( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR J s$ %3
%
c , O su | j � } | j \ } } t j | � t j | � t j | � } }
} | j d d � } | j d d � }
d | k r� | j d � } n* t j t j | j d d � � � } d } | j d d � } | j d d � } | j d d � } | j d
d � } | j d d � } | j d | d k � } | j d
d � } | rp| d k rp| d k rp| j
| | | � } n g } g } g } x�t j d | d | � D]�} x�t j 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 � }# | j
| d | d |" d |" d g � t j | |! |" |# f � }$ | j
|$ � q�Wt | � } t j g � }% g }& d }' x> | D]6 }( |( |% k r[|& j
|( � |( }% |' |( d 7}' q[q[W| j
|& � | d k r�| j
| | | � n | j
|' t |& � � | d k r�| r�t j |& d � t j |& d � }) t j |& d � t j |& d � }* | j
t j |) |* � � q�q�Wq�Wt j | | | � }+ | d k r�| r�| j | | � } n |+ j | � |+ j | � n� | r t j | � } |+ j | � | d k s�| d k r|+ j | | � n | d k rN|+ j | � qNn. | r;| j | | � } n | } |+ j | � | j |+ � | j | | | | � |+ 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*, *Z* Data values as numpy.arrays
*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.
*facecolors* Face colors for the individual patches
*norm* An instance of Normalize to map values to colors
*vmin* Minimum value to map
*vmax* Maximum value to map
*shade* Whether to shade the facecolors
============= ================================================
Other arguments are passed on to
:func:`~mpl_toolkits.mplot3d.art3d.Poly3DCollection.__init__`
t rstridei
t cstridet
facecolorst colort bt cmapt normt vmint vmaxt linewidtht shadet lightsourcei i Ni����g i ( R t shapeR t transposeR R R t to_rgbaR t gett _shade_colors_lightsourcet aranget mint appendt concatenateRF R t crossR� t Poly3DCollectiont
_shade_colorst set_facecolorst set_edgecolorst set_arrayt set_climt set_normt add_collectionR (, R
R{ R| R} R/ R0 R~ t rowst colst tXt tYt tZR R t fcolorsR R R R R R R R t polyst normalst colsett rst cst pst cornerst at tat ztopt zleftt zbaset zrightR; t lastpt ps2t avgzsumR� t v1t v2t polyc( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt plot_surfacez s� / $ ."39()
$$$
c C s| g } xo | D]g } t j | d � t j | d � } t j | d � t j | d � } | j t j | | � � 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 polygonsR5 t vertsRD RE ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt _generate_normals� s
$$ c
C sV g } xC | D]; } | t j | � } | j t j | d d d g � � q
Wt j | � } t j | � } t | | � d k rFt t | | � t
| | � � } t j | � r� | j
� } d | d <g | D] } | d | | � d ^ q� } qRg t | | � D]6 \ } } t j t j | � � d | | � d ^ q} n | j
� } | S( s�
Shade *color* using normal vectors given by *normals*.
*color* can also be an array of the same length as *normals*.
i����g �?i i i ( RR R� R# R R� R t isnanR R R" t maxR� t iscolort copyRF R R (
R
R R5 R t nt maskR t vt colorst c( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR' s
&#
.L c C s4 | d k r$ t d d d d � } n | j | | � S( Nt azdegi� t altdegi7 ( R R R ( R
t dataR R ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR s c O s0 | j d d � } | j d d � } | j � } | j \ } }
t j | � t j | � t j | � } } }
g t d | | � D] } | ^ q� | d g } g t d |
| � D] } | ^ q� |
d g } g | D] } | | ^ q� } g | D] } | | ^ q� } g | D] } | | ^ q} g | D] } | | ^ q,} g | D] } | | ^ qI} g | D] } |
| ^ qf} g t | | | � D]! \ } } } t | | | � ^ q�} | g t | | | � D]! \ } } } t | | | � ^ q�7} t j | | | � } | j | � | j
| | | | � | S( sl
Plot a 3D wireframe.
========== ================================================
Argument Description
========== ================================================
*X*, *Y*, Data values as numpy.arrays
*Z*
*rstride* Array row stride (step size)
*cstride* Array column stride (step size)
========== ================================================
Keyword arguments are passed on to
:func:`matplotlib.collections.LineCollection.__init__`.
Returns a :class:`~mpl_toolkits.mplot3d.art3d.Line3DCollection`
R i R i ( R R R R R t rangeRF R� t Line3DCollectionR- R ( R
R{ R| R} R/ R0 R R R~ R. R/ R0 R1 R2 Rk t riit ciit xlinest ylinest zlinest txlinest tylinest tzlinest xlt ylt zlR t linec( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt plot_wireframe# s* /00 78
i c C s� | j } | j } | d | d d } x2t | | � D]!\ } } t j | j � | | � } t j | j � | | � } | j � d }
g } g } t t | d � | � }
|
d k r� t | d � d k r8 d }
q� q8 n t | d � d |
d } x� t t
t |
� � d � D]� } t
t | | � � } t
t | d | � � } | j | d | | d | | d | | d | g � t j
| d | � t j
| d | � } t j
| d | � t j
| d | � } | j t j | | � � q W| j |
| � } | j |
| � } t j | d | d | �} | j | � | j | � q8 Wx | D] } | j j | � qdWd S( s4
Extend a contour in 3D by creating
i i i R t
edgecolorsN( t levelsR RF R� t paths_to_3d_segmentst get_pathst get_colort roundR RW t intR# R R R% R' R& t
set_sort_zpost add_collection3dt remove( R
t csett strideRg t collsR� R; Rd t topvertst botvertsR t polyvertsR5 t nstepst stepsizeRk t i1t i2RD RE RR t colors2t polycolRi ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt _3d_extend_contourS sD # ,,
c O s! | j d t � } | j d d � } | j d d � } | j d d � } | j � }
t j | | | | � \ } } }
t j | | | |
| | � } d | } | r� | j | | � nQ xN t | j
| j � D]7 \ } } | d k r� | } n t j | | d | �q� W| j
| | | |
� | S( sI
Create a 3D contour plot.
========== ================================================
Argument Description
========== ================================================
*X*, *Y*, Data values as numpy.arrays
*Z*
*extend3d* Whether to extend contour in 3D (default: False)
*stride* Stride (step size) for extending contour
*zdir* The direction to use: x, y or z (default)
*offset* If specified plot a projection of the contour
lines on this position in plane normal to zdir
========== ================================================
The positional and other keyword arguments are passed on to
:func:`~matplotlib.axes.Axes.contour`
Returns a :class:`~matplotlib.axes.Axes.contour`
t extend3dRq i R R; t offsett -N( R R[ R R R� t rotate_axesR t contourR| RF Rg R t line_collection_2d_to_3dR ( R
R{ R| R} R/ R0 R} Rq R R~ R~ t jXt jYt jZRp R; Rd ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� � s !
" c
O s� | j � } t j | | | | | | � } | j } | j } xD t | | d | � D], \ }
} } t j | |
� | j |
� qS W| j | | | | � | S( s�
Plot filled 3D contours.
*X*, *Y*, *Z*: data points.
The positional and keyword arguments are passed on to
:func:`~matplotlib.axes.Axes.contourf`
Returns a :class:`~matplotlib.axes.Axes.contourf`
i (
R R t contourfRg R RF R� t poly_collection_2d_to_3dRm R (
R
R{ R| R} R/ R0 R~ Rp Rg Rr R� t z2Rd ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� � s &i R; c C s� t | � t j k rD t j | d | d | �| j t | � � n� t | � t j k r� t j | d | d | �| j t | � � nD t | � t j k r� t j
| d | d | �| j t | � � n t j | | � d S( s)
Add a 3d collection object to the plot.
2D collection types are converted to a 3D version by
modifying the object and adding z coordinate information.
Supported are:
- PolyCollection
- LineColleciton
- PatchCollection
RP R N(
t typeR t PolyCollectionR� R� Rm R" t LineCollectionR� t PatchCollectiont patch_collection_2d_to_3dR R- ( R
Ri RP R ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRn � s
c
O s� | j � } t j | | | | | � } t j | � sX t } t j t | � � | } n t } t
j | d | d | �| s� | j | | | | � n | S( s
Create a scatter plot.
========== ================================================
Argument Description
========== ================================================
*xs*, *ys* Positions of data points.
*zs* Either an array of the same length as *xs* and
*ys* or a single value to place all points in
the same plane. Default is 0.
*zdir* Which direction to use as z ('x', 'y' or 'z')
when plotting a 2d set.
========== ================================================
Keyword arguments are passed on to
:func:`~matplotlib.axes.Axes.scatter`.
Returns a :class:`~mpl_toolkits.mplot3d.art3d.Patch3DCollection`
RP R (
R R t scatterR R� R& R R
R R[ R� R� R (
R
RN RO RP R R/ R0 R~ Rf t is_2d( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� � s c O s4 | j � } t j | | | | | � } t j | � sR t j t | � � | } n g } g }
x� t | | � D]u \ } } t j
| � }
| |
j � 7} |
| g t |
� 7}
t j | | | � d | k rn | j
| d � qn qn Wt | � \ } } t j | | |
| � \ } } }
| j | | |
| � | S( s
Add 2D bar(s).
========== ================================================
Argument Description
========== ================================================
*left* The x coordinates of the left sides of the bars.
*height* The height of the bars.
*zs* Z coordinate of bars, if one value is specified
they will all be placed at the same z.
*zdir* Which direction to use as z ('x', 'y' or 'z')
when plotting a 2d set.
========== ================================================
Keyword arguments are passed onto :func:`~matplotlib.axes.Axes.bar`.
Returns a :class:`~mpl_toolkits.mplot3d.art3d.Patch3DCollection`
t alpha( R R t barR R� R R
R RF R� t get_patch_vertst tolistt patch_2d_to_3dt set_alphat juggle_axesR ( R
t leftt heightRP R R/ R0 R~ Rf RI t verts_zsR� R; t vsRN RO ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� s" !R t averagec O s | j � } t j | � s' | g } n t j | � sB | g } n t j | � s] | g } n t j | � sx | g } n t j | � s� | g } n t j | � s� | g } n t | � d k r� | t | � } n t | � d k r� | t | � } n t | � d k r | t | � } n t | � t | � k sMt | � t | � k r]t j d � n d \ } }
} d
\ } } } g } xjt | | | | | | � D]M\ } } } } } } t | | � } t | | | � } t | |
� }
t | | | � } t | | � } t | | | � } | j | | | f | | | | f | | | | | f | | | | f f | | | | f | | | | | f | | | | | | f | | | | | f f | | | f | | | | f | | | | | f | | | | f f | | | | f | | | | | f | | | | | | f | | | | | f f | | | f | | | | f | | | | | f | | | | f f | | | | f | | | | | f | | | | | | f | | | | | f f g � q�Wg } | d k rd g t | � } n� t | � t | � k rVxi | D] } | j | g d � q5WnD t t j
| � � } t | � t | � k r�| d t | � 9} n | j | � } | j | | � } t j | d | d | | |
�} | j | � | j | | f |
| f | | f | � d S( sr
Generate a 3D bar, or multiple bars.
When generating multiple bars, x, y, z have to be arrays.
dx, dy, dz can be arrays or scalars.
*color* can be:
- A single color value, to color all bars the same color.
- An array of colors of length N bars, to color each bar
independently.
- An array of colors of length 6, to color the faces of the
bars similarly.
- An array of colors of length 6 * N bars, to color each face
independently.
When coloring the faces of the boxes specifically, this is
the order of the coloring:
1. -Z (bottom of box)
2. +Z (top of box)
3. -Y
4. +Y
5. -X
6. +X
Keyword arguments are passed onto
:func:`~mpl_toolkits.mplot3d.art3d.Poly3DCollection`
i s$ x, y, and z must be the same length.g@��x �Dg@��x ��i t zsortt facecolorN( g@��x �Dg@��x �Dg@��x �D( g@��x ��g@��x ��g@��x ��( R R R� R R� R� RF R" RL t extendR t listR t
to_rgba_arrayRJ R' R� R& R- R ( R
R9 R: R; R� R� R� R R� R/ R0 R~ RH RJ RL RI RK RM R4 t xit yit zit dxit dyit dziR RS R5 t sfacecolorsRi ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt bar3d6 sz " 04 '$/ '$/ '$:
c K sC t j | | | | � | j j � \ } } | j j d | � d S( Ngq=
ףp�?( R t set_titlet titlet get_positiont set_y( R
R� R� R0 R9 R: ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR� � s N(B t __name__t
__module__t __doc__t nameR R% R* R) RQ RV RY RZ R` Rs R R& Rz RE R� R� R� R� R� R� R� R� R� R� R# R] R( R� R� R� R� R� R� R� R� R� R� R� R R� R t text3DR t text2DR t plot3DRG RJ R' R Re R| R� t contour3DR� t
contourf3DRn R� t scatter3DR� R� R� ( ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR $ s| / $ / ( 9
. y 0 0 , %+ pg�������?c C s� d d l m } t j d d | � } } t j | | � \ } } | | | d d d d � } | | | d d d d � } | | } | d
} | d
} | d } | | | f S( s6
Return a tuple X, Y, Z with a test data set.
i����( t bivariate_normalg �g @g �?g g �?g �?i i
i� ( t matplotlib.mlabR� R R! t meshgrid( t deltaR� R9 R: R{ R| t Z1t Z2R} ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
get_test_data� s
( R� R� t matplotlib.axesR R t
matplotlibR t matplotlib.transformsR R t numpyR t matplotlib.colorsR R R R� RR R< R
R R R� t matplotlib.projectionst projectionst projt projection_registryt register( ( ( sN /opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt <module>
s( � � � � �