gO)DdZddlZddlmZddlmZGddZdS)z Tools for triangular grids. N)_api) TriangulationcZeZdZdZdZedZd dZd dZdZ e d Z d S) TriAnalyzera Define basic tools for triangular mesh analysis and improvement. A TriAnalyzer encapsulates a `.Triangulation` object and provides basic tools for mesh analysis and mesh improvement. Attributes ---------- scale_factors Parameters ---------- triangulation : `~matplotlib.tri.Triangulation` The encapsulated triangulation to analyze. cJtjt|||_dS)N) triangulation)rcheck_isinstancer_triangulation)selfrs N/granary/edev/quantum/lib/python3.11/site-packages/matplotlib/tri/_tritools.py__init__zTriAnalyzer.__init__s& m=IIII+cF|j}tjtj||jjjdk}dtj|jj|z dtj|jj|z fS)a- Factors to rescale the triangulation into a unit square. Returns ------- (float, float) Scaling factors (kx, ky) so that the triangulation ``[triangulation.x * kx, triangulation.y * ky]`` fits exactly inside a unit square.  minlengthr) r get_masked_trianglesnpbincountravelxsizeptpy)r compressed_triangles node_useds r scale_factorszTriAnalyzer.scale_factors s $2GGII[*>!?!?+/+>+@+EGGGJKL BF4.0;<<<BF4.0;<<<> >rTc|r |j\}}nd\}}tj|jj|z|jj|zgj}||jj}|dddddf|dddddfz }|dddddf|dddddfz }|dddddf|dddddfz }tj|dddf|dddf}tj|dddf|dddf}tj|dddf|dddf}||z|zdz} | ||z| z z||z| z z||z| z z} | dk} tj | rq|j d} tj | tj } tj | | <||z|z}|| d tj| | zz | | <n ||z|zd tj| zz } ||z|zd | z| zz }|| z }|jj}||Stj|| S) aN Return 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 `scale_factors` (Only if *rescale* is True, which is its default value). Parameters ---------- rescale : bool, default: True If True, internally rescale (based on `scale_factors`), so that the (unmasked) triangles fit exactly inside a unit square mesh. Returns ------- 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. )?rNrrg?gdtypeg@)mask)rrvstackr rrT triangleshypotanyshapeemptyfloat64infsqrtr#maarray)r rescalekxkyptstri_ptsabcsprod bool_flatntri circum_radiusabc in_radius circle_ratior#s r circle_ratioszTriAnalyzer.circle_ratios2s:  ")HR!HRi,.r1,.r134445 d)34 AAAq!!!G wqqq!QQQw/ / AAAq!!!G wqqq!QQQw/ / AAAq!!!G wqqq!QQQw/ / HQqqq!tWa1g & & HQqqq!tWa1g & & HQqqq!tWa1g & & qSUCK!A#a%y!A#a% !A#a%(RZ 6)   :=#DHT<<<-OI#^3J% 4L6*%%D.0OJM **,OJ ' h"'>  **>?$lL) rctjtj|dtj}tj|}tj|tj||<|S)a Parameters ---------- valid : 1D bool array Validity mask. Returns ------- int array Array so that (`valid_array` being a compressed array based on a `masked_array` with mask ~*valid*): - For all i with valid[i] = True: valid_array[renum[i]] = masked_array[i] - For all i with valid[i] = False: renum[i] = -1 (invalid value) rCr!)rfullrrJrLrI)validrenumn_valids r rYz$TriAnalyzer._total_to_compress_renumsP&"(;;;&--y999e  rN)T)rAT) __name__ __module__ __qualname____doc__r propertyrr@rWr` staticmethodrYrr rr s ,,,>>X>"A8A8A8A8FI0I0I0I0V///b\rr)rinumpyr matplotlibrmatplotlib.trirrrlrr rps}((((((||||||||||r