a
    f1b&                     @   s\   d Z ddlZddlmZ ddlmZ ddlm  m	Z	 dgZ
dddZG dd dejZdS )	z:UnitDblConverter module containing class UnitDblConverter.    N)cbookUnitDblConverterc                 C   sj   t | tj d d }|dkr&t| S |dkr2dS |dkr>dS |d dkrZd|d  d	S d| d
S dS )zRadian function formatter.g       @g      ?r      z$\pi/2$   z$\pi$$z\pi$z\pi/2$N)intnpZpistr)xposn r   Llib/python3.9/site-packages/matplotlib/testing/jpl_units/UnitDblConverter.pyrad_fn   s    r   c                   @   s@   e Zd ZdZddddZedd Zedd	 Zed
d ZdS )r   zS
    Provides Matplotlib conversion functionality for the Monte UnitDbl class.
    ZkmdegZsec)Zdistanceangletimec                 C   sn   dd l m  m} | r.t| tr$| n|  }nd }|dkrTt|jtjrTtj	 }n|j
dd}tj||dS )Nr   r   F)Z	useOffset)majfmtlabel)Zmatplotlib.testing.jpl_unitsZtestingZ	jpl_units
isinstancer	   r   axespolar	PolarAxesZThetaFormatterZUnitDblFormatterunitsZAxisInfo)unitaxisUr   r   r   r   r   axisinfo)   s    zUnitDblConverter.axisinfoc                    s`   t | s fdd| D S d u r2t|  t jtjrV|  dkrV| 	dS | 	S )Nc                    s   g | ]}t | qS r   )r   convert).0r
   r   r   r   r   
<listcomp>D       z,UnitDblConverter.convert.<locals>.<listcomp>r   Zrad)
r   is_scalar_or_stringr   default_unitsr   r   r   r   typer   )valuer   r   r   r    r   r   @   s    

zUnitDblConverter.convertc                 C   s,   t | rtj|   S t| d |S d S )Nr   )r   r#   r   defaultsr%   r$   )r&   r   r   r   r   r$   N   s    
zUnitDblConverter.default_unitsN)	__name__
__module____qualname____doc__r'   staticmethodr   r   r$   r   r   r   r   r      s   

)N)r+   Znumpyr   Z
matplotlibr   Zmatplotlib.unitsr   Zmatplotlib.projections.polarZprojectionsr   __all__r   ZConversionInterfacer   r   r   r   r   <module>   s   
