3jTSrSSKrSSKrSSKJr SSKJrJrJ r J r J r J rJrJr SSKJrJrJrJrJr SSKJr SSKJr S rS rS rS rS r "SS\RB5r"S5Sjr#"SS\ RH5r%S5Sjr&S6Sjr'S6Sjr(S6Sjr)S6Sjr*"SS\5r+S5Sjr,"SS\5r-S5Sjr."SS\5r/"S S!\/5r0S"r1S5S#jr2S6S$jr3"S%S&\5r4S'r5"S(S)\5r6S7SS*.S+jjr7"S,S-\5r8S5S.jr9S/r:S0r;S8S1jrS:S4jr?g);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,US-S-nUS:aUS- nU$)z?Return the given angle normalized to -180 < *a* <= 180 degrees.ihas T/home/wildlama/miniconda3/lib/python3.13/site-packages/mpl_toolkits/mplot3d/art3d.py _norm_anglers$ SCA3w G Hc,US-S-nUS:aUS- nU$)z=Return the given angle normalized to -90 < *a* <= 90 degrees.rZrrs r_norm_text_angler!s$ SCA2v G Hrc[R"US5(a[R"S5$[R"US5(a[R"S5$[R"US5(a[R"S5$Uc[R"S5$[R"U5(a%[ U5S:Xa[R"U5$[ S 5e) 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 X[RRXRR:XRR :XRR :XRR:X#RR:X#RR :45nU$)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 ==  oo&:&:!:!#oo&:&:!:!#oo&:&:!:!#oo&:&:!:!#oo&:&:!:!#oo&:&:!: !< =D Krc [RR[R"URR R U55[R"URR R U55[R"URR R U5545$)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:!< ==rc~^\rSrSrSrS SjrSrS U4SjjrSrS Sjr \ RS5r S S jr S rU=r$)Text3D|a 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 SU;a[R"S5 SU;a[R"S5 [RR"XX$40UD6 UR X5U5 g)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 rrLText3D.__init__se     ,  f $   ,   DQ77 q 3rcHURURUR4$)z*Return the (x, y, z) position of the text.)_x_y_zrNs rget_position_3dText3D.get_position_3dsww((rcz>[TU]USS5 URUS5 Ub[U5Ulgg)a0 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)rNxyzr* __class__s rset_position_3dText3D.set_position_3ds? S!W% 3q6  *40DM rcXlSUlg)zD Set the *z* position of the text. Parameters ---------- z : float TN)rUstale)rNr"s rr] Text3D.set_zs rcJXl[U5UlX0lSUlg)ao 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)rUr+r^ _axlim_cliprd)rNr"r*rOs rrMText3D.set_3d_propertiess" &t, % rc[URURURUR5nUR (a9U[ URURURUR5-n[R"U5(ab[RRURURUR/U[S9R[R5nO:[R"URURUR/[S9nX0R-n[R"X4/5n[ R""USS2S4USS2S4USS2S4UR5nUSSUSS- nUSSUSS- n[$R&"[$R("X55n [*R,"XSSUSS[/U 5S9 [0R2R5X5 SSS5 SUlg!,(df  N=f)N)r9dtyperjrrrZ)rSrT _rotationF)rBrSrTrUr8rgr:r%anymar&floatfillednanr^asarrayr_scale_proj_transformmathdegreesatan2r _setattr_cmrrJrKdrawrd) rNrendererr9pos3ddir_endpointsprojdxdyangles rrx Text3D.draws"477DGGTWWdiiH    M$''477DGGTYYG GD 66$<<EEKK$''477 ;%) 88>rvv HHdggtww8FE--'U,-++ 1a4L&A,q!t diiA !WQZ$q'!* $ !WQZ$q'!* $ TZZ/0   tQ tAwqz)9%)@B JJOOD +B BBs H<< I cgNr)rNrys r get_tightbboxText3D.get_tightbboxsr)rgr^rUrd)rrrr"Frrr"F)__name__ __module__ __qualname____firstlineno____doc__rLrWrar]rMrallow_rasterizationrxr__static_attributes__ __classcell__r`s@rrDrD|sE. 4)1" *  ,rrDc>[UlURXU5 g)ah 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`rM)objr"r*rOs r text_2d_to_3drs CM!:.rct^\rSrSrSrSS.U4SjjrS SjrSrSr\ RU4S j5r S r U=r $) Line3Di a 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)rOc`>[TU]"///UQ70UD6 URXU5 X@lg)z 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)r[rL set_data_3drg)rNr5r6r7rOargsrPr`s rrLLine3D.__init__s4 R1$1&1 $%rcUR5nUR5n[R"U5R 5n[ R "U[U55n[XEX5Ul X0l SUl g)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_verts3drgrd)rNr7r*rOr5r6s rrMLine3D.set_3d_properties&sd"^^  ^^   + +B / 5 5 7 __RR )#BB5 % rc[U5S:XaUSn[SU5H.up#[R"U5(aM"[ US35e XlSUlg)z 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' RuntimeErrorrrd)rNrnamer_s rrLine3D.set_data_3d?s[" t9>7DUD)ID;;s##"dV+>#?@@*  rcUR$)z} Get the current data Returns ------- verts3d : length-3 tuple or array-like The current data as a tuple or array-like. )rrVs r get_data_3dLine3D.get_data_3dXs}}rc>[/URQURP76nUR(a$U[ /URQURP76-n[ R "U5(a[ R"U[UR5/URSRQ75n[ RRUR[US9R[ R5upEnOURupEn[R "XEX`R5upxpUR#Xx5 [$T U]MU5 SUlg)Nrrjr9F)rBrr8rgr:r%rmrr(shapernr&rorprqr_scale_proj_transform_clipset_datar[rxrd) rNry scale_maskr9xs3dys3dzs3dr5r6r7tisr`s rrx Line3D.drawcs (C$--CC    -BB B BJ 66*  ??T]]#=dmmA&6&<&<=D "uu{{4==16T + CCI6"&&> D $}} D ;;DiiX b  X r)rgrrdr)rrrrrrLrMrrrrrxrrrs@rrr s@6;&&$22   rrc>[UlURXU5 g)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`rM)liner7r*rOs r line_2d_to_3drws"DN2Z0rc [R"U[U55nURSSS9n[ X15VVVVs/sHuuupEpgXEU4PM nnnnnUVVVs/sHupEn[ XEXr5PM n nnnU $s snnnnfs snnnf)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!?H03H0A B0A,.6A4A!90AC B9< =IQ1[q 'E = L C =s B !Bc "[R"U5(d![R"U[U55nO#[U5[U5:wa [ S5e[ X5VVs/sHup4[ X4U5PM nnnU$s snnf)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_segmentsrsz ;;r?? __RU , r7c%j LM M #5~ /-  T 2-  / K /s.B c d[R"U[U55nURSSS9n[ X15VVVVs/sHuuupEpgXEU4U4PM nnnnnU(a.[ U6upU VVVs/sHupEn[ XEXr5PM n nnnO/n /n U [ U 54$s snnnnfs snnnf)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; && 9sA5c\rSrSrSrSrSrg) Collection3DizA collection of 3D paths.c URVVs/sHupUPM nnnUVs/sH$n[/URQURP76PM& nnUR(aA[ XC5VVs/sH)upQU[ /URQURP76-PM+ nnn[ X45VVs/sH^up[R"U5(a<[RRU[R"XQR5S9OUPM` nnnUVs/sH;n[R"USS2S4USS2S4USS2S4UR5PM= nn[ X`R5VVVV s/sH@uupxnup)[R "[RR#Xx/5U 5PMB sn nnnUl[R&"UVV s/sHu p*U PM sn n5n [)U 5(aU R+5$S$s snnfs snfs snnfs snnfs snfs sn nnnfs sn nf)z0Project the points according to renderer matrix.r9NrrrZeA)_3dverts_codesrBTr8rgrr:r%rmrnr&rrrrsmpathPath column_stack_paths concatenater(min) rNvs_vs_listmasksm xyzs_listr5r6csr7s rdo_3d_projectionCollection3D.do_3d_projections#'#6#67#6%"2#67AHI2$6bdd6DII6I   "%e"57"588dii88"5 7!$G 35 3urffQii55;;r88(D;E%'( 3 5@GH?F11 q!tHbAh1a4$))5?F H47yBUBU3VX3V/;BAzz"%%"4"4bX">C3VX ^^ : HAqR : ;r77rvvx++8I75HX:s+H+H"10H'2A%H-AH3[TU]"U40UD6 X lgr)r[rLrg)rNrrOrPr`s rrLLine3DCollection.__init__s  )&)% rcXlSUlgz&Set the position to use for z-sorting.TN _sort_zposrdrNvals r set_sort_zposLine3DCollection.set_sort_zpos rc0>Xl[TU] /5 g)z Set 3D segments. N) _segments3dr[ set_segments)rNrr`s rrLine3DCollection.set_segments s$ R rc.URnUVs/sHn[U5PM nn[[U55S:nU(aN[SU55(a [R R U5nO-[R "U5nO[R"U5nURS:Xa'[R"U/5 [R$Sn[R RU5(a3UR[R RLa URn[USUSUSUR 5n[R"U5(a<U[R""US[R$4/UR&QS P75-nUR((ay[+USUSUSUR 5n[R"U5(a=[R""US[R$4/UR&QS P75nXW-n[,R."XR 5nUSLa[R R1XS 9nUSSS 24n U(a.[R2"U [R4"US S 55n [R"X 5 [U5S:a[7USR75S5n U $[Rn U $s snf)z2 Project the points according to renderer matrix. rc3`# UH$n[RRU5v M& g7fr)r%rnisMA).0segments r 4Line3DCollection.do_3d_projection..s ?h7255::g&&hs,.rF.r.r.rZ.r#rrZNr)rr(setrmr%rnr asanyarraysizer rrqrr9nomaskrBr8rnewaxisrrgr:r_scale_proj_transform_vectorsr&splitcumsumr) rNrrsegment_lengthsraggedr9r viewlim_maskxyzs segments_2dminzs rr!Line3DCollection.do_3d_projections##8@@xG3w>(3}}X.H ==A   ' 'b 166M 55::h  HMM$E==D(&)9)1&)9)1&)9)-4  66*   BOOJsBJJ$?$:j&6&6$:$:< [TU]"U0UD6 URXU5 g)a Parameters ---------- verts : zs : float The location along the *zdir* axis in 3D space to position the patch. zdir : {'x', 'y', 'z'} Plane to plot patch orthogonal to. Default: 'z'. 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 N)r[rLrM)rNr7r*rOrrPr`s rrLPatch3D.__init__[s' $)&) r4rc [R"U[U55n[X5VVVs/sHuupVn[ XVXs5PM snnnUlX@lgs snnnf)a Set the *z* position and direction of the patch. Parameters ---------- verts : zs : float The location along the *zdir* axis in 3D space to position the patch. zdir : {'x', 'y', 'z'} Plane to plot patch orthogonal to. Default: 'z'. 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 N)r%rr(rr _segment3drg)rNvertsr7r*rOr r!r"s rrMPatch3D.set_3d_propertiesnsS$__RU ,.1%n>.<{'qQ5.<>%>sAc[US5(d9URR5URlUR 5 UR $)N_path2d)hasattrr8get_projMrr(rVs rget_pathPatch3D.get_paths?tY'')),,.DIIK  ! ! #||rcRURn[U6up#n[X#X@R5nUR(aU[ X#X@R5-n[ R"U5(aJ[ RR[U6[US9R[ R5upgnOX#Upn[R"XgXR5upp[R "[ RR#X/55Ul['U 5$Nr)r$rrBr8rgr:r%rmrnr&rorprqrrrrrr(r rNsxs0ys0zs0r9r5r6r7vxsvysvzsviss rrPatch3D.do_3d_projections OOQ #"3S))<    M#C; ;D 66$<<S!W+0t%==CVBFF^ BB3BB#>>rr99U#zz"%%"4"4cZ"@A 3xr)rgr(r$r) rrrrrrLrMr,rrrrs@rrrVs/"$#%55&&.  rrc:\rSrSrSrSSSS.SjrS SjrS rS rg ) PathPatch3Diz 3D PathPatch object. rr"Fr c V[R"U40UD6 URXX45 g)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)rrLrM)rNr r7r*rOrPs rrLPathPatch3D.__init__s%" t&v& t:rcd[RXRX#US9 URUlg)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. axlim_clip : bool, default: False Whether to hide path patches with a point outside the axes view limits. .. versionadded:: 3.10 r N)rrMrr_code3d)rNr r7r*rOs rrMPathPatch3D.set_3d_propertiess-$ !!$ "-7 " 9zz rchURn[U6up#n[X#X@R5nUR(aU[ X#X@R5-n[ R"U5(aJ[ RR[U6[US9R[ R5upgnOX#Upn[R"XgXR5upp[R "[ RR#X/5UR$5Ul[)U 5$r/)r$rrBr8rgr:r%rmrnr&rorprqrrrrrr?r(rr0s rrPathPatch3D.do_3d_projections OOQ #"3S))<    M#C; ;D 66$<<S!W+0t%==CVBFF^ BB3BB#>>rr99U#zz"%%"4"4cZ"@$,,O 3xr)r?r(Nr) rrrrrrLrMrrrrrr;r;s$&CE;(", rr;cUR5nUR5nURU5n[U5(aUS$[R "/5$)z2Return a list of vertices for the path of a patch.r)get_patch_transformr, to_polygonsr(r%r&)patchtransr polygonss r_get_patch_vertsrIsK  % % 'E >> D&Hh--8A;9RXXb\9rcT[U5n[UlURXAX#5 g)z*Convert a `.Patch` to a `.Patch3D` object.N)rIrr`rM)rFr"r*rOr%s rpatch_2d_to_3drKs# U #EEO Ed7rcUR5nUR5nURU5n[UlUR XQU5 g)z2Convert a `.PathPatch` to a `.PathPatch3D` object.N)r,rDtransform_pathr;r`rM) pathpatchr"r*r rGrs rpathpatch_2d_to_3drOsF    D  ) ) +E   &E%I $/rc^\rSrSrSrSSSSSS.U4SjjrS rSS jrS rSS jr U4S jr Sr U4Sjr U4Sjr SrU=r$)Patch3DCollectioniz A collection of 3D patches. rr"NFr7r* depthshadedepthshade_minalpharOc>Uc [SnUc [SnX0lX@l[TU]"U0UD6 UR XU5 g)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_minalphar[rLrM rNr7r*rSrTrOrrPr`s rrLPatch3DCollection.__init__sV8  !"56J  &"*+G"H %$7! $)&) r4rcUR$rrXrVs rget_depthshade Patch3DCollection.get_depthshadercBUc [SnXlX lSUlg)aR 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 NrWTr rXrYrdrNrSrTs rset_depthshade Patch3DCollection.set_depthshade")$  &"*+G"H %$7! rcXlSUlgrrrs rrPatch3DCollection.set_sort_zpos:rrcUR5 UR5n[U5S:aURupVO/n/n[ XV[ R "U5U5Ul[S5Ul SUl X0l SUl g)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. axlim_clip : bool, default: False Whether to hide patches with a vertex outside the axes view limits. .. versionadded:: 3.10 rr NT) update_scalarmappable get_offsetsr(rrr% atleast_1d _offsets3dslice_z_markers_idx_vzsrgrdrNr7r*rOoffsetsr5r6s rrM#Patch3DCollection.set_3d_properties?sz( ""$""$ w[/URQURP76nUR(a$U[ /URQURP76-n[ R "U5(a,[ RRURUS9up#nOURup#n[R"X#X@R5upVpxXpl [ RRU5(a.[T U]9[ RRXV/55 O$[T U]9[ R"XV/55 UR S:a [#U5$[ R$$)Nrr)rBrmr8rgr:r%rmrnr&rrrprr[ set_offsetsrr rrq) rNr9r5r6r7r5r6r7r8r`s rr"Patch3DCollection.do_3d_projection`s"?DOO?TYY?    M>4??>DII> >D 66$<<T__4@JBBJBB#>>rr99U# 55::c?? G  2 2C: > ? G  ; < 88a<s8O66MrcURnURbQUR(a@[UURURS9nUbUR SS:Xa X!SS2S4-n[ U5S:aKXRn[R"U5S:a#[R"U5URn[R"X5$)N) min_alpharr#r) _alpharprX_zalpharYrr(ror%ndimrrmcolors to_rgba_arrayrN color_arrayalphas r"_maybe_depth_shade_and_sort_colors4Patch3DCollection._maybe_depth_shade_and_sort_colorsts  99 T%5%5! 33K  [%6%6q%9Q%>AqD 11 { a %&9&9:Kwwu~! 5)$*=*=>$$[88rc>>UR[TU] 55$rrr[ get_facecolorrNr`s rrPatch3DCollection.get_facecolor66uw7L7NOOrc>[R"URS5(aUR5$UR [ TU]55$Nfacerr$ _edgecolorsrrr[ get_edgecolorrs rrPatch3DCollection.get_edgecolorF   D,,f 5 5%%' '66uw7L7NOOr)rgrXrYrmrrprordrr)rrrrrrLr^rdrrMrrrrrrrs@rrQrQsZ  #5#5J !0 B(9,PPPrrQc\[RRU5(dgUR5UR 5- nUR5UR 5- nUR5UR 5- n[R "US-US--US--5$)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. rrZ)r%rncountmaxrsqrt)XYZptp_xptp_yptp_zs r_get_data_scalers 55;;q>> EEGaeeg E EEGaeeg E EEGaeeg E 775A: *UaZ7 88rc^\rSrSrSrSSSSSS.U4SjjrU4S jrS rSS jrSU4S jjr U4S jr Sr SSjr Sr \U4Sj5rSrU4SjrU4SjrSrU=r$)Path3DCollectioniz A collection of 3D paths. rr"NFrRc>Uc [SnUc [SnX0lX@lSUl[TU]"U0UD6 UR XU5 SUlg)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. NrVrWF)r rXrY_in_drawr[rLrM_offset_zorderedrZs rrLPath3DCollection.__init__se8  !"56J  &"*+G"H %$7!  $)&) r4 $rc>UR5 [R"USS9 [TU]U5 SSS5 SSS5 g!,(df  N=f!,(df  g=f)NT)r)_use_zordered_offsetrrwr[rx)rNryr`s rrxPath3DCollection.drawsH  & & (""4$7 X&8) (77) (s!AA A A A A(cXlSUlgrrrs rrPath3DCollection.set_sort_zposrrcUR5 UR5n[U5S:aURupVO/n/nX l[ XV[ R"U5U5UlURUl [ R"UR5Ul URupVn[S5UlSUlX0lSUlg)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 rr NT)rjrkr(r_zdirrr%rlrm_sizes_sizes3dr& _linewidths _linewidths3drnrorprgrdrqs rrM"Path3DCollection.set_3d_propertiess( ""$""$ w[TU]X5 UR(dXlggr)r[ set_sizesrr)rNsizesdpir`s rrPath3DCollection.set_sizess! %%}}!Mrc>[TU]U5 UR(d&[R"UR 5Ulggr)r[ set_linewidthrr%r&rr)rNlwr`s rrPath3DCollection.set_linewidths3 b!}}!#$*:*:!;D rcUR$rr]rVs rr^Path3DCollection.get_depthshade!r`rcBUc [SnXlX lSUlg)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 Sets the minimum alpha value used by depth-shading. .. versionadded:: 3.11 NrWTrbrcs rrdPath3DCollection.set_depthshade$rfrc,SnURH7n[RRU5(dM)XR-nM9 U[ /URQUR P76-nUR(a$U[/URQUR P76-n[R"U5(an[R"U[UR5/URSRQ75n[RRURUS9nO URn[R"/UQUR P76upEpg[!XEU5Ul[RR%U5SSS2=olX`l[UR*5S:aUR*UUl[UR.5S:aUR.UUl[2R4"U[RR7XE455 XhnXHnXXn[RR7XE45UlUR:(a[R<"U5$[R>$)NFrrr r) rmr%rnrr9rBr8rgr:rmrr(rr&rrr _data_scaleargsortrorprrrrrrurrr rrq) rNr9r_rr5r6r7r8 z_markers_idxs rr!Path3DCollection.do_3d_projection<s??Cuuzz#hh#)F4??FDIIFF   -DD$))DDD 66$<<??4$'$8#T4??1;M;S;S#TVD55;;tT;:D??D#>>PPdiiP#*3S9/1eemmC.@2.FF +  t}}  !-- 6DK t!! "Q &#11-@D ""4););SJ)GH   !# 2 2C: >!hhrvvc{2BFF2rc#># URcSv gURn[TU] UR5 Sv Xlg!Xlf=f7fr)r_offsetsr[ru)rN old_offsetr`s rr%Path3DCollection._use_zordered_offsetisE  ( J G  5 5 6 + * s8AAAAAcURnURb\UR(aK[UURURUR S9nUbUR SS:Xa X!SS2S4-n[U5S:aKXRn[R"U5S:a#[R"U5URn[R"X5$)N)rxrrryr#r)rzrprXr{rYrrr(ror%r|rrr}r~rs rr3Path3DCollection._maybe_depth_shade_and_sort_colorsws  99 T%5%5! 33 ,, K  [%6%6q%9Q%>AqD 11 { a %&9&9:Kwwu~! 5)$*=*=>$$[88rc>>UR[TU] 55$rrrs rrPath3DCollection.get_facecolorrrc>[R"URS5(aUR5$UR [ TU]55$rrrs rrPath3DCollection.get_edgecolorrr)rgrrXrYrrrrrrmrrrrprorrdr)gR@r)rrrrrrLrxrrMrrr^rdrrrrrrrrrs@rrrs~  %%%%N'  2h" <  !0+3Z + +9.PPPrr)rTc[U[5(a[UlSUlO [U[ 5(a [ UlUc [SnUc [SnX0lXPl SUl URXU5 g)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 NrVrWF) isinstancerrr`rrrQr rXrYrrM)rr7r*rSrOrTrs rpatch_collection_2d_to_3drszH#~&&( # C ) )) 12 "&'CD O2CL"J/rc^\rSrSrSrSSSSS.U4Sjjr\R\R\RS.r S r \ R"S 5S 5rS rSU4S jjrSrSSjrSrSrU4SjrU4SjrSrSrSrSrU=r$)Poly3DCollectionia 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 lightsourcerOc>U(ad[U5nURSS5n U b[XU5US'URSS5n U b[XU5US'U cU c [S5e[T U]"U/UQ70UD6 [ U[R5(aURS:wa [S5eO"[SU55(a [S5eURU5 SUl XPl g)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-likec3f# UH'n[[R"U55S:gv M) g7f)rZN)r(r%r)rverts rr,Poly3DCollection.__init__..s">3rxx~&!+s/1)_generate_normalsget _shade_colorsr)r[rLrr%ndarrayr|rm set_zsort_codes3drg) rNr%rrrrOrrPnormalsrrr`s rrLPoly3DCollection.__init__sH '.GL$7J%'4(|$ L$7J%'4(|$!j&8 %&& 000 eRZZ ( (zzQ !LMM>>>> !LMM u %r)rrrcHURUUlSUlSUlg)z 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 _zsortfuncrrd)rNrs rrPoly3DCollection.set_zsort(s$//6 rz3.10c$URU5$r) _get_vector)rNrs r get_vectorPoly3DCollection.get_vector6s ++rc[U[R5(a|[R"SUS9 [U[R R 5(a0URUlURRSS9Ul g XlSUl g [U5n[R"[[U5[RS9nUR!SS9n[R""X$S 45n[%U5HupgXuUS [U524'M XPl[R&"U5US S 2S 4:Ul g ) a& 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#)rr axisFrkr)initialr#N)rr%rr check_shapern MaskedArraydata_facesr9rm_invalid_verticesr(fromitermapintprempty enumeratearange)rNr num_faces num_verts max_vertsrirs rrPoly3DCollection._get_vector:s j"** - -   _ D*bee&7&788(oo )3)<)<")<)E&( ).&JI CZ$8HI! a 0Ixxq 9:H$Z0*.JSYJ'1"K%'YYy%9Yq$w=O%OD "rcT>URU5 [TU] /S5 X lg)af 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)rr[ set_verts_closed)rNr%closedr`s rrPoly3DCollection.set_vertsYs'  "e$ rc0URUSS9 X lg)z Set 3D vertices with path codes.F)rN)rr)rNr%rs rset_verts_and_codes$Poly3DCollection.set_verts_and_codesls uU+ rcUR5 SUlURS5 [R"U5Ul[R "U5Ul[R"U5Ul SUl g)NrT) rjrrrr _facecolor3dr _edgecolor3d get_alpha_alpha3drd)rNrOs rrM"Poly3DCollection.set_3d_propertiestsb ""$ y!*88>*88>&006  rcXlSUlgrrrs rrPoly3DCollection.set_sort_zposrrc URbTUR5 UR(aURUlUR (aUR Ul[UR5nUR[URSURSURSUR5-n[R"U5n[R"SSS9 [ R""URUR5nSSS5 UR$(a`['URSURSURSUR5n[R"U5(aSnX%-nWSnU(a[R(R+XbS9nURnURn[U5U:waUR-US S 9n[U5U:wa"[U5S :XaUnOUR-US S 9n[U5S :aUR/US S 9n OUn U(a U R0n [R2"U S S 9SSS 2n [U5S :a XJSS2SS 24n OUn UR4b[UR45S :akU(a.X*SS24)n [7X5V Vs/sH upXSS24PM n n nU Vs/sHoR4UPM nn[8R:"X U5 O}U(aU[U 5S :aF[R<"X*SS2S4U R>5n[R(R+U US9n [8R@"X URB5 [U5S :a XzUl"OXpl"[UR5[U5:Xa[U5S :a XUl#OURUl#URHb[[RJ"S /S /URH/S //5n[ RL"UURRN5nUS S $URPS :a[RR"U5$[RT$!,(df  GN=fs snn fs snf)z, Perform the 3D projection for this object. Nrrr ignore)invaliddivideTrrrr rZr)+_Arj_face_is_mapped _facecolorsr_edge_is_mappedrrr(rrrBr8r%rmerrstaterrrgr:rnrrepeatrrrrrrrrrrr _facecolors2d _edgecolors2drr&_proj_transform_vecr+r rrq)rNrr9 needs_maskingpfacesrpzscfacecedgeface_z face_orderfaces_2d segment_maskridxrinvalid_vertices_2dzvecztranss rr!Poly3DCollection.do_3d_projections 77   & & (##$($4$4!##$($4$4! $ %%(; KK V!4 KK ),,t [[( ;99$++tyyQF<   (V)Q)-V):C]]3':E>  . .tu EX!2&(ooQ,-NN'$#55,, ':-<  $ $TT\\ B u:>!&!2D !&  t  !SZ /CJN!&!2D !%!2!2D  ?? &88aS1#'81#>?D//diikkBF!9Q<  XX\66#; 66MS< ;P?>s.,R(R:0S( R7cZ>[TU]U5 [R"U5Ulgr)r[ set_facecolorrrrrNrr`s rr*Poly3DCollection.set_facecolor# f%*88>rcZ>[TU]U5 [R"U5Ulgr)r[ set_edgecolorrrrr+s rr/Poly3DCollection.set_edgecolorr-rc[RRX5 [R"UR UR 5Ul[R"URUR 5Ul SUl g![[[4a NPf=f![[[4a N:f=f)NT) rArtist set_alphar}r~rrzAttributeError TypeError IndexErrorrrrd)rNrs rr3Poly3DCollection.set_alphas ,  ' 5 5!!4;;!0D  &44%%t{{ 4D   :6     :6   s#0B 0B% B"!B"%B=<B=c[US5(d9URR5URlUR 5 [ R "UR5$)Nr)r)r8r*r+rr%rrrrVs rrPoly3DCollection.get_facecolorJt_--)),,.DIIK  ! ! #zz$,,--rc[US5(d9URR5URlUR 5 [ R "UR5$)Nr)r)r8r*r+rr%rrrrVs rrPoly3DCollection.get_edgecolor r:r)r rgrrrrrrrrrrrrd)Tr)rrrrrrLr%rrrrrr deprecatedrrrrrMrrr*r/r3rrrrrs@rrrs(,5E!e>&>&B::vvvv  __V,,P>&  cJ? ? ...rrc[UR5X5upE[UlUR XE5 UR 5 X0lg)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`rrMrg)rr7r*rO segments_3drs rpoly_collection_2d_to_3dr@sB: MMOR'K$CMK/ Orc\US:XaX U4$US:XaXU4$USS:Xa [XX#5$XU4$)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*sH s{rz rz aC22,,rzrc2US;aXU4$US;aX U4$XU4$)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!rrDs rrCrC;s2  {rz  rzrzrc[U5S:Xd[U5S:Xa[R"S5$[R"USS5nUbUS:Xa[R"U5nO5[R"SU[R "U5- U- - US5n[R "[R"U5[U5S45n[R"USS2SS24USS2S4U-/5$)z?Modify the alpha values of the color list according to z-depth.r)rryrNryr#) r(r%zerosclip ones_likerrr}r~r)rr7rxrsatsrgbas rr{r{Is 6{a3r7a<xx 1a(IkQ.||BwwqBO{::IqI ??70083r7A, GD ??DBQBKadd):; <T"T"U55$rr)r in_normout_norms rrQ_shade_colors..normsGAJ' 'rrr#)r} LightSourcer%rrPrQ directionrNrm Normalizeinverser~rr copy) colorrrrr9rQrrrnros @@rrrs# ))GD X &BIINN7TNJJ(() ' HHUO D xxzz##B*$$S!,44 (e %%e,ad eQ ]+e3q!t  Mu%**, M3 ' &s 0E Er)rr")rr"NF)rlN)g:0yE>r)@rrtnumpyr% contextlibr matplotlibrrrrr}rr rJr rr matplotlib.collectionsr r rrrmatplotlib.patchesrrrrrr+r:rBrKrDrLine2Drrrrrrrrrrrr;rIrKrOrQrrrrr@rrCr{r]rrrrrr}sf   %QQ$  OD2=0vUZZvr/(kU\\k\1*   ',:,* !g~gT!DeDN<'<~:80^P^PB9.jP~jP^  0000fC.~C.L !." " =8&+R)X#r