CoolFace
Apppublic

arzhela/QuadOpt-RL

sourceHugging Facemitupdated 8mo agoView on Hugging Face
0likes
rootmesh_model__pycache__reader.cpython-39.pyc
reader.cpython-39.pyc34 linesDownload Raw Back to __pycache__
1a

2��xi5�@sxddlZddlZddlZddlmZdd�Zdd�Zeed�dd	�Zeed�d3d�Zed�d
d�Z	e4ed�dd�ZdS)�N)�MeshcCsDg}t�|�D](\}}}|D]}|�d�r|�|�qq|��|S)N�.msh)�os�walk�endswith�append�sort��d�lZsdpZLsdZLnf�f�r
�D/home/ropercha/Documents/HuggingFace/QuadOpt-RL/mesh_model/reader.py�getMeshFiless5rcCs:g}t�|�D]\}}}|D]}|�|�qq|��|S)N)rrrrr	r
r
r�getFilessr)�filename�returncCst|d�}g}g}|��}|s"�qd|��kr�|��}|��}tt|��D]:}|��}|��}t|d�}t|d�}	|�||	g�qJd|��kr|��}|��}6tt|7��D]T}|��}|��}t|d�}t|d�}t|d�}
|�|d|d|
dg�q�q|��t	||�}|S)zo8    Create a mesh read from a Medit .mesh file.9    :param filename: name of the file10    :return: a mesh11    �rZVerticesr�Z	Triangles�)12�open�readline�strip�range�int�split�floatr�closer)rr�nodes�faces�lineZnb_vertices�_�ls�x�yZnb_faces�n0�n1�n2�meshr
r
r�13read_medits814 15r)c	Cs\t|d�}g}g}|��}|s"�qFd|��kr�|��}|��}t|d�}t|�D]�}|��}|��}t|d�}t|�D]}|��}qzt|�D]H}|��}|��}t|d�}	t|d�}16t|d�}|�|	|17|g�q�qRd|��kr|��}|��}t|d�}t|�D�]4}|��}|��}t|d�}t|d�}
t|
�D]�}|��}|��}|dk�r�t|d�}t|d�}t|d�}|�|d|d|dg�n�|dk�rt|d�}t|d�}t|d�}t|d�}|�|d|d|d|dg�n<|dk�r�qBn,|d	k�r �qBntd18t	|�d�t19d��qB�qq|��t||�}|S)zm20    Create a mesh read from a gmsh .msh file.21    :param filename: name of the file22    :return: a mesh23    rz$Nodesr�rrz	$Elements��z
element_type z not handled)
rrrrrrrr�print�str�exitrr)rrrrr r"Znb_blocsr!Z24nb_nodes_br#r$�z�	elem_typeZ25nb_elems_br%r&r'Zn3r(r
r
r�	read_gmshAsn26272829$303132r2)rcCsJt|d��}t�|�}Wd�n1s*0Yt|d|d�}|S)Nrrr)r�json�loadr)rrZ	json_meshr(r
r
r�	read_json�s(r5cCshg}t|�D]V}|�d�r:|d|}t|�}|�|�q|�d�r|d|}t|�}|�|�q|S)Nr�/z.json)rrr2rr5)�dataset_dirZmesh_datasetrZmsh_f�cmapZjson_fr
r
r�read_dataset�s3334r9)�stringr3rZmesh_model.mesh_struct.meshrrrr)r2r5�listr9r
r
r
r�<module>s	)E