expose all io methods in the meshio namespace

Aquan1412 · Dec 6, 2019 · f7fd546 · f7fd546
1 parent edfd72e
commit f7fd546
Show file tree

Hide file tree

Showing 59 changed files with 341 additions and 332 deletions.
diff --git a/meshio/__init__.py b/meshio/__init__.py
@@ -1,20 +1,67 @@
-from . import _cli
+from . import (
+    _cli,
+    abaqus,
+    ansys,
+    cgns,
+    dolfin,
+    exodus,
+    flac3d,
+    gmsh,
+    h5m,
+    mdpa,
+    med,
+    medit,
+    nastran,
+    neuroglancer,
+    obj,
+    off,
+    permas,
+    ply,
+    stl,
+    svg,
+    tetgen,
+    vtk,
+    vtu,
+    wkt,
+    xdmf,
+)
 from .__about__ import __author__, __author_email__, __version__, __website__
 from ._exceptions import ReadError, WriteError
 from ._helpers import read, write, write_points_cells
 from ._mesh import Mesh
-from ._xdmf import XdmfTimeSeriesReader, XdmfTimeSeriesWriter
 
 __all__ = [
+    "abaqus",
+    "ansys",
+    "cgns",
+    "dolfin",
+    "exodus",
+    "flac3d",
+    "gmsh",
+    "h5m",
+    "mdpa",
+    "med",
+    "medit",
+    "nastran",
+    "neuroglancer",
+    "obj",
+    "off",
+    "permas",
+    "ply",
+    "stl",
+    "svg",
+    "tetgen",
+    "vtk",
+    "vtu",
+    "wkt",
+    "xdmf",
     "_cli",
     "read",
     "write",
     "write_points_cells",
     "Mesh",
     "ReadError",
     "WriteError",
-    "XdmfTimeSeriesReader",
-    "XdmfTimeSeriesWriter",
     "__version__",
     "__author__",
     "__author_email__",

diff --git a/meshio/_common.py b/meshio/_common.py
@@ -110,3 +110,50 @@ def write_xml(filename, root, pretty_print=False):
     tree = ET.ElementTree(root)
     tree.write(filename, pretty_print=pretty_print)
     return
+
+
+# https://www.vtk.org/doc/nightly/html/vtkCellType_8h_source.html
+vtk_to_meshio_type = {
+    0: "empty",
+    1: "vertex",
+    # 2: 'poly_vertex',
+    3: "line",
+    # 4: 'poly_line',
+    5: "triangle",
+    # 6: 'triangle_strip',
+    7: "polygon",
+    # 8: 'pixel',
+    9: "quad",
+    10: "tetra",
+    # 11: 'voxel',
+    12: "hexahedron",
+    13: "wedge",
+    14: "pyramid",
+    15: "penta_prism",
+    16: "hexa_prism",
+    21: "line3",
+    22: "triangle6",
+    23: "quad8",
+    24: "tetra10",
+    25: "hexahedron20",
+    26: "wedge15",
+    27: "pyramid13",
+    28: "quad9",
+    29: "hexahedron27",
+    30: "quad6",
+    31: "wedge12",
+    32: "wedge18",
+    33: "hexahedron24",
+    34: "triangle7",
+    35: "line4",
+    #
+    # 60: VTK_HIGHER_ORDER_EDGE,
+    # 61: VTK_HIGHER_ORDER_TRIANGLE,
+    # 62: VTK_HIGHER_ORDER_QUAD,
+    # 63: VTK_HIGHER_ORDER_POLYGON,
+    # 64: VTK_HIGHER_ORDER_TETRAHEDRON,
+    # 65: VTK_HIGHER_ORDER_WEDGE,
+    # 66: VTK_HIGHER_ORDER_PYRAMID,
+    # 67: VTK_HIGHER_ORDER_HEXAHEDRON,
+}
+meshio_to_vtk_type = {v: k for k, v in vtk_to_meshio_type.items()}
diff --git a/meshio/_helpers.py b/meshio/_helpers.py
@@ -3,30 +3,30 @@
 import numpy
 
 from . import (
-    _abaqus,
-    _ansys,
-    _cgns,
-    _dolfin,
-    _exodus,
-    _flac3d,
-    _gmsh,
-    _h5m,
-    _mdpa,
-    _med,
-    _medit,
-    _nastran,
-    _neuroglancer,
-    _obj,
-    _off,
-    _permas,
-    _ply,
-    _stl,
-    _svg,
-    _tetgen,
-    _vtk,
-    _vtu,
-    _wkt,
-    _xdmf,
+    abaqus,
+    ansys,
+    cgns,
+    dolfin,
+    exodus,
+    flac3d,
+    gmsh,
+    h5m,
+    mdpa,
+    med,
+    medit,
+    nastran,
+    neuroglancer,
+    obj,
+    off,
+    permas,
+    ply,
+    stl,
+    svg,
+    tetgen,
+    vtk,
+    vtu,
+    wkt,
+    xdmf,
 )
 from ._common import num_nodes_per_cell
 from ._exceptions import ReadError, WriteError
@@ -158,55 +158,55 @@ def read(filename, file_format=None):
     :returns mesh{2,3}d: The mesh data.
     """
     format_to_reader = {
-        "ansys": _ansys,
-        "ansys-ascii": _ansys,
-        "ansys-binary": _ansys,
-        "cgns": _cgns,
+        "ansys": ansys,
+        "ansys-ascii": ansys,
+        "ansys-binary": ansys,
+        "cgns": cgns,
         #
-        "gmsh": _gmsh,
-        "gmsh-ascii": _gmsh,
-        "gmsh-binary": _gmsh,
-        "gmsh2": _gmsh,
-        "gmsh2-ascii": _gmsh,
-        "gmsh2-binary": _gmsh,
-        "gmsh4": _gmsh,
-        "gmsh4-ascii": _gmsh,
-        "gmsh4-binary": _gmsh,
+        "gmsh": gmsh,
+        "gmsh-ascii": gmsh,
+        "gmsh-binary": gmsh,
+        "gmsh2": gmsh,
+        "gmsh2-ascii": gmsh,
+        "gmsh2-binary": gmsh,
+        "gmsh4": gmsh,
+        "gmsh4-ascii": gmsh,
+        "gmsh4-binary": gmsh,
         #
-        "flac3d": _flac3d,
-        "wkt": _wkt,
-        "med": _med,
-        "medit": _medit,
-        "nastran": _nastran,
-        "neuroglancer": _neuroglancer,
-        "dolfin-xml": _dolfin,
-        "permas": _permas,
-        "moab": _h5m,
-        "obj": _obj,
-        "off": _off,
+        "flac3d": flac3d,
+        "wkt": wkt,
+        "med": med,
+        "medit": medit,
+        "nastran": nastran,
+        "neuroglancer": neuroglancer,
+        "dolfin-xml": dolfin,
+        "permas": permas,
+        "moab": h5m,
+        "obj": obj,
+        "off": off,
         #
-        "ply": _ply,
-        "ply-ascii": _ply,
-        "ply-binary": _ply,
+        "ply": ply,
+        "ply-ascii": ply,
+        "ply-binary": ply,
         #
-        "stl": _stl,
-        "stl-ascii": _stl,
-        "stl-binary": _stl,
+        "stl": stl,
+        "stl-ascii": stl,
+        "stl-binary": stl,
         #
-        "tetgen": _tetgen,
+        "tetgen": tetgen,
         #
-        "vtu-ascii": _vtu,
-        "vtu-binary": _vtu,
+        "vtu-ascii": vtu,
+        "vtu-binary": vtu,
         #
-        "vtk-ascii": _vtk,
-        "vtk-binary": _vtk,
+        "vtk-ascii": vtk,
+        "vtk-binary": vtk,
         #
-        "xdmf": _xdmf,
-        "exodus": _exodus,
+        "xdmf": xdmf,
+        "exodus": exodus,
         #
-        "abaqus": _abaqus,
+        "abaqus": abaqus,
         #
-        "mdpa": _mdpa,
+        "mdpa": mdpa,
     }
 
     if is_buffer(filename, "r"):
@@ -305,45 +305,45 @@ def write(filename, mesh, file_format=None, **kwargs):
 
 
 _writer_map = {
-    "moab": (_h5m, (), {}),
-    "ansys-ascii": (_ansys, (), {"binary": False}),
-    "ansys-binary": (_ansys, (), {"binary": True}),
-    "wkt": (_wkt, (), {}),
-    "gmsh2-ascii": (_gmsh, ("2",), {"binary": False}),
-    "gmsh2-binary": (_gmsh, ("2",), {"binary": True}),
-    "gmsh4-ascii": (_gmsh, ("4",), {"binary": False}),
-    "gmsh4-binary": (_gmsh, ("4",), {"binary": True}),
-    "med": (_med, (), {}),
-    "medit": (_medit, (), {}),
-    "dolfin-xml": (_dolfin, (), {}),
-    "neuroglancer": (_neuroglancer, (), {}),
-    "obj": (_obj, (), {}),
-    "off": (_off, (), {}),
-    "permas": (_permas, (), {}),
-    "ply-ascii": (_ply, (), {"binary": False}),
-    "ply-binary": (_ply, (), {"binary": True}),
-    "stl-ascii": (_stl, (), {"binary": False}),
-    "stl-binary": (_stl, (), {"binary": True}),
-    "tetgen": (_tetgen, (), {}),
-    "vtu-ascii": (_vtu, (), {"binary": False}),
-    "vtu": (_vtu, (), {"binary": True}),
-    "vtu-binary": (_vtu, (), {"binary": True}),
-    "vtk-ascii": (_vtk, (), {"binary": False}),
-    "vtk": (_vtk, (), {"binary": True}),
-    "vtk-binary": (_vtk, (), {"binary": True}),
-    "xdmf": (_xdmf, (), {}),
-    "xdmf3": (_xdmf, (), {}),
-    "xdmf-binary": (_xdmf, (), {"data_format": "Binary"}),
-    "xdmf3-binary": (_xdmf, (), {"data_format": "Binary"}),
-    "xdmf-hdf": (_xdmf, (), {"data_format": "HDF"}),
-    "xdmf3-hdf": (_xdmf, (), {"data_format": "HDF"}),
-    "xdmf-xml": (_xdmf, (), {"data_format": "XML"}),
-    "xdmf3-xml": (_xdmf, (), {"data_format": "XML"}),
-    "abaqus": (_abaqus, (), {}),
-    "exodus": (_exodus, (), {}),
-    "mdpa": (_mdpa, (), {}),
-    "svg": (_svg, (), {}),
-    "nastran": (_nastran, (), {}),
-    "flac3d": (_flac3d, (), {}),
-    "cgns": (_cgns, (), {}),
+    "moab": (h5m, (), {}),
+    "ansys-ascii": (ansys, (), {"binary": False}),
+    "ansys-binary": (ansys, (), {"binary": True}),
+    "wkt": (wkt, (), {}),
+    "gmsh2-ascii": (gmsh, ("2",), {"binary": False}),
+    "gmsh2-binary": (gmsh, ("2",), {"binary": True}),
+    "gmsh4-ascii": (gmsh, ("4",), {"binary": False}),
+    "gmsh4-binary": (gmsh, ("4",), {"binary": True}),
+    "med": (med, (), {}),
+    "medit": (medit, (), {}),
+    "dolfin-xml": (dolfin, (), {}),
+    "neuroglancer": (neuroglancer, (), {}),
+    "obj": (obj, (), {}),
+    "off": (off, (), {}),
+    "permas": (permas, (), {}),
+    "ply-ascii": (ply, (), {"binary": False}),
+    "ply-binary": (ply, (), {"binary": True}),
+    "stl-ascii": (stl, (), {"binary": False}),
+    "stl-binary": (stl, (), {"binary": True}),
+    "tetgen": (tetgen, (), {}),
+    "vtu-ascii": (vtu, (), {"binary": False}),
+    "vtu": (vtu, (), {"binary": True}),
+    "vtu-binary": (vtu, (), {"binary": True}),
+    "vtk-ascii": (vtk, (), {"binary": False}),
+    "vtk": (vtk, (), {"binary": True}),
+    "vtk-binary": (vtk, (), {"binary": True}),
+    "xdmf": (xdmf, (), {}),
+    "xdmf3": (xdmf, (), {}),
+    "xdmf-binary": (xdmf, (), {"data_format": "Binary"}),
+    "xdmf3-binary": (xdmf, (), {"data_format": "Binary"}),
+    "xdmf-hdf": (xdmf, (), {"data_format": "HDF"}),
+    "xdmf3-hdf": (xdmf, (), {"data_format": "HDF"}),
+    "xdmf-xml": (xdmf, (), {"data_format": "XML"}),
+    "xdmf3-xml": (xdmf, (), {"data_format": "XML"}),
+    "abaqus": (abaqus, (), {}),
+    "exodus": (exodus, (), {}),
+    "mdpa": (mdpa, (), {}),
+    "svg": (svg, (), {}),
+    "nastran": (nastran, (), {}),
+    "flac3d": (flac3d, (), {}),
+    "cgns": (cgns, (), {}),
 }
diff --git a/meshio/_xdmf/__init__.py b/meshio/_xdmf/__init__.py
diff --git a/meshio/_abaqus.py → meshio/abaqus/_abaqus.py b/meshio/_abaqus.py → meshio/abaqus/_abaqus.py
@@ -3,10 +3,10 @@
 """
 import numpy
 
-from .__about__ import __version__
-from ._exceptions import ReadError
-from ._files import open_file
-from ._mesh import Mesh
+from ..__about__ import __version__
+from .._exceptions import ReadError
+from .._files import open_file
+from .._mesh import Mesh
 
 abaqus_to_meshio_type = {
     # trusses

diff --git a/meshio/_ansys.py → meshio/ansys/_ansys.py b/meshio/_ansys.py → meshio/ansys/_ansys.py
@@ -7,10 +7,10 @@
 
 import numpy
 
-from .__about__ import __version__
-from ._exceptions import ReadError, WriteError
-from ._files import open_file
-from ._mesh import Mesh
+from ..__about__ import __version__
+from .._exceptions import ReadError, WriteError
+from .._files import open_file
+from .._mesh import Mesh
 
 
 def _skip_to(f, char):