gdZddlZddlmZdZ ddZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZeZdZdZdZdZdS)z, Various transforms used for by the 3D code Nc,tj|}||z }||z }||z}tj||d}|tjtj||z|z dd|z}||z dzddzS)a7 Return the distance(s) from point(s) *p* to segment(s) (*s0*, *s1*). Parameters ---------- p : (ndim,) or (N, ndim) array-like The points from which the distances are computed. s0, s1 : (ndim,) or (N, ndim) array-like The xy(z...) coordinates of the segment endpoints. r)axisg?)npasarraywheremultiplyouterclipsum)ps0s1s01s0pl2p1s Q/granary/edev/quantum/lib/python3.11/site-packages/mpl_toolkits/mplot3d/proj3d.py_line2d_seg_distr s BB r'C b&C sB "b!  B bkc B1 = =sCC CB VM  B  ' 'C 00c ||z }||z }||z } ||\} } } || z}|| z}| | z} tjd|z dd| |z gdd|z d| |z gddd| z | | z ggdgS)z Produce a matrix that scales homogeneous coords in the specified ranges to [0, 1], or [0, pb_aspect[i]] if the plotbox aspect ratio is specified. Nrrrrrrrarray) xminxmaxyminymaxzminzmax pb_aspectdxdydzaxayazs rworld_transformationr* s B B B B b b b 8adA!b1AbD!b1A!B$b1***, - --rc|tj|z \}}}tj|}tj|}dtj|dz dzz}tj||z|z|z||z|z||zz ||z|z||zzg||z|z||zz||z|z|z||z|z||zz g||z|z||zz ||z|z||zz||z|z|zgg}|S)zK Produce a rotation matrix for an angle in radians about a vector. r)rlinalgnormsincosr) vanglevxvyvzsctRs rrotation_about_vectorr96sRY^^A&&&JBB u A u A "&q//1 A  2b12b2a42b2a48 2b2a42b12b2a48 2b2a42b2a42b157 8 8A Hrct||z }|tj|z }tj||}|tj|z }tj||}|dkr;t || }tj||}tj||}|||fS)a Get the unit viewing axes in data coordinates. Parameters ---------- E : 3-element numpy array The coordinates of the eye/camera. R : 3-element numpy array The coordinates of the center of the view box. V : 3-element numpy array Unit vector in the direction of the vertical axis. roll : float The roll angle in radians. Returns ------- u : 3-element numpy array Unit vector pointing towards the right of the screen. v : 3-element numpy array Unit vector pointing towards the top of the screen. w : 3-element numpy array Unit vector pointing out of the screen. r)rr,r-crossr9dot)Er8Vrollwur0Rrolls r _view_axesrCGs0 QA ")..  A AA ")..  A AA qyy%a$// F5!   F5!   a7Nrctjd}tjd}|||g|ddddf<| |dddf<tj||}|S)a Return the view transformation matrix. Parameters ---------- u : 3-element numpy array Unit vector pointing towards the right of the screen. v : 3-element numpy array Unit vector pointing towards the top of the screen. w : 3-element numpy array Unit vector pointing out of the screen. E : 3-element numpy array The coordinates of the eye/camera. Nr)reyer<)rAr0r@r=MrMtMs r_view_transformation_uvwrKns` B BQBrr2A2vJBrr2vJ r2A HrcVt||||\}}}t||||}|S)az Return the view transformation matrix. Parameters ---------- E : 3-element numpy array The coordinates of the eye/camera. R : 3-element numpy array The coordinates of the center of the view box. V : 3-element numpy array Unit vector in the direction of the vertical axis. roll : float The roll angle in radians. )rCrK)r=r8r>r?rAr0r@rJs rview_transformationrMs5Aq$''GAq! Aq!,,A Hrc|}d}||z||z z }d||zz||z z }tj|dddgd||z ddgdd||ggdg}|S)Nrr)rrrrr)zfrontzback focal_lengtheabr6 proj_matrixs rpersp_transformationrWs~A A u %A F5L6%<(A(Q!aO!aO!aO+OO-..K rc b||z }||z }tjgdgdgddd||gg}|S)N)rrrr)rrrr)rrrOrrr)rPrQrTrUrVs rortho_transformationrYsW 5.A 5.A(MMM)MM)MMAqM+,,K rctj||}|d}|d|z |d|z |d|z }}}|||fSNrFrrr)rr<)vecrJvecwr@txstystzss r_proj_transform_vecrasK 6!S>>D QAGAItAwqy$q'!)cC S=rc0tj||}|d}|d|z |d|z |d|z }}}d|dk|ddkzd|dkz|ddkz}tj|r |ddk}||||fSr[)rr<any)r\rJr]r@r^r_r`tiss r_proj_transform_vec_clipres 6!S>>D QAGaKa1d1gkcC QalphacosasinaM1s rrot_xrsju $D <<<tdUA&tT1%<<! " "B 6"a==rrq)__doc__numpyr numpy.linalgr,rr*r9rCrKrMrWrYrarerorhru transformrwr|rzrrrrsJ111204----,   "$$$N   .   (    % % %444'''  ,,,999))) r