o iN@sddlmZddlZddlmZddlmZmZmZddl m Z ddl m ZddlZddlmZmZddlmZmZddlmZmZddlmZmZdd lmZerpdd lm Z dd l!m"Z"dd l#m$Z$dd l%m&Z&ddl'm(Z)GdddeZ*Gddde*Z+Gddde*Z,dS)) annotationsN)pairwise) TYPE_CHECKINGAnycast)FillTypeLineType)convert_filled convert_lines) as_fill_type as_line_type)filled_to_mpl_pathslines_to_mpl_paths)Renderer)Sequence)Axes)Figure) ArrayLikec@seZdZUdZded<ded<ded<   dXdYddZdZddZdZddZd[d!d"Z # $ %d\d]d.d/Z # 0 1 #d^d_d7d8Z # $ 9 d`dad?d@Z # 0dbdcdCdDZ dddedHdIZ dfdKdLZdZdMdNZdgdhdPdQZ # R S EdidjdVdWZd S)k MplRendereraUtility renderer using Matplotlib to render a grid of plots over the same (x, y) range. Args: nrows (int, optional): Number of rows of plots, default ``1``. ncols (int, optional): Number of columns of plots, default ``1``. figsize (tuple(float, float), optional): Figure size in inches, default ``(9, 9)``. show_frame (bool, optional): Whether to show frame and axes ticks, default ``True``. backend (str, optional): Matplotlib backend to use or ``None`` for default backend. Default ``None``. gridspec_kw (dict, optional): Gridspec keyword arguments to pass to ``plt.subplots``, default None. zSequence[Axes]_axesr_figbool _want_tight rTNnrowsintncolsfigsizetuple[float, float] show_framebackend str | None gridspec_kwdict[str, Any] | NonereturnNonec Cs|dur ddl}|||dddd}|dur||d<nddi|d<tj||fi|\|_} | |_|sB|jD]} | d q:d|_dS) NrFT)rZsqueezeZsharexZshareyr$ZaspectequalZ subplot_kwoff) Z matplotlibZusepltZsubplotsrflattenrZaxisr) selfrrrr!r"r$ZmplkwargsZaxesaxr/j/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/contourpy/util/mpl_renderer.py__init__,s        zMplRenderer.__init__cCst|dr t|jdSdS)Nr)hasattrr*closerr,r/r/r0__del__Hs zMplRenderer.__del__cCsT|jD]}t|ddr|jddd|_q|jr&t|jdkr(|jdSdSdS)N_need_autoscaleFT)Ztightr)rgetattrZautoscale_viewr6rlenrZ tight_layoutr,r.r/r/r0 _autoscaleLs   zMplRenderer._autoscaler. Axes | intrcCst|tr |j|}|SN) isinstancerrr9r/r/r0_get_axWs  zMplRenderer._get_axrC0ffffff?filledcpy.FillReturn fill_typeFillType | strcolorstralphafloatcCsDt|}||}t||}tj||dd|d}||d|_dS)a Plot filled contours on a single Axes. Args: filled (sequence of arrays): Filled contour data as returned by :meth:`~.ContourGenerator.filled`. fill_type (FillType or str): Type of :meth:`~.ContourGenerator.filled` data as returned by :attr:`~.ContourGenerator.fill_type`, or string equivalent ax (int or Maplotlib Axes, optional): Which axes to plot on, default ``0``. color (str, optional): Color to plot with. May be a string color or the letter ``"C"`` followed by an integer in the range ``"C0"`` to ``"C9"`` to use a color from the ``tab10`` colormap. Default ``"C0"``. alpha (float, optional): Opacity to plot with, default ``0.7``. nonerZ facecolorsZ edgecolorslwrGTN)r r>r mcollectionsPathCollectionadd_collectionr6)r,rArCr.rErGpaths collectionr/r/r0rA\s     zMplRenderer.filledblack皙?xry point_colorquad_as_tri_alphac Cs||}|||\}}||d}|j|||j|jfi||dkrd|ddddf|ddddf|ddddf|ddddf} d|ddddf|ddddf|ddddf|ddddf} ||d<|jt|ddddf| |ddddffdt|ddddf| |ddddffdt|ddddf| |ddddffdt|ddddf| |ddddffdfi||dur|j||||d dd d |_dS) aPlot quad grid lines on a single Axes. Args: x (array-like of shape (ny, nx) or (nx,)): The x-coordinates of the grid points. y (array-like of shape (ny, nx) or (ny,)): The y-coordinates of the grid points. ax (int or Matplotlib Axes, optional): Which Axes to plot on, default ``0``. color (str, optional): Color to plot grid lines, default ``"black"``. alpha (float, optional): Opacity to plot lines with, default ``0.1``. point_color (str, optional): Color to plot grid points or ``None`` if grid points should not be plotted, default ``None``. quad_as_tri_alpha (float, optional): Opacity to plot ``quad_as_tri`` grid, default 0. Colors may be a string color or the letter ``"C"`` followed by an integer in the range ``"C0"`` to ``"C9"`` to use a color from the ``tab10`` colormap. Warning: ``quad_as_tri_alpha > 0`` plots all quads as though they are unmasked. )rErGrg?NrrG)rWo)rErGmarkerrKT)r> _grid_as_2dplotTnpstackZreshaper6) r,rSrTr.rErGrUrVr-xmidymidr/r/r0gridys&  TT4444 zMplRenderer.grid?linescpy.LineReturn line_typeLineType | str linewidthc CsDt|}||}t||}tj|d|||d}||d|_dS)aWPlot contour lines on a single Axes. Args: lines (sequence of arrays): Contour line data as returned by :meth:`~.ContourGenerator.lines`. line_type (LineType or str): Type of :meth:`~.ContourGenerator.lines` data as returned by :attr:`~.ContourGenerator.line_type`, or string equivalent. ax (int or Matplotlib Axes, optional): Which Axes to plot on, default ``0``. color (str, optional): Color to plot lines. May be a string color or the letter ``"C"`` followed by an integer in the range ``"C0"`` to ``"C9"`` to use a color from the ``tab10`` colormap. Default ``"C0"``. alpha (float, optional): Opacity to plot lines with, default ``1.0``. linewidth (float, optional): Width of lines, default ``1``. rIrJTN)r r>rrLrMrNr6) r,rdrfr.rErGrhrOrPr/r/r0rds     zMplRenderer.linesz'ArrayLike | np.ma.MaskedArray[Any, Any]cCsTtj|}|tjjurdS||}|||\}}|j||||d|ddS)aPlot masked out grid points as circles on a single Axes. Args: x (array-like of shape (ny, nx) or (nx,)): The x-coordinates of the grid points. y (array-like of shape (ny, nx) or (ny,)): The y-coordinates of the grid points. z (masked array of shape (ny, nx): z-values. ax (int or Matplotlib Axes, optional): Which Axes to plot on, default ``0``. color (str, optional): Circle color, default ``"black"``. NrY)c)r^maZgetmaskZnomaskr>r[r\)r,rSrTrir.rEmaskr/r/r0rms   zMplRenderer.maskFfilename transparentcCs||jj||ddS)zSave plots to SVG or PNG file. Args: filename (str): Filename to save to. transparent (bool, optional): Whether background should be transparent, default ``False``. )roN)r:rsavefig)r,rnror/r/r0saveszMplRenderer.save io.BytesIOcCs.|t}|jj|dd|d|S)zhSave plots to an ``io.BytesIO`` buffer. Return: BytesIO: PNG image buffer. Zpng)formatr)r:ioBytesIOrrpseek)r,bufr/r/r0save_to_buffers  zMplRenderer.save_to_buffercCs|tdS)zMShow plots in an interactive window, in the usual Matplotlib manner. N)r:r*showr4r/r/r0rys zMplRenderer.showtitlecCs0|r||j||ddS|||dS)aSet the title of a single Axes. Args: title (str): Title text. ax (int or Matplotlib Axes, optional): Which Axes to set the title of, default ``0``. color (str, optional): Color to set title. May be a string color or the letter ``"C"`` followed by an integer in the range ``"C0"`` to ``"C9"`` to use a color from the ``tab10`` colormap. Default is ``None`` which uses Matplotlib's default title color that depends on the stylesheet in use. )rEN)r> set_title)r,rzr.rEr/r/r0rzs zMplRenderer.titlegreen.1ffmt quad_as_tric Cs@||}|||\}}t|}|j\}} t|D]$} t| D]} |j|| | f|| | f|| | f|dd|ddq!q|rt|dD]U} t| dD]J} tj|| | d| | dftjd} tj|| | d| | dftjd} t|| | d| | df}|j| | ||dd|ddqPqHdSdS)aShow ``z`` values on a single Axes. Args: x (array-like of shape (ny, nx) or (nx,)): The x-coordinates of the grid points. y (array-like of shape (ny, nx) or (ny,)): The y-coordinates of the grid points. z (array-like of shape (ny, nx): z-values. ax (int or Matplotlib Axes, optional): Which Axes to plot on, default ``0``. color (str, optional): Color of added text. May be a string color or the letter ``"C"`` followed by an integer in the range ``"C0"`` to ``"C9"`` to use a color from the ``tab10`` colormap. Default ``"green"``. fmt (str, optional): Format to display z-values, default ``".1f"``. quad_as_tri (bool, optional): Whether to show z-values at the ``quad_as_tri`` centers of quads. Warning: ``quad_as_tri=True`` shows z-values for all quads, even if masked. centerTZhavarEZclip_onrZdtypeN) r>r[r^asarrayshaperangetextmeanZfloat64)r,rSrTrir.rEr~rnynxjixxyyzzr/r/r0z_values s,     ,(("zMplRenderer.z_values)rrrTNN)rrrrrr r!rr"r#r$r%r&r')r&r')r.r;r&r)rr?r@) rArBrCrDr.r;rErFrGrHr&r')rrQrRNr)rSrrTrr.r;rErFrGrHrUr#rVrHr&r')rr?rcr)rdrerfrgr.r;rErFrGrHrhrHr&r')rrQ) rSrrTrrirjr.r;rErFr&r')F)rnrFrorr&r')r&rr)rN)rzrFr.r;rEr#r&r')rr|r}F)rSrrTrrirr.r;rErFr~rFrrr&r')__name__ __module__ __qualname____doc____annotations__r1r5r:r>rArbrdrmrqrxryrzrr/r/r/r0rsT       ! 3 $  rcs*eZdZdZ   d dfd d ZZS)MplTestRendererzTest renderer implemented using Matplotlib. No whitespace around plots and no spines/ticks displayed. Uses Agg backend, so can only save to file/buffer, cannot call ``show()``. rrrrrrr r&r'cshddddddd}tj|||dd|d|jD]}|d|d|g|gqd|_dS) Ng{Gz?gGz?)leftrighttopbottomZwspaceZhspaceTZAgg)r!r"r$F)superr1rZ set_xmarginZ set_ymarginZ set_xticksZ set_yticksr)r,rrrZgridspecr. __class__r/r0r1?s        zMplTestRenderer.__init__)rrr)rrrrrr r&r')rrrrr1 __classcell__r/r/rr0r9s rcseZdZdZ    dEdFfdd ZdGddZ       !dHdIfd+d, Z   -   !dJdKfd3d4 Z  dLdMd9d:Z  ;dNdMd  ?dOdPdCdDZ Z S)QMplDebugRendererzDebug renderer implemented using Matplotlib. Extends ``MplRenderer`` to add extra information to help in debugging such as markers, arrows, text, etc. rrTrrrrr r!rr&r'cst||||dSr<)rr1)r,rrrr!rr/r0r1`szMplDebugRenderer.__init__r.r line_startcpy.CoordinateArrayline_endrErFrGrH arrow_sizec Csd||}||}|tt||}t|d|d f} t||d| |||d|||d| |f} |j| dddf| dddfd||ddS)Ng?rr-rkrG)r^sqrtdotrr_r\) r,r.rrrErGrmidZalongrarrowr/r/r0_arrowis 0zMplDebugRenderer._arrowrC1r@r?redrRrArBrCrDr; line_color line_alpharUstart_point_colorc  st|}t||||||dur|durdS||}t||tj}|dur{t|D]N\} } | dur5q,t| D]@\} }| | |}|j |dddf|dddf||d| dkryt |}t |dD]}| |||||d||| qfq9q,|durt|D]d\} } | durqt j| dtd}d|| ddd<| dur| dd}d||<|j | dddf|| dddf|d||d| dur|j | dddf|| dddf|d| |dqdSdS) NrrrrrWrFrY)r rrAr>r rZChunkCombinedOffsetziprr\r8rrr^Zonesr)r,rArCr.rErGrrrUrrZpointsoffsetsstartendZxysnrrmZ start_indicesrr/r0rA}sL   *"  **zMplDebugRenderer.filledrcrdrerfrgrhc  sTt|}t||||||| dkr|durdS||}t||tj} tr-tt j | } | dkrQ| D]} t t | dD]} | || | | | d||| q=q3|dur| D]P} d} t | }|dur|j| d| dd||dd} | dd| ddkr| dd| ddkr|d8}|j| | |df| | |dfd||dqWdSdS) Nrrr)rr)rrrYrrW)r rrdr>r rZSeparaterrcpyZLineReturn_Separaterr8rr\)r,rdrfr.rErGrhrUrrZseparate_linesliner start_indexZ end_indexrr/r0rds6   "0"zMplDebugRenderer.linesrSrrTric Cs||}|||\}}t|}|j\}}t|D]%}t|D]} | ||} |j||| f||| ft| dd|ddq!qdS)NrrTrr>r[r^rrrrrF) r,rSrTrir.rErrrrquadr/r/r0 point_numberss      $zMplDebugRenderer.point_numbersbluec Cs||}|||\}}t|}|j\}}td|D]F}td|D]>} | ||} ||d|d| d| df} ||d|d| d| df} |j| | t| dd|ddq#qdS)NrrTr) r>r[r^rrrrrrF) r,rSrTrir.rErrrrrr`rar/r/r0 quad_numberss    ((zMplDebugRenderer.quad_numbersNr| lower_level upper_level float | Nonec Cs||}|||\}}t|}|j\}} t|D]9} t| D]2} || | f} |dur4| |kr4d} n | |kr;d} nd} |j|| | f|| | ft| dd|ddq!qdS)NrrrrrTrr)r,rSrTrirrr.rErrrrrZz_levelr/r/r0z_levelss"      $zMplDebugRenderer.z_levels)rrrT) rrrrrr r!rr&r')r.rrrrrrErFrGrHrrHr&r')rrr@r?r@r?rrR)rArBrCrDr.r;rErFrGrHrrFrrHrUrFrrFrrHr&r')rr?rcrr?rrR)rdrerfrgr.r;rErFrGrHrhrHrUrFrrFrrHr&r')rr) rSrrTrrirr.r;rErFr&r')rr)Nrr|)rSrrTrrirrrHrrr.r;rErFr&r') rrrrr1rrArdrrrrr/r/rr0rZsF 9-  r)- __future__rrt itertoolsrtypingrrrZmatplotlib.collections collectionsrLZmatplotlib.pyplotZpyplotr*numpyr^Z contourpyrrZcontourpy.convertr r Zcontourpy.enum_utilr r Zcontourpy.util.mpl_utilr rZcontourpy.util.rendererrcollections.abcrZmatplotlib.axesrZmatplotlib.figurerZ numpy.typingrZcontourpy._contourpyZ _contourpyrrrrr/r/r/r0s.           !