U
    i2e                     @   sT   d Z ddlZddlmZ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units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nppistr)xposn r   Q/tmp/pip-unpacked-wheel-qhyxngld/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.
    kmdeg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UnitDblFormatterr   ZAxisInfo)unitaxisUr   r   r   r   r   axisinfo(   s    zUnitDblConverter.axisinfoc                    s`   t | s fdd| D S d k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>C   s     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&   M   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   r   Zmatplotlib.projections.polarZprojectionsr   __all__r   ZConversionInterfacer   r   r   r   r   <module>   s   
