OwlCyberSecurity - MANAGER
Edit File: triangulation.cpython-38.pyc
U ��]� � @ s d dl ZG dd� de�ZdS )� Nc @ sf e Zd ZdZddd�Zdd� Zedd� �Zd d � Zd d � Z e d d� �Z dd� Z edd� �Z dd� ZdS )� Triangulationa� An unstructured triangular grid consisting of npoints points and ntri triangles. The triangles can either be specified by the user or automatically generated using a Delaunay triangulation. Parameters ---------- x, y : array-like of shape (npoints) Coordinates of grid points. triangles : integer array_like of shape (ntri, 3), optional For each triangle, the indices of the three points that make up the triangle, ordered in an anticlockwise manner. If not specified, the Delaunay triangulation is calculated. mask : boolean array-like of shape (ntri), optional Which triangles are masked out. Attributes ---------- edges : int array of shape (nedges, 2) See `~.Triangulation.edges` neighbors : int array of shape (ntri, 3) See `~.Triangulation.neighbors` mask : bool array of shape (ntri, 3) Masked out triangles. is_delaunay : bool Whether the Triangulation is a calculated Delaunay triangulation (where `triangles` was not specified) or not. Notes ----- For a Triangulation to be valid it must not have duplicate points, triangles formed from colinear points, or overlapping triangles. Nc C sD ddl m} tj|tjd�| _tj|tjd�| _| jj| jjksL| jjdkrTt d��d | _ d | _ d | _ d| _ |d kr�|�||�\| _| _ d| _ njtj|tjdd �| _| jjd ks�| jjd d kr�t d ��| j�� t| j�kr�t d ��| j�� dk r�t d��|d k �r4tj|td�| _ | j j| jjd fk�r4t d��d | _d | _d S )Nr )�_qhull��dtype� z'x and y must be equal-length 1-D arraysFT�C)r �order� � z triangles must be a (?,3) arrayz&triangles max element is out of boundsz&triangles min element is out of bounds�3mask array must have same length as triangles array)� matplotlibr �np�asarrayZfloat64�x�y�shape�ndim� ValueError�mask�_edges� _neighbors� is_delaunayZdelaunay� trianglesZarray�int32�max�len�min�bool�_cpp_triangulation� _trifinder)�selfr r r r r � r! �>/usr/lib/python3/dist-packages/matplotlib/tri/triangulation.py�__init__& s2 zTriangulation.__init__c C s | � � �|�S )au Calculate plane equation coefficients for all unmasked triangles from the point (x, y) coordinates and specified z-array of shape (npoints). The returned array has shape (npoints, 3) and allows z-value at (x, y) position in triangle tri to be calculated using ``z = array[tri, 0] * x + array[tri, 1] * y + array[tri, 2]``. )�get_cpp_triangulation� calculate_plane_coefficients)r �zr! r! r" r% O s z*Triangulation.calculate_plane_coefficientsc C s | j dkr| �� �� | _ | j S )aN Return integer array of shape (nedges, 2) containing all edges of non-masked triangles. Each row defines an edge by it's start point index and end point index. Each edge appears only once, i.e. for an edge between points *i* and *j*, there will only be either *(i, j)* or *(j, i)*. N)r r$ Z get_edges�r r! r! r" �edgesY s zTriangulation.edgesc C sD ddl m} | jdkr>|�| j| j| j| j| j| j | j �| _| jS )zc Return the underlying C++ Triangulation object, creating it if necessary. r )�_triN) r r) r r r r r r r r r )r r) r! r! r" r$ g s �z#Triangulation.get_cpp_triangulationc C s" | j dk r| j| j S | jS dS )zC Return an array of triangles that are not masked. N)r r r' r! r! r" �get_masked_triangless s z"Triangulation.get_masked_trianglesc O s� t | d t�r$| d }| dd� } n�| d }| d }| dd� } |�dd�}d}|dkrh| rh| d }d}|dk r�ztj|tjd�}W n tk r� d}Y nX |dk r�|jdks�|jd d kr�d}|dk r�|r�| dd� } |�d d�}t||||�}|| |fS ) a� Return a Triangulation object from the args and kwargs, and the remaining args and kwargs with the consumed values removed. There are two alternatives: either the first argument is a Triangulation object, in which case it is returned, or the args and kwargs are sufficient to create a new Triangulation to return. In the latter case, see Triangulation.__init__ for the possible args and kwargs. r r Nr r FTr r r ) � isinstancer �popr r r r r r )�args�kwargsZ triangulationr r r Z from_argsr r! r! r" �get_from_args_and_kwargs| s2 � z&Triangulation.get_from_args_and_kwargsc C s& | j dkr ddlm} || �| _ | j S )z� Return the default :class:`matplotlib.tri.TriFinder` of this triangulation, creating it if necessary. This allows the same TriFinder object to be easily shared. Nr )�TrapezoidMapTriFinder)r Zmatplotlib.tri.trifinderr0 )r r0 r! r! r" � get_trifinder� s zTriangulation.get_trifinderc C s | j dkr| �� �� | _ | j S )a� Return integer array of shape (ntri, 3) containing neighbor triangles. For each triangle, the indices of the three triangles that share the same edges, or -1 if there is no such neighboring triangle. neighbors[i,j] is the triangle that is the neighbor to the edge from point index triangles[i,j] to point index triangles[i,(j+1)%3]. N)r r$ Z get_neighborsr' r! r! r" � neighbors� s zTriangulation.neighborsc C sz |dkrd| _ n.tj|td�| _ | j j| jjd fkr>td��| jdk rV| j�| j � d| _ d| _ | j dk rv| j � � dS )zp Set or clear the mask array. This is either None, or a boolean array of shape (ntri). Nr r r ) r r r r r r r r �set_maskr r r Z _initialize)r r r! r! r" r3 � s zTriangulation.set_mask)NN)�__name__� __module__� __qualname__�__doc__r# r% �propertyr( r$ r* � staticmethodr/ r1 r2 r3 r! r! r! r" r s ! ) - r )Znumpyr �objectr r! r! r! r" �<module> s