U Ev fT@sdZddlZddlZddlmZddlmZmZm Z m Z m Z mZddlmZmZmZmZddlmZddlmZdd lmZd d Zd d ZddZGddde jZdDddZ Gddde j!Z"dEddZ#dFddZ$dGddZ%dHddZ&dIdd Z'Gd!d"d"eZ(dJd#d$Z)Gd%d&d&eZ*Gd'd(d(e*Z+d)d*Z,dKd+d,Z-dLd-d.Z.Gd/d0d0eZ/Gd1d2d2eZ0dMd4d5Z1Gd6d7d7eZ2dNd8d9Z3d:d;Z4dd?Z6d@dAZ7dOdBdCZ8dS)Pzx Module containing 3D artist code and functions to convert 2D artists into 3D versions which can be added to an Axes3D. N)contextmanager)artistcbookcolorslinestextpath)LineCollectionPolyCollectionPatchCollectionPathCollection) Normalize)Patch)proj3dcCs |dd}|dkr|d}|S)z?Return the given angle normalized to -180 < *a* <= 180 degrees.iharr>/tmp/pip-unpacked-wheel-7vhvci0g/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_angle"s rcCst|dkrtdS|dkr$tdS|dkr6tdS|dkrHtdSt|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 : float 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||||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_3ddszText3D.get_position_3dNcs8t|dd||d|dk r4t||_dS)ax 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. See `.get_dir_vector` for a description of the values. N)superZ set_positionset_zr$_dir_vec)r,xyzr# __class__rrset_position_3dhs zText3D.set_position_3dcCs||_d|_dS)zl Set the *z* position of the text. Parameters ---------- z : float TN)r0stale)r,rrrrr5ysz Text3D.set_zcCs||_t||_d|_dS)aB Set the *z* position and direction of the text. Parameters ---------- z : float The z-position in 3D space. zdir : {'x', 'y', 'z', 3-tuple} The direction of the text. Default: 'z'. See `.get_dir_vector` for a description of the values. TN)r0r$r6r;)r,rr#rrrr+s  zText3D.set_3d_propertiesc Cst|j|j|jf}t|||jg|jj }|dd|dd}|dd|dd}t t ||}t j||dd|ddt|dtj||W5QRXd|_dS)Nrr)r.r/Z _rotationF)rrr.r/r0rproj_trans_pointsr6axesMmathdegreesatan2r _setattr_cmrr(r)drawr;)r,rendererZ position3dZprojZdxZdyZanglerrrrCs z Text3D.drawcCsdSr'rr,rDrrr get_tightbboxszText3D.get_tightbbox)rrrr&r)N)rr)N)__name__ __module__ __qualname____doc__r*r2r:r5r+rallow_rasterizationrCrF __classcell__rrr8rr%Ls   r%rcCst|_|||dS)a Convert a `.Text` to a `.Text3D` object. Parameters ---------- z : float The z-position in 3D space. zdir : {'x', 'y', 'z', 3-tuple} The direction of the text. Default: 'z'. See `.get_dir_vector` for a description of the values. N)r%r9r+)objrr#rrr text_2d_to_3ds rNcsLeZdZdZfddZdddZdd Zd d Zej fd d Z Z S)Line3Da 3D line object. .. note:: Use `get_data_3d` to obtain the data associated with the line. `~.Line2D.get_data`, `~.Line2D.get_xdata`, and `~.Line2D.get_ydata` return the x- and y-coordinates of the projected 2D-line, not the x- and y-data of the 3D-line. Similarly, use `set_data_3d` to set the data, not `~.Line2D.set_data`, `~.Line2D.set_xdata`, and `~.Line2D.set_ydata`. cs(tjggf||||||dS)aO Parameters ---------- xs : array-like The x-data to be plotted. ys : array-like The y-data to be plotted. zs : array-like The z-data to be plotted. *args, **kwargs : Additional arguments are passed to `~matplotlib.lines.Line2D`. N)r4r* set_data_3d)r,xsyszsargsr-r8rrr*szLine3D.__init__rrcCsH|}|}t|}t|t|}t|||||_ d|_ dS)a Set the *z* position and direction of the line. Parameters ---------- zs : float or array of floats The location along the *zdir* axis in 3D space to position the line. zdir : {'x', 'y', 'z'} Plane to plot line orthogonal to. Default: 'z'. See `.get_dir_vector` for a description of the values. TN) Z get_xdataZ get_ydatarZ_to_unmasked_float_arrayZravelr broadcast_tor juggle_axes_verts3dr;)r,rSr#rQrRrrrr+s zLine3D.set_3d_propertiescGsPt|dkr|d}td|D] \}}t|st|dq||_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) rrr7z must be a sequenceTN)r ziprr RuntimeErrorrWr;)r,rTnamer7rrrrPs  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. )rWr1rrr get_data_3ds zLine3D.get_data_3dcsH|j\}}}t||||jj\}}}|||t|d|_dS)NF) rWrZproj_transformr=r>set_datar4rCr;)r,rDZxs3dZys3dZzs3drQrRrSr8rrrC s    z Line3D.draw)rr) rGrHrIrJr*r+rPr[rrKrCrLrrr8rrOs   rOcCst|_|||dS)aA Convert a `.Line2D` to a `.Line3D` object. Parameters ---------- zs : float The location along the *zdir* axis in 3D space to position the line. zdir : {'x', 'y', 'z'} Plane to plot line orthogonal to. Default: 'z'. See `.get_dir_vector` for a description of the values. N)rOr9r+)linerSr#rrr line_2d_to_3ds r^csHt|t|}|jddd}ddt||D}fdd|D}|S)zConvert a path to a 3D segment.FZsimplifyZcurvescSs"g|]\\\}}}}|||fqSrr.0rrcoderrrr *sz'_path_to_3d_segment..cs g|]\}}}t|||qSrrVrarrrr"rrrc+s)rrUr iter_segmentsrX)rrSr#pathsegssegseg3drr"r_path_to_3d_segment%s rjcsPt|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)rjrarZpathzr"rrrc8sz)_paths_to_3d_segments..)rrrUr r!rX)pathsrSr#Zsegsrr"r_paths_to_3d_segments/s  rmcsjt|t|}|jddd}ddt||D}|rVt|\}}fdd|D}ng}g}|t|fS)z/Convert a path to a 3D segment with path codes.Fr_cSs&g|]\\\}}}}|||f|fqSrrr`rrrrcBsz2_path_to_3d_segment_with_codes..cs g|]\}}}t|||qSrrdrer"rrrcEs)rrUr rfrXlist)rrSr#rgZ seg_codesrhcodesrirr"r_path_to_3d_segment_with_codes=s rpcsTt|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)rprkr"rrrcRsz4_paths_to_3d_segments_with_codes..)rrUr rXrn)rlrSr#Zsegments_codessegmentsrorr"r _paths_to_3d_segments_with_codesLs  rrcs0eZdZdZddZfddZddZZS)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) _segments3dr4 set_segments)r,rqr8rrr{eszLine3DCollection.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>)raZpointsr1rrrcpsz5Line3DCollection.do_3d_projection..cSs g|]\}}}t||gqSr)r column_stack)rarQrRrSrrrrcrsgeA)rzr r{min)r,Zxyslist segments_2dZminzrQrRrSrr1rdo_3d_projectionls  z!Line3DCollection.do_3d_projection)rGrHrIrJryr{rrLrrr8rrs[s rscCs$t|||}t|_||dS)z.N)rrUr rX _segment3d)r,vertsrSr#rr"rr+s zPatch3D.set_3d_propertiescCs|jSr')_path2dr1rrrget_pathszPatch3D.get_pathc CsN|j}t|\}}}t||||jj\}}}}tt ||g|_ t |Sr') rrXrproj_transform_clipr=r>mpathPathrr|rr} r,srQrRrSvxsvysvzsvisrrrrs  zPatch3D.do_3d_projection)rr) rGrHrIrJr*r+rrrLrrr8rrs  rc@s2eZdZdZdddddZd dd Zd d Zd S) PathPatch3Dz 3D PathPatch object. rrrcKs tj|f|||||dS)aa Parameters ---------- path : zs : float The location along the *zdir* axis in 3D space to position the path patch. zdir : {'x', 'y', 'z', 3-tuple} Plane to plot path patch orthogonal to. Default: 'z'. See `.get_dir_vector` for a description of the values. N)rr*r+)r,rrSr#r-rrrr*s zPathPatch3D.__init__rcCs tj||j||d|j|_dS)a Set the *z* position and direction of the path patch. Parameters ---------- path : zs : float The location along the *zdir* axis in 3D space to position the path patch. zdir : {'x', 'y', 'z', 3-tuple} Plane to plot path patch orthogonal to. Default: 'z'. See `.get_dir_vector` for a description of the values. rN)rr+Zverticesro_code3d)r,rrSr#rrrr+szPathPatch3D.set_3d_propertiesc CsR|j}t|\}}}t||||jj\}}}}tt ||g|j |_ t |Sr') rrXrrr=r>rrrr|rrr}rrrrrs  zPathPatch3D.do_3d_projectionN)rr)rGrHrIrJr*r+rrrrrrs rcCs4|}|}||}t|r*|dStgS)z2Return a list of vertices for the path of a patch.r)get_patch_transformrZ to_polygonsr rr)patchtransrpolygonsrrr_get_patch_vertss rcCs t|}t|_||||dS)z*Convert a `.Patch` to a `.Patch3D` object.N)rrr9r+)rrr#rrrrpatch_2d_to_3dsrcCs2|}|}||}t|_||||dS)z2Convert a `.PathPatch` to a `.PathPatch3D` object.N)rrZtransform_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. rrTrSr# 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 to shade the patches in order to give the appearance of depth (default is *True*). This is typically desired in scatter plots. N) _depthshader4r*r+r,rSr#rrTr-r8rrr*szPatch3DCollection.__init__cCs|jSr'rr1rrrget_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|_dSrtrurwrrrry%szPatch3DCollection.set_sort_zposcCs`||}t|dkr(|j\}}ng}g}t||t|||_td|_ d|_ d|_ dS)a Set the *z* positions and direction of the patches. Parameters ---------- zs : float or array of floats The location or locations to place the patches in the collection along the *zdir* axis. zdir : {'x', 'y', 'z'} Plane to plot patches orthogonal to. All patches must have the same direction. See `.get_dir_vector` for a description of the values. rNT) update_scalarmappable get_offsetsr TrVr atleast_1d _offsets3dslice_z_markers_idx_vzsr;r,rSr#offsetsrQrRrrrr+*s   z#Patch3DCollection.set_3d_propertiescs`|j\}}}t||||jj\}}}}||_tt ||g|j dkrVt |Stj SdSNr) rrrr=r>rr4 set_offsetsrr|sizer}nan)r,rQrRrSrrrrr8rrrFs    z"Patch3DCollection.do_3d_projectioncCsD|jdk r|jrt||jn|}t|dkr6||j}t||jSNrrr_zalphar rmcolors to_rgba_array_alphar,Z color_arrayrrr"_maybe_depth_shade_and_sort_colorsRs  z4Patch3DCollection._maybe_depth_shade_and_sort_colorscs|tSr'rr4 get_facecolorr1r8rrr\szPatch3DCollection.get_facecolorcs&t|jdr|S|tSNfacerZ _str_equal _edgecolorsrrr4 get_edgecolorr1r8rrr_szPatch3DCollection.get_edgecolor)rGrHrIrJr*rrryr+rrrrrLrrr8rrs   rcseZdZdZddddfdd Zfdd Zd d Zd d Zd!fdd ZfddZ ddZ ddZ ddZ e fddZddZfddZfdd ZZS)"Path3DCollectionz# A collection of 3D paths. rrTrcs0||_d|_tj|||||d|_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 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_drawr4r*r+_offset_zorderedrr8rrr*ms  zPath3DCollection.__init__c s>|,tj|ddt|W5QRXW5QRXdS)NT)r)_use_zordered_offsetrrBr4rCrEr8rrrCs zPath3DCollection.drawcCs||_d|_dSrtrurwrrrryszPath3DCollection.set_sort_zposcCs||}t|dkr(|j\}}ng}g}t||t|||_|j|_ t |j |_ |j\}}}t d|_d|_d|_dS)a Set the *z* positions and direction of the paths. Parameters ---------- zs : float or array of floats The location or locations to place the paths in the collection along the *zdir* axis. zdir : {'x', 'y', 'z'} Plane to plot paths orthogonal to. All paths must have the same direction. See `.get_dir_vector` for a description of the values. rrNT)rrr rrVrrr_sizes_sizes3dr _linewidths _linewidths3drrrr;rrrrr+s     z"Path3DCollection.set_3d_propertiesR@cst|||js||_dSr')r4 set_sizesrr)r,ZsizesZdpir8rrrszPath3DCollection.set_sizescs$t||js t|j|_dSr')r4 set_linewidthrrrrr)r,Zlwr8rrrs zPath3DCollection.set_linewidthcCs|jSr'rr1rrrrszPath3DCollection.get_depthshadecCs||_d|_dSrrrrrrrs zPath3DCollection.set_depthshadec Cs|j\}}}t||||jj\}}}}t|ddd}|_||_t |j dkrb|j ||_ t |j dkr||j ||_ t|t||f||}||}||}t||f|_|jrt|StjS)Nrr)rrrr=r>rZargsortrrr rrrrr rr|rrr}r) r,rQrRrSrrrrZ z_markers_idxrrrrs      z!Path3DCollection.do_3d_projectionc#s>|jdkrdVn(|j}t|jz dVW5||_XdSr')rZ_offsetsr4r)r,Z old_offsetr8rrrs  z%Path3DCollection._use_zordered_offsetcCsD|jdk r|jrt||jn|}t|dkr6||j}t||jSrrrrrrrs  z3Path3DCollection._maybe_depth_shade_and_sort_colorscs|tSr'rr1r8rrr szPath3DCollection.get_facecolorcs&t|jdr|S|tSrrr1r8rrrszPath3DCollection.get_edgecolor)r)rGrHrIrJr*rCryr+rrrrrrrrrrrLrrr8rrhs -  !  rTcCsDt|trt|_d|_nt|tr(t|_||_d|_| ||dS)a6 Convert a `.PatchCollection` into a `.Patch3DCollection` object (or a `.PathCollection` into a `.Path3DCollection` object). Parameters ---------- zs : float or array of floats The location or locations to place the patches in the collection along the *zdir* axis. Default: 0. zdir : {'x', 'y', 'z'} The axis in which to place the patches. Default: "z". See `.get_dir_vector` for a description of the values. depthshade Whether to shade the patches to give a sense of depth. Default: *True*. NF) isinstancer rr9rr rrrr+)rrSr#rrrrpatch_collection_2d_to_3ds  rcseZdZdZddddfdd ZejejejdZ d d Z d d Z d"fdd Z ddZ ddZddZddZfddZfddZddZddZd d!ZZS)#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. averageFN)zsortshade lightsourcec s|rlt|}|dd}|dk r0t||||d<|dd} | dk rTt| |||d<|dkrl| dkrltdtj|f||t|tjr|j dkrtdnt dd|Drtd| |d|_ dS) a" Parameters ---------- verts : list of (N, 3) array-like The sequence of polygons [*verts0*, *verts1*, ...] where each element *verts_i* defines the vertices of polygon *i* as a 2D array-like of shape (N, 3). zsort : {'average', 'min', 'max'}, default: 'average' The calculation method for the z-order. See `~.Poly3DCollection.set_zsort` for details. shade : bool, default: False Whether to shade *facecolors* and *edgecolors*. When activating *shade*, *facecolors* and/or *edgecolors* must be provided. .. versionadded:: 3.7 lightsource : `~matplotlib.colors.LightSource`, optional The lightsource to use when *shade* is True. .. versionadded:: 3.7 *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. facecolorsN edgecolorszCYou must provide facecolors, edgecolors, or both for shade to work.rz)verts must be a list of (N, 3) array-likecss |]}tt|dkVqdS)r3N)r rshape)raZvertrrr }sz,Poly3DCollection.__init__..) _generate_normalsget _shade_colorsr!r4r*rrndarrayndimany set_zsort_codes3d) r,rrrrrTr-normalsrrr8rrr*Hs8      zPoly3DCollection.__init__)rr}maxcCs|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 _zsortfuncrvr;)r,rrrrrs zPoly3DCollection.set_zsortcCst|rt|j\}}}nggg}}}tt|}t||||g|_dtdd|D}tt |dd|dd|_ dS)zOptimize points for projection.rcSsg|] }t|qSr)r )rasegmentrrrrcsz/Poly3DCollection.get_vector..Nrr)r) r rZ row_stackronesr_vecZcumsummapr _segslices)r,rrQrRrSrindicesrrr get_vectorszPoly3DCollection.get_vectorTcs"||tgd||_dS)a Set 3D vertices. Parameters ---------- verts : list of (N, 3) array-like The sequence of polygons [*verts0*, *verts1*, ...] where each element *verts_i* defines the vertices of polygon *i* as a 2D array-like of shape (N, 3). closed : bool, default: True Whether the polygon should be closed by adding a CLOSEPOLY connection at the end. FN)rr4 set_verts_closed)r,rclosedr8rrrs zPoly3DCollection.set_vertscCs|j|dd||_dS)z Set 3D vertices with path codes.F)rN)rr)r,rrorrrset_verts_and_codessz$Poly3DCollection.set_verts_and_codescCsF|d|_|dt||_t||_t||_ d|_ dS)NrT) rrvrr r _facecolor3dr _edgecolor3dZ get_alphaZ_alpha3dr;r1rrrr+s    z"Poly3DCollection.set_3d_propertiescCs||_d|_dSrtrurwrrrryszPoly3DCollection.set_sort_zposc sjdk r.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}jdk rXfd d|D}t||nt|jtjt|krj_jdk rtdgdgjgdgg} t | j j } | ddSjdkrtStj SdS)z< Perform the 3D projection for this object. Ncs"g|]}|||fqSrr)rasl)txstystzsrrrcsz5Poly3DCollection.do_3d_projection..r)axisc3s<|]4\}\\}}}}}|t||g|||fVqdSr')rrr|)raidxrQrRrSZfcZecr1rrrsz4Poly3DCollection.do_3d_projection..cSs|dSrrrrrrz3Poly3DCollection.do_3d_projection..T)keyreversercsg|]}j|qSr)r)rarr1rrrcsrr3)!Z_ArZ_face_is_mappedZ _facecolorsrZ_edge_is_mappedrrrZ_proj_transform_vecrr=r>rr repeatsorted enumeraterX _facecolors2d _edgecolors2dremptyrr rrrrvrrr}r) r,ZxyzlistZcfaceZcedgeZ z_segments_2d_r~ZidxsroZzvecZztransr)r,rrrrrsV          z!Poly3DCollection.do_3d_projectioncst|t||_dSr')r4 set_facecolorr rrr,rr8rrrs zPoly3DCollection.set_facecolorcst|t||_dSr')r4 set_edgecolorr rrr r8rrr s zPoly3DCollection.set_edgecolorc Cs|tj||zt|j|j|_Wnttt fk r>YnXzt|j |j|_ Wnttt fk rpYnXd|_ dS)NT) rZArtist set_alpharrrrAttributeError TypeError IndexErrorrrr;)r,alpharrrr s   zPoly3DCollection.set_alphacCs&t|ds |j|j_||jS)Nr)hasattrr=get_projr>rrr1rrrr+s zPoly3DCollection.get_facecolorcCs&t|ds |j|j_||jS)Nr)rr=rr>rrr1rrrr3s zPoly3DCollection.get_edgecolor)T)rGrHrIrJr*rrr}rrrrrrr+ryrrr r rrrLrrr8rr3s(;  E  rcCs2t|||\}}t|_||||dS)a Convert a `.PolyCollection` into a `.Poly3DCollection` object. Parameters ---------- zs : float or array of floats The location or locations to place the polygons in the collection along the *zdir* axis. Default: 0. zdir : {'x', 'y', 'z'} The axis in which to place the patches. Default: 'z'. See `.get_dir_vector` for a description of the values. N)rrrrr9rr+)rrSr#Z segments_3drorrrpoly_collection_2d_to_3d<s  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_axesrQrRrSr#rrrrVPs   rVcCs2|dkr|||fS|dkr$|||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-y)z-xrNrrrrrras   rcCst|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) r rzerosr r}rrUrrr|)rrSnormZsatsZrgbarrrros  rc Cs:t|tjr||jd}d|dd|d}}}|d|ddf|d|ddf}|d|ddf|d|ddf}ntt|df}tt|df}t|D]\}}t|}d|dd|d}}}||ddf||ddf||ddf<||ddf||ddf||ddf<qt||S)aA Compute the normals of a list of polygons, one normal per polygon. Normals point towards the viewer for a face with its vertices in counterclockwise order, following the right hand rule. Uses three points equally spaced around the polygon. This method assumes that the points are in a plane. Otherwise, more than one shade is required, which is not supported. Parameters ---------- polygons : list of (M_i, 3) array-like, or (..., M, 3) array-like A sequence of polygons to compute normals for, which can have varying numbers of vertices. If the polygons all have the same number of vertices and array is passed, then the operation will be vectorized. Returns ------- normals : (..., 3) array A normal vector estimated for the polygon. rrr3.N)rrrrrr rZcross) rni1i2Zi3Zv1Zv2Zpoly_iZpsrrrr}s  $&,.rc s|dkrtjddd}tjdd"|tjj|ddd |j}W5QRXt|}|rt d dt d dj fd d }d||<t |}|dddf}||ddtj f|}||dddf<nt |}|S)z Shade *color* using normal vectors given by *normals*, assuming a *lightsource* (using default position if not given). *color* can also be an array of the same length as *normals*. Ng -x3@)ZazdegZaltdegignore)invalidrT)rZkeepdimsrg333333?cs |Sr'rrZin_normZout_normrrrsz_shade_colors..normrr)rZ LightSourcerZerrstateZlinalgr directionisnanrr ZinverserZnewaxisZ asanyarraycopy)colorrrrmaskrrrrrrrs$    r)rr)rr)rr)rr)rr)rr)rr)rr)rr)rrT)rr)N)9rJr?Znumpyr contextlibrZ matplotlibrrrrrrr(rrZmatplotlib.collectionsr r r r Zmatplotlib.colorsr Zmatplotlib.patchesrr&rrrr$r)r%rNZLine2DrOr^rjrmrprrrsrrrrrrrrrrrrVrrrrrrrrsN     "[ ]    ! 2/  k1  +