Module simple_3dviz.io

Provide a common interface for loading mesh files or other useful files.

"""Provide a common interface for loading mesh files or other useful files."""

from os import path

from .mesh import ObjMeshReader, OffMeshReader, PlyMeshReader, StlMeshReader
from .material import MtlMaterialReader


def read_mesh_file(filename, ext=None):
    if isinstance(filename, str):
        _, ext = path.splitext(filename)
    try:
        return {
            ".ply": PlyMeshReader,
            ".obj": ObjMeshReader,
            ".off": OffMeshReader,
            ".stl": StlMeshReader
        }[ext.lower()](filename)
    except KeyError as e:
        raise ValueError("{} mesh file extension is not supported".format(ext))


def read_material_file(filename, ext=None):
    if isinstance(filename, str):
        _, ext = path.splitext(filename)
    try:
        return {
            ".mtl": MtlMaterialReader
        }[ext.lower()](filename)
    except KeyError as e:
        raise ValueError("{} material file extension is not supported".format(ext))

Sub-modules

simple_3dviz.io.material

Load a material file into ambient, diffuse and specular colors and a set of texture maps.

simple_3dviz.io.mesh

Load a mesh into three arrays vertices, faces and vertex_colors.

simple_3dviz.io.utils

simple_3dviz.io.voxels

Functions

def read_material_file(filename, ext=None)

Expand source code

def read_material_file(filename, ext=None):
    if isinstance(filename, str):
        _, ext = path.splitext(filename)
    try:
        return {
            ".mtl": MtlMaterialReader
        }[ext.lower()](filename)
    except KeyError as e:
        raise ValueError("{} material file extension is not supported".format(ext))

def read_mesh_file(filename, ext=None)

Expand source code

def read_mesh_file(filename, ext=None):
    if isinstance(filename, str):
        _, ext = path.splitext(filename)
    try:
        return {
            ".ply": PlyMeshReader,
            ".obj": ObjMeshReader,
            ".off": OffMeshReader,
            ".stl": StlMeshReader
        }[ext.lower()](filename)
    except KeyError as e:
        raise ValueError("{} mesh file extension is not supported".format(ext))