However, there `cubit.get_block_hexes(block_id)` does not return anything other than an empty set when these sets of commands were last tested (`Cubit v13.2`).
By contrast the script verified to work with `Cubit v15.2` includes this logic as the bug has been corrected in later versions of the software.
-The following version of the above script, instead, produces a VTK file:
-```python
-#!python
-#!python
-# This script will output whatever mesh you have currently in CUBIT
-# in the VTK ascii format
-#
-# In this script:
-# - block id => material id
-# - sideset id => boundary id
-# - nodeset id => manifold id
-import cubit
-
-vtkfile = '/tmp/output.vtk'
-# def exportVTK(vtkfile):
-if True:
- outfile = open(vtkfile,"w")
- # ============================================================
- # Global dictionaries
- # ============================================================
- node_ids = {}
- #
- hex_material_ids = {}
- hex_manifold_ids = {}
- #
- quad_material_ids = {}
- quad_manifold_ids = {}
- #
- edge_material_ids = {}
- edge_manifold_ids = {}
- #
- n_nodes = 0
- n_hexes = 0
- n_quads = 0
- n_edges = 0
- #
- block_ids = cubit.get_block_id_list()
- nodeset_ids = cubit.get_nodeset_id_list()
- sideset_ids = cubit.get_sideset_id_list()
- #
- # ============================================================
- # Collect nodes
- # ============================================================
- group_id = cubit.get_id_from_name("temp_nodes")
- if group_id != 0:
- cubit.cmd("delete group " + str(group_id))
- cubit.cmd("group 'temp_nodes' add node all")
- group_id = cubit.get_id_from_name("temp_nodes")
- node_list = cubit.get_group_nodes(group_id)
- cubit.cmd("delete group " + str(group_id))
- n_nodes = len(node_list)
- #
- i = 0
- for node in node_list:
- node_ids[node] = i
- i +=1
- #
- # ============================================================
- # Collect all hexes for 3d grids
- # set default material and manifold ids
- # ============================================================
- group_id = cubit.get_id_from_name("temp_hexes")
- if group_id != 0:
- cubit.cmd("delete group " + str(group_id))
- cubit.cmd("group 'temp_hexes' add hex all")
- group_id = cubit.get_id_from_name("temp_hexes")
- hex_list = cubit.get_group_hexes(group_id)
- cubit.cmd("delete group " + str(group_id))
- n_hexes = len(hex_list)
- #
- for hex in hex_list:
- hex_material_ids[hex] = 0
- hex_manifold_ids[hex] = -1
- #
- # ============================================================
- # Collect hexes by block id ( => Material ID)
- # ============================================================
- for block_id in block_ids:
- volumes = cubit.get_block_volumes(block_id)
- for vol in volumes:
- hexes = cubit.get_volume_hexes(vol)
- print len(hexes), 'hexes in volume', vol
- for hex in hexes:
- hex_material_ids[hex] = block_id
- #
- # ============================================================
- # Collect hexes by nodeset id ( => Manifold ID)
- # ============================================================
- for nodeset_id in nodeset_ids:
- volumes = cubit.get_nodeset_volumes(nodeset_id)
- for vol in volumes:
- hexes = cubit.get_volume_hexes(vol)
- for hex in hexes:
- hex_manifold_ids[hex] = nodeset_id
- #
- # ============================================================
- # Collect all quads (for 2d grids)
- # set default material and manifold ids
- # ============================================================
- if n_hexes == 0:
- surface_list = ()
- group_id = cubit.get_id_from_name("temp_surfs")
- if group_id != 0:
- cubit.cmd("delete group " + str(group_id))
- cubit.cmd("group 'temp_surfs' add surf all")
- group_id = cubit.get_id_from_name("temp_surfs")
- surface_list = cubit.get_group_surfaces(group_id)
- cubit.cmd("delete group " + str(group_id))
- for surface_id in surface_list:
- quads = cubit.get_surface_quads(surface_id)
- for quad in quads:
- quad_material_ids[quad] = 0
- quad_manifold_ids[quad] = -1
- #
- n_quads = len(quad_material_ids)
- #
- # ============================================================
- # Collect quads by block id for 2D grids ( => Material ID)
- # and by sideset_id for 3d grids ( => Material ID == Boundary ID)
- # ============================================================
- if n_hexes == 0:
- for block_id in block_ids:
- surfaces = cubit.get_block_surfaces(block_id)
- for surface_id in surfaces:
- quads = cubit.get_surface_quads(surface_id)
- quad_material_ids[quad] = block_id
- else:
- for sideset_id in sideset_ids:
- surfaces = cubit.get_sideset_surfaces(sideset_id)
- for surface_id in surfaces:
- quads = cubit.get_surface_quads(surface_id)
- for quad in quads:
- quad_material_ids[quad] = sideset_id
- quad_manifold_ids[quad] = -1
- #
- # ============================================================
- # Collect quads by nodeset id ( => Manifold ID)
- # ============================================================
- for nodeset_id in nodeset_ids:
- surfaces = cubit.get_nodeset_surfaces(nodeset_id)
- for surface_id in surfaces:
- quads = cubit.get_surface_quads(surface_id)
- quad_manifold_ids[quad] = nodeset_id
- if quad not in quad_material_ids:
- # this can only happen in 3d, so this must be an internal
- # face
- quad_material_ids[quad] = -1
- #
- n_quads = len(quad_material_ids)
- #
- # ============================================================
- # Collect edges by sideset id ( => Material ID)
- # ============================================================
- for sideset_id in sideset_ids:
- curves = cubit.get_sideset_curves(sideset_id)
- for curve in curves:
- # edges = get_curve_edges(curve)
- # the above does not work. Get around this bug
- group_id = cubit.get_id_from_name("temp_edges")
- if group_id != 0:
- cubit.cmd("delete group " + str(group_id))
- cubit.cmd("group 'temp_edges' add edge all in curve " + str(curve))
- group_id = cubit.get_id_from_name("temp_edges")
- edges = cubit.get_group_edges(group_id)
- cubit.cmd("delete group " + str(group_id))
- for edge in edges:
- edge_material_ids[edge] = sideset_id
- edge_manifold_ids[edge] = -1
- #
- # ============================================================
- # Collect edges by nodeset id ( => Manifold ID)
- # ============================================================
- for nodeset_id in nodeset_ids:
- curves = cubit.get_nodeset_curves(nodeset_id)
- for curve in curves:
- # edges = get_curve_edges(curve)
- # the above does not work. Get around this bug
- group_id = cubit.get_id_from_name("temp_edges")
- if group_id != 0:
- cubit.cmd("delete group " + str(group_id))
- cubit.cmd("group 'temp_edges' add edge all in curve " + str(curve))
- group_id = cubit.get_id_from_name("temp_edges")
- edges = cubit.get_group_edges(group_id)
- cubit.cmd("delete group " + str(group_id))
- for edge in edges:
- edge_manifold_ids[edge] = nodeset_id
- if edge not in edge_material_ids:
- edge_material_ids[edge] = -1
- #
- n_edges = len(edge_material_ids)
- #
- # ============================================================
- # Now we write the header.
- # ============================================================
- outfile.write("# vtk DataFile Version 3.0\n")
- outfile.write("File generated with Trelis\nASCII\n")
- outfile.write("DATASET UNSTRUCTURED_GRID\nPOINTS " + str(n_nodes) + " double\n")
- #
- n_elements = n_hexes + n_quads + n_edges
- #
- # ============================================================
- # The node list.
- # ============================================================
- for node_num in node_list:
- node_coord = cubit.get_nodal_coordinates(node_num)
- if abs(node_coord[0])<1e-15:
- outfile.write("0 ")
- else :
- outfile.write(str(node_coord[0]) + " ")
- if abs(node_coord[1])<1e-15:
- outfile.write("0 ")
- else :
- outfile.write(str(node_coord[1]) + " ")
- if abs(node_coord[2])<1e-15:
- outfile.write("0")
- else :
- outfile.write(str(node_coord[2]))
- outfile.write("\n")
- #
- # ============================================================
- # Now we write the for the CELLS.
- # ============================================================
- storage = n_hexes*9 + n_quads*5 + n_edges*3
- outfile.write("CELLS " + str(n_elements) + " " + str(storage) + "\n")
- #
- # ============================================================
- # The hex list
- # ============================================================
- for hex in hex_manifold_ids:
- outfile.write("8")
- n = cubit.get_connectivity("Hex", hex)
- i = 0
- while i<8:
- outfile.write(" " + str(node_ids[n[i]]))
- i += 1
- outfile.write("\n")
- #
- #
- # ============================================================
- # The quad list
- # ============================================================
- for quad in quad_manifold_ids:
- outfile.write("4")
- n = cubit.get_connectivity("Quad", quad)
- i = 0
- while i<4:
- outfile.write(" " + str(node_ids[n[i]]))
- i += 1
- outfile.write("\n")
- #
- # ============================================================
- # The edge list
- # ============================================================
- for edge in edge_manifold_ids:
- outfile.write("2")
- n = cubit.get_connectivity("Edge", edge)
- i = 0
- while i<2:
- outfile.write(" " + str(node_ids[n[i]]))
- i += 1
- outfile.write("\n")
- #
- # ============================================================
- # The CELL_TYPES section
- # ============================================================
- cell_types = "12 "* n_hexes + "9 "*n_quads + "3 "*n_edges+"\n"
- outfile.write("CELL_TYPES " + str(n_elements) + "\n" + cell_types)
- #
- # ============================================================
- # The DATA_SET set
- # ============================================================
- outfile.write("CELL_DATA " + str(n_elements) +"\n")
- outfile.write("SCALARS MaterialID int 1\nLOOKUP_TABLE default\n")
- for hex in hex_material_ids:
- outfile.write(str(hex_material_ids[hex]) + " ")
- for quad in quad_material_ids:
- outfile.write(str(quad_material_ids[quad]) + " ")
- for edge in edge_material_ids:
- outfile.write(str(edge_material_ids[edge]) + " ")
- outfile.write("\n")
- #
- outfile.write("SCALARS ManifoldID int 1\nLOOKUP_TABLE default\n")
- for hex in hex_manifold_ids:
- outfile.write(str(hex_manifold_ids[hex]) + " ")
- for quad in quad_manifold_ids:
- outfile.write(str(quad_manifold_ids[quad]) + " ")
- for edge in edge_manifold_ids:
- outfile.write(str(edge_manifold_ids[edge]) + " ")
- outfile.write("\n")
- #
- print n_nodes, "nodes"
- print n_hexes, "hexes"
- print n_quads, "quads"
- print n_edges, "edges"
-```
-In principle, it should be possible to add a shortcut in a Custom Toolbar to load the script into memory. Then when you are
+Another version of the above script is available that produces a VTK file. You can find it here:
+
+https://gist.github.com/luca-heltai/b127a69b86b4b140c87a01110053f751
+
+You can add a shortcut in a Custom Toolbar to load the script into memory. Then when you are
ready to export, you can type the following command in the script tab:
saveVTK("filename.vtk")
\ No newline at end of file
projects / dealii.wiki.git / commitdiff