B ]t\/a@sdZddlZddlZddlmZmZmZm Z m Z m Z ddlmZmZmZmZmZddlmZddlmZddlmZd d Zd d Zd dZGddde jZdAddZGddde j Z!dBddZ"dCddZ#dDddZ$dEddZ%dFddZ&Gd d!d!eZ'dGd"d#Z(Gd$d%d%eZ)Gd&d'd'e)Z*d(d)Z+dHd*d+Z,dId,d-Z-Gd.d/d/eZ.Gd0d1d1eZ/dJd3d4Z0Gd5d6d6eZ1dKd7d8Z2d9d:Z3d;d<Z4d=d>Z5d?d@Z6dS)Lzx Module containing 3D artist code and functions to convert 2D artists into 3D versions which can be added to an Axes3D. N)artistcbookcolorslinestextpath) CollectionLineCollectionPolyCollectionPatchCollectionPathCollection) Normalize)Patch)proj3dcCs |dd}|dkr|d}|S)z?Return the given angle normalized to -180 < *a* <= 180 degrees.ih)arr9lib/python3.7/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.rZr)rrrrnorm_text_angle s rcCsn|dkrtdS|dkr$tdS|dkr6tdS|dkrHtdSt|rbt|d krb|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': euqivalent to (0, 1, 0) - 'z': equivalent to (0, 0, 1) - *None*: euqivalent to (0, 0, 0) - an iterable (x, y, z) is returned unchanged. Returns ------- x, y, z : array-like The direction vector. This is either a numpy.array or *zdir* itself if *zdir* is already a length-3 iterable. x)rrry)rrrz)rrrN)rrrz2'x', 'y', 'z', None or vector of length 3 expected)nparrayriterablelen ValueError)zdirrrrget_dir_vector(s    r"c@s:eZdZdZdddZdddZejd d Zd d Z d 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||||f||||dS)N)mtextText__init__set_3d_properties)selfrrrrr!kwargsrrrr'_szText3D.__init__cCs2|\}}t|||f|_t||_d|_dS)NT)Z get_positionrr _position3dr"_dir_vecstale)r)rr!rrrrrr(cs  zText3D.set_3d_propertiescCst|j|j|jg|j}|dd|dd}|dd|dd}|dkrb|dkrbd}ntt||}||dd|ddf| t |t j ||d|_dS)NrrgF)rproj_trans_pointsr+r,MmathZdegreesZatan2Z set_positionZ set_rotationrr%r&drawr-)r)rendererZprojZdxZdyZanglerrrr1isz Text3D.drawcCsdS)Nr)r)r2rrr get_tightbboxyszText3D.get_tightbboxN)rrrr$r)rr) __name__ __module__ __qualname____doc__r'r(rallow_rasterizationr1r3rrrrr#Ks   r#rcCst|_|||dS)z"Convert a Text to a Text3D object.N)r# __class__r()objrr!rrr text_2d_to_3dsr;c@s0eZdZdZddZd ddZejdd Zd S) Line3Dz 3D line object. cOs(tjj|ggf|||||f|_dS)zU Keyword arguments are passed onto :func:`~matplotlib.lines.Line2D`. N)rLine2Dr'_verts3d)r)xsyszsargsr*rrrr'szLine3D.__init__rrcCsV|}|}ytj|t|d}Wntk r:YnXt|||||_d|_dS)N)Z fill_valueT) Z get_xdataZ get_ydatarZ full_likefloat TypeError juggle_axesr>r-)r)rAr!r?r@rrrr(szLine3D.set_3d_propertiescCsH|j\}}}t||||j\}}}|||tj||d|_dS)NF) r>rZproj_transformr/set_datarr=r1r-)r)r2Zxs3dZys3dZzs3dr?r@rArrrr1s   z Line3D.drawN)rr) r4r5r6r7r'r(rr8r1rrrrr<s r<cCst|_|||dS)zConvert a 2D line to 3D.N)r<r9r()linerAr!rrr line_2d_to_3dsrHcsHt|t|}|jddd}ddt||D}fdd|D}|S)zConvert a path to a 3D segment.F)simplifycurvescSs"g|]\\\}}}}|||fqSrr).0rrcoderrrr sz&path_to_3d_segment..cs g|]\}}}t|||qSr)rE)rKrrr)r!rrrMs)r broadcast_tor iter_segmentszip)rrAr!pathsegssegseg3dr)r!rpath_to_3d_segments rTcs,t|t|}fddt||D}|S)z6Convert paths from a collection object to 3D segments.csg|]\}}t||qSr)rT)rKrpathz)r!rrrMsz(paths_to_3d_segments..)rrNrrP)pathsrAr!segsr)r!rpaths_to_3d_segmentss rXc szt|t|}g}g}|jddd}x8t||D]*\\\}}}} |||| f||q2Wfdd|D} | |fS)z/Convert a path to a 3D segment with path codes.F)rIrJcs g|]\}}}t|||qSr)rE)rKrrr)r!rrrMsz1path_to_3d_segment_with_codes..)rrNrrOrPappend) rrAr!rRcodesrQrrrLrrSr)r!rpath_to_3d_segment_with_codessr[c Cs\t|t|}g}g}x:t||D],\}}t|||\}}||||q$W||fS)zP Convert paths from a collection object to 3D segments with path codes. )rrNrrPr[rY) rVrAr!segmentsZ codes_listrrUrWrZrrrpaths_to_3d_segments_with_codess r]c@s8eZdZdZddZddZddZejd d d Z d S) Line3DCollectionz# A collection of 3D lines. cCs||_d|_dS)z&Set the position to use for z-sorting.TN) _sort_zposr-)r)valrrr set_sort_zposszLine3DCollection.set_sort_zposcCst||_t|gdS)z" Set 3D segments. N)rZ asanyarray _segments3dr set_segments)r)r\rrrrcs zLine3DCollection.set_segmentscsXfdd|jD}dd|D}t||d}x |D]\}}}t|t|}q8W|S)zB Project the points according to renderer matrix. csg|]}t|jqSr)rr.r/)rKpoints)r2rrrMsz5Line3DCollection.do_3d_projection..cSs g|]\}}}t||gqSr)r column_stack)rKr?r@rArrrrMsgeA)rbr rcmin)r)r2Zxyslist segments_2dZminzr?r@rAr)r2rdo_3d_projections   z!Line3DCollection.do_3d_projectionFcCs|r||t||dS)N)rhr r1)r)r2Zprojectrrrr1s zLine3DCollection.drawN)F) r4r5r6r7rarcrhrr8r1rrrrr^s r^cCs$t|||}t|_||dS)z6Convert a LineCollection to a Line3DCollection object.N)rX get_pathsr^r9rc)colrAr! segments3drrrline_collection_2d_to_3dsrlc@sBeZdZdZdddddZddd Zd d Zd d ZddZdS)Patch3Dz 3D patch object. rr)rAr!cOs"tj|f|||||dS)N)rr'r()r)rAr!rBr*rrrr'szPatch3D.__init__rcs:t|t|}fddt||D|_t||_dS)Ncs"g|]\\}}}t|||qSr)rE)rKrrr)r!rrrMsz-Patch3D.set_3d_properties..)rrNrrP _segment3dr get_facecolor _facecolor3d)r)vertsrAr!r)r!rr(s zPatch3D.set_3d_propertiescCs|jS)N)_path2d)r)rrrget_pathszPatch3D.get_pathcCs|jS)N) _facecolor2d)r)rrrroszPatch3D.get_facecolorc CsT|j}t|\}}}t||||j\}}}} tt||g|_ |j |_ t |S)N) rnrPrproj_transform_clipr/mpathPathrrerrrprtrf) r)r2sr?r@rAvxsvysvzsvisrrrrh"s zPatch3D.do_3d_projectionN)rr) r4r5r6r7r'r(rsrorhrrrrrm s  rmc@s2eZdZdZdddddZd dd Zd d Zd S) PathPatch3Dz 3D PathPatch object. rr)rAr!cKs tj|f|||||dS)N)rr'r()r)rrAr!r*rrrr'1szPathPatch3D.__init__rcCs tj||j||d|j|_dS)N)rAr!)rmr(ZverticesrZ_code3d)r)rrAr!rrrr(5szPathPatch3D.set_3d_propertiesc CsX|j}t|\}}}t||||j\}}}} tt||g|j |_ |j |_ t |S)N)rnrPrrur/rvrwrrer~rrrprtrf) r)r2rxr?r@rAryrzr{r|rrrrh9s zPathPatch3D.do_3d_projectionN)rr)r4r5r6r7r'r(rhrrrrr},s r}cCs2|}|}||}t|r*|dSgSdS)z2Return a list of vertices for the path of a patch.rN)get_patch_transformrsZ to_polygonsr)patchtransrZpolygonsrrrget_patch_vertsCs  rcCs t|}t|_||||dS)z$Convert a Patch to a Patch3D object.N)rrmr9r()rrr!rqrrrpatch_2d_to_3dNsrcCs2|}|}||}t|_||||dS)z,Convert a PathPatch to a PathPatch3D object.N)rsrZtransform_pathr}r9r()Z pathpatchrr!rrrvrrrpathpatch_2d_to_3dUs  rcsBeZdZdZddddfdd Zdd Zd d Zd d ZZS)Patch3DCollectionz% A collection of 3D patches. rrT)rAr! depthshadecs$||_tj|||||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) _depthshadesuperr'r()r)rAr!rrBr*)r9rrr'dszPatch3DCollection.__init__cCs||_d|_dS)z&Set the position to use for z-sorting.TN)r_r-)r)r`rrrraxszPatch3DCollection.set_sort_zposcCsd||}t|dkr(|j\}}ng}g}t||t|||_||_ | |_ d|_ dS)NrT) update_scalarmappable get_offsetsrTrEr atleast_1d _offsets3drorp get_edgecolor _edgecolor3dr-)r)rAr!offsetsr?r@rrrr(}s    z#Patch3DCollection.set_3d_propertiesc Cs|j\}}}t||||j\}}}}|jr8t|j|n|j} t| |j } | | |jrht|j |n|j } t| |j } | | t |t||g|jdkrt|StjSdS)Nr)rrrur/rzalpharpmcolors to_rgba_array_alphaset_facecolorsrset_edgecolorsr set_offsetsrresizerfnan) r)r2r?r@rAryrzr{r|fcsecsrrrrhs    z"Patch3DCollection.do_3d_projection) r4r5r6r7r'rar(rh __classcell__rr)r9rr_s rcsBeZdZdZddddfdd Zdd Zd d Zd d ZZS)Path3DCollectionz# A collection of 3D paths. rrT)rAr!rcs$||_tj|||||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. N)rrr'r()r)rAr!rrBr*)r9rrr'szPath3DCollection.__init__cCs||_d|_dS)z&Set the position to use for z-sorting.TN)r_r-)r)r`rrrraszPath3DCollection.set_sort_zposcCsd||}t|dkr(|j\}}ng}g}t||t|||_||_ | |_ d|_ dS)NrT) rrrrrErrrrorprrr-)r)rAr!rr?r@rrrr(s    z"Path3DCollection.set_3d_propertiesc Cs|j\}}}t||||j\}}}}|jr8t|j|n|j} t| |j } | | |jrht|j |n|j } t| |j } | | t |t||g|jdkrt|StjSdS)Nr)rrrur/rrrprrrrrrr rrrerrfr) r)r2r?r@rAryrzr{r|rrrrrrhs    z!Path3DCollection.do_3d_projection) r4r5r6r7r'rar(rhrrr)r9rrs rTcCs8t|trt|_nt|tr"t|_||_|||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*. N) isinstancer rr9r rrr()rjrAr!rrrrpatch_collection_2d_to_3ds   rcseZdZdZddfdd ZejejejdZ ddZ d d Z dd d Z d dZ ddZddZddZddZddZddZddZddZZS) Poly3DCollectionz& A collection of 3D polygons. T)zsortcs(tj|f||||d|_dS)a  Create a Poly3DCollection. *verts* should contain 3D coordinates. Keyword arguments: zsort, see set_zsort for options. Note that this class does a bit of magic with the _facecolors and _edgecolors properties. N)rr' set_zsort_codes3d)r)rqrrBr*)r9rrr's  zPoly3DCollection.__init__)averagerfmaxcCsP|dkr d}|dk r0||jkr*|j|}q4dSnd}||_d|_||_d|_dS)au Sets the calculation method for the z-order. Parameters ---------- zsort : bool or {'average', 'min', 'max'} For 'average', 'min', 'max' the z-order is determined by applying the function to the z-coordinates of the vertices in the viewer's coordinate system. *True* is equivalent to 'average'. TrFN)_zsort_functions_zsortr_ _zsortfuncr-)r)rZ zsortfuncrrrrs   zPoly3DCollection.set_zsortc Csd}d}g}g}x4|D],}|||t|}|||f|}qWt|r^t|\}}} nggg}}} tt|} t||| | g|_||_dS)zOptimize points for projection.rN) extendrrYrPronesr_vec_segis) r)rksieiZsegisrdpr?r@rArrrr get_vector4s   zPoly3DCollection.get_vectorcCs"||t|gd||_dS)zSet 3D vertices.FN)rr set_verts_closed)r)rqclosedrrrrJs zPoly3DCollection.set_vertscCs|j|dd||_dS)z!Sets 3D vertices with path codes.F)rN)rr)r)rqrZrrrset_verts_and_codesQsz$Poly3DCollection.set_verts_and_codescCsF|d|_|dt||_t||_t||_ d|_ dS)NT) rr_rr ro _facecolors3dr _edgecolors3dZ get_alphaZ_alpha3dr-)r)rrrr(Ys    z"Poly3DCollection.set_3d_propertiescCs||_d|_dS)z&Set the position to use for z-sorting.TN)r_r-)r)r`rrrradszPoly3DCollection.set_sort_zposc sjdk rj_tj|j\fddjD}j}j }t |t |krv|j t |dd}t |t |krt |dkr|}n|j t |dd}j rt fddtt|||Ddd d d }ntd d d|D}jdk r"fdd|D}t||nt|jdd|D_t j t |krhdd|D_nj _jdk rtdgdgjgdgg}t||j} | ddSjdkrtStjSdS)z< Perform the 3D projection for this object. Ncs2g|]*\}}||||||fqSrr)rKrr)txstystzsrrrMssz5Poly3DCollection.do_3d_projection..r)Zaxisc3s<|]4\}\\}}}}}|t||g|||fVqdS)N)rrre)rKidxr?r@rAfcec)r)rr sz4Poly3DCollection.do_3d_projection..cSs|dS)Nrr)rrrrsz3Poly3DCollection.do_3d_projection..T)keyreverseZwhoopscSsg|]\}}}}}|qSrr)rKrrxrrrrrrrMscs g|]\}}}}}j|qSr)r)rKrrxrrr)r)rrrMscSsg|]\}}}}}|qSrr)rKrrxrrrrrrrMscSsg|]\}}}}}|qSrr)rKrrxrrrrrrrMsr)Z_Ar _facecolorsrrZproj_transform_vecrr/rrrrepeatrsorted enumeraterPr rr rrr _facecolors2d _edgecolors2dr_rrrrfr) r)r2ZxyzlistZcfaceZcedgeZ z_segments_2drgrZZzvecZztransr)r)rrrrrhisH         z!Poly3DCollection.do_3d_projectioncCst||t||_dS)N)r set_facecolorror)r)rrrrrs zPoly3DCollection.set_facecolorcCst||t||_dS)N)r set_edgecolorrr)r)rrrrrs zPoly3DCollection.set_edgecolorc Cs|dk r2y t|Wntk r0tdYnXtj||yt|j|j|_ Wnt tt fk rpYnXyt|j |j|_ Wnt tt fk rYnXd|_dS)z Set the alpha transparencies of the collection. Parameters ---------- alpha : float or None Nzalpha must be a float or NoneT)rCrDrZArtist set_alpharrrrrAttributeError IndexErrorrZ _edgecolorsr-)r)Zalpharrrrs" zPoly3DCollection.set_alphacCs|jS)N)r)r)rrrroszPoly3DCollection.get_facecolorcCs|jS)N)r)r)rrrrszPoly3DCollection.get_edgecolor)T)r4r5r6r7r'rrrfrrrrrrr(rarhrrrrorrrr)r9rrs"   <rcCs2t|||\}}t|_||||dS)z6Convert a PolyCollection to a Poly3DCollection object.N)r]rirr9rr()rjrAr!Z segments_3drZrrrpoly_collection_2d_to_3ds   rcCsL|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_axes)r?r@rAr!rrrrEs   rEcCsV|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-yNr)r?r@rAr!rrrrs    rcCs*tt|rt|n ddddg|dfS)z@Stretch the color argument to provide the required number *num*.r)rrNrrr)cZnumrrr get_colorssrcCsPt|t|}t|rLtt|t|}d||d}ddt||D}|S)z7Modify the alphas of the color list according to depth.rgffffff?cSs0g|](\}}|d|d|d|d|fqS)rrrrr)rKrrxrrrrM szzalpha..)rrr rfrrP)rrAZnormZsatsrrrrs r)rr)rr)rr)rr)rr)rr)rr)rr)rr)rrT)rr)7r7r0ZnumpyrZ matplotlibrrrrrrr%rrvZmatplotlib.collectionsrr r r r Zmatplotlib.colorsr Zmatplotlib.patchesrr$rrrr"r&r#r;r=r<rHrTrXr[r]r^rlrmr}rrrrrrrrrErrrrrrrsF    #4 "   (   BB R