�
/&XLc@s�dZd
dl
Z
d
dlmZmZ
d
dlmZ
d
dlmZ
d
dlm	Z	
d
dl
Zd
dlm
Z
mZmZ
d
dlZd
dlZd
dlZd�Zd	�Zd
ef
d��YZdd
�
Zd
dljZejje�

dS(sZ
Module containing Axes3D, an object which can plot 3D objects on a
2D matplotlib figure.
i����N(tAxestrcParams(
tcbook(
tBbox(
tcollections(t	NormalizetcolorConvertertLightSourcecCsGt|
�
d
ks$t|
�
dkr;d|
}|j
|�
Sd|
SdS(s:Used to generate more comprehensible numbers in status barg��@g����MbP?s%1.4es%4.3fN(tabst_formatSciNotation(tselftvaluet
s((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytsensible_format_datas$



c
Cs+tt
jd
d
gddgg�
�
}|S(Nii
(Rtnptarray(
tbox((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt	unit_bbox s
'
tAxes3DcBsxeZ
dZd
Zd>d�
Zd�Zd�Zd>d�
Zd>d>d�Z	d>d>d�Z
d�Zd	�Zd
�Z
d>d>d�Zeeed�Zd
�Zd>d>d�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd>d>d�Zd�Zddd�Zd�Zd�Z d�Z!d �Z"d!�Z#d"�Z$d#�Z%d$�Z&d%�Z'd>d&�
Z(d>d'�
Z)d>d(�
Z*ed)�
Z+d>d*�
Z,e,Z-e.j,Z/d+�Z0e0Z1d,�Z2d-�Z3d.�Z4d/�Z5d0�Z6d1d2�
Z7d3�Z8e8Z9d4�Z:e:Z;d5d6d7�Z<d5d6d8�Z=e=Z>d5d6d9�Z?d:d;d<�Z@d>d=�
ZARS(?s
    3D axes object.
    t3dcOs.
|dkr!d
d
ddg}n|
|_
g|_|jdd�|_|jdd�|_t�|_t�|_t�|_	t�|_
|j|j|j�
d|_t
j||j
|dtd	gd
g||�
d|_d|_|j�
|j�
|j�
|jjd�

|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)
          ================   =========================================
        gg�?tazimi���televiitframeontxtickstyticksi
N(tNonetfigt_cidstpoptinitial_azimtinitial_elevRt
xy_viewLimt
zz_viewLimt
xy_dataLimt
zz_dataLimt	view_initt_readyRt__init__tTruet
Mt
mouse_inittcreate_axestset_top_viewt	axesPatcht
set_linewidthtadd_axes(R
Rtrecttargstkwargs((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR%*s,

	
	




	


		






c
Csld
|j}
d|j}d
|j}d|j}t
j||
|dd�

t
j|||dd�

dS(Ngffffff�?g�������?tauto(tdistRtset_xlimRtset_ylim(R
txdwltxdwtydwltydw((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR*Ys







c

Csytj
d
|jj|jj|�|_tjd|jj|jj|�|_tj	d|j
j|jj|�|_dS(Nt
xt
yt
z(
taxis3dtXAxisRt	intervalxR!tw_xaxistYAxist	intervalytw_yaxistZAxisR R"tw_zaxis(
R
((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR)ds





cCs�|
p|j�\}}}}}}||||||||g||||||||g||||||||g}}	}
t
||	|
�S(
N(t
get_w_limstzip(R
tvalstminxtmaxxtminytmaxytminztmaxztxstystzs((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt	unit_cubels

$


&
cCs=|dkr|j
}n|j|
�
}tj||�}|S(
N(RR'RQtproj3dtproj_points(R
RGR'txyzsttcube((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
tunit_cubess





cCs�|j|
|�}|d
|df|d|df|d|df|d|d
f|d
|df|d|df|d|df|d|df|d|df|d|df|d|df|d|dfg}|S(	Nii
iiiiii(
RV(R
RGR'ttctedges((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyttunit_edgeszs











c	Cs�
|jj
|
�

t|_|j�|_|j|
_|j|
_|j|
_|j|
_g|j	D]}|j
|
�
|f^qb}|j�
|j�
x)t
|�
D]\}\}}||_q�Wg|jD]}|j
|
�
|f^q�}|j�
|j�
x)t
|�
D]\}\}}||_q
W|j|j|jf}x|D]}|j|
�

qM
Wx|D]}|j
|
�

qk
Wtj
||
�
dS(
N(R+tdrawtFalset_frameontget_projR'tvvecteyetget_axis_positionRtdo_3d_projectiontsorttreverset	enumeratetzordertpatchesR?RBRDt	draw_paneR(	R
trenderertcoltzlistt
iR;tpatchtaxestax((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRZ�s2
	




(







(












c
Cs�|j�}
|j
|
|j�}|d
d|ddk}|dd|ddk}|dd|ddk}|||fS(Ni
iii(RERVR'(R
RGRWtxhightyhightzhigh((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR`�s





c
KsdS(
N((R
txysR0((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytupdate_datalim�s
cCs�tt
j|
||f�\}}}y8|j�|j�}}|dk	r[|j�}nWntk
rr


�n
X|jjt
j||g�
j	|�
|dk	r�|j
jt
j||g�
j	|�
n|j�
dS(
N(tmapRtasarraytflattenRtAttributeErrorR!tupdate_from_data_xyRt
TR"tautoscale_view(R
t
Xt
Yt
Zthad_dataR9R:R;((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytauto_scale_xyz�s
$






&

)cKs|j�
|j
sdS|j�s'dS|
rC|j|jj�

n|r_|j|jj�

n|r{|j|j	j�

ndS(
N(
R*R$tget_autoscale_ont
set_xlim3dR!R>t
set_ylim3dRAt
set_zlim3dR"(R
tscalextscaleytscaleztkw((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRz�s

	







c
CsL|j�\}
}|j
�\}}|j�\}}|
|||||fS(
sGet 3d world limits.(t
get_xlim3dt
get_ylim3dt
get_zlim3d(R
RHRIRJRKRLRM((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRE�s


cOsW|dkr*t
j|
�
r*|
\}
}n|
|krM|
d
8}
|d
7}n|
|fS(Ng�?(RRtiterable(R
txmintxmaxR/R0((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt_determine_lims�s







c
Os"|j|
|�}||j
_|S(
sSet 3D x limits.(R�RR>(R
R/R0tlims((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR��s

c
Os"|j|
|�}||j
_|S(
sSet 3D y limits.(R�RRA(R
R/R0R�((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR��s

c
Os"|j|
|�}||j
_|S(
sSet 3D z limits.(R�R R>(R
R/R0R�((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR��s

c


Cs
|jj
S(
sGet 3D x limits.(RR>(
R
((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR�
sc


Cs
|jj
S(
sGet 3D y limits.(RRA(
R
((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR�
sc


Cs
|jj
S(
sGet 3D z limits.(R R>(
R
((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR�
sc

OsdS(
N(
R(R
R/R0((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytclabel
s
c
Cs
d
G|
GHdS(Ntnumsteps((R
R�((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytpany
s
c
Cs
d
G|
GHdS(NR�((R
R�((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytpanpy
s
cCsUd
|_|
dkr$|j|_n	|
|_|dkrH|j|_n	||_dS(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(R2RRRRR(R
RR((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR#
s
	
	
c
Cs�
tj
|jd
tj
|jd
}
}|j�\}}|j�\}}|j�\}}tj||||||�}	tj	dddg�
}
|
dtj
|�
tj
|
�
|j}|
dtj|�
tj
|
�
|j}|
dtj|
�
|j}
tj	|||
f�
}||_
|
||_|jtj|j�
|_t|
�
tj
dkr~
tj	d�
}ntj	d�
}|j|j}}tj||
|�}tj||�}tj||	�}tj||�}|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�?ii
ii����(iii����(iii
(RtpiRRR�R�R�RRtworld_transformationRtcosR2tsinR_R^tmodRtview_transformationtpersp_transformationtdot(R
trelevtrazimR�R�tymintymaxtzmintzmaxtworldMt
Rtxptyptzpt
Et
VtzfronttzbacktviewMtperspMtM0R'((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR].
s0)



+
+

	







i
icCs�d|_
|jj}|dkru|jd
|j�}|jd|j�}|jd|j�}|||g|_n
t	j
d�

tj|
�
|_
tj|�
|_dS(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.
        ============  =======================================================

        tmotion_notify_eventtbutton_press_eventtbutton_release_eventscAxes3D.figure.canvas is 'None', mouse rotation disabled.  Set canvas then call Axes3D.mouse_init().N(Rtbutton_pressedtfiguretcanvastmpl_connectt_on_movet
_button_presst_button_releaseRtwarningstwarnRt
atleast_1dt_rotate_btnt	_zoom_btn(R
t
rotate_btntzoom_btntcanvtc1tc2tc3((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR(]
s	







c

Cs,|j�
t
j|�

|jtd
�

dS(s7Clear axes and disable mouse button callbacks.
        saxes3d.gridN(tdisable_mouse_rotationRtclatgridR
(
R
((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR�|
s



c
Cs4x$|jD]}
|j
jj|
�

q
Wg|_d
S(s(Disable mouse button callbacks.
        N(RR�R�tmpl_disconnect(R
tcid((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR��
s
cCs;|
j|kr7|
j
|_|
j|
j|_|_ndS(
N(tinaxestbuttonR�txdatatydatatsxtsy(R
tevent((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR��
s


cCs
d|_
dS(
N(RR�(R
R�((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR��
s
cCsBy|j|
�
SWn*t
k
r=


|jj�}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?tget_major_formatterR
(R
R9tfmt((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytformat_xdata�
s





cCsBy|j|
�
SWn*t
k
r=


|jj�}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�RBR�R
(R
R:R�((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytformat_ydata�
s





cCsHy|j|
�
SWn0t
tfk
rC


|jj�}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_zdataRwR�RDR�R
(R
R;R�((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytformat_zdata�
s




cCs�
|jdkrd
S|j|jkr9d|j|jfS|
|f}|j�}gt|�
D]-\}\}}tj	|||�|f^q^}|j
�
|dd}	||	\}}|\}
}}|\}
}}tj|
|
||�}tj|
|
||�}||}||||||}tj
|
|||j�\}}}|j|�
}|j|�
}|j|�
}d|||fS(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.
        ts!azimuth=%d deg, elevation=%d deg ii
sx=%s, y=%s, z=%sN(R'RR�R�RRRYRdRRtline2d_seg_distRbRthypott
inv_transformR�R�(R
txdtydt
pRXRktp0tp1tldiststedgeitx0ty0tz0tx1ty1tz1td0td1tdtR;R9R:RNRORP((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytformat_coord�
s,



=







$


cCs|js
d
S|j
d
kr d
S|
j|
j}}|d
krCd
S||j||j}}|j�\}}|j�\}}	||}
|	|}|||_|_|j|j	kr6
|dkr�|dkr�d
St
j|j||d�
|_t
j|j
||
d�
|_
|j�
|jjj�
n�|j|jkr|j�\}}
}}}}d|||}|
||}|||}|||}|j|||
|�
|j||||�
|j||||�
|j�
|jjj�
nd
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().
        Nii�i
(R�R'RR�R�R�R�tget_xlimtget_ylimR�tart3dt
norm_angleRRR]R�R�RZR�RER�R�R�(R
R�R9R:tdxtdyR�R�R�R�t
wt
hRHRIRJRKRLRMtdftdz((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR��
s>	











!
!


	









cKsI|jj
�}|j|
�

|d
k	r8|j|�

n|j|�

|S(sSet xlabel.N(R?t	get_labeltset_textRtupdate(R
txlabeltfontdictR0tlabel((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
set_xlabels






cKsI|jj
�}|j|
�

|d
k	r8|j|�

n|j|�

|S(sSet ylabel.N(RBR�R�RR�(R
tylabelR�R0R�((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
set_ylabel#s






cKsI|jj
�}|j|
�

|d
k	r8|j|�

n|j|�

|S(sSet zlabel.N(RDR�R�RR�(R
tzlabelR�R0R�((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
set_zlabel-s






cKs
|
|_d
S(s&
        Set / unset 3D grid.
        N(
t
_draw_grid(R
tonR0((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR�7scKs2tj
||
|||�}tj|||�
|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.
        (RttextR�t
text_2d_to_3d(R
R9R:R;RtzdirR0R
((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR
=s

cOso
|j�}|j
d
d�}|j
dd�}d}t|�
dkr�tj|d�
r�t|
�
t|d�
kr�tj|dd�
r�||}|d7}n<t|�
dkr�tj|d�
r�||}|d7}ntj|�
s
tjt|
�
�
|}ntj	||
||||�}	x'|	D]}
t
j|
d
|d|�

q2
W|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`
        RPiR
R;i
(
thas_dataRtlenRR�t	is_scalarRtonesRtplotR�t
line_2d_to_3dR(R
RNROR/R0R~RPR
targsitlinestline((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR
Js$

%
3



%








c,Osu|j�}|j
\}}tj|
�
tj|�
tj|�
}	}
}|jd
d�}|jdd�}
d|kr�|jd�
}n*tjtj|jdd��
�
}d}|j	dd�}|jdd�}|jd	d�}|jd
d�}|j	dd�}|jd|dk�}|jd
d�}|rp
|dk	rp
|dk	rp
|j
|||�}ng}g}g}x�tjd|d|�D]�}x�tjd|d|
�D]�}g}g}x\
|
|	f||
f||fgD]9
\}}|||t|||
d�!} |t|d||
�|t|||d�!}!|t|d||�|t|||
d�d�}"|"ddd�}"|||t|||d�d�}#|#ddd�}#|j
| d| d|"d|"dg�

tj| |!|"|#f�
}$|j
|$�

q�
Wt|�}tjg�
}%g}&d}'x>|D]6}(|(|%kr[|&j
|(�

|(}%|'|(d7}'q[q[W|j
|&�

|dk	r�|j
|||�

n|j
|'t|&�
�

|dkr�
|r�
tj|&d�
tj|&d�
})tj|&d�
tj|&d�
}*|j
tj|)|*��

q�
q�
Wq�
Wtj|||�
}+|dk	r�|r�|j||�}n|+j|�

|+j|�

n�|r tj|�
}|+j|�

|dk	s�|dk	r
|+j||�
n|dk	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__`
        trstridei
tcstridet
facecolorstcolort
btcmaptnormtvmintvmaxt	linewidthtshadetlightsourceii
Ni����gi(R
tshapeRt	transposeRRRtto_rgbaRtgett_shade_colors_lightsourcetarangetmintappendtconcatenateRFR
tcrossR�tPoly3DCollectiont
_shade_colorstset_facecolorstset_edgecolorst	set_arraytset_climtset_normtadd_collectionR(,R
R{R|R}R/R0R~trowstcolsttXttYttZR
R
tfcolorsR
R
R
R
R
R
R
R
tpolystnormalstcolsettrstcstpstcornerst
attatztoptzlefttzbasetzrightR;tlastptps2tavgzsumR�tv1tv2tpolyc((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytplot_surfacezs�
/


$









 
 


.
"
3
9

(

)















$
$
$

















cCs|g}xo|
D]g}tj
|d
�
tj
|d�
}tj
|d�
tj
|d
�
}|jtj||��

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.
        ii
i(RRR#
R%
(R
tpolygonsR5
tvertsRD
RE
((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt_generate_normals�s


$
$

c
CsV
g}xC|D];}|tj
|�
}|jtj|d
d
dg��

q
Wtj|�
}tj|�
}t||�
dkrF
tt	||�
t
||�
�}tj|
�
r�|
j
�}
d|
d<g|D]}|
d||�
d^q�}qR
gt|
|�D]6\}	}tjtj|	�
�
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�?ii
i(RRR�R#
RR�RtisnanR
RR"
tmaxR�tiscolortcopyRFRR
(
R
R
R5
R
t
ntmaskR
t
vtcolorst
c((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR'

s 



&

#




.LcCs4|dkr$t
d
ddd�}n|j|
|�S(Ntazdegi�taltdegi7(RRR
(R
tdataR
R
((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR 
s


cOs0|jd
d�}|jdd�}|j
�}|j\}	}
tj|
�
tj|�
tj|�
}}}
gtd|	|�D]}|^q�|	dg
}gtd|
|�D]}|^q�|
dg
}g|D]}|
|^q�}g|D]}||^q�}g|D]}||^q
}g|D]}||^q,
}g|D]}||^qI
}g|D]}|
|^qf
}gt|||�D]!\}}}t|||�^q�
}|gt|||�D]!\}}}t|||�^q�
7}tj|||�
}|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(RR
R
RR
trangeRFR�tLine3DCollectionR-
R(R
R{R|R}R/R0R
R
R~R.
R/
R0
R1
R2
RktriitciitxlinestylinestzlinesttxlinesttylinesttzlinestxltyltzlR
tlinec((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytplot_wireframe#s*

/0
0





7

8


icCs�|
j}|
j
}|d
|dd}x2t||�D]!\}}tj|j�||�}tj|j�||�}	|j�d}
g}g}tt|d�
|�
}
|
d
k
r�t|d�
d
kr8d}
q�q8nt|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|�
}|jtj||��

q
W|j|
|�}|j|
|�}tj|d|d|�
}|j|�

|j|�

q8Wx|D]}|j
j|�

qdWdS(s4
        Extend a contour in 3D by creating
        i
iiR
t
edgecolorsN(tlevelsRRFR�tpaths_to_3d_segmentst	get_pathst	get_colortroundR
RW
tintR#
RRR%
R'
R&
t
set_sort_zpostadd_collection3dtremove(R
tcsettstrideRg
tcollsR�R;Rd
ttopvertstbotvertsR
t	polyvertsR5
tnstepststepsizeRkti1ti2RD
RE
RR
tcolors2tpolycolRi((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt_3d_extend_contourSsD	
	







	
#





,
,




	




cOs!
|jd
t
�}|jdd�}|jdd�}|jdd�}	|j�}
tj|
|||�\}}}
tj||||
||�}d|}|r�|j||�
nQxNt	|j
|j�D]7\}}|	dk	r�|	}ntj||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`
        textend3dRq
iR
R;toffsett
-N(RR[RR
R�trotate_axesRtcontourR|
RFRg
Rtline_collection_2d_to_3dR(R
R{R|R}R/R0R}
Rq
R
R~
R~tjXtjYtjZRp
R;Rd
((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR�
�s 


!



"

	

c
Os�|j�}t
j||
||||�}|j}|j}	xDt||d
|	�D],\}
}}tj||
�
|j|
�

qSW|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
RtcontourfRg
RRFR�tpoly_collection_2d_to_3dRm
R(
R
R{R|R}R/R0R~Rp
Rg
Rr
R�tz2Rd
((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR�
�s
	
	
&


iR;cCs�t|
�
t
jkrDtj|
d
|d|�

|
jt|�
�

n�t|
�
t
jkr�tj|
d
|d|�

|
jt|�
�

nDt|
�
t
j	kr�tj
|
d
|d|�

|
jt|�
�

ntj||
�
dS(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
        RPR
N(
ttypeRtPolyCollectionR�R�
Rm
R"
tLineCollectionR�
tPatchCollectiontpatch_collection_2d_to_3dRR-
(R
RiRPR
((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRn
�s








c
Os�|j�}t
j||
|||�}tj|�
sXt}	tjt|
�
�
|}nt	}	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`
        RPR
(
R
RtscatterRR�R&RR

R
R[R�R�
R(
R
RNRORPR
R/R0R~Rftis_2d((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR�
�s




cOs4
|j�}t
j||
|||�}tj|�
sRtjt|
�
�
|}ng}	g}
x�t||�D]u\}}t	j
|�
}
|	|
j�7}	|
|g
t|
�
7}
t	j|||�
d
|krn|j
|d
�

qnqnWt|	�\}}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`
        talpha(R
RtbarRR�RR

R
RFR�tget_patch_vertsttolisttpatch_2d_to_3dt	set_alphatjuggle_axesR(R
tlefttheightRPR
R/R0R~RfRI
tverts_zsR�R;tvsRNRO((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR�
s"









!
R
taveragec	Os|j�}t
j|
�
s'|
g
}
nt
j|�
sB|g
}nt
j|�
s]|g
}nt
j|�
sx|g
}nt
j|�
s�|g
}nt
j|�
s�|g
}nt|�
d
kr�|t|
�
}nt|�
d
kr�|t|�
}nt|�
d
kr
|t|�
}nt|
�
t|�
ksM
t|
�
t|�
kr]
tjd�

nd	\}}
}d
\}}}g}xjt|
|||||�D]M\}}}}}}t||�}t|||�}t||
�}
t|||�}t||�}t|||�}|j	|||f||||f|||||f||||ff||||f|||||f||||||f|||||ff|||f||||f|||||f||||ff||||f|||||f||||||f|||||ff|||f||||f|||||f||||ff||||f|||||f||||||f|||||ffg�

q�
Wg}|dkrdg
t|
�
}n�t|�
t|
�
krVxi|D]}|j	|g
d�

q5WnDttj
|�
�
}t|�
t|
�
kr�|dt|
�
9}n|j|�
}|j||�}tj|d|d||	|
�
}|j|�

|j||f|
|f||f|�
dS(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��itzsortt	facecolorN(g@��x�Dg@��x�Dg@��x�D(g@��x��g@��x��g@��x��(R
RR�R
R�R�RFR"
RL
textendRtlistRt
to_rgba_arrayRJ
R'
R�R&
R-
R(R
R9R:R;R�R�R�R
R�
R/R0R~RHRJRLRIRKRMR4
txityitzitdxitdyitdziR
RS
R5
tsfacecolorsRi((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytbar3d6sz"














0


4







'
$
/
'
$
/
'
$
:












cKsCtj
||
||�
|jj�\}}|jjd
|�

dS(Ngq=
ףp�?(Rt	set_titlettitletget_positiontset_y(R
R�R�R0R9R:((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR�
�s


N(Bt__name__t
__module__t__doc__tnameRR%R*R)RQRVRYRZR`RsRR&RzRER�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
ttext3DRttext2DR
tplot3DRG
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
	Cs�d
dlm
}

tjdd|�}}tj||�\}}|
||dddd�}|
||ddd	d	�}||}|d
}|d
}|d}|||fS(s6
    Return a tuple X, Y, Z with a test data set.
    i����(
tbivariate_normalg�g@g�?gg�?g�?i
i
i�
(tmatplotlib.mlabR�
RR!
tmeshgrid(	tdeltaR�
R9R:R{R|tZ1tZ2R}((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt
get_test_data�s










(R�
R�tmatplotlib.axesRR
t
matplotlibRtmatplotlib.transformsRRtnumpyRtmatplotlib.colorsRRRR�RRR<R
RRR�
tmatplotlib.projectionstprojectionstprojtprojection_registrytregister(((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt<module>
s(







		�����