o
    -h                     @   s\   d dl mZ d dlmZ d dlmZmZ G dd dZG dd dej	Z	G dd	 d	eZ
dS )
    N)Artist)XAxisYAxisc                   @   s$   e Zd Zdd Zdd Zdd ZdS )SimpleChainedObjectsc                 C   s
   || _ d S N_objects)selfobjects r   T/var/www/vscode/kcb/lib/python3.10/site-packages/mpl_toolkits/axes_grid1/mpl_axes.py__init__   s   
zSimpleChainedObjects.__init__c                    s   t  fdd| jD }|S )Nc                       g | ]}t | qS r   getattr).0akr   r   
<listcomp>   s    z4SimpleChainedObjects.__getattr__.<locals>.<listcomp>)r   r   )r	   r   _ar   r   r   __getattr__
   s   z SimpleChainedObjects.__getattr__c                 O   s   | j D ]	}||i | qd S r   r   )r	   argskwargsmr   r   r   __call__   s   
zSimpleChainedObjects.__call__N)__name__
__module____qualname__r   r   r   r   r   r   r   r      s    r   c                       s8   e Zd ZG dd deZedd Z fddZ  ZS )Axesc                       s0   e Zd Z fddZ fddZdd Z  ZS )zAxes.AxisDictc                    s   || _ t   d S r   )axessuperr   )r	   r    	__class__r   r   r      s   zAxes.AxisDict.__init__c                    sr   t |trt fdd|D }|S t |tr3|jd u r/|jd u r/|jd u r/tt  S t	dt
 |S )Nc                    s   g | ]}t tj|qS r   )r!   r   AxisDict__getitem__)r   k1)r#   r	   r   r   r      s    z-Axes.AxisDict.__getitem__.<locals>.<listcomp>zUnsupported slice)
isinstancetupler   slicestartstopsteplistvalues
ValueErrordictr%   )r	   r   rr"   r	   r   r%      s   

zAxes.AxisDict.__getitem__c                 O   s   t jj| jg|R i |S r   )maxesr   axisr    )r	   vr   r   r   r   r   (   s   zAxes.AxisDict.__call__)r   r   r   r   r%   r   __classcell__r   r   r"   r   r$      s    r$   c                 C   s   | j S r   )
_axislinesr2   r   r   r   r4   +   s   z	Axes.axisc              	      sn   t    | | | _| jjt| jd| jd t| jd| jd t| jd| jd t| jd| jd d d S )N   bottom   topleftright)r9   r;   r<   r=   )	r!   clearr$   r7   updateSimpleAxisArtistxaxisspinesyaxisr2   r"   r   r   r>   /   s   

z
Axes.clear)	r   r   r   r0   r$   propertyr4   r>   r6   r   r   r"   r   r      s
    
r   c                       s^   e Zd Z fddZedd Zedd Zedd Z fd	d
Zdd Z	dddZ
  ZS )r@   c                    sj   || _ || _|| _t|trddg|d  | _nt|tr'ddg|d  | _ntd| t 	  d S )Nr9   r;   r8   r<   r=   z1axis must be instance of XAxis or YAxis, but got )
_axis_axisnumliner'   r   _axis_directionr   r/   r!   r   )r	   r4   axisnumspiner"   r   r   r   <   s   

zSimpleAxisArtist.__init__c                    &   d| j   t fdd| j D S )Nz
tick%dlinec                    r   r   r   r   ticktickliner   r   r   M       z0SimpleAxisArtist.major_ticks.<locals>.<listcomp>rF   r   rE   get_major_ticksr2   r   rN   r   major_ticksJ      
zSimpleAxisArtist.major_ticksc                    rK   )Nzlabel%dc                    r   r   r   rL   labelr   r   r   S   rP   z5SimpleAxisArtist.major_ticklabels.<locals>.<listcomp>rQ   r2   r   rU   r   major_ticklabelsP   rT   z!SimpleAxisArtist.major_ticklabelsc                 C   s   | j jS r   )rE   rV   r2   r   r   r   rV   V   s   zSimpleAxisArtist.labelc                    s4   | j |d | j| | jd t | d S )N)allT)togglerG   set_visiblerE   r!   )r	   br"   r   r   rZ   Z   s   zSimpleAxisArtist.set_visiblec                 C   s   | j | d S r   )rE   set_label_text)r	   txtr   r   r   	set_label`   s   zSimpleAxisArtist.set_labelNc           
      C   s  |rd\}}}n|d urd\}}}nd\}}}|d ur|}|d ur#|}|d ur)|}|d ur?d| j  d|i}| jjd	i | |d urUd| j  d|i}| jjd	i | |d ur| j }	|	| jkrn|sn| jjd d S |r| jjd | j| j d S d S d S )
N)TTT)FFF)NNNrM   OnrV   FTr   )rF   rE   set_tick_paramsget_label_positionrH   rV   rZ   set_label_position)
r	   rX   ticks
ticklabelsrV   _ticks_ticklabels_label	tickparamposr   r   r   rY   c   s4   

zSimpleAxisArtist.toggle)NNNN)r   r   r   r   rD   rS   rW   rV   rZ   r^   rY   r6   r   r   r"   r   r@   ;   s    


r@   )matplotlib.axesr    r3   matplotlib.artistr   matplotlib.axisr   r   r   r   r@   r   r   r   r   <module>   s    (