OwlCyberSecurity - MANAGER
Edit File: tritools.cpython-38.pyc
U ��]�0 � @ s, d Z ddlZddlmZ G dd� de�ZdS )z Tools for triangular grids. � N)� Triangulationc @ sP e Zd ZdZdd� Zedd� �Zddd�Zdd d �Zdd d�Z e ddd��Z dS )� TriAnalyzera� Define basic tools for triangular mesh analysis and improvement. A TriAnalyzer encapsulates a :class:`~matplotlib.tri.Triangulation` object and provides basic tools for mesh analysis and mesh improvement. Parameters ---------- triangulation : :class:`~matplotlib.tri.Triangulation` object The encapsulated triangulation to analyze. Attributes ---------- `scale_factors` c C s t |t�std��|| _d S )Nz Expected a Triangulation object)� isinstancer � ValueError�_triangulation)�selfZ triangulation� r �9/usr/lib/python3/dist-packages/matplotlib/tri/tritools.py�__init__ s zTriAnalyzer.__init__c C sT | j �� }tjt�|�| j jjd�dk}dt�| j j| � dt�| j j| � fS )ar Factors to rescale the triangulation into a unit square. Returns *k*, tuple of 2 scale factors. Returns ------- k : tuple of 2 floats (kx, ky) Tuple of floats that would rescale the triangulation : ``[triangulation.x * kx, triangulation.y * ky]`` fits exactly inside a unit square. �Z minlengthr � ) r �get_masked_triangles�np�bincount�ravel�x�sizeZptp�y)r �compressed_trianglesZ node_usedr r r � scale_factors s ���zTriAnalyzer.scale_factorsTc C s: |r| j \}}nd\}}t�| jj| | jj| g�j}|| jj }|dd�ddd�f |dd�ddd�f }|dd�ddd�f |dd�ddd�f }|dd�ddd�f |dd�ddd�f }t�|dd�df |dd�df �}t�|dd�df |dd�df �}t�|dd�df |dd�df �}|| | d } | || | || | || | } | dk} t� | ��r�|j d } tj | tj d�} tj | | < || | }|| d t�| | � | | < n|| | d t�| � } || | d | | }|| }| jj}|dk�r&|S tjj||d �S dS ) a� Returns a measure of the triangulation triangles flatness. The ratio of the incircle radius over the circumcircle radius is a widely used indicator of a triangle flatness. It is always ``<= 0.5`` and ``== 0.5`` only for equilateral triangles. Circle ratios below 0.01 denote very flat triangles. To avoid unduly low values due to a difference of scale between the 2 axis, the triangular mesh can first be rescaled to fit inside a unit square with :attr:`scale_factors` (Only if *rescale* is True, which is its default value). Parameters ---------- rescale : boolean, optional If True, a rescaling will be internally performed (based on :attr:`scale_factors`, so that the (unmasked) triangles fit exactly inside a unit square mesh. Default is True. Returns ------- circle_ratios : masked array Ratio of the incircle radius over the circumcircle radius, for each 'rescaled' triangle of the encapsulated triangulation. Values corresponding to masked triangles are masked out. )� �?r Nr r � g �?g �Zdtypeg @)�mask)r r Zvstackr r r �T� trianglesZhypot�any�shape�emptyZfloat64�infZsqrtr �maZarray)r �rescaleZkxZkyZptsZtri_pts�a�b�c�sZprodZ bool_flat�ntriZ circum_radius�abcZ in_radiusZ circle_ratior r r r � circle_ratios5 s> � ,,,$$$( � zTriAnalyzer.circle_ratios�{�G�z�?c C s� | j jjd }| �|�|k }| j j}|dkr:tj|td�}t�| j j �}tj |tj d�}d}|dkr�tj |dd�dk| @ } | |@ } | |B }t� | �}d|| dd�f< d|| < t�|dkd|| �}q\tj�|d�S )a� Eliminates excessively flat border triangles from the triangulation. Returns a mask *new_mask* which allows to clean the encapsulated triangulation from its border-located flat triangles (according to their :meth:`circle_ratios`). This mask is meant to be subsequently applied to the triangulation using :func:`matplotlib.tri.Triangulation.set_mask`. *new_mask* is an extension of the initial triangulation mask in the sense that an initially masked triangle will remain masked. The *new_mask* array is computed recursively; at each step flat triangles are removed only if they share a side with the current mesh border. Thus no new holes in the triangulated domain will be created. Parameters ---------- min_circle_ratio : float, optional Border triangles with incircle/circumcircle radii ratio r/R will be removed if r/R < *min_circle_ratio*. Default value: 0.01 rescale : boolean, optional If True, a rescaling will first be internally performed (based on :attr:`scale_factors` ), so that the (unmasked) triangles fit exactly inside a unit square mesh. This rescaling accounts for the difference of scale which might exist between the 2 axis. Default (and recommended) value is True. Returns ------- new_mask : array-like of booleans Mask to apply to encapsulated triangulation. All the initially masked triangles remain masked in the *new_mask*. Notes ----- The rationale behind this function is that a Delaunay triangulation - of an unstructured set of points - sometimes contains almost flat triangles at its border, leading to artifacts in plots (especially for high-resolution contouring). Masked with computed *new_mask*, the encapsulated triangulation would contain no more unmasked border triangles with a circle ratio below *min_circle_ratio*, thus improving the mesh quality for subsequent plots or interpolation. r Nr ���r )ZaxisT)r r r r( r r Zzeros�bool�copyZ neighbors�arange�int32�min�sum�wherer Zfilled) r Zmin_circle_ratior! r& Zmask_bad_ratioZ current_maskZvalid_neighborsZrenum_neighborsZnaddZ wavefrontZ added_maskr r r �get_flat_tri_maskz s& 1 �z TriAnalyzer.get_flat_tri_maskFc C s� | j j}| j �� }| j jjd }| �||�}tjt�|�| j j j d�dk}| j j | }| j j | } | �|�} | | }|s�|s�||| fS ||| | fS n |s�||| |fS ||| || fS dS )a8 Compress (if masked) the encapsulated triangulation. Returns minimal-length triangles array (*compressed_triangles*) and coordinates arrays (*compressed_x*, *compressed_y*) that can still describe the unmasked triangles of the encapsulated triangulation. Parameters ---------- return_tri_renum : boolean, optional Indicates whether a renumbering table to translate the triangle numbers from the encapsulated triangulation numbering into the new (compressed) renumbering will be returned. return_node_renum : boolean, optional Indicates whether a renumbering table to translate the nodes numbers from the encapsulated triangulation numbering into the new (compressed) renumbering will be returned. Returns ------- compressed_triangles : array-like the returned compressed triangulation triangles compressed_x : array-like the returned compressed triangulation 1st coordinate compressed_y : array-like the returned compressed triangulation 2nd coordinate tri_renum : array-like of integers renumbering table to translate the triangle numbers from the encapsulated triangulation into the new (compressed) renumbering. -1 for masked triangles (deleted from *compressed_triangles*). Returned only if *return_tri_renum* is True. node_renum : array-like of integers renumbering table to translate the point numbers from the encapsulated triangulation into the new (compressed) renumbering. -1 for unused points (i.e. those deleted from *compressed_x* and *compressed_y*). Returned only if *return_node_renum* is True. r r N) r r r r r �_total_to_compress_renumr r r r r r ) r Zreturn_tri_renumZreturn_node_renumZtri_maskr r& Z tri_renumZ node_maskZ compressed_xZ compressed_yZ node_renumr r r � _get_compressed_triangulation� s6 ) �� �� �z)TriAnalyzer._get_compressed_triangulationNc C sv |dkrt �| �}| dk rbt j|dt jd�}t j|t jd�| }t jt �|d�t jd�||< |S t j|t jd�S dS )aT Parameters ---------- mask : 1d boolean array or None mask n : integer length of the mask. Useful only id mask can be None Returns ------- renum : integer array array so that (`valid_array` being a compressed array based on a `masked_array` with mask *mask*) : - For all i such as mask[i] = False: valid_array[renum[i]] = masked_array[i] - For all i such as mask[i] = True: renum[i] = -1 (invalid value) Nr* r r )r r Zfullr. r- )r �nZrenumZvalidr r r r3 s z$TriAnalyzer._total_to_compress_renum)T)r) T)FF)N) �__name__� __module__� __qualname__�__doc__r �propertyr r( r2 r4 � staticmethodr3 r r r r r s E L � Gr )r9 Znumpyr Zmatplotlib.trir �objectr r r r r �<module> s