o
    ˆ‚¡e”  ã                   @   sÊ   d Z ddlmZ ddlZddlZddlZddlZddl	m
Z
mZmZ e e¡ZG dd„ dejƒZG dd„ dejƒZG d	d
„ d
ejƒZG dd„ dƒZeƒ eje< eƒ ejej< eƒ eje< eƒ ejej< dS )a·  
Plotting of string "category" data: ``plot(['d', 'f', 'a'], [1, 2, 3])`` will
plot three points with x-axis values of 'd', 'f', 'a'.

See :doc:`/gallery/lines_bars_and_markers/categorical_variables` for an
example.

The module uses Matplotlib's `matplotlib.units` mechanism to convert from
strings to integers and provides a tick locator, a tick formatter, and the
`.UnitData` class that creates and stores the string-to-integer mapping.
é    )ÚOrderedDictN)Ú_apiÚtickerÚunitsc                   @   s<   e Zd Zedd„ ƒZedd„ ƒZedd„ ƒZedd„ ƒZd	S )
ÚStrCategoryConverterc                 C   sP   |du rt dƒ‚t |¡ t tj| td¡}| |¡ tj|j	j
tgd|ƒS )a  
        Convert strings in *value* to floats using mapping information stored
        in the *unit* object.

        Parameters
        ----------
        value : str or iterable
            Value or list of values to be converted.
        unit : `.UnitData`
            An object mapping strings to integers.
        axis : `~matplotlib.axis.Axis`
            The axis on which the converted value is plotted.

            .. note:: *axis* is unused.

        Returns
        -------
        float or `~numpy.ndarray` of float
        NzMissing category information for StrCategoryConverter; this might be caused by unintendedly mixing categorical and numeric data©Údtype)Úotypes)Ú
ValueErrorr   Ú_validate_unitÚnpÚ
atleast_1dÚarrayÚobjectÚupdateÚ	vectorizeÚ_mappingÚ__getitem__Úfloat)ÚvalueÚunitÚaxisÚvalues© r   úB/var/www/myenv/lib/python3.10/site-packages/matplotlib/category.pyÚconvert   s   ÿ

zStrCategoryConverter.convertc                 C   s,   t  | ¡ t| jƒ}t| jƒ}tj||dS )a´  
        Set the default axis ticks and labels.

        Parameters
        ----------
        unit : `.UnitData`
            object string unit information for value
        axis : `~matplotlib.axis.Axis`
            axis for which information is being set

            .. note:: *axis* is not used

        Returns
        -------
        `~matplotlib.units.AxisInfo`
            Information to support default tick labeling

        )ÚmajlocÚmajfmt)r   r   ÚStrCategoryLocatorr   ÚStrCategoryFormatterr   ÚAxisInfo)r   r   r   r   r   r   r   Úaxisinfo<   s   


zStrCategoryConverter.axisinfoc                 C   s0   |j du r| t| ƒ¡ |j S |j  | ¡ |j S )aO  
        Set and update the `~matplotlib.axis.Axis` units.

        Parameters
        ----------
        data : str or iterable of str
        axis : `~matplotlib.axis.Axis`
            axis on which the data is plotted

        Returns
        -------
        `.UnitData`
            object storing string to integer mapping
        N)r   Ú	set_unitsÚUnitDatar   )Údatar   r   r   r   Údefault_unitsW   s
   
ÿz"StrCategoryConverter.default_unitsc                 C   s   t | dƒstd| › dƒ‚d S )Nr   zProvided unit "zU" is not valid for a categorical converter, as it does not have a _mapping attribute.)Úhasattrr
   )r   r   r   r   r   n   s
   

ÿÿz#StrCategoryConverter._validate_unitN)Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r!   r%   r   r   r   r   r   r      s    
 

r   c                   @   s(   e Zd ZdZdd„ Zdd„ Zdd„ ZdS )	r   z1Tick at every integer mapping of the string data.c                 C   ó
   || _ dS ©zŠ
        Parameters
        ----------
        units_mapping : dict
            Mapping of category names (str) to indices (int).
        N©Ú_units©ÚselfÚunits_mappingr   r   r   Ú__init__x   ó   
zStrCategoryLocator.__init__c                 C   s   t | j ¡ ƒS ©N)Úlistr.   r   ©r0   r   r   r   Ú__call__   s   zStrCategoryLocator.__call__c                 C   s   | ƒ S r4   r   )r0   ÚvminÚvmaxr   r   r   Útick_values…   s   zStrCategoryLocator.tick_valuesN)r'   r(   r)   Ú__doc__r2   r7   r:   r   r   r   r   r   v   s
    	r   c                   @   s6   e Zd ZdZdd„ Zddd„Zdd„ Zed	d
„ ƒZdS )r   z0String representation of the data at every tick.c                 C   r+   r,   r-   r/   r   r   r   r2   Œ   r3   zStrCategoryFormatter.__init__Nc                 C   s   |   |g¡d S )Nr   )Úformat_ticks)r0   ÚxÚposr   r   r   r7   •   s   zStrCategoryFormatter.__call__c                    s*   ‡fdd„ˆj  ¡ D ƒ‰ ‡ fdd„|D ƒS )Nc                    s   i | ]
\}}|ˆ   |¡“qS r   )Ú_text)Ú.0ÚkÚvr6   r   r   Ú
<dictcomp>›   ó    z5StrCategoryFormatter.format_ticks.<locals>.<dictcomp>c                    s   g | ]
}ˆ   t|ƒd ¡‘qS )Ú )ÚgetÚround)r@   Úval)Ú	r_mappingr   r   Ú
<listcomp>œ   rD   z5StrCategoryFormatter.format_ticks.<locals>.<listcomp>)r.   Úitems)r0   r   r   )rI   r0   r   r<   ™   s   z!StrCategoryFormatter.format_ticksc                 C   s0   t | tƒr| jdd} | S t | tƒst| ƒ} | S )z0Convert text values into utf-8 or ascii strings.zutf-8)Úencoding)Ú
isinstanceÚbytesÚdecodeÚstr©r   r   r   r   r?   ž   s   

þzStrCategoryFormatter._textr4   )	r'   r(   r)   r;   r2   r7   r<   r*   r?   r   r   r   r   r   Š   s    
	r   c                   @   s*   e Zd Zddd„Zedd„ ƒZdd„ ZdS )	r#   Nc                 C   s,   t ƒ | _t ¡ | _|dur|  |¡ dS dS )z¸
        Create mapping between unique categorical values and integer ids.

        Parameters
        ----------
        data : iterable
            sequence of string values
        N)r   r   Ú	itertoolsÚcountÚ_counterr   )r0   r$   r   r   r   r2   ©   s
   	
ÿzUnitData.__init__c                 C   sN   zt | ƒ W dS  ty&   z
tj | ¡ W Y dS  ttfy%   Y Y dS w w )zY
        Helper method to check whether a string can be parsed as float or date.
        FT)r   r
   ÚdateutilÚparserÚparseÚ	TypeError)rH   r   r   r   Ú_str_is_convertible·   s   
úýþýzUnitData._str_is_convertiblec                 C   s‚   t  t j|td¡}d}t |¡D ]}tjtt	f|d |r#|  
|¡}|| jvr0t| jƒ| j|< q|jr=|r?t d¡ dS dS dS )zõ
        Map new values to integer identifiers.

        Parameters
        ----------
        data : iterable of str or bytes

        Raises
        ------
        TypeError
            If elements in *data* are neither str nor bytes.
        r   TrQ   z½Using categorical units to plot a list of strings that are all parsable as floats or dates. If these strings should be plotted as numbers, cast to the appropriate data type before plotting.N)r   r   r   r   r   Úfromkeysr   Úcheck_isinstancerP   rN   rY   r   ÚnextrT   ÚsizeÚ_logÚinfo)r0   r$   ÚconvertiblerH   r   r   r   r   Æ   s   

€
ÿzUnitData.updater4   )r'   r(   r)   r2   r*   rY   r   r   r   r   r   r#   ¨   s
    

r#   )r;   Úcollectionsr   Údateutil.parserrU   rR   ÚloggingÚnumpyr   Ú
matplotlibr   r   r   Ú	getLoggerr'   r^   ÚConversionInterfacer   ÚLocatorr   Ú	Formatterr   r#   ÚregistryrP   Ústr_rN   Úbytes_r   r   r   r   Ú<module>   s     
\>