Source code for polpo.preprocessing.mesh._meshio

import os

import meshio

from polpo.preprocessing.base import PreprocessingStep




class MeshioReader(PreprocessingStep):


    def apply(self, path):
        return meshio.read(path)






class MeshioWriter(PreprocessingStep):
    def __init__(self, ext=None, dirname="", **kwargs):
        self.dirname = dirname
        self.ext = ext
        self.kwargs = kwargs



    def apply(self, data):
        # TODO: maybe do opposite instead (give preference to arguments)
        # filename extension ignored if ext is not None
        filename, mesh = data
        if self.ext is not None:
            if "." in filename:
                filename = filename.split(".")[0]
            filename += f".{self.ext}"

        path = os.path.join(self.dirname, filename)
        mesh.write(path, **self.kwargs)

        return path