o li@sdZddlZddlmZddZ d%ddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZddZddZeZddZdd Zd!d"Zd#d$ZdS)&z, Various transforms used for by the 3D code NcCslt|}||}||}||}t||d}|tjt|||dd|}||djdddS)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)ps0s1s01s0pl2p1rY/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/mpl_toolkits/mplot3d/proj3d.py_line2d_seg_dist s $rc Cs||}||}||} |dur!|\} } } || }|| }| | } td|dd| |gdd|d| |gddd| | | 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_aspectdxdydzaxayazrrrworld_transformation s r'c Cs|tj|\}}}t|}t|}dt|dd}t||||||||||||||g||||||||||||||g||||||||||||||gg}|S)zK Produce a rotation matrix for an angle in radians about a vector. r)rlinalgnormsincosr) vanglevxvyvzsctRrrrrotation_about_vector6s  444r5cCsv||}|tj|}t||}|tj|}t||}|dkr6t|| }t||}t||}|||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)crossr5dot)Er4Vrollwur,Rrollrrr _view_axesGs      r>cCsPtd}td}|||g|ddddf<| |dddf<t||}|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)reyer7)r<r,r;r8MrMtMrrr_view_transformation_uvwns   rEcCs&t||||\}}}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. )r>rE)r8r4r9r:r<r,r;rDrrrview_transformationsrFcCsb|}d}||||}d||||}t|dddgd||ddgdd||ggdg}|S)Nrr)rrrrr)zfrontzback focal_lengtheabr2 proj_matrixrrrpersp_transformations rOc Cs>|| }|| }tgdgdgddd||gg}|S)N)rrrr)rrrr)rrrGrrr)rHrIrLrMrNrrrortho_transformations    rPcCsFt||}|d}|d||d||d|}}}|||fSNr@rrr)rr7)vecrDvecwr;txstystzsrrr_proj_transform_vecs ( rWcCst||}|d}|d||d||d|}}}d|dk|ddk@d|dk@|ddk@}t|rA|ddk}||||fSrQ)rr7any)rRrDrSr;rTrUrVtisrrr_proj_transform_vec_clips (0   rZcCs\t|}t|||}t||}z||d}Wn ty"Ynw|d|d|dfS)zK Transform the points by the inverse of the projection matrix *M*. r@rrr)r(inv _vec_pad_onesrr7 OverflowError)xsyszsrDiMrRvecrrrr inv_transforms    rccCst|||t|gSN)rr ones_like)r^r_r`rrrr\sr\cCt|||}t||S)z< Transform the points by the projection matrix *M*. )r\rWr^r_r`rDrRrrrproj_transforms  rhcCrf)zy Transform the points by the projection matrix and return the clipping result returns txs, tys, tzs, tis )r\rZrgrrrproj_transform_clips  ricCstt||Srd)r column_stackproj_trans_points)pointsrDrrr proj_pointssrmcCst|\}}}t||||Srd)ziprh)rlrDr^r_r`rrrrksrkcCsNt|t|}}tgdd|| dgd||dggdg}t||S)N)rrrrrr)rr+r*rr7)r9alphacosasinaM1rrrrot_xs    rsrd)__doc__numpyr numpy.linalgr(rr'r5r>rErFrOrPrWrZrcr\rh transformrirmrkrsrrrrs,  '