Как конвертировать VTK в STL в Python? - PullRequest
Новая
       28

Как конвертировать VTK в STL в Python?

0 голосов
/ 28 июня 2018

У меня есть около 500 файлов VTK, которые необходимо преобразовать в файл STL. Я обычно использую Paraview, чтобы конвертировать их вручную, но это занимает вечность. Интересно, есть ли инструмент для преобразования файлов vtk в файлы stl в Python? Я прикрепил скриншот моего файла VTK.

Заранее спасибо.

Мой файл VTK

1 Ответ

0 голосов
/ 17 июля 2018

Как насчет чего-то вроде следующего? Сохраните код как file_converter.py и запустите скрипт с аргументами: 

python file_converter.py -h
python file_converter.py <path-to-input-dir> -o <path-to-output-dir>

А вот и код для конвертера. 

#!/usr/bin/env python
import os
import vtk
import argparse

def convertFile(filepath, outdir):
    if not os.path.isdir(outdir):
        os.makedirs(outdir)
    if os.path.isfile(filepath):
        basename = os.path.basename(filepath)
        print("Copying file:", basename)
        basename = os.path.splitext(basename)[0]
        outfile = os.path.join(outdir, basename+".stl")
        reader = vtk.vtkGenericDataObjectReader()
        reader.SetFileName(filepath)
        reader.Update()
        writer = vtk.vtkSTLWriter()
        writer.SetInputConnection(reader.GetOutputPort())
        writer.SetFileName(outfile)
        return writer.Write()==1
    return False

def convertFiles(indir, outdir):
    files = os.listdir(indir)
    files = [ os.path.join(indir,f) for f in files if f.endswith('.vtk') ]
    ret = 0
    print("In:", indir)
    print("Out:", outdir)
    for f in files:
        ret += convertFile(f, outdir)
    print("Successfully converted %d out of %d files." % (ret, len(files)))

def run(args):
    convertFiles(args.indir, args.outdir)

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description="VTK to STL converter")
    parser.add_argument('indir', help="Path to input directory.")
    parser.add_argument('--outdir', '-o', default='output', help="Path to output directory.")
    parser.set_defaults(func=run)
    args = parser.parse_args()
    ret = args.func(args)
...