o
    h                      @  s2  d dl mZ d dlmZmZ d dlZd dlmZmZ d dl	Z	d dl
Zd dlZ
d dlZd dlmZ d dlmZ d dlmZ d dlm  mZ erMd dlmZ 				d<d=ddZd>ddZd?ddZd@dd ZdAd"d#ZdBd%d&Z dCd(d)Z!dDd+d,Z"dCd-d.Z#dEd/d0Z$dFd1d2Z%dFd3d4Z&dGd7d8Z'dHd:d;Z(dS )I    )annotations)
CollectionIteratorN)TYPE_CHECKINGcast)MatplotlibColor)find_stack_level)is_list_like)Colormapdefault
num_colorsintcolormapColormap | None
color_typestrcolor3dict[str, Color] | Color | Collection[Color] | Nonec                 C  s.   t |tr|S t|||| d}tt|| d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.
    r   r   r   r   r   )
isinstancedict_derive_colorslist_cycle_colors)r   r   r   r   colors r   u/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/pandas/plotting/_matplotlib/style.pyget_standard_colors   s   
&r    Color | Collection[Color] | Nonestr | Colormap | Nonereturnlist[Color]c                 C  sP   | du r|durt ||dS | dur"|durtjdt d t| S t||d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_colormapwarningswarnr   _get_colors_from_color_get_colors_from_color_typer   r   r   r   r   P   s   'r   r   Iterator[Color]c                 c  s,    t |t| }tt| |E dH  dS )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)maxlen	itertoolsislicecycle)r   r   
max_colorsr   r   r   r      s   r   str | Colormapc                   s&   t |   fddtjdd|dD S )zGet colors from colormap.c                   s   g | ]} |qS r   r   .0numcmapr   r   
<listcomp>       z-_get_colors_from_colormap.<locals>.<listcomp>r      )r3   )_get_cmap_instancenplinspace)r   r   r   r4   r   r$      s   r$   r
   c                 C  s4   t | tr| }tj|  } | du rtd| d| S )z$Get instance of matplotlib colormap.Nz	Colormap z is not recognized)r   r   mpl	colormaps
ValueError)r   r5   r   r   r   r9      s   

r9   Color | Collection[Color]c                 C  sL   t | dkrtd|  t| rtt| } | gS ttt | } tt| S )z!Get colors from user input color.r   zInvalid color argument: )r+   r>   _is_single_colorr   Colorr   r   !_gen_list_of_colors_from_iterabler   r   r   r   r'      s   
r'   boolc                 C  s&   t | trt| rdS t| rdS dS )a0  Check 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_colorrC   r   r   r   r@      s
   r@   Collection[Color]c                 c  s,    | D ]}t |r|V  qtd| dS )zS
    Yield colors from string of several letters or from collection of colors.
    zInvalid color N)r@   r>   )r   xr   r   r   rB      s   rB   c                 C  s6   t t| ot| dkpt| dkotdd | D S )zACheck if color comprises a sequence of floats representing color.      c                 s  s    | ]
}t |ttfV  qd S )N)r   r   float)r2   rH   r   r   r   	<genexpr>   s    z#_is_floats_color.<locals>.<genexpr>)rD   r	   r+   allrC   r   r   r   rF      s   rF   c                 C  s(   | dkrt |S | dkrt|S td)z&Get colors from user input color type.r   randomz/color_type must be either 'default' or 'random')_get_default_colors_get_random_colorsr>   )r   r   r   r   r   r(      s
   r(   c                 C  s,   ddl m} dd |jd D }|d|  S )z=Get `num_colors` of default colors from matplotlib rc params.r   Nc                 S  s   g | ]}|d  qS rC   r   )r2   cr   r   r   r6      r7   z'_get_default_colors.<locals>.<listcomp>zaxes.prop_cycle)matplotlib.pyplotpyplotrcParams)r   pltr   r   r   r   rO      s   rO   c                 C  s   dd t | D S )z"Get `num_colors` of random colors.c                 S  s   g | ]}t |qS r   )_random_colorr1   r   r   r   r6      r7   z&_get_random_colors.<locals>.<listcomp>)ranger   r   r   r   rP      s   rP   columnlist[float]c                 C  s   t | }|d S )z4Get a random color represented as a list of length 3rI   )comrandom_staterandtolist)rX   rsr   r   r   rV      s   
rV   rA   c                 C  s0   t j }z||  W dS  ty   Y dS w )a  Check 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.
    FT)
matplotlibr   ColorConverterto_rgbar>   )r   convr   r   r   rE      s   
rE   )Nr   N)r   r   r   r   r   r   r   r   )
r   r   r   r    r   r   r   r   r!   r"   )r   r"   r   r   r!   r)   )r   r0   r   r   r!   r"   )r   r0   r!   r
   )r   r?   r!   r"   )r   r?   r!   rD   )r   rG   r!   r)   )r   r   r   r   r!   r"   )r   r   r!   r"   )rX   r   r!   rY   )r   rA   r!   rD   ))
__future__r   collections.abcr   r   r,   typingr   r   r%   r_   r<   matplotlib.colorsnumpyr:   pandas._typingr   rA   pandas.util._exceptionsr   pandas.core.dtypes.commonr	   pandas.core.commoncorecommonrZ   r
   r   r   r   r$   r9   r'   r@   rB   rF   r(   rO   rP   rV   rE   r   r   r   r   <module>   s>    
3
4


	





	



