OwlCyberSecurity - MANAGER
Edit File: geo.cpython-38.pyc
U ��](F � @ s� d dl Zd dlmZmZ d dlmZ d dlmZ d dl m Z d dl m Z d dlmZ d dlmZmZmZmZ d dlmZmZmZ G dd � d e�ZG d d � d e�ZG d d � d e�Z G dd� de�Z G dd� de�Z G dd� de�Z dS )� N)�cbook�rcParams)�Axes��Circle)�Path)� Formatter� NullLocator� FixedLocator� NullFormatter)�Affine2D�BboxTransformTo� Transformc @ s� e Zd ZdZG dd� de�ZdZdd� Zdd� Zd d � Z d d � Z d7dd�Z dd� Z dd� Z d8dd�Zdd� Zdd� Zdd� Zd d � Zd d � ZeZd d!� ZeZd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Z d.d/� Z d0d1� Z d2d3� Z d4d5� Z d6S )9�GeoAxesz2An abstract base class for geographic projections.c @ s$ e Zd ZdZddd�Zd dd�ZdS ) zGeoAxes.ThetaFormatterz� Used to format the theta tick labels. Converts the native unit of radians into degrees and adds a degree symbol. � �?c C s || _ d S �N)� _round_to)�selfZround_to� r �</usr/lib/python3/dist-packages/matplotlib/projections/geo.py�__init__ s z GeoAxes.ThetaFormatter.__init__Nc C sH |t j d }t �|| j �| j }td r<td s<d| S d| S d S )Ng �f@z text.usetexztext.latex.unicodez $%0.0f^\circ$u %0.0f°)�np�pi�roundr r )r �x�pos�degreesr r r �__call__ s z GeoAxes.ThetaFormatter.__call__)r )N)�__name__� __module__� __qualname__�__doc__r r r r r r �ThetaFormatter s r" �K c C s$ t �| �| _t �| �| _| �� d S r )�maxisZXAxis�xaxisZYAxis�yaxisZ_update_transScale�r r r r � _init_axis" s zGeoAxes._init_axisc C s� t �| � | �d� | �d� | �d� | j�t� � | j�t� � | j� d� | j� d� | jj dd� | � t d � t � | tj tj� t �| tj d tjd � d S ) N� � r# ZnoneT)Zlabel1Onz axes.grid� @)r �cla�set_longitude_grid�set_latitude_grid�set_longitude_grid_endsr% Zset_minor_locatorr r& Zset_ticks_positionZset_tick_params�gridr �set_xlimr r �set_ylimr' r r r r, * s z GeoAxes.clac C s0 | � | j�| _| �� | _t| j�| _| j| j | j | _t � � d| j d �� d| j �| _ | j | j | _t � � dd�| j t � � dd� | _t � � dd�| j t � � dd� | _t � � tjd d�� tj d�}t � � dd�}|| j | _|| j || j | j }|t � � dd� | _|t � � dd� | _d S ) N� � r � ���g�������?i����� )�_get_core_transform� RESOLUTIONZtransProjection�_get_affine_transformZ transAffiner ZbboxZ transAxesZ transDatar �scale�_longitude_cap� translate�_xaxis_pretransform�_xaxis_transform�_xaxis_text1_transform�_xaxis_text2_transformr r �_yaxis_transform�_yaxis_text1_transform�_yaxis_text2_transform)r Z yaxis_stretchZ yaxis_spaceZyaxis_text_baser r r �_set_lim_and_transforms= sr ��� � ���� � �� � �� ������� �� ��z GeoAxes._set_lim_and_transformsc C sT | � d�}|�tjdf�\}}|�dtjd f�\}}t� �d| d| ��dd�S )Nr3 r r4 � �?)r8 Ztransform_pointr r r r; r= )r � transformZxscale�_Zyscaler r r r: p s � �z GeoAxes._get_affine_transformr0 c C s t jdddg|d� | jS �NZtick1Ztick2r0 )�which)r �_check_in_listr? �r rJ r r r �get_xaxis_transformx s zGeoAxes.get_xaxis_transformc C s | j ddfS )NZbottom�center)r@ �r Zpadr r r �get_xaxis_text1_transform| s z!GeoAxes.get_xaxis_text1_transformc C s | j ddfS )N�toprN )rA rO r r r �get_xaxis_text2_transform s z!GeoAxes.get_xaxis_text2_transformc C s t jdddg|d� | jS rI )r rK rB rL r r r �get_yaxis_transform� s zGeoAxes.get_yaxis_transformc C s | j ddfS )NrN �right)rC rO r r r �get_yaxis_text1_transform� s z!GeoAxes.get_yaxis_text1_transformc C s | j ddfS )NrN �left)rD rO r r r �get_yaxis_text2_transform� s z!GeoAxes.get_yaxis_text2_transformc C s t dd�S )N�rF rF rF r r' r r r �_gen_axes_patch� s zGeoAxes._gen_axes_patchc C s dt j�| dd�iS )NZgeorX rF )�mspinesZSpineZcircular_spiner' r r r �_gen_axes_spines� s zGeoAxes._gen_axes_spinesc O s |d dkrt �d S )Nr Zlinear)�NotImplementedError�r �args�kwargsr r r � set_yscale� s zGeoAxes.set_yscalec O s t d��d S )NznIt is not possible to change axes limits for geographic projections. Please consider using Basemap or Cartopy.)� TypeErrorr] r r r r1 � s zGeoAxes.set_xlimc C sN t �||g�\}}|dkr d}nd}|dkr2d}nd}dt|�|t|�|f S )z0return a format string formatting the coordinate� �N�S�E�Wu %f°%s, %f°%s)r Zrad2deg�abs)r �lon�lat�nsZewr r r � format_coord� s �zGeoAxes.format_coordc C s> t �d| d|�}| j�tt �|��� | j�| �|�� dS )zH Set the number of degrees between each longitude grid. iL���� N)r �aranger% �set_major_locatorr �deg2rad�set_major_formatterr" �r r r0 r r r r- � s zGeoAxes.set_longitude_gridc C s> t �d| d|�}| j�tt �|��� | j�| �|�� dS )zG Set the number of degrees between each latitude grid. i�����Z N)r rm r& rn r ro rp r" rq r r r r. � s zGeoAxes.set_latitude_gridc C s4 t �|�| _| j�� �d| jd ��d| j � dS )zS Set the latitude(s) at which to stop drawing the longitude grids. r r+ rb N)r ro r< r> �clearr; r= )r r r r r r/ � s � �z GeoAxes.set_longitude_grid_endsc C s dS )z= Return the aspect ratio of the data itself. r r r' r r r �get_data_ratio� s zGeoAxes.get_data_ratioc C s dS )z� Return *True* if this axes supports the zoom box button functionality. This axes object does not support interactive zoom box. Fr r' r r r �can_zoom� s zGeoAxes.can_zoomc C s dS )z� Return *True* if this axes supports the pan/zoom button functionality. This axes object does not support interactive pan/zoom. Fr r' r r r �can_pan� s zGeoAxes.can_panc C s d S r r )r r �y�buttonr r r � start_pan� s zGeoAxes.start_panc C s d S r r r' r r r �end_pan� s zGeoAxes.end_panc C s d S r r )r rx �keyr rw r r r �drag_pan� s zGeoAxes.drag_panN)r0 )r0 )!r r r r! r r" r9 r( r, rE r: rM rP rR rS rU rW rY r[ r` Z set_xscaler1 r2 rk r- r. r/ rt ru rv ry rz r| r r r r r s: 3 r c @ s0 e Zd ZdZdZdZdd� Zdd� Zdd� Zd S ) � _GeoTransformr4 Fc C s t �| � || _dS )z� Create a new geographical transform. Resolution is the number of steps to interpolate between each input line segment to approximate its path in curved space. N)r r � _resolution�r � resolutionr r r r � s z_GeoTransform.__init__c C s d� t| �j| j�S )Nz{}({}))�format�typer r~ r' r r r �__str__� s z_GeoTransform.__str__c C s |� | j�}t| �|j�|j�S r )Z interpolatedr~ r rG ZverticesZcodes)r �pathZipathr r r �transform_path_non_affine� s z'_GeoTransform.transform_path_non_affineN) r r r Z input_dimsZ output_dimsZ is_separabler r� r� r r r r r} � s r} c @ s@ e Zd ZdZG dd� de�ZG dd� de�Zdd� Zdd � Zd S ) � AitoffAxesZaitoffc @ s e Zd ZdZdd� Zdd� ZdS )zAitoffAxes.AitoffTransformzThe base Aitoff transform.c C s� |d d �df }|d d �df }|d }t �|�}t �|t �|� �}d||dk< t �|�| }t �|t�}|t �|� | |d d �df< t �|�| |d d �df< |S )Nr r3 r+ g#B� ���;rb )r �cosZarccos�sinZ empty_like�float) r �ll� longitude�latitude� half_long� cos_latitude�alphaZ sinc_alpha�xyr r r �transform_non_affine s z/AitoffAxes.AitoffTransform.transform_non_affinec C s t �| j�S r )r� �InvertedAitoffTransformr~ r' r r r �inverted s z#AitoffAxes.AitoffTransform.invertedN�r r r r! r� r� r r r r �AitoffTransform s r� c @ s e Zd Zdd� Zdd� ZdS )z"AitoffAxes.InvertedAitoffTransformc C s |S r r )r r� r r r r� $ s z7AitoffAxes.InvertedAitoffTransform.transform_non_affinec C s t �| j�S r )r� r� r~ r' r r r r� ) s z+AitoffAxes.InvertedAitoffTransform.invertedN�r r r r� r� r r r r r� " s r� c O s: t jd | _tj| f|�|� | jdddd� | �� d S �Nr+ rF �box�C�Z adjustableZanchor�r r r< r r � set_aspectr, r] r r r r - s zAitoffAxes.__init__c C s | � |�S r )r� r r r r r8 3 s z AitoffAxes._get_core_transformN) r r r �namer} r� r� r r8 r r r r r� s r� c @ s@ e Zd ZdZG dd� de�ZG dd� de�Zdd� Zdd � Zd S ) � HammerAxesZhammerc @ s e Zd ZdZdd� Zdd� ZdS )zHammerAxes.HammerTransformzThe base Hammer transform.c C s� |d d �dd�f }|d d �dd�f }|d }t �|�}t �d�}t �d|t �|� �}d| |t �|� | }|t �|� | } t �|| fd�S )Nr r3 r4 r+ r )r r� �sqrtr� � concatenate) r r� r� r� r� r� Zsqrt2r� r rw r r r r� = s z/HammerAxes.HammerTransform.transform_non_affinec C s t �| j�S r )r� �InvertedHammerTransformr~ r' r r r r� L s z#HammerAxes.HammerTransform.invertedNr� r r r r �HammerTransform: s r� c @ s e Zd Zdd� Zdd� ZdS )z"HammerAxes.InvertedHammerTransformc C sn |j \}}t�d|d d |d d �}dt�|| dd|d d � }t�|| �}t�||g�S )Nr3 r5 r4 )�Tr r� �arctan�arcsinZ column_stack)r r� r rw �zr� r� r r r r� R s "&z7HammerAxes.InvertedHammerTransform.transform_non_affinec C s t �| j�S r )r� r� r~ r' r r r r� Z s z+HammerAxes.InvertedHammerTransform.invertedNr� r r r r r� P s r� c O s: t jd | _tj| f|�|� | jdddd� | �� d S r� r� r] r r r r ^ s zHammerAxes.__init__c C s | � |�S r )r� r r r r r8 d s z HammerAxes._get_core_transformN) r r r r� r} r� r� r r8 r r r r r� 7 s r� c @ s@ e Zd ZdZG dd� de�ZG dd� de�Zdd� Zdd � Zd S ) � MollweideAxesZ mollweidec @ s e Zd ZdZdd� Zdd� ZdS )z MollweideAxes.MollweideTransformz The base Mollweide transform.c sx � fdd�}|d d �df }|d d �df }t jd t �|� }|dk }| }t j|jtd�}|�� r�t jt �|| � � d|| } || �\} } t �| �r�| | | | 7 < || �\} } q�| d ||< |�� �r|| } d d t j | d d }t jd | t �|| � ||< t j|jtd�} dt � d� t j | t � |� | d d �df< t � d�t �|� | d d �df< | S ) Nc s4 | t �| � � dt �| � }|t �|�dkfS )Nr3 g����MbP?)r r� r� rg )�theta�delta�Zpi_sin_lr r �dp s �z@MollweideAxes.MollweideTransform.transform_non_affine.<locals>.dr r3 r4 g��ʡE�?)Zdtyper+ rF � gUUUUUU�?) r r rg �empty�shaper� �anyr� Zsignr� r� )r r� r� r� r� �clatZihighZilowZauxr� r� Z large_delta�er� r r� r r� n s. . z5MollweideAxes.MollweideTransform.transform_non_affinec C s t �| j�S r )r� �InvertedMollweideTransformr~ r' r r r r� � s z)MollweideAxes.MollweideTransform.invertedNr� r r r r �MollweideTransformk s #r� c @ s e Zd Zdd� Zdd� ZdS )z(MollweideAxes.InvertedMollweideTransformc C s� |d d �dd�f }|d d �dd�f }t �|t �d� �}t jdt �d� | t �|� }t �d| t �d| � t j �}t �||fd�S )Nr r3 r4 )r r� r� r r� r� r� )r r� r rw r� rh ri r r r r� � s ""z=MollweideAxes.InvertedMollweideTransform.transform_non_affinec C s t �| j�S r )r� r� r~ r' r r r r� � s z1MollweideAxes.InvertedMollweideTransform.invertedNr� r r r r r� � s r� c O s: t jd | _tj| f|�|� | jdddd� | �� d S r� r� r] r r r r � s zMollweideAxes.__init__c C s | � |�S r )r� r r r r r8 � s z!MollweideAxes._get_core_transformN) r r r r� r} r� r� r r8 r r r r r� h s *r� c @ sX e Zd ZdZG dd� de�ZG dd� de�Zddd�dd �Zd d � Zd d � Z dd� Z dS )� LambertAxesZlambertc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) z LambertAxes.LambertTransformzThe base Lambert transform.c C s t �| |� || _|| _dS )z� Create a new Lambert transform. Resolution is the number of steps to interpolate between each input line segment to approximate its path in curved Lambert space. N�r} r �_center_longitude�_center_latitude�r �center_longitude�center_latituder� r r r r � s z%LambertAxes.LambertTransform.__init__c C s� |d d �dd�f }|d d �dd�f }| j }| j}t�|�}t�|�}|| }t�|�} t�dt�|�| t�|�| | d�} t�d| �} | | t�|� } | t�|�| t�|�| | } t�| | fd�S )Nr r3 r4 gV瞯�<)r� r� r r� r� �maximumr� r� )r r� r� r� �clongr� Zcos_latZsin_latZ diff_longZ cos_diff_longZinner_k�kr rw r r r r� � s "�$z1LambertAxes.LambertTransform.transform_non_affinec C s t �| j| j| j�S r )r� �InvertedLambertTransformr� r� r~ r' r r r r� � s �z%LambertAxes.LambertTransform.invertedN)r r r r! r r� r� r r r r �LambertTransform� s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z$LambertAxes.InvertedLambertTransformc C s t �| |� || _|| _d S r r� r� r r r r � s z-LambertAxes.InvertedLambertTransform.__init__c C s� |d d �dd�f }|d d �dd�f }| j }| j}t�t�||�d�}dt�d| � }t�|�}t�|�} t�| t�|� || t�|� | �} |t�|| |t�|� | |t�|� | � } t� | | fd�S )Nr r3 r4 g��&� .>rF ) r� r� r r� Zhypotr� r� r� r� r� ) r r� r rw r� r� �p�cZsin_cZcos_cri rh r r r r� � s �*�z9LambertAxes.InvertedLambertTransform.transform_non_affinec C s t �| j| j| j�S r )r� r� r� r� r~ r' r r r r� � s �z-LambertAxes.InvertedLambertTransform.invertedN)r r r r r� r� r r r r r� � s r� r )r� r� c O sF t jd | _|| _|| _tj| f|�|� | jdddd� | �� d S )Nr4 Zequalr� r� r� ) r r r< r� r� r r r� r, )r r� r� r^ r_ r r r r � s zLambertAxes.__init__c C s t �| � | j�t� � d S r )r r, r& rp r r' r r r r, s zLambertAxes.clac C s | � | j| j|�S r )r� r� r� r r r r r8 s �z LambertAxes._get_core_transformc C s t � �d��dd�S )Ng �?rF )r r; r= r' r r r r: s � �z!LambertAxes._get_affine_transformN) r r r r� r} r� r� r r, r8 r: r r r r r� � s ( r� ) Znumpyr Z matplotlibr r Zmatplotlib.axesr Zmatplotlib.axisZaxisr$ Zmatplotlib.patchesr Zmatplotlib.pathr Zmatplotlib.spinesZspinesrZ Zmatplotlib.tickerr r r r Zmatplotlib.transformsr r r r r} r� r� r� r� r r r r �<module> s ^31J