a Sicz@sdZddlZddlZddlmZmZmZm Z m Z m Z ddlmZmZmZmZddlmZddlmZddlmZd d Zd d Zd dZGddde jZd?ddZGddde jZ d@ddZ!dAddZ"dBddZ#dCddZ$dDddZ%Gd d!d!eZ&dEd"d#Z'Gd$d%d%eZ(Gd&d'd'e(Z)d(d)Z*dFd*d+Z+dGd,d-Z,Gd.d/d/eZ-Gd0d1d1eZ.dHd3d4Z/Gd5d6d6eZ0dId7d8Z1d9d:Z2d;d<Z3d=d>Z4dS)Jzx Module containing 3D artist code and functions to convert 2D artists into 3D versions which can be added to an Axes3D. N)artistcbookcolorslinestextpath)LineCollectionPolyCollectionPatchCollectionPathCollection) Normalize)Patch)proj3dcCs |dd}|dkr|d}|S)z?Return the given angle normalized to -180 < *a* <= 180 degrees.iharrV/var/www/html/django/DPS/env/lib/python3.9/site-packages/mpl_toolkits/mplot3d/art3d.py _norm_angles rcCs |dd}|dkr|d}|S)z=Return the given angle normalized to -90 < *a* <= 90 degrees.rZrrrrr_norm_text_angles rcCst|dkrtdS|dkr$tdS|dkr6tdS|durHtdSt|rht|d krht|Std dS) a Return a direction vector. Parameters ---------- zdir : {'x', 'y', 'z', None, 3-tuple} The direction. Possible values are: - 'x': equivalent to (1, 0, 0) - 'y': equivalent to (0, 1, 0) - 'z': equivalent to (0, 0, 1) - *None*: equivalent to (0, 0, 0) - an iterable (x, y, z) is converted to a NumPy array, if not already Returns ------- x, y, z : array-like The direction vector. x)rrry)rrrz)rrrN)rrrz2'x', 'y', 'z', None or vector of length 3 expected)nparrayiterablelen ValueErrorzdirrrrget_dir_vector's     r#cs^eZdZdZdddZddZdfd d Zd d ZdddZe j ddZ dddZ Z S)Text3Da Text object with 3D position and direction. Parameters ---------- x, y, z The position of the text. text : str The text string to display. zdir : {'x', 'y', 'z', None, 3-tuple} The direction of the text. See `.get_dir_vector` for a description of the values. Other Parameters ---------------- **kwargs All other parameters are passed on to `~matplotlib.text.Text`. rrcKs*tjj||||fi||||dSN)mtextText__init__set_3d_properties)selfrrrrr"kwargsrrrr)]szText3D.__init__cCs|j|j|jfS)z*Return the (x, y, z) position of the text.)_x_y_zr+rrrget_position_3daszText3D.get_position_3dNcs8t|dd||d|dur4t||_dS)a? Set the (*x*, *y*, *z*) position of the text. Parameters ---------- xyz : (float, float, float) The position in 3D space. zdir : {'x', 'y', 'z', None, 3-tuple} The direction of the text. If unspecified, the zdir will not be changed. N)super set_positionset_zr#_dir_vec)r+xyzr" __class__rrset_position_3des zText3D.set_position_3dcCs||_d|_dS)zl Set the *z* position of the text. Parameters ---------- z : float TN)r/stale)r+rrrrr5vsz Text3D.set_zcCs||_t||_d|_dSNT)r/r#r6r;)r+rr"rrrr*s zText3D.set_3d_propertiescCst|j|j|jf}t|||jg|jj }|dd|dd}|dd|dd}t t ||}t j||dd|ddt|dtj||Wdn1s0Yd|_dS)Nrr)r-r. _rotationF)rrr-r.r/rproj_trans_pointsr6axesMmathdegreesatan2r _setattr_cmrr'r(drawr;)r+rendererZ position3dZprojdxdyanglerrrrEs,z Text3D.drawcCsdSr&rr+rFrrr get_tightbboxszText3D.get_tightbbox)rrrr%r)N)rr)N)__name__ __module__ __qualname____doc__r)r1r:r5r*rallow_rasterizationrErK __classcell__rrr8rr$Is   r$rcCst|_|||dS)z"Convert a Text to a Text3D object.N)r$r9r*)objrr"rrr text_2d_to_3dsrScsLeZdZdZfddZdddZdd Zd d Zej fd d Z Z S)Line3Dz 3D line object. cs,tjggg|Ri||||f|_dS)zU Keyword arguments are passed onto :func:`~matplotlib.lines.Line2D`. N)r3r)_verts3d)r+xsyszsargsr,r8rrr)szLine3D.__init__rrcCsH|}|}t|}t|t|}t|||||_ d|_ dSr<) get_xdata get_ydatar_to_unmasked_float_arrayravelr broadcast_tor juggle_axesrUr;)r+rXr"rVrWrrrr*s zLine3D.set_3d_propertiescGs(t|dkr|d|_n||_d|_dS)ag Set the x, y and z data Parameters ---------- x : array-like The x-data to be plotted. y : array-like The y-data to be plotted. z : array-like The z-data to be plotted. Notes ----- Accepts x, y, z arguments or a single array-like (x, y, z) rrTN)rrUr;)r+rYrrr set_data_3ds  zLine3D.set_data_3dcCs|jS)z Get the current data Returns ------- verts3d : length-3 tuple or array-like The current data as a tuple or array-like. )rUr0rrr get_data_3ds zLine3D.get_data_3dcsH|j\}}}t||||jj\}}}|||t|d|_dS)NF) rUrZproj_transformr?r@set_datar3rEr;)r+rFZxs3dZys3dZzs3drVrWrXr8rrrEs    z Line3D.draw)rr) rLrMrNrOr)r*r`rarrPrErQrrr8rrTs   rTcCst|_|||dS)zConvert a 2D line to 3D.N)rTr9r*)linerXr"rrr line_2d_to_3dsrdcsHt|t|}|jddd}ddt||D}fdd|D}|S)zConvert a path to a 3D segment.FsimplifycurvescSs"g|]\\\}}}}|||fqSrr.0rrcoderrrr z'_path_to_3d_segment..cs g|]\}}}t|||qSrr_rirrrr!rrrkrl)rr^r iter_segmentszip)rrXr"pathsegssegseg3drr!r_path_to_3d_segments rtcsPt|st|t|}nt|t|kr4tdfddt||D}|S)z6Convert paths from a collection object to 3D segments.z-Number of z-coordinates does not match paths.csg|]\}}t||qSr)rtrirZpathzr!rrrksz)_paths_to_3d_segments..)rrr^rr rp)pathsrXr"segsrr!r_paths_to_3d_segmentss  rxcsjt|t|}|jddd}ddt||D}|rVt|\}}fdd|D}ng}g}|t|fS)z/Convert a path to a 3D segment with path codes.FrecSs&g|]\\\}}}}|||f|fqSrrrhrrrrkrlz2_path_to_3d_segment_with_codes..cs g|]\}}}t|||qSrrmrnr!rrrkrl)rr^rrorplist)rrXr"rqZ seg_codesrrcodesrsrr!r_path_to_3d_segment_with_codess r{csTt|t|}fddt||D}|r:t|\}}n gg}}t|t|fS)zP Convert paths from a collection object to 3D segments with path codes. csg|]\}}t||qSr)r{rur!rrrksz4_paths_to_3d_segments_with_codes..)rr^rrpry)rvrXr"Zsegments_codessegmentsrzrr!r _paths_to_3d_segments_with_codes s  r}cs0eZdZdZddZfddZddZZS)Line3DCollectionz# A collection of 3D lines. cCs||_d|_dSz&Set the position to use for z-sorting.TN _sort_zposr;r+valrrr set_sort_zpos szLine3DCollection.set_sort_zposcs||_tgdS)z" Set 3D segments. N) _segments3dr3 set_segments)r+r|r8rrr%szLine3DCollection.set_segmentscsTfddjD}dd|D}t|d}|D]\}}}t|t|}q6|S)zB Project the points according to renderer matrix. csg|]}t|jjqSr)rr>r?r@)ripointsr0rrrk0sz5Line3DCollection.do_3d_projection..cSs g|]\}}}t||gqSr)r column_stack)rirVrWrXrrrrk2rlgeA)rrrmin)r+Zxyslist segments_2dminzrVrWrXrr0rdo_3d_projection,s  z!Line3DCollection.do_3d_projection)rLrMrNrOrrrrQrrr8rr~s r~cCs$t|||}t|_||dS)z6Convert a LineCollection to a Line3DCollection object.N)rx get_pathsr~r9r)colrXr" segments3drrrline_collection_2d_to_3d<srcsBeZdZdZdddfdd Zddd Zd d Zd d ZZS)Patch3Dz 3D patch object. rrrXr"cs"tj|i||||dSr&)r3r)r*)r+rXr"rYr,r8rrr)HszPatch3D.__init__rcs.t|t|}fddt||D|_dS)Ncs"g|]\\}}}t|||qSrrmrnr!rrrkNs z-Patch3D.set_3d_properties..)rr^rrp _segment3d)r+vertsrXr"rr!rr*Ls zPatch3D.set_3d_propertiescCs|jSr&)_path2dr0rrrget_pathQszPatch3D.get_pathc CsN|j}t|\}}}t||||jj\}}}}tt ||g|_ t |Sr&) rrprproj_transform_clipr?r@mpathPathrrrr r+srVrWrXvxsvysvzsvisrrrrTs  zPatch3D.do_3d_projection)rr) rLrMrNrOr)r*rrrQrrr8rrCs  rc@s2eZdZdZdddddZd dd Zd d Zd S) PathPatch3Dz 3D PathPatch object. rrrcKs$tj|fi|||||dSr&)r r)r*)r+rrXr"r,rrrr)bszPathPatch3D.__init__rcCs tj||j||d|j|_dS)Nr)rr*verticesrz_code3d)r+rrXr"rrrr*gszPathPatch3D.set_3d_propertiesc CsR|j}t|\}}}t||||jj\}}}}tt ||g|j |_ t |Sr&) rrprrr?r@rrrrrrrrrrrrks  zPathPatch3D.do_3d_projectionN)rr)rLrMrNrOr)r*rrrrrr]s rcCs4|}|}||}t|r*|dStgS)z2Return a list of vertices for the path of a patch.r)get_patch_transformr to_polygonsrrr)patchtransrpolygonsrrr_get_patch_vertsts rcCs t|}t|_||||dS)z$Convert a Patch to a Patch3D object.N)rrr9r*)rrr"rrrrpatch_2d_to_3d|srcCs2|}|}||}t|_||||dS)z,Convert a PathPatch to a PathPatch3D object.N)rrtransform_pathrr9r*)Z pathpatchrr"rrrrrrpathpatch_2d_to_3ds  rcsveZdZdZddddfdd Zdd Zd d Zd d ZddZfddZ ddZ fddZ fddZ Z S)Patch3DCollectionz% A collection of 3D patches. rrTrXr" depthshadecs(||_tj|i||||dS)a Create a collection of flat 3D patches with its normal vector pointed in *zdir* direction, and located at *zs* on the *zdir* axis. 'zs' can be a scalar or an array-like of the same length as the number of patches in the collection. Constructor arguments are the same as for :class:`~matplotlib.collections.PatchCollection`. In addition, keywords *zs=0* and *zdir='z'* are available. Also, the keyword argument *depthshade* is available to indicate whether or not to shade the patches in order to give the appearance of depth (default is *True*). This is typically desired in scatter plots. N) _depthshader3r)r*r+rXr"rrYr,r8rrr)szPatch3DCollection.__init__cCs|jSr&rr0rrrget_depthshadesz Patch3DCollection.get_depthshadecCs||_d|_dSz Set whether depth shading is performed on collection members. Parameters ---------- depthshade : bool Whether to shade the patches in order to give the appearance of depth. TNrr;r+rrrrset_depthshades z Patch3DCollection.set_depthshadecCs||_d|_dSrrrrrrrszPatch3DCollection.set_sort_zposcCs`||}t|dkr(|j\}}ng}g}t||t|||_td|_ d|_ d|_ dSNrT) update_scalarmappable get_offsetsrTr_r atleast_1d _offsets3dslice_z_markers_idx_vzsr;r+rXr"offsetsrVrWrrrr*s   z#Patch3DCollection.set_3d_propertiescs`|j\}}}t||||jj\}}}}||_tt ||g|j dkrVt |Stj SdSNr) rrrr?r@rr3 set_offsetsrrsizernan)r+rVrWrXrrrrr8rrrs    z"Patch3DCollection.do_3d_projectioncCsD|jdur|jrt||jn|}t|dkr6||j}t||jSNrrr_zalpharrmcolors to_rgba_array_alphar+Z color_arrayrrr"_maybe_depth_shade_and_sort_colorss  z4Patch3DCollection._maybe_depth_shade_and_sort_colorscs|tSr&rr3 get_facecolorr0r8rrrszPatch3DCollection.get_facecolorcs&t|jdr|S|tSNfacer _str_equal _edgecolorsrrr3 get_edgecolorr0r8rrrszPatch3DCollection.get_edgecolor)rLrMrNrOr)rrrr*rrrrrQrrr8rrs   rcseZdZdZddddfdd Zfdd Zd d Zd d Zdfdd ZfddZ ddZ ddZ ddZ ddZ fddZfddZZS) Path3DCollectionz# A collection of 3D paths. rrTrcs.||_d|_tj|i||||dS)a Create a collection of flat 3D paths with its normal vector pointed in *zdir* direction, and located at *zs* on the *zdir* axis. 'zs' can be a scalar or an array-like of the same length as the number of paths in the collection. Constructor arguments are the same as for :class:`~matplotlib.collections.PathCollection`. In addition, keywords *zs=0* and *zdir='z'* are available. Also, the keyword argument *depthshade* is available to indicate whether or not to shade the patches in order to give the appearance of depth (default is *True*). This is typically desired in scatter plots. FN)r_in_drawr3r)r*rr8rrr)szPath3DCollection.__init__cs>tj|ddt|Wdn1s00YdS)NT)r)rrDr3rErJr8rrrEszPath3DCollection.drawcCs||_d|_dSrrrrrrr szPath3DCollection.set_sort_zposcCs||}t|dkr(|j\}}ng}g}t||t|||_|j|_ t |j |_ |j\}}}t d|_d|_d|_dSr)rrrrr_rrr_sizes_sizes3dr _linewidths _linewidths3drrrr;rrrrr*s     z"Path3DCollection.set_3d_propertiesR@cst|||js||_dSr&)r3 set_sizesrr)r+sizesdpir8rrr/szPath3DCollection.set_sizescs$t||js t|j|_dSr&)r3 set_linewidthrrrrr)r+lwr8rrr4s zPath3DCollection.set_linewidthcCs|jSr&rr0rrrr9szPath3DCollection.get_depthshadecCs||_d|_dSrrrrrrr<s zPath3DCollection.set_depthshadec Cs|j\}}}t||||jj\}}}}t|ddd}|_||_t |j dkrb|j ||_ t |j dkr||j ||_ ||}||}||}t|t||f|jrt|StjS)Nrr)rrrr?r@rargsortrrrrrrrr rrrrr) r+rVrWrXrrrrZ z_markers_idxrrrrIs     z!Path3DCollection.do_3d_projectioncCsD|jdur|jrt||jn|}t|dkr6||j}t||jSrrrrrrrgs  z3Path3DCollection._maybe_depth_shade_and_sort_colorscs|tSr&rr0r8rrrqszPath3DCollection.get_facecolorcs&t|jdr|S|tSrrr0r8rrrtszPath3DCollection.get_edgecolor)r)rLrMrNrOr)rErr*rrrrrrrrrQrrr8rrs     rTcCs>t|trt|_nt|tr"t|_||_d|_|||dS)a Convert a :class:`~matplotlib.collections.PatchCollection` into a :class:`Patch3DCollection` object (or a :class:`~matplotlib.collections.PathCollection` into a :class:`Path3DCollection` object). Parameters ---------- za The location or locations to place the patches in the collection along the *zdir* axis. Default: 0. zdir The axis in which to place the patches. Default: "z". depthshade Whether to shade the patches to give a sense of depth. Default: *True*. FN) isinstancer rr9r rrrr*)rrXr"rrrrpatch_collection_2d_to_3d}s  rcseZdZdZddfdd ZejejejdZ ddZ d d Z d fd d Z ddZ ddZddZddZfddZfddZddZddZddZZS)!Poly3DCollectiona A collection of 3D polygons. .. note:: **Filling of 3D polygons** There is no simple definition of the enclosed surface of a 3D polygon unless the polygon is planar. In practice, Matplotlib fills the 2D projection of the polygon. This gives a correct filling appearance only for planar polygons. For all other polygons, you'll find orientations in which the edges of the polygon intersect in the projection. This will lead to an incorrect visualization of the 3D area. If you need filled areas, it is recommended to create them via `~mpl_toolkits.mplot3d.axes3d.Axes3D.plot_trisurf`, which creates a triangulation and thus generates consistent surfaces. average)zsortcshtj|g|Ri|t|tjr:|jdkrTtdntdd|DrTtd||d|_ dS)aS Parameters ---------- verts : list of (N, 3) array-like Each element describes a polygon as a sequence of ``N_i`` points ``(x, y, z)``. zsort : {'average', 'min', 'max'}, default: 'average' The calculation method for the z-order. See `~.Poly3DCollection.set_zsort` for details. *args, **kwargs All other parameters are forwarded to `.PolyCollection`. Notes ----- Note that this class does a bit of magic with the _facecolors and _edgecolors properties. rz)verts must be a list of (N, 3) array-likecss |]}tt|dkVqdS)r2N)rrshape)rivertrrr rlz,Poly3DCollection.__init__..N) r3r)rrndarrayndimr any set_zsort_codes3d)r+rrrYr,r8rrr)s    zPoly3DCollection.__init__)rrmaxcCs|j||_d|_d|_dS)a Set the calculation method for the z-order. Parameters ---------- zsort : {'average', 'min', 'max'} The function applied on the z-coordinates of the vertices in the viewer's coordinate system, to determine the z-order. NT)_zsort_functions _zsortfuncrr;)r+rrrrrs zPoly3DCollection.set_zsortcCst|rt|j\}}}nggg}}}tt|}t||||g|_dgtdd|D}gtt |dd|dd|_ dS)zOptimize points for projection.rcSsg|] }t|qSr)r)risegmentrrrrkrlz/Poly3DCollection.get_vector..Nrr) rr row_stackronesr_veccumsummapr _segslices)r+rrVrWrXrindicesrrr get_vectorszPoly3DCollection.get_vectorTcs"||tgd||_dS)zSet 3D vertices.FN)rr3 set_verts_closed)r+rclosedr8rrrs zPoly3DCollection.set_vertscCs|j|dd||_dS)z Set 3D vertices with path codes.F)rN)rr)r+rrzrrrset_verts_and_codessz$Poly3DCollection.set_verts_and_codescCsF|d|_|dt||_t||_t||_ d|_ dS)NrT) rrrr r _facecolor3dr _edgecolor3d get_alphaZ_alpha3dr;r0rrrr*s    z"Poly3DCollection.set_3d_propertiescCs||_d|_dSrrrrrrrszPoly3DCollection.set_sort_zposc sjdur.jr j_jr.j_t j j j \fddj D}j}j}t|t|kr|jt|dd}t|t|krt|dkr|}n|jt|dd}|r tfddtt|||Ddd d d }t|\}}__}n g}td _td _g}jdurXfd d|D}t||nt|jtjt|krj_jdurtdgdgjgdgg} t | j j } | ddSjdkrtStj SdS)z< Perform the 3D projection for this object. Ncs"g|]}|||fqSrr)risl)txstystzsrrrkrlz5Poly3DCollection.do_3d_projection..r)axisc3s<|]4\}\\}}}}}|t||g|||fVqdSr&)rrr)riidxrVrWrXfcecr0rrr*sz4Poly3DCollection.do_3d_projection..cSs|dSrr)rrrr-rlz3Poly3DCollection.do_3d_projection..T)keyreversercsg|]}j|qSr)r)rir r0rrrk8rlrr2)!_Ar_face_is_mapped _facecolorsr_edge_is_mappedrrrZ_proj_transform_vecrr?r@rrrepeatsorted enumeraterp _facecolors2d _edgecolors2dremptyrr rrrrrrrr) r+ZxyzlistZcfaceZcedgeZ z_segments_2d_ridxsrzZzvecZztransr)r+rrrrrsT          z!Poly3DCollection.do_3d_projectioncst|t||_dSr&)r3 set_facecolorr rrr+rr8rrrMs zPoly3DCollection.set_facecolorcst|t||_dSr&)r3 set_edgecolorr rrrr8rrrRs zPoly3DCollection.set_edgecolorc Csxtj||zt|j|j|_Wnttt fy<Yn0zt|j |j|_ Wnttt fylYn0d|_ dSr<) rArtist set_alpharrrrAttributeError TypeError IndexErrorrrr;)r+alpharrrr!Ws  zPoly3DCollection.set_alphacCs&t|ds |j|j_||jS)Nr)hasattrr?get_projr@rrr0rrrrfs zPoly3DCollection.get_facecolorcCs&t|ds |j|j_||jS)Nr)r&r?r'r@rrr0rrrrns zPoly3DCollection.get_edgecolor)T)rLrMrNrOr)rrrrrrrrrr*rrrrr!rrrQrrr8rrs$  E  rcCs2t|||\}}t|_||||dS)z6Convert a PolyCollection to a Poly3DCollection object.N)r}rrr9rr*)rrXr"Z segments_3drzrrrpoly_collection_2d_to_3dws   r(cCsL|dkr|||fS|dkr$|||fS|ddkr>t||||S|||fSdS)z Reorder coordinates so that 2D xs, ys can be plotted in the plane orthogonal to zdir. zdir is normally x, y or z. However, if zdir starts with a '-' it is interpreted as a compensation for rotate_axes. rrr-N) rotate_axesrVrWrXr"rrrr_s   r_cCsV|dkr|||fS|dkr$|||fS|dkr6|||fS|dkrH|||fS|||fSdS)z Reorder coordinates so that the axes are rotated with zdir along the original z axis. Prepending the axis with a '-' does the inverse transform, so zdir can be x, -x, y, -y, z or -z rz-xrz-yNrr+rrrr*s    r*cCst|dkst|dkr"tdStt|t|}d||d}tt|t|df}t |ddddf|dddf|gS)z7Modify the alphas of the color list according to depth.rrrgffffff?rNr) rrzerosr rrr^rrr)rrXnormZsatsrgbarrrrs  r)rr)rr)rr)rr)rr)rr)rr)rr)rr)rrT)rr)5rOrAnumpyr matplotlibrrrrrrr'rrmatplotlib.collectionsrr r r matplotlib.colorsr matplotlib.patchesr r%rrrr#r(r$rSLine2DrTrdrtrxr{r}r~rrrrrrrrrrr(r_r*rrrrrsF    "P ?    !   _ `