PiM!`ddlmZddlmZmZmZddlZddlmZm Z m Z ddl Z ddl Z ddlZ ddlZddlmZddlmZddlmZerddlmZe d1d2dZe d1d dd3dZe d1d dd4dZ d5ddd4dZd6dZd7d"Zd8d$Zd9d&Zd:d(Zd;d*Zdd0Z#dS)?) annotations) CollectionIteratorSequenceN) TYPE_CHECKINGcastoverload)MatplotlibColor)find_stack_level) is_list_like)Colormap. num_colorsintcolormapColormap | None color_typestrcolordict[str, Color]returncdSNrrrrs u/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/plotting/_matplotlib/style.pyget_standard_colorsrs srColor | Sequence[Color] | None list[Color]cdSrrrs rrr's #r1dict[str, Color] | Color | Sequence[Color] | Nonedict[str, Color] | list[Color]cdSrrrs rrr1s &)Srdefaultct|tr|St||||}tt ||S)a Get standard colors based on `colormap`, `color_type` or `color` inputs. Parameters ---------- num_colors : int Minimum number of colors to be returned. Ignored if `color` is a dictionary. colormap : :py:class:`matplotlib.colors.Colormap`, optional Matplotlib colormap. When provided, the resulting colors will be derived from the colormap. color_type : {"default", "random"}, optional Type of colors to derive. Used if provided `color` and `colormap` are None. Ignored if either `color` or `colormap` are not None. color : dict or str or sequence, optional Color(s) to be used for deriving sequence of colors. Can be either be a dictionary, or a single color (single color string, or sequence of floats representing a single color), or a sequence of colors. Returns ------- dict or list Standard colors. Can either be a mapping if `color` was a dictionary, or a list of colors with a length of `num_colors` or more. Warns ----- UserWarning If both `colormap` and `color` are provided. Parameter `color` will override. rrrrr) isinstancedict_derive_colorslist _cycle_colors)rrrrcolorss rrr;sYN%  F  f<<< = ==r Color | Collection[Color] | Nonestr | Colormap | Nonec||t||S|3|"tjdtt |St ||S)aa Derive colors from either `colormap`, `color_type` or `color` inputs. Get a list of colors either from `colormap`, or from `color`, or from `color_type` (if both `colormap` and `color` are None). Parameters ---------- color : str or sequence, optional Color(s) to be used for deriving sequence of colors. Can be either be a single color (single color string, or sequence of floats representing a single color), or a sequence of colors. colormap : :py:class:`matplotlib.colors.Colormap`, optional Matplotlib colormap. When provided, the resulting colors will be derived from the colormap. color_type : {"default", "random"}, optional Type of colors to derive. Used if provided `color` and `colormap` are None. Ignored if either `color` or `colormap`` are not None. num_colors : int Number of colors to be extracted. Returns ------- list List of colors extracted. Warns ----- UserWarning If both `colormap` and `color` are provided. Parameter `color` will override. Nr(zC'color' and 'colormap' cannot be used simultaneously. Using 'color') stacklevel)_get_colors_from_colormapwarningswarnr _get_colors_from_color_get_colors_from_color_typer's rr+r+osvN }-(jIIII     MU+--    &e,,,*:*MMMMrr.Iterator[Color]c#Kt|t|}tjtj||Ed{VdS)zCycle colors until achieving max of `num_colors` or length of `colors`. Extra colors will be ignored by matplotlib if there are more colors than needed and nothing needs to be done here. N)maxlen itertoolsislicecycle)r.r max_colorss rr-r-sR ZV--J  7 7DDDDDDDDDDDrstr | Colormapcht|fdtjdd|DS)zGet colors from colormap.c&g|] }|Srr).0numcmaps r z-_get_colors_from_colormap..s! C C C#DDII C C Crr)rD)_get_cmap_instancenplinspace)rrrEs @rr3r3s= h ' 'D C C C CQz!B!B!B C C CCrr ct|tr)|}tj|}|t d|d|S)z$Get instance of matplotlib colormap.Nz Colormap z is not recognized)r)rmpl colormaps ValueError)rrEs rrHrHsK(C  C=*  AAAABB B OrColor | Collection[Color]ct|dkrtd|t|rtt|}|gStt t|}t t|S)z!Get colors from user input color.rzInvalid color argument: )r;rN_is_single_colorrColorrr,!_gen_list_of_colors_from_iterablers rr6r6s{ 5zzQ;E;;<<<UE""w E"E * *E 1%88 9 99rboolctt|trt|rdSt|rdSdS)a0Check if `color` is a single color, not a sequence of colors. Single color is of these kinds: - Named color "red", "C0", "firebrick" - Alias "g" - Sequence of floats, such as (0.1, 0.2, 0.3) or (0.1, 0.2, 0.3, 0.4). See Also -------- _is_single_string_color TF)r)r_is_single_string_color_is_floats_colorrs rrQrQsE%"9%"@"@tt 5rCollection[Color]c#^K|D]'}t|r|Vtd|dS)zS Yield colors from string of several letters or from collection of colors. zInvalid color N)rQrN)rxs rrSrSsS33 A   3GGGG1a1122 2 33rctt|o>t|dkpt|dkotd|DS)zACheck if color comprises a sequence of floats representing color.c3NK|] }t|ttfV!dSr)r)rfloat)rCrZs r z#_is_floats_color..s0;; 1sEl++;;;;;;r)rTr r;allrs rrWrWs_ U < ZZ1_ /E a < ;;U;;; ; ;  rc \|dkrLtjd}dtj|t |t |DS|dkrFt j||df Std)z&Get colors from user input color type.r%zaxes.prop_cyclecg|] }|d Srr)rCcs rrFz/_get_colors_from_color_type..s,    gJ   rrandomr\z/color_type must be either 'default' or 'random') rLrcParamsr<r=minr;rIre default_rngtolistrN)rr prop_cycles rr7r7sY\"34   %j#j#j//2R2RSS     x  y$$Z0077QHHOOQQQJKKKrrRctj} ||dS#t$rYdSwxYw)aCheck if `color` is a single string color. Examples of single string colors: - 'r' - 'g' - 'red' - 'green' - 'C3' - 'firebrick' Parameters ---------- color : Color Color string or sequence of floats. Returns ------- bool True if `color` looks like a valid color. False otherwise. TF) matplotlibr.ColorConverterto_rgbarN)rconvs rrVrVs\,   + + - -D Ut uus7 AA)..) rrrrrrrrrr) rrrrrrrrrr ) rrrrrrrr"rr#)Nr%) rr/rr0rrrrrr )r.r rrrr8)rr@rrrr )rr@rr )rrOrr )rrOrrT)rrXrr8)rrrrrr )rrRrrT)$ __future__rcollections.abcrrrr<typingrrr r4rlrLmatplotlib.colorsnumpyrIpandas._typingr rRpandas.util._exceptionsr pandas.core.dtypes.commonr r rr+r-r3rHr6rQrSrWr7rVrrrrxs""""""  333333444444222222+****** !$  !$ -0   !$) @C ))))) )!%1> @D 1>1>1>1>1>1>h1N1N1N1NhEEEEDDDD : : : :,3333 L L L Lr