�
[�3Lc@s6
dZd
dl
mZ
d
dlmZ
d
dlZd
dljZd�Z	d�Z
d�Zd�Zd	�Z
d
�Zd�Zd�Zd
�Zd�Zd�Zd�Zd�Zd�Zd�ZeZd�Zd�Zd�Zdd�
Zdd�
Zd�Zd�Z d�Z!e"dkr2
e�
ndS(s,
Various transforms used for by the 3D code
i����(
tLineCollection(
tCircleNc	Cs�|d
 \}}|
d
 \}}||krAd}d}|}nH||krcd}d}|}n&||}||}||||}|||fS(s<
    Return 2D equation of line in the form ax+by+c = 0
    ii����ii
((	tp0tp1tx0ty0tx1ty1t
at
bt
c((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytline2ds



	









cCsN|\}}}|
\}}t|||||t
j|d
|d
�
�
S(sO
    Distance from line to point
    line is a tuple of coefficients a,b,c
    i(tabstnptsqrt(t
lt
pRR	R
RR((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytline2d_dist%s

c	Cs�|
d
|d
}|
d|d}tj
|d
�
|d
}tj
|d�
|d}||||tt|d|d�
�
}tj|d
d�}tj|||d|||d�
}|S(s�distance(s) from line defined by p1 - p2 to point(s) p0

    p0[0] = x(s)
    p0[1] = y(s)

    intersection point p = p1 + u*(p2-p1)
    and intersection point lies within segement if u is between 0 and 1
    ii
i(R
tasarraytfloatRtclipR(	Rtp2Rtx21ty21tx01ty01t
ut
d((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytline2d_seg_dist/s



.

+cCs*
d
dl}|j
�}
d
d}}|j||�
t||�}|\}}dd
}}|j||�
t|||d|df�}t||tj||f�
�}xKt|||�D]7\}}	}
t||	f|
dd�
}|
j	|�

q�W|j
d	d�
|jd	d�
|j�
dS(Ni����iiii�idi�tfilli8���(ii(ii�(iiii(idi�ii�(
tpylabtgcatplottziptscatterRR
tarrayR
t	add_patchtxlimtylimtshow(RtaxtxstystpointsRRtdistt
xt
yRR
((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pyttest_lines_distsDs 







 
!
"



c

Cs-tj
|d
d|dd|dd�
S(s3d vector lengthiii
(R
R(
t
v((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytmodZsc	Cs�|
|||||}}}tj
d
|dd||gdd
|d||gddd
|||gdddd
gg�
S(Ng�?i(R
R#(	txmintxmaxtymintymaxtzmintzmaxtdxtdytdz((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytworld_transformation^s 




cCsHd\}}
d\}}d\}}t||
||||�}|GHdS(	Nidixi����g�������?g�������?(idix(i����id(g�������?g�������?(
R;(R2R3R4R5R6R7t
M((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pyt
test_worldhs





cCs
||
}|t|�
}t
j||�}|t|�
}t
j||�}|d
|d|dd
g|d
|d|dd
g|d
|d|dd
gd
d
d
dgg}dd
d
|d
gd
dd
|dgd
d
d|dgd
d
d
dgg}t
j||�S(Nii
i(R1R
tcrosstdot(t
Et
Rt
Vt
nRR0tMrtMt((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytview_transformationos












cCsq||
||
}d
||
||
}tj
ddddgddddgdd||gddddgg�
S(Ni����i
ii����(R
R#(tzfronttzbackRR	((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytpersp_transformation�s





cCsUtj
|
|�}|d
}|d||d||d|}}}|||fS(Niii
i(R
R?(tvecR<tvecwt
wttxsttysttzs((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytproj_transform_vec�s


,
cCs�tj
|
|�}|d
}|d||d||d|}}}|ddk|ddk
|ddk|ddk
}tj|�
r�|ddk}n||||fS(Niii
i(R
R?tsometrue(RJR<RKRLRMRNROttis((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytproj_transform_vec_clip�s


,
:


cCsrtj
|�
}t||
|�}tj||�}y||d
}Wntk
rX


n
X|d|d|dfS(Niii
i(tlinalgtinvtvec_pad_onesR
R?t
OverflowError(R)R*tzsR<tiMRJtvecr((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pyt
inv_transform�s








cCs�ysy+tj
||
|tj|j�
g�
}WnAttfk
rq


tj
||
|tjt|�
�
g�
}n
XWn,tk
r�


tj
||
|d
g�
}n
X|S(Ni
(R
R#tonestshapetAttributeErrort	TypeErrortlen(R)R*RXRJ((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pyRV�s


+

2



cCst||
|�}t
||�S(
s7
    Transform the points by the projection matrix
    (RVRP(R)R*RXR<RJ((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytproj_transform�s
cCst||
|�}t
||�S(
sv
    Transform the points by the projection matrix
    and return the clipping result
    returns txs,tys,tzs,tis
    (RVRS(R)R*RXR<RJ((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytproj_transform_clip�s
cCstt
||
��S(
N(R!tproj_trans_points(R+R<((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytproj_points�s
cCs(t|�\}}}t
||||
�S(
N(R!Ra(R+R<R)R*RX((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pyRc�s

cCs(t|�\}}}t
||||
�S(
N(R!Rb(R+R<R)R*RX((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytproj_trans_clip_points�s

i
cCsG
d
dl}d|
ddgdd|
dgddd|
g}}}t
||||�\}}}|d|df|d|df|d|df|d|dff\}	}
}}|	|
f|	|f|	|fg}
|j�}
t|
�
}|
j|�

xBt||dddd	g�D]"\}}}|j|||�
q
WdS(
Ni����ii
iit
oR-R.t
z(RRaRRtadd_collectionR!ttext(R<t
sRR)R*RXRMRNRORfR(taytaztlinestlinecR-R.t
t((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pyttest_proj_draw_axes�s

8

"
4
!



.
c
Cs�|ptj
d
ddg�
d}tj
d
d
d
g�
d}
tj
ddd
g�
}t||
|�}tdd�}tj||�}|S(Ni
i����ii�idii����(R
R#RFRIR?(R@RARBtviewMtperspMR<((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pyttest_proj_make_M�s"




cCs�
d
dl}t
�}
gddddddddd	dg
D]}d
|^q:}ddddddddddg
ddddddddddg
ddddddddddg
}}}g|||fD]}tj|�
d^q�\}}}t|
dd
�


t||||
�\}}	}
t||	|
|
�\}}}
|j||	d|
�

|j||	dd�

x6t	||	|�D]"\}}}|j
|||�
q�
W|jdd�
|jdd�
|j
�
dS(Ni����ii
iiiiiis%di,
Rji�
R
t
rg������ɿg�������?(RRsR
R#RpRaR[R"R R!RiR%R&R'(RR<t
ittsR)R*RXR0RMRNROtixstiystizsR-R.Ro((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pyt	test_proj�s 

	;
B
,
8



"

cCs{tj
|
�
tj|
�
}}tjd
dddgd||dgd||dgddddgg�
}tj||�S(Ni
i(R
tcostsinR#R?(RBtalphatcosatsinatM1((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pytrot_x	
s




c
CsRd
ddd
g}t|t
jd�GHdd
dd
g}t|t
jd�GHdS(Ni
ii(R�R
tpi(
RB((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pyttest_rot
s



t__main__(#t__doc__tmatplotlib.collectionsRtmatplotlib.patchesR
tnumpyR
tnumpy.linalgRTRRRR/R1R;R=RFRIRPRSR[RVRaRbt	transformRdRcReRptNoneRsRzR�R�t__name__(((sN/opt/alt/python27/lib64/python2.7/site-packages/mpl_toolkits/mplot3d/proj3d.pyt<module>s<