^j?dZddlmZddlmZddlmZmZddlZ ddl m Z ddl m Z erddlmZGd d eZd e e j$d e e j&d dfdZGddZy)z0Python module to interface with wrapped meshfix.) find_spec)Path) TYPE_CHECKINGAnyN)NDArray)_meshfixPolyDatac eZdZdfd ZxZS)InvalidMeshFixInputErrorc,|d}t||y)NzInvalid input. Please input a surface mesh, vertex and face arrays, or a file name. Note that pyvista is required when loading anything other than arrays.)super__init__)selfmessage __class__s J/opt/ringagent/.cad-venv/lib/python3.12/site-packages/pymeshfix/meshfix.pyrz!InvalidMeshFixInputError.__init__s! ?  !)N)__name__ __module__ __qualname__r __classcell__)rs@rr r s ""rr pointsfacesreturnr ctd tdddlm}ddlm}ddlm}ddlm }|jdk7r td |}||_ |j}tjd|j d z|j"d tj$ }||d | } ||j'd | } |} | j)| | |j+| |S)a Generate a polydata from a faces array containing no padding and all triangles. Parameters ---------- points : np.ndarray Points array. faces : np.ndarray ``(n, 3)`` faces array. Returns ------- PolyData New mesh. pyvista.corezCTo use this feature install pyvista with: `pip install pyvistarr ) numpy_to_vtk)vtkTypeInt32Array) vtkCellArrayz&Expected a two dimensional face array.)dtypeF)deep array_type)rModuleNotFoundErrorpyvista.core.pointsetr vtkmodules.util.numpy_supportrvtkmodules.vtkCommonCorervtkmodules.vtkCommonDataModelr ndim ValueErrorr GetDataTypenparangesizeshapeint32ravelSetDataSetPolys) rrr rrr pdata vtk_dtypeoffset offset_vtk faces_vtkcarrs r_polydata_from_facesr<s" (! S  /::: zzQABB JEEL"#//1I YYq%**q.%++a. IFf5YGJU[[]9MI >DLLY' NN4 LrceZdZdZdddefdZedefdZde e jd e e jdd fd Z dee e je e jffd Zed$d Zd$dZede e jfdZede e jfdZd%dedefdZ d&dededd fdZd'dededefdZd(dZd(dZd)dededefdZd*dedefd Zd*dedefd!Zd%d"eezfd#Zy )+MeshFixaClean and tetrahedralize surface meshes using MeshFix. Parameters ---------- args : pyvista.PolyData | (np.ndarray, np.ndarray) | pathlib.Path | str Either a pyvista surface mesh :class:`pyvista.PolyData` or a ``n x 3`` vertex array and ``n x 3`` face array (indices of the triangles). Also supports reading directly from a file. verbose : bool, default: False Set this to ``True`` to enable additional output from MeshFix. Examples -------- Create a meshfix object from a pyvista mesh. >>> from pyvista import examples >>> from pymeshfix import MeshFix >>> cow = examples.download_cow() >>> mfix = MeshFix(cow) Create a meshfix object from two numpy arrays. This example is incomplete as it does not contain the entire array. >>> import numpy as np >>> points = np.array( ... [ ... [3.71636, 2.343387, 0.0], ... [4.126565, 0.642027, 0.0], ... [3.454971, 2.169877, 0.0], ... ..., ... [4.12616, 2.12093, 1.17252], ... [4.133175, 2.175231, 1.259323], ... [4.232341, 1.903079, 0.534362], ... ], ... dtype=float32, ... ) >>> faces = np.array( ... [ ... [210, 252, 251], ... [250, 251, 252], ... [201, 253, 210], ... ..., ... [1965, 2193, 2194], ... [2391, 2398, 970], ... [966, 961, 970], ... ] ... ) >>> mfix = MeshFix(points, faces) F)verboser?ctj|_|jj| t |dk(r t t |dtjrOt |dk7st |dtjs td|j|d|dytd t ddl m }t |d|jr|j|d}nft |dt t"frB|j%|d}t ||js#t dt'|dt d |j(j+tj,d }|j.s|j1}|j2j5d d j+tj6d }|j||y)zInitialize meshfix.rr!r"z?If first argument is an array, second argument must be an arrayrNz:Expected to load a `pyvista.PolyData` from file, but got ``z\Invalid input. Please input a pyvista.PolyData, vertex and face arrays, or a path to a file.Fcopy)rPyTMesh_mfix set_quietlenr isinstancer.ndarray TypeError load_arraysr pyvista.corecorer wraprstrreadtyperastypefloat64is_all_triangles triangulate_connectivity_arrayreshaper2)rr?argspvmeshvfs rrzMeshFix.__init__~s%%'  [) t9>*, , d1grzz *4yA~ZQ%D abb   T!Wd1g .(0.00 %$q'2;;/wwtAw'DGdC[1wwtAw'!$ 42TUYZ^U_T``ab/+ ""2::E":A(('')((00Q7>>rxxe>TA   Q "rrc.|jjS)z5Return the number of boundaries (holes) in this mesh.)rG n_boundariesrs rr`zMeshFix.n_boundariesszz&&&rr]r^Nc|jj|jtjd|jtj dy)a Load triangular mesh from vertex and face numpy arrays. Both vertex and face arrays should be 2D arrays with each vertex containing XYZ data and each face containing three points. Parameters ---------- v : np.ndarray[np.float64] ``(n, 3)`` vertex array. f : np.ndarray[np.int32] ``(m, 3)`` face array. Examples -------- Create a meshfix object from two numpy arrays. This example is incomplete as it does not contain the entire array. >>> from pymeshfix import MeshFix >>> import numpy as np >>> points = np.array( ... [ ... [3.71636, 2.343387, 0.0], ... [4.126565, 0.642027, 0.0], ... [3.454971, 2.169877, 0.0], ... ..., ... [4.12616, 2.12093, 1.17252], ... [4.133175, 2.175231, 1.259323], ... [4.232341, 1.903079, 0.534362], ... ], ... dtype=float32, ... ) >>> faces = np.array( ... [ ... [210, 252, 251], ... [250, 251, 252], ... [201, 253, 210], ... ..., ... [1965, 2193, 2194], ... [2391, 2398, 970], ... [966, 961, 970], ... ] ... ) >>> mfix = MeshFix(points, faces) FrBN)rG load_arrayrTr.rUr2)rr]r^s rrMzMeshFix.load_arrayss?^  HHRZZeH , HHRXXEH * rc6|jjS)z Return the arrays from the mesh fix instance. Returns ------- np.ndarray[np.float64] Array of points shaped ``(n, 3)``. np.ndarray[np.float64] Array of faces shaped ``(m, 3)``. )rG return_arraysras r_return_arrayszMeshFix._return_arraysszz''))rcBt|j|jS)aY Return the surface mesh. Returns ------- pyvista.PolyData Surface mesh as a PolyData object. Examples -------- Create a meshfix object from a pyvista mesh. >>> from pyvista import examples >>> from pymeshfix import MeshFix >>> cow = examples.download_cow() >>> mfix = MeshFix(cow) >>> mfix.mesh PolyData (0x7fa3c735f8e0) N Cells: 5804 N Points: 2903 N Strips: 0 X Bounds: -4.446e+00, 5.998e+00 Y Bounds: -3.637e+00, 2.760e+00 Z Bounds: -1.701e+00, 1.701e+00 N Arrays: 0 Access the underlying mesh and plot it. >>> mfix.mesh.plot() Access the underlying mesh and export it. >>> mfix.mesh.save("my_mesh.ply") )r<rrras rr\z MeshFix.meshsJ$DKK<|jjdddS)zPExtract the boundaries of the holes in this mesh to a new PyVista mesh of lines.TF)boundary_edges feature_edgesmanifold_edges)r\extract_feature_edgesras r extract_holeszMeshFix.extract_holess%yy..uU/  rc6|jjS)a" Return the points of the mesh. Returns ------- numpy.ndarray The points of the mesh. Examples -------- Generate a MeshFix object and return the points. >>> from pyvista import examples >>> from pymeshfix import MeshFix >>> cow = examples.download_cow() >>> mfix = MeshFix(cow) >>> mfix.points pyvista_ndarray([[3.71636 , 2.343387, 0. ], [4.126565, 0.642027, 0. ], [3.454971, 2.169877, 0. ], ..., [4.12616 , 2.12093 , 1.17252 ], [4.133175, 2.175231, 1.259323], [4.232341, 1.903079, 0.534362]], dtype=float32) )rG return_pointsras rrzMeshFix.pointss8zz''))rc6|jjS)a Return the indices of the faces of the mesh. Returns ------- numpy.ndarray The points of the mesh. Examples -------- Generate a MeshFix object and return the points. >>> from pyvista import examples >>> from pymeshfix import MeshFix >>> cow = examples.download_cow() >>> mfix = MeshFix(cow) >>> mfix.faces array([[ 210, 252, 251], [ 250, 251, 252], [ 201, 253, 210], ..., [1965, 2193, 2194], [2391, 2398, 970], [ 966, 961, 970]]) )rG return_facesras rrz MeshFix.faces:s8zz&&((r show_holeskwargsc  ddl}|r|j}|jdd}|jdd}|j |}d|vr|jdd|_|j |jfd d i||jr$|j |d d |j|j|S|jjdi|S#t$r tdwxYw)aPlot the mesh. Parameters ---------- show_holes : bool, default: True Shows boundaries. **kwargs : keyword arguments Additional keyword arguments. See :func:`pyvista.plot`. Examples -------- Load the example bunny mesh and plot it. >>> from pyvista import examples >>> from pymeshfix import MeshFix >>> mesh = examples.download_bunny() >>> mfix = MeshFix(mesh) >>> mfix.plot(show_holes=True) rNz#Install pyvista to use this feature off_screenF screenshot)rucposlabelr\redges)rx)rv) pyvistar&rmpopPlottercamera_positionadd_meshr\n_points add_legendshowplot)rrrrsr[rzrurvplotters rrz MeshFix.plotXs, M  &&(EL%8JL$7JjjJj7G*0**VT*B' G  TYY ?f ? ?~~  7 ;""$<<:<6 6tyy~~'''%# M%&KL L Ms C&&C;joincompremove_smallest_componentsc|jjdd|r|jj|r|jj|jj y)aK Perform mesh repair using MeshFix's default repair process. Parameters ---------- fill_holes : bool, default: True Call meshfix's internal ``fill_small_boundaries``. joincomp : bool, default: False Attempts to join nearby open components. remove_smallest_components : bool, default: True Remove all but the largest isolated component from the mesh before beginning the repair process. Notes ----- Vertex and face arrays can be accessed from: * :attr:`Meshfix.points` * :attr:`Meshfix.faces` You can alternatively call individual repair functions if you desire specific fixes. For example :func:`MeshFix.fill_holes`. Examples -------- Repair using the verbose option. >>> from pyvista import examples >>> from pymeshfix import MeshFix >>> mesh = examples.download_bunny() >>> mfix = MeshFix(mesh, verbose=True) >>> mfix.repair() >>> mfix.plot(show_holes=True) rTN)rGfill_small_boundariesjoin_closest_componentsrclean)rrrs rrepairzMeshFix.repairsOP ((D1  JJ . . 0 % JJ 1 1 3 rn_edgesrefinec:|jj||S)ae Fill small boundary loops (holes) in the mesh. Parameters ---------- nbe : int, default: 0 Maximum number of boundary edges to fill. If 0, fill all. refine : bool, default: True Refine filled regions. Returns ------- int Number of holes filled. )rGr)rrrs r fill_holeszMeshFix.fill_holess"zz//@@rc8|jjy)z'Attempt to join nearby open components.N)rGrras rrzMeshFix.join_closest_componentss **,rc8|jjy)z/Remove all but the largest connected component.N)rGrras rrz"MeshFix.remove_smallest_componentss --/r max_iters inner_loopsc:|jj||S)a Remove degenerate triangles and self-intersections. Iteratively calls :func:`MeshFix.strong_degeneracy_removal` and :func:MeshFix.strong_intersection_removal to produce a clean mesh without degeneracies and intersections. The two aforementioned methods are called up to max_iter times and each of them is called using 'inner_loops' as a parameter. Returns ``True`` only if the mesh could be completely cleaned. Parameters ---------- max_iters )rGr)rrrs rrz MeshFix.cleans zz ;77rmax_iterc8|jj|S)a Remove degenerate triangles. Parameters ---------- max_iter : int, default: 3 Maximum number of iterations to perform. Returns ------- bool ``True`` when successful. )rGstrong_degeneracy_removalrrs rdegeneracy_removalzMeshFix.degeneracy_removals zz33H==rc8|jj|S)a  Remove self-intersecting triangles. Parameters ---------- max_iter : int, default: 3 Maximum number of iterations to perform. Returns ------- bool ``True`` when successful. )rGstrong_intersection_removalrs rintersection_removalzMeshFix.intersection_removalszz55h??rfilenamec:|jj||S)a Write the points and faces as a surface mesh to disk using PyVista. This is a a simple wrapper for :pyvista:`pyvista.PolyData.save`. Parameters ---------- filename : str Filename of mesh to be written. Filetype is inferred from the extension of the filename unless overridden with ftype. Generally one of the following types: - ``".ply"`` - ``".stl"`` - ``".vtk"`` binary: bool, default: True Write the file using a binary writer. Binary files read and write much faster than ASCII. )r\save)rrbinarys rrz MeshFix.saves,yy~~h//r)rr )T)FT)rT)rN) rE)rE) rrr__doc__boolrpropertyintr`rr.rUr2rMtuplerfr\rmrrrrrrrrrrrrQrrr{rrr>r>Js1f/4(#t(#T'c''2 WRZZ02 WRXX5F2 42 h *gbjj&97288;L&L M *$=$=L * +**:)wrxx()):*(t*(c*(\+/--%)-  -^A#A4A3A&-08s8c8$8$>3>t>$@S@@"0S4Z0rr>)rimportlib.utilrpathlibrtypingrrnumpyr. numpy.typingr pymeshfixrr'r rLr rUr2r<r>r{rrrsf6$% ."y",!4,WRXX=N,S],^O0O0r