Coreform-Cubit-Mesh-Export 1.1.2__py3-none-any.whl → 1.3__py3-none-any.whl

{coreform_cubit_mesh_export-1.1.2.dist-info → coreform_cubit_mesh_export-1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Coreform_Cubit_Mesh_Export
-Version: 1.1.2
+Version: 1.3
 Summary: Cubit mesh export to various formats including Gmsh, Nastran, and VTK
 Author-email: Kengo Sugahara <ksugahar@gmail.com>
 Maintainer-email: Kengo Sugahara <ksugahar@gmail.com>

coreform_cubit_mesh_export-1.3.dist-info/RECORD

@@ -0,0 +1,9 @@
+cubit_mesh_export.py,sha256=STR8WvHLmVoUaWISfpn2oXhiE8ZpwF5denx8e2wEZbs,25050
+sample_script/cubit_mesh_export_2d_geo_lukas_format.py,sha256=XJh9K2nMYkpqBgSzTNhxy1dn4ZLczw0lecCCNBGaTHk,2543
+sample_script/cubit_mesh_export_3d_cdb.py,sha256=Kw6ViJ7wXBjqL5FvglvtY0ozqc_liIPEcsUVNfLAdes,4925
+sample_script/cubit_mesh_export_3d_freefem_mesh.py,sha256=doBAxvORdAj5NJ07Z2POwz_bue-AtEqDMBUajV7mk8k,2095
+sample_script/cubit_mesh_export_3d_gmesh_v4.py,sha256=PVKQB0_RBhG3qEhUo6RbzjsH6ZDf23VSyHIxqfcF5QU,6947
+coreform_cubit_mesh_export-1.3.dist-info/METADATA,sha256=JooncWOJ7jSGW6Vgj5Wslbj_r3YbNGF7HwFaL9jIaFA,4953
+coreform_cubit_mesh_export-1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+coreform_cubit_mesh_export-1.3.dist-info/top_level.txt,sha256=kKpvsJ-GNEjfO5z-ekTyHF0-sf9m62KYMGMsv4qLWJM,32
+coreform_cubit_mesh_export-1.3.dist-info/RECORD,,

cubit_mesh_export.py

@@ -14,7 +14,9 @@ def export_Gmsh_ver2(cubit, FileName):
 		fid.write(f'{cubit.get_block_count()}\n')
 		for block_id in cubit.get_block_id_list():
 			name = cubit.get_exodus_entity_name("block", block_id)
-			if len(cubit.get_block_edges(block_id)) > 0:
+			if len(cubit.get_block_nodes(block_id)) > 0:
+				fid.write(f'0 {block_id} "{name}"\n')
+			elif len(cubit.get_block_edges(block_id)) > 0:
 				fid.write(f'1 {block_id} "{name}"\n')
 			elif len(cubit.get_block_tris(block_id)) + len(cubit.get_block_faces(block_id))> 0:
 				fid.write(f'2 {block_id} "{name}"\n')
@@ -25,7 +27,7 @@ def export_Gmsh_ver2(cubit, FileName):
 		fid.write("$Nodes\n")
 		node_list = set()
 		for block_id in cubit.get_block_id_list():
-			elem_types = ["hex", "tet", "wedge", "pyramid", "tri", "face", "edge"]
+			elem_types = ["hex", "tet", "wedge", "pyramid", "tri", "face", "edge", "node"]
 			for elem_type in elem_types:
 				if elem_type == "hex":
 					func = getattr(cubit, f"get_block_{elem_type}es")
@@ -48,6 +50,7 @@ def export_Gmsh_ver2(cubit, FileName):
 		tri_list = set()
 		quad_list = set()
 		edge_list = set()
+		node_list = set()
 
 		for block_id in cubit.get_block_id_list():
 			tet_list.update(cubit.get_block_tets(block_id))
@@ -57,10 +60,11 @@ def export_Gmsh_ver2(cubit, FileName):
 			tri_list.update(cubit.get_block_tris(block_id))
 			quad_list.update(cubit.get_block_faces(block_id))
 			edge_list.update(cubit.get_block_edges(block_id))
+			node_list.update(cubit.get_block_nodes(block_id))
 
 		element_id = 0
 		fid.write('$Elements\n')
-		fid.write(f'{len(hex_list) + len(tet_list) + len(wedge_list) + len(pyramid_list) + len(tri_list) + len(quad_list) + len(edge_list)}\n')
+		fid.write(f'{len(hex_list) + len(tet_list) + len(wedge_list) + len(pyramid_list) + len(tri_list) + len(quad_list) + len(edge_list) + len(node_list)}\n')
 
 		for block_id in cubit.get_block_id_list():
 
@@ -133,140 +137,20 @@ def export_Gmsh_ver2(cubit, FileName):
 				if len(node_list)==3:
 					fid.write(f'{element_id} {8} {2} {block_id} {block_id} {node_list[0]} {node_list[1]} {node_list[2]}\n')
 
+			node_list = cubit.get_block_nodes(block_id)
+			for node_id in node_list:
+				element_id += 1
+				fid.write(f'{element_id} {15} {2} {block_id} {block_id} {node_id}\n')
+
 		fid.write('$EndElements\n')
 		fid.close()
 	return cubit
 
 ########################################################################
-###	NASTRAN 2D file
-########################################################################
-
-def export_2D_Nastran(cubit, FileName):
-	import datetime
-	formatted_date_time = datetime.datetime.now().strftime("%d-%b-%y at %H:%M:%S")
-
-	fid = open(FileName,'w',encoding='UTF-8')
-	fid.write("$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$\n")
-	fid.write("$\n")
-	fid.write("$                         CUBIT NX Nastran Translator\n")
-	fid.write("$\n")
-	fid.write(f"$            File: {FileName}\n")
-	fid.write(f"$      Time Stamp: {formatted_date_time}\n")
-	fid.write("$\n")
-	fid.write("$\n")
-	fid.write("$                        PLEASE CHECK YOUR MODEL FOR UNITS CONSISTENCY.\n")
-	fid.write("$\n")
-	fid.write("$       It should be noted that load ID's from CUBIT may NOT correspond to Nastran SID's\n")
-	fid.write("$ The SID's for the load and restraint sets start at one and increment by one:i.e.,1,2,3,4...\n")
-	fid.write("$\n")
-	fid.write("$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$\n")
-	fid.write("$\n")
-	fid.write("$\n")
-	fid.write("$ -------------------------\n")
-	fid.write("$ Executive Control Section\n")
-	fid.write("$ -------------------------\n")
-	fid.write("$\n")
-	fid.write("SOL 101\n")
-	fid.write("CEND\n")
-	fid.write("$\n")
-	fid.write("$\n")
-	fid.write("$ --------------------\n")
-	fid.write("$ Case Control Section\n")
-	fid.write("$ --------------------\n")
-	fid.write("$\n")
-	fid.write("ECHO = SORT\n")
-	fid.write("$\n")
-	fid.write("$\n")
-	fid.write("$ Name: Initial\n")
-	fid.write("$\n")
-	fid.write("$\n")
-	fid.write("$ Name: Default Set\n")
-	fid.write("$\n")
-	fid.write("SUBCASE = 1\n")
-	fid.write("$\n")
-	fid.write("LABEL = Default Set\n")
-	fid.write("$\n")
-	fid.write("$ -----------------\n")
-	fid.write("$ Bulk Data Section\n")
-	fid.write("$ -----------------\n")
-	fid.write("$\n")
-	fid.write("BEGIN BULK\n")
-	fid.write("$\n")
-	fid.write("$ Params\n")
-	fid.write("$\n")
-	fid.write("$\n")
-	fid.write("$ Node cards\n")
-	fid.write("$\n")
-
-	node_list = set()
-	for block_id in cubit.get_block_id_list():
-		elem_types = ["tri", "face", "edge"]
-		for elem_type in elem_types:
-			func = getattr(cubit, f"get_block_{elem_type}s")
-			for element_id in func(block_id):
-				node_ids = cubit.get_connectivity(elem_type, element_id)
-				node_list.update(node_ids)
-	for node_id in node_list:
-		coord = cubit.get_nodal_coordinates(node_id)
-		fid.write(f"GRID*   {node_id:>16}{0:>16}{coord[0]:>16.5f}{coord[1]:>16.5f}\n*       {coord[2]:>16.5f}\n")
-
-	element_id = 0
-	fid.write("$\n")
-	fid.write("$ Element cards\n")
-	fid.write("$\n")
-	for block_id in cubit.get_block_id_list():
-		name = cubit.get_exodus_entity_name("block",block_id)
-		fid.write("$\n")
-		fid.write(f"$ Name: {name}\n")
-		fid.write("$\n")
-
-		tri_list = cubit.get_block_tris(block_id)
-		for tri_id in tri_list:
-			element_id += 1
-			surface_id = int(cubit.get_geometry_owner("tri", tri_id).split()[1])
-			normal = cubit.get_surface_normal(surface_id)
-			node_list = cubit.get_connectivity('tri',tri_id)
-			if normal[2] > 0:
-				fid.write(f"CTRIA3  {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[1]:<8}{node_list[2]:<8}\n")
-			else:
-				fid.write(f"CTRIA3  {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[2]:<8}{node_list[1]:<8}\n")
-
-		quad_list = cubit.get_block_faces(block_id)
-		for quad_id in quad_list:
-			element_id += 1
-			surface_id = int(cubit.get_geometry_owner("quad", quad_id).split()[1])
-			normal = cubit.get_surface_normal(surface_id)
-			node_list = cubit.get_connectivity('quad',quad_id)
-			if normal[2] > 0:
-				fid.write(f"CQUAD4  {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[1]:<8}{node_list[2]:<8}{node_list[3]:<8}\n")
-			else:
-				fid.write(f"CQUAD4  {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[3]:<8}{node_list[2]:<8}{node_list[1]:<1}\n")
-	
-		edge_list = cubit.get_block_edges(block_id)
-		for edge_id in edge_list:
-			element_id += 1
-			node_list = cubit.get_connectivity('edge', edge_id)
-			if len(node_list)==2:
-				fid.write(f"CROD    {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[1]:<8}\n")
-	fid.write("$\n")
-	fid.write("$ Property cards\n")
-	fid.write("$\n")
-
-	for block_id in cubit.get_block_id_list():
-		name = cubit.get_exodus_entity_name("block",block_id)
-		fid.write("$\n")
-		fid.write(f"$ Name: {name}\n")
-		fid.write("$\n")
-		fid.write(f"MAT1    {block_id:< 8}{1:<8}{1:<8}{1:<8}\n")
-	fid.write("ENDDATA\n")
-	fid.close()
-	return cubit
-
-########################################################################
-###	3D Nastran file
+###	Nastran file
 ########################################################################
 
-def export_3D_Nastran(cubit, FileName, PYRAM=True):
+def export_Nastran(cubit, FileName, DIM="3D", PYRAM=True):
 
 	import datetime
 	formatted_date_time = datetime.datetime.now().strftime("%d-%b-%y at %H:%M:%S")
@@ -325,7 +209,7 @@ def export_3D_Nastran(cubit, FileName, PYRAM=True):
 
 	node_list = set()
 	for block_id in cubit.get_block_id_list():
-		elem_types = ["hex", "tet", "wedge", "pyramid", "tri", "face", "edge"]
+		elem_types = ["hex", "tet", "wedge", "pyramid", "tri", "face", "edge", "node"]
 		for elem_type in elem_types:
 			if elem_type == "hex":
 				func = getattr(cubit, f"get_block_{elem_type}es")
@@ -334,10 +218,12 @@ def export_3D_Nastran(cubit, FileName, PYRAM=True):
 			for element_id in func(block_id):
 				node_ids = cubit.get_connectivity(elem_type, element_id)
 				node_list.update(node_ids)
-	fid.write(f'{len(node_list)}\n')
 	for node_id in node_list:
 		coord = cubit.get_nodal_coordinates(node_id)
-		fid.write(f"GRID*   {node_id:>16}{0:>16}{coord[0]:>16.5f}{coord[1]:>16.5f}\n*       {coord[2]:>16.5f}\n")
+		if DIM == "3D":
+			fid.write(f"GRID*   {node_id:>16}{0:>16}{coord[0]:>16.5f}{coord[1]:>16.5f}\n*       {coord[2]:>16.5f}\n")
+		else:
+			fid.write(f"GRID*   {node_id:>16}{0:>16}{coord[0]:>16.5f}{coord[1]:>16.5f}\n*       {0}\n")
 
 	element_id = 0
 	fid.write("$\n")
@@ -349,46 +235,68 @@ def export_3D_Nastran(cubit, FileName, PYRAM=True):
 		fid.write(f"$ Name: {name}\n")
 		fid.write("$\n")
 		tet_list = cubit.get_block_tets(block_id)
-		for tet_id in tet_list:
-			node_list = cubit.get_connectivity('tet',tet_id)
-			element_id += 1
-			fid.write(f"CTETRA  {element_id:>8}{block_id:>8}{node_list[0]:>8}{node_list[1]:>8}{node_list[2]:>8}{node_list[3]:>8}\n")
-		hex_list = cubit.get_block_hexes(block_id)
-		for hex_id in hex_list:
-			node_list = cubit.get_connectivity('hex',hex_id)
-			element_id += 1
-			fid.write(f"CHEXA   {element_id:>8}{block_id:>8}{node_list[0]:>8}{node_list[1]:>8}{node_list[2]:>8}{node_list[3]:>8}{node_list[4]:>8}{node_list[5]:>8}+\n+       {node_list[6]:>8}{node_list[7]:>8}\n")
-		wedge_list = cubit.get_block_wedges(block_id)
-		for wedge_id in wedge_list:
-			node_list = cubit.get_connectivity('wedge',wedge_id)
-			element_id += 1
-			fid.write(f"CPENTA  {element_id:>8}{block_id:>8}{node_list[0]:>8}{node_list[1]:>8}{node_list[2]:>8}{node_list[3]:>8}{node_list[4]:>8}{node_list[5]:>8}\n")
-		pyramid_list = cubit.get_block_pyramids(block_id)
-		for pyramid_id in pyramid_list:
-			node_list = cubit.get_connectivity('pyramid',pyramid_id)
-			if PYRAM:
+
+		if DIM=="3D":
+			for tet_id in tet_list:
+				node_list = cubit.get_connectivity('tet',tet_id)
 				element_id += 1
-				fid.write(f"CPYRAM  {element_id:>8}{block_id:>8}{node_list[0]:>8}{node_list[1]:>8}{node_list[2]:>8}{node_list[3]:>8}{node_list[4]:>8}\n")
-			else:
+				fid.write(f"CTETRA  {element_id:>8}{block_id:>8}{node_list[0]:>8}{node_list[1]:>8}{node_list[2]:>8}{node_list[3]:>8}\n")
+			hex_list = cubit.get_block_hexes(block_id)
+			for hex_id in hex_list:
+				node_list = cubit.get_connectivity('hex',hex_id)
 				element_id += 1
-				fid.write(f"CHEXA   {element_id:>8}{block_id:>8}{node_list[0]:>8}{node_list[1]:>8}{node_list[2]:>8}{node_list[3]:>8}{node_list[4]:>8}{node_list[4]:>8}+\n+       {node_list[4]:>8}{node_list[4]:>8}\n")
+				fid.write(f"CHEXA   {element_id:>8}{block_id:>8}{node_list[0]:>8}{node_list[1]:>8}{node_list[2]:>8}{node_list[3]:>8}{node_list[4]:>8}{node_list[5]:>8}+\n+       {node_list[6]:>8}{node_list[7]:>8}\n")
+			wedge_list = cubit.get_block_wedges(block_id)
+			for wedge_id in wedge_list:
+				node_list = cubit.get_connectivity('wedge',wedge_id)
+				element_id += 1
+				fid.write(f"CPENTA  {element_id:>8}{block_id:>8}{node_list[0]:>8}{node_list[1]:>8}{node_list[2]:>8}{node_list[3]:>8}{node_list[4]:>8}{node_list[5]:>8}\n")
+			pyramid_list = cubit.get_block_pyramids(block_id)
+			for pyramid_id in pyramid_list:
+				node_list = cubit.get_connectivity('pyramid',pyramid_id)
+				if PYRAM:
+					element_id += 1
+					fid.write(f"CPYRAM  {element_id:>8}{block_id:>8}{node_list[0]:>8}{node_list[1]:>8}{node_list[2]:>8}{node_list[3]:>8}{node_list[4]:>8}\n")
+				else:
+					element_id += 1
+					fid.write(f"CHEXA   {element_id:>8}{block_id:>8}{node_list[0]:>8}{node_list[1]:>8}{node_list[2]:>8}{node_list[3]:>8}{node_list[4]:>8}{node_list[4]:>8}+\n+       {node_list[4]:>8}{node_list[4]:>8}\n")
+
 		tri_list = cubit.get_block_tris(block_id)
 		for tri_id in tri_list:
 			node_list = cubit.get_connectivity('tri',tri_id)
 			element_id += 1
-			fid.write(f"CTRIA3  {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[1]:<8}{node_list[2]:<8}\n")
+			if DIM=="3D":
+				fid.write(f"CTRIA3  {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[1]:<8}{node_list[2]:<8}\n")
+			else:
+				surface_id = int(cubit.get_geometry_owner("tri", tri_id).split()[1])
+				normal = cubit.get_surface_normal(surface_id)
+				if normal[2] > 0:
+					fid.write(f"CTRIA3  {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[1]:<8}{node_list[2]:<8}\n")
+				else:
+					fid.write(f"CTRIA3  {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[2]:<8}{node_list[1]:<8}\n")
 		quad_list = cubit.get_block_faces(block_id)
 		for quad_id in quad_list:
 			node_list = cubit.get_connectivity('quad',quad_id)
 			element_id += 1
-			fid.write(f"CQUAD4  {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[1]:<8}{node_list[2]:<8}{node_list[3]:<8}\n")
+			if DIM=="3D":
+				fid.write(f"CQUAD4  {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[1]:<8}{node_list[2]:<8}{node_list[3]:<8}\n")
+			else:
+				surface_id = int(cubit.get_geometry_owner("quad", quad_id).split()[1])
+				normal = cubit.get_surface_normal(surface_id)
+				node_list = cubit.get_connectivity('quad',quad_id)
+				if normal[2] > 0:
+					fid.write(f"CQUAD4  {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[1]:<8}{node_list[2]:<8}{node_list[3]:<8}\n")
+				else:
+					fid.write(f"CQUAD4  {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[3]:<8}{node_list[2]:<8}{node_list[1]:<1}\n")
 		edge_list = cubit.get_block_edges(block_id)
-		if len(edge_list)>0:
-			for edge_id in edge_list:
-				element_id += 1
-				node_list = cubit.get_connectivity('edge', edge_id)
-				if len(node_list)==2:
-					fid.write(f"CROD    {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[1]:<8}\n")
+		for edge_id in edge_list:
+			element_id += 1
+			node_list = cubit.get_connectivity('edge', edge_id)
+			fid.write(f"CROD    {element_id:<8}{block_id:<8}{node_list[0]:<8}{node_list[1]:<8}\n")
+		node_list = cubit.get_block_nodes(block_id)
+		for node_id in node_list:
+			element_id += 1
+			fid.write(f"CMASS   {element_id:<8}{block_id:<8}{node_id:<8}\n")
 	fid.write("$\n")
 	fid.write("$ Property cards\n")
 	fid.write("$\n")
@@ -397,8 +305,16 @@ def export_3D_Nastran(cubit, FileName, PYRAM=True):
 		name = cubit.get_exodus_entity_name("block",block_id)
 		fid.write("$\n")
 		fid.write(f"$ Name: {name}\n")
+		if len(cubit.get_block_nodes(block_id)) > 0:
+			fid.write(f"PMASS    {block_id:< 8}{block_id:< 8}\n")
+		elif len(cubit.get_block_edges(block_id)) > 0:
+			fid.write(f"PROD     {block_id:< 8}{block_id:< 8}\n")
+		elif len(cubit.get_block_tris(block_id)) + len(cubit.get_block_faces(block_id))> 0:
+			fid.write(f"PSHELL   {block_id:< 8}{block_id:< 8}\n")
+		else:
+			fid.write(f"PSOLID   {block_id:< 8}{block_id:< 8}\n")
 		fid.write("$\n")
-		fid.write(f"MAT1    {block_id:< 8}{1:<8}{1:<8}{1:<8}\n")
+
 	fid.write("ENDDATA\n")
 	fid.close()
 	return cubit
@@ -441,29 +357,30 @@ def export_meg(cubit, FileName, DIM='T', MGR2=[]):
 	for block_id in cubit.get_block_id_list():
 		name = cubit.get_exodus_entity_name("block",block_id)
 
-		tet_list = cubit.get_block_tets(block_id)
-		for tet_id in tet_list:
-			node_list = cubit.get_connectivity('tet',tet_id)
-			element_id += 1
-			fid.write(f"{name[0:4]}{DIM} {element_id} 0 {block_id} {node_list[0]} {node_list[1]} {node_list[2]} {node_list[3]}\n")
+		if DIM=='T':
+			tet_list = cubit.get_block_tets(block_id)
+			for tet_id in tet_list:
+				node_list = cubit.get_connectivity('tet',tet_id)
+				element_id += 1
+				fid.write(f"{name[0:4]}{DIM} {element_id} 0 {block_id} {node_list[0]} {node_list[1]} {node_list[2]} {node_list[3]}\n")
 
-		hex_list = cubit.get_block_hexes(block_id)
-		for hex_id in hex_list:
-			node_list = cubit.get_connectivity('hex',hex_id)
-			element_id += 1
-			fid.write(f"{name[0:4]}{DIM} {element_id} 0 {block_id} {node_list[0]} {node_list[1]} {node_list[2]} {node_list[3]} {node_list[4]} {node_list[5]} {node_list[6]} {node_list[7]}\n")
+			hex_list = cubit.get_block_hexes(block_id)
+			for hex_id in hex_list:
+				node_list = cubit.get_connectivity('hex',hex_id)
+				element_id += 1
+				fid.write(f"{name[0:4]}{DIM} {element_id} 0 {block_id} {node_list[0]} {node_list[1]} {node_list[2]} {node_list[3]} {node_list[4]} {node_list[5]} {node_list[6]} {node_list[7]}\n")
 
-		wedge_list = cubit.get_block_wedges(block_id)
-		for wedge_id in wedge_list:
-			node_list = cubit.get_connectivity('wedge',wedge_id)
-			element_id += 1
-			fid.write(f"{name[0:4]}{DIM} {element_id} 0 {block_id} {node_list[0]} {node_list[1]} {node_list[2]} {node_list[3]} {node_list[4]} {node_list[5]}\n")
+			wedge_list = cubit.get_block_wedges(block_id)
+			for wedge_id in wedge_list:
+				node_list = cubit.get_connectivity('wedge',wedge_id)
+				element_id += 1
+				fid.write(f"{name[0:4]}{DIM} {element_id} 0 {block_id} {node_list[0]} {node_list[1]} {node_list[2]} {node_list[3]} {node_list[4]} {node_list[5]}\n")
 
-		pyramid_list = cubit.get_block_pyramids(block_id)
-		for pyramid_id in pyramid_list:
-			node_list = cubit.get_connectivity('pyramid',pyramid_id)
-			element_id += 1
-			fid.write(f"{name[0:4]}{DIM} {element_id} 0 {block_id} {node_list[0]} {node_list[1]} {node_list[2]} {node_list[3]} {node_list[4]} {node_list[4]} {node_list[4]} {node_list[4]}\n")
+			pyramid_list = cubit.get_block_pyramids(block_id)
+			for pyramid_id in pyramid_list:
+				node_list = cubit.get_connectivity('pyramid',pyramid_id)
+				element_id += 1
+				fid.write(f"{name[0:4]}{DIM} {element_id} 0 {block_id} {node_list[0]} {node_list[1]} {node_list[2]} {node_list[3]} {node_list[4]} {node_list[4]} {node_list[4]} {node_list[4]}\n")
 
 		tri_list = cubit.get_block_tris(block_id)
 		for tri_id in tri_list:
@@ -483,6 +400,11 @@ def export_meg(cubit, FileName, DIM='T', MGR2=[]):
 			element_id += 1
 			fid.write(f"{name[0:4]}{DIM} {element_id} 0 {block_id} {node_list[0]} {node_list[1]}\n")
 
+		node_list = cubit.get_block_nodes(block_id)
+		for node_id in node_list:
+			element_id += 1
+			fid.write(f"{name[0:4]}{DIM} {element_id} 0 {block_id} {node_id}\n")
+
 	fid.write("* NODE\n")
 	for node_id in range(len(MGR2)):
 		fid.write(f"MGR2 {node_id+1} 0 {MGR2[node_id][0]} {MGR2[node_id][1]} {MGR2[node_id][2]}\n")
@@ -494,7 +416,7 @@ def export_meg(cubit, FileName, DIM='T', MGR2=[]):
 ###	vtk format
 ########################################################################
 
-def export_vtk(cubit, FileName):
+def export_vtk(cubit, FileName, ORDER="2nd"):
 
 	fid = open(FileName,'w')
 	fid.write('# vtk DataFile Version 3.0\n')
@@ -515,6 +437,7 @@ def export_vtk(cubit, FileName):
 	tri_list = set()
 	quad_list = set()
 	edge_list = set()
+	nodes_list = set()
 
 	for block_id in cubit.get_block_id_list():
 		tet_list.update(cubit.get_block_tets(block_id))
@@ -524,61 +447,112 @@ def export_vtk(cubit, FileName):
 		tri_list.update(cubit.get_block_tris(block_id))
 		quad_list.update(cubit.get_block_faces(block_id))
 		edge_list.update(cubit.get_block_edges(block_id))
+		nodes_list.update(cubit.get_block_nodes(block_id))
 
-	fid.write(f'CELLS {len(tet_list) + len(hex_list) + len(wedge_list) + len(pyramid_list) + len(tri_list) + len(quad_list) + len(edge_list)} {5*len(tet_list) + 9*len(hex_list) + 7*len(wedge_list) + 6*len(pyramid_list) + 4*len(tri_list) + 5*len(quad_list) + 3*len(edge_list)}\n' )
-	for tet_id in tet_list:
-		node_list = cubit.get_connectivity("tet", tet_id)
-		fid.write(f'4 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1}\n')
-	for hex_id in hex_list:
-		node_list = cubit.get_connectivity("hex", hex_id)
-		fid.write(f'8 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} {node_list[4]-1} {node_list[5]-1} {node_list[6]-1} {node_list[7]-1}\n')
-	for wedge_id in wedge_list:
-		node_list = cubit.get_connectivity("wedge", wedge_id)
-		fid.write(f'6 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} {node_list[4]-1} {node_list[5]-1} \n')
-	for pyramid_id in pyramid_list:
-		node_list = cubit.get_connectivity("pyramid", pyramid_id)
-		fid.write(f'5 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} {node_list[4]-1} \n')
-	for tri_id in tri_list:
-		node_list = cubit.get_connectivity("tri", tri_id)
-		fid.write(f'3 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} \n')
-	for quad_id in quad_list:
-		node_list = cubit.get_connectivity("quad", quad_id)
-		fid.write(f'4 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} \n')
-	for edge_id in edge_list:
-		node_list = cubit.get_connectivity("edge", edge_id)
-		fid.write(f'2 {node_list[0]-1} {node_list[1]-1} \n')
+	if ORDER=="2nd":
+		fid.write(f'CELLS {len(tet_list) + len(hex_list) + len(wedge_list) + len(pyramid_list) + len(tri_list) + len(quad_list) + len(edge_list) + len(nodes_list)} {11*len(tet_list) + 21*len(hex_list) + 16*len(wedge_list) + 14*len(pyramid_list) + 7*len(tri_list) + 9*len(quad_list) + 4*len(edge_list) + 2*len(nodes_list)}\n' )
+	else:
+		fid.write(f'CELLS {len(tet_list) + len(hex_list) + len(wedge_list) + len(pyramid_list) + len(tri_list) + len(quad_list) + len(edge_list) + len(nodes_list)} { 5*len(tet_list) +  9*len(hex_list) +  7*len(wedge_list) +  6*len(pyramid_list) + 4*len(tri_list) + 5*len(quad_list) + 3*len(edge_list) + 2*len(nodes_list)}\n' )
 
-	fid.write(f'CELL_TYPES {len(tet_list) + len(hex_list) + len(wedge_list) + len(pyramid_list) + len(tri_list) + len(quad_list) + len(edge_list)}\n')
 	for tet_id in tet_list:
-		fid.write('10\n')
+		node_list = cubit.get_expanded_connectivity("tet", tet_id)
+		if len(node_list)==4:
+			fid.write(f'4 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1}\n')
+		if len(node_list)==10:
+			fid.write(f'10 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} {node_list[4]-1} {node_list[5]-1} {node_list[6]-1} {node_list[7]-1} {node_list[8]-1} {node_list[9]-1}\n')
 	for hex_id in hex_list:
-		fid.write('12\n')
+		node_list = cubit.get_expanded_connectivity("hex", hex_id)
+		if len(node_list)==8:
+			fid.write(f'8 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} {node_list[4]-1} {node_list[5]-1} {node_list[6]-1} {node_list[7]-1}\n')
+		if len(node_list)==20:
+			fid.write(f'20 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} {node_list[4]-1} {node_list[5]-1} {node_list[6]-1} {node_list[7]-1} {node_list[8]-1} {node_list[9]-1} {node_list[10]-1} {node_list[11]-1} {node_list[16]-1} {node_list[17]-1} {node_list[18]-1} {node_list[19]-1} {node_list[12]-1} {node_list[13]-1} {node_list[14]-1} {node_list[15]-1}\n')
 	for wedge_id in wedge_list:
-		fid.write('13\n')
+		node_list = cubit.get_expanded_connectivity("wedge", wedge_id)
+		if len(node_list)==6:
+			fid.write(f'6 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} {node_list[4]-1} {node_list[5]-1} \n')
+		if len(node_list)==15:
+			fid.write(f'15 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} {node_list[4]-1} {node_list[5]-1} {node_list[6]-1} {node_list[7]-1} {node_list[8]-1} {node_list[12]-1} {node_list[13]-1} {node_list[14]-1} {node_list[9]-1} {node_list[10]-1} {node_list[11]-1} \n')
+
 	for pyramid_id in pyramid_list:
-		fid.write('14\n')
+		node_list = cubit.get_expanded_connectivity("pyramid", pyramid_id)
+		if len(node_list)==5:
+			fid.write(f'5 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} {node_list[4]-1} \n')
+		if len(node_list)==13:
+			fid.write(f'13 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} {node_list[4]-1} {node_list[5]-1} {node_list[6]-1} {node_list[7]-1} {node_list[8]-1} {node_list[9]-1} {node_list[10]-1} {node_list[11]-1} {node_list[12]-1} \n')
 	for tri_id in tri_list:
-		fid.write('5\n')
+		node_list = cubit.get_expanded_connectivity("tri", tri_id)
+		if len(node_list)==3:
+			fid.write(f'3 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} \n')
+		if len(node_list)==6:
+			fid.write(f'6 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} {node_list[4]-1} {node_list[5]-1} \n')
 	for quad_id in quad_list:
-		fid.write('9\n')
+		node_list = cubit.get_expanded_connectivity("quad", quad_id)
+		if len(node_list)==4:
+			fid.write(f'4 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} \n')
+		if len(node_list)==8:
+			fid.write(f'8 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} {node_list[3]-1} {node_list[4]-1} {node_list[5]-1} {node_list[6]-1} {node_list[7]-1}\n')
 	for edge_id in edge_list:
-		fid.write('3\n')
-	fid.write(f'CELL_DATA {len(tet_list) + len(hex_list) + len(wedge_list) + len(pyramid_list) + len(tri_list) + len(quad_list) + len(edge_list)}\n')
+		node_list = cubit.get_expanded_connectivity("edge", edge_id)
+		if len(node_list)==2:
+			fid.write(f'2 {node_list[0]-1} {node_list[1]-1} \n')
+		if len(node_list)==3:
+			fid.write(f'3 {node_list[0]-1} {node_list[1]-1} {node_list[2]-1} \n')
+	for node_id in nodes_list:
+		fid.write(f'1 {node_id-1} \n')
+
+	fid.write(f'CELL_TYPES {len(tet_list) + len(hex_list) + len(wedge_list) + len(pyramid_list) + len(tri_list) + len(quad_list) + len(edge_list) + len(nodes_list)}\n')
+	if ORDER=="2nd":
+		for tet_id in tet_list:
+			fid.write('24\n')
+		for hex_id in hex_list:
+			fid.write('25\n')
+		for wedge_id in wedge_list:
+			fid.write('26\n')
+		for pyramid_id in pyramid_list:
+			fid.write('27\n')
+		for tri_id in tri_list:
+			fid.write('22\n')
+		for quad_id in quad_list:
+			fid.write('23\n')
+		for edge_id in edge_list:
+			fid.write('21\n')
+		for node_id in nodes_list:
+			fid.write('1\n')
+	else:
+		for tet_id in tet_list:
+			fid.write('10\n')
+		for hex_id in hex_list:
+			fid.write('12\n')
+		for wedge_id in wedge_list:
+			fid.write('13\n')
+		for pyramid_id in pyramid_list:
+			fid.write('14\n')
+		for tri_id in tri_list:
+			fid.write('5\n')
+		for quad_id in quad_list:
+			fid.write('9\n')
+		for edge_id in edge_list:
+			fid.write('3\n')
+		for node_id in nodes_list:
+			fid.write('1\n')
+	fid.write(f'CELL_DATA {len(tet_list) + len(hex_list) + len(wedge_list) + len(pyramid_list) + len(tri_list) + len(quad_list) + len(edge_list) + len(nodes_list)}\n')
 	fid.write('SCALARS scalars float\n')
 	fid.write('LOOKUP_TABLE default\n')
 	for tet_id in tet_list:
-		fid.write(f'{4}\n')
+		fid.write('1\n')
 	for hex_id in hex_list:
-		fid.write(f'{8}\n')
+		fid.write('2\n')
 	for wedge_id in wedge_list:
-		fid.write(f'{6}\n')
+		fid.write('3\n')
 	for pyramid_id in pyramid_list:
-		fid.write(f'{5}\n')
+		fid.write('4\n')
 	for tri_id in tri_list:
-		fid.write(f'{3}\n')
+		fid.write('5\n')
 	for quad_id in quad_list:
-		fid.write(f'{4}\n')
+		fid.write('6\n')
 	for edge_id in edge_list:
-		fid.write(f'{2}\n')
+		fid.write('0\n')
+	for node_id in nodes_list:
+		fid.write('-1\n')
 	return cubit
 

coreform_cubit_mesh_export-1.1.2.dist-info/RECORD

@@ -1,9 +0,0 @@
-cubit_mesh_export.py,sha256=_xMcjTzjfKyGnllg3dksioYFHpnPyFai2q2h16leW24,24606
-sample_script/cubit_mesh_export_2d_geo_lukas_format.py,sha256=XJh9K2nMYkpqBgSzTNhxy1dn4ZLczw0lecCCNBGaTHk,2543
-sample_script/cubit_mesh_export_3d_cdb.py,sha256=Kw6ViJ7wXBjqL5FvglvtY0ozqc_liIPEcsUVNfLAdes,4925
-sample_script/cubit_mesh_export_3d_freefem_mesh.py,sha256=doBAxvORdAj5NJ07Z2POwz_bue-AtEqDMBUajV7mk8k,2095
-sample_script/cubit_mesh_export_3d_gmesh_v4.py,sha256=PVKQB0_RBhG3qEhUo6RbzjsH6ZDf23VSyHIxqfcF5QU,6947
-coreform_cubit_mesh_export-1.1.2.dist-info/METADATA,sha256=7ZV_kU9Sp4j80pVisfhBVdcLPGdzR0q34gsztjFF_7E,4955
-coreform_cubit_mesh_export-1.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-coreform_cubit_mesh_export-1.1.2.dist-info/top_level.txt,sha256=kKpvsJ-GNEjfO5z-ekTyHF0-sf9m62KYMGMsv4qLWJM,32
-coreform_cubit_mesh_export-1.1.2.dist-info/RECORD,,