ply2vtp.py

#! /usr/bin/env python
"""Generate a VTK style vtp file from a ply file

usage: ply2vtp.py mesh.ply mesh.vtp"""

__author__ = "Benjamin Barad"
__email__ = "benjamin.barad@gmail.com"
__license__ = "GPLv3"

import gc
import glob
import os

import click
import numpy as np
from pycurv import pycurv_io as io
from scipy.ndimage.morphology import distance_transform_edt
import vtk

@click.command()
@click.argument('input_ply', type=str)
@click.argument('output_vtp', type=str)
def convert_from_CLI(input_ply, output_vtp):
    """Click wrapper for convert script"""
    ply_to_vtp(input_ply, output_vtp)

def ply_to_vtp(plyfilename, vtpfilename):
    """Convert an input ply file to a vtp file"""
    print("open")
    plyfile = vtk.vtkPLYReader()
    plyfile.SetFileName(plyfilename)
    plyfile.Update()
    surf = plyfile.GetOutput() 
    writer = vtk.vtkXMLPolyDataWriter()
    fname = vtpfilename
    writer.SetFileName(fname)
    writer.SetInputData(surf)
    if writer.Write() != 1:
        raise pexceptions.PySegInputError(
            expr='save_vtp', msg='Error writing the file {}.'.format(fname))
    gc.collect()






if __name__ == '__main__':
    convert_from_CLI()