OwlCyberSecurity - MANAGER
Edit File: trirefine.cpython-38.pyc
U ��]6 � @ sD d Z ddlZddlmZ ddlZG dd� de�ZG dd� de�Z dS )z' Mesh refinement for triangular grids. � N)� Triangulationc @ s e Zd ZdZdd� ZdS )� TriRefinera� Abstract base class for classes implementing mesh refinement. A TriRefiner encapsulates a Triangulation object and provides tools for mesh refinement and interpolation. Derived classes must implements: - ``refine_triangulation(return_tri_index=False, **kwargs)`` , where the optional keyword arguments *kwargs* are defined in each TriRefiner concrete implementation, and which returns: - a refined triangulation - optionally (depending on *return_tri_index*), for each point of the refined triangulation: the index of the initial triangulation triangle to which it belongs. - ``refine_field(z, triinterpolator=None, **kwargs)`` , where: - *z* array of field values (to refine) defined at the base triangulation nodes - *triinterpolator* is a :class:`~matplotlib.tri.TriInterpolator` (optional) - the other optional keyword arguments *kwargs* are defined in each TriRefiner concrete implementation and which returns (as a tuple) a refined triangular mesh and the interpolated values of the field at the refined triangulation nodes. c C s t |t�std��|| _d S )Nz Expected a Triangulation object)� isinstancer � ValueError�_triangulation��self� triangulation� r �:/usr/lib/python3/dist-packages/matplotlib/tri/trirefine.py�__init__) s zTriRefiner.__init__N)�__name__� __module__� __qualname__�__doc__r r r r r r s r c @ s: e Zd ZdZdd� Zd dd�Zdd d �Zedd d ��ZdS )�UniformTriRefinerz� Uniform mesh refinement by recursive subdivisions. Parameters ---------- triangulation : :class:`~matplotlib.tri.Triangulation` The encapsulated triangulation (to be refined) c C s t �| |� d S )N)r r r r r r r = s zUniformTriRefiner.__init__F� c C s� | j }|jjd }tj|tjd�}t|�D ]}| �||�\}}q*|jjd }|j}|r�tj |dtjd�} | j j } | dkr�t� |d�� dd�| |< n\| | } t� || d�� dd�| || dd�f < t� || d�� dd�| || dd�f < || fS |S dS )ao Computes an uniformly refined triangulation *refi_triangulation* of the encapsulated :attr:`triangulation`. This function refines the encapsulated triangulation by splitting each father triangle into 4 child sub-triangles built on the edges midside nodes, recursively (level of recursion *subdiv*). In the end, each triangle is hence divided into ``4**subdiv`` child triangles. The default value for *subdiv* is 3 resulting in 64 refined subtriangles for each triangle of the initial triangulation. Parameters ---------- return_tri_index : boolean, optional Boolean indicating whether an index table indicating the father triangle index of each point will be returned. Default value False. subdiv : integer, optional Recursion level for the subdivision. Defaults value 3. Each triangle will be divided into ``4**subdiv`` child triangles. Returns ------- refi_triangulation : :class:`~matplotlib.tri.Triangulation` The returned refined triangulation found_index : array-like of integers Index of the initial triangulation containing triangle, for each point of *refi_triangulation*. Returned only if *return_tri_index* is set to True. r �Zdtype���Nr ) r � triangles�shape�np�arange�int32�range�_refine_triangulation_once�xZfull�mask�repeatZreshape) r �return_tri_index�subdivZrefi_triangulation�ntri� ancestors�_� refi_nptsZrefi_triangles� found_indexZtri_maskZ ancestor_maskr r r �refine_triangulation@ sJ ! � � � � �� � ��z&UniformTriRefiner.refine_triangulationNc C sf |dkrt j�| j|�}nt|t jj�s0td��|}| j|dd�\}}|j|j |j |d�d }||fS )a� Refines a field defined on the encapsulated triangulation. Returns *refi_tri* (refined triangulation), *refi_z* (interpolated values of the field at the node of the refined triangulation). Parameters ---------- z : 1d-array-like of length ``n_points`` Values of the field to refine, defined at the nodes of the encapsulated triangulation. (``n_points`` is the number of points in the initial triangulation) triinterpolator : :class:`~matplotlib.tri.TriInterpolator`, optional Interpolator used for field interpolation. If not specified, a :class:`~matplotlib.tri.CubicTriInterpolator` will be used. subdiv : integer, optional Recursion level for the subdivision. Defaults to 3. Each triangle will be divided into ``4**subdiv`` child triangles. Returns ------- refi_tri : :class:`~matplotlib.tri.Triangulation` object The returned refined triangulation refi_z : 1d array of length: *refi_tri* node count. The returned interpolated field (at *refi_tri* nodes) Nz!Expected a TriInterpolator objectT)r r )Z tri_indexr ) � matplotlibZtriZCubicTriInterpolatorr r ZTriInterpolatorr r& Z_interpolate_multikeysr �y)r �zZtriinterpolatorr ZinterpZrefi_trir% Zrefi_zr r r � refine_field� s, �� � ��z UniformTriRefiner.refine_fieldc $ C s� | j }| j}| j}| j}t�|�d }t�|�d }|dk rrt�|�}t�|�|fkrrtd�t�|�t�|����t� |dk�}d| | d } || } t� | �} t� | �} || d|�< || d|�< t� tj |tj d�d�} t�tj dtj d�|�}|| |f }| |k}| | }|| }||||f ||||d d f d }||||f ||||d d f d }|| |d�< || |d�< |}tj|dgtj d�}|}td�D ]L}||k}t� |�}|| }tj |tj d�| |dd�|f |< ||7 }�q�t�|�} | | } || } t�|| dd�f t�| tjdtj d�� �} tj| dd �} || }!|| | f || |!f< tj|d dgtj d�}"t�|dd�df |dd�df |dd�df g�j|"ddd �dd�f< t�|dd�df |dd�df |dd�df g�j|"ddd �dd�f< t�|dd�df |dd�df |dd�df g�j|"ddd �dd�f< t�|dd�df |dd�df |dd�df g�j|"ddd �dd�f< t| | |"�}#| jdk �r�|#�t�| jd �� |dk�r�|#S |#t�|d �fS dS ) a This function refines a matplotlib.tri *triangulation* by splitting each triangle into 4 child-masked_triangles built on the edges midside nodes. The masked triangles, if present, are also split but their children returned masked. If *ancestors* is not provided, returns only a new triangulation: child_triangulation. If the array-like key table *ancestor* is given, it shall be of shape (ntri,) where ntri is the number of *triangulation* masked_triangles. In this case, the function returns (child_triangulation, child_ancestors) child_ancestors is defined so that the 4 child masked_triangles share the same index as their father: child_ancestors.shape = (4 * ntri,). r NzYIncompatible shapes provide for triangulation.masked_triangles and ancestors: {0} and {1}r r � r � g �?)Zaxis� )r r( � neighborsr r r Zasarrayr �format�sumZzerosZtiler r r �emptyr Z logical_not�absZouterZonesZargminZvstack�Tr r Zset_mask)$r r" r r( r. r Znptsr! ZbordersZ added_ptsr$ Zrefi_xZrefi_yZ edge_elemsZ edge_apexesZedge_neighborsZ mask_mastersZmastersZ apex_mastersZx_addZy_addZ new_pt_cornerZnew_pt_midsideZcum_sumZimidZ mask_st_locZ n_masters_locZelem_masters_locZ mask_slavesZslavesZslaves_mastersZ diff_tableZslave_masters_apexZ slaves_apexZchild_trianglesZchild_triangulationr r r r � s� �� ���� �� �� � � � � z,UniformTriRefiner._refine_triangulation_once)Fr )Nr )N) r r r r r r& r* � staticmethodr r r r r r / s H +r ) r Znumpyr Z matplotlib.tri.triangulationr Z matplotlib.tri.triinterpolater' �objectr r r r r r �<module> s %