iM!ddlmZddlmZmZmZddlZddlmZm Z m Z ddl Z ddl Z ddlZ ddlZddlmZddlmZddlmZerddlmZe d dd Ze dd d  dd Ze dd d  dd Z ddd  ddZ ddZddZ d dZd!dZ d"dZd#dZd$dZ d#dZ!d%dZ"d&dZ#y)') annotations) CollectionIteratorSequenceN) TYPE_CHECKINGcastoverload)MatplotlibColor)find_stack_level) is_list_like)Colormap.cyN num_colorscolormap color_typecolors v/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pandas/plotting/_matplotlib/style.pyget_standard_colorsrsrcyrrrs rrr'srcyrrrs rrr1s&)rcpt|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;s>N%  F  f< ==rc|| t||S|,|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. rzC'color' and 'colormap' cannot be used simultaneously. Using 'color') stacklevel)_get_colors_from_colormapwarningswarnr _get_colors_from_color_get_colors_from_color_typers rr!r!osWN }-(jII     MMU+- &e,,*:*MMrc#Kt|t|}tjtj||Ed{y7w)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#s7 ZV-J  7DDDsAA A A czt|}tjdd|Dcgc] }|| c}Scc}w)zGet colors from colormap.r)num)_get_cmap_instancenplinspace)rrcmapr5s rr'r's9 h 'D!#Qz!B C!B#DI!B CC Cs8crt|tr&|}tj|}|t d|d|S)z$Get instance of matplotlib colormap.z Colormap z is not recognized)rstrmpl colormaps ValueError)rr9s rr6r6s?(C ==*  y.@AB B Orct|dk(rtd|t|rtt|}|gStt t|}t t|S)z!Get colors from user input color.rzInvalid color argument: )r.r>_is_single_colorrColorrr"!_gen_list_of_colors_from_iterablers rr*r*s_ 5zQ3E7;<<UE"w E"E *E 1%8 99rcTt|tr t|ryt|ryy)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)rr;_is_single_string_color_is_floats_colorrs rr@r@s'%"9%"@ rc#TK|D]}t|r|td|yw)zS Yield colors from string of several letters or from collection of colors. zInvalid color N)r@r>)rxs rrBrBs1 A G~aS12 2 s&(ctt|xr2t|dk(xst|dk(xrtd|DS)zACheck if color comprises a sequence of floats representing color.c3HK|]}t|ttfywr)rintfloat).0rGs r z#_is_floats_color..s;U 1sEl+Us ")boolr r.allrs rrErEsG U < Z1_ /E a < ;U; ; rc L|dk(rMtjd}tj|t |t |Dcgc]}|d c}S|dk(r>t jj|j|dfjStdcc}w)z&Get colors from user input color type.defaultzaxes.prop_cyclerrandomrIz/color_type must be either 'default' or 'random') r<rcParamsr/r0minr.r7rT default_rngtolistr>)rr prop_cyclecs rr+r+sY\\"34 %%j#j#j/2RS S gJS   x yy$$Z077QHOOQQJKK s B!ctjj} |j|y#t$rYywxYw)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_rgbar>)rconvs rrDrDsD,    + + -D U s 2 >>)..) rrLrColormap | Nonerr;rdict[str, Color]returnra) rrLrr`rr;rzColor | Sequence[Color] | Nonerb list[Color]) rrLrr`rr;rz1dict[str, Color] | Color | Sequence[Color] | Nonerbzdict[str, Color] | list[Color])NrS) rz Color | Collection[Color] | Nonerzstr | Colormap | Nonerr;rrLrbrc)r$rcrrLrbIterator[Color])rstr | ColormaprrLrbrc)rrerbr )rColor | Collection[Color]rbrc)rrfrbrP)rzCollection[Color]rbrd)rr;rrLrbrc)rrArbrP)$ __future__rcollections.abcrrrr/typingrrr r(r\r<matplotlib.colorsnumpyr7pandas._typingr rApandas.util._exceptionsr pandas.core.dtypes.commonr r rr!r#r'r6r*r@rBrEr+rDrrrros"  342* !$      !$ -0  *     !$) @C )))) = ) $ ) )!%1> @D 1>1>1>1> = 1> $ 1>h1N +1N$1N 1N  1N  1NhEDDDD : $ : :,3 Lr