o
    :a&                     @   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   sf   t | tj d d }|dkrt| S |dkrdS |dkrdS |d dkr-d|d  d	S d| d
S )zRadian function formatter.g       @g      ?r      z$\pi/2$   z$\pi$$z\pi$z\pi/2$)intnpZpistr)xposn r   O/usr/lib/python3/dist-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r*t|jtjr*t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 rt|  t jtjr+|  dkr+| 	dS | 	S )Nc                    s   g | ]	}t | qS r   )r   convert).0r
   r   r   r   r   
<listcomp>D   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 )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    
