OwlCyberSecurity - MANAGER
Edit File: grid_helper_curvelinear.cpython-38.pyc
U ��]!: � @ s� d Z ddlmZ ddlZddlmZ ddlmZm Z ddl m Z m Z ddl mZ dd lmZ G d d � d e j�ZG d d � d e j�ZG dd� de �ZdS )z/ An experimental support for curvilinear grid. � )�chainN)�Path)�Affine2D�IdentityTransform� )�AxisArtistHelper�GridHelperBase)� AxisArtist)� GridFinderc sD e Zd ZdZd � fdd� Zdd� Zddd�Zd d � Zd d � Z� Z S )�FixedAxisArtistHelperz( Helper class for a fixed axis. Nc s8 t � j|d� || _|dkr"| j}|| _|| _d| _dS )�} nth_coord = along which coordinate value varies. nth_coord = 0 -> x axis, nth_coord = 1 -> y axis )�locNF)�super�__init__� grid_helper� nth_coord�nth_coord_ticks�side�_limits_inverted)�selfr r r �� __class__� �Q/usr/lib/python3/dist-packages/mpl_toolkits/axisartist/grid_helper_curvelinear.pyr s z FixedAxisArtistHelper.__init__c C sJ | j �|� | jdkr$|�� \}}n |�� \}}||kr@d| _nd| _d S )Nr TF)r � update_limr �get_ylim�get_xlimr )r �axesZxy1Zxy2r r r r $ s z FixedAxisArtistHelper.update_limc C s2 |d krd| j | _ n|dkr&|| _ ntd��d S )Nr )r r zwrong coord number)r � Exception)r Z coord_numberr r r �change_tick_coord1 s z'FixedAxisArtistHelper.change_tick_coordc C s |j S �N�� transData�r r r r r �get_tick_transform9 s z(FixedAxisArtistHelper.get_tick_transformc C s^ | j }| jr"ddddd�| j }n| j}|�| j|�}|jd| j |dd�}t||�tg �fS ) z tick_loc, tick_angle, tick_label�right�left�bottom�top)r& r% r( r' r T)�minor)r r r �get_tick_iteratorr r �iter)r r �gr Zti1Zti2r r r �get_tick_iterators< s ��z(FixedAxisArtistHelper.get_tick_iterators)N)N) �__name__� __module__� __qualname__�__doc__r r r r$ r- � __classcell__r r r r r s r c s^ e Zd Zd� fdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z � Z S )�FloatingAxisArtistHelperNc s0 t � �||� || _|| _d| _d| _d| _dS )r �NNN�d )r r �valuer � _extremes�_get_line_path�_line_num_points)r r r r6 �axis_directionr r r r O s z!FloatingAxisArtistHelper.__init__c C s ||f| _ d S r )r7 )r �e1�e2r r r � set_extremes] s z%FloatingAxisArtistHelper.set_extremesc C s� | j �|� |�� \}}|�� \}}| j j}|�|j||||�}t|�}| j\}} | j dkr�|d k rvt ||d �|d< | d k r�t | |d �|d< n>| j dkr�|d k r�t ||d �|d< | d k r�t | |d �|d< t � } |\} } } }|� | | �\}}}|�| |�\}}}|| d< |||f| d< |||f| d< |�d||�| d < |�d||�| d < | j j}| j dk�r�t�| j| j| j�}t�|d |d | j�}|�||�\}}nH| j dk�r�t�|d |d | j�}t�| j| j| j�}|�||�\}}||f| d < | | _d S ) Nr � � r �extremes�lon_info�lat_infor' � lon_labels� lat_labels�line_xy)r r r r � grid_finder�extreme_finderZinv_transform_xy�listr7 r �max�min�dict� grid_locator1� grid_locator2�tick_formatter1�tick_formatter2�npZlinspacer6 r9 � transform_xy� grid_info)r r �x1�x2�y1�y2rF r@ r; r< rR Zlon_minZlon_maxZlat_minZlat_max�lon_levs�lon_n� lon_factor�lat_levs�lat_n� lat_factor�xx0�yy0ZxxZyyr r r r ` sd � � ��� z#FloatingAxisArtistHelper.update_limc C s t � S r )r r# r r r �get_axislabel_transform� s z0FloatingAxisArtistHelper.get_axislabel_transformc C s� | j d }| jdkrN| j}|d |d d }dt|d |d �d }}nB| jdkr�|d |d d }| j}t|d |d �d d }}| jj}|�|g|g�\}} |j|j� � } | � |d | d g�} d| d kr�dk�r�n n�d| d k�r dk�r�n nz|j� |d | d g�\} } |�|| g|| g�\}}|j� |d |d g�\}}| | ft � || || �t j d fS d S d S ) Nr@ r r> r? g @g g @�@r g �f@r4 )rR r r6 �absr rF rQ r" � transAxes�inverted�transform_pointrP �arctan2�pi)r r r@ r] r^ ZdxxZdyyrF �xx1�yy1Ztrans_passingthrough_point�pZxx1cZyy1cZxx2Zyy2Zxx2cZyy2cr r r �get_axislabel_pos_angle� s& > &z0FloatingAxisArtistHelper.get_axislabel_pos_anglec C s t � S r )r r# r r r r$ � s z+FloatingAxisArtistHelper.get_tick_transformc s� �j j��jd \}}}t�|�}|dk r<|| }d| }n|}d}�jd \}}} t�|�}| dk rx|| } d| } n|} d} d�jkr��j\} } nt�j�\} } | dkr�tj } | dkr�tj} �jdkr�| |k|| k@ }|| }n$�jdk�r | | k| | k@ }| | } � �fdd�}�jdk�r�t� |�j �} || |�\�� | � � }|| | | k | 8 < |||�\�� ||| |�\�� || |�\�� || || �\�� �jd �d d � t �|�D ��n��jdk�rft� | �j �}|| |�\�� || |�\�� || || �\�� | � � }|| | | k | 8 < |||�\�� ||| |�\�� �jd �d d � t �|�D ��� �������� � � � � f dd�}|� t g �fS )z9tick_loc, tick_angle, tick_label, (optionally) tick_labelrB Ng{�G�z�?rA r r c s> �� | |�\}}� j�t�||g��� �}|�� \}}||fS r )rQ r" Z transformrP ZarrayZ transpose)�x�yrS rU Zx2y2rT rV )r rF r r rQ � s zAFloatingAxisArtistHelper.get_tick_iterators.<locals>.transform_xyrD c S s g | ]\}}|r|�qS r r ��.0�l�mr r r � <listcomp>� s z?FloatingAxisArtistHelper.get_tick_iterators.<locals>.<listcomp>rC c S s g | ]\}}|r|�qS r r rl r r r rp s c 3 s t �� � �� �} t �� � �� �}� � dk�� dk@ }|| t jd | |< ��� �}|� j�� }t��| |��D ]�\}}}}} |�||f�} d} d| | d kr�d| kr~n q~d| | d kr�d| kr~n q~t �||g�\} }||g| || fV q~d S )Nr r> g�h㈵��>r ) rP rd re r$ ra rb �ziprc Zrad2deg) ZddZdd2ZmmZ trans_tickZtr2axrj rk �dZd2ZlabZc2ZdeltaZd1) r �labelsr rf �xx1a�xx1b�xx2a�xx2brg �yy1a�yy1b�yy2a�yy2br r �f1 s Hz7FloatingAxisArtistHelper.get_tick_iterators.<locals>.f1)r rF rR rP Zasarrayr7 �sorted�infr Z full_liker6 �copyrq r+ )r r rZ r[ r\ r^ ZdyrW rX rY r] ZdxZe0r; �maskrQ Zxx00r| r )r rF rs r rf rt ru rv rw rg rx ry rz r{ r r- � sj $z+FloatingAxisArtistHelper.get_tick_iteratorsc C s |j S r r! r# r r r �get_line_transform s z+FloatingAxisArtistHelper.get_line_transformc C sF | � |� | jd \}}| jd kr4tt�||g��S | �|||�S d S )NrE )r rR r8 r rP Z column_stack)r r rj rk r r r �get_line s z!FloatingAxisArtistHelper.get_line)N) r. r/ r0 r r= r r_ ri r$ r- r� r� r2 r r r r r3 M s 7 ar3 c s` e Zd Zd� fdd� Zddd�Zdd� Zddd �Zdd d �Zd d� Zddd�Z ddd�Z � Z S ) �GridHelperCurveLinearNc s4 t � �� d| _d| _|| _t||||||�| _dS )a� aux_trans : a transform from the source (curved) coordinate to target (rectilinear) coordinate. An instance of MPL's Transform (inverse transform should be defined) or a tuple of two callable objects which defines the transform and its inverse. The callables need take two arguments of array of source coordinates and should return two target coordinates. e.g., ``x2, y2 = trans(x1, y1)`` N)r r rR � _old_valuesZ _aux_transr rF )r � aux_transrG rL rM rN rO r r r r ) s �z GridHelperCurveLinear.__init__c K s. |d k r| j �|� | j jf |� | �� d S r )rF Zupdate_transform�updateZ invalidate)r r� �kwr r r �update_grid_finderG s z(GridHelperCurveLinear.update_grid_finderc C sF | � � r | j||||fkr dS | �||||� ||||f| _d| _dS )z!bbox in 0-based image coordinatesNF)Zvalidr� � _update_gridZ _force_update)r rS rT rU rV r r r �_updateO s z GridHelperCurveLinear._updatec C s: |d kr| j }|d kr|}t| ||d�}t|||d�}|S )N)r )r: )r r r )r r r r: �offsetr �_helper�axisliner r r �new_fixed_axis\ s �z$GridHelperCurveLinear.new_fixed_axisr' c C sF |d kr| j }t| |||�}t||�}|j�d� |j�|j j� |S )NT)r r3 r �lineZ set_clip_onZ set_clip_boxZbbox)r r r6 r r: r� r� r r r �new_floating_axisp s � z'GridHelperCurveLinear.new_floating_axisc C s | j �||||�| _d S r )rF Z get_grid_inforR )r rS rU rT rV r r r r� � s z"GridHelperCurveLinear._update_grid�major�bothc C sT g }|dkr*| j d d D ]}|�|� q|dkrP| j d d D ]}|�|� q@|S )N)r� rj �lon�lines)r� rk �lat)rR �extend)r ZwhichZaxisZ grid_linesZglr r r � get_gridlines� s z#GridHelperCurveLinear.get_gridlinesFc c s� t ddddd�| }ddg| }|slt| j| d | | j| d | �D ] \\}}}|} || ||fV qJnFt| j| d | | j| d | �D ] \\}}}|} || |dfV q�d S ) N�Z r )r& r% r' r( r� r� Z tick_locsZ tick_labels� )rK rq rR ) r r Z axis_sider) Z angle_tangentZ lon_or_latZxy�arn Z angle_normalr r r r* � s ��z'GridHelperCurveLinear.get_tick_iterator)NNNNN)N)NNNN)Nr' )r� r� )F) r. r/ r0 r r� r� r� r� r� r� r* r2 r r r r r� ' s& � � � r� )r1 � itertoolsr ZnumpyrP Zmatplotlib.pathr Zmatplotlib.transformsr r Z axislinesr r Z axis_artistr rF r ZFixedr ZFloatingr3 r� r r r r �<module> s > [