:j4dZddlZddlZddlmZddlmZmZm Z m Z m Z m ZmZmZddlmZmZmZmZmZddlmZddlmZd Zd Zd Zd Zd Z GddejBZ"d5dZ#Gdde jHZ%d5dZ&d6dZ'd6dZ(d6dZ)d6dZ*GddeZ+d5dZ,GddeZ-d5dZ.GddeZ/Gd d!e/Z0d"Z1d5d#Z2d6d$Z3Gd%d&eZ4d'Z5Gd(d)eZ6 d7dd*d+Z7Gd,d-eZ8d5d.Z9d/Z:d0Z; d8d1Zd:d4Z?y);zx Module containing 3D artist code and functions to convert 2D artists into 3D versions which can be added to an Axes3D. N)contextmanager)_apiartistcbookcolorslinestextpathrcParams) CollectionLineCollectionPolyCollectionPatchCollectionPathCollection)Patch)proj3dc*|dzdz}|dkDr|dz }|S)z?Return the given angle normalized to -180 < *a* <= 180 degrees.ihas o/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/mpl_toolkits/mplot3d/art3d.py _norm_anglers$ SCA3w G Hc*|dzdz}|dkDr|dz }|S)z=Return the given angle normalized to -90 < *a* <= 90 degrees.rZrrs r_norm_text_angler!s$ SCA2v G Hrctj|drtjdStj|drtjdStj|drtjdS|tjdStj|r#t |dk(rtj|St d ) 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 an array Returns ------- x, y, z : array The direction vector. x)rrry)rrrz)rrr)rrrz2'x', 'y', 'z', None or vector of length 3 expected)r _str_equalnparrayiterablelen ValueError)zdirs rget_dir_vectorr+)s( c"xx ""   $ $xx ""   $ $xx "" xx "" T s4yA~xx~MNNrc dtjj||jjk||jj kD||jj k||jjkD||jjk||jj kDf}|S)a] Return the mask of the points outside the axes view limits. Parameters ---------- xs, ys, zs : array-like The points to mask. These should be in data coordinates. axes : Axes3D The axes to use for the view limits. Returns ------- mask : np.array The mask of the points as a bool array. ) r% logical_orreduce xy_viewLimxminxmaxyminymax zz_viewLim)xsyszsaxesmasks r _viewlim_maskr:Ks ==  doo&:&:!:!#doo&:&:!:!#doo&:&:!:!#doo&:&:!:!#doo&:&:!:!#doo&:&:!: !< =D Krc tjjtj|jj j |tj|jj j |tj|jj j |fS)a Return the mask of points whose coordinates are invalid for the axis scale they live on (e.g. <=0 on a log axis). Parameters ---------- xs, ys, zs : array-like The points to check, in data coordinates. axes : Axes3D The axes whose scales are queried. Returns ------- mask : np.ndarray Boolean array, ``True`` where any of x/y/z is out of its scale's valid domain. ) r%r-r. logical_notxaxis_scale val_in_rangeyaxiszaxis)r5r6r7r8s r_scale_invalid_maskrBds$ ==   tzz((55b9: tzz((55b9: tzz((55b9:!< ==rcleZdZdZd dZdZd fd ZdZd dZe jdZ d dZ xZ S) Text3DaJ 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. axlim_clip : bool, default: False Whether to hide text outside the axes view limits. .. versionadded:: 3.10 Other Parameters ---------------- **kwargs All other parameters are passed on to `~matplotlib.text.Text`. c d|vrtjdd|vrtjdtjj||||fi||j |||y)NrotationzOThe `rotation` parameter has not yet been implemented and is currently ignored. rotation_modezTThe `rotation_mode` parameter has not yet been implemented and is currently ignored.)r warn_externalmtextText__init__set_3d_properties)selfr r!r"r r* axlim_clipkwargss rrKzText3D.__init__sk     ,  f $   ,   D!Q77 q$ 3rcH|j|j|jfS)z*Return the (x, y, z) position of the text.)_x_y_zrMs rget_position_3dzText3D.get_position_3dsww((rcxt||dd|j|d|t||_yy)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)super set_positionset_zr+_dir_vec)rMxyzr* __class__s rset_position_3dzText3D.set_position_3ds? S!W% 3q6  *40DM rc ||_d|_y)zl Set the *z* position of the text. Parameters ---------- z : float TN)rSstale)rMr"s rrZz Text3D.set_zs rcN||_t||_||_d|_y)a 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. axlim_clip : bool, default: False Whether to hide text outside the axes view limits. .. versionadded:: 3.10 TN)rSr+r[ _axlim_clipr`)rMr"r*rNs rrLzText3D.set_3d_propertiess& &t, % rct|j|j|j|j}|j r9|t |j|j|j|jz}tj|retjj|j|j|jg|tjtj}nrvv HHdggtww8FE$--'UG,-++ 1a4L&A,q!t diiA !WQZ$q'!* $ !WQZ$q'!* $ TZZB/0   tQ tAwqz)9%)@B , JJOOD( + ,  , s !H99I cyNr)rMrss r get_tightbboxzText3D.get_tightbboxsr)rrrr"Fr|rr"F)__name__ __module__ __qualname____doc__rKrUr^rZrLrallow_rasterizationrrr} __classcell__r]s@rrDrD|s@. 4)1" *  ,rrDc@t|_|j|||y)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. axlim_clip : bool, default: False Whether to hide text outside the axes view limits. .. versionadded:: 3.10 N)rDr]rL)objr"r*rNs r text_2d_to_3drs CM!T:.rcdeZdZdZddfd Zd dZdZdZejfdZ xZ 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`. F)rNcbt|ggg|i||j|||||_y)aM 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)rXrK set_data_3drb)rMr5r6r7rNargsrOr]s rrKzLine3D.__init__s8 R1$1&1 R$%rc |j}|j}tj|j }t j |t|}t|||||_ ||_ d|_ y)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. axlim_clip : bool, default: False Whether to hide lines with an endpoint outside the axes view limits. .. versionadded:: 3.10 TN) get_xdata get_ydatar_to_unmasked_float_arrayravelr% broadcast_tor( juggle_axes_verts3drbr`)rMr7r*rNr5r6s rrLzLine3D.set_3d_properties&sj"^^  ^^   + +B / 5 5 7 __RR )#BB5 % rct|dk(r|d}td|D](\}}tj|rt |d||_d|_y)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) rrr\z must be a sequenceTN)r(zipr%r' RuntimeErrorrr`)rMrnamer\s rrzLine3D.set_data_3d?sc" t9>7DUD) AID#;;s#"dV+>#?@@ A  rc|jS)z Get the current data Returns ------- verts3d : length-3 tuple or array-like The current data as a tuple or array-like. )rrTs r get_data_3dzLine3D.get_data_3dXs}}rctg|j|j}|jr$|t g|j|jz}t j |rt j|t|jg|jdj}t jj|jt|jt j\}}}n|j\}}}tj ||||j\}}} } |j#||t$ |M|d|_y)Nrrdr9F)rBrr8rbr:r%rgrr(shaperhr&rirjrkr_scale_proj_transform_clipset_datarXrrr`) rMrs scale_maskr9xs3dys3dzs3dr5r6r7tisr]s rrrz Line3D.drawcs(C$--CC    -BB B BJ 66* ??T]]#=dmmA&6&<&<=D "uu{{4==16T + CCI6"&&> D$ $}} D$ ;;D$diiXBC b"  X rr) rrrrrKrLrrrrrrrrs@rrr s;6;&$22   rrc@t|_|j|||y)a 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. axlim_clip : bool, default: False Whether to hide lines with an endpoint outside the axes view limits. .. versionadded:: 3.10 N)rr]rL)liner7r*rNs r line_2d_to_3drws"DN2tZ0rc $tj|t|}|jdd}t ||Dcgc]\\\}}}}|||f}}}}}|Dcgc]\}}}t ||||} }}}| Scc}}}}wcc}}}w)zConvert a path to a 3D segment.Fsimplifycurves)r%rr( iter_segmentsrr) r r7r*pathsegsr r!coder"segseg3ds r_path_to_3d_segmentrs SY 'B!!5!?H03Hb0A B B,.6Aq4!Aq!9 BC B9< = =IQ1[Aq$ ' =E = L C =s B $B c tj|s tj|t|}n"t|t|k7r t dt ||Dcgc]\}}t |||}}}|Scc}}w)z6Convert paths from a collection object to 3D segments.z-Number of z-coordinates does not match paths.)r%r'rr(r)rr)pathsr7r*r pathzsegss r_paths_to_3d_segmentsrs{ ;;r? __RU , r7c%j LM M #5"~ /e eT 2 /D / K /s'Bc btj|t|}|jdd}t ||Dcgc]\\\}}}}|||f|f}}}}}|r/t |\} } | Dcgc]\}}}t ||||} }}}ng} g} | t | fScc}}}}wcc}}}w)z/Convert a path to a 3D segment with path codes.Fr)r%rr(rrrlist) r r7r*rr r!rr" seg_codesrcodesrs r_path_to_3d_segment_with_codesrs SY 'B!!5!?H4; && 9sA1ceZdZdZdZy) Collection3DzA collection of 3D paths.c V|jDcgc]\}}| }}}|Dcgc]#}tg|j|j%}}|jr@t ||Dcgc])\}}|t g|j|jz+}}}t ||Dcgc][\}}tj|r?tjj|tj||jn|]}}}|Dcgc]9}tj|dddf|dddf|dddf|j;}}t ||jD cgc]A\\}}}\}} tj tjj#||g| Cc} }}}|_tj&|D cgc]\}}} |  c} }} t)| r| j+SdScc}}wcc}wcc}}wcc}}wcc}wcc} }}}wcc} }w)z0Project the points according to renderer matrix.r9NrrrWeA)_3dverts_codesrBTr8rbrr:r%rgrhr&rrrrmmpathPath column_stack_paths concatenater(min) rMvs_vs_listmasksm xyzs_listr5r6csr7s rdo_3d_projectionzCollection3D.do_3d_projections#'#6#67%"a277AHI2$6bdd6DII6II   "%eW"57B88dii887E7!$GU 35r1ffQi55;;r288(D;E%'(55@GH9;11 q!tHbAh1a4$))5H H47y$BUBU3VXX/;BABzz"%%"4"4b"X">CX ^^ :HAq"R: ;r7rvvx++8I75HX:s* H(H(.H (A H>H*AH  H% N)rrrrrrrrrrs #,rrc,tj|t|j}t |j|Dcgc]\}}tj t gtj |jtj|t|jgj||jfc}}|_ tjtdt||_||_ycc}}w)z4Convert a `.Collection` to a `.Collection3D` object.z{}3DN)r%rr( get_pathsrrrverticesrrrr_make_class_factoryrtyper]rb)colr7r*rNpr"s rcollection_2d_to_3drs S1 2B  , . Aq  __ajj"//!S_*MN O Q Q    .C DE--lFCDINCM CO.s BDc:eZdZdZdfd ZdZfdZdZxZS)Line3DCollectionz# A collection of 3D lines. c 4t||fi|||_yr|)rXrKrb)rMrrNrOr]s rrKzLine3DCollection.__init__s" )&)% rc ||_d|_yz&Set the position to use for z-sorting.TN _sort_zposr`rMvals r set_sort_zposzLine3DCollection.set_sort_zpos rc2||_t| gy)z" Set 3D segments. N) _segments3drX set_segments)rMrr]s rrzLine3DCollection.set_segments s$ R rc|j}|Dcgc] }t|}}tt|dkD}|rHtd|Dr tj j |}n+t j |}nt j|}|jdk(r&tj|gtjSd}tj j|r2|jtj jur |j}t|d|d|d|j }t j|r;|t j"|dtj$fg|j&d z}|j(rst+|d|d|d|j }t j|r=t j"|dtj$fg|j&d }||z}t-j.||j }|dur!tj j1|| }|ddd f} |r,t j2| t j4|d d } tj|| t|dkDrt7|dj7d} | Stj} | Scc}w)zB Project the points according to renderer matrix. rc3ZK|]#}tjj|%ywr|)r%rhisMA).0segments r z4Line3DCollection.do_3d_projection..s?7255::g&?s)+rF.r.r.rW.r#rrWNr)rr(setrgr%rhr asanyarraysizer rrkrr9nomaskrBr8rnewaxisrrbr:r_scale_proj_transform_vectorsr&splitcumsumr) rMrrsegment_lengthsraggedr9r viewlim_maskxyzs segments_2dminzs rrz!Line3DCollection.do_3d_projections{##8@@G3w<@@S)*Q. ?h??55,,X6>>(3}}X.H ==A   ' 'b 166M 55::h HMM$E==D(&)9)1&)9)1&)9)-4  66*  BOOJsBJJ$?$:j&6&6$:$:< >*A'q!Q5>%>sAct|ds9|jj|j_|j |j S)N_path2d)hasattrr8get_projMrrrTs rget_pathzPatch3D.get_paths<tY')),,.DIIK  ! ! #||rcX|j}t|\}}}t||||j}|jr|t ||||jz}t j|rNt jjt|t|jt j\}}}n|||}}}tj||||j\} } } } tj t jj#| | g|_t'| SNr)rrrBr8rbr:r%rgrhr&rirjrkrrrrrrr rMsxs0ys0zs0r9r5r6r7vxsvysvzsviss rrzPatch3D.do_3d_projections OOQ S#"3S$))<    M#sC; ;D 66$<S!W+0t%==CVBFF^ BBc3BB#>>r2r499US#szz"%%"4"4c3Z"@A 3xrr) rrrrrKrLrrrrs@rr r Vs%"$#%5&&. rr c.eZdZdZdddddZd dZdZy ) PathPatch3Dz 3D PathPatch object. rr"Fr c Xtj|fi||j||||y)a 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. axlim_clip : bool, default: False Whether to hide path patches with a point outside the axes view limits. .. versionadded:: 3.10 N)rrKrL)rMr r7r*rNrOs rrKzPathPatch3D.__init__s)" t&v& tRz:rcntj||j||||j|_y)aB 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. axlim_clip : bool, default: False Whether to hide path patches with a point outside the axes view limits. .. versionadded:: 3.10 r N)r rLrr_code3d)rMr r7r*rNs rrLzPathPatch3D.set_3d_propertiess1$ !!$ "4-7 " 9zz rcn|j}t|\}}}t||||j}|jr|t ||||jz}t j|rNt jjt|t|jt j\}}}n|||}}}tj||||j\} } } } tj t jj#| | g|j$|_t)| Sr)rrrBr8rbr:r%rgrhr&rirjrkrrrrrr%rrrs rrzPathPatch3D.do_3d_projections OOQ S#"3S$))<    M#sC; ;D 66$<S!W+0t%==CVBFF^ BBc3BB#>>r2r499US#szz"%%"4"4c3Z"@$,,O 3xrNr)rrrrrKrLrrrrr"r"s$&CE;(", rr"c|j}|j}|j|}t|r|dSt j gS)z2Return a list of vertices for the path of a patch.r)get_patch_transformr to_polygonsr(r%r&)patchtransr polygonss r_get_patch_vertsr-sI  % % 'E >> D&Hh-8A;9RXXb\9rcXt|}t|_|j||||y)z*Convert a `.Patch` to a `.Patch3D` object.N)r-r r]rL)r*r"r*rNrs rpatch_2d_to_3dr/s' U #EEO E1dJ7rc|j}|j}|j|}t|_|j |||y)z2Convert a `.PathPatch` to a `.PathPatch3D` object.N)rr(transform_pathr"r]rL) pathpatchr"r*r r+rs rpathpatch_2d_to_3dr3sH    D  ) ) +E   &E%I q$/rcreZdZdZddddddfd ZdZ dd Zd Zdd Zfd Z d Z fdZ fdZ xZ S)Patch3DCollectionz% A collection of 3D patches. rr"NFr7r* depthshadedepthshade_minalpharNc| td}| td}||_||_t||i||j |||y)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. 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. *depthshade_minalpha* sets the minimum alpha value applied by depth-shading. Naxes3d.depthshadeaxes3d.depthshade_minalpha)r _depthshade_depthshade_minalpharXrKrL rMr7r*r7r8rNrrOr]s rrKzPatch3DCollection.__init__s\8  !"56J  &"*+G"H %$7! $)&) r44rc|jSr|r<rTs rget_depthshadez Patch3DCollection.get_depthshadercD| td}||_||_d|_y)a 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. depthshade_minalpha : float, default: :rc:`axes3d.depthshade_minalpha` Sets the minimum alpha value used by depth-shading. .. versionadded:: 3.11 Nr;Tr r<r=r`rMr7r8s rset_depthshadez Patch3DCollection.set_depthshade"-$  &"*+G"H %$7! rc ||_d|_yrrrs rrzPatch3DCollection.set_sort_zpos:rrc|j|j}t|dkDr|j\}}ng}g}t ||t j |||_td|_ d|_ ||_ d|_ y)ai 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. axlim_clip : bool, default: False Whether to hide patches with a vertex outside the axes view limits. .. versionadded:: 3.10 rrNT) update_scalarmappable get_offsetsr(rrr% atleast_1d _offsets3dslice_z_markers_idx_vzsrbr`rMr7r*rNoffsetsr5r6s rrLz#Patch3DCollection.set_3d_properties?s~( ""$""$ w4??>DII> >D 66$<T__4@JBBJBB#>>r2r499US#s 55::c? G  2 2C: > ? G c ; < 88a<s8O66Mrc|j}|jN|jrBt||j|j}||j ddk(r ||dddfz}t |dkDrI||j}tj|dkDr"tj||j}tj||S)N) min_alpharr#r) _alpharPr<_zalphar=rr(rOr%ndimrlmcolors to_rgba_arrayrM color_arrayalphas r"_maybe_depth_shade_and_sort_colorsz4Patch3DCollection._maybe_depth_shade_and_sort_colorsts  99 T%5%5! 33K  [%6%6q%9Q%> AqD 11 { a %d&9&9:Kwwu~! 5)$*=*=>$$[%88rc>|jt| Sr|r`rX get_facecolorrMr]s rrczPatch3DCollection.get_facecolor66uw7L7NOOrctj|jdr|jS|j t |SNfacerr$ _edgecolorsrcr`rX get_edgecolorrds rrkzPatch3DCollection.get_edgecolorC   D,,f 5%%' '66uw7L7NOOrr|r)rrrrrKrArFrrLrr`rcrkrrs@rr5r5sU  #5J !0 B(9,PPPrr5cPtjj|sy|j|j z }|j|j z }|j|j z }tj |dz|dzz|dzzS)z Estimate the scale of the 3D data for use in depth shading Parameters ---------- X, Y, Z : masked arrays The data to estimate the scale of. rrW)r%rhcountmaxrsqrt)XYZptp_xptp_yptp_zs r_get_data_scalerws 55;;q> EEGaeeg E EEGaeeg E EEGaeeg E 775A: *UaZ7 88rceZdZdZddddddfd ZfdZd Zdd Zdfd Zfd Z d Z ddZ dZ e fdZdZfdZfdZxZS)Path3DCollectionz# A collection of 3D paths. rr"NFr6c| td}| td}||_||_d|_t ||i||j |||d|_y)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. *depthshade_minalpha* sets the minimum alpha value applied by depth-shading. Nr:r;F)r r<r=_in_drawrXrKrL_offset_zorderedr>s rrKzPath3DCollection.__init__sk8  !"56J  &"*+G"H %$7!  $)&) r44 $rc|j5tj|d5t||ddddddy#1swYxYw#1swYyxYw)NT)r{)_use_zordered_offsetrrqrXrr)rMrsr]s rrrzPath3DCollection.drawsZ  & & ( '""4$7 ' X& ' ' ' ' ' ' 's!AA A A AA c ||_d|_yrrrs rrzPath3DCollection.set_sort_zposrrc|j|j}t|dkDr|j\}}ng}g}||_t ||t j|||_|j|_ t j|j|_ |j\}}}td|_d|_||_d|_y)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. axlim_clip : bool, default: False Whether to hide paths with a vertex outside the axes view limits. .. versionadded:: 3.10 rrNT)rJrKr(r_zdirrr%rLrM_sizes_sizes3dr& _linewidths _linewidths3drNrOrPrbr`rQs rrLz"Path3DCollection.set_3d_propertiess( ""$""$ wS)NFrrrr) rMr%rhrr9rBr8rbr:rgrr(rr&rrrw _data_scaleargsortrOrPrrrrrrTrr|rrrk) rMr9r\rrrrr  z_markers_idxs rrz!Path3DCollection.do_3d_projection<s ?? 'Cuuzz#chh ')F4??FDIIFF   -DD$))DDD 66$<??4$'$8#T4??1;M;S;S#TVD55;;tT;:D??D#>>PPdiiPS#s*3S9/1eemmC.@2.FF +  t}}  !-- 6DK t!! "Q &#11-@D ""4););S#J)GH- - - !# 2 2C: >!hhrvvc{2BFF2rc#K|jdy|j}t| |j d||_y#||_wxYwwr|)r|_offsetsrXrT)rM old_offsetr]s rr~z%Path3DCollection._use_zordered_offsetisJ  ( J G  5 5 6 + *  s7AAA AAc|j}|jY|jrMt||j|j|j }||j ddk(r ||dddfz}t|dkDrI||j}tj|dkDr"tj||j}tj||S)N)rVrrrWr#r)rXrPr<rYr=rrr(rOr%rZrlr[r\r]s rr`z3Path3DCollection._maybe_depth_shade_and_sort_colorsws  99 T%5%5! 33 ,, K  [%6%6q%9Q%> AqD 11 { a %d&9&9:Kwwu~! 5)$*=*=>$$[%88rc>|jt| Sr|rbrds rrczPath3DCollection.get_facecolorrerctj|jdr|jS|j t |Srgrirds rrkzPath3DCollection.get_edgecolorrlrr)gR@r|)rrrrrKrrrrLrrrArFrrr~r`rcrkrrs@rryrysy  %%N'  2h" <  !0+3Z + +9.PPPrry)r8ct|trt|_d|_nt|t r t |_| td}| td}||_||_ d|_ |j|||y)a  Convert a `.PatchCollection` into a `.Patch3DCollection` object (or a `.PathCollection` into a `.Path3DCollection` object). Parameters ---------- col : `~matplotlib.collections.PatchCollection` or `~matplotlib.collections.PathCollection` The collection to convert. 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 : bool, default: :rc:`axes3d.depthshade` Whether to shade the patches to give a sense of depth. axlim_clip : bool, default: False Whether to hide patches with a vertex outside the axes view limits. .. versionadded:: 3.10 depthshade_minalpha : float, default: :rc:`axes3d.depthshade_minalpha` Sets the minimum alpha value used by depth-shading. .. versionadded:: 3.11 Nr:r;F) isinstancerryr]r|rr5r r<r=r{rL)rr7r*r7rNr8rs rpatch_collection_2d_to_3drszH#~&( # C )) 12 "&'CD CO2CCL"dJ/rceZdZdZdddddfd Zej ejejdZ dZ e jd d Z d Zdfd Zd ZddZdZdZfdZfdZdZdZdZxZS)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 lightsourcerNc|rbt|}|jdd} | t| |||d<|jdd} | t| |||d<| | tdt ||g|i|t |tjr|jdk7r(tdtd|Dr td|j|d|_ ||_ y)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 axlim_clip : bool, default: False Whether to hide polygons with a vertex outside the view limits. .. versionadded:: 3.10 *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.r#z)verts must be a list of (N, 3) array-likec3^K|]%}ttj|dk7'yw)rWN)r(r%r)rverts rrz,Poly3DCollection.__init__..s">3rxx~&!+>s+-)_generate_normalsget _shade_colorsr)rXrKrr%ndarrayrZrg set_zsort_codes3drb) rMrrrrrNrrOnormalsrrr]s rrKzPoly3DCollection.__init__sH '.GL$7J%'4(|$ L$7J%'4(|$!j&8 %&& 000 eRZZ (zzQ !LMM>>> !LMM u %r)rrrocH|j||_d|_d|_y)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`)rMrs rrzPoly3DCollection.set_zsort(s$//6 rz3.10c$|j|Sr|) _get_vector)rMrs r get_vectorzPoly3DCollection.get_vector6s ++rct|tjr}tjd|t|tj j r3|j|_|jjd|_ y ||_d|_ y t|}tjtt|tj}|j!d}tj"||d f}t%|D]\}}|||d t|f<||_tj&||d d d fk\|_ y ) af Optimize points for projection. Parameters ---------- segments3d : NumPy array or list of NumPy arrays List of vertices of the boundary of every segment. If all paths are of equal length and this argument is a NumPy array, then it should be of shape (num_faces, num_vertices, 3). )NNr#)rraxisFrer)initialr#N)rr%rr check_shaperh MaskedArraydata_facesr9rg_invalid_verticesr(fromitermapintproempty enumeratearange)rMr num_faces num_verts max_vertsrirhs rrzPoly3DCollection._get_vector:s  j"** -   _ D*bee&7&78(oo )3)<)<")<)E&( ).&JI CZ$8HI! a 0IxxIq 9:H$Z0 /4*.JSYJ' /"DK%'YYy%9Yq$w=O%OD "rcV|j|t| gd||_y)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)rrX set_verts_closed)rMrclosedr]s rrzPoly3DCollection.set_vertsYs)  "e$ rc8|j|d||_y)z Set 3D vertices with path codes.F)rN)rr)rMrrs rset_verts_and_codesz$Poly3DCollection.set_verts_and_codesls uU+ rc|jd|_|jdtj||_tj ||_tj||_ d|_ y)NrT) rJrrrrc _facecolor3drk _edgecolor3d get_alpha_alpha3dr`)rMrNs rrLz"Poly3DCollection.set_3d_propertiestsb ""$ y!*88>*88>&006  rc ||_d|_yrrrs rrzPoly3DCollection.set_sort_zposrrc8 |jJ|j|jr|j|_|j r|j |_t|j}|jt|jd|jd|jd|jz}tj|}tjdd5t!j"|j|j}ddd|j$r[t'|jd|jd|jd|j}tj|rd}||z}d}|r!tj(j+||}|j}|j}t||k7r|j-|d }t||k7r$t|d k(r|}n|j-|d }t|d kDr|j/|d } n|} |r | j0} tj2| d ddd } t|d kDr || dddd f} n|} |j4}t|j4d kDre|r/|| ddf} t7| | D cgc]\} }| |ddf} } }| Dcgc]}|j4|}}t9j:|| |nz|rWt| d kDrItj<|| dddf| j>}tj(j+| |} t9j@|| |jBt|d kDr || |_"n||_"t|jt|k(rt|d kDr || |_#n|j|_#|jHYtjJd gd g|jHgd gg}t!jL||jjN}|d d S|jPd kDrtjR|StjTS#1swYwxYwcc}} wcc}w)z< Perform the 3D projection for this object. Nrrrignore)invaliddivideTrrrrrWr)+_ArJ_face_is_mapped _facecolorsr_edge_is_mappedrjrr(rrrBr8r%rgerrstaterrrbr:rhrrepeatrrrrrrrrrrr _facecolors2d _edgecolors2drr&_proj_transform_vecrrrrk)rMrr9 needs_maskingpfacesrpzscfacecedgeface_z face_orderfaces_2d segment_maskrhidxrinvalid_vertices_2dzvecztranss rrz!Poly3DCollection.do_3d_projections  77   & & (##$($4$4!##$($4$4! $ %%(; KK V!4 KK ),,t [[( ; R99$++tyyQF R   (V)Q)-V)CT]]3'>E>  . .tXu EX!2&(ooQ,-NN'$#55,, ':-<  $ $T8T\\ B u:>!&z!2D !&D  t  !SZ /CJN!&z!2D !%!2!2D  ?? &88aS1#'81#>?D//diikkBF!9Q<  XX\66#; 66MS R RP?>s$+R<RRRcXt||tj||_yr|)rX set_facecolorrrcrrMrr]s rrzPoly3DCollection.set_facecolor# f%*88>rcXt||tj||_yr|)rX set_edgecolorrrkrrs rrzPoly3DCollection.set_edgecolorrrctjj|| tj|j |j |_ tj|j|j |_ d|_ y#tttf$rYNwxYw#tttf$r Yd|_ ywxYw)NT) rArtist set_alphar[r\rrXAttributeError TypeError IndexErrorrrjr`)rMr_s rrzPoly3DCollection.set_alphas e,  ' 5 5!!4;;!0D  &44%%t{{ 4D   :6     :6    s#/B /B# B B #CCct|ds9|jj|j_|j t j |jS)Nr)rr8rrrr%rlrrTs rrczPoly3DCollection.get_facecolorGt_-)),,.DIIK  ! ! #zz$,,--rct|ds9|jj|j_|j t j |jS)Nr)rr8rrrr%rlrrTs rrkzPoly3DCollection.get_edgecolor rr)Tr)rrrrrKr%rrrorrr deprecatedrrrrrLrrrrrrcrkrrs@rrrs(,5E!e>&B::vvvv T__V,,P>&  cJ? ? ..rrct|j||\}}t|_|j |||j ||_y)a Convert a `.PolyCollection` into a `.Poly3DCollection` object. Parameters ---------- col : `~matplotlib.collections.PolyCollection` The collection to convert. 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)rrrr]rrLrb)rr7r*rN segments_3drs rpoly_collection_2d_to_3drsJ: MMOR'K$CMK/ COrc`|dk(r|||fS|dk(r|||fS|ddk(rt||||S|||fS)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`. r r!r-) rotate_axesr5r6r7r*s rrr*sR s{2rz 2rz aC2r2t,,2rzrc0|dvr|||fS|dvr|||fS|||fS)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'. )r z-y)z-xr!rrs rrr;s8  {2rz  2rz2rzrct|dk(st|dk(rtjdStj|dd}||dk(rtj|}n3tjd|tj |z |z z |d}tj tj|t|df}tj|ddddf|dddf|zgS)z?Modify the alpha values of the color list according to z-depth.r)rrWrNrWr#) r(r%zerosclip ones_likerrr[r\r)rr7rVrsatsrgbas rrYrYIs 6{a3r7a<xx 1a(IkQ.||BwwqBO{::IqI ??70083r7A, GD ??DBQBKadd):; <.normsGAJ' 'rrr#)r[ LightSourcer%rrr directionr rg Normalizeinverser\rrcopy) colorrrrr9rr_rr-r.s @@rrrs, ))GD X &*BIINN7TNJJ(()* HHUO D xxz##B*$$S!,44 (te %%e,ad eQ ]+e3q!t  Mu%**, M3**s 3EE r)rr")rr"NF)r+N)g:0yE>r|)@rrnnumpyr% contextlibr matplotlibrrrrr[rr rIr rr matplotlib.collectionsr r rrrmatplotlib.patchesrr~rrrr+r:rBrJrDrLine2Drrrrrrrrrrr r"r-r/r3r5rwryrrrrrrYrrrrrrr;sf   %QQ$  OD2=0vUZZvr/(kU\\k\1*   ',:,* !g~gT!DeDN<'<~:80^P^PB9.jP~jP^  0000fC.~C.L !." " =8&+R)X#r