U
    ^{f                     @  s  d dl mZ d dlZd dlmZmZ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rd dlmZ d=d	d
dddddZdddd	ddddZdd	ddddZdd	ddddZdddd d!Zd"dd#d$d%Zd"d&d#d'd(Z d)dd#d*d+Z!d"d&d#d,d-Z"dd	dd.d/d0Z#d	dd1d2d3Z$d	dd1d4d5Z%d	d6d7d8d9Z&d:d&d#d;d<Z'dS )>    )annotationsN)TYPE_CHECKING
CollectionIteratorcast)MatplotlibColor)find_stack_level)is_list_like)ColormapdefaultintzColormap | Nonestrz3dict[str, Color] | Color | Collection[Color] | None)
num_colorscolormap
color_typecolorc                 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   `/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/pandas/plotting/_matplotlib/style.pyget_standard_colors   s    &
r   z Color | Collection[Color] | Nonezstr | Colormap | Nonezlist[Color])r   r   r   r   returnc                 C  sT   | dkr|dk	rt ||dS | dk	rD|dk	r<tjdt d t| S t||dS 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   N   s    'r   zIterator[Color])r   r   r   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   Z
max_colorsr   r   r   r      s    r   zstr | Colormap)r   r   r   c                   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>   s     z-_get_colors_from_colormap.<locals>.<listcomp>r      )r+   )_get_cmap_instancenpZlinspace)r   r   r   r,   r   r      s    r   r
   )r   r   c                 C  s4   t | tr0| }tj|  } | dkr0td| d| S )z$Get instance of matplotlib colormap.Nz	Colormap z is not recognized)r   r   mplZ	colormaps
ValueError)r   r-   r   r   r   r0      s    

r0   zColor | Collection[Color])r   r   c                 C  sL   t | dkrtd|  t| r2tt| } | gS ttt | } tt| S )z!Get colors from user input color.r   zInvalid color argument: )r%   r3   _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_colorr7   r   r   r   r4      s
    r4   zCollection[Color]c                 c  s,   | D ]"}t |r|V  qtd| qdS )zS
    Yield colors from string of several letters or from collection of colors.
    zInvalid color N)r4   r3   )r   xr   r   r   r6      s    r6   c                 C  s6   t t| o2t| dks"t| dko2t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)r*   r;   r   r   r   	<genexpr>   s     z#_is_floats_color.<locals>.<genexpr>)r8   r	   r%   allr7   r   r   r   r:      s    r:   )r   r   r   c                 C  s,   | dkrt |S | dkr t|S tddS )z&Get colors from user input color type.r   randomz/color_type must be either 'default' or 'random'N)_get_default_colors_get_random_colorsr3   )r   r   r   r   r   r#      s
    r#   )r   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 r7   r   )r*   cr   r   r   r.      s     z'_get_default_colors.<locals>.<listcomp>zaxes.prop_cycle)Zmatplotlib.pyplotZpyplotZrcParams)r   Zpltr   r   r   r   rB      s    rB   c                 C  s   dd t | D S )z"Get `num_colors` of random colors.c                 S  s   g | ]}t |qS r   )_random_colorr)   r   r   r   r.      s     z&_get_random_colors.<locals>.<listcomp>)ranger   r   r   r   rC      s    rC   zlist[float])columnr   c                 C  s   t | }|d S )z4Get a random color represented as a list of length 3r<   )comZrandom_stateZrandtolist)rG   rsr   r   r   rE      s    
rE   r5   c                 C  s8   t j }z||  W n tk
r.   Y dS X dS dS )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.
    FTN)
matplotlibr   ZColorConverterZto_rgbar3   )r   convr   r   r   r9      s    
r9   )Nr   N)(
__future__r   r&   typingr   r   r   r   r    rK   r2   Zmatplotlib.colorsnumpyr1   Zpandas._typingr   r5   Zpandas.util._exceptionsr   Zpandas.core.dtypes.commonr	   Zpandas.core.commoncorecommonrH   r
   r   r   r   r   r0   r"   r4   r6   r:   r#   rB   rC   rE   r9   r   r   r   r   <module>   s:      34
	
	
